Function train

animals/scripts/main_animal3d.py:45–46

(opt, actions, train_loader, model, optimizer, epoch)

    CFM = get_model(args.model_type)
     
def train(opt, actions, train_loader, model, optimizer, epoch):
    return step('train', opt, actions, train_loader, model, optimizer, epoch)

def val(opt, actions, val_loader, model, steps=None):
    with torch.no_grad():