search

Shaoli-Huang / SnapMix

SnapMix: Semantically Proportional Mixing for Augmenting Fine-grained Data (AAAI 2021)
129 stars 25 forks source link

Why PyTorch #8

Open BorScho opened 3 years ago

BorScho commented 3 years ago

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?

innat commented 3 years ago

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.