stevinc / thecoloroutofspace Goto Github PK

I follow your instruction to train with Colorization/main.py and Multi_label_classification/main.py. After i had trained the Multi_label_classification.main for 3 bands rgb and 9 bands spectral, i run the main_ensemble.py with checkpoint load from two mentioned model but got the error bellow:

Traceback (most recent call last):
File "/home/thongnq/miniconda3/envs/thongnq/lib/python3.10/runpy.py", line 196, in _run_module_as_main
return _run_code(code, main_globals, None,
File "/home/thongnq/miniconda3/envs/thongnq/lib/python3.10/runpy.py", line 86, in _run_code
exec(code, run_globals)
File "/home/thongnq/color/TheColorOutOfSpace/Multi_label_classification/main_ensemble.py", line 114, in
main(args)
File "/home/thongnq/color/TheColorOutOfSpace/Multi_label_classification/main_ensemble.py", line 97, in main
test(model=model, test_loader=test_loader, loss_fn=loss_fn,
File "/home/thongnq/color/TheColorOutOfSpace/Multi_label_classification/test.py", line 29, in test
out, out_sigmoid = model(in_bands)
File "/home/thongnq/miniconda3/envs/thongnq/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1501, in _call_impl
return forward_call(*args, **kwargs)
File "/home/thongnq/color/TheColorOutOfSpace/Multi_label_classification/models/Ensemble.py", line 22, in forward
spectral = torch.index_select(input=x, dim=1, index=indices_spectral.to(self.device))
RuntimeError: INDICES element is out of DATA bounds, id=9 axis_dim=9

Can you help me specify the error?