MCPcopy Create free account
hub / github.com/MTLab/MorphMLP / train_epoch

Function train_epoch

tools/train_net.py:28–185  ·  view source on GitHub ↗

Perform the video training for one epoch. Args: train_loader (loader): video training loader. model (model): the video model to train. optimizer (optim): the optimizer to perform optimization on the model's parameters. loss_scaler (scaler): scaler

(
    train_loader, model, optimizer, loss_scaler, train_meter, cur_epoch, cfg, writer=None
)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

trainFunction · 0.85

Calls 9

MixUpClass · 0.90
add_scalarsMethod · 0.80
iter_ticMethod · 0.45
data_tocMethod · 0.45
update_statsMethod · 0.45
iter_tocMethod · 0.45
log_iter_statsMethod · 0.45
log_epoch_statsMethod · 0.45
resetMethod · 0.45

Tested by

no test coverage detected