David Tweedle
David Tweedle
Hi, I have been working on this. - The code I have is running on cifar (on kaggle), and it seems to be fine. - The wrapping I used is...
Hi, OK for now I will disable the model checkpoints. Here is a gist of the logs for this run. https://gist.github.com/davidtweedle/a870a7dd0d409e920604565a2e08b638 I am not sure what to make of this...
Hi, I hope it is appropriate to give a quick update on what could be going on here. When the batch norm is updated during the training step, "module.apply" is...
Here is a log from attempting to run the jitted ogbg workload: [ogbg.log](https://github.com/user-attachments/files/19349644/ogbg.log)