61421e8d341312e02dff23b46acde4261808dab3,train.py,,,#,15

Before Change

                train_batch[key] = train_batch[key].cuda()
        
        // forward pass to encoder
        img = Variable(train_batch["img_feat"])
        ques = Variable(train_batch["ques_fwd"])
        hist = Variable(train_batch["hist"])
        enc_out = encoder(img, ques, hist)

        // forward pass to decoder
        options = Variable(train_batch["opt"])
        ans_ind = Variable(train_batch["ans_ind"])

After Change

// ----------------------------------------------------------------------------

args = parser.parse_args()
for arg in vars(args):
    print("{:<20}: {}".format(arg, getattr(args, arg)))

// seed for reproducibility
torch.manual_seed(1234)

// set device and default tensor type
if args.gpuid >= 0:

Instances