fixed paths
This commit is contained in:
parent
e52a2a4b81
commit
b740819757
|
@ -12,7 +12,7 @@ from torchvision import transforms
|
|||
from tqdm import tqdm
|
||||
|
||||
from aiia import AIIABase
|
||||
from aiunn.upsampler import Upsampler
|
||||
from upsampler import Upsampler
|
||||
|
||||
# Define a simple EarlyStopping class to monitor the epoch loss.
|
||||
class EarlyStopping:
|
||||
|
|
Loading…
Reference in New Issue