fix(train): discriminator B aliased to discriminator A in (+1 more)

[andrewwhitecdw]

Small fixes in train.py:

fix: discriminator B aliased to discriminator A in DataParallel

Fix: Replace:

if num_gpu>1:
    print('Now you are using %d gpus.'%num_gpu)
    trainer.dis_a = torch.nn.DataParallel(trainer.dis_a, gpu_ids)
    trainer.dis_b = trainer.dis_a
    trainer = torch.nn.DataParallel(trainer, gpu_ids)

with:

if num_gpu>1:
    print('Now you are using %d gpus.'%num_gpu)
    trainer.dis_a = torch.nn.DataParallel(trainer.dis_a, gpu_ids)
    trainer.dis_b = torch.nn.DataParallel(trainer.dis_b, gpu_ids)
    trainer = torch.nn.DataParallel(trainer, gpu_ids)

fix: random.seed(7) only seeds numpy, not torch RNG

Fix: Replace:

random.seed(7) #fix random result

with:

random.seed(7) #fix random result
torch.manual_seed(7)

Files changed

• train.py

[andrewwhitecdw]

Closing this sweep-generated PR: sole commit is missing a valid Signed-off-by trailer. It does not meet the sweep requirements (single signed-off commit).