search

antoine77340 / S3D_HowTo100M

S3D Text-Video model trained on HowTo100M using MIL-NCE
Apache License 2.0
191 stars 21 forks source link

space_to_depth value #6

Closed thematrixduo closed 3 years ago

thematrixduo commented 3 years ago

In this repository 'space_to_depth' is set to True in S3D model. However in the training repository (https://github.com/antoine77340/MIL-NCE_HowTo100M/blob/) 'space_to_depth' is set to False. Could you confirm if this is intentionally set differently? Thanks!

antoine77340 commented 3 years ago

Yes it is intentional. space_to_depth was set to True here so we could load the weights we have trained using TPUs. This trick is useful to speedup the training process on TPU.

However, if you train on GPU, you will not need this space_to_depth trick.

thematrixduo commented 3 years ago

Got it, Thanks for the prompt reply!