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
|