use ignite for exp2
This commit is contained in:
@@ -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()
|
||||
|
||||
|
||||
Reference in New Issue
Block a user