Aki van Galen

Results 2 comments of Aki van Galen

Hi John, I had also trouble training the conversion model. In my case, the reason was that mask was not correctly created in parse_ourput() in model.py. The following code fixed...

I am using torch==1.4. The original code suggested to install torch==1.1 but apex which is needed for fp16 cannot be installed then. I will check torch==1.1 to be sure.