fallcat commited on
Commit
55bf9d9
1 Parent(s): c17b88c

Add hf model

Browse files
Files changed (4) hide show
  1. .gitignore +1 -0
  2. README.md +7 -0
  3. config.json +8 -0
  4. pytorch_model.bin +0 -0
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .ipynb_checkpoints
README.md CHANGED
@@ -1,3 +1,10 @@
1
  ---
2
  license: mit
3
  ---
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
  ---
4
+
5
+ To load the pretrained model:
6
+ ```
7
+ from exlib.datasets.massmaps import MassMapsConvnetRegModel
8
+
9
+ model = MassMapsConvnetForImageRegression.from_pretrained(f'BrachioLab/MassMapsConvnet')
10
+ ```
config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "MassMapsConvnetForImageRegression"
4
+ ],
5
+ "num_classes": 2,
6
+ "torch_dtype": "float32",
7
+ "transformers_version": "4.34.0"
8
+ }
pytorch_model.bin ADDED
Binary file (793 kB). View file