temporal-shift-module
temporal-shift-module copied to clipboard
test failed when using RGBDiff modality
input_view = input.view((-1, self.num_segments, self.new_length + 1, input_c,) + input.size()[2:]) RuntimeError: shape '[-1, 16, 6, 3, 18, 224, 224]' is invalid for input of size 231211008
my segments num is 16, and training stage has no error.
Hi, sorry for the late reply. I did not specially take care of the RGBDiff modality since it is not used. Have you figured out the problem yet? I think maybe there are some missing steps when setting channel numbers.