--data_path /data/zzhao/uav_regression/mainFlow/fushion/data_tasks.npy --data_label_path /data/zzhao/uav_regression/mainFlow/fushion/data_subnet.npy --init_path /data/zzhao/uav_regression/mainFlow/fushion/data_init.npy --label_path /data/zzhao/uav_regression/mainFlow/fushion/label_mainnet.npy --lr 0.005 --momentum 0.9 --weight_decay 0.0001 --batch_size 4 --num_epochs 30 --split_ratio 0.9 --checkpoint_dir /home/zzhao/data/uav_regression/check_point --model_checkpoint_name snet_main_flow_2D --image_save_folder mainFlow