I am trying to run GAN in a coursera course. But I am stuck at this error. How to solve it$?$

gen$\_$loss$\_$tensor: tensor$(0\mathrm{.}4753,$ device$=$'cuda:$0\text{'})$

$---------------------------------------------------------------------------$

AssertionError Traceback $($most recent call last$)$

Input In $[21],$ in $()$

$33$ assert not torch.all$($torch$.$eq$($old$\_$weight, new$\_$weight$)),$ "Generator weights did not update"

$36$ # Run tests

$--->37$ test$\_$gen$\_$reasonable$(10)$

$38$ test$\_$gen$\_$loss$(18)$

$39$ print$("$Success$!")$

Input In $[21],$ in test$\_$gen$\_$reasonable$($num$\_$images$)$

$12$ assert torch.all$($torch$.$abs$($gen$\_$loss$\_$tensor$)<1),$ "gen$\_$loss$\_$tensor values are not close to zero"

$15$ # Verify shape. Related to gen$\_$noise parametrization

$--->16$ assert tuple$($gen$\_$loss$\_$tensor.shape$)==($num$\_$images, $1)$

AssertionError: