diff --git a/torch_receptive_field/receptive_field.py b/torch_receptive_field/receptive_field.py index 4dc629f..1db5bff 100644 --- a/torch_receptive_field/receptive_field.py +++ b/torch_receptive_field/receptive_field.py @@ -79,7 +79,7 @@ def hook(module, input, output): receptive_field[m_key]["r"] = 0 receptive_field[m_key]["start"] = 0 else: - raise ValueError("module {} not ok".format(class_name)) + raise ValueError("module {} not supported yet".format(class_name)) pass receptive_field[m_key]["input_shape"] = list(input[0].size()) # only one receptive_field[m_key]["input_shape"][0] = batch_size