Merge pull request #136 from kipoi/Deepsea-remove-legacy-dep

Fix pytorch 1.0 compatibility of DeepSEA and Basset models
kipoi · Dec 10, 2018 · 581e40b · 581e40b
2 parents 899d5de + 2a3a6b9
commit 581e40b
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 2 deletions.
diff --git a/Basset/pretrained_model_reloaded_th.py b/Basset/pretrained_model_reloaded_th.py
@@ -1,7 +1,6 @@
 
 import torch
 import torch.nn as nn
-import torch.legacy.nn as lnn
 
 from functools import reduce
 from torch.autograd import Variable

diff --git a/DeepSEA/model_architecture.py b/DeepSEA/model_architecture.py
@@ -1,6 +1,5 @@
 import torch
 import torch.nn as nn
-import torch.legacy.nn as lnn
 import numpy as np
 from functools import reduce
 from torch.autograd import Variable