hub / github.com/VITA-Group/TransGAN / train

Function train

functions.py:70–241 · view source on GitHub ↗

(args, gen_net: nn.Module, dis_net: nn.Module, gen_optimizer, dis_optimizer, gen_avg_param, train_loader,
          epoch, writer_dict, fixed_z, schedulers=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

main_workerFunction · 0.90

Calls 6

tqdmFunction · 0.85

compute_gradient_penaltyFunction · 0.85

meanMethod · 0.80

writeMethod · 0.80

backwardMethod · 0.45

stepMethod · 0.45

Tested by

no test coverage detected