armed.tfutils#

Utilities for Tensorflow sessions

Functions

set_gpu(gpu[, mem_frac])

Assign GPU to Tensorflow session and limit memory usage