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