liuyu/qlib
1
0
Fork 0
mirror of https://github.com/microsoft/qlib.git synced 2026-07-15 16:56:54 +08:00
Code Activity

black format

Browse Source
This commit is contained in:
bxdd
2020-11-27 13:09:40 +08:00
parent 8b275a6006
commit ae757a4b51
4 changed files with 8 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
qlib/contrib/model/pytorch_nn.py
View File

@@ -22,6 +22,7 @@ from ...utils import unpack_archive_with_buffer, save_multiple_parts_file, creat
from ...log import get_module_logger, TimeInspector
from ...workflow import R
class DNNModelPytorch(Model):
"""DNN Model
@@ -349,7 +350,7 @@ class Net(nn.Module):
def _weight_init(self):
for m in self.modules():
if isinstance(m, nn.Linear):
nn.init.kaiming_normal_(m.weight, a=0.1, mode='fan_in', nonlinearity='leaky_relu')
nn.init.kaiming_normal_(m.weight, a=0.1, mode="fan_in", nonlinearity="leaky_relu")
def forward(self, x):
cur_output = x