kkahatapitiya / X3D-Multigrid

PyTorch implementation of X3D models with Multigrid training.
MIT License
92 stars 13 forks source link

Pretrained models #3

Closed iliasprc closed 3 years ago

iliasprc commented 3 years ago

Hello, which configurations of X3d have you trained and included in the repo ? (X3D-M , X3D- L , X3D-XL ....)

tonyduan commented 3 years ago

Looks like both x3d_charades_rgb_sgd_024000.pt and x3d_multigrid_kinetics_rgb_sgd_204000.pt under the models/ directory are trained with the X3D-L configuration.

kkahatapitiya commented 3 years ago

In fact, the pretrained models included are for X3D-M configuration