Update Q Model

2021-03-04 05:42:52 +00:00
parent ead6ae0842
commit e329b78cf4
4 changed files with 44 additions and 22 deletions
--- a/lib/layers/init.py
+++ b/lib/layers/init.py
@@ -1,4 +1,5 @@
 from .drop import DropBlock2d, DropPath
+from .mlp import MLP
 from .weight_init import trunc_normal_

 from .positional_embedding import PositionalEncoder