mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-09-11 13:59:08 +00:00
[devops] remove post commit ci (#5566)
* [devops] remove post commit ci * [misc] run pre-commit on all files * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
import torch
|
||||
import pytest
|
||||
import torch
|
||||
|
||||
from colossalai.nn.optimizer import CPUAdam, HybridAdam
|
||||
from colossalai.testing import clear_cache_before_run, parameterize
|
||||
@@ -17,6 +17,7 @@ def check_params_equal(model, torch_model):
|
||||
for p, torch_p in zip(model.parameters(), torch_model.parameters()):
|
||||
assert torch.allclose(p, torch_p, atol=1e-3), f"diff: {torch.abs(p - torch_p)}"
|
||||
|
||||
|
||||
# TODO Something wrong with ci when running this test.
|
||||
@pytest.mark.skip(reason="skip because of something wrong with CI")
|
||||
@clear_cache_before_run()
|
||||
|
Reference in New Issue
Block a user