cs230-stanford/cs230-code-examples

async=true fails since async is a keyword

johntiger1 opened this issue · 0 comments

train_batch, labels_batch = train_batch.cuda(async=True), labels_batch.cuda(async=True)