Upload to Main

2025-10-07 22:42:55 +08:00
commit d3ddab7c5d
218 changed files with 125815 additions and 0 deletions
--- a/deepcore/nets/nets_utils/parallel.py
+++ b/deepcore/nets/nets_utils/parallel.py
@@ -0,0 +1,16 @@
+from torch.nn import DataParallel
+
+
+class MyDataParallel(DataParallel):
+    def __getattr__(self, name):
+        try:
+            return super().__getattr__(name)
+        except AttributeError:
+            return getattr(self.module, name)
+    def __setattr__(self, name, value):
+        try:
+            if name == "no_grad":
+                return setattr(self.module, name, value)
+            return super().__setattr__(name, value)
+        except AttributeError:
+            return setattr(self.module, name, value)