Function restore_collection

tensorpack/tfutils/collection.py:37–46

Restore from a collection backup. Args: backup (dict):

(backup)

def restore_collection(backup):
    """
    Restore from a collection backup.

    Args:
        backup (dict):
    """
    for k, v in six.iteritems(backup):
        del tf.get_collection_ref(k)[:]
        tf.get_collection_ref(k).extend(v)


@contextmanager