from config.svhn_avg.base import * from src.model.classifier import LMTraining from src.extension.c_scheduler import CScheduler from src.extension.c_scheduler import GoyalSchedule mode = ['lmt', 'lmt-fc'] start_c = 1e-5 end_c = 1e-1 model = LMTraining(predictor, preprocess, c=start_c) extension += [(CScheduler(GoyalSchedule(start_c, end_c, 5)), (1, 'iteration'))]
from config.outer_polytope.base import * from src.model.classifier import LMTraining mode = ['lmt', 'lmt-fc'] model = LMTraining(predictor, preprocess, c=50.)
from config.outer_polytope.base import * from src.model.classifier import LMTraining from config.attack.cw_30_1_1_1 import attacker, attacker_args, attacker_kwargs attacker = attacker(predictor, *attacker_args, **attacker_kwargs) mode = ['default'] model = LMTraining(predictor, preprocess, attacker=attacker) attacker.register_model(model)