fix cifardata bug

tensorpack · Apr 10, 2016 · 17d8feb · 17d8feb
1 parent a41f9e8
commit 17d8feb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/tensorpack/dataflow/dataset/cifar10.py b/tensorpack/dataflow/dataset/cifar10.py
@@ -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)

diff --git a/tensorpack/tfutils/common.py b/tensorpack/tfutils/common.py
@@ -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():