Closed xuebaobizui1 closed 1 year ago
当时我在Finetune on CARPK时,确实由于hub提供的服务不稳定,有时会断联,所以时不时有些图片的load可能会需要比较长的时间,但也不至于到了一个小时2个epoch这么夸张的地步,并且没有出现内存占用高而显存占用低的问题。我不确定你是否把device设置成CPU了?如果不是的话,那可能就是服务器网络的问题。
如果你们服务器确实不能很好地使用hub的服务的话,你也可以在这里下载CARPK数据集。然后主要是要把FSC_finetune_CARPK.py 中的line 204到line 240这部分数据预处理模仿FSC_finetune_cross.py中line 122对FSC147做的那样,改写一下。剩下的可能就是要改一下line 261这里关于gt的部分。我当时没有在本地CARPK上训练,所以可能没法提出很具体的建议,希望我上面说的这些能帮到你。
非常感谢!!!这对我很有帮助
在Finetune on CARPK部分,我发现训练速度非常的缓慢,大约一小时2个epoch,按照这个速度在3090上需要训练几十天,这显然是不正常的。与此同时内存的占用很高,但是显存的占用很低。 我尝试修改num_workers的数值以及batch size的大小,发现会因为输入形状问题报错。不知道这是否是因为使用了deeplake的缘故。 或许应该在本地CARPK上进行训练?如果在本地训练的话,我该怎么修改FSC_finetune_CARPK.py,或许您能给我一些建议!!感激不尽!!!