utilities

Shreyasi2002 · Dec 20, 2023 · 123e087 · 123e087
1 parent dfb2e1b
commit 123e087
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -6,4 +6,11 @@ def load_checkpoint(checkpoint, model, optimizer):
     return checkpoint['train_losses'], checkpoint['val_losses']
 
 def save_checkpoint(checkpoint, model_path):
-    torch.save(checkpoint, model_path)
+    torch.save(checkpoint, model_path)
+
+def to_numpy_array(tensor):
+    '''Convert torch.Tensor to np.ndarray'''
+    tensor_ = tensor.cpu()
+    tensor_ = tensor_.squeeze(0)
+    tensor_ = tensor_.detach().numpy()
+    return tensor_