Open kowalcorp opened 1 year ago
correction of small mistake in row 54:
before: self.device = T.device('cuda:0' if T.cuda.is_available() else 'cuda:1')
after: self.device = T.device('cuda:0' if T.cuda.is_available() else 'cpu')
correction of small mistake in row 54:
before: self.device = T.device('cuda:0' if T.cuda.is_available() else 'cuda:1')
after: self.device = T.device('cuda:0' if T.cuda.is_available() else 'cpu')