nowittynamesleft / protein_function_description

Describing protein sets' functions with natural language.
0 stars 0 forks source link

pytorch lightning integration #5

Closed nowittynamesleft closed 2 years ago

nowittynamesleft commented 3 years ago

should be able to scale all models with lightning plug-and-play like advertised in the lightning docs

nowittynamesleft commented 3 years ago

pytorch-lightning-integration branch has working training code. automatic lightning logging functions not tested yet

nowittynamesleft commented 2 years ago

tensorboard logging on the cluster is having an issue where connection is refused when starting tensorboard locally. restarting and re-forwarding haven't solved the issue. might need to email nick or ian.

nowittynamesleft commented 2 years ago

reinstalled tensorboard in a separate conda environment; problem was with a bad tensorboard version, and there was a conflicting tensorflow/tensorboard combination in my other conda environment

nowittynamesleft commented 2 years ago

pytorch lightning has been integrated, logging etc is a separate issue if it comes up again