code restructure and remove parser

[yasharora102]

• Removes the unnecessary arg parser
• PSNR only initialized once
• unnecessary print statements removed

Regarding the wandb logging, we are already providing the loss as well as the PSNR. Can you please clarify this? Thank You! @saisriteja

Fixes #1 #2