Граф коммитов

11 Коммитов

Автор SHA1 Сообщение Дата
Caleb Robinson 3e296991e2 Removing some headers 2021-08-31 13:25:05 -05:00
Adam J. Stewart da7e120350 Add license header to all files 2021-08-31 13:25:05 -05:00
Caleb Robinson e460d5af23
Pytorch lightning based training framework (#42)
* Initial commit of lightning based model training framework

* Made save directories work correctly

* Add pytorch-lightning dependency and some comments

* More documentation and cosmetic tweaks

* Typo fix

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>

* Fix some style issues

* Fix pydocstyle

* Add missing sklearn dependency

* Try to get conda environment working

* Add documentation

* Ignore missing target reference

* Make train.py executable

* Ignore logs and output dirs

* Raise exceptions instead of returning

* Move all argparse stuff to set_up_parser

* Add tests for train.py

* Fix Python 3.6 compatibility

* Fix support for older versions of pytorch-lightning

Co-authored-by: Adam J. Stewart <ajstewart426@gmail.com>
2021-07-17 16:57:18 -07:00
Adam J. Stewart 2ea2bf6eef Ignore AML files 2021-07-14 14:45:03 -05:00
Adam J. Stewart 282fe805dc
Add tests for COWC datasets 2021-06-18 22:10:38 +00:00
Adam J. Stewart 7b06347424 Address review comments 2021-06-07 13:50:34 -05:00
Caleb Robinson c5c17d1781 Removing .ipynb from .gitignore 2021-06-07 13:50:34 -05:00
Caleb Robinson fb34d1906e Added visual studio code ignores to .gitignore 2021-06-07 13:50:34 -05:00
Adam J. Stewart 3a3b5d80af Add Spack gitignores 2021-06-07 15:25:54 +00:00
Adam J. Stewart eb81387814
Add config files 2021-05-11 20:54:53 -05:00
Adam J. Stewart 2330aad537
Initial commit 2021-05-11 14:46:50 -05:00