diff --git a/distiller/utils.py b/distiller/utils.py index 29e8bde9760eccdf022e42343daffb2f8d60c7f7..17d1097c52b02fdb87ced151ba86fc78d08d3e65 100755 --- a/distiller/utils.py +++ b/distiller/utils.py @@ -72,8 +72,12 @@ def assign_layer_fq_names(container, name=None): """ is_leaf = True for key, module in container._modules.items(): - is_leaf = False - assign_layer_fq_names(module, ".".join([name, key]) if name is not None else key) + try: + assign_layer_fq_names(module, ".".join([name, key]) if name is not None else key) + is_leaf = False + except AttributeError: + if module is not None: + raise if is_leaf: container.distiller_name = name