Args: keys (list): list of collection keys to backup. Defaults to all keys in the graph. Returns: dict: the backup
(keys=None)
| 17 | |
| 18 | |
| 19 | def backup_collection(keys=None): |
| 20 | """ |
| 21 | Args: |
| 22 | keys (list): list of collection keys to backup. |
| 23 | Defaults to all keys in the graph. |
| 24 | |
| 25 | Returns: |
| 26 | dict: the backup |
| 27 | """ |
| 28 | if keys is None: |
| 29 | keys = tf.get_default_graph().get_all_collection_keys() |
| 30 | ret = {} |
| 31 | assert isinstance(keys, (list, tuple, set)) |
| 32 | for k in keys: |
| 33 | ret[k] = copy(tf.get_collection(k)) |
| 34 | return ret |
| 35 | |
| 36 | |
| 37 | def restore_collection(backup): |
no test coverage detected