def load(self, path):
# dont remove this line. This is important to init the PyTorch nn
self.init_modules()
print('loading from: ' + path)
model = PyTorchAug.load(path)
print(model)
return model
def test_save_load():
np.random.seed(123)
a_np = np.random.randn(3, 2).astype(np.float32)
a = PyTorch.asFloatTensor(a_np)
print('a', a)
filename = '/tmp/foo.t7' # TODO: should use tempfile to get this
PyTorchAug.save(filename, a)
b = PyTorchAug.load(filename)
print('type(b)', type(b))
print('b', b)
assert np.abs(a_np - b.asNumpyTensor()).max() < 1e-4
