Update utils.py

utils.py

@@ -8,6 +8,7 @@ from io import BytesIO
 from PIL import Image
 import requests
 import re
+from copy import deepcopy
 
 class ImageType(Enum):
     REAL_UP_L = 0
@@ -74,3 +75,13 @@ def image_to_base64(image_array):
     image_array.save(buffered, format="PNG")
     image_b64 = base64.b64encode(buffered.getvalue()).decode("utf-8")
     return f"data:image/png;base64, {image_b64}"
+
+
+def copy_G_params(model):
+    flatten = deepcopy(list(p.data for p in model.parameters()))
+    return flatten
+
+
+def load_params(model, new_param):
+    for p, new_p in zip(model.parameters(), new_param):
+        p.data.copy_(new_p)