Skip to content

Generator Cuda Call #2

@jaymefosa

Description

@jaymefosa

had to add a call to set the x variable to use cuda

    def forward(self, n, y):
        x = Variable(torch.randn((n, 32-10)), requires_grad=False)
        if use_cuda:
            x = x.cuda()
        x = torch.cat([x, y], 1)
        for layer in self.layers:
            x = layer(x)
        return x

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions