mmenendezg commited on
Commit
d475eba
1 Parent(s): 90c2cd7

DeTR successfully trained!

Browse files
README.md CHANGED
@@ -32,8 +32,8 @@ More information needed
32
  ### Training hyperparameters
33
 
34
  The following hyperparameters were used during training:
35
- - learning_rate: 1e-05
36
- - train_batch_size: 16
37
  - eval_batch_size: 8
38
  - seed: 42
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 
32
  ### Training hyperparameters
33
 
34
  The following hyperparameters were used during training:
35
+ - learning_rate: 9.764150575493936e-06
36
+ - train_batch_size: 4
37
  - eval_batch_size: 8
38
  - seed: 42
39
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
preprocessor_config.json CHANGED
@@ -3,7 +3,6 @@
3
  "do_pad": true,
4
  "do_rescale": true,
5
  "do_resize": true,
6
- "feature_extractor_type": "DetrFeatureExtractor",
7
  "format": "coco_detection",
8
  "image_mean": [
9
  0.485,
 
3
  "do_pad": true,
4
  "do_rescale": true,
5
  "do_resize": true,
 
6
  "format": "coco_detection",
7
  "image_mean": [
8
  0.485,
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:99f9b2a082ac82cf5a7fde48779f3a0351600c85dce24528913c63aaeeaf4233
3
  size 166608277
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ef0c46d2ea8c04dabb73712cca6027787fefc7731d9cc84a8f2bd24191ac861
3
  size 166608277
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:6d13815816776e268cf313688dc811a3e8d3a7abbccbcccc519253ae3e330846
3
  size 4027
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:76c064ca28baaefc67de052ae6054d436c610947efbfb7e323297d72d29a3459
3
  size 4027