Niksa Praljak commited on
Commit ·
0354528
1
Parent(s): f782d11
Update PenCL README
Browse files- weights/PenCL/README.md +2 -2
weights/PenCL/README.md
CHANGED
|
@@ -17,6 +17,7 @@ To download the **PenCL epoch 20 pre-trained weights** as a `.bin` file from Goo
|
|
| 17 |
```bash
|
| 18 |
pip install gdown
|
| 19 |
gdown --id 1Lup7Xqwa1NjJpoM2uvvBAdghoM-fecEj -O BioM3_PenCL_epoch20.bin
|
|
|
|
| 20 |
|
| 21 |
---
|
| 22 |
|
|
@@ -65,5 +66,4 @@ if __name__ == '__main__':
|
|
| 65 |
model.load_state_dict(torch.load(model_weights_path, map_location="cpu"))
|
| 66 |
model.eval()
|
| 67 |
print("Model loaded successfully with weights!")
|
| 68 |
-
|
| 69 |
-
|
|
|
|
| 17 |
```bash
|
| 18 |
pip install gdown
|
| 19 |
gdown --id 1Lup7Xqwa1NjJpoM2uvvBAdghoM-fecEj -O BioM3_PenCL_epoch20.bin
|
| 20 |
+
```
|
| 21 |
|
| 22 |
---
|
| 23 |
|
|
|
|
| 66 |
model.load_state_dict(torch.load(model_weights_path, map_location="cpu"))
|
| 67 |
model.eval()
|
| 68 |
print("Model loaded successfully with weights!")
|
| 69 |
+
```
|
|
|