python3 train_mlp.py \ --multiprocessing-distributed \ --num_thread_reader 8 \ --cudnn_benchmark 1 \ --pin_memory \ --checkpoint_dir whl \ --batch_size 256 \ --batch_size_val 256 \ --evaluate \ --dataset crosstask \ --resume \ --horizon 3 \ --action_dim 105 \ --class_dim 18 \ --observation_dim 1536 \ --json_path_val dataset/crosstask/crosstask_release/test_split_T3.json \ --json_path_train dataset/crosstask/crosstask_release/train_split_T3.json