dl4ds
dl4ds copied to clipboard
static_vars = None fails for the CGANTrainer
If static_vars is assigned, then the aux_hr variable is defined. However, for static_vars = None, aux_hr is not defined and the code fails. See lines 338-341 (https://github.com/carlos-gg/dl4ds/blob/232ae495151a4d3607402b3e7c8afd51d6bfa3de/dl4ds/training/cgan.py#L338). I added a aux_hr=None on line 342 and got the code to run.
There is a similar issue on line 440 where hr_arrtest is not defined if the static_vars is None.