{
    "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
    }
}