9108041562e1b7a4fb159a8c0afe0caf54fe2a6d,snli/train.py,,,#,16

Before Change

if config.birnn:
    config.n_cells *= 2

model = SNLIClassifier(config)
if args.word_vectors:
    model.embed.weight.data = inputs.vocab.vectors
model.cuda()
criterion = nn.CrossEntropyLoss()
opt = O.Adam(model.parameters(), lr=args.lr)

iterations = 0

After Change

if config.birnn:
    config.n_cells *= 2

if args.resume_snapshot:
    model = torch.load(args.resume_snapshot, map_location=lambda storage, locatoin: storage.cuda(args.gpu))
else:
    model = SNLIClassifier(config)
    if args.word_vectors:
        model.embed.weight.data = inputs.vocab.vectors
        model.cuda(args.gpu)

criterion = nn.CrossEntropyLoss()
opt = O.Adam(model.parameters(), lr=args.lr)

iterations = 0

Instances