-
sdarkhovsky authored
* updated the build_generator call to include the models argument * fixed RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same
b6a61c3f
* updated the build_generator call to include the models argument * fixed RuntimeError: Input type (torch.FloatTensor) and weight type (torch.cuda.FloatTensor) should be the same