feat: import torch and numpy

Bjarten · Dec 30, 2018 · b345d0b · b345d0b
1 parent bae0775
commit b345d0b
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorchtools.py b/pytorchtools.py
@@ -1,3 +1,6 @@
+import numpy as np
+import torch
+
 class EarlyStopping:
     """Early stops the training if validation loss dosen't improve after a given patience."""
     def __init__(self,patience=3):