yunjey

fixed to be compatible with python 3

......@@ -93,7 +93,7 @@ class DTN(object):
# spatial size for convolution
s = self.output_size
s2, s4, s8, s16 = s/2, s/4, s/8, s/16 # 32, 16, 8, 4
s2, s4, s8, s16 = int(s/2), int(s/4), int(s/8), int(s/16) # 32, 16, 8, 4
# project and reshape z
h1= linear(z, s16*s16*self.dim_gf*8, name='g_h1') # (batch_size, 2*2*512)
......
......@@ -51,7 +51,11 @@ class Solver(object):
else:
image_file = os.path.join(image_path, 'test.images.hkl')
images = hickle.load(image_file)
try:
images = hickle.load(image_file)
except:
hickle.load(images, image_file)
images = images / 127.5 - 1
print ('finished loading mnist image dataset..!')
return images
......