`flame.oneflow.trainer`#

Submodules#

Package Contents#

Classes#

class flame.oneflow.trainer.BaseTrainer(args)#

Parameters:: args (flame.oneflow.arguments.BaseArgs) –

T#

to_device(x)#

Parameters:: x (T) –
Return type:: T

run(max_epochs)#

Parameters:: max_epochs (int) –

train(loader, prefix='train')#

Parameters:: prefix (str) –

validate(loader, prefix='val')#

Parameters:: prefix (str) –

test(loader, prefix='test')#

Parameters:: prefix (str) –

_disable_tqdm()#

Return type:: bool

epoch_range(max_epochs)#

Parameters:: max_epochs (int) –

progress_meter(prefix)#

Parameters:: prefix (str) –
Return type:: flame.oneflow.trainer.progress_meter.ProgressMeter

set_sampler_epoch(loader)#

Parameters:: loader (oneflow.utils.data.dataloader.DataLoader) –

property name#

Return type:: str

property module_name#

lib.xxx_trainer -> xxx_trainer

Return type:: str