dvc.lock

schema: '2.0'
stages:
  tripletloss:
    cmd: python -m src.auxilliary.train clustering
    deps:
    - path: data/annotations/
      hash: md5
      md5: 3d83dbeb25cfaaacd791caf6a0462b40.dir
      size: 215163
      nfiles: 6
    - path: src/auxilliary/loss.py
      hash: md5
      md5: 7fbc29b62b2dc8a879dde20a3b49eddd
      size: 545
    - path: src/auxilliary/train.py
      hash: md5
      md5: 9b2f538ed94ed6c054ca6ba31b48aacc
      size: 3121
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/siamese.py
      hash: md5
      md5: 938d75ebec30eaa7a2c7c7e068534829
      size: 1669
    outs:
    - path: models/clustering.pt
      hash: md5
      md5: 5a75283d5ecd9709a380cd12dcfece9d
      size: 14122
  default-train:
    cmd: python -m src.model.train default
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/arch.py
      hash: md5
      md5: 8ceccba1d6b7127d1e24ac5eef027dd6
      size: 2087
    - path: src/model/loss.py
      hash: md5
      md5: 8fa191c1eea16977be4d5c13240e4df5
      size: 1775
    - path: src/model/train.py
      hash: md5
      md5: a524e64c7950cb94ff3501754f6c0e44
      size: 4402
    outs:
    - path: models/default.pt
      hash: md5
      md5: 35841284222c2ef19a6d67353b3593ab
      size: 856128
  multiloss-train:
    cmd: python -m src.model.train multiloss
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: models/clustering.pt
      hash: md5
      md5: 5a75283d5ecd9709a380cd12dcfece9d
      size: 14122
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/arch.py
      hash: md5
      md5: 8ceccba1d6b7127d1e24ac5eef027dd6
      size: 2087
    - path: src/model/loss.py
      hash: md5
      md5: 8fa191c1eea16977be4d5c13240e4df5
      size: 1775
    - path: src/model/train.py
      hash: md5
      md5: a524e64c7950cb94ff3501754f6c0e44
      size: 4402
    outs:
    - path: models/multiloss.pt
      hash: md5
      md5: 07ff6a56b46ca0556de67276d84adef7
      size: 856600
  default-visualize:
    cmd: python -m src.model.evaluate visualize default
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: models/default.pt
      hash: md5
      md5: 35841284222c2ef19a6d67353b3593ab
      size: 856128
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/evaluate.py
      hash: md5
      md5: 0b7421f9443f349a4df511bc59f33554
      size: 2463
    outs:
    - path: data/evals/default.png
      hash: md5
      md5: 4abdbdb93b2044cacf60aca75faf1a46
      size: 2157710
  default-group:
    cmd: python -m src.model.evaluate group default
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: models/default.pt
      hash: md5
      md5: 35841284222c2ef19a6d67353b3593ab
      size: 856128
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/evaluate.py
      hash: md5
      md5: 0b7421f9443f349a4df511bc59f33554
      size: 2463
    outs:
    - path: data/evals/default.txt
      hash: md5
      md5: 35a96f129ab90c6d5b37fbfd33cd00e6
      size: 454
  multiloss-visualize:
    cmd: python -m src.model.evaluate visualize multiloss
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: models/multiloss.pt
      hash: md5
      md5: 07ff6a56b46ca0556de67276d84adef7
      size: 856600
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/evaluate.py
      hash: md5
      md5: 0b7421f9443f349a4df511bc59f33554
      size: 2463
    outs:
    - path: data/evals/multiloss.png
      hash: md5
      md5: 6e9dc3f10ba987d3771c07ac5d30bc57
      size: 1995343
  multiloss-group:
    cmd: python -m src.model.evaluate group multiloss
    deps:
    - path: data/waterbirds/
      hash: md5
      md5: 37b8a1f7249ecb1a68d5f39206d888fe.dir
      size: 492707912
      nfiles: 11789
    - path: models/multiloss.pt
      hash: md5
      md5: 07ff6a56b46ca0556de67276d84adef7
      size: 856600
    - path: src/const.py
      hash: md5
      md5: 08ebae9f12d5a9df85d2174a19585eba
      size: 1128
    - path: src/data/waterbirds.py
      hash: md5
      md5: 426c6ed08c3c34fea8e50677f31c2ba5
      size: 1623
    - path: src/model/evaluate.py
      hash: md5
      md5: 0b7421f9443f349a4df511bc59f33554
      size: 2463
    outs:
    - path: data/evals/multiloss.txt
      hash: md5
      md5: e245db041a5ed8795052c10c6ccba743
      size: 450