Skip to content

GitLab

Unverified Commit a5883aa7 authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[test] fixed codefactor format report (#4026)

parent 822c3d4d
Show whitespace changes
Inline Side-by-side
Showing with 14 additions and 4 deletions
tests/test_checkpoint_io/test_general_checkpoint_io.py
View file @ a5883aa7
...@@ -165,8 +165,13 @@ def test_sharded_optimizer_multiple_param_groups(): ...@@ -165,8 +165,13 @@ def test_sharded_optimizer_multiple_param_groups():
# create a model and optimizer # create a model and optimizer
model = resnet18() model = resnet18()
optimizer = Adam([{'params': model.layer1.parameters()}, \ optimizer = Adam([{
{'params': model.layer2.parameters(), 'lr': 0.002}], lr=0.001) 'params': model.layer1.parameters()
}, {
'params': model.layer2.parameters(),
'lr': 0.002
}],
lr=0.001)
# create test data sample # create test data sample
x = torch.randn(1, 3, 224, 224) x = torch.randn(1, 3, 224, 224)
...@@ -189,8 +194,13 @@ def test_sharded_optimizer_multiple_param_groups(): ...@@ -189,8 +194,13 @@ def test_sharded_optimizer_multiple_param_groups():
# create new model # create new model
new_model = resnet18() new_model = resnet18()
new_optimizer = Adam([{'params': new_model.layer1.parameters()}, \ new_optimizer = Adam([{
{'params': new_model.layer2.parameters(), 'lr': 0.002}], lr=0.001) 'params': new_model.layer1.parameters()
}, {
'params': new_model.layer2.parameters(),
'lr': 0.002
}],
lr=0.001)
ckpt_io.load_model(new_model, str(model_ckpt_dir.name), strict=True) ckpt_io.load_model(new_model, str(model_ckpt_dir.name), strict=True)
ckpt_io.load_optimizer(new_optimizer, str(optimizer_ckpt_dir.name)) ckpt_io.load_optimizer(new_optimizer, str(optimizer_ckpt_dir.name))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment