File "/home/kevin/PE3R/modules/pe3r/demo.py", line 433, in get_reconstructed_scene
cog_seg_maps, rev_cog_seg_maps, cog_feats = get_cog_feats(images, pe3r)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/kevin/anaconda3/envs/pe3r/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 124, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/home/kevin/PE3R/modules/pe3r/demo.py", line 381, in get_cog_feats
image_features = image_features / image_features.norm(dim=-1, keepdim=True)
^^^^^^^^^^^^^^^^^^^
AttributeError: 'BaseModelOutputWithPooling' object has no attribute 'norm'
