wandb.py

with warnings.catch_warnings():
    warnings.filterwarnings("ignore", category=FutureWarning)
    import tensorflow as tf

# WandB – Login to your wandb account so you can log all your metrics
wandb.login()

wandb.init(project="hierarchical_cma", sync_tensorboard=True)

wb_config = wandb.config
wb_config.LR = 1e-4
wb_config.EPOCHS = 20
wb_config.BATCH_SIZE = 1

with TensorboardWriter(
    wandb.run.dir, flush_secs=self.flush_secs, purge_step=0
) as writer: