Sync for release

GilesStrong · Jun 28, 2021 · c69427a · c69427a
1 parent 0907737
commit c69427a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nbs/01_model_wrapper.ipynb b/nbs/01_model_wrapper.ipynb
@@ -55,7 +55,7 @@
     "from fastprogress import master_bar, progress_bar\n",
     "import numpy as np\n",
     "\n",
-    "from torch.tensor import Tensor\n",
+    "from torch import Tensor\n",
     "import torch\n",
     "import torch.nn as nn\n",
     "from torch import optim"

diff --git a/pytorch_inferno/model_wrapper.py b/pytorch_inferno/model_wrapper.py
@@ -12,7 +12,7 @@
 from fastprogress import master_bar, progress_bar
 import numpy as np
 
-from torch.tensor import Tensor
+from torch import Tensor
 import torch
 import torch.nn as nn
 from torch import optim