hezhangsprinter / DCPDN

Densely Connected Pyramid Dehazing Network (CVPR'2018)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

train.py problem

ImageWQ opened this issue · comments

Hello, according to your reminder, we have successfully trained the network today. The following problems are encountered:

  1. The training command "python train.py --dataroot ./facades/train512 --valDataroot ./facades/test512 --exp ./checkpoints_new --netG ./demo_model/netG_epoch_8.pth" provided in readme, in which the test data set folder was mistaken by me, and the folder should not be "test512". After decompressing the training data set, the file path should be "./facades/val512".
  2. Training data download slowly, today only 200 downloads. When I trained with 200 data, I reported an error: "Runtime Error: unable to write to file". The query found that there was insufficient shared memory. Later, 100 data were used to train normally. I hope to know your training environment and configuration for reference.
    Tomorrow morning I'll look at the training results and try to use more data to train. I hope I don't make any more mistakes. It's a great honor to communicate with you!