hub / github.com/OpenGVLab/InternImage / train_one_epoch

Function train_one_epoch

classification/main.py:382–521 · view source on GitHub ↗

(config,
                    model,
                    criterion,
                    data_loader,
                    optimizer,
                    epoch,
                    mixup_fn,
                    lr_scheduler,
                    amp_autocast=suppress,
                    loss_scaler=None,
                    model_ema=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

mainFunction · 0.85

Calls 8

updateMethod · 0.95

MyAverageMeterClass · 0.90

get_grad_normFunction · 0.90

obsolete_torch_versionFunction · 0.85

stepMethod · 0.80

trainMethod · 0.45

backwardMethod · 0.45

infoMethod · 0.45

Tested by

no test coverage detected