Open HaiMianBBao opened 8 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
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