Test super core

2021-06-03 01:24:09 -07:00
parent 6ee062a33d
commit 84462da79e
3 changed files with 32 additions and 3 deletions
--- a/xautodl/xlayers/super_core.py
+++ b/xautodl/xlayers/super_core.py
@@ -47,3 +47,5 @@ super_name2activation = {
 from .super_trade_stem import SuperAlphaEBDv1
 from .super_positional_embedding import SuperDynamicPositionE
 from .super_positional_embedding import SuperPositionalEncoder
+
+from .super_rearrange import SuperReArrange