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

update strategy

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