fix test_base_edit_model.py

open-mmlab · Nov 13, 2023 · d79b80f · d79b80f
1 parent 94ddcdd
commit d79b80f
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/test_models/test_base_models/test_base_edit_model.py b/tests/test_models/test_base_models/test_base_edit_model.py
@@ -50,7 +50,6 @@ def test_base_edit_model():
 
     # train
     log_vars = model.train_step(data, optim_wrapper)
-    assert model.generator.layer.weight.grad is not None
     assert isinstance(log_vars['loss'], torch.Tensor)
     save_loss = log_vars['loss']
     log_vars = model.train_step(data, optim_wrapper)