Skip to content

Commit 16ea52e

Browse files
committed
remove outdated test split_resnet_feat_clf
1 parent 56de667 commit 16ea52e

File tree

1 file changed

+0
-15
lines changed

1 file changed

+0
-15
lines changed

tests/test_cnn.py

Lines changed: 0 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -621,21 +621,6 @@ def test_eval_raises_bad_labels(train_df):
621621
model.eval(train_df.values, scores.values)
622622

623623

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-
639624
def test_train_no_validation(train_df):
640625
model = cnn.CNN(architecture="resnet18", classes=[0, 1], sample_duration=2)
641626
model.train(train_df, save_path="tests/models")

0 commit comments

Comments
 (0)