SharoonArshad's picture
Upload trained ASL Transformer 84-class 408D model
30609f7 verified
Raw
History Blame Contribute Delete
641 Bytes
{
"data_dir": "/kaggle/working/training_model23_final_train_ready",
"output_dir": "/kaggle/working/asl_transformer_84class_run",
"seq_len": 50,
"input_dim": 408,
"num_classes": 84,
"d_model": 256,
"nhead": 4,
"num_layers": 3,
"dim_feedforward": 512,
"dropout": 0.3,
"batch_size": 64,
"epochs": 60,
"max_lr": 0.0006,
"weight_decay": 0.0001,
"label_smoothing": 0.05,
"grad_clip": 1.0,
"patience": 10,
"num_workers": 2,
"pin_memory": true,
"use_augmentation": true,
"noise_std": 0.01,
"feature_dropout_prob": 0.03,
"time_mask_prob": 0.15,
"time_mask_max_len": 6,
"seed": 42,
"use_amp": true
}