_uniform_, constant_ etc.

fa900595 · zzhang-cn · ca0622b2 · fa900595
Commit fa900595 authored May 11, 2018 by zzhang-cn
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

model.py model.py +2 -2

No files found.
--- a/model.py
+++ b/model.py
@@ -29,8 +29,8 @@ def build_cnn(**config):
            kernel_size,
            padding=tuple((_ - 1) // 2 for _ in kernel_size),
            )
-        INIT.xavier_uniform(module.weight)
-        INIT.constant(module.bias, 0)
+        INIT.xavier_uniform_(module.weight)
+        INIT.constant_(module.bias, 0)
        cnn_list.append(module)
        if i < len(filters) - 1:
            cnn_list.append(NN.LeakyReLU())