search

DRAGNLabs / 301r_retnet

2 stars 1 forks source link

Make Models Trainable Locally #62

Closed nprisbrey closed 3 months ago

nprisbrey commented 3 months ago

There was previously errors when trying to train a model with Torch Lightning locally. The fix, as seen in the first commit, is to specify the strategy for training specifically to "ddp". This fix was found after reading through Torch Lightning Pull Requests.

The second and third commits are to simply standardize script permissions so that they're all executable!

This code now works locally and I've trained a model.

nprisbrey commented 3 months ago

Seems like it is already in the config, right?