storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-18 07:33:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

tracker logging

Browse Source
This commit is contained in:
sayakpaul
2024-12-27 12:25:11 +05:30
parent 739f3d8535
commit 4461cdd98f
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
finetrainers/trainer.py
+9 -1
View File
@@ -923,7 +923,15 @@ class Trainer:
tracker_key = "final" if final_validation else "validation" tracker_key = "final" if final_validation else "validation"
for tracker in accelerator.trackers: for tracker in accelerator.trackers:
if tracker.name == "wandb": if tracker.name == "wandb":
tracker.log({tracker_key: all_artifacts}, step=step) image_artifacts = [artifact for artifact in all_artifacts if isinstance(artifact, wandb.Image)]
video_artifacts = [artifact for artifact in all_artifacts if isinstance(artifact, wandb.Video)]
tracker.log(
{
tracker_key: {"images": image_artifacts, "videos": video_artifacts},
},
step=step,
)
# Remove all hooks that might have been added during pipeline initialization to the models # Remove all hooks that might have been added during pipeline initialization to the models
pipeline.remove_all_hooks() pipeline.remove_all_hooks()