diff --git a/scripts/tools/train.py b/scripts/tools/train.py index 5acbfef..f305352 100644 --- a/scripts/tools/train.py +++ b/scripts/tools/train.py @@ -255,8 +255,6 @@ def train( } executor_kwargs = { - "static_graph": True, - "find_unused_parameters": False, "gradient_as_bucket_view": True, "broadcast_buffers": False, }