Update receptive_field.py

Fangyh09 · May 9, 2024 · 0aeb7f8 · 0aeb7f8
1 parent cdeb9c4
commit 0aeb7f8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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