cycle-gan训练时报错：RuntimeError: CUDA out of memory. Tried to allocate 16.00 MiB（如图）

img
看其他文章说要加一个with.torch.no_grad().没看明白是在哪里加，怎么找？