hantman-lab / animal-soup

Hantman Lab automated behavioral classification system.
GNU General Public License v3.0
4 stars 0 forks source link

keeping track of flow_window #32

Closed clewis7 closed 1 year ago

clewis7 commented 1 year ago

when you go to train or infer with a feature extractor, it is necessary to know the flow window for the optic flow of the flow generator going to be used

a checkpoint needs to store this information in the state_dict so that when a model checkpoint is loaded for training the feature extractor that the correct flow window, wouldn't matter for inference because a user would simply specify the flow window to use in inferring optic flow