File size: 269 Bytes
bdce880 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | python exp_darcy.py \
--gpu 4 \
--model Transolver_Structured_Mesh_2D \
--n-hidden 128 \
--n-heads 8 \
--n-layers 8 \
--lr 0.001 \
--max_grad_norm 0.1 \
--batch-size 4 \
--slice_num 64 \
--unified_pos 1 \
--ref 8 \
--eval 0 \
--downsample 5 \
--save_name darcy_UniPDE
|