Failed to reproduce 'train': output 'model.pt' does not exist

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? :frowning: