-
Notifications
You must be signed in to change notification settings - Fork 0
/
auto_trainv1.py
58 lines (50 loc) · 1.43 KB
/
auto_trainv1.py
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
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
import os
import shutil
import torch
from torchvision.models.resnet import resnet152
from torch.nn import DataParallel
import time
import sys
def run_msg(msg):
print(msg)
os.system(msg)
time.sleep(3)
try:
run_msg(
"python3 mae_finetune_base_affectnet.py "
"--exp 0507_maskfeat599_base "
"--dataset affectnet7 "
"--data_path /raid/libraboli/FER/small/ "
"--output_dir /raid/libraboli/FER/log_results "
"--batch_size 640 "
"--finetune /raid/libraboli/FER/pretrain_models/MaskFeat_40//checkpoint-599.pth"
)
run_msg(
"python3 mae_finetune_rcloss_affectnet.py "
"--exp 0507_maskfeat599_rc "
"--dataset affectnet7 "
"--data_path /raid/libraboli/FER/small/ "
"--output_dir /raid/libraboli/FER/log_results "
"--batch_size 640 "
"--finetune /raid/libraboli/FER/pretrain_models/MaskFeat_40//checkpoint-599.pth --labelset "
)
run_msg(
"python3 mae_finetune_base.py "
"--exp 0507_mae599_base "
"--dataset affectnet7 "
"--data_path /raid/libraboli/FER/small/ "
"--output_dir /raid/libraboli/FER/log_results "
"--batch_size 640 "
"--finetune /raid/libraboli/FER/pretrain_models/MAE//checkpoint-599.pth"
)
run_msg(
"python3 mae_finetune_rcloss_affectnet.py "
"--exp 0507_mae599_rc "
"--dataset affectnet7 "
"--data_path /raid/libraboli/FER/small/ "
"--output_dir /raid/libraboli/FER/log_results "
"--batch_size 640 "
"--finetune /raid/libraboli/FER/pretrain_models/MAE/checkpoint-599.pth"
)
except:
pass