wiiiktor

This line in my custom recipe does not work (the only one that I have added): from torchtune.datasets import text_completion_dataset When I run tune, the message is: ImportError: cannot import...

I am using Gemma-2B and it is not saving checkpoints at all. It hangs (no error, just waiting forever). I use 4 gpus, but even if memory usage is very...