| { | |
| "datasetCommonParams": { | |
| "dataset_root_dir": "../dataset", | |
| "mosi": { | |
| "aligned": { | |
| "featurePath": "MOSI/Processed/aligned_50.pkl", | |
| "feature_dims": [768, 5, 20], | |
| "train_samples": 1284, | |
| "num_classes": 3, | |
| "language": "en", | |
| "KeyEval": "Loss" | |
| }, | |
| "unaligned": { | |
| "featurePath": "MOSI/Processed/unaligned_50.pkl", | |
| "feature_dims": [768, 5, 20], | |
| "train_samples": 1284, | |
| "num_classes": 3, | |
| "language": "en", | |
| "KeyEval": "Loss" | |
| } | |
| }, | |
| "mosei": { | |
| "aligned": { | |
| "featurePath": "MOSEI/Processed/aligned_50.pkl", | |
| "feature_dims": [768, 74, 35], | |
| "train_samples": 16326, | |
| "num_classes": 3, | |
| "language": "en", | |
| "KeyEval": "Loss" | |
| }, | |
| "unaligned": { | |
| "featurePath": "MOSEI/Processed/unaligned_50.pkl", | |
| "feature_dims": [768, 74, 35], | |
| "train_samples": 16326, | |
| "num_classes": 3, | |
| "language": "en", | |
| "KeyEval": "Loss" | |
| } | |
| } | |
| }, | |
| "DLF": { | |
| "commonParams": { | |
| "need_data_aligned": true, | |
| "need_model_aligned": true, | |
| "early_stop": 10, | |
| "use_bert": true, | |
| "use_finetune": true, | |
| "attn_mask": true, | |
| "update_epochs": 10 | |
| }, | |
| "datasetParams": { | |
| "mosi": { | |
| "attn_dropout_a": 0.2, | |
| "attn_dropout_v": 0.0, | |
| "relu_dropout": 0.0, | |
| "embed_dropout": 0.2, | |
| "res_dropout": 0.0, | |
| "dst_feature_dim_nheads": [50, 10], | |
| "batch_size": 16, | |
| "learning_rate": 0.0001, | |
| "nlevels": 2, | |
| "conv1d_kernel_size_l": 5, | |
| "conv1d_kernel_size_a": 5, | |
| "conv1d_kernel_size_v": 5, | |
| "text_dropout": 0.5, | |
| "attn_dropout": 0.3, | |
| "output_dropout": 0.5, | |
| "grad_clip": 0.6, | |
| "patience": 5, | |
| "weight_decay": 0.005, | |
| "transformers": "bert", | |
| "pretrained": "bert-base-uncased" | |
| }, | |
| "mosei": { | |
| "attn_dropout_a": 0.0, | |
| "attn_dropout_v": 0.0, | |
| "relu_dropout": 0.0, | |
| "embed_dropout": 0.0, | |
| "res_dropout": 0.0, | |
| "dst_feature_dim_nheads": [50, 10], | |
| "batch_size": 16, | |
| "learning_rate": 0.0001, | |
| "nlevels": 2, | |
| "conv1d_kernel_size_l": 3, | |
| "conv1d_kernel_size_a": 3, | |
| "conv1d_kernel_size_v": 3, | |
| "text_dropout": 0.1, | |
| "attn_dropout": 0.5, | |
| "output_dropout": 0.5, | |
| "grad_clip": 0.6, | |
| "patience": 5, | |
| "weight_decay": 0.001, | |
| "transformers": "bert", | |
| "pretrained": "bert-base-uncased" | |
| } | |
| } | |
| } | |
| } |