Spaces:
Paused
Paused
Dean
Finished training of model, saving before qualitative testing. Seems model has actually learned something. Need to add metrics and params to the pipeline.
479e632
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: | |
- path: src/code/make_dataset.py | |
md5: b15a09b30657f303d62215ceb73c6f4f | |
- path: src/data/raw/nyu_depth_v2_labeled.mat | |
md5: 520609c519fba3ba5ac58c8fefcc3530 | |
- path: src/data/raw/splits.mat | |
md5: 08e3c3aea27130ac7c01ffd739a4535f | |
outs: | |
- path: src/data/processed/ | |
md5: 78d932b016048d527f9e73e1781127cd.dir | |
train: | |
cmd: python3 src/code/training.py src/data/processed | |
deps: | |
- path: src/code/training.py | |
md5: d36fea3f2f8524ea040fe7330ce8ba40 | |
- path: src/data/processed/ | |
md5: 78d932b016048d527f9e73e1781127cd.dir | |
outs: | |
- path: src/models/ | |
md5: 48f953dd81c67ea0f6958fe4e4226e01.dir | |