Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

请问用了预训练的resnet101,后边部分的网络需要初始化嘛?怎么初始化啊? #1

Open
zhangtao22 opened this issue Jul 10, 2018 · 2 comments

Comments

@zhangtao22
Copy link

没看到你代码初始化后边其他系数啊

@jkznst
Copy link
Owner

jkznst commented Jul 10, 2018

cls_subnet和bbox_subnet中的weight和bias给出了初始化方式,FPN中的其他参数在mx.mod.Module.fit中采用Xavier初始化。

@zhangtao22
Copy link
Author

mod.fit(train_iter,
val_iter,
eval_metric=MultiBoxMetric(),
validation_metric=valid_metric,
batch_end_callback=batch_end_callback,
eval_end_callback=eval_end_callback,
epoch_end_callback=epoch_end_callback,
optimizer=opt,
optimizer_params=opt_params,
begin_epoch=begin_epoch,
num_epoch=end_epoch,
initializer=mx.init.Xavier(),
arg_params=args,
aux_params=auxs,
allow_missing=True,
monitor=monitor)
这样写是把所有的参数都xavier初始化吧?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants