Skip to content

Commit

Permalink
add prefix when logging domain loss
Browse files Browse the repository at this point in the history
  • Loading branch information
bdvllrs committed Sep 20, 2024
1 parent 9811d43 commit af6461c
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion shimmer/modules/global_workspace.py
Original file line number Diff line number Diff line change
Expand Up @@ -533,7 +533,7 @@ def generic_step(self, batch: RawDomainGroupsT, mode: ModelModeT) -> STEP_OUTPUT
if unimodal_losses is not None:
for name, metric in unimodal_losses.all.items():
self.log(
f"{mode}/{name}",
f"{mode}/domain_loss/{name}",
metric,
batch_size=batch_size,
add_dataloader_idx=False,
Expand Down

0 comments on commit af6461c

Please sign in to comment.