'GeneralTracker' object has no attribute 'log_images'

[RohollahHS]

Hi, when trying to log the reconstructed images in the stage 1 training, I got this error

  File "/home/rohol/projects/1d-tokenizer/utils/train_utils.py", line 762, in reconstruct_images
    accelerator.get_tracker("wandb").log_images(
AttributeError: 'GeneralTracker' object has no attribute 'log_images'

[cornettoyu]

It seems to me there is some problem at accelerator lib which should get WandBTracker instance that supports log_images

Could you double check the accelerate and wandb version? Or alternatively, you can switch to tensorboard by setting config.training.enable_wandb=False. Or, you can just disable all loggers as we also implement saving visualization to local storage so it should work as well.

[RohollahHS]

Thanks for your response. Yes, I also tested tensorboard and it also returns an error, so I commented out both wand and tensorboard and continued with saving them locally.