请问用了预训练的resnet101,后边部分的网络需要初始化嘛?怎么初始化啊?
Opened this issue · 2 comments
zhangtao22 commented
没看到你代码初始化后边其他系数啊
jkznst commented
cls_subnet和bbox_subnet中的weight和bias给出了初始化方式,FPN中的其他参数在mx.mod.Module.fit中采用Xavier初始化。
zhangtao22 commented
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初始化吧?