Skip to content

S
seminar-breakout
Commit 17d8feb5 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

fix cifardata bug

parent a41f9e83
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/dataflow/dataset/cifar10.py
View file @ 17d8feb5
...@@ -28,7 +28,7 @@ def maybe_download_and_extract(dest_directory): ...@@ -28,7 +28,7 @@ def maybe_download_and_extract(dest_directory):
logger.info("Found cifar10 data in {}.".format(dest_directory)) logger.info("Found cifar10 data in {}.".format(dest_directory))
return return
else: else:
download(URL, dest_directory) download(DATA_URL, dest_directory)
filename = DATA_URL.split('/')[-1] filename = DATA_URL.split('/')[-1]
filepath = os.path.join(dest_directory, filename) filepath = os.path.join(dest_directory, filename)
tarfile.open(filepath, 'r:gz').extractall(dest_directory) tarfile.open(filepath, 'r:gz').extractall(dest_directory)
......
tensorpack/tfutils/common.py
View file @ 17d8feb5
...@@ -18,6 +18,7 @@ def get_default_sess_config(mem_fraction=0.5): ...@@ -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.per_process_gpu_memory_fraction = mem_fraction
conf.gpu_options.allocator_type = 'BFC' conf.gpu_options.allocator_type = 'BFC'
conf.allow_soft_placement = True conf.allow_soft_placement = True
#conf.log_device_placement = True
return conf return conf
def get_global_step_var(): 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