| { |
| "data": { |
| "motion_loader": { |
| "_target_": "src.data.motion.AMASSMotionLoader", |
| "base_dir": "datasets/motions/guoh3dfeats", |
| "normalizer": { |
| "_target_": "src.data.motion.Normalizer", |
| "base_dir": "stats/amass-part-20/guoh3dfeats", |
| "eps": 1e-12 |
| }, |
| "fps": 20.0, |
| "nfeats": 263 |
| }, |
| "_target_": "src.data.text_motion.TextMotionDataset", |
| "path": "datasets/annotations/amass-part-20", |
| "text_to_token_emb": { |
| "_target_": "src.data.text.TokenEmbeddings", |
| "path": "datasets/annotations/amass-part-20", |
| "modelname": "distilbert-base-uncased", |
| "modelpath": null, |
| "preload": true |
| }, |
| "text_to_sent_emb": { |
| "_target_": "src.data.text.SentenceEmbeddings", |
| "path": "datasets/annotations/amass-part-20", |
| "modelname": "sentence-transformers/all-mpnet-base-v2", |
| "modelpath": null, |
| "preload": true |
| }, |
| "preload": true, |
| "min_seconds": 1.0, |
| "max_seconds": 20.0, |
| "part": "sequence_caption" |
| }, |
| "model": { |
| "_target_": "src.model.TMR", |
| "motion_encoder": { |
| "_target_": "src.model.ACTORStyleEncoder", |
| "nfeats": 263, |
| "vae": true, |
| "latent_dim": 256, |
| "ff_size": 1024, |
| "num_layers": 6, |
| "num_heads": 4, |
| "dropout": 0.1, |
| "activation": "gelu" |
| }, |
| "text_encoder": { |
| "_target_": "src.model.ACTORStyleEncoder", |
| "nfeats": 768, |
| "vae": true, |
| "latent_dim": 256, |
| "ff_size": 1024, |
| "num_layers": 6, |
| "num_heads": 4, |
| "dropout": 0.1, |
| "activation": "gelu" |
| }, |
| "motion_decoder": { |
| "_target_": "src.model.ACTORStyleDecoder", |
| "nfeats": 263, |
| "latent_dim": 256, |
| "ff_size": 1024, |
| "num_layers": 6, |
| "num_heads": 4, |
| "dropout": 0.1, |
| "activation": "gelu" |
| }, |
| "vae": true, |
| "lmd": { |
| "recons": 1.0, |
| "latent": 1e-05, |
| "kl": 1e-05, |
| "contrastive": 0.1 |
| }, |
| "lr": 0.0001, |
| "threshold_selfsim_metrics": 0.95, |
| "contrastive_loss": { |
| "_target_": "src.model.losses.InfoNCE_with_filtering", |
| "temperature": 0.1, |
| "threshold_selfsim": 0.8 |
| } |
| }, |
| "trainer": { |
| "_target_": "pytorch_lightning.Trainer", |
| "max_epochs": 500, |
| "log_every_n_steps": 50, |
| "num_sanity_val_steps": 0, |
| "check_val_every_n_epoch": 1, |
| "accelerator": "gpu", |
| "devices": 1, |
| "callbacks": [ |
| { |
| "_target_": "pytorch_lightning.callbacks.ModelCheckpoint", |
| "filename": "latest-{epoch}", |
| "every_n_epochs": 1, |
| "save_top_k": 1, |
| "save_last": true |
| }, |
| { |
| "_target_": "pytorch_lightning.callbacks.ModelCheckpoint", |
| "filename": "latest-{epoch}", |
| "monitor": "step", |
| "mode": "max", |
| "every_n_epochs": 100, |
| "save_top_k": -1, |
| "save_last": false |
| }, |
| { |
| "_target_": "src.callback.progress.ProgressLogger", |
| "precision": 3 |
| }, |
| { |
| "_target_": "src.callback.tqdmbar.TQDMProgressBar" |
| } |
| ], |
| "logger": [ |
| { |
| "_target_": "src.logger.csv.CSVLogger", |
| "save_dir": "outputs/tmr_amass-part-20_guoh3dfeats_sequence_caption", |
| "name": "logs" |
| }, |
| { |
| "_target_": "pytorch_lightning.loggers.WandbLogger", |
| "entity": "coral79", |
| "project": "motion-evaluator", |
| "name": "amass-part-20_sequence_caption_0920_0111", |
| "save_dir": "outputs/tmr_amass-part-20_guoh3dfeats_sequence_caption", |
| "offline": false, |
| "tags": [ |
| "evaluation", |
| "amass-part-20", |
| "sequence_caption" |
| ] |
| } |
| ] |
| }, |
| "run_dir": "outputs/tmr_amass-part-20_guoh3dfeats_sequence_caption", |
| "seed": 1234, |
| "logger_level": "INFO", |
| "ckpt": "last", |
| "resume_dir": null, |
| "ckpt_path": null, |
| "dataloader": { |
| "_target_": "torch.utils.data.DataLoader", |
| "batch_size": 32, |
| "num_workers": 8 |
| } |
| } |
