提交 0b51a847 编写于 作者: M meixiaowei

modify weight init

上级 73bd2e9a
......@@ -14,7 +14,6 @@
# ============================================================================
"""train_imagenet."""
import os
import math
import argparse
import random
import numpy as np
......@@ -64,7 +63,6 @@ if __name__ == '__main__':
epoch_size = config.epoch_size
net = resnet101(class_num=config.class_num)
# weight init
default_recurisive_init(net)
for _, cell in net.cells_and_names():
if isinstance(cell, nn.Conv2d):
cell.weight.default_input = weight_init.initializer(weight_init.XavierUniform(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册