Skip to content

Latest commit

 

History

History
6 lines (6 loc) · 313 Bytes

training.md

File metadata and controls

6 lines (6 loc) · 313 Bytes

Training the model

The command to train the network is as follows

python train.py --dataset Dataset_name --cuda --batch_size BatchSize --seed RandomSeed --alpha Alpha --g_lr G_lr --d_lr D_lr --lr_policy Policy --lr_decay_iters Iters

Note that those options should be changed according to your settings.