RF Parameter fix for bool dtype

rwth-i6 · Nov 17, 2023 · a9a6690 · a9a6690
1 parent 55ede50
commit a9a6690
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/returnn/torch/frontend/_backend.py b/returnn/torch/frontend/_backend.py
@@ -644,7 +644,7 @@ def create_parameter_raw(tensor: rf.Parameter, *, device: Optional[str] = None)
             dtype=TorchBackend.as_dtype_raw(tensor.dtype),
             device=device or rf.get_default_device(),
         )
-        if tensor.dtype.startswith("int"):
+        if tensor.dtype.startswith("int") or tensor.dtype == "bool":
             requires_grad = False
         else:
             requires_grad = True