File tree Expand file tree Collapse file tree 1 file changed +0
-15
lines changed Expand file tree Collapse file tree 1 file changed +0
-15
lines changed Original file line number Diff line number Diff line change @@ -621,21 +621,6 @@ def test_eval_raises_bad_labels(train_df):
621
621
model .eval (train_df .values , scores .values )
622
622
623
623
624
- def test_split_resnet_feat_clf (train_df ):
625
- # TODO: will add attributes to cnn architectures
626
- # specifying which layers are feature and which are classifier,
627
- # to make this API more generic
628
- model = cnn .CNN (architecture = "resnet18" , classes = [0 , 1 ], sample_duration = 2 )
629
- cnn .separate_resnet_feat_clf (model )
630
- assert "feature" in model .optimizer_params ["kwargs" ]
631
- model .optimizer_params ["kwargs" ]["feature" ]["lr" ] = 0.1
632
- model .train (train_df , epochs = 0 , save_path = "tests/models" )
633
- # shutil.rmtree("tests/models/")
634
-
635
-
636
- # test load_outdated_model?
637
-
638
-
639
624
def test_train_no_validation (train_df ):
640
625
model = cnn .CNN (architecture = "resnet18" , classes = [0 , 1 ], sample_duration = 2 )
641
626
model .train (train_df , save_path = "tests/models" )
You can’t perform that action at this time.
0 commit comments