7d8f3f66495979f1bbd27205d422d673991709f2,GANs/stylegan2-distillation/main.py,,,#,31

Before Change

    nn.set_default_context(ctx)
          
    di = get_data_iterator(args.data_root, comm, config.train.batch_size, config.model.base_image_shape, 
                            img_exts=["_o.png", "_y.png"])
    
    trainer = Trainer(config.train, config.model, comm, di)
    

After Change

    if args.load_path:
        nn.load_parameters(args.load_path)

    trainer = Trainer(config.train, config.model, comm, di, face_morph=args.face_morph)
    
    trainer.train()

    if comm.rank == 0:

Instances