Praveen Tirupattur

Use the following code to load the model. Works with PyTorch version 1.2..0 ``` def build_densenet_model(trained_model_file, num_classes): model = densenet.densenet201(num_classes=num_classes, sample_size=params.input_size, sample_duration=params.frames_per_clip) pretrained_model = torch.load(trained_model_file) pretrained_kvpair = pretrained_model['state_dict'] model_kvpair =...

I have the same exact issue. I tried Tensorflow version 1.5 and 1.6 and it is same in both the cases. Really appreciate your help in resolving this.

@mangoooooooo and @Wyz2927 Can you share the training log? Also, what is the PyTorch version you are using?

Hello @piergiaj I used your code to extract I3D features (RGB only) from Multi-Thumos dataset and trained the TGM model with 3 layer, L = 5. The result I get...