优化训练过程

2025-03-20 16:41:51 +08:00
parent 358bb40a2a
commit bba39adcfc
7 changed files with 16 additions and 12 deletions
--- a/toch_test.py
+++ b/toch_test.py
@ -0,0 +1,5 @@
+import torch
+
+is_gpu = torch.cuda.is_available()
+
+print(str(is_gpu))