Update README.md
Browse files
README.md
CHANGED
|
@@ -60,10 +60,10 @@ To run this model, you need the proper environment. We recommend the following v
|
|
| 60 |
> [!NOTE]
|
| 61 |
> Make sure you have installed the compatible versions of **torch** and **torchvision** with your **CUDA driver** to leverage GPU.
|
| 62 |
## Pretrained Models
|
| 63 |
-
You can download the pretrained models from this [link](https://huggingface.co/
|
| 64 |
-
You can use latest checkpoint `latest_net_G.pth` for inference.
|
| 65 |
> [!NOTE]
|
| 66 |
-
> Please ensure your path to the checkpoint is correct in the script files before running.
|
| 67 |
|
| 68 |
## Usage Guide
|
| 69 |
|
|
|
|
| 60 |
> [!NOTE]
|
| 61 |
> Make sure you have installed the compatible versions of **torch** and **torchvision** with your **CUDA driver** to leverage GPU.
|
| 62 |
## Pretrained Models
|
| 63 |
+
You can download the pretrained models from this [link](https://huggingface.co/val-utehy/CR-Net/tree/main/checkpoints_v2/ast_rafael_v2_sharpening).
|
| 64 |
+
You can use latest checkpoint `latest_net_G.pth` and `opt.pkl` for inference.
|
| 65 |
> [!NOTE]
|
| 66 |
+
> Please ensure your path to the checkpoint and config (opt.pkl) is correct in the script files before running.
|
| 67 |
|
| 68 |
## Usage Guide
|
| 69 |
|