Skip to content

Instantly share code, notes, and snippets.

View TimZaman's full-sized avatar

Tim Zaman TimZaman

View GitHub Profile
@TimZaman
TimZaman / tf_rotating_gpu_buffer.py
Created December 11, 2016 17:47
Tensorflow Rotating GPU Buffer
# Implementation of a rotating buffer on the GPU of size 2.
import threading
import tensorflow as tf
from tensorflow.python.client import timeline
import numpy as np
import time
params = {
'batch_size': 128,
'seg_len': 4000,