Open zychen2016 opened 4 years ago
Any one could help me?
Could you give some advice about how to train ssd_mobilenetv2_fpn using Imagenet classification checkpoint? Thanks
to summarize: " Please train on CPU for a few steps, the error message is much better." from https://github.com/tensorflow/tensor2tensor/issues/574#issuecomment-364722156
hello @zychen2016 i also encounted this error when training quantized model(i noticed you add 'graph_rewriter' lines in the end of the config file ). For me, when training without quantized lines(code lines in the end of the config file), it works well; but when i want to do Quantization-aware Training(using cofig just like yours), error occured. The solution for me is to uninstall tensorflow-gpu and download tensorflow(cpu version instead), it worked for me.
Reduce one or more of the following value:
You'll have to experiment with different values and different combinations of the above to get it just right
System information
Describe the problem
Use
ssd_mobilenet_v2_fpnlite_quantized_shared_box_predictor_256x256_depthmultiplier_75_coco14_sync.config
train model.Just comment the sync options,Because I train model On one Gpu.
For transfering learning,I download mobilenet_v2_1.0_224 from
https://github.com/tensorflow/models/tree/master/research/slim/nets/mobilenet
When training,Then error reported.
errors