src.train package

Subpackages

• src.train.data namespace
  • Submodules
  • src.train.data.dataloader module
  • src.train.data.dataset module
  • src.train.data.mytransforms module
• src.train.utils namespace
  • Submodules
  • src.train.utils.common module
  • src.train.utils.dist_utils module
  • src.train.utils.factory module
  • src.train.utils.loss module
  • src.train.utils.metrics module

Submodules

src.train.train module

src.train.train.calc_loss(loss_dict, results, logger, global_step)

src.train.train.inference(net, data_label, use_aux)

src.train.train.main()

src.train.train.resolve_val_data(results, use_aux)

src.train.train.train(net, data_loader, loss_dict, optimizer, scheduler, logger, epoch, metric_dict, use_aux)

Module contents