cb413bcf7b58223c6c4335f2eaaf3dc53c0b0c53,train.py,,main,#,190

Before Change

    device = torch.device("cuda" if opt.cuda else "cpu")
    //========= Loading Dataset =========//
    data = pickle.load(open(opt.data, "rb"))
    opt.max_token_seq_len = data["settings"].max_len
    opt.src_pad_idx = data["vocab"]["src"].vocab.stoi[Constants.PAD_WORD]
    opt.trg_pad_idx = data["vocab"]["trg"].vocab.stoi[Constants.PAD_WORD]

    opt.src_vocab_size = len(data["vocab"]["src"].vocab)
    opt.trg_vocab_size = len(data["vocab"]["trg"].vocab)
    training_data, validation_data = prepare_dataloaders(data, device, opt.batch_size)

    //========= Preparing Model =========//

After Change

Instances