[C1_W2_Assignment] Error trying to initialize generator, discriminator, optimizers

Hello DeepLearners and staff,

I’m trying to pass the assignment for the second week of the course and despite I pass all the unit test related to class Generator and class Discriminator, I get an error when I try to initialize the generator executing the next section of the code:

And the error is below:

Anyone can give me a clue about what is happening here?

Thanks in advance!!

I found it!!
So, if its useful to somebody, there is the explanation:

I set up to false the affine param so the code fail getting the weight of the BatchNorm in weights_init function, which were unexistent. Setting to True the affine param solved the problem.