From 1012cdec3111821dfff4becb57f7771af69bf166 Mon Sep 17 00:00:00 2001 From: abdessaied Date: Thu, 17 Oct 2024 14:11:35 +0200 Subject: [PATCH] Update models/utils.py --- models/utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/models/utils.py b/models/utils.py index 8a5ff26..cc1dbe8 100644 --- a/models/utils.py +++ b/models/utils.py @@ -46,7 +46,7 @@ def seperate_nextqa_input_modalities( Returns: _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_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :] @@ -90,7 +90,7 @@ def seperate_input_modalities( Returns: _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_flow_hidden = features_copy[:, i3d_flow_interval[0]+1:i3d_flow_interval[1], :] sam_hidden = features_copy[:, sam_interval[0]+1:sam_interval[1], :]