npvinHnivqn commited on
Commit
f9f92f3
·
verified ·
1 Parent(s): cb12c20

library_name: transformers
tags: []
---

## Original result
```
IoU metric: bbox
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.001
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.003
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.008
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.008
```

## After training result
```
IoU metric: bbox
Average Precision (AP) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.75 | area= all | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Precision (AP) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 1 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets= 10 ] = 0.006
Average Recall (AR) @[ IoU=0.50:0.95 | area= all | maxDets=100 ] = 0.015
Average Recall (AR) @[ IoU=0.50:0.95 | area= small | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area=medium | maxDets=100 ] = 0.000
Average Recall (AR) @[ IoU=0.50:0.95 | area= large | maxDets=100 ] = 0.017
```

## Config
- dataset: NIH
- original model: facebook/detr-resnet-50
- lr: 0.0001
- max_epochs: 1

## Logging

### Training process

{'training_loss': tensor(2.9624, device='cuda:0'), 'train_loss_ce': tensor(0.4469, device='cuda:0'), 'train_loss_bbox': tensor(0.2411, device='cuda:0'), 'train_loss_giou': tensor(0.6551, device='cuda:0'), 'train_cardinality_error': tensor(1.1250, device='cuda:0'), 'validation_loss': tensor(2.4818, device='cuda:0'), 'validation_loss_ce': tensor(0.5116, device='cuda:0'), 'validation_loss_bbox': tensor(0.1740, device='cuda:0'), 'validation_loss_giou': tensor(0.5502, device='cuda:0'), 'validation_cardinality_error': tensor(1.0955, device='cuda:0')}

### Validation process
{'validation_loss': tensor(5.8176, device='cuda:0'), 'validation_loss_ce': tensor(2.3980, device='cuda:0'), 'validation_loss_bbox': tensor(0.4030, device='cuda:0'), 'validation_loss_giou': tensor(0.7024, device='cuda:0'), 'validation_cardinality_error': tensor(98.5312, device='cuda:0')}
{'training_loss': tensor(2.9624, device='cuda:0'), 'train_loss_ce': tensor(0.4469, device='cuda:0'), 'train_loss_bbox': tensor(0.2411, device='cuda:0'), 'train_loss_giou': tensor(0.6551, device='cuda:0'), 'train_cardinality_error': tensor(1.1250, device='cuda:0'), 'validation_loss': tensor(2.4818, device='cuda:0'), 'validation_loss_ce': tensor(0.5116, device='cuda:0'), 'validation_loss_bbox': tensor(0.1740, device='cuda:0'), 'validation_loss_giou': tensor(0.5502, device='cuda:0'), 'validation_cardinality_error': tensor(1.0955, device='cuda:0')}

Files changed (1) hide show
  1. model.safetensors +1 -1
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8b4056303d99d545c0f701d46ee086eade368026600f32de82a1a60731c2dc7b
3
  size 166510924
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c8181776ff898a19a1432c48c09c44d464001e1e0315076f3266e164589baedf
3
  size 166510924