path # lines of code train.py 275 split_data.py 60 src/slurm.py 105 src/__init__.py 1 src/logger.py 38 src/utils.py 112 src/model/__init__.py 29 src/model/transformer.py 369 src/trainer.py 397 src/envs/__init__.py 14 src/envs/ode.py 1437 src/optim.py 209 src/evaluator.py 574