Skip to content

Commit 14acc2d

Browse files
committed
minor fix
1 parent 82f7df5 commit 14acc2d

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tpcav/tpcav_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -346,7 +346,7 @@ def hook_fn(_module, _inputs, output):
346346
handle.remove()
347347

348348
if not cache:
349-
raise RuntimeError(f"No activation captured for layer {self.layer_name}")
349+
raise RuntimeError(f"No activation captured for layer {self.layer_name}; Are you sure it's used in forward function?")
350350
return cache[0]
351351

352352
def _resolve_layer(self, name: str):

0 commit comments

Comments
 (0)