Error with DLC iteration > 0 #18
Description
Hi all,
Ran into an error with the run_dgp_demo pipeline when I tried to use it on a project that had two DLC iterations (iteration-0 and iteration-1) --
File "[my home directory]/deepgraphpose/src/deepgraphpose/utils_model.py", line 93, in get_train_config
TrainingFraction = cfg['TrainingFraction'][iteration]
IndexError: list index out of range
My project's config.yaml has "iteration: 1", which I think is causing this problem. It only has one value to index for TrainingFraction.
So far, I've fixed it in a hacky way by changing line 93 of utils_model.py to index using 0 rather than iteration. Runs just fine for me after that. (Not sure if that fix would cause other problems for folks with more than one TrainingFraction value...)