8g显存，bs设置为4，爆显存

[ray7jq]

bs设为多少都爆显存

[ray7jq]

不爆显存了，但是一直卡在Epoch 1/50: 0%| | 0/113 [00:00<?, ?it/s<class 'dict'>]，卡在model_train(images)，不知道什么原因

[bubbliiiing]

这种环境没配好的

[ray7jq]

感谢up深夜回复，我使用你给的数据集测试是正常的 ，我训练自己的数据集会出现这个问题，我输入的图片是（720，720，3），进入model_train(images)就卡住了。

[ray7jq]

一开始爆显存是因为我输入额图片是(720,1280,3),以为是新手，所以说还在找原因。

[bubbliiiing]

0 0 一般卡住都是没配好，康康环境

[ray7jq]

up今天研究了下代码，发现在dataloader.py271行的batch_images_path的len是固定的，后面_convert_path_list_to_images_and_labels函数中的number_of_pairs固定为2了。还有就是__getitem__这个函数的index参数没用，那么训练的图片永远都是随机取了。

[bubbliiiing]

嗯是

[bubbliiiing]

你可能是每个类的图片太少

[ray7jq]

感谢up的回复，因为比较小白所以有时候问的比较傻，代码写的很不错！

[bubbliiiing]

加油啊