-
您好! 我事先把brisque_svm_weights.pth文件下载到本地,然后想初始化时从本地加载brisque_svm_weights.pth,基本代码逻辑如下:
device = torch.device("cuda") if torch.cuda.is_available() else torch.device("cpu")
model_BRISQUE = pyiqa.create_…
-
@Wang-Yu-Qing hello there! 👋
Absolutely, you can assign class weights directly in the code. If you're using a Python script to train your model, you can set the class weights in the `…
-
### System Info
- CPU: x86_64
- GPU: A10 (24G)
### Who can help?
_No response_
### Information
- [x] The official example scripts
- [ ] My own modified scripts
### Tasks
- [x] An officially su…
-
can you release the query weight? thanks
-
Dear all,
It would be great to see an end-to-end practical example of LoTR. By "practical" I mean that one takes, for example some existing LLM weights file, compresses it into a smaller weights fi…
-
Hello!
Great work!
I tried to load the light weight model following
`https://github.com/smiles724/VQMAE/blob/master/infer_new_pdb.py#L62C5-L62C47`
using the code
```
ckpt = torch.lo…
-
### Your current environment
The output of `python collect_env.py`
```text
Collecting environment information...
PyTorch version: 2.4.0a0+git74cd574
Is debug build: False
CUDA used to build …
-
I am having some issues with the DeepMreye demo using the exemplary data from the 2 first participants from the sample dataset as instructed in the notebook "deepmreye_example_usage_pretrained_model_w…
-
### 💡 Your Question
Hello,
I am experiencing an issue while trying to load pre-trained weights for YOLO-NAS using the SuperGradients library.
Here is the code snippet I am using:
python
Copy …
-
I noticed there are model weights for the "chair" and "car" classes, but none for "airplane." Can you please provide the model weights for the "airplane" class?