f6415170f7f90124e9cdb9ccd37cd867852b7657,train.py,,,#,19

Before Change

            for key in batch:
                batch[key] = Variable(batch[key].cuda())
        
        enc_out = encoder(batch["img_feat"], batch["ques_fwd"], batch["hist"])
        dec_out = decoder(enc_out, batch["opt"])
        cur_loss = criterion(dec_out, batch["ans_ind"].view(-1))
        cur_loss.backward()

After Change

        if args.gpuid >= 0:
            for key in batch:
                if not isinstance(batch[key], list):
                    batch[key] = Variable(batch[key].cuda())

        enc_out = encoder(batch)
        dec_out = decoder(enc_out, batch)
        cur_loss = criterion(dec_out, batch["ans_ind"].view(-1))
        cur_loss.backward()

Instances