Skip to content

GitLab

Commit fa900595 authored by zzhang-cn's avatar zzhang-cn
Browse files

_uniform_, constant_ etc.

parent ca0622b2
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
model.py
View file @ fa900595
...@@ -29,8 +29,8 @@ def build_cnn(**config): ...@@ -29,8 +29,8 @@ def build_cnn(**config):
kernel_size, kernel_size,
padding=tuple((_ - 1) // 2 for _ in kernel_size), padding=tuple((_ - 1) // 2 for _ in kernel_size),
) )
INIT.xavier_uniform(module.weight) INIT.xavier_uniform_(module.weight)
INIT.constant(module.bias, 0) INIT.constant_(module.bias, 0)
cnn_list.append(module) cnn_list.append(module)
if i < len(filters) - 1: if i < len(filters) - 1:
cnn_list.append(NN.LeakyReLU()) cnn_list.append(NN.LeakyReLU())
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment