JSSL/SSL vsharp engine

NKI-AI · May 17, 2024 · 1ff2de9 · 1ff2de9
1 parent 66dd5cd
commit 1ff2de9
Show file tree

Hide file tree

Showing 3 changed files with 635 additions and 4 deletions.
diff --git a/direct/nn/ssl/mri_models.py b/direct/nn/ssl/mri_models.py
@@ -19,7 +19,7 @@
 from direct.utils import detach_dict, dict_to_device, normalize_image
 from direct.utils.events import get_event_storage
 
-__all__ = ["SSLMRIModelEngine"]
+__all__ = ["SSLMRIModelEngine", "JSSLMRIModelEngine"]
 
 
 class SSLMRIModelEngine(MRIModelEngine):