nnclib package

Submodules

nnclib.count_all_params module

nnclib.download_keras_models module

nnclib.generators module

nnclib.make_dense_weights module

nnclib.test_gpu module

nnclib.utils module

Utility functions for various purposes.

nnclib.utils.get_results_dir(file, base='.')

Generate a string for the file observer.

nnclib.utils.reshape_weights(weights)

Takes a \(d_1 \times d_2 \times \ldots \times d_{n-1} \times d_n\) dimensional tensor, and reshapes it to a \(d_1 \cdots d_{n-2} \cdot d_n \times d_{n-1}\) dimensional matrix.

nnclib.utils.sum_weights(pairs)

Calculate the sum of weights from a list of pairs. This function is used to summarise the compression of weights. It process the output of TODO.

Module contents