Model save

Browse files

Files changed (9) hide show

README.md +120 -0
all_results.json +54 -0
config.json +125 -0
model.safetensors +3 -0
preprocessor_config.json +29 -0
test_results.json +49 -0
train_results.json +8 -0
trainer_state.json +1182 -0
training_args.bin +3 -0

README.md ADDED Viewed

	@@ -0,0 +1,120 @@

+---
+library_name: transformers
+license: apache-2.0
+base_model: PekingU/rtdetr_r50vd
+tags:
+- generated_from_trainer
+model-index:
+- name: suas-2025-rtdetr-finetuned-b16-lr1e-4
+  results: []
+---
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+# suas-2025-rtdetr-finetuned-b16-lr1e-4
+This model is a fine-tuned version of [PekingU/rtdetr_r50vd](https://huggingface.co/PekingU/rtdetr_r50vd) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 5.5169
+- Map: 0.7975
+- Map 50: 0.9083
+- Map 75: 0.8523
+- Map Baseball-bat: 0.2923
+- Map Basketball: 0.8923
+- Map Car: -1.0
+- Map Football: 0.5264
+- Map Human: 0.9058
+- Map Large: 0.8257
+- Map Luggage: 0.8227
+- Map Mattress: 0.9768
+- Map Medium: 0.7526
+- Map Motorcycle: 0.7936
+- Map Skis: 0.8144
+- Map Small: 0.6611
+- Map Snowboard: 0.9197
+- Map Soccer-ball: 0.8408
+- Map Stop-sign: 0.9465
+- Map Tennis-racket: 0.6958
+- Map Umbrella: 0.8818
+- Map Volleyball: 0.8557
+- Mar 1: 0.7621
+- Mar 10: 0.8843
+- Mar 100: 0.8899
+- Mar 100 Baseball-bat: 0.6389
+- Mar 100 Basketball: 0.9235
+- Mar 100 Car: -1.0
+- Mar 100 Football: 0.6421
+- Mar 100 Human: 0.9585
+- Mar 100 Luggage: 0.9242
+- Mar 100 Mattress: 0.9944
+- Mar 100 Motorcycle: 0.957
+- Mar 100 Skis: 0.9614
+- Mar 100 Snowboard: 0.9525
+- Mar 100 Soccer-ball: 0.8735
+- Mar 100 Stop-sign: 0.9754
+- Mar 100 Tennis-racket: 0.8176
+- Mar 100 Umbrella: 0.9516
+- Mar 100 Volleyball: 0.8882
+- Mar Large: 0.9814
+- Mar Medium: 0.8664
+- Mar Small: 0.7154
+## Model description
+More information needed
+## Intended uses & limitations
+More information needed
+## Training and evaluation data
+More information needed
+## Training procedure
+### Training hyperparameters
+The following hyperparameters were used during training:
+- learning_rate: 0.0001
+- train_batch_size: 16
+- eval_batch_size: 16
+- seed: 1337
+- optimizer: Use adamw_torch with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
+- lr_scheduler_type: linear
+- num_epochs: 10.0
+- mixed_precision_training: Native AMP
+### Training results
+| Training Loss | Epoch | Step | Validation Loss | Map    | Map 50 | Map 75 | Map Baseball-bat | Map Basketball | Map Car | Map Football | Map Human | Map Large | Map Luggage | Map Mattress | Map Medium | Map Motorcycle | Map Skis | Map Small | Map Snowboard | Map Soccer-ball | Map Stop-sign | Map Tennis-racket | Map Umbrella | Map Volleyball | Mar 1  | Mar 10 | Mar 100 | Mar 100 Baseball-bat | Mar 100 Basketball | Mar 100 Car | Mar 100 Football | Mar 100 Human | Mar 100 Luggage | Mar 100 Mattress | Mar 100 Motorcycle | Mar 100 Skis | Mar 100 Snowboard | Mar 100 Soccer-ball | Mar 100 Stop-sign | Mar 100 Tennis-racket | Mar 100 Umbrella | Mar 100 Volleyball | Mar Large | Mar Medium | Mar Small |
+|:-------------:|:-----:|:----:|:---------------:|:------:|:------:|:------:|:----------------:|:--------------:|:-------:|:------------:|:---------:|:---------:|:-----------:|:------------:|:----------:|:--------------:|:--------:|:---------:|:-------------:|:---------------:|:-------------:|:-----------------:|:------------:|:--------------:|:------:|:------:|:-------:|:--------------------:|:------------------:|:-----------:|:----------------:|:-------------:|:---------------:|:----------------:|:------------------:|:------------:|:-----------------:|:-------------------:|:-----------------:|:---------------------:|:----------------:|:------------------:|:---------:|:----------:|:---------:|
+| 19.0657       | 1.0   | 219  | 5.6722          | 0.6892 | 0.7627 | 0.7569 | 0.6662           | 0.7899         | -1.0    | 0.1757       | 0.9219    | 0.8153    | 0.64        | 0.9656       | 0.7254     | 0.9096         | 0.4186   | 0.5623    | 0.5418        | 0.6531          | 0.9522        | 0.4801            | 0.9224       | 0.6124         | 0.7137 | 0.8856 | 0.8913  | 0.7836               | 0.8709             | -1.0        | 0.703            | 0.9551        | 0.9195          | 0.9945           | 0.9427             | 0.9871       | 0.9854            | 0.8679              | 0.9777            | 0.771                 | 0.954            | 0.7657             | 0.9702    | 0.9112     | 0.7427    |
+| 6.7339        | 2.0   | 438  | 5.6625          | 0.742  | 0.8333 | 0.8117 | 0.7301           | 0.787          | -1.0    | 0.3376       | 0.9024    | 0.8916    | 0.6101      | 0.9744       | 0.7798     | 0.8818         | 0.7438   | 0.5812    | 0.857         | 0.4723          | 0.9349        | 0.6734            | 0.884        | 0.5999         | 0.717  | 0.868  | 0.8722  | 0.8446               | 0.8614             | -1.0        | 0.6179           | 0.9545        | 0.9334          | 0.992            | 0.9245             | 0.9871       | 0.9869            | 0.6388              | 0.9578            | 0.8684                | 0.9388           | 0.7047             | 0.9659    | 0.9105     | 0.7034    |
+| 5.8898        | 3.0   | 657  | 4.3034          | 0.838  | 0.9206 | 0.9089 | 0.7363           | 0.8553         | -1.0    | 0.5598       | 0.9044    | 0.955     | 0.7481      | 0.9755       | 0.8569     | 0.9085         | 0.9331   | 0.7084    | 0.9585        | 0.8474          | 0.9861        | 0.7094            | 0.9098       | 0.6997         | 0.7942 | 0.9197 | 0.9236  | 0.8602               | 0.9017             | -1.0        | 0.7879           | 0.9702        | 0.9451          | 0.993            | 0.9622             | 0.9965       | 0.9945            | 0.9126              | 0.9936            | 0.8788                | 0.9515           | 0.7824             | 0.9788    | 0.9432     | 0.8018    |
+| 5.6445        | 4.0   | 876  | 4.7474          | 0.8119 | 0.9135 | 0.8838 | 0.6429           | 0.8321         | -1.0    | 0.6007       | 0.9296    | 0.8501    | 0.8492      | 0.9742       | 0.8402     | 0.9266         | 0.6816   | 0.642     | 0.9515        | 0.7901          | 0.9841        | 0.7362            | 0.9255       | 0.5431         | 0.7707 | 0.8935 | 0.8977  | 0.7837               | 0.8853             | -1.0        | 0.7635           | 0.9672        | 0.9507          | 0.9902           | 0.9637             | 0.9941       | 0.9953            | 0.853               | 0.993             | 0.8337                | 0.957            | 0.638              | 0.9918    | 0.9332     | 0.7188    |
+| 5.624         | 5.0   | 1095 | 4.6554          | 0.8246 | 0.9344 | 0.9007 | 0.6562           | 0.7609         | -1.0    | 0.5938       | 0.89      | 0.956     | 0.8797      | 0.9671       | 0.8605     | 0.9113         | 0.8424   | 0.6468    | 0.9674        | 0.8123          | 0.9741        | 0.7671            | 0.9256       | 0.5956         | 0.7729 | 0.8907 | 0.8939  | 0.7985               | 0.8355             | -1.0        | 0.7333           | 0.9497        | 0.9368          | 0.9806           | 0.9621             | 0.9856       | 0.9888            | 0.8548              | 0.986             | 0.8674                | 0.9523           | 0.6838             | 0.9866    | 0.9296     | 0.7144    |
+| 5.3591        | 6.0   | 1314 | 4.4766          | 0.8456 | 0.9325 | 0.9116 | 0.5066           | 0.8924         | -1.0    | 0.7547       | 0.9151    | 0.9694    | 0.8623      | 0.9712       | 0.7744     | 0.843          | 0.8972   | 0.7299    | 0.986         | 0.859           | 0.9637        | 0.6545            | 0.9286       | 0.8045         | 0.7963 | 0.9169 | 0.9202  | 0.8016               | 0.9287             | -1.0        | 0.8219           | 0.9692        | 0.9339          | 0.9858           | 0.9567             | 0.9797       | 0.9961            | 0.8943              | 0.9838            | 0.828                 | 0.9576           | 0.8451             | 0.9875    | 0.9297     | 0.7973    |
+| 5.1554        | 7.0   | 1533 | 5.4352          | 0.8049 | 0.9055 | 0.8645 | 0.506            | 0.8712         | -1.0    | 0.3751       | 0.8879    | 0.9469    | 0.8829      | 0.9662       | 0.8023     | 0.8948         | 0.8169   | 0.6245    | 0.9745        | 0.8381          | 0.9435        | 0.6993            | 0.8829       | 0.7293         | 0.7557 | 0.874  | 0.8768  | 0.726                | 0.9031             | -1.0        | 0.5281           | 0.953         | 0.9372          | 0.9822           | 0.9515             | 0.954        | 0.988             | 0.8728              | 0.9696            | 0.8031                | 0.9372           | 0.7696             | 0.9782    | 0.9096     | 0.6926    |
+| 5.1876        | 8.0   | 1752 | 4.8210          | 0.8373 | 0.9334 | 0.8999 | 0.5239           | 0.8396         | -1.0    | 0.636        | 0.915     | 0.9611    | 0.8947      | 0.9864       | 0.8289     | 0.9182         | 0.9147   | 0.6663    | 0.9729        | 0.8374          | 0.9701        | 0.6451            | 0.8846       | 0.783          | 0.7803 | 0.9013 | 0.9038  | 0.7665               | 0.8765             | -1.0        | 0.7182           | 0.9674        | 0.9511          | 0.9956           | 0.9654             | 0.9782       | 0.9883            | 0.8725              | 0.9874            | 0.8135                | 0.9473           | 0.8252             | 0.9883    | 0.9242     | 0.735     |
+| 4.9537        | 9.0   | 1971 | 4.7137          | 0.838  | 0.9376 | 0.9021 | 0.4846           | 0.8804         | -1.0    | 0.6231       | 0.9015    | 0.9476    | 0.8638      | 0.9507       | 0.8454     | 0.9206         | 0.898    | 0.6902    | 0.951         | 0.8526          | 0.978         | 0.7464            | 0.8486       | 0.8327         | 0.7894 | 0.9057 | 0.9086  | 0.738                | 0.9128             | -1.0        | 0.7469           | 0.9636        | 0.9232          | 0.9778           | 0.9648             | 0.9653       | 0.9741            | 0.8889              | 0.9925            | 0.858                 | 0.9418           | 0.8723             | 0.9834    | 0.9191     | 0.7541    |
+| 4.9224        | 10.0  | 2190 | 4.5997          | 0.8651 | 0.9611 | 0.9319 | 0.5229           | 0.8898         | -1.0    | 0.6333       | 0.9194    | 0.9665    | 0.8892      | 0.9732       | 0.8569     | 0.9065         | 0.9781   | 0.7128    | 0.9808        | 0.8364          | 0.9768        | 0.8425            | 0.9263       | 0.8363         | 0.8041 | 0.9164 | 0.9189  | 0.7741               | 0.9223             | -1.0        | 0.7222           | 0.9623        | 0.9477          | 0.995            | 0.9632             | 0.9931       | 0.9919            | 0.871               | 0.988             | 0.899                 | 0.9617           | 0.8728             | 0.9791    | 0.933      | 0.7708    |
+| 4.7256        | 11.0  | 2409 | 4.8142          | 0.8399 | 0.9462 | 0.9028 | 0.4084           | 0.8985         | -1.0    | 0.577        | 0.93      | 0.9667    | 0.9061      | 0.9792       | 0.8373     | 0.9326         | 0.8848   | 0.6739    | 0.9532        | 0.792           | 0.9554        | 0.8222            | 0.9108       | 0.8089         | 0.7824 | 0.8985 | 0.9028  | 0.6793               | 0.9339             | -1.0        | 0.6901           | 0.9718        | 0.9503          | 0.9933           | 0.9776             | 0.9594       | 0.9749            | 0.8324              | 0.9774            | 0.8829                | 0.9604           | 0.8549             | 0.9876    | 0.9168     | 0.734     |
+| 4.6993        | 12.0  | 2628 | 4.7597          | 0.8217 | 0.9297 | 0.8787 | 0.332            | 0.8934         | -1.0    | 0.4632       | 0.9409    | 0.9484    | 0.9098      | 0.954        | 0.8053     | 0.949          | 0.84     | 0.6514    | 0.9579        | 0.8407          | 0.9612        | 0.7405            | 0.9017       | 0.8189         | 0.7709 | 0.8931 | 0.8971  | 0.6972               | 0.9273             | -1.0        | 0.5876           | 0.9713        | 0.9524          | 0.98             | 0.9798             | 0.9604       | 0.9783            | 0.8748              | 0.9807            | 0.8492                | 0.961            | 0.8588             | 0.9739    | 0.9146     | 0.7196    |
+| 4.7266        | 13.0  | 2847 | 5.4190          | 0.7638 | 0.8926 | 0.8072 | 0.2568           | 0.8598         | -1.0    | 0.2283       | 0.9259    | 0.9286    | 0.8451      | 0.9628       | 0.7222     | 0.7728         | 0.8578   | 0.5811    | 0.8954        | 0.8364          | 0.9659        | 0.6612            | 0.7892       | 0.8364         | 0.7303 | 0.8583 | 0.8635  | 0.6368               | 0.8976             | -1.0        | 0.38             | 0.9653        | 0.9222          | 0.9822           | 0.9603             | 0.9535       | 0.9387            | 0.8686              | 0.9863            | 0.8057                | 0.9188           | 0.8735             | 0.9675    | 0.873      | 0.6569    |
+| 4.7442        | 14.0  | 3066 | 5.6973          | 0.7497 | 0.8723 | 0.7933 | 0.2398           | 0.8922         | -1.0    | 0.3593       | 0.9187    | 0.787     | 0.7688      | 0.88         | 0.6939     | 0.7618         | 0.7407   | 0.6262    | 0.8981        | 0.8576          | 0.9551        | 0.5716            | 0.8151       | 0.837          | 0.7227 | 0.8473 | 0.8528  | 0.5856               | 0.9204             | -1.0        | 0.4725           | 0.964         | 0.8604          | 0.9307           | 0.9673             | 0.9          | 0.9324            | 0.8869              | 0.9816            | 0.7316                | 0.9325           | 0.8733             | 0.8429    | 0.8393     | 0.6824    |
+| 4.5738        | 15.0  | 3285 | 5.4357          | 0.7908 | 0.9047 | 0.8408 | 0.2947           | 0.8977         | -1.0    | 0.4669       | 0.9032    | 0.9224    | 0.8532      | 0.9445       | 0.7435     | 0.8411         | 0.7558   | 0.6625    | 0.9395        | 0.8668          | 0.9471        | 0.6622            | 0.8522       | 0.8465         | 0.7527 | 0.8804 | 0.8853  | 0.6655               | 0.9291             | -1.0        | 0.5869           | 0.9612        | 0.9217          | 0.9773           | 0.9613             | 0.9347       | 0.9585            | 0.8933              | 0.9827            | 0.7984                | 0.9458           | 0.8777             | 0.9781    | 0.8732     | 0.7268    |
+| 4.4781        | 16.0  | 3504 | 5.6429          | 0.7929 | 0.9102 | 0.8519 | 0.3248           | 0.892          | -1.0    | 0.5374       | 0.9101    | 0.925     | 0.845       | 0.9494       | 0.7235     | 0.7888         | 0.7988   | 0.6634    | 0.9226        | 0.8371          | 0.9334        | 0.6908            | 0.8184       | 0.8523         | 0.7585 | 0.8821 | 0.8864  | 0.6622               | 0.9182             | -1.0        | 0.6483           | 0.9607        | 0.9234          | 0.9758           | 0.9597             | 0.9361       | 0.9571            | 0.8722              | 0.9718            | 0.8109                | 0.9299           | 0.8836             | 0.9743    | 0.8641     | 0.7226    |
+| 4.3981        | 17.0  | 3723 | 5.4522          | 0.8124 | 0.9224 | 0.8705 | 0.2978           | 0.8935         | -1.0    | 0.5512       | 0.9205    | 0.9528    | 0.8544      | 0.9725       | 0.7791     | 0.8652         | 0.8152   | 0.6647    | 0.9437        | 0.8396          | 0.9579        | 0.7453            | 0.8685       | 0.8482         | 0.7712 | 0.8924 | 0.8969  | 0.6459               | 0.9246             | -1.0        | 0.6725           | 0.9653        | 0.9346          | 0.9885           | 0.9694             | 0.945        | 0.9665            | 0.873               | 0.9757            | 0.8549                | 0.9541           | 0.8868             | 0.985     | 0.8836     | 0.725     |
+| 4.299         | 18.0  | 3942 | 5.5161          | 0.8209 | 0.9261 | 0.8788 | 0.3434           | 0.8958         | -1.0    | 0.5914       | 0.9229    | 0.9299    | 0.8504      | 0.9673       | 0.7851     | 0.8517         | 0.8319   | 0.6775    | 0.9347        | 0.8423          | 0.9591        | 0.7675            | 0.8792       | 0.8551         | 0.7741 | 0.8968 | 0.9012  | 0.6663               | 0.9246             | -1.0        | 0.6943           | 0.9677        | 0.9318          | 0.9861           | 0.9643             | 0.9599       | 0.9632            | 0.8748              | 0.9799            | 0.8637                | 0.9536           | 0.8858             | 0.9719    | 0.8805     | 0.7305    |
+| 4.3295        | 19.0  | 4161 | 5.5718          | 0.7992 | 0.9122 | 0.8546 | 0.2997           | 0.8944         | -1.0    | 0.4731       | 0.907     | 0.8165    | 0.8348      | 0.9801       | 0.7526     | 0.8236         | 0.8092   | 0.6442    | 0.9193        | 0.8264          | 0.9554        | 0.7577            | 0.8659       | 0.8424         | 0.7612 | 0.8855 | 0.8904  | 0.6517               | 0.9223             | -1.0        | 0.6112           | 0.9627        | 0.925           | 0.9951           | 0.9629             | 0.9624       | 0.9572            | 0.862               | 0.976             | 0.842                 | 0.9544           | 0.8806             | 0.8579    | 0.8766     | 0.7056    |
+| 4.3268        | 20.0  | 4380 | 5.5169          | 0.7975 | 0.9083 | 0.8523 | 0.2923           | 0.8923         | -1.0    | 0.5264       | 0.9058    | 0.8257    | 0.8227      | 0.9768       | 0.7526     | 0.7936         | 0.8144   | 0.6611    | 0.9197        | 0.8408          | 0.9465        | 0.6958            | 0.8818       | 0.8557         | 0.7621 | 0.8843 | 0.8899  | 0.6389               | 0.9235             | -1.0        | 0.6421           | 0.9585        | 0.9242          | 0.9944           | 0.957              | 0.9614       | 0.9525            | 0.8735              | 0.9754            | 0.8176                | 0.9516           | 0.8882             | 0.9814    | 0.8664     | 0.7154    |
+### Framework versions
+- Transformers 4.47.0
+- Pytorch 2.5.1+cu124
+- Datasets 3.1.0
+- Tokenizers 0.21.0

all_results.json ADDED Viewed

	@@ -0,0 +1,54 @@

+{
+    "epoch": 20.0,
+    "test_loss": 4.6487956047058105,
+    "test_map": 0.8602,
+    "test_map_50": 0.9588,
+    "test_map_75": 0.9287,
+    "test_map_baseball-bat": 0.5145,
+    "test_map_basketball": 0.8925,
+    "test_map_car": -1.0,
+    "test_map_football": 0.6313,
+    "test_map_human": 0.9186,
+    "test_map_large": 0.9699,
+    "test_map_luggage": 0.8921,
+    "test_map_mattress": 0.9812,
+    "test_map_medium": 0.8498,
+    "test_map_motorcycle": 0.9013,
+    "test_map_skis": 0.9669,
+    "test_map_small": 0.702,
+    "test_map_snowboard": 0.9779,
+    "test_map_soccer-ball": 0.8238,
+    "test_map_stop-sign": 0.9746,
+    "test_map_tennis-racket": 0.8141,
+    "test_map_umbrella": 0.9101,
+    "test_map_volleyball": 0.8435,
+    "test_mar_1": 0.8039,
+    "test_mar_10": 0.9143,
+    "test_mar_100": 0.9167,
+    "test_mar_100_baseball-bat": 0.7644,
+    "test_mar_100_basketball": 0.9259,
+    "test_mar_100_car": -1.0,
+    "test_mar_100_football": 0.7329,
+    "test_mar_100_human": 0.9633,
+    "test_mar_100_luggage": 0.9482,
+    "test_mar_100_mattress": 0.994,
+    "test_mar_100_motorcycle": 0.9575,
+    "test_mar_100_skis": 0.9935,
+    "test_mar_100_snowboard": 0.9889,
+    "test_mar_100_soccer-ball": 0.8611,
+    "test_mar_100_stop-sign": 0.9887,
+    "test_mar_100_tennis-racket": 0.8852,
+    "test_mar_100_umbrella": 0.9526,
+    "test_mar_100_volleyball": 0.8783,
+    "test_mar_large": 0.9887,
+    "test_mar_medium": 0.9276,
+    "test_mar_small": 0.7594,
+    "test_runtime": 72.2106,
+    "test_samples_per_second": 20.773,
+    "test_steps_per_second": 0.651,
+    "total_flos": 4.4130028142592e+19,
+    "train_loss": 5.691855750236337,
+    "train_runtime": 6795.0216,
+    "train_samples_per_second": 20.603,
+    "train_steps_per_second": 0.645
+}

config.json ADDED Viewed

	@@ -0,0 +1,125 @@

+{
+  "_name_or_path": "PekingU/rtdetr_r50vd",
+  "activation_dropout": 0.0,
+  "activation_function": "silu",
+  "anchor_image_size": null,
+  "architectures": [
+    "RTDetrForObjectDetection"
+  ],
+  "attention_dropout": 0.0,
+  "auxiliary_loss": true,
+  "backbone": null,
+  "backbone_config": {
+    "model_type": "rt_detr_resnet",
+    "out_features": [
+      "stage2",
+      "stage3",
+      "stage4"
+    ],
+    "out_indices": [
+      2,
+      3,
+      4
+    ]
+  },
+  "backbone_kwargs": null,
+  "batch_norm_eps": 1e-05,
+  "box_noise_scale": 1.0,
+  "d_model": 256,
+  "decoder_activation_function": "relu",
+  "decoder_attention_heads": 8,
+  "decoder_ffn_dim": 1024,
+  "decoder_in_channels": [
+    256,
+    256,
+    256
+  ],
+  "decoder_layers": 6,
+  "decoder_n_points": 4,
+  "disable_custom_kernels": true,
+  "dropout": 0.0,
+  "encode_proj_layers": [
+    2
+  ],
+  "encoder_activation_function": "gelu",
+  "encoder_attention_heads": 8,
+  "encoder_ffn_dim": 1024,
+  "encoder_hidden_dim": 256,
+  "encoder_in_channels": [
+    512,
+    1024,
+    2048
+  ],
+  "encoder_layers": 1,
+  "eos_coefficient": 0.0001,
+  "eval_size": null,
+  "feat_strides": [
+    8,
+    16,
+    32
+  ],
+  "focal_loss_alpha": 0.75,
+  "focal_loss_gamma": 2.0,
+  "freeze_backbone_batch_norms": true,
+  "hidden_expansion": 1.0,
+  "id2label": {
+    "0": "baseball-bat",
+    "1": "basketball",
+    "2": "car",
+    "3": "football",
+    "4": "human",
+    "5": "luggage",
+    "6": "mattress",
+    "7": "motorcycle",
+    "8": "skis",
+    "9": "snowboard",
+    "10": "soccer-ball",
+    "11": "stop-sign",
+    "12": "tennis-racket",
+    "13": "umbrella",
+    "14": "volleyball"
+  },
+  "initializer_bias_prior_prob": null,
+  "initializer_range": 0.01,
+  "is_encoder_decoder": true,
+  "label2id": {
+    "baseball-bat": 0,
+    "basketball": 1,
+    "car": 2,
+    "football": 3,
+    "human": 4,
+    "luggage": 5,
+    "mattress": 6,
+    "motorcycle": 7,
+    "skis": 8,
+    "snowboard": 9,
+    "soccer-ball": 10,
+    "stop-sign": 11,
+    "tennis-racket": 12,
+    "umbrella": 13,
+    "volleyball": 14
+  },
+  "label_noise_ratio": 0.5,
+  "layer_norm_eps": 1e-05,
+  "learn_initial_query": false,
+  "matcher_alpha": 0.25,
+  "matcher_bbox_cost": 5.0,
+  "matcher_class_cost": 2.0,
+  "matcher_gamma": 2.0,
+  "matcher_giou_cost": 2.0,
+  "model_type": "rt_detr",
+  "normalize_before": false,
+  "num_denoising": 100,
+  "num_feature_levels": 3,
+  "num_queries": 300,
+  "positional_encoding_temperature": 10000,
+  "torch_dtype": "float32",
+  "transformers_version": "4.47.0",
+  "use_focal_loss": true,
+  "use_pretrained_backbone": false,
+  "use_timm_backbone": false,
+  "weight_loss_bbox": 5.0,
+  "weight_loss_giou": 2.0,
+  "weight_loss_vfl": 1.0,
+  "with_box_refine": true
+}

model.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:d7a99f49ae9d24fa6230d5d05f35637621a84eb8818b62b1bf7c4a3849ccd7e9
+size 171641556

preprocessor_config.json ADDED Viewed

	@@ -0,0 +1,29 @@

+{
+  "do_convert_annotations": true,
+  "do_normalize": false,
+  "do_pad": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "format": "coco_detection",
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "RTDetrImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "pad_size": {
+    "height": 640,
+    "width": 640
+  },
+  "resample": 2,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "max_height": 640,
+    "max_width": 640
+  }
+}

