Skip to content

Only calculating response loss when response_predict config is set t…#105

Merged
WilliamYue37 merged 4 commits intomainfrom
fix/pi05_response_predict_false
Feb 5, 2026
Merged

Only calculating response loss when response_predict config is set t…#105
WilliamYue37 merged 4 commits intomainfrom
fix/pi05_response_predict_false

Conversation

@akshay18iitg
Copy link
Contributor

@akshay18iitg akshay18iitg commented Feb 5, 2026

…o true

What this does

Close #104

How it was tested

Ran training on local computer

Examples:
opentau-train --config_path=configs/examples/pi05_training_config.json

set the predict_response = False

How to checkout & try? (for the reviewer)

opentau-train --config_path=configs/examples/pi05_training_config.json

Checklist

  • I have added Google-style docstrings to important functions and ensured function parameters are typed.
  • My PR includes policy-related changes.
    • If the above is checked: I have run the GPU pytests (pytest -m "gpu") and regression tests.

Note: Before submitting this PR, please read the contributor guideline.

@akshay18iitg akshay18iitg self-assigned this Feb 5, 2026
@WilliamYue37 WilliamYue37 changed the title Only calculating respoonse loss when response_predict config is set t… Only calculating response loss when response_predict config is set t… Feb 5, 2026
@WilliamYue37 WilliamYue37 merged commit 6414dd9 into main Feb 5, 2026
5 checks passed
@WilliamYue37 WilliamYue37 deleted the fix/pi05_response_predict_false branch February 5, 2026 23:27
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Turning predict_response off during training causes a error

2 participants