Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/transformers.git" did not exist on "b5ba758ba983af2b0486dda02ae68109846afdb7"
Commit a8ec52ef authored by Sylvain Gugger's avatar Sylvain Gugger
Browse files

Style

parent b35e7b68
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_trainer.py
View file @ a8ec52ef
...@@ -992,7 +992,7 @@ class TrainerIntegrationTest(unittest.TestCase): ...@@ -992,7 +992,7 @@ class TrainerIntegrationTest(unittest.TestCase):
# should be about half of fp16_init # should be about half of fp16_init
# perfect world: fp32_init/2 == fp16_eval # perfect world: fp32_init/2 == fp16_eval
self.assertAlmostEqual(fp16_eval, fp32_init / 2, delta=5_000) self.assertAlmostEqual(fp16_eval, fp32_init / 2, delta=5_000)
def test_no_wd_param_group(self): def test_no_wd_param_group(self):
model = torch.nn.Sequential(TstLayer(128), torch.nn.ModuleList([TstLayer(128), TstLayer(128)])) model = torch.nn.Sequential(TstLayer(128), torch.nn.ModuleList([TstLayer(128), TstLayer(128)]))
trainer = Trainer(model=model) trainer = Trainer(model=model)
......
tests/test_trainer_utils.py
View file @ a8ec52ef
...@@ -30,7 +30,7 @@ if is_torch_available(): ...@@ -30,7 +30,7 @@ if is_torch_available():
DistributedTensorGatherer, DistributedTensorGatherer,
LabelSmoother, LabelSmoother,
LengthGroupedSampler, LengthGroupedSampler,
get_parameter_names get_parameter_names,
) )
class TstLayer(torch.nn.Module): class TstLayer(torch.nn.Module):
......
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