test_results.json ADDED Viewed

	@@ -0,0 +1,49 @@

+{
+    "epoch": 20.0,
+    "test_loss": 4.6487956047058105,
+    "test_map": 0.8602,
+    "test_map_50": 0.9588,
+    "test_map_75": 0.9287,
+    "test_map_baseball-bat": 0.5145,
+    "test_map_basketball": 0.8925,
+    "test_map_car": -1.0,
+    "test_map_football": 0.6313,
+    "test_map_human": 0.9186,
+    "test_map_large": 0.9699,
+    "test_map_luggage": 0.8921,
+    "test_map_mattress": 0.9812,
+    "test_map_medium": 0.8498,
+    "test_map_motorcycle": 0.9013,
+    "test_map_skis": 0.9669,
+    "test_map_small": 0.702,
+    "test_map_snowboard": 0.9779,
+    "test_map_soccer-ball": 0.8238,
+    "test_map_stop-sign": 0.9746,
+    "test_map_tennis-racket": 0.8141,
+    "test_map_umbrella": 0.9101,
+    "test_map_volleyball": 0.8435,
+    "test_mar_1": 0.8039,
+    "test_mar_10": 0.9143,
+    "test_mar_100": 0.9167,
+    "test_mar_100_baseball-bat": 0.7644,
+    "test_mar_100_basketball": 0.9259,
+    "test_mar_100_car": -1.0,
+    "test_mar_100_football": 0.7329,
+    "test_mar_100_human": 0.9633,
+    "test_mar_100_luggage": 0.9482,
+    "test_mar_100_mattress": 0.994,
+    "test_mar_100_motorcycle": 0.9575,
+    "test_mar_100_skis": 0.9935,
+    "test_mar_100_snowboard": 0.9889,
+    "test_mar_100_soccer-ball": 0.8611,
+    "test_mar_100_stop-sign": 0.9887,
+    "test_mar_100_tennis-racket": 0.8852,
+    "test_mar_100_umbrella": 0.9526,
+    "test_mar_100_volleyball": 0.8783,
+    "test_mar_large": 0.9887,
+    "test_mar_medium": 0.9276,
+    "test_mar_small": 0.7594,
+    "test_runtime": 72.2106,
+    "test_samples_per_second": 20.773,
+    "test_steps_per_second": 0.651
+}

