infogan line 102 when I try to apply it on celebA with size (channel,height,width) as (3,180,180)
as the code below, the data from dataloader size is (batch,channel,height,width), but when choose the [0] the data size become (3,180,180), so at the
self.D = discriminator(input_dim=data.shape[1]
the input_dim become 180 rather than 3,I remove the [0] to fix it
infogan line 102 when I try to apply it on celebA with size (channel,height,width) as (3,180,180) as the code below, the data from dataloader size is (batch,channel,height,width), but when choose the [0] the data size become (3,180,180), so at the
the input_dim become 180 rather than 3,I remove the [0] to fix it