1
0

use ignite for exp2

This commit is contained in:
2025-12-02 23:07:27 +08:00
parent 43b807679f
commit 65c56e938c
15 changed files with 246 additions and 794 deletions

View File

@@ -5,7 +5,7 @@ import tkinter as tk
from predict import PredictResult, Predictor
sys.path.append(str(Path(__file__).resolve().parent.parent.parent))
import pytorch_gpu_utils
import gpu_utils
class SketchpadApp:
@@ -181,7 +181,7 @@ class SketchpadApp:
# endregion
if __name__ == "__main__":
pytorch_gpu_utils.print_gpu_availability()
gpu_utils.print_gpu_availability()
predictor = Predictor()