public-projects/MST-MIXER
2
0
Fork 0
Code Issues Pull requests Releases Wiki Activity Actions

Update models/utils.py

Browse source
This commit is contained in:
Adnen Abdessaied 2024-10-17 14:11:35 +02:00
parent 611640acae
commit 1012cdec31
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
models/utils.py
View file

@ -46,7 +46,7 @@ def seperate_nextqa_input_modalities(
Returns: Returns:
_type_: _description_ _type_: _description_
""" """
features_copy = features.clone() # .detach() features_copy = features.clone()
i3d_rgb_hidden = features_copy[:, i3d_rgb_interval[0]+1:i3d_rgb_interval[1], :] i3d_rgb_hidden = features_copy[:, i3d_rgb_interval[0]+1:i3d_rgb_interval[1], :]
i3d_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :] i3d_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :]
@ -90,7 +90,7 @@ def seperate_input_modalities(
Returns: Returns:
_type_: _description_ _type_: _description_
""" """
features_copy = features.clone() # .detach() features_copy = features.clone()
i3d_rgb_hidden = features_copy[:, i3d_rgb_interval[0]+1:i3d_rgb_interval[1], :] i3d_rgb_hidden = features_copy[:, i3d_rgb_interval[0]+1:i3d_rgb_interval[1], :]
i3d_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :] i3d_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :]
sam_hidden = features_copy[:, sam_interval[0]+1:sam_interval[1], :] sam_hidden = features_copy[:, sam_interval[0]+1:sam_interval[1], :]