BlidReview's picture
weights, code, eval script
bdce880 verified
Raw
History Blame Contribute Delete
270 Bytes
python exp_pipe.py \
--gpu 7 \
--model Transolver_Structured_Mesh_2D \
--n-hidden 128 \
--n-heads 8 \
--n-layers 8 \
--mlp_ratio 2 \
--lr 0.001 \
--max_grad_norm 0.1 \
--batch-size 8 \
--slice_num 64 \
--unified_pos 0 \
--ref 8 \
--eval 0 \
--save_name pipe_Transolver