temporary commit for lowercase fix
This commit is contained in:
parent
736886021c
commit
2883e67e1c
|
@ -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:
|
||||
|
|
|
@ -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)
|
Loading…
Reference in New Issue