diff --git a/_spaun/modules/stim/imagenet/data.py b/_spaun/modules/stim/imagenet/data.py index fba6171..55c2b90 100644 --- a/_spaun/modules/stim/imagenet/data.py +++ b/_spaun/modules/stim/imagenet/data.py @@ -106,8 +106,8 @@ def __init__(self, data_filepath=None): # --- Handle subsampling of probe data --- self.probe_subsample = 4 self.probe_image_shape = (self.image_shape[0], - self.image_shape[1] / self.probe_subsample, - self.image_shape[2] / self.probe_subsample) + self.image_shape[1] // self.probe_subsample, + self.image_shape[2] // self.probe_subsample) subsample_trfm = np.arange(np.cumprod(self.image_shape)[-1]) subsample_inds = \ diff --git a/_spaun/modules/stim/mnist/data.py b/_spaun/modules/stim/mnist/data.py index 390794a..8a5d4c9 100644 --- a/_spaun/modules/stim/mnist/data.py +++ b/_spaun/modules/stim/mnist/data.py @@ -60,8 +60,8 @@ def __init__(self, data_filepath=None): self.max_pixel_value = 1.0 self.probe_subsample = 1 - self.probe_image_shape = (self.image_shape[0] / self.probe_subsample, - self.image_shape[1] / self.probe_subsample) + self.probe_image_shape = (self.image_shape[0] // self.probe_subsample, + self.image_shape[1] // self.probe_subsample) subsample_trfm = np.arange(np.cumprod(self.image_shape)[-1]) subsample_inds = \ diff --git a/run_spaun.py b/run_spaun.py index f2b6b34..99f78d1 100644 --- a/run_spaun.py +++ b/run_spaun.py @@ -892,7 +892,7 @@ # Log subprocess call logger.write("\n#\n# To display graphs of the recorded probe data:") - logger.write("\n# " + " ".join(subprocess_call_list)) + logger.write("\n# > " + " ".join(subprocess_call_list)) if args.showgrph: # Open subprocess