introduce tcnn to motion_encoder

nnaakkaaii · Jun 25, 2024 · 1286a5b · 1286a5b
1 parent 6384940
commit 1286a5b
Show file tree

Hide file tree

Showing 2 changed files with 130 additions and 97 deletions.
diff --git a/hrdae/__init__.py b/hrdae/__init__.py
@@ -41,8 +41,6 @@
 from .models.networks.motion_encoder import (
     MotionConv2dEncoder1dOption,
     MotionConv3dEncoder2dOption,
-    MotionGuidedEncoder1dOption,
-    MotionGuidedEncoder2dOption,
     MotionNormalEncoder1dOption,
     MotionNormalEncoder2dOption,
     MotionRNNEncoder1dOption,
@@ -187,16 +185,6 @@
     name="conv3d",
     node=MotionConv3dEncoder2dOption,
 )
-cs.store(
-    group="config/experiment/model/network/motion_encoder",
-    name="guided1d",
-    node=MotionGuidedEncoder1dOption,
-)
-cs.store(
-    group="config/experiment/model/network/motion_encoder",
-    name="guided2d",
-    node=MotionGuidedEncoder2dOption,
-)
 cs.store(
     group="config/experiment/model/network/motion_encoder/rnn",
     name="conv_lstm1d",