Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Embd_dim #3

Open
Petr-Byv opened this issue Oct 1, 2019 · 1 comment
Open

Embd_dim #3

Petr-Byv opened this issue Oct 1, 2019 · 1 comment

Comments

@Petr-Byv
Copy link

Petr-Byv commented Oct 1, 2019

Hi, thank you for sharing the model.
For the evaluation the line suggest to use 6144 for the embedding:
python eval.py --eval_msrvtt=1 --eval_youcook=1 --eval_lsmdc=1 --num_thread_reader=8 --embd_dim=6144 --pretrain_path=model/howto100m_pt_model.pth.

Is that indeed so, and how is it different to the 4096 recommended for the training:
python train.py --num_thread_reader=8 --epochs=15 --batch_size=32 --n_pair=64 --n_display=100 --embd_dim=4096 --checkpoint_dir=ckpt --features_path_2D=folder_for_2d_features --features_path_3D=folder_for_3d_features --train_csv=train.csv
?

Thank you.

@zhenliulv
Copy link

Hi, i have the same question, can you give us some idea ?
Thank you.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants