danieltanhx

and need to set the initial value for --lambda_ds because of line 95 and line 143 of core/solver.py # remember the initial value of ds weight initial_lambda_ds = args.lambda_ds

try to modify the Dataset class just like in the link->https://github.com/Hramchenko/simplified_pix2pixHD/blob/main/simplified_pix2pixHD.ipynb

import os os.environ[“WANDB_DISABLED”] = “true” will do the job https://discuss.huggingface.co/t/how-to-turn-wandb-off-in-trainer/6237