search

philtabor / Youtube-Code-Repository

Repository for most of the code from my YouTube channel
859 stars 479 forks source link

small corrections in row 54 #53

Open kowalcorp opened 1 year ago

kowalcorp commented 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')