Add wandb logging (Life is more peacefull)

Add update logic

Sloving conflicts

Update disc_loss update logic

Made changes in the training script

Re training will finish this time for sure

More samples

Halfway through

Change checkpoint dir

Modify training code

Move to gpu and add checkpoint images to attributes

Print statements for logging

Make changes

Resolve run time errors

Change any output errors