torchctr.trainer package

Submodules

torchctr.trainer.trainer module

class torchctr.trainer.trainer.Trainer(model, dataset, param={})[source]

Bases: object

build_param(param)[source]
build_trainer()[source]
save_model(file_fullpath)[source]
train(dashboard_address=None)[source]
train_step(epoch)[source]
valid_step()[source]

Module contents

class torchctr.trainer.Trainer(model, dataset, param={})[source]

Bases: object

build_param(param)[source]
build_trainer()[source]
save_model(file_fullpath)[source]
train(dashboard_address=None)[source]
train_step(epoch)[source]
valid_step()[source]