mirror of https://github.com/InternLM/InternLM
name (#514)
parent
b59641715a
commit
06e8301861
|
@ -411,4 +411,4 @@ def test_loss():
|
|||
|
||||
|
||||
if __name__ == "__main__":
|
||||
pytest.main(["-s", "-q", "test_diff_num_bsz_loss.py"])
|
||||
pytest.main(["-s", "-q", "test_swap_nb_loss_and_gradnorm.py"])
|
||||
|
|
Loading…
Reference in New Issue