unfreeze

catalyst-team · May 27, 2020 · 9f945d7 · 9f945d7
1 parent bec2593
commit 9f945d7
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 11 deletions.
diff --git a/catalyst/contrib/dl/callbacks/trace.py b/catalyst/contrib/dl/callbacks/trace.py
@@ -108,7 +108,7 @@ def _trace(self, state: _Runner):
             checkpoint_name_to_restore = "best"
 
         traced_model = trace_model_from_runner(
-            state=state,
+            runner=state,
             checkpoint_name=checkpoint_name_to_restore,
             method_name=self.method_name,
             mode=self.trace_mode,

diff --git a/catalyst/core/runner.py b/catalyst/core/runner.py
@@ -60,7 +60,8 @@ def __init__(
         self._model = None
         self._prepare_inner_state(device=device, model=model)
         self._init()
-        self._freeze()
+        # @TODO: how to fix?
+        # self._freeze()
 
     def _prepare_inner_state(
         self,

diff --git a/docs/api/core.rst b/docs/api/core.rst
@@ -51,13 +51,6 @@ Callback
     :undoc-members:
     :show-inheritance:
 
-State
-~~~~~~~~~~~~~~~~~~~~~~
-.. automodule:: catalyst.core.state
-    :members:
-    :undoc-members:
-    :show-inheritance:
-
 Callbacks
 ----------------------
 

diff --git a/docs/api/dl.rst b/docs/api/dl.rst
@@ -16,7 +16,7 @@ Experiment
 
 Experiment
 ~~~~~~~~~~~~~~~~~~~~~~
-.. automodule:: catalyst.dl.experiment.core
+.. automodule:: catalyst.dl.experiment.experiment
     :members:
     :undoc-members:
     :show-inheritance:
@@ -45,7 +45,7 @@ Runner
 
 Runner
 ~~~~~~~~~~~~~~~~~~~~~~
-.. automodule:: catalyst.dl.runner.core
+.. automodule:: catalyst.dl.runner.runner
     :members:
     :undoc-members:
     :show-inheritance: