Youtube-Code-Repository
Youtube-Code-Repository copied to clipboard
small corrections in row 54
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')