Implement Wasserstein Critic Loss — easy Loss Functions problem | Incognition