B-1368
B-1368
4G显存是不是有点太小了?可以换一个大点显存的显卡或者试着在main.py里面把训练时输入的图像尺寸调小
感觉还是换个大点显存的显卡试试吧,双卡12g显存的应该够
用这个: https://docs.google.com/forms/d/e/1FAIpQLSehr5Dh_i3TwACmFFi8QEgIVNYGmSPwV0GueIcsUev0NEfUug/viewform?usp=sf_link
> import sat 是先import sat再import SwissArmyTransformer吗?还是不太行啊 ``` >>> import sat [2023-09-09 08:40:09,351] [INFO] [real_accelerator.py:158:get_accelerator] Setting ds_accelerator to cuda (auto detect) [2023-09-09 08:40:11,026] [INFO] webdataset not install, use pip to install,...