Use set_default_tensor_type for Cuda #74

Open

opened

on Feb 18, 2021

if torch.cuda.is_available():
        torch.set_default_tensor_type('torch.cuda.FloatTensor')

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests