Im using Hydra and Data Version Control for the first time, and I’ve got some problems. I’ve written dvc.yaml file but I get failed to reproduce ‘train’: output ‘model.pt’ does not exist as the result of dvc repro
My dvc.yaml looks like this.
stages:
prepare_data:
cmd: python prepare_data.py
deps:
- prepare_data.py
outs:
- cifar10
train:
cmd: python train.py
deps:
- cifar10
- modeling
- train.py
outs:
- model.pt
generate_sample:
cmd: python generate_sample.py
deps:
- modeling
- generate_sample.py
- model.pt
outs:
- samples
train.py fits model and then saves it to model.pt:
@hydra.main(config_path='conf', config_name='config')
def main(cfg: DictConfig):
...
torch.save(ddpm.state_dict(), hydra.utils.get_original_cwd() + "/model.pt")
What am I missing?