提交 260a2423 编写于 作者: S Steffy-zxf

update strategy

上级 49ad8879
......@@ -638,8 +638,9 @@ class ULMFiTStrategy(CombinedStrategy):
ratio=32,
dis_blocks=3,
factor=2.6,
dis_params_layer=None,
frz_blocks=3,
params_layer=None,
frz_params_layer=None,
**kwargs):
scheduler = {
......@@ -649,12 +650,12 @@ class ULMFiTStrategy(CombinedStrategy):
},
"gradual_unfreeze": {
"blocks": frz_blocks,
"params_layer": params_layer
"params_layer": frz_params_layer
},
"discriminative": {
"blocks": dis_blocks,
"factor": factor,
"params_layer": params_layer
"params_layer": dis_params_layer
}
}
regularization = {}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册