JiahuiYu / slimmable_networks

Slimmable Networks, AutoSlim, and Beyond, ICLR 2019, and ICCV 2019
Other
914 stars 131 forks source link

Slimmable Network using Mixed Precision #53

Open pprp opened 3 years ago

pprp commented 3 years ago

@JiahuiYu Thank you for your great work, I have some questions about slimmable network using mixed precision.

How to use Mixed Precision when training the Slimmable Network ?

How to make the convolution below half-precision?

https://github.com/JiahuiYu/slimmable_networks/blob/5dc14d0357ccfc596d706281acdc8a5b0b66c6d6/models/slimmable_ops.py#L50-L53