From 0b51a847ca29a80c3d8d22ef636f15ce5aab4b7d Mon Sep 17 00:00:00 2001 From: meixiaowei Date: Wed, 29 Apr 2020 22:27:27 +0800 Subject: [PATCH] modify weight init --- example/resnet101_imagenet2012/train.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/example/resnet101_imagenet2012/train.py b/example/resnet101_imagenet2012/train.py index 3d0a23f93a..365a859395 100755 --- a/example/resnet101_imagenet2012/train.py +++ b/example/resnet101_imagenet2012/train.py @@ -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(),