Minor fix

NKI-AI · Nov 4, 2024 · f0a2b4f · f0a2b4f
1 parent 5ca113b
commit f0a2b4f
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/direct/nn/registration/registration.py b/direct/nn/registration/registration.py
@@ -480,6 +480,7 @@ def __init__(
             gpsa_interval=gpsa_interval,
             locality_strength=locality_strength,
             use_pos_embedding=use_pos_embedding,
+            normalized=False,
         )
         self.max_seq_len = max_seq_len
         self.warp_num_integration_steps = warp_num_integration_steps