KeyError: 'multi_scale_loss' on training the backbone

[hachreak]

Hi @wgcban ,
thank you very much to publish the code for the paper!

I was trying to train the backbone (first stage):

python train.py --config configs/config_HSIT_PRE.json

But, the training script is raising an error:

Traceback (most recent call last):
  File "train.py", line 346, in <module>
    train(epoch)
  File "train.py", line 198, in train
    if config[config["train_dataset"]]["multi_scale_loss"]:
KeyError: 'multi_scale_loss'

Is the configuration value for multi_scale_loss missing?
Which value do I need to set?
Thanks for your help! :D