I have tried the train.py. But after 97 epoch of training, the result is as the pic below:
![epoch-97-iteration-200](https://user-images.githubusercontent.com/19465753/111021356-a05c3600-8406-11eb-8146-0f36a15bc28e.jpg)
and the training loss is as:
loss (epoch-93-iteration-300) : 13.986238
save image: ./samples/epoch-93-iteration-400.jpg
loss (epoch-93-iteration-400) : 1.800856
save image: ./samples/epoch-94-iteration-100.jpg
loss (epoch-94-iteration-100) : 7.462033
save image: ./samples/epoch-94-iteration-200.jpg
loss (epoch-94-iteration-200) : 1.865844
save image: ./samples/epoch-94-iteration-300.jpg
loss (epoch-94-iteration-300) : 2.170855
save image: ./samples/epoch-94-iteration-400.jpg
loss (epoch-94-iteration-400) : 3.993443
save image: ./samples/epoch-95-iteration-100.jpg
loss (epoch-95-iteration-100) : 15.650680
save image: ./samples/epoch-95-iteration-200.jpg
loss (epoch-95-iteration-200) : 1.744222
save image: ./samples/epoch-95-iteration-300.jpg
loss (epoch-95-iteration-300) : 1.496086
save image: ./samples/epoch-95-iteration-400.jpg
loss (epoch-95-iteration-400) : 1.489223
save image: ./samples/epoch-96-iteration-100.jpg
loss (epoch-96-iteration-100) : 2.223736
save image: ./samples/epoch-96-iteration-200.jpg
loss (epoch-96-iteration-200) : 1.954349
save image: ./samples/epoch-96-iteration-300.jpg
loss (epoch-96-iteration-300) : 2.035094
save image: ./samples/epoch-96-iteration-400.jpg
loss (epoch-96-iteration-400) : 26.680279
save image: ./samples/epoch-97-iteration-100.jpg
loss (epoch-97-iteration-100) : 5.275980
save image: ./samples/epoch-97-iteration-200.jpg
loss (epoch-97-iteration-200) : 9.626511
I do not know if everything is going well. So please provide the training results for reference.