stages:
  train:
    cmd: python ./train.py 10
    deps:
    - Customers.csv
    - train.py
  test:
    cmd: python ./test.py
    deps:
    - cifar_net.pth
    - test.py