diff --git a/MyTrain.py b/MyTrain.py index 160de2a..10e575b 100644 --- a/MyTrain.py +++ b/MyTrain.py @@ -3,7 +3,7 @@ import os import argparse from datetime import datetime -from lib.Fuzzy_Res2Net import FuzzyNet +from lib.FuzzyNet_Res2Net import FuzzyNet from utils.dataloader import get_loader,test_dataset from utils.utils import clip_gradient, adjust_lr, AvgMeter import torch.nn.functional as F