train.py File Reference

Go to the source code of this file.

Namespaces
namespace	smrunner
namespace	smrunner.train

Functions
	smrunner.train.adjust_learning_rate (optimizer, epoch, lr, adjust_epoch, adjust_rate)
	smrunner.train.tmp_pipeline (sample, keys, device)
	smrunner.train.train_sample (model, loss_cls, optimizer, sample)
	smrunner.train.train_epoch (model, loss_cls, optimizer, data_loader, collect_keys, device)
	smrunner.train.validation_sample (model, loss_cls, sample, metric_args)
	smrunner.train.validation (model, loss_cls, data_loader, collect_keys, device, metric_args)
	smrunner.train.train (cfg, device_ids)