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/quant_transformer.py
View File

@@ -27,8 +27,10 @@ import torch.utils.data as th_data
from log_utils import AverageMeter
from utils import count_parameters
from trade_models.transformers import DEFAULT_NET_CONFIG
from trade_models.transformers import get_transformer
from xlayers import super_core
from .transformers import DEFAULT_NET_CONFIG
from .transformers import get_transformer
from qlib.model.base import Model
@@ -90,6 +92,7 @@ class QuantTransformer(Model):
torch.cuda.manual_seed_all(self.seed)
self.model = get_transformer(self.net_config)
self.model.set_super_run_type(super_core.SuperRunMode.FullModel)
self.logger.info("model: {:}".format(self.model))
self.logger.info("model size: {:.3f} MB".format(count_parameters(self.model)))