Skip to content

Commit

Permalink
change cifar download dir
Browse files Browse the repository at this point in the history
Change download directory to the local directory to avoid excessive download

Signed-off-by: Norman Heckscher <[email protected]>
  • Loading branch information
normanheckscher committed Jan 22, 2017
1 parent e916abd commit a17bc1c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions util.py
Original file line number Diff line number Diff line change
Expand Up @@ -98,15 +98,15 @@ def get_config(problem_name, path=None):
net_assignments = None
elif problem_name == "cifar":
mode = "train" if path is None else "test"
problem = problems.cifar10("/tmp/cifar10",
problem = problems.cifar10("cifar10",
conv_channels=(16, 16, 16),
linear_layers=(32,),
mode=mode)
net_config = {"cw": get_default_net_config("cw", path)}
net_assignments = None
elif problem_name == "cifar-multi":
mode = "train" if path is None else "test"
problem = problems.cifar10("/tmp/cifar10",
problem = problems.cifar10("cifar10",
conv_channels=(16, 16, 16),
linear_layers=(32,),
mode=mode)
Expand Down

0 comments on commit a17bc1c

Please sign in to comment.