Skip to content

Instantly share code, notes, and snippets.

@virtualdvid
Last active January 15, 2019 19:21
Show Gist options
  • Save virtualdvid/eef3e24d0fb97ced30d7353ae2d1d4b9 to your computer and use it in GitHub Desktop.
Save virtualdvid/eef3e24d0fb97ced30d7353ae2d1d4b9 to your computer and use it in GitHub Desktop.
Gap: Flow with Augmentation
print("Gap: Flow with Augmentation")
dataset = Images(augment=['rotate=-30,30'])
dataset.load('flowers')
start_mem = psutil.virtual_memory().used
start_time = time.time()
dataset.minibatch = 32
train_generator = dataset.minibatch
print("PRE-TIME", time.time() - start_time)
start_time = time.time()
step = 0
for x_batch, y_batch in train_generator:
step += 1
if step > STEPS: break
print("TIME", time.time() - start_time)
curr_mem = psutil.virtual_memory().used
print("Memory Used: %.2f GB" % ((curr_mem - start_mem) / GB))
# Release unused memory
gc.collect()
time.sleep(5)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment