-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsystem-config.yaml
167 lines (158 loc) · 3.91 KB
/
system-config.yaml
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
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
# File containing all system related information (things to be configured only once)
# TODO: Think about everything that needs to be configured
# TODO: Specify the format of the file
# NOTE: Local execution must be performed using the following command:
# torchrun --nproc_per_node=4 main.py
system: server # (server | local)
ddp:
port: 29500 # port for ddp setup
models:
dummy:
torch.hub.load:
repo:
model:
ann:
torch.hub.load:
repo:
model:
deeper-ann:
torch.hub.load:
repo:
model:
alexnet:
torch.hub.load:
repo: pytorch/vision:v0.10.0
model: alexnet
resnet18:
torch.hub.load:
repo: pytorch/vision:v0.10.0
model: resnet18
resnet50:
torch.hub.load:
repo: pytorch/vision:v0.10.0
model: resnet50
mobilenet_v3_small:
torch.hub.load:
repo: pytorch/vision:v0.10.0
model: mobilenet_v3_small
deit-base:
torch.hub.load:
repo: facebookresearch/deit:main
model: deit_base_patch16_224
deit-small:
torch.hub.load:
repo: facebookresearch/deit:main
model: deit_small_patch16_224
# Describe datasets that are available
datasets:
ImageNet:
path: /hkfs/home/dataset/datasets/imagenet-2012/original/imagenet-raw/ILSVRC/Data/CLS-LOC
num-classes: 1000
load-function:
module: torchvision.datasets
type: imagenet # (generic | built-in | imagenet)
name: ImageFolder
transforms:
train:
- name: RandomResizedCrop
kwargs:
size: [224, 224]
- name: RandomHorizontalFlip
kwargs: {}
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
test:
- name: Resize
kwargs:
size: [256, 256]
- name: CenterCrop
kwargs:
size: [224, 224]
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.485, 0.456, 0.406]
std: [0.229, 0.224, 0.225]
CIFAR10:
path: /hkfs/home/dataset/datasets/CIFAR10
num-classes: 10
load-function:
module: torchvision.datasets
type: built-in
name: CIFAR10
transforms:
train:
- name: Resize
kwargs:
size: [70, 70]
- name: RandomCrop
kwargs:
size: [64, 64]
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.4914, 0.4822, 0.4465]
std: [0.247, 0.243, 0.261]
test:
- name: Resize
kwargs:
size: [70, 70]
- name: CenterCrop
kwargs:
size: [64, 64]
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.4914, 0.4822, 0.4465]
std: [0.247, 0.243, 0.261]
CIFAR100:
path: /hkfs/home/dataset/datasets/CIFAR100/train
num-classes: 100
load-function:
module: torchvision.datasets
type: built-in
name: CIFAR100
transforms:
train:
- name: Resize
kwargs:
size: [70, 70]
- name: RandomCrop
kwargs:
size: [64, 64]
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.4914, 0.4822, 0.4465]
std: [0.247, 0.243, 0.261]
test:
- name: Resize
kwargs:
size: [70, 70]
- name: CenterCrop
kwargs:
size: [64, 64]
- name: ToTensor
kwargs: {}
- name: Normalize
kwargs:
mean: [0.4914, 0.4822, 0.4465]
std: [0.247, 0.243, 0.261]
SUSY:
path: /hkfs/home/dataset/datasets/susy
num-classes: 2
load-function:
module: src.data.datasets
type: built-in
name: SUSYDataset
transforms:
train:
test: