Types & classes — github.com/cybertronai/gradient-checkpointing

CodeHub MCPcopy Create free account

hub / github.com/cybertronai/gradient-checkpointing / types & classes

Types & classes7 in github.com/cybertronai/gradient-checkpointing

⨍ Functions 280◇ Types & classes 7 ↳ Endpoints 2

↓ 10 callersClassOrderedSet

test/linearize.py:269

↓ 6 callersClassBufferedWriter

Class that aggregates multiple writes and flushes periodically.

test/deep_resnet_benchmark.py:151

↓ 2 callersClassDataLoader

an object that generates batches of CIFAR-10 data for training

test/data/cifar10_data.py:55

↓ 2 callersClassTemporaryFileHelper

Provides a way to fetch contents of temporary file.

test/util.py:301

ClassDataLoader

an object that generates batches of CIFAR-10 data for training

test/data/imagenet_data.py:87

Classcapture_stderr

Utility to capture output, use as follows with util.capture_stderr() as stderr: sess = tf.Session() print("Captured:", stderr.getval

test/util.py:258

Classcapture_stdout

Utility to capture output, use as follows with util.capture_stderr() as stderr: sess = tf.Session() print("Captured:", stderr.getval

test/util.py:279