@@ -115,7 +115,7 @@ class VideoClipSampler(nn.Module):
115
115
Args:
116
116
video_args (`VideoArgs`): The video args
117
117
sampler_args (`SamplerArgs`): The sampler args. Can be TimeBasedSamplerArgs or IndexBasedSamplerArgs
118
- decorder_args (`DecoderArgs`): Decoder args contain value needs for decoder, for example, thread count
118
+ decoder_args (`DecoderArgs`): Decoder args contain value needs for decoder, for example, thread count
119
119
120
120
Example:
121
121
>>> video_args = VideoArgs(desired_width=224, desired_height=224)
@@ -130,12 +130,12 @@ def __init__(
130
130
self ,
131
131
video_args : VideoArgs ,
132
132
sampler_args : SamplerArgs ,
133
- decorder_args : Union [None , DecoderArgs ] = None ,
133
+ decoder_args : Union [None , DecoderArgs ] = None ,
134
134
) -> None :
135
135
super ().__init__ ()
136
136
self .video_args = video_args
137
137
self .sampler_args = sampler_args
138
- self .decorder_args = DecoderArgs () if decorder_args is None else decorder_args
138
+ self .decoder_args = DecoderArgs () if decoder_args is None else decoder_args
139
139
140
140
def forward (self , video_data : Tensor ) -> Union [List [Any ]]:
141
141
"""Sample video clips from the video data
@@ -162,7 +162,7 @@ def forward(self, video_data: Tensor) -> Union[List[Any]]:
162
162
video_decoder ,
163
163
width = target_width ,
164
164
height = target_height ,
165
- num_threads = self .decorder_args .num_threads ,
165
+ num_threads = self .decoder_args .num_threads ,
166
166
)
167
167
168
168
clips : List [Any ] = []
0 commit comments