search

bubbliiiing / Siamese-pytorch

这是一个孪生神经网络（Siamese network）的库，可进行图片的相似性比较。
MIT License
598 stars 127 forks source link

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

Open ray7jq opened 2 years ago

ray7jq commented 2 years ago

bs设为多少都爆显存

ray7jq commented 2 years ago

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

bubbliiiing commented 2 years ago

这种环境没配好的

ray7jq commented 2 years ago

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

ray7jq commented 2 years ago

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

bubbliiiing commented 2 years ago

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

ray7jq commented 2 years ago

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

bubbliiiing commented 2 years ago

嗯是

bubbliiiing commented 2 years ago

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

ray7jq commented 2 years ago

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

bubbliiiing commented 2 years ago

加油啊