Include check for cuda in seed utility
Closed this issue · 0 comments
einbandi commented
Seed utility function throws an error if torch.verison.cuda does not return anything. Include a check for cuda and also check other places in the code that assume cuda.