96d5ad9bf60fb840269e9a1a6c4189480da769f4,examples/dump_activations_to_disk.py,,,#,8

Before Change

if __name__ == "__main__":
    // Check for GPUs and set them to dynamically grow memory as needed
    // Avoids OOM from tensorflow greedily allocating GPU memory
    physical_devices = tf.config.list_physical_devices("GPU")
    if physical_devices:
        for dev in physical_devices:
            tf.config.experimental.set_memory_growth(dev, True)

    // define the model.
    model = Sequential()
    model.add(Dense(16, input_shape=(10,)))
    model.add(Dropout(0.5))
    model.add(Dense(10, activation="softmax"))

After Change

from keract import get_activations, persist_to_json_file, load_activations_from_json_file

if __name__ == "__main__":
    gpu_dynamic_mem_growth()

    // define the model.
    model = Sequential()
    model.add(Dense(16, input_shape=(10,)))

Instances