search

hellozhuo / pidinet

Code for the ICCV 2021 paper "Pixel Difference Networks for Efficient Edge Detection" (Oral).
Other
445 stars 69 forks source link

关于多GPU训练的询问 #33

Open renxuzhiqiuyue opened 2 years ago

renxuzhiqiuyue commented 2 years ago

您好，我看到在参数里面给了多GPU的选项，但是数据集的batch_size始终为1。如果强行更改会出现，尺寸不相符的问题。请问这个模型是不是无法进行多GPU训练

zhuoinoulu commented 2 years ago

@renxuzhiqiuyue 对的,由于输入图片分辨率不是固定的,目前不支持多GPU.