p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

modify weight init

This commit is contained in:
meixiaowei 2020-04-29 22:27:27 +08:00
parent 73bd2e9afb
commit 0b51a847ca
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
example/resnet101_imagenet2012/train.py
View File

@ -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(),