temporary commit for lowercase fix

This commit is contained in:
Falko Victor Habel 2025-02-23 19:26:36 +01:00
parent 736886021c
commit 2883e67e1c
2 changed files with 2 additions and 1 deletions

View File

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

View File

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