hankkung / logpred Goto Github PK

(wmlce-1.7.0) c699c056:LogPred ccyang$ python train.py -model dl -num_layers 2 -hidden_size 128 -window_size 10 -dataset hd -epoch 300
Traceback (most recent call last):
  File "train.py", line 11, in <module>
    from deeplog.model import *
  File "/gpfs/wscgpfs01/ccyang/LogPred/deeplog/model.py", line 1, in <module>
    class DL(nn.Module):
NameError: name 'nn' is not defined

(wmlce-1.7.0) c699login01:LogPred ccyang$ python train_vae.py 
Number of sessions: 4855
Traceback (most recent call last):
  File "train_vae.py", line 90, in <module>
    val_dataset = DataLoader(val_dataset, batch_size=4096, shuffle=False, pin_memory=True)
NameError: name 'val_dataset' is not defined