{ "best_config": { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.5, "smooth": 0.0 }, "per": 10.122729029764205 }, "all_results": [ { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.0, "smooth": 0.0 }, "per": 10.45129493621956 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.0, "smooth": 0.5 }, "per": 10.794356397371473 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.5, "smooth": 0.0 }, "per": 11.620603015075377 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.5, "smooth": 0.5 }, "per": 11.850115964437572 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.0, "smooth": 0.0 }, "per": 10.470622342481638 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.0, "smooth": 0.5 }, "per": 10.760533436412834 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.5, "smooth": 0.0 }, "per": 11.42974487823734 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.5, "smooth": 0.5 }, "per": 11.569868573637418 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.0, "smooth": 0.0 }, "per": 10.463374565133359 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.0, "smooth": 0.5 }, "per": 10.772613065326633 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.5, "smooth": 0.0 }, "per": 11.42249710088906 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.0, "smooth": 0.0 }, "per": 10.485117897178199 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.5, "smooth": 0.5 }, "per": 11.565036722071898 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.0, "smooth": 0.5 }, "per": 10.753285659064554 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.0, "smooth": 0.0 }, "per": 10.183127174333205 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.5, "smooth": 0.0 }, "per": 11.379010436799382 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.0, "smooth": 0.5 }, "per": 10.183127174333205 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.5, "smooth": 0.0 }, "per": 10.134808658678006 }, { "gru_weight": 0.8, "lstm_weight": 0.19999999999999996, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.5, "smooth": 0.5 }, "per": 11.48289524545806 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.0, "smooth": 0.0 }, "per": 10.173463471202165 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.0, "shift": 0.5, "smooth": 0.5 }, "per": 10.41747197526092 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.0, "smooth": 0.5 }, "per": 10.168631619636646 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.5, "smooth": 0.0 }, "per": 10.122729029764205 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.0, "smooth": 0.0 }, "per": 10.185543100115964 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.0, "scale": 0.5, "shift": 0.5, "smooth": 0.5 }, "per": 10.190374951681484 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.0, "smooth": 0.5 }, "per": 10.200038654812523 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.5, "smooth": 0.0 }, "per": 10.127560881329726 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.0, "shift": 0.5, "smooth": 0.5 }, "per": 10.192790877464244 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.0, "smooth": 0.0 }, "per": 10.168631619636646 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.0, "smooth": 0.5 }, "per": 10.183127174333205 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.5, "smooth": 0.0 }, "per": 10.137224584460766 }, { "gru_weight": 1.0, "lstm_weight": 0.0, "tta_weights": { "original": 1.0, "noise": 0.5, "scale": 0.5, "shift": 0.5, "smooth": 0.5 }, "per": 10.161383842288364 } ], "args": { "gru_model_path": "/root/autodl-tmp/nejm-brain-to-text/data/t15_pretrained_rnn_baseline", "lstm_model_path": "/root/autodl-tmp/nejm-brain-to-text/model_training_lstm/trained_models/baseline_rnn", "data_dir": "../data/hdf5_data_final", "csv_path": "../data/t15_copyTaskData_description.csv", "eval_type": "val", "gpu_number": 0, "gru_weights": "0.8,1.0", "tta_noise_weights": "0.0,0.5", "tta_scale_weights": "0.0,0.5", "tta_shift_weights": "0.0,0.5", "tta_smooth_weights": "0.0,0.5", "tta_noise_std": 0.01, "tta_smooth_range": 0.5, "tta_scale_range": 0.05, "tta_cut_max": 3, "cache_file": "tta_cache.pkl", "force_recache": false, "output_file": "test_results.json", "max_workers": 4, "batch_size": 100 }, "timestamp": "2025-09-17 22:25:20" }