The color loss function doesn't seem to clear self.loss. It keeps appending to the list each time before it calculates a mean. Simply added a clear().
Also removed pycache directories from being tracked by git so that the repo doesn't contain compiled files and doesn't see them as changed after recompilation.
The color loss function doesn't seem to clear self.loss. It keeps appending to the list each time before it calculates a mean. Simply added a clear().
Also removed pycache directories from being tracked by git so that the repo doesn't contain compiled files and doesn't see them as changed after recompilation.