Unfortunately I was not able to run the repository (I tried on iNaturalist). I had CPU memory issues when trying to load a batch, so I tried to modify the repository:
I used torch.utils.data.Subset to load sub-batches;
I used mixed-precision to speed-up training.
But the results were way bellow the ones reported in your paper. Were you planing to release the weights of your models ?
Hi,
Unfortunately I was not able to run the repository (I tried on iNaturalist). I had CPU memory issues when trying to load a batch, so I tried to modify the repository:
torch.utils.data.Subset
to load sub-batches;Thank you!