Machine-Learning
/
aiuNN
1
0
Fork 0
Code Issues 2 Pull Requests Actions Packages Projects Releases Wiki Activity

develop #4

Merged
Fabel merged 103 commits from develop into main 2025-03-01 21:47:17 +00:00
Conversation 0 Commits 103 Files Changed 13 +2 -2
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 6a9b4afd91 - Show all commits

4
src/aiunn/finetune.py
View File

@ -3,7 +3,7 @@ import io
from PIL import Image from PIL import Image
from torch.utils.data import Dataset from torch.utils.data import Dataset
from torchvision import transforms from torchvision import transforms
from aiia import AIIA from aiia import AIIABase
import csv import csv
from tqdm import tqdm from tqdm import tqdm
@ -39,7 +39,7 @@ import torch
pretrained_model_path = "/root/vision/AIIA/AIIA-base-512" pretrained_model_path = "/root/vision/AIIA/AIIA-base-512"
# Load the model using the AIIA.load class method (the implementation copied in your query) # Load the model using the AIIA.load class method (the implementation copied in your query)
model = AIIA.load(pretrained_model_path) model = AIIABase.load(pretrained_model_path)
device = torch.device("cuda" if torch.cuda.is_available() else "cpu") device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
model = model.to(device) model = model.to(device)
from torch import nn, optim from torch import nn, optim