Skip to content

DiffusionModelEncoder output layer is set outside of the initiation and left out of trainable parameters and set to the wrong device #8577

@virginiafdez

Description

@virginiafdez

Describe the bug
This issue has been observed in the tutorials: Project-MONAI/tutorials#2030 (comment).

Expected behavior
The out layer should not be set in the forward loop because:

  • The device is wrong in this scenario
  • The parameters from this layer are not included in the optimizer trainable args.

Additional context
This was caused by fix #8514.

Metadata

Metadata

Assignees

Labels

GenerativeModelsIssues relating to generative modelsbugSomething isn't working

Type

No fields configured for Bug.

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions