jku-vds-lab/paradime

Include check for cuda in seed utility

Closed this issue · 0 comments

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.