(self, _distributed_cache_data=None)
| 138 | |
| 139 | class CustomDynamicCacheOnCPU(CustomDynamicCache): |
| 140 | def __init__(self, _distributed_cache_data=None): |
| 141 | super().__init__(_distributed_cache_data) |
| 142 | |
| 143 | def record_kwargs(self, layer_idx, kwargs): |
| 144 | d = {} |