diff --git a/sot/opcode_translator/skip_files.py b/sot/opcode_translator/skip_files.py index fd38b425..58053b61 100644 --- a/sot/opcode_translator/skip_files.py +++ b/sot/opcode_translator/skip_files.py @@ -119,7 +119,7 @@ def _module_dir(m: types.ModuleType): customed_skip_code = set() -no_skip_code = {paddle.nn.Sequential.forward.__code__} +no_skip_code = {paddle.nn.Sequential.forward.__code__, paddle.nn.Layer.__call__} def need_skip_path(filepath: str) -> bool: diff --git a/sot/opcode_translator/transform.py b/sot/opcode_translator/transform.py index b17d6bbb..3a9606f4 100644 --- a/sot/opcode_translator/transform.py +++ b/sot/opcode_translator/transform.py @@ -47,7 +47,7 @@ def eval_frame_callback(frame, **kwargs) -> CustomCode: return CustomCode(None, True) if need_skip(frame): - return CustomCode(None, False) + return CustomCode(None, True) log( 2,