When I followed the first step of training with a specified checkpoint, it reported the error" TypeError: 'bool' object is not subscriptable" .
Traceback (most recent call last):
File "/root/autodl-tmp/PIC/src/train.py", line 404, in
main(sys.argv[1:])
File "/root/autodl-tmp/PIC/src/train.py", line 247, in main
counter ,loss_train, bpp_loss_train, mse_lss_train, bpp_scalable_train = train_one_epoch( net,
File "/root/autodl-tmp/PIC/src/training/step.py", line 86, in train_one_epoch
out_net = model(d, quality = list_quality)
File "/root/miniconda3/envs/pic/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/PIC/src/models/rem_pic.py", line 365, in forward
if self.enable_rem[right_index] and y_checkpoint_hat is not None:
TypeError: 'bool' object is not subscriptable
When I followed the first step of training with a specified checkpoint, it reported the error" TypeError: 'bool' object is not subscriptable" .
Traceback (most recent call last):
File "/root/autodl-tmp/PIC/src/train.py", line 404, in
main(sys.argv[1:])
File "/root/autodl-tmp/PIC/src/train.py", line 247, in main
counter ,loss_train, bpp_loss_train, mse_lss_train, bpp_scalable_train = train_one_epoch( net,
File "/root/autodl-tmp/PIC/src/training/step.py", line 86, in train_one_epoch
out_net = model(d, quality = list_quality)
File "/root/miniconda3/envs/pic/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl
return forward_call(*input, **kwargs)
File "/root/autodl-tmp/PIC/src/models/rem_pic.py", line 365, in forward
if self.enable_rem[right_index] and y_checkpoint_hat is not None:
TypeError: 'bool' object is not subscriptable