Error with DLC iteration > 0

[marierosegreaney]

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...)