iicd/xautodl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix black errors

Browse Source
This commit is contained in:
D-X-Y
2021-03-21 05:59:56 -07:00
parent b8c173eb76
commit 1acd1e9f9b
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
lib/trade_models/transformers.py
View File

@@ -17,7 +17,7 @@ from xlayers import trunc_normal_
from xlayers import super_core
__all__ = ["DefaultSearchSpace"]
__all__ = ["DefaultSearchSpace", "DEFAULT_NET_CONFIG", "get_transformer"]
def _get_mul_specs(candidates, num):
@@ -41,6 +41,7 @@ def _assert_types(x, expected_types):
)
DEFAULT_NET_CONFIG = None
_default_max_depth = 5
DefaultSearchSpace = dict(
d_feat=6,
@@ -163,7 +164,9 @@ class SuperTransformer(super_core.SuperModule):
else:
stem_dim = spaces.get_determined_value(self._stem_dim)
cls_tokens = self.cls_token.expand(batch, -1, -1)
cls_tokens = F.interpolate(cls_tokens, size=(stem_dim), mode="linear", align_corners=True)
cls_tokens = F.interpolate(
cls_tokens, size=(stem_dim), mode="linear", align_corners=True
)
feats_w_ct = torch.cat((cls_tokens, feats), dim=1)
feats_w_tp = self.pos_embed(feats_w_ct)
xfeats = self.backbone(feats_w_tp)