Laura Cabayol Garcia commited on
Commit
503dfe6
·
1 Parent(s): 4e1e7f9

move some files

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. train.py → scripts/train.py +0 -3
README.md CHANGED
@@ -57,7 +57,7 @@ jupytext --to ipynb notebooks/*.py
57
  The model can be trained using the train.py script at the repo main directory.
58
 
59
  ```bash
60
- python train.py --config-file data/config.yml
61
  ```
62
  More information on the training script can be found in the [docs](https://lauracabayol.github.io/TEMPS/docs/docs/index.html)
63
 
 
57
  The model can be trained using the train.py script at the repo main directory.
58
 
59
  ```bash
60
+ python scripts/train.py --config-file data/config.yml
61
  ```
62
  More information on the training script can be found in the [docs](https://lauracabayol.github.io/TEMPS/docs/docs/index.html)
63
 
train.py → scripts/train.py RENAMED
@@ -1,9 +1,6 @@
1
- import numpy as np
2
  import torch
3
  import argparse
4
  from loguru import logger
5
- from astropy.table import Table
6
- import pandas as pd
7
  from pathlib import Path
8
 
9
  from temps.archive import Archive
 
 
1
  import torch
2
  import argparse
3
  from loguru import logger
 
 
4
  from pathlib import Path
5
 
6
  from temps.archive import Archive