Hello,
Thank you for releasing the training code. Could you please provide the instructions for training?
I tried to use the instructions for training STDF - python -m torch.distributed.launch train_rf_ft.py --opt_path config/Final_QP37.yml
and ran into an error with the STDF module's in_conv layer:
Expected 4-dimensional input for 4-dimensional weight [32, 7, 3, 3], but got 5-dimensional input of size [8, 15, 1, 96, 96] instead
Could you please tell me what radius should be specified in the config file for training? The currently released config file specifies a radius of 7 but from the error log, it seems the training code expects R=3.
Additionally, your paper mentions two-stage training where you train the STDF with radius 3 and then the full network with 15 frame clips. Does your training code support doing that?
Thank you!