Error when trying to train STEAL model

[ToadTWP666]

I was able to train the baseline model using the following command: python train.py --dataset_type ped2

When I execute the following command:
python train.py --dataset_type ped2 --pseudo_anomaly_jump 0.01 --jump 2 3 4 5

I get the following error:

rain.py --dataset_type ped2 --pseudo_anomaly_jump 0.01 --jump 2 3 4 5
exp_dir: logweight_recon_pajump0.01_jump[2,3,4,5]
Traceback (most recent call last):
File "train.py", line 115, in
net_in[b] = imgsjump[0][b]
IndexError: index 1 is out of bounds for dimension 0 with size 1

Please advise

[aseuteurideu]

Where do you get your dataset? What is the size of imgsjump and net_in when the error happen?

[zdl12345678]

I also encountered the same problem, did you solve it?