This repository has been archived by the owner on Mar 19, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 330
/
benchmark_in1k_linear_simclr.json
176 lines (176 loc) · 7.66 KB
/
benchmark_in1k_linear_simclr.json
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
168
169
170
171
172
173
174
175
176
{
"params": {
"benchmarks": [
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=50",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50_100ep_simclr_8node_resnet_16_07_20.8edb093e/model_final_checkpoint_phase99.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_in1k_simclr_100ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=50",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50_200ep_simclr_8node_resnet_16_07_20.a816c0ef/model_final_checkpoint_phase199.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_in1k_simclr_200ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=50",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50_400ep_simclr_8node_resnet_16_07_20.36b338ef/model_final_checkpoint_phase399.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_in1k_simclr_400ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=50",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50_800ep_simclr_8node_resnet_16_07_20.7e8feed1/model_final_checkpoint_phase799.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_in1k_simclr_800ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=50",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50_1000ep_simclr_8node_resnet_16_07_20.afe428c7/model_final_checkpoint_phase999.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_in1k_simclr_1000ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"+config/benchmark/linear_image_classification/imagenet1k/models=resnet50_w2",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50w2_100ep_simclr_8node_resnet_16_07_20.05b37ec3/model_final_checkpoint_phase99.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_w2_in1k_simclr_100ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"+config/benchmark/linear_image_classification/imagenet1k/models=resnet50_w2",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50w2_1000ep_simclr_8node_resnet_16_07_20.e1e3bbf0/model_final_checkpoint_phase999.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_w2_in1k_simclr_1000ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"+config/benchmark/linear_image_classification/imagenet1k/models=resnet50_w4",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn50w4_1000ep_bs32_16node_simclr_8node_resnet_28_07_20.9e20b0ae/model_final_checkpoint_phase999.torch"
],
"workflow_type": "engine",
"expt_name": "rn50_w4_in1k_simclr_1000ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": "V100_32G"
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=101",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn101_100ep_simclr_8node_resnet_16_07_20.1ff6cb4b/model_final_checkpoint_phase99.torch"
],
"workflow_type": "engine",
"expt_name": "rn101_in1k_simclr_100ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
},
{
"config_files": [
"config=benchmark/linear_image_classification/imagenet1k/eval_resnet_8gpu_transfer_in1k_linear",
"config.MODEL.TRUNK.RESNETS.DEPTH=101",
"config.MODEL.WEIGHTS_INIT.STATE_DICT_KEY_NAME=classy_state_dict",
"config.MODEL.WEIGHTS_INIT.PARAMS_FILE=https://dl.fbaipublicfiles.com/vissl/model_zoo/simclr_rn101_1000ep_simclr_8node_resnet_16_07_20.35063cea/model_final_checkpoint_phase999.torch"
],
"workflow_type": "engine",
"expt_name": "rn101_in1k_simclr_1000ep",
"machine_config": {
"num_gpus": 8,
"memory": "110g",
"num_cpus": 48,
"num_nodes": 1,
"gpu_type": ""
}
}
]
}
}