@ -4,7 +4,7 @@ import sys
import torch
sys.path.append('../')
from models.model import MNISTmodel, MNISTmodel_plain
from core.dann import train_dann
from core.train import train_dann
from utils.utils import get_data_loader, init_model, init_random_seed
from core.test import test
from models.model import AlexModel
@ -2,7 +2,7 @@ import os
import sys
@ -5,7 +5,7 @@ import torch
from models.model import SVHNmodel