Closed Yueziyu closed 4 years ago
我在使用”python trainer.py --dataset div2k --eval_datasets div2k set5 bsds100 urban100 --model scn --num_blocks 32 --num_residual_units 128 --scale 4 --job_dir ./scn_x4_retrain” 这条命令进行训练后，程序显示GPU显存占用1000多M，但是没有显示进行到多少步，我在train.py程序中加了epoch的显示，但是一直卡在第一个epoch，请问训练大概时间是多久，如果不是训练很慢是哪里出了问题？
应该是图片太大，没有裁剪导致训练很慢
Yueziyu
commented
4 years ago Yueziyu
commented
4 years ago
我在使用”python trainer.py --dataset div2k --eval_datasets div2k set5 bsds100 urban100 --model scn --num_blocks 32 --num_residual_units 128 --scale 4 --job_dir ./scn_x4_retrain” 这条命令进行训练后，程序显示GPU显存占用1000多M，但是没有显示进行到多少步，我在train.py程序中加了epoch的显示，但是一直卡在第一个epoch，请问训练大概时间是多久，如果不是训练很慢是哪里出了问题？