train_results.json ADDED Viewed

	@@ -0,0 +1,8 @@

+{
+    "epoch": 20.0,
+    "total_flos": 4.4130028142592e+19,
+    "train_loss": 5.691855750236337,
+    "train_runtime": 6795.0216,
+    "train_samples_per_second": 20.603,
+    "train_steps_per_second": 0.645
+}

trainer_state.json ADDED Viewed

	@@ -0,0 +1,1182 @@

+{
+  "best_metric": 0.8651,
+  "best_model_checkpoint": "rtdetr-finetuned/checkpoint-2190",
+  "epoch": 20.0,
+  "eval_steps": 500,
+  "global_step": 4380,
+  "is_hyper_param_search": false,
+  "is_local_process_zero": true,
+  "is_world_process_zero": true,
+  "log_history": [
+    {
+      "epoch": 1.0,
+      "grad_norm": 24.248611450195312,
+      "learning_rate": 4.908675799086758e-05,
+      "loss": 19.0657,
+      "step": 219
+    },
+    {
+      "epoch": 1.0,
+      "eval_loss": 5.672151565551758,
+      "eval_map": 0.6892,
+      "eval_map_50": 0.7627,
+      "eval_map_75": 0.7569,
+      "eval_map_baseball-bat": 0.6662,
+      "eval_map_basketball": 0.7899,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.1757,
+      "eval_map_human": 0.9219,
+      "eval_map_large": 0.8153,
+      "eval_map_luggage": 0.64,
+      "eval_map_mattress": 0.9656,
+      "eval_map_medium": 0.7254,
+      "eval_map_motorcycle": 0.9096,
+      "eval_map_skis": 0.4186,
+      "eval_map_small": 0.5623,
+      "eval_map_snowboard": 0.5418,
+      "eval_map_soccer-ball": 0.6531,
+      "eval_map_stop-sign": 0.9522,
+      "eval_map_tennis-racket": 0.4801,
+      "eval_map_umbrella": 0.9224,
+      "eval_map_volleyball": 0.6124,
+      "eval_mar_1": 0.7137,
+      "eval_mar_10": 0.8856,
+      "eval_mar_100": 0.8913,
+      "eval_mar_100_baseball-bat": 0.7836,
+      "eval_mar_100_basketball": 0.8709,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.703,
+      "eval_mar_100_human": 0.9551,
+      "eval_mar_100_luggage": 0.9195,
+      "eval_mar_100_mattress": 0.9945,
+      "eval_mar_100_motorcycle": 0.9427,
+      "eval_mar_100_skis": 0.9871,
+      "eval_mar_100_snowboard": 0.9854,
+      "eval_mar_100_soccer-ball": 0.8679,
+      "eval_mar_100_stop-sign": 0.9777,
+      "eval_mar_100_tennis-racket": 0.771,
+      "eval_mar_100_umbrella": 0.954,
+      "eval_mar_100_volleyball": 0.7657,
+      "eval_mar_large": 0.9702,
+      "eval_mar_medium": 0.9112,
+      "eval_mar_small": 0.7427,
+      "eval_runtime": 87.176,
+      "eval_samples_per_second": 17.207,
+      "eval_steps_per_second": 0.539,
+      "step": 219
+    },
+    {
+      "epoch": 2.0,
+      "grad_norm": 19.983396530151367,
+      "learning_rate": 9.908675799086759e-05,
+      "loss": 6.7339,
+      "step": 438
+    },
+    {
+      "epoch": 2.0,
+      "eval_loss": 5.662532329559326,
+      "eval_map": 0.742,
+      "eval_map_50": 0.8333,
+      "eval_map_75": 0.8117,
+      "eval_map_baseball-bat": 0.7301,
+      "eval_map_basketball": 0.787,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.3376,
+      "eval_map_human": 0.9024,
+      "eval_map_large": 0.8916,
+      "eval_map_luggage": 0.6101,
+      "eval_map_mattress": 0.9744,
+      "eval_map_medium": 0.7798,
+      "eval_map_motorcycle": 0.8818,
+      "eval_map_skis": 0.7438,
+      "eval_map_small": 0.5812,
+      "eval_map_snowboard": 0.857,
+      "eval_map_soccer-ball": 0.4723,
+      "eval_map_stop-sign": 0.9349,
+      "eval_map_tennis-racket": 0.6734,
+      "eval_map_umbrella": 0.884,
+      "eval_map_volleyball": 0.5999,
+      "eval_mar_1": 0.717,
+      "eval_mar_10": 0.868,
+      "eval_mar_100": 0.8722,
+      "eval_mar_100_baseball-bat": 0.8446,
+      "eval_mar_100_basketball": 0.8614,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6179,
+      "eval_mar_100_human": 0.9545,
+      "eval_mar_100_luggage": 0.9334,
+      "eval_mar_100_mattress": 0.992,
+      "eval_mar_100_motorcycle": 0.9245,
+      "eval_mar_100_skis": 0.9871,
+      "eval_mar_100_snowboard": 0.9869,
+      "eval_mar_100_soccer-ball": 0.6388,
+      "eval_mar_100_stop-sign": 0.9578,
+      "eval_mar_100_tennis-racket": 0.8684,
+      "eval_mar_100_umbrella": 0.9388,
+      "eval_mar_100_volleyball": 0.7047,
+      "eval_mar_large": 0.9659,
+      "eval_mar_medium": 0.9105,
+      "eval_mar_small": 0.7034,
+      "eval_runtime": 70.7584,
+      "eval_samples_per_second": 21.199,
+      "eval_steps_per_second": 0.664,
+      "step": 438
+    },
+    {
+      "epoch": 3.0,
+      "grad_norm": 20.1086368560791,
+      "learning_rate": 9.931086433113413e-05,
+      "loss": 5.8898,
+      "step": 657
+    },
+    {
+      "epoch": 3.0,
+      "eval_loss": 4.303356647491455,
+      "eval_map": 0.838,
+      "eval_map_50": 0.9206,
+      "eval_map_75": 0.9089,
+      "eval_map_baseball-bat": 0.7363,
+      "eval_map_basketball": 0.8553,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5598,
+      "eval_map_human": 0.9044,
+      "eval_map_large": 0.955,
+      "eval_map_luggage": 0.7481,
+      "eval_map_mattress": 0.9755,
+      "eval_map_medium": 0.8569,
+      "eval_map_motorcycle": 0.9085,
+      "eval_map_skis": 0.9331,
+      "eval_map_small": 0.7084,
+      "eval_map_snowboard": 0.9585,
+      "eval_map_soccer-ball": 0.8474,
+      "eval_map_stop-sign": 0.9861,
+      "eval_map_tennis-racket": 0.7094,
+      "eval_map_umbrella": 0.9098,
+      "eval_map_volleyball": 0.6997,
+      "eval_mar_1": 0.7942,
+      "eval_mar_10": 0.9197,
+      "eval_mar_100": 0.9236,
+      "eval_mar_100_baseball-bat": 0.8602,
+      "eval_mar_100_basketball": 0.9017,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7879,
+      "eval_mar_100_human": 0.9702,
+      "eval_mar_100_luggage": 0.9451,
+      "eval_mar_100_mattress": 0.993,
+      "eval_mar_100_motorcycle": 0.9622,
+      "eval_mar_100_skis": 0.9965,
+      "eval_mar_100_snowboard": 0.9945,
+      "eval_mar_100_soccer-ball": 0.9126,
+      "eval_mar_100_stop-sign": 0.9936,
+      "eval_mar_100_tennis-racket": 0.8788,
+      "eval_mar_100_umbrella": 0.9515,
+      "eval_mar_100_volleyball": 0.7824,
+      "eval_mar_large": 0.9788,
+      "eval_mar_medium": 0.9432,
+      "eval_mar_small": 0.8018,
+      "eval_runtime": 74.6719,
+      "eval_samples_per_second": 20.088,
+      "eval_steps_per_second": 0.629,
+      "step": 657
+    },
+    {
+      "epoch": 4.0,
+      "grad_norm": 18.903017044067383,
+      "learning_rate": 9.719978134831332e-05,
+      "loss": 5.6445,
+      "step": 876
+    },
+    {
+      "epoch": 4.0,
+      "eval_loss": 4.747404098510742,
+      "eval_map": 0.8119,
+      "eval_map_50": 0.9135,
+      "eval_map_75": 0.8838,
+      "eval_map_baseball-bat": 0.6429,
+      "eval_map_basketball": 0.8321,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.6007,
+      "eval_map_human": 0.9296,
+      "eval_map_large": 0.8501,
+      "eval_map_luggage": 0.8492,
+      "eval_map_mattress": 0.9742,
+      "eval_map_medium": 0.8402,
+      "eval_map_motorcycle": 0.9266,
+      "eval_map_skis": 0.6816,
+      "eval_map_small": 0.642,
+      "eval_map_snowboard": 0.9515,
+      "eval_map_soccer-ball": 0.7901,
+      "eval_map_stop-sign": 0.9841,
+      "eval_map_tennis-racket": 0.7362,
+      "eval_map_umbrella": 0.9255,
+      "eval_map_volleyball": 0.5431,
+      "eval_mar_1": 0.7707,
+      "eval_mar_10": 0.8935,
+      "eval_mar_100": 0.8977,
+      "eval_mar_100_baseball-bat": 0.7837,
+      "eval_mar_100_basketball": 0.8853,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7635,
+      "eval_mar_100_human": 0.9672,
+      "eval_mar_100_luggage": 0.9507,
+      "eval_mar_100_mattress": 0.9902,
+      "eval_mar_100_motorcycle": 0.9637,
+      "eval_mar_100_skis": 0.9941,
+      "eval_mar_100_snowboard": 0.9953,
+      "eval_mar_100_soccer-ball": 0.853,
+      "eval_mar_100_stop-sign": 0.993,
+      "eval_mar_100_tennis-racket": 0.8337,
+      "eval_mar_100_umbrella": 0.957,
+      "eval_mar_100_volleyball": 0.638,
+      "eval_mar_large": 0.9918,
+      "eval_mar_medium": 0.9332,
+      "eval_mar_small": 0.7188,
+      "eval_runtime": 73.188,
+      "eval_samples_per_second": 20.495,
+      "eval_steps_per_second": 0.642,
+      "step": 876
+    },
+    {
+      "epoch": 5.0,
+      "grad_norm": 24.344205856323242,
+      "learning_rate": 9.374930190505666e-05,
+      "loss": 5.624,
+      "step": 1095
+    },
+    {
+      "epoch": 5.0,
+      "eval_loss": 4.6554155349731445,
+      "eval_map": 0.8246,
+      "eval_map_50": 0.9344,
+      "eval_map_75": 0.9007,
+      "eval_map_baseball-bat": 0.6562,
+      "eval_map_basketball": 0.7609,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5938,
+      "eval_map_human": 0.89,
+      "eval_map_large": 0.956,
+      "eval_map_luggage": 0.8797,
+      "eval_map_mattress": 0.9671,
+      "eval_map_medium": 0.8605,
+      "eval_map_motorcycle": 0.9113,
+      "eval_map_skis": 0.8424,
+      "eval_map_small": 0.6468,
+      "eval_map_snowboard": 0.9674,
+      "eval_map_soccer-ball": 0.8123,
+      "eval_map_stop-sign": 0.9741,
+      "eval_map_tennis-racket": 0.7671,
+      "eval_map_umbrella": 0.9256,
+      "eval_map_volleyball": 0.5956,
+      "eval_mar_1": 0.7729,
+      "eval_mar_10": 0.8907,
+      "eval_mar_100": 0.8939,
+      "eval_mar_100_baseball-bat": 0.7985,
+      "eval_mar_100_basketball": 0.8355,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7333,
+      "eval_mar_100_human": 0.9497,
+      "eval_mar_100_luggage": 0.9368,
+      "eval_mar_100_mattress": 0.9806,
+      "eval_mar_100_motorcycle": 0.9621,
+      "eval_mar_100_skis": 0.9856,
+      "eval_mar_100_snowboard": 0.9888,
+      "eval_mar_100_soccer-ball": 0.8548,
+      "eval_mar_100_stop-sign": 0.986,
+      "eval_mar_100_tennis-racket": 0.8674,
+      "eval_mar_100_umbrella": 0.9523,
+      "eval_mar_100_volleyball": 0.6838,
+      "eval_mar_large": 0.9866,
+      "eval_mar_medium": 0.9296,
+      "eval_mar_small": 0.7144,
+      "eval_runtime": 74.9781,
+      "eval_samples_per_second": 20.006,
+      "eval_steps_per_second": 0.627,
+      "step": 1095
+    },
+    {
+      "epoch": 6.0,
+      "grad_norm": 33.08514404296875,
+      "learning_rate": 8.903269206302188e-05,
+      "loss": 5.3591,
+      "step": 1314
+    },
+    {
+      "epoch": 6.0,
+      "eval_loss": 4.476555347442627,
+      "eval_map": 0.8456,
+      "eval_map_50": 0.9325,
+      "eval_map_75": 0.9116,
+      "eval_map_baseball-bat": 0.5066,
+      "eval_map_basketball": 0.8924,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.7547,
+      "eval_map_human": 0.9151,
+      "eval_map_large": 0.9694,
+      "eval_map_luggage": 0.8623,
+      "eval_map_mattress": 0.9712,
+      "eval_map_medium": 0.7744,
+      "eval_map_motorcycle": 0.843,
+      "eval_map_skis": 0.8972,
+      "eval_map_small": 0.7299,
+      "eval_map_snowboard": 0.986,
+      "eval_map_soccer-ball": 0.859,
+      "eval_map_stop-sign": 0.9637,
+      "eval_map_tennis-racket": 0.6545,
+      "eval_map_umbrella": 0.9286,
+      "eval_map_volleyball": 0.8045,
+      "eval_mar_1": 0.7963,
+      "eval_mar_10": 0.9169,
+      "eval_mar_100": 0.9202,
+      "eval_mar_100_baseball-bat": 0.8016,
+      "eval_mar_100_basketball": 0.9287,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.8219,
+      "eval_mar_100_human": 0.9692,
+      "eval_mar_100_luggage": 0.9339,
+      "eval_mar_100_mattress": 0.9858,
+      "eval_mar_100_motorcycle": 0.9567,
+      "eval_mar_100_skis": 0.9797,
+      "eval_mar_100_snowboard": 0.9961,
+      "eval_mar_100_soccer-ball": 0.8943,
+      "eval_mar_100_stop-sign": 0.9838,
+      "eval_mar_100_tennis-racket": 0.828,
+      "eval_mar_100_umbrella": 0.9576,
+      "eval_mar_100_volleyball": 0.8451,
+      "eval_mar_large": 0.9875,
+      "eval_mar_medium": 0.9297,
+      "eval_mar_small": 0.7973,
+      "eval_runtime": 73.5138,
+      "eval_samples_per_second": 20.404,
+      "eval_steps_per_second": 0.639,
+      "step": 1314
+    },
+    {
+      "epoch": 7.0,
+      "grad_norm": 79.52643585205078,
+      "learning_rate": 8.320605485908638e-05,
+      "loss": 5.1554,
+      "step": 1533
+    },
+    {
+      "epoch": 7.0,
+      "eval_loss": 5.435186386108398,
+      "eval_map": 0.8049,
+      "eval_map_50": 0.9055,
+      "eval_map_75": 0.8645,
+      "eval_map_baseball-bat": 0.506,
+      "eval_map_basketball": 0.8712,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.3751,
+      "eval_map_human": 0.8879,
+      "eval_map_large": 0.9469,
+      "eval_map_luggage": 0.8829,
+      "eval_map_mattress": 0.9662,
+      "eval_map_medium": 0.8023,
+      "eval_map_motorcycle": 0.8948,
+      "eval_map_skis": 0.8169,
+      "eval_map_small": 0.6245,
+      "eval_map_snowboard": 0.9745,
+      "eval_map_soccer-ball": 0.8381,
+      "eval_map_stop-sign": 0.9435,
+      "eval_map_tennis-racket": 0.6993,
+      "eval_map_umbrella": 0.8829,
+      "eval_map_volleyball": 0.7293,
+      "eval_mar_1": 0.7557,
+      "eval_mar_10": 0.874,
+      "eval_mar_100": 0.8768,
+      "eval_mar_100_baseball-bat": 0.726,
+      "eval_mar_100_basketball": 0.9031,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.5281,
+      "eval_mar_100_human": 0.953,
+      "eval_mar_100_luggage": 0.9372,
+      "eval_mar_100_mattress": 0.9822,
+      "eval_mar_100_motorcycle": 0.9515,
+      "eval_mar_100_skis": 0.954,
+      "eval_mar_100_snowboard": 0.988,
+      "eval_mar_100_soccer-ball": 0.8728,
+      "eval_mar_100_stop-sign": 0.9696,
+      "eval_mar_100_tennis-racket": 0.8031,
+      "eval_mar_100_umbrella": 0.9372,
+      "eval_mar_100_volleyball": 0.7696,
+      "eval_mar_large": 0.9782,
+      "eval_mar_medium": 0.9096,
+      "eval_mar_small": 0.6926,
+      "eval_runtime": 76.6654,
+      "eval_samples_per_second": 19.566,
+      "eval_steps_per_second": 0.613,
+      "step": 1533
+    },
+    {
+      "epoch": 8.0,
+      "grad_norm": 16.643890380859375,
+      "learning_rate": 7.647911489121759e-05,
+      "loss": 5.1876,
+      "step": 1752
+    },
+    {
+      "epoch": 8.0,
+      "eval_loss": 4.820957660675049,
+      "eval_map": 0.8373,
+      "eval_map_50": 0.9334,
+      "eval_map_75": 0.8999,
+      "eval_map_baseball-bat": 0.5239,
+      "eval_map_basketball": 0.8396,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.636,
+      "eval_map_human": 0.915,
+      "eval_map_large": 0.9611,
+      "eval_map_luggage": 0.8947,
+      "eval_map_mattress": 0.9864,
+      "eval_map_medium": 0.8289,
+      "eval_map_motorcycle": 0.9182,
+      "eval_map_skis": 0.9147,
+      "eval_map_small": 0.6663,
+      "eval_map_snowboard": 0.9729,
+      "eval_map_soccer-ball": 0.8374,
+      "eval_map_stop-sign": 0.9701,
+      "eval_map_tennis-racket": 0.6451,
+      "eval_map_umbrella": 0.8846,
+      "eval_map_volleyball": 0.783,
+      "eval_mar_1": 0.7803,
+      "eval_mar_10": 0.9013,
+      "eval_mar_100": 0.9038,
+      "eval_mar_100_baseball-bat": 0.7665,
+      "eval_mar_100_basketball": 0.8765,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7182,
+      "eval_mar_100_human": 0.9674,
+      "eval_mar_100_luggage": 0.9511,
+      "eval_mar_100_mattress": 0.9956,
+      "eval_mar_100_motorcycle": 0.9654,
+      "eval_mar_100_skis": 0.9782,
+      "eval_mar_100_snowboard": 0.9883,
+      "eval_mar_100_soccer-ball": 0.8725,
+      "eval_mar_100_stop-sign": 0.9874,
+      "eval_mar_100_tennis-racket": 0.8135,
+      "eval_mar_100_umbrella": 0.9473,
+      "eval_mar_100_volleyball": 0.8252,
+      "eval_mar_large": 0.9883,
+      "eval_mar_medium": 0.9242,
+      "eval_mar_small": 0.735,
+      "eval_runtime": 75.2242,
+      "eval_samples_per_second": 19.94,
+      "eval_steps_per_second": 0.625,
+      "step": 1752
+    },
+    {
+      "epoch": 9.0,
+      "grad_norm": 22.301359176635742,
+      "learning_rate": 6.899470923656964e-05,
+      "loss": 4.9537,
+      "step": 1971
+    },
+    {
+      "epoch": 9.0,
+      "eval_loss": 4.713701248168945,
+      "eval_map": 0.838,
+      "eval_map_50": 0.9376,
+      "eval_map_75": 0.9021,
+      "eval_map_baseball-bat": 0.4846,
+      "eval_map_basketball": 0.8804,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.6231,
+      "eval_map_human": 0.9015,
+      "eval_map_large": 0.9476,
+      "eval_map_luggage": 0.8638,
+      "eval_map_mattress": 0.9507,
+      "eval_map_medium": 0.8454,
+      "eval_map_motorcycle": 0.9206,
+      "eval_map_skis": 0.898,
+      "eval_map_small": 0.6902,
+      "eval_map_snowboard": 0.951,
+      "eval_map_soccer-ball": 0.8526,
+      "eval_map_stop-sign": 0.978,
+      "eval_map_tennis-racket": 0.7464,
+      "eval_map_umbrella": 0.8486,
+      "eval_map_volleyball": 0.8327,
+      "eval_mar_1": 0.7894,
+      "eval_mar_10": 0.9057,
+      "eval_mar_100": 0.9086,
+      "eval_mar_100_baseball-bat": 0.738,
+      "eval_mar_100_basketball": 0.9128,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7469,
+      "eval_mar_100_human": 0.9636,
+      "eval_mar_100_luggage": 0.9232,
+      "eval_mar_100_mattress": 0.9778,
+      "eval_mar_100_motorcycle": 0.9648,
+      "eval_mar_100_skis": 0.9653,
+      "eval_mar_100_snowboard": 0.9741,
+      "eval_mar_100_soccer-ball": 0.8889,
+      "eval_mar_100_stop-sign": 0.9925,
+      "eval_mar_100_tennis-racket": 0.858,
+      "eval_mar_100_umbrella": 0.9418,
+      "eval_mar_100_volleyball": 0.8723,
+      "eval_mar_large": 0.9834,
+      "eval_mar_medium": 0.9191,
+      "eval_mar_small": 0.7541,
+      "eval_runtime": 73.5443,
+      "eval_samples_per_second": 20.396,
+      "eval_steps_per_second": 0.639,
+      "step": 1971
+    },
+    {
+      "epoch": 10.0,
+      "grad_norm": 21.906387329101562,
+      "learning_rate": 6.100912018849448e-05,
+      "loss": 4.9224,
+      "step": 2190
+    },
+    {
+      "epoch": 10.0,
+      "eval_loss": 4.59971809387207,
+      "eval_map": 0.8651,
+      "eval_map_50": 0.9611,
+      "eval_map_75": 0.9319,
+      "eval_map_baseball-bat": 0.5229,
+      "eval_map_basketball": 0.8898,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.6333,
+      "eval_map_human": 0.9194,
+      "eval_map_large": 0.9665,
+      "eval_map_luggage": 0.8892,
+      "eval_map_mattress": 0.9732,
+      "eval_map_medium": 0.8569,
+      "eval_map_motorcycle": 0.9065,
+      "eval_map_skis": 0.9781,
+      "eval_map_small": 0.7128,
+      "eval_map_snowboard": 0.9808,
+      "eval_map_soccer-ball": 0.8364,
+      "eval_map_stop-sign": 0.9768,
+      "eval_map_tennis-racket": 0.8425,
+      "eval_map_umbrella": 0.9263,
+      "eval_map_volleyball": 0.8363,
+      "eval_mar_1": 0.8041,
+      "eval_mar_10": 0.9164,
+      "eval_mar_100": 0.9189,
+      "eval_mar_100_baseball-bat": 0.7741,
+      "eval_mar_100_basketball": 0.9223,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.7222,
+      "eval_mar_100_human": 0.9623,
+      "eval_mar_100_luggage": 0.9477,
+      "eval_mar_100_mattress": 0.995,
+      "eval_mar_100_motorcycle": 0.9632,
+      "eval_mar_100_skis": 0.9931,
+      "eval_mar_100_snowboard": 0.9919,
+      "eval_mar_100_soccer-ball": 0.871,
+      "eval_mar_100_stop-sign": 0.988,
+      "eval_mar_100_tennis-racket": 0.899,
+      "eval_mar_100_umbrella": 0.9617,
+      "eval_mar_100_volleyball": 0.8728,
+      "eval_mar_large": 0.9791,
+      "eval_mar_medium": 0.933,
+      "eval_mar_small": 0.7708,
+      "eval_runtime": 73.7262,
+      "eval_samples_per_second": 20.346,
+      "eval_steps_per_second": 0.637,
+      "step": 2190
+    },
+    {
+      "epoch": 11.0,
+      "grad_norm": 37.53829574584961,
+      "learning_rate": 5.2764985829313986e-05,
+      "loss": 4.7256,
+      "step": 2409
+    },
+    {
+      "epoch": 11.0,
+      "eval_loss": 4.814239501953125,
+      "eval_map": 0.8399,
+      "eval_map_50": 0.9462,
+      "eval_map_75": 0.9028,
+      "eval_map_baseball-bat": 0.4084,
+      "eval_map_basketball": 0.8985,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.577,
+      "eval_map_human": 0.93,
+      "eval_map_large": 0.9667,
+      "eval_map_luggage": 0.9061,
+      "eval_map_mattress": 0.9792,
+      "eval_map_medium": 0.8373,
+      "eval_map_motorcycle": 0.9326,
+      "eval_map_skis": 0.8848,
+      "eval_map_small": 0.6739,
+      "eval_map_snowboard": 0.9532,
+      "eval_map_soccer-ball": 0.792,
+      "eval_map_stop-sign": 0.9554,
+      "eval_map_tennis-racket": 0.8222,
+      "eval_map_umbrella": 0.9108,
+      "eval_map_volleyball": 0.8089,
+      "eval_mar_1": 0.7824,
+      "eval_mar_10": 0.8985,
+      "eval_mar_100": 0.9028,
+      "eval_mar_100_baseball-bat": 0.6793,
+      "eval_mar_100_basketball": 0.9339,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6901,
+      "eval_mar_100_human": 0.9718,
+      "eval_mar_100_luggage": 0.9503,
+      "eval_mar_100_mattress": 0.9933,
+      "eval_mar_100_motorcycle": 0.9776,
+      "eval_mar_100_skis": 0.9594,
+      "eval_mar_100_snowboard": 0.9749,
+      "eval_mar_100_soccer-ball": 0.8324,
+      "eval_mar_100_stop-sign": 0.9774,
+      "eval_mar_100_tennis-racket": 0.8829,
+      "eval_mar_100_umbrella": 0.9604,
+      "eval_mar_100_volleyball": 0.8549,
+      "eval_mar_large": 0.9876,
+      "eval_mar_medium": 0.9168,
+      "eval_mar_small": 0.734,
+      "eval_runtime": 77.5232,
+      "eval_samples_per_second": 19.349,
+      "eval_steps_per_second": 0.606,
+      "step": 2409
+    },
+    {
+      "epoch": 12.0,
+      "grad_norm": 16.9416446685791,
+      "learning_rate": 4.4512800009799146e-05,
+      "loss": 4.6993,
+      "step": 2628
+    },
+    {
+      "epoch": 12.0,
+      "eval_loss": 4.759703636169434,
+      "eval_map": 0.8217,
+      "eval_map_50": 0.9297,
+      "eval_map_75": 0.8787,
+      "eval_map_baseball-bat": 0.332,
+      "eval_map_basketball": 0.8934,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.4632,
+      "eval_map_human": 0.9409,
+      "eval_map_large": 0.9484,
+      "eval_map_luggage": 0.9098,
+      "eval_map_mattress": 0.954,
+      "eval_map_medium": 0.8053,
+      "eval_map_motorcycle": 0.949,
+      "eval_map_skis": 0.84,
+      "eval_map_small": 0.6514,
+      "eval_map_snowboard": 0.9579,
+      "eval_map_soccer-ball": 0.8407,
+      "eval_map_stop-sign": 0.9612,
+      "eval_map_tennis-racket": 0.7405,
+      "eval_map_umbrella": 0.9017,
+      "eval_map_volleyball": 0.8189,
+      "eval_mar_1": 0.7709,
+      "eval_mar_10": 0.8931,
+      "eval_mar_100": 0.8971,
+      "eval_mar_100_baseball-bat": 0.6972,
+      "eval_mar_100_basketball": 0.9273,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.5876,
+      "eval_mar_100_human": 0.9713,
+      "eval_mar_100_luggage": 0.9524,
+      "eval_mar_100_mattress": 0.98,
+      "eval_mar_100_motorcycle": 0.9798,
+      "eval_mar_100_skis": 0.9604,
+      "eval_mar_100_snowboard": 0.9783,
+      "eval_mar_100_soccer-ball": 0.8748,
+      "eval_mar_100_stop-sign": 0.9807,
+      "eval_mar_100_tennis-racket": 0.8492,
+      "eval_mar_100_umbrella": 0.961,
+      "eval_mar_100_volleyball": 0.8588,
+      "eval_mar_large": 0.9739,
+      "eval_mar_medium": 0.9146,
+      "eval_mar_small": 0.7196,
+      "eval_runtime": 73.1101,
+      "eval_samples_per_second": 20.517,
+      "eval_steps_per_second": 0.643,
+      "step": 2628
+    },
+    {
+      "epoch": 13.0,
+      "grad_norm": 14.29014778137207,
+      "learning_rate": 3.650330122026836e-05,
+      "loss": 4.7266,
+      "step": 2847
+    },
+    {
+      "epoch": 13.0,
+      "eval_loss": 5.418997287750244,
+      "eval_map": 0.7638,
+      "eval_map_50": 0.8926,
+      "eval_map_75": 0.8072,
+      "eval_map_baseball-bat": 0.2568,
+      "eval_map_basketball": 0.8598,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.2283,
+      "eval_map_human": 0.9259,
+      "eval_map_large": 0.9286,
+      "eval_map_luggage": 0.8451,
+      "eval_map_mattress": 0.9628,
+      "eval_map_medium": 0.7222,
+      "eval_map_motorcycle": 0.7728,
+      "eval_map_skis": 0.8578,
+      "eval_map_small": 0.5811,
+      "eval_map_snowboard": 0.8954,
+      "eval_map_soccer-ball": 0.8364,
+      "eval_map_stop-sign": 0.9659,
+      "eval_map_tennis-racket": 0.6612,
+      "eval_map_umbrella": 0.7892,
+      "eval_map_volleyball": 0.8364,
+      "eval_mar_1": 0.7303,
+      "eval_mar_10": 0.8583,
+      "eval_mar_100": 0.8635,
+      "eval_mar_100_baseball-bat": 0.6368,
+      "eval_mar_100_basketball": 0.8976,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.38,
+      "eval_mar_100_human": 0.9653,
+      "eval_mar_100_luggage": 0.9222,
+      "eval_mar_100_mattress": 0.9822,
+      "eval_mar_100_motorcycle": 0.9603,
+      "eval_mar_100_skis": 0.9535,
+      "eval_mar_100_snowboard": 0.9387,
+      "eval_mar_100_soccer-ball": 0.8686,
+      "eval_mar_100_stop-sign": 0.9863,
+      "eval_mar_100_tennis-racket": 0.8057,
+      "eval_mar_100_umbrella": 0.9188,
+      "eval_mar_100_volleyball": 0.8735,
+      "eval_mar_large": 0.9675,
+      "eval_mar_medium": 0.873,
+      "eval_mar_small": 0.6569,
+      "eval_runtime": 88.7032,
+      "eval_samples_per_second": 16.91,
+      "eval_steps_per_second": 0.53,
+      "step": 2847
+    },
+    {
+      "epoch": 14.0,
+      "grad_norm": 95.85160827636719,
+      "learning_rate": 2.8979854028439745e-05,
+      "loss": 4.7442,
+      "step": 3066
+    },
+    {
+      "epoch": 14.0,
+      "eval_loss": 5.6973114013671875,
+      "eval_map": 0.7497,
+      "eval_map_50": 0.8723,
+      "eval_map_75": 0.7933,
+      "eval_map_baseball-bat": 0.2398,
+      "eval_map_basketball": 0.8922,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.3593,
+      "eval_map_human": 0.9187,
+      "eval_map_large": 0.787,
+      "eval_map_luggage": 0.7688,
+      "eval_map_mattress": 0.88,
+      "eval_map_medium": 0.6939,
+      "eval_map_motorcycle": 0.7618,
+      "eval_map_skis": 0.7407,
+      "eval_map_small": 0.6262,
+      "eval_map_snowboard": 0.8981,
+      "eval_map_soccer-ball": 0.8576,
+      "eval_map_stop-sign": 0.9551,
+      "eval_map_tennis-racket": 0.5716,
+      "eval_map_umbrella": 0.8151,
+      "eval_map_volleyball": 0.837,
+      "eval_mar_1": 0.7227,
+      "eval_mar_10": 0.8473,
+      "eval_mar_100": 0.8528,
+      "eval_mar_100_baseball-bat": 0.5856,
+      "eval_mar_100_basketball": 0.9204,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.4725,
+      "eval_mar_100_human": 0.964,
+      "eval_mar_100_luggage": 0.8604,
+      "eval_mar_100_mattress": 0.9307,
+      "eval_mar_100_motorcycle": 0.9673,
+      "eval_mar_100_skis": 0.9,
+      "eval_mar_100_snowboard": 0.9324,
+      "eval_mar_100_soccer-ball": 0.8869,
+      "eval_mar_100_stop-sign": 0.9816,
+      "eval_mar_100_tennis-racket": 0.7316,
+      "eval_mar_100_umbrella": 0.9325,
+      "eval_mar_100_volleyball": 0.8733,
+      "eval_mar_large": 0.8429,
+      "eval_mar_medium": 0.8393,
+      "eval_mar_small": 0.6824,
+      "eval_runtime": 70.8312,
+      "eval_samples_per_second": 21.177,
+      "eval_steps_per_second": 0.664,
+      "step": 3066
+    },
+    {
+      "epoch": 15.0,
+      "grad_norm": 29.083290100097656,
+      "learning_rate": 2.2200198345783707e-05,
+      "loss": 4.5738,
+      "step": 3285
+    },
+    {
+      "epoch": 15.0,
+      "eval_loss": 5.435721397399902,
+      "eval_map": 0.7908,
+      "eval_map_50": 0.9047,
+      "eval_map_75": 0.8408,
+      "eval_map_baseball-bat": 0.2947,
+      "eval_map_basketball": 0.8977,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.4669,
+      "eval_map_human": 0.9032,
+      "eval_map_large": 0.9224,
+      "eval_map_luggage": 0.8532,
+      "eval_map_mattress": 0.9445,
+      "eval_map_medium": 0.7435,
+      "eval_map_motorcycle": 0.8411,
+      "eval_map_skis": 0.7558,
+      "eval_map_small": 0.6625,
+      "eval_map_snowboard": 0.9395,
+      "eval_map_soccer-ball": 0.8668,
+      "eval_map_stop-sign": 0.9471,
+      "eval_map_tennis-racket": 0.6622,
+      "eval_map_umbrella": 0.8522,
+      "eval_map_volleyball": 0.8465,
+      "eval_mar_1": 0.7527,
+      "eval_mar_10": 0.8804,
+      "eval_mar_100": 0.8853,
+      "eval_mar_100_baseball-bat": 0.6655,
+      "eval_mar_100_basketball": 0.9291,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.5869,
+      "eval_mar_100_human": 0.9612,
+      "eval_mar_100_luggage": 0.9217,
+      "eval_mar_100_mattress": 0.9773,
+      "eval_mar_100_motorcycle": 0.9613,
+      "eval_mar_100_skis": 0.9347,
+      "eval_mar_100_snowboard": 0.9585,
+      "eval_mar_100_soccer-ball": 0.8933,
+      "eval_mar_100_stop-sign": 0.9827,
+      "eval_mar_100_tennis-racket": 0.7984,
+      "eval_mar_100_umbrella": 0.9458,
+      "eval_mar_100_volleyball": 0.8777,
+      "eval_mar_large": 0.9781,
+      "eval_mar_medium": 0.8732,
+      "eval_mar_small": 0.7268,
+      "eval_runtime": 73.5722,
+      "eval_samples_per_second": 20.388,
+      "eval_steps_per_second": 0.639,
+      "step": 3285
+    },
+    {
+      "epoch": 16.0,
+      "grad_norm": 10.778564453125,
+      "learning_rate": 1.6308290593265478e-05,
+      "loss": 4.4781,
+      "step": 3504
+    },
+    {
+      "epoch": 16.0,
+      "eval_loss": 5.64291524887085,
+      "eval_map": 0.7929,
+      "eval_map_50": 0.9102,
+      "eval_map_75": 0.8519,
+      "eval_map_baseball-bat": 0.3248,
+      "eval_map_basketball": 0.892,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5374,
+      "eval_map_human": 0.9101,
+      "eval_map_large": 0.925,
+      "eval_map_luggage": 0.845,
+      "eval_map_mattress": 0.9494,
+      "eval_map_medium": 0.7235,
+      "eval_map_motorcycle": 0.7888,
+      "eval_map_skis": 0.7988,
+      "eval_map_small": 0.6634,
+      "eval_map_snowboard": 0.9226,
+      "eval_map_soccer-ball": 0.8371,
+      "eval_map_stop-sign": 0.9334,
+      "eval_map_tennis-racket": 0.6908,
+      "eval_map_umbrella": 0.8184,
+      "eval_map_volleyball": 0.8523,
+      "eval_mar_1": 0.7585,
+      "eval_mar_10": 0.8821,
+      "eval_mar_100": 0.8864,
+      "eval_mar_100_baseball-bat": 0.6622,
+      "eval_mar_100_basketball": 0.9182,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6483,
+      "eval_mar_100_human": 0.9607,
+      "eval_mar_100_luggage": 0.9234,
+      "eval_mar_100_mattress": 0.9758,
+      "eval_mar_100_motorcycle": 0.9597,
+      "eval_mar_100_skis": 0.9361,
+      "eval_mar_100_snowboard": 0.9571,
+      "eval_mar_100_soccer-ball": 0.8722,
+      "eval_mar_100_stop-sign": 0.9718,
+      "eval_mar_100_tennis-racket": 0.8109,
+      "eval_mar_100_umbrella": 0.9299,
+      "eval_mar_100_volleyball": 0.8836,
+      "eval_mar_large": 0.9743,
+      "eval_mar_medium": 0.8641,
+      "eval_mar_small": 0.7226,
+      "eval_runtime": 72.8768,
+      "eval_samples_per_second": 20.583,
+      "eval_steps_per_second": 0.645,
+      "step": 3504
+    },
+    {
+      "epoch": 17.0,
+      "grad_norm": 80.79771423339844,
+      "learning_rate": 1.1516049617182226e-05,
+      "loss": 4.3981,
+      "step": 3723
+    },
+    {
+      "epoch": 17.0,
+      "eval_loss": 5.452239036560059,
+      "eval_map": 0.8124,
+      "eval_map_50": 0.9224,
+      "eval_map_75": 0.8705,
+      "eval_map_baseball-bat": 0.2978,
+      "eval_map_basketball": 0.8935,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5512,
+      "eval_map_human": 0.9205,
+      "eval_map_large": 0.9528,
+      "eval_map_luggage": 0.8544,
+      "eval_map_mattress": 0.9725,
+      "eval_map_medium": 0.7791,
+      "eval_map_motorcycle": 0.8652,
+      "eval_map_skis": 0.8152,
+      "eval_map_small": 0.6647,
+      "eval_map_snowboard": 0.9437,
+      "eval_map_soccer-ball": 0.8396,
+      "eval_map_stop-sign": 0.9579,
+      "eval_map_tennis-racket": 0.7453,
+      "eval_map_umbrella": 0.8685,
+      "eval_map_volleyball": 0.8482,
+      "eval_mar_1": 0.7712,
+      "eval_mar_10": 0.8924,
+      "eval_mar_100": 0.8969,
+      "eval_mar_100_baseball-bat": 0.6459,
+      "eval_mar_100_basketball": 0.9246,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6725,
+      "eval_mar_100_human": 0.9653,
+      "eval_mar_100_luggage": 0.9346,
+      "eval_mar_100_mattress": 0.9885,
+      "eval_mar_100_motorcycle": 0.9694,
+      "eval_mar_100_skis": 0.945,
+      "eval_mar_100_snowboard": 0.9665,
+      "eval_mar_100_soccer-ball": 0.873,
+      "eval_mar_100_stop-sign": 0.9757,
+      "eval_mar_100_tennis-racket": 0.8549,
+      "eval_mar_100_umbrella": 0.9541,
+      "eval_mar_100_volleyball": 0.8868,
+      "eval_mar_large": 0.985,
+      "eval_mar_medium": 0.8836,
+      "eval_mar_small": 0.725,
+      "eval_runtime": 70.9069,
+      "eval_samples_per_second": 21.154,
+      "eval_steps_per_second": 0.663,
+      "step": 3723
+    },
+    {
+      "epoch": 18.0,
+      "grad_norm": 14.495157241821289,
+      "learning_rate": 7.96908523460124e-06,
+      "loss": 4.299,
+      "step": 3942
+    },
+    {
+      "epoch": 18.0,
+      "eval_loss": 5.5161237716674805,
+      "eval_map": 0.8209,
+      "eval_map_50": 0.9261,
+      "eval_map_75": 0.8788,
+      "eval_map_baseball-bat": 0.3434,
+      "eval_map_basketball": 0.8958,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5914,
+      "eval_map_human": 0.9229,
+      "eval_map_large": 0.9299,
+      "eval_map_luggage": 0.8504,
+      "eval_map_mattress": 0.9673,
+      "eval_map_medium": 0.7851,
+      "eval_map_motorcycle": 0.8517,
+      "eval_map_skis": 0.8319,
+      "eval_map_small": 0.6775,
+      "eval_map_snowboard": 0.9347,
+      "eval_map_soccer-ball": 0.8423,
+      "eval_map_stop-sign": 0.9591,
+      "eval_map_tennis-racket": 0.7675,
+      "eval_map_umbrella": 0.8792,
+      "eval_map_volleyball": 0.8551,
+      "eval_mar_1": 0.7741,
+      "eval_mar_10": 0.8968,
+      "eval_mar_100": 0.9012,
+      "eval_mar_100_baseball-bat": 0.6663,
+      "eval_mar_100_basketball": 0.9246,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6943,
+      "eval_mar_100_human": 0.9677,
+      "eval_mar_100_luggage": 0.9318,
+      "eval_mar_100_mattress": 0.9861,
+      "eval_mar_100_motorcycle": 0.9643,
+      "eval_mar_100_skis": 0.9599,
+      "eval_mar_100_snowboard": 0.9632,
+      "eval_mar_100_soccer-ball": 0.8748,
+      "eval_mar_100_stop-sign": 0.9799,
+      "eval_mar_100_tennis-racket": 0.8637,
+      "eval_mar_100_umbrella": 0.9536,
+      "eval_mar_100_volleyball": 0.8858,
+      "eval_mar_large": 0.9719,
+      "eval_mar_medium": 0.8805,
+      "eval_mar_small": 0.7305,
+      "eval_runtime": 71.4404,
+      "eval_samples_per_second": 20.997,
+      "eval_steps_per_second": 0.658,
+      "step": 3942
+    },
+    {
+      "epoch": 19.0,
+      "grad_norm": 8.617731094360352,
+      "learning_rate": 5.775170163436809e-06,
+      "loss": 4.3295,
+      "step": 4161
+    },
+    {
+      "epoch": 19.0,
+      "eval_loss": 5.571784019470215,
+      "eval_map": 0.7992,
+      "eval_map_50": 0.9122,
+      "eval_map_75": 0.8546,
+      "eval_map_baseball-bat": 0.2997,
+      "eval_map_basketball": 0.8944,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.4731,
+      "eval_map_human": 0.907,
+      "eval_map_large": 0.8165,
+      "eval_map_luggage": 0.8348,
+      "eval_map_mattress": 0.9801,
+      "eval_map_medium": 0.7526,
+      "eval_map_motorcycle": 0.8236,
+      "eval_map_skis": 0.8092,
+      "eval_map_small": 0.6442,
+      "eval_map_snowboard": 0.9193,
+      "eval_map_soccer-ball": 0.8264,
+      "eval_map_stop-sign": 0.9554,
+      "eval_map_tennis-racket": 0.7577,
+      "eval_map_umbrella": 0.8659,
+      "eval_map_volleyball": 0.8424,
+      "eval_mar_1": 0.7612,
+      "eval_mar_10": 0.8855,
+      "eval_mar_100": 0.8904,
+      "eval_mar_100_baseball-bat": 0.6517,
+      "eval_mar_100_basketball": 0.9223,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6112,
+      "eval_mar_100_human": 0.9627,
+      "eval_mar_100_luggage": 0.925,
+      "eval_mar_100_mattress": 0.9951,
+      "eval_mar_100_motorcycle": 0.9629,
+      "eval_mar_100_skis": 0.9624,
+      "eval_mar_100_snowboard": 0.9572,
+      "eval_mar_100_soccer-ball": 0.862,
+      "eval_mar_100_stop-sign": 0.976,
+      "eval_mar_100_tennis-racket": 0.842,
+      "eval_mar_100_umbrella": 0.9544,
+      "eval_mar_100_volleyball": 0.8806,
+      "eval_mar_large": 0.8579,
+      "eval_mar_medium": 0.8766,
+      "eval_mar_small": 0.7056,
+      "eval_runtime": 70.3979,
+      "eval_samples_per_second": 21.307,
+      "eval_steps_per_second": 0.668,
+      "step": 4161
+    },
+    {
+      "epoch": 20.0,
+      "grad_norm": 16.554805755615234,
+      "learning_rate": 5.000965402951676e-06,
+      "loss": 4.3268,
+      "step": 4380
+    },
+    {
+      "epoch": 20.0,
+      "eval_loss": 5.516897678375244,
+      "eval_map": 0.7975,
+      "eval_map_50": 0.9083,
+      "eval_map_75": 0.8523,
+      "eval_map_baseball-bat": 0.2923,
+      "eval_map_basketball": 0.8923,
+      "eval_map_car": -1.0,
+      "eval_map_football": 0.5264,
+      "eval_map_human": 0.9058,
+      "eval_map_large": 0.8257,
+      "eval_map_luggage": 0.8227,
+      "eval_map_mattress": 0.9768,
+      "eval_map_medium": 0.7526,
+      "eval_map_motorcycle": 0.7936,
+      "eval_map_skis": 0.8144,
+      "eval_map_small": 0.6611,
+      "eval_map_snowboard": 0.9197,
+      "eval_map_soccer-ball": 0.8408,
+      "eval_map_stop-sign": 0.9465,
+      "eval_map_tennis-racket": 0.6958,
+      "eval_map_umbrella": 0.8818,
+      "eval_map_volleyball": 0.8557,
+      "eval_mar_1": 0.7621,
+      "eval_mar_10": 0.8843,
+      "eval_mar_100": 0.8899,
+      "eval_mar_100_baseball-bat": 0.6389,
+      "eval_mar_100_basketball": 0.9235,
+      "eval_mar_100_car": -1.0,
+      "eval_mar_100_football": 0.6421,
+      "eval_mar_100_human": 0.9585,
+      "eval_mar_100_luggage": 0.9242,
+      "eval_mar_100_mattress": 0.9944,
+      "eval_mar_100_motorcycle": 0.957,
+      "eval_mar_100_skis": 0.9614,
+      "eval_mar_100_snowboard": 0.9525,
+      "eval_mar_100_soccer-ball": 0.8735,
+      "eval_mar_100_stop-sign": 0.9754,
+      "eval_mar_100_tennis-racket": 0.8176,
+      "eval_mar_100_umbrella": 0.9516,
+      "eval_mar_100_volleyball": 0.8882,
+      "eval_mar_large": 0.9814,
+      "eval_mar_medium": 0.8664,
+      "eval_mar_small": 0.7154,
+      "eval_runtime": 70.5374,
+      "eval_samples_per_second": 21.265,
+      "eval_steps_per_second": 0.666,
+      "step": 4380
+    },
+    {
+      "epoch": 20.0,
+      "step": 4380,
+      "total_flos": 4.4130028142592e+19,
+      "train_loss": 5.691855750236337,
+      "train_runtime": 6795.0216,
+      "train_samples_per_second": 20.603,
+      "train_steps_per_second": 0.645
+    }
+  ],
+  "logging_steps": 500,
+  "max_steps": 4380,
+  "num_input_tokens_seen": 0,
+  "num_train_epochs": 20,
+  "save_steps": 500,
+  "stateful_callbacks": {
+    "TrainerControl": {
+      "args": {
+        "should_epoch_stop": false,
+        "should_evaluate": false,
+        "should_log": false,
+        "should_save": true,
+        "should_training_stop": true
+      },
+      "attributes": {}
+    }
+  },
+  "total_flos": 4.4130028142592e+19,
+  "train_batch_size": 32,
+  "trial_name": null,
+  "trial_params": null
+}

training_args.bin ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:d376879dd051fb357ea7fecabf245405828e80767f79d19329504424688e74e9
+size 5432