Traceback (most recent call last):


 "train_classes_v2.py", line 338, in <module>

_main(parser.parse_args())


 "train_classes_v2.py", line 292, in _main

use_multiprocessing=True,


 "/home/michael/.local/lib/python3.5/site-packages/keras/legacy/interfaces.py", line 91, in wrapper

return func(*args, **kwargs)


 "/home/michael/.local/lib/python3.5/site-packages/keras/engine/training.py", line 1415, in fit_generator

initial_epoch=initial_epoch)


 "/home/michael/.local/lib/python3.5/site-packages/keras/engine/training_generator.py", line 247, in fit_generator

callbacks.on_epoch_end(epoch, epoch_logs)


 "/home/michael/.local/lib/python3.5/site-packages/keras/callbacks.py", line 77, in on_epoch_end

callback.on_epoch_end(epoch, logs)


 "/home/michael/.local/lib/python3.5/site-packages/keras/callbacks.py", line 455, in on_epoch_end

self.model.save(filepath, overwrite=True)


 "/home/michael/.local/lib/python3.5/site-packages/keras/engine/network.py", line 1085, in save

save_model(self, filepath, overwrite, include_optimizer)


 "/home/michael/.local/lib/python3.5/site-packages/keras/engine/saving.py", line 174, in save_model

'weight_names'] = weight_names


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


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


 "/home/michael/.local/lib/python3.5/site-packages/h5py/_hl/attrs.py", line 95, in __setitem__

self.create(name, data=value, dtype=base.guess_dtype(value))


 "/home/michael/.local/lib/python3.5/site-packages/h5py/_hl/attrs.py", line 188, in create

attr = h5a.create(self._id, self._e(tempname), htype, space)


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


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


 "h5py/h5a.pyx", line 47, in h5py.h5a.create
RuntimeError: Unable to create attribute (object header message is too large)

Traceback (most recent call last):


 "main.py", line 83, in <module>

main()


 "main.py", line 75, in main

gan.train()


 "/home/p0wx/prj/UGATIT-pytorch/UGATIT.py", line 209, in train

fake_B2B, fake_B2B_cam_logit, _ = self.genA2B(real_B)


 "/home/p0wx/.local/share/virtualenvs/p0wx-AcgHNkMk/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call

result = self.forward(*input, **kwargs)


 "/home/p0wx/prj/UGATIT-pytorch/networks.py", line 108, in forward

out = self.UpBlock2(x)


 "/home/p0wx/.local/share/virtualenvs/p0wx-AcgHNkMk/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call

result = self.forward(*input, **kwargs)


 "/home/p0wx/.local/share/virtualenvs/p0wx-AcgHNkMk/lib/python3.6/site-packages/torch/nn/modules/container.py", line 92, in forward

input = module(input)


 "/home/p0wx/.local/share/virtualenvs/p0wx-AcgHNkMk/lib/python3.6/site-packages/torch/nn/modules/module.py", line 493, in call

result = self.forward(*input, **kwargs)


 "/home/p0wx/prj/UGATIT-pytorch/networks.py", line 191, in forward

out = self.rho.expand(input.shape[0], -1, -1, -1) * out_in + (1-self.rho.expand(input.shape[0], -1, -1, -1)) * out_ln
RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB (GPU 0; 10.76 GiB total capacity; 9.32 GiB already allocated; 5.56 MiB free; 621.27 MiB cached)```
