# HG changeset patch # User fsavard # Date 1271606097 14400 # Node ID 82dae7c4604684716077f2f5ca35a0af7c3d3346 # Parent b25ad1670ff7644560e63826fd676166621d1c1a Last few bugfixes before launching diff -r b25ad1670ff7 -r 82dae7c46046 deep/crbm/mnist_crbm.py --- a/deep/crbm/mnist_crbm.py Sun Apr 18 11:47:35 2010 -0400 +++ b/deep/crbm/mnist_crbm.py Sun Apr 18 11:54:57 2010 -0400 @@ -32,11 +32,13 @@ import utils -if not os.path.exists(IMAGE_OUTPUT_DIR): - os.mkdir(IMAGE_OUTPUT_DIR) -elif os.path.isfile(IMAGE_OUTPUT_DIR): - print "IMAGE_OUTPUT_DIR is not a directory!" - sys.exit(1) +if not (len(sys.argv) > 1 and sys.argv[1] in \ + ('jobman_insert',)): + if not os.path.exists(IMAGE_OUTPUT_DIR): + os.mkdir(IMAGE_OUTPUT_DIR) + elif os.path.isfile(IMAGE_OUTPUT_DIR): + print "IMAGE_OUTPUT_DIR is not a directory!" + sys.exit(1) #def filename_from_time(suffix): # import datetime diff -r b25ad1670ff7 -r 82dae7c46046 deep/crbm/utils.py --- a/deep/crbm/utils.py Sun Apr 18 11:47:35 2010 -0400 +++ b/deep/crbm/utils.py Sun Apr 18 11:54:57 2010 -0400 @@ -3,6 +3,7 @@ from __future__ import with_statement +import jobman from jobman import DD from pylearn.io.seriestables import *