ONNX
josh-gregory commited on
Commit
9d1a3f1
·
verified ·
1 Parent(s): c0bb35a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -6
README.md CHANGED
@@ -23,7 +23,6 @@ There are several models that were trained on the ClotSimNet dataset:
23
  The EfficientNet and ResNet models were chosen due to their precedence in the biomedical imaging field. ConvNeXt-Tiny was selected to explore a new architecture based on vision transformers to assess its performance relative to the more conventional CNN architectures.
24
 
25
 
26
-
27
  - **Developed by:** Josh Gregory
28
 
29
 
@@ -51,11 +50,11 @@ As of this publication, these models are not intended to treat, diagnose, or pre
51
 
52
  Each model was exported in [ONNX](https://onnx.ai/) (`.onnx`) and [Safetensors](https://huggingface.co/docs/safetensors/index) (`.safetensors`).
53
 
54
- See the `inference_examples` directory for how to inference these models in safetensors format. For futher examples in ONNX, refer to the GitHub repository, with examples located in the path `/poresimnet/ml/inference`.
55
 
56
  ### Loading the Models
57
 
58
- See the `inference_examples` directory. Models in the safetensors format were used the most for internal inference work.
59
 
60
  ## Training Details
61
 
@@ -83,7 +82,7 @@ Refer to the Dataset card to find the relevant testing data, as different kinds
83
 
84
  #### Summary
85
 
86
- The `base` directory contains all trained models with base hyperparameters. The `tuned` directory contains the models after hyperparameter tuning. The `hyperparameters` folder itself contains each model's tuned hyperparameters in a JSON file, and the `inference_examples` directory contains examples for how to inference our models in safetensors format.
87
 
88
  ## Model Examination
89
 
@@ -115,7 +114,7 @@ Several architectures were considered. Both MLPs and CNNs were selected due to t
115
 
116
  ### Compute Infrastructure
117
 
118
- This work utilized the Alpine high performance computing resource at the University of Colorado Boulder. Alpine is jointly funded by the University of Colorado Boulder, the University of Colorado Anschutz, and Colorado State University and with support from NSF grants OAC-2201538 and OAC-2322260
119
 
120
  #### Hardware
121
 
@@ -139,7 +138,7 @@ All models were written in PyTorch Lightning.
139
 
140
  ## Model Card Authors
141
 
142
- Josh Gregory
143
 
144
  ## Model Card Contact
145
 
 
23
  The EfficientNet and ResNet models were chosen due to their precedence in the biomedical imaging field. ConvNeXt-Tiny was selected to explore a new architecture based on vision transformers to assess its performance relative to the more conventional CNN architectures.
24
 
25
 
 
26
  - **Developed by:** Josh Gregory
27
 
28
 
 
50
 
51
  Each model was exported in [ONNX](https://onnx.ai/) (`.onnx`) and [Safetensors](https://huggingface.co/docs/safetensors/index) (`.safetensors`).
52
 
53
+ See the `inference_examples` directory for how to inference these models in Safetensors format. For futher examples in ONNX, refer to the GitHub repository, with examples located in the path `/poresimnet/ml/inference`.
54
 
55
  ### Loading the Models
56
 
57
+ See the `inference_examples` directory. Models in the Safetensors format were used the most for internal inference work.
58
 
59
  ## Training Details
60
 
 
82
 
83
  #### Summary
84
 
85
+ The `base` directory contains all trained models with base hyperparameters. The `tuned` directory contains the models after hyperparameter tuning. The `hyperparameters` folder itself contains each model's tuned hyperparameters in a JSON file, and the `inference_examples` directory contains examples for how to inference our models in Safetensors format.
86
 
87
  ## Model Examination
88
 
 
114
 
115
  ### Compute Infrastructure
116
 
117
+ This work utilized the Alpine high performance computing resource at the University of Colorado Boulder. Alpine is jointly funded by the University of Colorado Boulder, the University of Colorado Anschutz, and Colorado State University and with support from NSF grants OAC-2201538 and OAC-2322260.
118
 
119
  #### Hardware
120
 
 
138
 
139
  ## Model Card Authors
140
 
141
+ Josh Gregory: josh.a.gregory42@gmail.com
142
 
143
  ## Model Card Contact
144