Traceback (most recent call last):


 "/Users/Sami/btgym/btgym/algorithms/aac.py", line 1499, in __init__

super(Unreal, self).__init__(name='UNREAL', **kwargs)


 "/Users/Sami/btgym/btgym/algorithms/aac.py", line 497, in __init__

raise RuntimeError(msg)
RuntimeError: Base class __init__() exception occurred.

Traceback (most recent call last):


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

train(args, loader, generator, discriminator, g_optim, d_optim, g_ema, device)


 "train.py", line 225, in train

(args.r1 / 2 * r1_loss * args.d_reg_every + 0 * real_pred[0]).backward()


 "/home/leo/code/core/venv_leo/lib/python3.6/site-packages/torch/tensor.py", line 185, in backward

torch.autograd.backward(self, gradient, retain_graph, create_graph)


 "/home/leo/code/core/venv_leo/lib/python3.6/site-packages/torch/autograd/__init__.py", line 127, in backward

allow_unreachable=True)
# allow_unreachable flag
RuntimeError: CUDA out of memory. Tried to allocate 512.00 MiB (GPU 2; 10.76 GiB total capacity; 8.85 GiB already allocated; 212.31 MiB free; 9.60 GiB reserved in total by PyTorch)
