Sheng Wang
Sheng Wang
``` pretrained_v = timm.create_model('vit_base_patch16_224', pretrained=True) pretrained_v.head = nn.Linear(768,2) ``` I tried Kaggle Cats vs Dogs Dataset for binary classification. Didn't work, output is all cat or all dog. Any idea...
I run: ```python test_expVAE.py``` And get error as follows: ``` cuda available Traceback (most recent call last): File "test_expVAE.py", line 99, in main() File "test_expVAE.py", line 77, in main gcam.backward(mu,...
重参数化是自己写的,想知道有人LoRA+int4成功的吗?