kkirchheim commited on
Commit
757c3fc
·
verified ·
1 Parent(s): 7f68701

Add VOS model (resnet-18/imagenet200/vos/s2)

Browse files
resnet-18/imagenet200/vos/s2/config.yaml ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ model:
2
+ _target_: pytorch_ood.model.ResNet18
3
+ arch_name: resnet-18
4
+ num_classes: 200
5
+ weights: null
6
+ dataset:
7
+ name: imagenet200
8
+ kind: imagenet
9
+ num_classes: 200
10
+ root: /data/fast/scratch/imagenet-2012
11
+ mean:
12
+ - 0.485
13
+ - 0.456
14
+ - 0.406
15
+ std:
16
+ - 0.229
17
+ - 0.224
18
+ - 0.225
19
+ image_size: 224
20
+ pre_size: 256
21
+ loss:
22
+ _target_: pytorch_ood.loss.VOSRegLoss
23
+ method_name: vos
24
+ alpha: 0.1
25
+ device: cuda:0
26
+ optimizer:
27
+ _target_: torch.optim.SGD
28
+ lr: 0.001
29
+ momentum: 0.9
30
+ weight_decay: 0.0005
31
+ nesterov: true
32
+ outliers:
33
+ name: imagenet800
34
+ _target_: pytorch_ood.dataset.img.ImageNet800
35
+ root: /data/fast/scratch/imagenet-2012
36
+ seed: 2
37
+ deterministic: true
38
+ imagenet_root: /data/fast/scratch/imagenet-2012
39
+ init_from: resnet-18/imagenet200/crossentropy/s2
40
+ epochs: 10
41
+ batch_size: 128
42
+ oe_batch_size: 256
43
+ num_workers: 8
44
+ device: cuda:0
45
+ scheduler:
46
+ _target_: scheduler.per_iteration_cosine
47
+ lr_min: 1.0e-06
resnet-18/imagenet200/vos/s2/metrics.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "final_accuracy": 0.8389999866485596,
3
+ "best_accuracy": 0.843999981880188
4
+ }
resnet-18/imagenet200/vos/s2/model-8f9642c7f6d66d39.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8f9642c7f6d66d39658e36bcc898c25e55cf24dd04434fdc27860d86709ad58f
3
+ size 45184322