from sacred import Experiment
from pytorch_utils.sacred_utils import get_model_path, read_config, import_source
from visdom_observer.visdom_observer import VisdomObserver
from sacred.observers import FileStorageObserver
from training_functions import validate
import adversarial as adv
# %%
ATTACKS = {'fgsm': adv.fgsm}
# %%
ex = Experiment('odenet-adv-mnist_time')
SAVE_DIR = 'runs/ODEnet-adversarial-test-mnist'
ex.observers.append(FileStorageObserver.create(SAVE_DIR))
ex.observers.append(VisdomObserver())
@ex.config
def input_config():
"""Parameters for sampling using the given model"""
run_dir = 'runs/ODEMnistClassification/17'
epoch = 'latest'
device = 'cpu'
epsilon = 0.3 # epsilon for attack
attack = 'fgsm' # type of attack, currently: [fgsm]
end_time_start = 0.1
end_time_end = 1000
num_times = 1000
tol = 1e-3
batches = -1
def apply(cls, args, run):
run.observers.append(VisdomObserver())