jactorch.quickstart.train#

Classes

Functions

simple_fit(model, loss_function, dataset, ...)

Class ModelTrainer

class ModelTrainer[source]#

Bases: object

__init__(model, optimizer, lr=0.01, weight_decay=0, **opt_kwargs)[source]#
__new__(**kwargs)#
train(data_loader, nr_epochs, verbose=True, meters=None, early_stop=None, print_interval=1)[source]#
train_epoch(data_loader, meters=None)[source]#
train_step(feed_dict, meters=None)[source]#
validate(data_loader, metric, meters=None)[source]#
validate_step(feed_dict, metric, meters=None)[source]#

Functions

simple_fit(model, loss_function, dataset, optimizer, epochs, lr=0.01, weight_decay=0, print_interval=1, batch_size=None, **opt_kwargs)[source]#