Open lucasjinreal opened 6 years ago
Maybe you should update the batch norm variable first. Btw, if you use this code in your github repo, you need to refer to this repository.
@hellochick Of course, that is only the inference code which works fine, but the training actually not works. If so, I'd better delete it.
I have tried to train on different dataset, and the code can work. There are several things you should notice:
--update_mean_var
I also have not been able to get this code to train on my own data. Could you give us the details of how you got it to work?
I have tried to train on different dataset, and the code can work. There are several things you should notice:
- Update the mean & var first using the flag
--update_mean_var
- Train the beta & gamma variable for batch norm layer.
- Freeze those variables and keep finetuning on your dataset.
@hellochick Thanks for the guidelines for training own dataset. May I know how do you separate these three steps? Are they three different training?
I have tried to train on different dataset, and the code can work. There are several things you should notice:
- Update the mean & var first using the flag
--update_mean_var
- Train the beta & gamma variable for batch norm layer.
- Freeze those variables and keep finetuning on your dataset.
The third, where is reflected in the code
This is what I got, really bad result, the car can not segment out. I have trained with pretuned model about 4750 epochs.
Here is my predict codes, what have I miss?