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

finetune_class #1

Merged
Fabel merged 96 commits from finetune_class into develop 2025-02-26 12:13:09 +00:00
Conversation 0 Commits 96 Files Changed 15 +2 -1
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files
Showing only changes of commit 2883e67e1c - Show all commits

2
src/aiunn/finetune.py
View File

@ -12,7 +12,7 @@ from torchvision import transforms
from tqdm import tqdm from tqdm import tqdm
from aiia import AIIABase from aiia import AIIABase
from upsampler import Upsampler from aiunn.upsample import Upsampler
# Define a simple EarlyStopping class to monitor the epoch loss. # Define a simple EarlyStopping class to monitor the epoch loss.
class EarlyStopping: class EarlyStopping:

1
src/aiunn/Upsampler.py → src/aiunn/upsample.py
View File

@ -2,6 +2,7 @@ import torch
import torch.nn as nn import torch.nn as nn
from aiia import AIIA, AIIAConfig, AIIABase from aiia import AIIA, AIIAConfig, AIIABase
class Upsampler(AIIA): class Upsampler(AIIA):
def init(self, base_model: AIIA): def init(self, base_model: AIIA):
# base_model must be a fully instantiated model (with a .config attribute) # base_model must be a fully instantiated model (with a .config attribute)