Dhruv Naik

I am using Resize inside Pytorch lightning training step and it makes my code terribly slow. Is there a solution for this?

@JackCaoG I am using Resize in my training step, more specifically `transforms.Resize(size=224, interpolation=transforms.InterpolationMode.BICUBIC, max_size=None, antialias=None)`. I am using the metric report function, but the training seems to be stuck at...

@sherlock42 check this out for training on TPU VMs with accelerate

@patrickvonplaten are there any open items for Jax/Flax integration? I'd love to contribute

Hi, I have recently faced a similar high memory problem with xgboost. I am using 'gpu_hist' for training. I notice large system memory spikes when `train()` method is executed, which...

cc: @patil-suraj if you or someone else could review this, It'd be great. Trying to fix the image result, but a review can help fix any other issues.

Fixed it. Forgot to update the for loop with the right timesteps. 

@patrickvonplaten I am mostly done with this. I just need to update the docs, which I will finish tonight. I have tested this on a TPU v3-8 pod and it...

Should be good to go @patrickvonplaten

Hi, I am interested in contributing to the research project.