search

saisriteja / DiffusionFlare

1 stars 0 forks source link

Code Quality #2

Open saisriteja opened 4 months ago

saisriteja commented 4 months ago
  1. init the PSNR again in the train loop is not necessary here
  2. Remove the print statements in the train loop here
  3. PSNR is one of the metrics, we should be ideally tracking the loss function most of the cases, so log the wandb of loss and then weighted loss as well here

@yasharora102

yasharora102 commented 4 months ago

Sure will make the necessary changes