Commit 17d8feb5 authored by Yuxin Wu's avatar Yuxin Wu

fix cifardata bug

parent a41f9e83
......@@ -28,7 +28,7 @@ def maybe_download_and_extract(dest_directory):
logger.info("Found cifar10 data in {}.".format(dest_directory))
return
else:
download(URL, dest_directory)
download(DATA_URL, dest_directory)
filename = DATA_URL.split('/')[-1]
filepath = os.path.join(dest_directory, filename)
tarfile.open(filepath, 'r:gz').extractall(dest_directory)
......
......@@ -18,6 +18,7 @@ def get_default_sess_config(mem_fraction=0.5):
conf.gpu_options.per_process_gpu_memory_fraction = mem_fraction
conf.gpu_options.allocator_type = 'BFC'
conf.allow_soft_placement = True
#conf.log_device_placement = True
return conf
def get_global_step_var():
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment