Traceback (most recent call last):


 "train.py", line 381, in <module>

model = model.cuda() # model = torch.nn.DataParallel(model, device_ids=[0]).cuda() # Multiple GPU parallelism


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 216, in cuda

return self._apply(lambda t: t.cuda(device))


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 146, in _apply

module._apply(fn)


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 146, in _apply

module._apply(fn)


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 146, in _apply

module._apply(fn)


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 152, in _apply

param.data = fn(param.data)


 "/home/user/.local/lib/python2.7/site-packages/torch/nn/modules/module.py", line 216, in <lambda>

return self._apply(lambda t: t.cuda(device))


 "/home/user/.local/lib/python2.7/site-packages/torch/_utils.py", line 69, in _cuda

return new_type(self.size()).copy_(self, async)
RuntimeError: cuda runtime error (2) : out of memory at /pytorch/torch/lib/THC/generic/THCStorage.cu:58

Traceback (most recent call last):


 "save_model.py", line 19, in <module>

classifier.save(save_here, save_format='h5')


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/tensorflow_core/python/keras/engine/network.py", line 986, in save

signatures, options)


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/save.py", line 112, in save_model

model, filepath, overwrite, include_optimizer)


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py", line 109, in save_model_to_hdf5

save_weights_to_hdf5_group(model_weights_group, model_layers)


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/tensorflow_core/python/keras/saving/hdf5_format.py", line 631, in save_weights_to_hdf5_group

param_dset = g.create_dataset(name, val.shape, dtype=val.dtype)


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/h5py/_hl/group.py", line 139, in create_dataset

self[name] = dset


 "/Users/alexryan/miniconda3/envs/tf2/lib/python3.7/site-packages/h5py/_hl/group.py", line 373, in __setitem__

h5o.link(obj.id, self.id, name, lcpl=lcpl, lapl=self._lapl)


 "h5py/_objects.pyx", line 54, in h5py._objects.with_phil.wrapper


 "h5py/_objects.pyx", line 55, in h5py._objects.with_phil.wrapper


 "h5py/h5o.pyx", line 202, in h5py.h5o.link
RuntimeError: Unable to create link (name already exists)
