added a simple routine to warm up the weights of the different model components

a44b6842 · jrzaurin · 04cb73a3 · a44b6842 · a44b6842
展开全部隐藏空白更改
内联并排

Showing with 279 addition and 251 deletion

pytorch_widedeep/models/_wdmodel_type.py pytorch_widedeep/models/_wdmodel_type.py +8 -0

pytorch_widedeep/models/wide_deep.py pytorch_widedeep/models/wide_deep.py +271 -251

未找到文件。
--- a/pytorch_widedeep/models/_wdmodel_type.py
+++ b/pytorch_widedeep/models/_wdmodel_type.py
+from typing import Union
+
+from .wide import Wide
+from .deep_dense import DeepDense
+from .deep_text import DeepText
+from .deep_image import DeepImage
+
+WDModel = Union[Wide, DeepDense, DeepText, DeepImage]
--- a/pytorch_widedeep/models/wide_deep.py
+++ b/pytorch_widedeep/models/wide_deep.py