I am busy with an audio generation diffusion project and would like to save and image and an audio file every epoch and track their evolving as artifacts, here is my current code:
Log media table
wandb_table_media = wandb.Table(
columns=['Epoch', 'Step', 'Clean-Images',
'Generated-Mel-Images', 'Generated-Audio'])
img_shape = np.reshape(images[0], (1, 256, 256))
wandb_table_media.add_data(
epoch,
global_step, wandb.Image(clean_images[0]),
wandb.Image(img_shape),
wandb.Audio(normalize(audios[0]), sample_rate=sample_rate))
wandb.log({'wandb_table_media': wandb_table_media})
# Log media artifact
media_artifact = wandb.Artifact(
f'media-table-{args.project_name}',
type='table',
description='media-table'
)
media_artifact.add(wandb_table_media, "media-table-sonic-diffusion")
wandb.log_artifact(media_artifact,
aliases=[f'step_{global_step}', f'epoch_{epoch}'])
It logs the table artifact fine in WANDB…how do i create a table in WANDB to view these images and audio files?