the output of I3D model and the calculation of loss function

tomrunia / PyTorchConv3D

I3D and 3D-ResNets in PyTorch

Apache License 2.0

189 stars 27 forks source link

the output of I3D model and the calculation of loss function #13

Open punknownq opened 5 years ago

punknownq commented 5 years ago

Hi, Thank you for your work, firstly. Do we need to add x = x.mean(2) in forward function of the I3D model? And why do you add if config.model == 'i3d': targets = torch.unsqueeze(targets, -1) in train_epoch while it has not been added in validation_epoch?