6324799d51a451995a91b76306eae40ccd11f55e,eval.py,,main,#Any#,120

Before Change

    nets = (net_encoder, net_decoder, crit)
    for net in nets:
        net.cuda()

    // Main loop
    evaluate(nets, loader_val, args)

After Change

def main(args):
    torch.cuda.set_device(args.gpu_id)
    
    // Network Builders
    builder = ModelBuilder()
    net_encoder = builder.build_encoder(arch=args.arch_encoder,

Instances