Laughing-q/xinye_competition

运行时出现的问题

epiphany97 opened this issue · 3 comments

你好:
我在代码复现时,运行python tools/train_regressor.py --epochs 150 --input_size 224 --batch_size 200 --backbone swin_transformer --swin_type small --loss_head Circleloss --use_cgd 时,由于自己显卡不太够,于是将bs改为了50,结果出现了
CUDA error: device-side assert triggered 的错误,请问你有遇到这个问题吗

@epiphany97 我之前遇到过这个问题,好像是我的标签索引没设置正确,对了这个代码当时上传的时候是可以按照readme运行的,但是后来有一些inference的更新, emmmmm就现在不太确定是否有影响训练的流程(当时也没想到会有人关注,所以可能做的不是很完善), 可能会有bug哈

@epiphany97 我之前遇到过这个问题,好像是我的标签索引没设置正确,对了这个代码当时上传的时候是可以按照readme运行的,但是后来有一些inference的更新, emmmmm就现在不太确定是否有影响训练的流程(当时也没想到会有人关注,所以可能做的不是很完善), 可能会有bug哈

好滴,我在网上查貌似也是这个问题,想问下你的那个训练算力用的什么条件的呀

@epiphany97 swin transformer比较耗显存,两张3090