File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Expand file tree Collapse file tree 1 file changed +6
-0
lines changed Original file line number Diff line number Diff line change @@ -39,11 +39,17 @@ def test_load_lr_scheduler_invalid(invalid_lr_scheduler_names):
3939
4040def test_get_supported_optimizers ():
4141 assert len (get_supported_optimizers ()) == 76
42+ assert len (get_supported_optimizers ('adam*' )) == 7
43+ assert len (get_supported_optimizers (['adam*' , 'ranger*' ])) == 9
4244
4345
4446def test_get_supported_lr_schedulers ():
4547 assert len (get_supported_lr_schedulers ()) == 16
48+ assert len (get_supported_lr_schedulers ('cosine*' )) == 4
49+ assert len (get_supported_lr_schedulers (['cosine*' , '*warm*' ])) == 5
4650
4751
4852def test_get_supported_loss_functions ():
4953 assert len (get_supported_loss_functions ()) == 13
54+ assert len (get_supported_loss_functions ('*focal*' )) == 4
55+ assert len (get_supported_loss_functions (['*focal*' , 'bce*' ])) == 5
You can’t perform that action at this time.
0 commit comments