diff --git a/creator/dvc.lock b/creator/dvc.lock index b9e1540..d3f9e76 100644 --- a/creator/dvc.lock +++ b/creator/dvc.lock @@ -10,30 +10,30 @@ stages: params.yaml: train.batch_size: 32 train.custom_loss: true - train.dancers: maria - train.epochs: 1 - train.look_back: 10 - train.lstm_layer: 32 - train.mdn_layer: 3 + train.dancers: all + train.epochs: 10 + train.look_back: 70 + train.lstm_layer: 128 + train.mdn_layer: 5 train.tags: impro - train.validation_split: 0.8 + train.validation_split: 0.3 outs: - path: ../data/metrics/deep-dance-loss.json - md5: 2155ee8a3302a9b2f84205c3d116bc1e - size: 127 + md5: 7d27a97224cef73232884e99f296b32d + size: 1069 - path: ../data/metrics/deep-dance-scores.json - md5: c33ecdbd0a9d444e04ce4f8c2b973272 - size: 79 + md5: 8c6119531a271e1ad0c1f1f8b2c5d75d + size: 80 - path: ../data/models - md5: 1c45a89bf8ac228f24d996b04b3744dc.dir - size: 535704 + md5: 23c86df5c65b17219dd96560c0823d59.dir + size: 5331768 nfiles: 1 generate: cmd: python src/generate.py deps: - path: ../data/models - md5: 9024a2e0efc48d91e7a4ca82a71ad2fe.dir - size: 535704 + md5: 23c86df5c65b17219dd96560c0823d59.dir + size: 5331768 nfiles: 1 - path: src/generate.py md5: 923a628ff5fcbaed9e01f35f397ebb27 @@ -41,13 +41,13 @@ stages: params: params.yaml: generate.custom_loss: true - generate.look_back: 10 + generate.look_back: 70 generate.random_state: 42 generate.seed: 1500 generate.steps_limit: 1500 - generate.validation_split: 0.7 + generate.validation_split: 0.3 outs: - path: ../data/generated - md5: 9d3d915de708ebaeb2521fe7847e8306.dir - size: 1620568 + md5: d923df9eb1c4b3d0d96822dbd1a9b3fb.dir + size: 1655384 nfiles: 1 diff --git a/creator/exp_run.sh b/creator/exp_run.sh index a751d3a..b7de939 100644 --- a/creator/exp_run.sh +++ b/creator/exp_run.sh @@ -32,6 +32,7 @@ dvc exp run --queue -S train.epochs=10 -S train.batch_size=32 -S train.look_back dvc exp run --queue -S train.epochs=10 -S train.batch_size=32 -S train.look_back=10 -S train.lstm_layer=128 -S train.mdn_layer=5 -S train.dancers=all -S generate.look_back=10 -S generate.dancers=all dvc exp run --run-all --jobs 1 + # Run queued experiments # --------------------------- # Or run in parallel diff --git a/creator/params.yaml b/creator/params.yaml index 473cf45..049740f 100644 --- a/creator/params.yaml +++ b/creator/params.yaml @@ -6,25 +6,25 @@ prepare: tags: impro train: - epochs: 1 + epochs: 10 batch_size: 32 - look_back: 10 - lstm_layer: 32 - mdn_layer: 3 + look_back: 70 + lstm_layer: 128 + mdn_layer: 5 custom_loss: true random_state: 42 - dancers: maria + dancers: all tags: impro test_size: 0.05 - validation_split: 0.9 + validation_split: 0.3 generate: - look_back: 10 + look_back: 70 random_state: 42 seed: 1500 steps_limit: 1500 custom_loss: true - dancers: maria + dancers: all tags: impro test_size: 0.05 - validation_split: 0.9 + validation_split: 0.3 diff --git a/data/metrics/deep-dance-loss.json b/data/metrics/deep-dance-loss.json index 1d265f3..fdbaaa1 100644 --- a/data/metrics/deep-dance-loss.json +++ b/data/metrics/deep-dance-loss.json @@ -1,8 +1,44 @@ { "perf": [ { - "loss": -24.531362533569336, - "val_loss": 3.6198043823242188 + "loss": -102.2507095336914, + "val_loss": -111.24691772460938 + }, + { + "loss": -133.21434020996094, + "val_loss": -137.85562133789062 + }, + { + "loss": -145.06207275390625, + "val_loss": -137.01634216308594 + }, + { + "loss": -140.25210571289062, + "val_loss": -144.203125 + }, + { + "loss": -154.2703857421875, + "val_loss": -144.431396484375 + }, + { + "loss": -158.3099365234375, + "val_loss": -149.65609741210938 + }, + { + "loss": -161.09478759765625, + "val_loss": -150.35093688964844 + }, + { + "loss": -142.21139526367188, + "val_loss": -111.67668151855469 + }, + { + "loss": -136.19578552246094, + "val_loss": -124.02241516113281 + }, + { + "loss": -137.40200805664062, + "val_loss": -117.71631622314453 } ] } \ No newline at end of file diff --git a/data/metrics/deep-dance-scores.json b/data/metrics/deep-dance-scores.json index 71d241c..954c7da 100644 --- a/data/metrics/deep-dance-scores.json +++ b/data/metrics/deep-dance-scores.json @@ -1,4 +1,4 @@ { - "end_loss": -24.531362533569336, - "end_val_loss": 3.6198043823242188 + "end_loss": -137.40200805664062, + "end_val_loss": -117.71631622314453 } \ No newline at end of file diff --git a/detector/VideoPose3D b/detector/VideoPose3D deleted file mode 160000 index d988b3a..0000000 --- a/detector/VideoPose3D +++ /dev/null @@ -1 +0,0 @@ -Subproject commit d988b3afa6563bb4451410b2a2581de255868f85 diff --git a/detector/detectron2 b/detector/detectron2 deleted file mode 160000 index 123366f..0000000 --- a/detector/detectron2 +++ /dev/null @@ -1 +0,0 @@ -Subproject commit 123366fe7fd412963bee6404c69c1e06a5cdf06c