Update README.md
Browse files
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
|
| 55 |
|
| 56 |
### Loading the Models
|
| 57 |
|
| 58 |
-
See the `inference_examples` directory. Models in the
|
| 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
|
| 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 |
|