model/train.py [1221:1228]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                save_checkpoint(
                    args,
                    model,
                    optimizer,
                    dis_optimizer,
                    gen_optimizer,
                    dataset.vocab,
                    train_step,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



model/train.py [1259:1266]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
                    save_checkpoint(
                        args,
                        model,
                        optimizer,
                        dis_optimizer,
                        gen_optimizer,
                        dataset.vocab,
                        train_step,
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -