Add JSSL experiments (#278)

* JSSL/SSL vsharp engine * Documentation updates
NKI-AI · May 22, 2024 · 575b33a · 575b33a
1 parent 66dd5cd
commit 575b33a
Show file tree

Hide file tree

Showing 40 changed files with 21,690 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):