Skip to content

Conversation

@jeipollack
Copy link
Collaborator

This PR closes #1 by adding the missing initialization of self.epistemic_optimizer in the DiffusionActiveInference class, which triggered an AttributeError during training.

The following line was added:

self.epistemic_optimizer = torch.optim.Adam(
    self.epistemic_estimator.parameters(),
    lr=self.config.epistemic_lr if hasattr(self.config, "epistemic_lr") else 1e-4
)

To test:
Code was launched and training has proceeded well over 4 hours without the error

@jeipollack jeipollack requested a review from neuronphysics June 15, 2025 13:09
@jeipollack jeipollack self-assigned this Jun 15, 2025
@jeipollack jeipollack added the bug Something isn't working label Jun 15, 2025
@jeipollack jeipollack changed the title Add attribute self.epistemic_optimizer with learning_rate option Fix bug by adding attribute self.epistemic_optimizer with learning_rate option Jun 15, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug Something isn't working

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Bug: missing DiffusionActiveInference object attribute self.epistemic_optimizer

3 participants