Co-authored-by: Steven Palma <imstevenpmwork@ieee.org>
This commit is contained in:
Martino Russi
2026-07-30 15:25:26 +02:00
committed by GitHub
parent 2939168c33
commit 0187856202
+1 -1
View File
@@ -150,7 +150,7 @@ class XVLAModel(nn.Module):
# Freeze or unfreeze policy transformer
if not self.config.train_policy_transformer:
for name, param in self.transformer.named_parameters():
if "soft_prompts" not in name:
if "soft_prompt" not in name:
param.requires_grad = False
# Freeze or unfreeze soft prompts