Vincent Micheli

Hey, the default configuration should require around 34 GB of memory. You can reduce the memory requirements by decreasing the batch size `batch_num_samples` and increasing the number of gradient accumulation...

Hey, Apologies for the late reply, I did not receive a notification for some reason. We did not encounter issues with these requirements on a couple different OSes and Python...

Hey, Thanks for the kind words and the work you have put in this port! We will make sure to add a pointer to your contribution and the notebook once...