17 lines
287 B
YAML
17 lines
287 B
YAML
|
stages:
|
||
|
|
prepare:
|
||
|
|
cmd: python3 create_dataset.py
|
||
|
|
deps:
|
||
|
|
- create_dataset.py
|
||
|
|
outs:
|
||
|
|
- train_dataset.pt
|
||
|
|
- val_dataset.pt
|
||
|
|
train:
|
||
|
|
cmd: python3 dlgssdpytorch.py
|
||
|
|
deps:
|
||
|
|
- dlgssdpytorch.py
|
||
|
|
- train_dataset.pt
|
||
|
|
- val_dataset.pt
|
||
|
|
outs:
|
||
|
|
- model.pt
|