schema: '2.0'
stages:
  prepare:
    cmd: python3 create_dataset.py
    deps:
    - path: create_dataset.py
      md5: 4d9e4ab135195a41948fb916eb6727a8
      size: 2091
    outs:
    - path: train_dataset.pt
      md5: 0da09964bb62e996ad8b77ed47a4d01d
      size: 15797
      isexec: true
    - path: val_dataset.pt
      md5: 90daa3482176c8e3911935e3aa28d93c
      size: 13365
      isexec: true
  train:
    cmd: python3 dlgssdpytorch.py
    deps:
    - path: dlgssdpytorch.py
      md5: 14e2d549d164c18e69e45c445a8e0982
      size: 4906
    - path: train_dataset.pt
      md5: 0da09964bb62e996ad8b77ed47a4d01d
      size: 15797
    - path: val_dataset.pt
      md5: 90daa3482176c8e3911935e3aa28d93c
      size: 13365
    outs:
    - path: model.pt
      md5: edd3adcee97259e2f23731dfdf5356c7
      size: 1397
      isexec: true