Training layers saved in train_save.py

Xilinx / Vitis-AI-Tutorials

MIT License

381 stars 145 forks source link

Training layers saved in train_save.py #18

Open gjkunde opened 3 years ago

gjkunde commented 3 years ago

Hello,

it appears that the dropout layers are saved in the train_save.py tutorial, the quantizer seems to crash finding those. Would be nice if the tutorial could be updated.