def main()

in source/train.py [0:0]

• 9 lines of code
• 4 McCabe index (conditional complexity)

def main(sm_args, world):
    
    cfg = _setup(sm_args)
    
    is_zero_rank = comm.get_local_rank()==0
    
    trainer = CocoTrainer(cfg)
    resume = True if sm_args.resume == "True" else False
    trainer.resume_or_load(resume=resume)
    trainer.train()
    
    if world["is_master"] and is_zero_rank:
        _save_model()