cast_input_to_torch(self, x, volatile=False)
cast_target_to_torch
cast_target_to_torch(self, y, volatile=False)
cast_output_to_numpy
cast_output_to_numpy(self, preds)
forward
forward(self)
train_on_batch
train_on_batch(self, x, target, optimizer, loss_fn, metrics=())
validate_on_batch
validate_on_batch(self, x, target, metrics)
predict_on_batch
predict_on_batch(self, x)
fit_generator
fit_generator(self, generator, steps_per_epoch, epochs, optimizer, loss_fn, validation_generator=None, validation_steps=0, metrics=(), callbacks=(), initial_epoch=0)
validate_generator
validate_generator(self, val_generator, validation_steps, loss_fn=None, metrics=())
predict_generator
predict_generator(self, generator, prediction_steps, verbose=0)
load_state
load_state(self, load_path)
save_state
save_state(self, save_path)