Commit Graph

4 Commits

Author SHA1 Message Date
Brett Kuprel
5aa6fe49bf use cuda if available 2022-06-28 12:47:11 -04:00
Brett Kuprel
8544f59576 use cuda if available 2022-06-28 12:38:31 -04:00
Brett Kuprel
aef24ea157 torch.no_grad(), cleanup 2022-06-28 12:16:44 -04:00
Brett Kuprel
e7001f063c simplified 2022-06-27 15:46:04 -04:00