File size: 228 Bytes
bdce880 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | python exp_ns.py \
--gpu 2 \
--model Transolver_Structured_Mesh_2D \
--n-hidden 256 \
--n-heads 8 \
--n-layers 8 \
--lr 0.001 \
--batch-size 2 \
--slice_num 32 \
--unified_pos 1 \
--ref 8 \
--eval 0 \
--save_name ns_Transolver
|