index out of bounds when directions_count is less than 512

anvoynov / GANLatentDiscovery

The authors official implementation of Unsupervised Discovery of Interpretable Directions in the GAN Latent Space

416 stars 52 forks source link

index out of bounds when directions_count is less than 512 #10

Open eps696 opened 4 years ago

eps696 commented 4 years ago

max_dim to inspect in inspect_all_directions is set to G.dim_shift; if directions_count is set to lower value (say, 200 instead of 512), there's "index out of bounds" error

lazavgeridis commented 3 years ago

@eps696 @anvoynov this can be fixed by changing the line max_dim = G.dim_shift to max_dim = deformator.input_dim in inspect_all_directions()