bug fixes

johnolafenwa · Jun 8, 2019 · 518ea86 · 518ea86
1 parent 24f8c30
commit 518ea86
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from setuptools import setup,find_packages
 
 setup(name="torchfusion-utils",
-      version='0.1.4',
+      version='0.1.5',
       description='A pytorch helper library for Mixed Precision Training, Initialization, Metrics and More Utilities to simplify training of deep learning models',
       url="https://github.com/johnolafenwa/TorchFusion-Utils",
       author='John Olafenwa and Moses Olafenwa (DeepQuest AI)',

diff --git a/torchfusion_utils/fp16/fp16.py b/torchfusion_utils/fp16/fp16.py
@@ -8,6 +8,7 @@
 from torch.autograd import Variable
 from torch._utils import _flatten_dense_tensors, _unflatten_dense_tensors
 from torch.optim import Optimizer
+import copy
 
 class MultiSequential(nn.Sequential):
     def __init__(self, *args):