e4ae2c6e5c4cf6f3b02fb592d03c1caaddcfcd99,examples/pytorch_lstm_tagger.py,,main,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,105

Before Change

         min_batch_size=1, max_batch_size=4, learn_rate=0.001,
         momentum=0.9, dropout=0.5, dropout_decay=1e-4,
         nb_epoch=20, L2=1e-6):
    if CupyOps.xp != None:
        print("Use GPU")
        torch.set_default_tensor_type("torch.cuda.FloatTensor")
        Model.ops = CupyOps()
        Model.Ops = CupyOps
    cfg = dict(locals())
    print(cfg)
    train_data, check_data, nr_tag = ancora_pos_tags()
    train_data = list(train_data)

After Change

         min_batch_size=16, max_batch_size=16, learn_rate=0.001,
         momentum=0.9, dropout=0.5, dropout_decay=1e-4,
         nb_epoch=20, L2=1e-6):
    using_gpu = prefer_gpu()
    if using_gpu:
        torch.set_default_tensor_type("torch.cuda.FloatTensor")
    cfg = dict(locals())
    print(cfg)

Instances