model/train.py [1221:1234]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                save_checkpoint(
                    args,
                    model,
                    optimizer,
                    dis_optimizer,
                    gen_optimizer,
                    dataset.vocab,
                    train_step,
                    val_nll,
                    scheduler,
                    dis_scheduler,
                    gen_scheduler,
                    name,
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



model/train.py [1239:1252]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                save_checkpoint(
                    args,
                    model,
                    optimizer,
                    dis_optimizer,
                    gen_optimizer,
                    dataset.vocab,
                    train_step,
                    val_nll,
                    scheduler,
                    dis_scheduler,
                    gen_scheduler,
                    name,
                )
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -