-
Notifications
You must be signed in to change notification settings - Fork 5
/
running.sh
35 lines (24 loc) · 1.68 KB
/
running.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
###
# @Author: Shuangchi He / Yulv
# @Email: [email protected]
# @Date: 2022-04-05 20:56:19
# @Motto: Entities should not be multiplied unnecessarily.
# @LastEditors: Shuangchi He
# @LastEditTime: 2022-04-08 23:48:29
# @FilePath: /Awesome-Ultrasound-Standard-Plane-Detection/code/DCNN-MF-SP/running.sh
# @Description:
###
CUDA_VISIBLE_DEVICES=0 python main.py --model_type base_model --model_name base_model
CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name VGG19
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name VGG19 --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name MobileNet
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name MobileNet --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name InceptionV3
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name InceptionV3 --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name ResNet50
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name ResNet50 --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name DenseNet121 --batch_size 48
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type DCNN --model_name DenseNet121 --batch_size 48 --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type EfficientNet --model_name EfficientNetB6 --batch_size 8
# CUDA_VISIBLE_DEVICES=1 python main.py --model_type EfficientNet --model_name EfficientNetB6 --batch_size 8 --imagenet_pretrained True
CUDA_VISIBLE_DEVICES=1 python main.py --model_type ViT --model_name ViT --batch_size 4