dvc.lock

schema: '2.0'
stages:
  ae_train:
    cmd: python src/autoencoder/train.py
    deps:
    - path: data/external/sprites
      md5: 23e143a3d4f7b30548cd966a889e3b87.dir
      size: 387131152
      nfiles: 45310
    - path: src/autoencoder
      md5: 01db33eca6234137ac19172e6818b387.dir
      size: 11029
      nfiles: 7
    - path: src/data
      md5: 19f3b6e442619e47134b34f6230f7148.dir
      size: 5322
      nfiles: 4
    - path: src/utils.py
      md5: bcc6681e7b87d2563003ca8dd54de48a
      size: 509
    params:
      params.yaml:
        autoencoder:
          epochs: 500
          data_dir: data/external/sprites
          log_dir: reports/autoencoder
          latent_size: 8
          batch_size: 128
    outs:
    - path: reports/autoencoder/gen
      md5: f8a8dfb30880036eb326480134bd4e3d.dir
      size: 560019
      nfiles: 500
    - path: reports/autoencoder/gen.gif
      md5: 97826c752a798816deca8ed17463a3f1
      size: 2066867
    - path: reports/autoencoder/logs
      md5: c686d10a0f59ad2034c3e45cb603dca2.dir
      size: 19736
      nfiles: 1
    - path: reports/autoencoder/logs.json
      md5: 2e09b42eef202d311b99da3c3ffc5e49
      size: 53
  data:
    cmd: tar -xzf data/raw/sprites.tar.gz -C data/raw/
    deps:
    - path: data/raw/sprites.tar.gz
      md5: 3ef8b4485374026c0896299bd33a7bd6
      size: 273138408
    outs:
    - path: data/raw/sprites
      md5: 23e143a3d4f7b30548cd966a889e3b87.dir
      size: 387131152
      nfiles: 45310
  sprites_data:
    cmd: tar -xf data/raw/sprites.tar -C data/raw/
    deps:
    - path: data/raw/sprites.tar
      md5: 84ec42a376e159573338ef93b4d6892f
      size: 421591040
    outs:
    - path: data/raw/sprites
      md5: 23e143a3d4f7b30548cd966a889e3b87.dir
      size: 387131152
      nfiles: 45310
  ae_conv_train:
    cmd: python src/autoencoder/vanilla_model_conv.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/conv.py
      md5: b504d408e6316d0c2bf1fde8f69a3e21
      size: 1319
    - path: src/autoencoder/encoders/conv.py
      md5: 1e9b050b285b3e7da795a7970156b90a
      size: 1060
    - path: src/autoencoder/models/vanilla_model.py
      md5: 32aee41e83bf49e176d18b7d8d9f4c43
      size: 550
    - path: src/autoencoder/train.py
      md5: 5d5f425b41fcacbc12face2c12ba6357
      size: 2657
    - path: src/autoencoder/vanilla_model_conv.py
      md5: 206827a01a55ea5754657b67c5f290ae
      size: 790
    - path: src/data/sprites.py
      md5: 31947df2ec8067da21c09378f9700cd5
      size: 3594
    - path: src/utils.py
      md5: ead46112488eed4252f3f205e0450946
      size: 858
    params:
      params.yaml:
        autoencoder_conv:
          latent_size: 16
          batch_size: 128
          train_kwargs:
            epochs: 100
            log_dir: reports/autoencoder/conv
          loader_kwargs:
    outs:
    - path: reports/autoencoder/conv/gen/0
      md5: 784a0a4861346767d7439e833bb5336a.dir
      size: 332585
      nfiles: 100
    - path: reports/autoencoder/conv/gen/0.gif
      md5: 6e7e74560909dad55360f4e3617adf4a
      size: 555393
    - path: reports/autoencoder/conv/gen/1
      md5: 3477e47a171db216a5248c92cb64df28.dir
      size: 354518
      nfiles: 100
    - path: reports/autoencoder/conv/gen/1.gif
      md5: 56bf6b165e81b216f700c55a710bf59a
      size: 600960
    - path: reports/autoencoder/conv/gen/2
      md5: 99e45abeac60d72e91374c2b28426041.dir
      size: 89995
      nfiles: 100
    - path: reports/autoencoder/conv/gen/2.gif
      md5: 5c75b08fb523b18e251cfbcbb311b445
      size: 369802
    - path: reports/autoencoder/conv/gen/3
      md5: a7bffb7ab9a9f6fdd01e0ddde9879abe.dir
      size: 104952
      nfiles: 100
    - path: reports/autoencoder/conv/gen/3.gif
      md5: 6040c9e8e331c6630243686730efce34
      size: 390111
    - path: reports/autoencoder/conv/logs
      md5: 215feead67d85314deddb5148fbe621a.dir
      size: 3871
      nfiles: 1
    - path: reports/autoencoder/conv/logs.json
      md5: 1159db2952ed4e5f114b6c03a23b6922
      size: 52
  ae_dense_train:
    cmd: python src/autoencoder/vanilla_model_dense.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/dense.py
      md5: bb0e46a7e49569854642a164c095a3d7
      size: 838
    - path: src/autoencoder/encoders/dense.py
      md5: b915f41a2934c5086507545d9ca6ea74
      size: 807
    - path: src/autoencoder/models/vanilla_model.py
      md5: 2269cef39cc263f80f6fab4dcd93d149
      size: 608
    - path: src/autoencoder/train.py
      md5: 75cf371b5717bda15818ddbafffff4d7
      size: 3176
    - path: src/autoencoder/vanilla_model_dense.py
      md5: 74bd0fdc571173727745fcb3391023af
      size: 959
    - path: src/data/sprites.py
      md5: 31947df2ec8067da21c09378f9700cd5
      size: 3594
    - path: src/utils.py
      md5: ead46112488eed4252f3f205e0450946
      size: 858
    params:
      params.yaml:
        autoencoder_dense:
          latent_size: 16
          batch_size: 128
          train_kwargs:
            epochs: 100
            log_dir: reports/autoencoder/dense
    outs:
    - path: reports/autoencoder/dense/gen/0
      md5: c962e482db0ec1f91e2c69a7543eb585.dir
      size: 404643
      nfiles: 100
    - path: reports/autoencoder/dense/gen/0.gif
      md5: 618acb63754f8316d1085b9a2a083bc6
      size: 1030063
    - path: reports/autoencoder/dense/gen/1
      md5: 187f3b8ad640e8031fffe989005548dd.dir
      size: 423795
      nfiles: 100
    - path: reports/autoencoder/dense/gen/1.gif
      md5: e65af6b23f465a7e6747bec6d0adb900
      size: 1075961
    - path: reports/autoencoder/dense/gen/2
      md5: e28d7506c398160d532af0389d358c71.dir
      size: 274751
      nfiles: 100
    - path: reports/autoencoder/dense/gen/2.gif
      md5: 3002a484ac5bcae321fbbad37defee79
      size: 548901
    - path: reports/autoencoder/dense/gen/3
      md5: b48bc1d481454ee4c3a46a2b30c4d3c7.dir
      size: 131784
      nfiles: 100
    - path: reports/autoencoder/dense/gen/3.gif
      md5: f84c38fc6744edb0ba0a3863eef2dec8
      size: 164173
    - path: reports/autoencoder/dense/logs
      md5: 439fdfde958b2d9adef5a72cc97d3421.dir
      size: 3872
      nfiles: 1
    - path: reports/autoencoder/dense/logs.json
      md5: 67d8c3d3f1177c9c4ec85b6b744239c7
      size: 52
  ae_dense:
    cmd: python src/autoencoder/vanilla_model_dense.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/dense.py
      md5: 7ff447d9b8d56d8ce82637c123b28d44
      size: 866
    - path: src/autoencoder/encoders/dense.py
      md5: 53c0be2894ae27af711dda768c687783
      size: 865
    - path: src/autoencoder/models/vanilla_model.py
      md5: 1db2b1b9f21021106975e1b2f8c3462f
      size: 886
    - path: src/autoencoder/train.py
      md5: f4240b973a43285f50fd44fb79ea0af7
      size: 4398
    - path: src/autoencoder/vanilla_model_dense.py
      md5: 74bd0fdc571173727745fcb3391023af
      size: 959
    - path: src/data/sprites.py
      md5: eb3b24072b0bd1f3b0357c8ca5f922ff
      size: 3595
    - path: src/utils.py
      md5: 51de3b46e88e4e72336e64cb7fa95b71
      size: 1072
    params:
      params.yaml:
        autoencoder_dense:
          latent_size: 16
          batch_size: 256
          train_kwargs:
            epochs: 10000
            log_dir: reports/autoencoder/dense
    outs:
    - path: reports/autoencoder/dense/gen/0
      md5: c286d904cad71411736e98ac420ab5c6.dir
      size: 32463
      nfiles: 30
    - path: reports/autoencoder/dense/gen/0.gif
      md5: 5f9f8b2aa8f7b8dfeec6e4912ec61de5
      size: 59178
    - path: reports/autoencoder/dense/gen/1
      md5: e270624a09b88ebdb692918080d037ed.dir
      size: 33023
      nfiles: 30
    - path: reports/autoencoder/dense/gen/1.gif
      md5: ce45e01897c02b51f92def2f85d3f871
      size: 65732
    - path: reports/autoencoder/dense/gen/2
      md5: c8ca47da563e648f46aa270843bacd93.dir
      size: 34402
      nfiles: 30
    - path: reports/autoencoder/dense/gen/2.gif
      md5: bada51ba05fd147f1e6fc04f65f5a171
      size: 81340
    - path: reports/autoencoder/dense/gen/3
      md5: 6654eb84b409999fe300267d80041108.dir
      size: 32220
      nfiles: 30
    - path: reports/autoencoder/dense/gen/3.gif
      md5: f6bc6a848e504307971da0e8503ab045
      size: 53804
    - path: reports/autoencoder/dense/logs
      md5: 3c49a8cd379e16833fca8c24af84a851.dir
      size: 803359
      nfiles: 2
    - path: reports/autoencoder/dense/logs.json
      md5: aa1c27cae47ff699bb6c735a8d6ad14b
      size: 84
    - path: reports/autoencoder/dense/results
      md5: 6ae801866777a2c62deef145a3fc5527.dir
      size: 44248
      nfiles: 16
  ae_conv:
    cmd: python src/autoencoder/vanilla_model_conv.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/conv.py
      md5: 9f974c78db50f2ed3e65106b60a74336
      size: 1605
    - path: src/autoencoder/encoders/conv.py
      md5: 28864e63d259e0c9f93e8d6ce06dc384
      size: 1549
    - path: src/autoencoder/models/vanilla_model.py
      md5: 1db2b1b9f21021106975e1b2f8c3462f
      size: 886
    - path: src/autoencoder/train.py
      md5: 425df3c56c5dd9c0bd74d07ae9386658
      size: 4028
    - path: src/autoencoder/vanilla_model_conv.py
      md5: 206827a01a55ea5754657b67c5f290ae
      size: 790
    - path: src/data/sprites.py
      md5: eb3b24072b0bd1f3b0357c8ca5f922ff
      size: 3595
    - path: src/utils.py
      md5: 51de3b46e88e4e72336e64cb7fa95b71
      size: 1072
    params:
      params.yaml:
        autoencoder_conv:
          latent_size: 16
          batch_size: 64
          train_kwargs:
            epochs: 10
            log_dir: reports/autoencoder/conv
    outs:
    - path: reports/autoencoder/conv/gen/0
      md5: 4668be6a7a3f3eb0c310d21bb89f9598.dir
      size: 11991
      nfiles: 10
    - path: reports/autoencoder/conv/gen/0.gif
      md5: 1749a2af2295fa97bce975fc5ac91659
      size: 51811
    - path: reports/autoencoder/conv/gen/1
      md5: 4a46b00a2967dc6efa8341c1025f2a60.dir
      size: 11482
      nfiles: 10
    - path: reports/autoencoder/conv/gen/1.gif
      md5: a99aedc81f5d97460979cd16e2d2e165
      size: 46685
    - path: reports/autoencoder/conv/gen/2
      md5: 0bfbcc1b08a504f94f5546b9bf44c3f0.dir
      size: 14991
      nfiles: 10
    - path: reports/autoencoder/conv/gen/2.gif
      md5: e6b0ad3925d9a6576e891f8fab382af7
      size: 55204
    - path: reports/autoencoder/conv/gen/3
      md5: 569add7b280082c228bce6207a1b5657.dir
      size: 11904
      nfiles: 10
    - path: reports/autoencoder/conv/gen/3.gif
      md5: 98b1b606ad491e99ae326a8f9d7e252e
      size: 42868
    - path: reports/autoencoder/conv/logs
      md5: 62ae753c56586a67f159bb0e33976d83.dir
      size: 752
      nfiles: 2
    - path: reports/autoencoder/conv/logs.json
      md5: 7017f28cee0ea07e8d4235468ba1a4ff
      size: 82
    - path: reports/autoencoder/conv/results
      md5: 8d4bab4df27d0a62598e6fdffe6d326f.dir
      size: 24360
      nfiles: 16
  vae_dense:
    cmd: python src/autoencoder/vae_dense.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/dense.py
      md5: 7ff447d9b8d56d8ce82637c123b28d44
      size: 866
    - path: src/autoencoder/encoders/dense.py
      md5: 53c0be2894ae27af711dda768c687783
      size: 865
    - path: src/autoencoder/models/variational_model.py
      md5: 47ce9ebb2953752db1f0a4cdd0786ec5
      size: 2135
    - path: src/autoencoder/train.py
      md5: f4240b973a43285f50fd44fb79ea0af7
      size: 4398
    - path: src/autoencoder/vae_dense.py
      md5: f90dad471834547f8f636b99860418c6
      size: 883
    - path: src/data/sprites.py
      md5: eb3b24072b0bd1f3b0357c8ca5f922ff
      size: 3595
    - path: src/utils.py
      md5: 51de3b46e88e4e72336e64cb7fa95b71
      size: 1072
    params:
      params.yaml:
        vae_dense:
          latent_size: 16
          batch_size: 256
          train_kwargs:
            epochs: 10000
            log_dir: reports/vae/dense
    outs:
    - path: reports/vae/dense/gen/0
      md5: 3a87b3a3a57f437a1bc1552868b7d2a3.dir
      size: 35032
      nfiles: 30
    - path: reports/vae/dense/gen/0.gif
      md5: 873be28fb01e05905f2b718c96ac02a9
      size: 97062
    - path: reports/vae/dense/gen/1
      md5: 638253eba64fdac4b1fac22625f3f69f.dir
      size: 34127
      nfiles: 30
    - path: reports/vae/dense/gen/1.gif
      md5: 469a3ea9bf2fc1d2616ecf2faacd5f9d
      size: 89464
    - path: reports/vae/dense/logs
      md5: 9768e698566f6bd4bf5f23e458ef1a52.dir
      size: 803222
      nfiles: 2
    - path: reports/vae/dense/logs.json
      md5: 47e2076a9a1f7366a0e00c2d6b86a8ac
      size: 85
    - path: reports/vae/dense/results
      md5: 1c7d6dee97b76f3d7ae6a60cce7f0f3c.dir
      size: 37502
      nfiles: 16
  vae_conv:
    cmd: python src/autoencoder/vae_conv.py
    deps:
    - path: data/raw/sprites
      md5: 447107e2f601c815f08b87b0eea34530.dir
      size: 242202638
      nfiles: 30118
    - path: src/autoencoder/decoders/conv.py
      md5: 9f974c78db50f2ed3e65106b60a74336
      size: 1605
    - path: src/autoencoder/encoders/conv.py
      md5: 28864e63d259e0c9f93e8d6ce06dc384
      size: 1549
    - path: src/autoencoder/models/variational_model.py
      md5: 47ce9ebb2953752db1f0a4cdd0786ec5
      size: 2135
    - path: src/autoencoder/train.py
      md5: 425df3c56c5dd9c0bd74d07ae9386658
      size: 4028
    - path: src/autoencoder/vae_conv.py
      md5: d38246cd2de65fa35319c988bad3c64c
      size: 929
    - path: src/data/sprites.py
      md5: eb3b24072b0bd1f3b0357c8ca5f922ff
      size: 3595
    - path: src/utils.py
      md5: 51de3b46e88e4e72336e64cb7fa95b71
      size: 1072
    params:
      params.yaml:
        vae_conv:
          latent_size: 16
          batch_size: 64
          train_kwargs:
            epochs: 10
            log_dir: reports/vae/conv
    outs:
    - path: reports/vae/conv/gen/0
      md5: c96ae7d218bd08e293bbb8f63115bdd1.dir
      size: 13102
      nfiles: 10
    - path: reports/vae/conv/gen/0.gif
      md5: 0b719ac3e9e0c31f4801fb6aa6a4449b
      size: 51584
    - path: reports/vae/conv/gen/1
      md5: 5561a6088553df5671460e1ca0b845fd.dir
      size: 12856
      nfiles: 10
    - path: reports/vae/conv/gen/1.gif
      md5: 346d05022cddfc94a9e7e5d0c48088d9
      size: 49963
    - path: reports/vae/conv/logs
      md5: f48a5e0785173313741e0aed8f260691.dir
      size: 749
      nfiles: 2
    - path: reports/vae/conv/logs.json
      md5: acbd153c448db4884e9411fa434e037d
      size: 82
    - path: reports/vae/conv/results
      md5: 2b98b54437a2f2268e4958b72f78e684.dir
      size: 23788
      nfiles: 16
  train@ae_dense:
    cmd: "python src/autoencoder/train.py --encoder-type dense --decoder-type dense\
      \ --ae-type ae --model-path models/ae_dense --epochs 500 --batch-size 64 --latent-size\
      \ 32 --log-dir reports/ae/dense --lr 0.0001 --val-ratio 0.1 --reg-rate 0.01\
      \ --reg-type None --gen-gifs true\n"
    deps:
    - path: data/external/sprites
      md5: f1c584fff98b529e69fe3de6a14e7e1b.dir
      size: 721084950
      nfiles: 48236
    - path: src/autoencoder/decoders/dense.py
      md5: 2fafc97da7c9fdcdff11f37341bb3f80
      size: 1746
    - path: src/autoencoder/encoders/dense.py
      md5: 3e9a7ba5d6065112ef145f426984abcc
      size: 1262
    - path: src/autoencoder/models/vanilla_model.py
      md5: 5e5157d73888ec5ce0e503b6ac778d1d
      size: 2808
    - path: src/autoencoder/train.py
      md5: c8aa30276c495fa0dd495a712aa16576
      size: 9248
    - path: src/data/sprites.py
      md5: debea5b0298af285804f6917ff812887
      size: 11205
    - path: src/utils.py
      md5: 6b884894e60db1e2b61da9e2510b5f44
      size: 13515
    outs:
    - path: models/ae_dense.pt
      md5: a5e56f15a442ecf59fa333df178c2fea
      size: 106327447
    - path: models/ae_dense_kwargs.yaml
      md5: 9b6eb5ae349431ec3491b1cd8cacb405
      size: 128
    - path: reports/ae/dense/gen
      md5: 41786a45d1dc522e008f522b34acff74.dir
      size: 8292349
      nfiles: 520
    - path: reports/ae/dense/latent
      md5: 8f343314f233c3710476ff1800b20b7e.dir
      size: 85132
      nfiles: 2
    - path: reports/ae/dense/logs
      md5: bddf4cabf0186d4be456e9002e4c78c4.dir
      size: 80404
      nfiles: 4
    - path: reports/ae/dense/logs.json
      md5: 31216d80ea79a9cbdb82f1e603476af7
      size: 164
    - path: reports/ae/dense/results
      md5: ef25851f2a5dc63920b30279581777b1.dir
      size: 23371
      nfiles: 16
    - path: reports/ae/dense/summary.txt
      md5: 1b1df3c3046bcf305b79ecd39e3051d7
      size: 1610
  train@vae_dense:
    cmd: "python src/autoencoder/train.py --encoder-type dense --decoder-type dense\
      \ --ae-type vae --model-path models/vae_dense --epochs 500 --batch-size 64 --latent-size\
      \ 32 --log-dir reports/vae/dense --lr 0.0001 --val-ratio 0.1 --reg-rate 0.01\
      \ --reg-type None --gen-gifs true\n"
    deps:
    - path: data/external/sprites
      md5: f1c584fff98b529e69fe3de6a14e7e1b.dir
      size: 721084950
      nfiles: 48236
    - path: src/autoencoder/decoders/dense.py
      md5: 2fafc97da7c9fdcdff11f37341bb3f80
      size: 1746
    - path: src/autoencoder/encoders/dense.py
      md5: 3e9a7ba5d6065112ef145f426984abcc
      size: 1262
    - path: src/autoencoder/models/variational_model.py
      md5: 3679ffc0734e74b7d7f6fc48ac2d899c
      size: 7203
    - path: src/autoencoder/train.py
      md5: c8aa30276c495fa0dd495a712aa16576
      size: 9248
    - path: src/data/sprites.py
      md5: debea5b0298af285804f6917ff812887
      size: 11205
    - path: src/utils.py
      md5: 6b884894e60db1e2b61da9e2510b5f44
      size: 13515
    outs:
    - path: models/vae_dense.pt
      md5: dc3496f58ddf678e09a134dc009b74ee
      size: 106318874
    - path: models/vae_dense_kwargs.yaml
      md5: 9b6eb5ae349431ec3491b1cd8cacb405
      size: 128
    - path: reports/vae/dense/gen
      md5: e39e64dab254cf72978fce82375ddadc.dir
      size: 4726279
      nfiles: 520
    - path: reports/vae/dense/latent
      md5: a81efdffe49927cc2403f788ec028186.dir
      size: 251512
      nfiles: 6
    - path: reports/vae/dense/logs
      md5: f22ce48c23dee21c8e6bd0a6a33d16bc.dir
      size: 92827
      nfiles: 5
    - path: reports/vae/dense/logs.json
      md5: bfb513d8128927fcdddb850f5efbe111
      size: 180
    - path: reports/vae/dense/results
      md5: a4e2fbe13a44cdcfa32bec278ab573b4.dir
      size: 21404
      nfiles: 16
    - path: reports/vae/dense/summary.txt
      md5: 70e08f936e7a9d1fe86546173322d6bf
      size: 1546
  train@vae_conv:
    cmd: "python src/autoencoder/train.py --encoder-type conv --decoder-type conv\
      \ --ae-type vae --model-path models/vae_conv --epochs 500 --batch-size 64 --latent-size\
      \ 32 --log-dir reports/vae/conv --lr 0.0001 --val-ratio 0.1 --reg-rate 0.001\
      \ --reg-type None --gen-gifs true\n"
    deps:
    - path: data/external/sprites
      md5: f1c584fff98b529e69fe3de6a14e7e1b.dir
      size: 721084950
      nfiles: 48236
    - path: src/autoencoder/decoders/conv.py
      md5: bc5c9e028db667257636f3a7c2408ba0
      size: 3591
    - path: src/autoencoder/encoders/conv.py
      md5: ec95ea5e066ff35f3a46cac7e666f0d2
      size: 3456
    - path: src/autoencoder/models/variational_model.py
      md5: 3679ffc0734e74b7d7f6fc48ac2d899c
      size: 7203
    - path: src/autoencoder/train.py
      md5: c8aa30276c495fa0dd495a712aa16576
      size: 9248
    - path: src/data/sprites.py
      md5: debea5b0298af285804f6917ff812887
      size: 11205
    - path: src/utils.py
      md5: 6b884894e60db1e2b61da9e2510b5f44
      size: 13515
    outs:
    - path: models/vae_conv.pt
      md5: 74f66e62a4f9334110a46bcf43604ee7
      size: 18930698
    - path: models/vae_conv_kwargs.yaml
      md5: 3f0de1dcb968ad63f73353610fdcd5f5
      size: 127
    - path: reports/vae/conv/gen
      md5: 28c2349516ed442355a93dbecbe7f4ab.dir
      size: 6316021
      nfiles: 520
    - path: reports/vae/conv/latent
      md5: abaa195f6c19b2b9004ecee99b9a5d7f.dir
      size: 233861
      nfiles: 6
    - path: reports/vae/conv/logs
      md5: 2e05fab48d04e6f2bb87b0b7e63ff7c4.dir
      size: 92678
      nfiles: 5
    - path: reports/vae/conv/logs.json
      md5: f9b3655365aacb6a70572ec9f6e9a28c
      size: 181
    - path: reports/vae/conv/results
      md5: c43784c0d593bd99cc78f699a562f925.dir
      size: 20518
      nfiles: 16
    - path: reports/vae/conv/summary.txt
      md5: 1b28a71b74d86dc2445801ba23b6da03
      size: 2438
  train@ae_conv:
    cmd: "python src/autoencoder/train.py --encoder-type conv --decoder-type conv\
      \ --ae-type ae --model-path models/ae_conv --epochs 500 --batch-size 64 --latent-size\
      \ 32 --log-dir reports/ae/conv --lr 0.0001 --val-ratio 0.1 --reg-rate 0.001\
      \ --reg-type None --gen-gifs true\n"
    deps:
    - path: data/external/sprites
      md5: f1c584fff98b529e69fe3de6a14e7e1b.dir
      size: 721084950
      nfiles: 48236
    - path: src/autoencoder/decoders/conv.py
      md5: bc5c9e028db667257636f3a7c2408ba0
      size: 3591
    - path: src/autoencoder/encoders/conv.py
      md5: ec95ea5e066ff35f3a46cac7e666f0d2
      size: 3456
    - path: src/autoencoder/models/vanilla_model.py
      md5: 5e5157d73888ec5ce0e503b6ac778d1d
      size: 2808
    - path: src/autoencoder/train.py
      md5: c8aa30276c495fa0dd495a712aa16576
      size: 9248
    - path: src/data/sprites.py
      md5: debea5b0298af285804f6917ff812887
      size: 11205
    - path: src/utils.py
      md5: 6b884894e60db1e2b61da9e2510b5f44
      size: 13515
    outs:
    - path: models/ae_conv.pt
      md5: d06c1882e7be22fbd3173228e326bd3c
      size: 18864781
    - path: models/ae_conv_kwargs.yaml
      md5: 3f0de1dcb968ad63f73353610fdcd5f5
      size: 127
    - path: reports/ae/conv/gen
      md5: 2bd113f6563bb717b6b46bf180e5e9e7.dir
      size: 8695695
      nfiles: 520
    - path: reports/ae/conv/latent
      md5: 0f9ac172cc5ff80102a4fbe671bee24a.dir
      size: 93234
      nfiles: 2
    - path: reports/ae/conv/logs
      md5: 6a4a2b252ee04eed7755485700f5df96.dir
      size: 80448
      nfiles: 4
    - path: reports/ae/conv/logs.json
      md5: cf431951805fcb490ac8729bb7b084b9
      size: 162
    - path: reports/ae/conv/results
      md5: 01ca49a41442dc0ac1696fd406704c98.dir
      size: 21262
      nfiles: 16
    - path: reports/ae/conv/summary.txt
      md5: ece526dcc278d211e6ac4725711385df
      size: 2438
  create_serving_model@vae_dense:
    cmd: "torch-model-archiver  --model-file src/autoencoder/models/variational_model.py\
      \ --model-name vae_dense  --serialized-file models/vae_dense.pt -v 1.0  --handler\
      \ image_classifier  --export-path models/serving\n"
    deps:
    - path: models/vae_dense.pt
      md5: 9f662f1854c83954fc18b1eca79f85f5
      size: 57089388
    - path: models/vae_dense_kwargs.yaml
      md5: a474762901fc9bf3f553cbd53fe369b7
      size: 126
    outs:
    - path: models/serving/vae_dense.mar
      md5: 76a89a10638ff8513a927e53034c2b90
      size: 52970343
  create_serving_model@vae_conv:
    cmd: "torch-model-archiver  --model-file src/autoencoder/models/variational_model.py\
      \ --model-name vae_conv  --serialized-file models/vae_conv.pt -v 1.0  --handler\
      \ image_classifier  --export-path models/serving\n"
    deps:
    - path: models/vae_conv.pt
      md5: c3a046d9de6a9ac4322436e97d4d933a
      size: 11296079
    - path: models/vae_conv_kwargs.yaml
      md5: 8418b20882bd5c9107199f5545ec489b
      size: 125
    outs:
    - path: models/serving/vae_conv.mar
      md5: 0044e4e00f4eafca979690d53df55d68
      size: 10461620
  create_serving_model@ae_dense:
    cmd: "torch-model-archiver  --model-file src/autoencoder/models/vanilla_model.py\
      \ --model-name ae_dense  --serialized-file models/ae_dense.pt -v 1.0  --handler\
      \ image_classifier  --export-path models/serving\n"
    deps:
    - path: models/ae_dense.pt
      md5: 015d38b44be5f9926240bd10063bfe9d
      size: 57091753
    - path: models/ae_dense_kwargs.yaml
      md5: a474762901fc9bf3f553cbd53fe369b7
      size: 126
    outs:
    - path: models/serving/ae_dense.mar
      md5: 0c7df83a150b9f54bc26efa094b85c65
      size: 53018038
  create_serving_model@ae_conv:
    cmd: "torch-model-archiver  --model-file src/autoencoder/models/vanilla_model.py\
      \ --model-name ae_conv  --serialized-file models/ae_conv.pt -v 1.0  --handler\
      \ image_classifier  --export-path models/serving\n"
    deps:
    - path: models/ae_conv.pt
      md5: 3f361d7b483326e116fe60a5bfe28c80
      size: 10706001
    - path: models/ae_conv_kwargs.yaml
      md5: 8418b20882bd5c9107199f5545ec489b
      size: 125
    outs:
    - path: models/serving/ae_conv.mar
      md5: 4a6a7812aec5a07d76a6b9d3d57402b2
      size: 9901544
  train@vae:
    cmd: "python src/autoencoder/train.py --encoder-type conv --decoder-type conv\
      \ --ae-type vae --model-path models/vae --epochs 500 --batch-size 64 --latent-size\
      \ 32 --log-dir reports/vae/vae --lr 0.0001 --val-ratio 0.1 --reg-rate 0.01 --reg-type\
      \ l1 --gen-gifs true\n"
    deps:
    - path: data/external/sprites
      md5: f1c584fff98b529e69fe3de6a14e7e1b.dir
      size: 721084950
      nfiles: 48236
    - path: src/autoencoder/decoders/conv.py
      md5: bc5c9e028db667257636f3a7c2408ba0
      size: 3591
    - path: src/autoencoder/encoders/conv.py
      md5: ec95ea5e066ff35f3a46cac7e666f0d2
      size: 3456
    - path: src/autoencoder/models/variational_model.py
      md5: 3679ffc0734e74b7d7f6fc48ac2d899c
      size: 7203
    - path: src/autoencoder/train.py
      md5: c8aa30276c495fa0dd495a712aa16576
      size: 9248
    - path: src/data/sprites.py
      md5: debea5b0298af285804f6917ff812887
      size: 11205
    - path: src/utils.py
      md5: 6b884894e60db1e2b61da9e2510b5f44
      size: 13515
    outs:
    - path: models/vae.pt
      md5: 3bdc0bc7140c51a6395257c3c8ca7e64
      size: 18930698
    - path: models/vae_kwargs.yaml
      md5: 24aef72315932f23c5ec72f437c5bcfc
      size: 124
    - path: reports/vae/vae/gen
      md5: 7f1ddbfd3185d433b3c10174263fc181.dir
      size: 2233172
      nfiles: 520
    - path: reports/vae/vae/latent
      md5: 3289613fe8079998bdda2943121b5394.dir
      size: 219142
      nfiles: 6
    - path: reports/vae/vae/logs
      md5: 39649c378c71efc7e0c7701360825973.dir
      size: 88383
      nfiles: 5
    - path: reports/vae/vae/logs.json
      md5: e3b2bda046ba26f36df3defc9c3e258c
      size: 173
    - path: reports/vae/vae/results
      md5: 6df0ac7fbcdebb4ff60f1a935c2931a0.dir
      size: 17346
      nfiles: 16
    - path: reports/vae/vae/summary.txt
      md5: 1b28a71b74d86dc2445801ba23b6da03
      size: 2438