bug fixes

Author: johnolafenwa

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)',

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):