beeldengeluid / dane-visual-feature-extraction-worker

Uses the VizXP feature extraction to produce features/vector spaces
MIT License
0 stars 0 forks source link

18 assert implementation is correct #19

Closed Veldhoen closed 1 year ago

Veldhoen commented 1 year ago

Fixes #18: Wrote unit tests, asserted implementation was incorrect, and fixed it by small modifications: model.eval() in feature_extraction module (switches model state from train to eval) store intermediate results in a list rather than in a concatenated tensor (maintain order & presumably more efficient)