Update model card for new model version

#3
by AndiLindner - opened
Files changed (1) hide show
  1. README.md +8 -3
README.md CHANGED
@@ -29,8 +29,12 @@ it were only 31. Not all butterfly and moth species can be determined based on i
29
 
30
  ## Training Details
31
 
32
- The model was trained on the EuroHPC supercomputers LUMI, hosted by CSC (Finland) and the LUMI consortium through a EuroHPC Regular Access call.
33
- Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library. More information on model training can be found here (link to publication) and scripts are availble here (link to github).
 
 
 
 
34
 
35
  ### Training Data
36
 
@@ -40,7 +44,7 @@ Species with less than 50 images were excluded from training. The final dataset
40
 
41
  ## Citation
42
 
43
- The model was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
44
 
45
  ```bibtex
46
  @Article{Barkmannetal2025a,
@@ -64,3 +68,4 @@ url={https://doi.org/10.1038/s41597-025-05708-z}
64
  }
65
  ```
66
 
 
 
29
 
30
  ## Training Details
31
 
32
+ The first model version ([a9ba52f](https://huggingface.co/RikeB/MaxViT_butterfly_identification/commit/a9ba52fd730b165b8fcd58265076fa82e763ebc2)) was trained on the EuroHPC supercomputer LUMI, hosted by CSC (Finland) and the LUMI consortium through a EuroHPC Regular Access call.
33
+ The second model version ([2371ff8](https://huggingface.co/RikeB/MaxViT_butterfly_identification/commit/2371ff89263bcc0319117d2a6e452e1dc5a82b4e)) was trained on the EuroHPC supercomputer LEONARDO, hosted by CINECA (Italy) and the LEONARDO consortium, also through a EuroHPC Regular Access call.
34
+
35
+ Training was parallelized using the Pytorch DDP framework and the Hugging Face Accelerate library.
36
+ More information on model training can be found in the publications below.
37
+ Scripts are availble on [GitHub](https://github.com/AndiLindner/butterfly_identification).
38
 
39
  ### Training Data
40
 
 
44
 
45
  ## Citation
46
 
47
+ The first model version was trained in the context of a data paper in which the butterfly and moth images dataset it was trained on was published:
48
 
49
  ```bibtex
50
  @Article{Barkmannetal2025a,
 
68
  }
69
  ```
70
 
71
+ Another publication is in preparation.