Skip to content

Commit

Permalink
rm tr_detr coz it use .cuda() function
Browse files Browse the repository at this point in the history
  • Loading branch information
awkrail committed Sep 18, 2024
1 parent 8c699fd commit aa87601
Showing 1 changed file with 8 additions and 5 deletions.
13 changes: 8 additions & 5 deletions tests/test_models.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
CGDETRPredictor, TRDETRPredictor, UVCOMPredictor)

FEATURES = ['clip', 'clip_slowfast']
MODELS = ['moment_detr', 'qd_detr', 'eatr', 'cg_detr', 'uvcom', 'tr_detr']
MODELS = ['moment_detr', 'qd_detr', 'eatr', 'cg_detr', 'uvcom']
DATASETS = ['qvhighlight']
MIN_DURATION = 10
MAX_DURATION = 151
Expand Down Expand Up @@ -53,16 +53,19 @@ def test_load_slowfast_pann_weights():
def test_model_prediction():
"""
Test all of the trained models, except for resnet_glove features and taskweave
ResNet+GloVe is skipped due to their low performance.
CLIP+Slowfast+PANNs is skipped due to their low latency.
Taskweave is skiiped because two strategies are neccesary for prediction.
Untested features:
- ResNet+GloVe is skipped due to their low performance.
- CLIP+Slowfast+PANNs is skipped due to their low latency.
Untested models:
- TR-DETR is skipped because model use .cuda() function. We need to remove it.
- Taskweave is skiped because two strategies are neccesary for prediction.
"""
model_loaders = {
'moment_detr': MomentDETRPredictor,
'qd_detr': QDDETRPredictor,
'eatr': EaTRPredictor,
'cg_detr': CGDETRPredictor,
'tr_detr': TRDETRPredictor,
'uvcom': UVCOMPredictor,
}

Expand Down

0 comments on commit aa87601

Please sign in to comment.