Shaoli-Huang / SnapMix

SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Why PyTorch

BorScho opened this issue · comments

I am quite new to ML and started to implement SnapMix in Keras as an exercise (not finished yet - though...). While doing this, I wondered, if there was a special reason for You to use PyTorch?

With PyTorch is easy to implement the research idea you have. You can do it in Keras as well. It's a matter of preference.