Spaces:
Paused
Paused
Dean
Fixed a bug in the training stage where the model was not saved, commiting before training on colab
79fd7d0
| stages: | |
| process_data: | |
| cmd: python3 src/code/make_dataset.py src/data/raw/nyu_depth_v2_labeled.mat src/data/raw/splits.mat | |
| src/data/processed | |
| deps: | |
| - src/code/make_dataset.py | |
| - src/data/raw/nyu_depth_v2_labeled.mat | |
| - src/data/raw/splits.mat | |
| outs: | |
| - src/data/processed/ | |
| train: | |
| cmd: python3 src/code/training.py src/data/processed | |
| deps: | |
| - src/code/training.py | |
| - src/data/processed/ | |
| outs: | |
| - src/models/ | |