Reloading Alex LPIPS model every 777 timesteps

NoahVl / Explaining-In-Style-Reproducibility-Study

Re-implementation of the StylEx paper, training a GAN to explain a classifier in StyleSpace, paper by Lang et al. (2021).

Other

36 stars 8 forks source link

Reloading Alex LPIPS model every 777 timesteps #4

Open NoahVl opened 2 years ago

NoahVl commented 2 years ago

The LPIPS models seem to get reloaded every 777 timesteps, as if all the globals are then reloaded for some reason. Might be fixed if we place these LPIPS loss model in the Trainer class instead of in the globals. Also strange that it seems to be loaded 4 times everytime, does this have to with the amount of workers?

NoahVl commented 2 years ago

Seems to be a windows issue, might investigate later.