-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathvit.sh
18 lines (7 loc) · 998 Bytes
/
vit.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
# python /root/武汉科技大学作业/num—9/homework-9/main.py --epoch 100 --batch_size 512 --num_layers 3
# python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 512 --epoch 100 --batch_size 512 --num_layers 3
# python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 256 --epoch 100 --batch_size 512 --num_layers 3
# python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 256 --epoch 100 --num_heads 4 --batch_size 512 --num_layers 3
# python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 256 --epoch 100 --num_heads 8 --batch_size 512 --num_layers 3
# python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 256 --epoch 100 --num_heads 8 --batch_size 512 --num_layers 8
python /root/武汉科技大学作业/num—9/homework-9/main.py --d_model 256 --epoch 100 --num_heads 8 --batch_size 2048 --num_layers 8 --learning_rate 0.0001