datnguyentien204 commited on
Commit
f0d78f1
·
verified ·
1 Parent(s): 20b545a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
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/datasets/datnguyentien204/CR-Net).
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