Devavrat Tomar

Hi, Thank you for your great work. I have a question regarding training using Cityscape dataset. The segmentation maps are one-hot (n_classes channels) while the corresponding images are RGB. How...

Nan loss for ResNext backbone trained on cifar 100

1

Thank you for your work. While trying your code for the Resnext backbone on cifar100, I get nan values for the training loss. As mentioned in the published paper, I...

Hi, I was trying jaxline with 2 local GPUs, but I receive ` RESOURCE_EXHAUSTED: Out of memory while trying to allocate 285212672 bytes.` errors. I have sufficient memory available on...