text_html_output_4.png text_html_output_6.png text_html_output_2.png text_html_output_5.png text_html_output_1.png text_html_output_8.png text_html_output_3.png text_html_output_7.png
from pytorch_lightning.loggers import WandbLogger

wandb_logger = WandbLogger(project='gender-detection-vit')

model = LitModel(2, wandb_logger=wandb_logger)