Argument name mismatch due to epoch_based_runner.py

[skbhat]

Namaste,

First of all thanks for presenting a wonderful active learning method.

What I observed in the code is that in the file single_stage.py, the function forward_train uses argument names x, y_loc_img, y_cls_img. This is different from what gets loaded by data loader in epoch_based_runner.py of the mmcv library. You have a local version of epoch_based_runner.py where you update these arguments. But by default it takes the code in mmcv library and we get error:

forward() missing 1 required positional argument: 'x'

Currently I have a work around of changing names of the keys in the dictionary. But a clear version is expected.

regard
skbhat

[skbhat]

This is a duplicate of #5