codeslake / IFAN

[CVPR 2021] Official PyTorch Implementation for "Iterative Filter Adaptive Network for Single Image Defocus Deblurring"
GNU Affero General Public License v3.0
222 stars 37 forks source link

Warning! No positional inputs found for a module, assuming batch size is 1. #19

Open HaiMianBBao opened 3 months ago

HaiMianBBao commented 3 months ago

Hello,I used the following command to train the model, but there was an warning : Warning! No positional inputs found for a module, assuming batch size is 1.

CUDA_VISIBLE_DEVICES=0 python run.py --is_train --mode IFAN --config config_IFAN --trainer trainer --network IFAN -b 2 -th 8 -ss