1. 介绍
[img2img-turbo]是[pytorch-CycleGAN-and-pix2pix]推荐的更新的图像变换的代码实现;
2. 配置信息
Conda环境名称:img2img-turbo
3. 问题描述
当前在我们尝试使用了官方推荐的训练命令在2080Ti上进行训练,
3.1 出现了 CUDA out of memory(OOM)错误,训练无法正常启动
相关信息如图所示,

在官方命令中,train_batch_size=1
,
并开启--enable_xformers_memory_efficient_attention
;