search

bdqnghi / infercode

[ICSE 2021] - InferCode: Self-Supervised Learning of Code Representations by Predicting Subtrees
89 stars 24 forks source link

How to specific the output vector size? #20

Open allomy opened 2 years ago

allomy commented 2 years ago

Thank you for your wonderful work.

I have a question that, it seems the output is a 100-dimensional vector by default, how can I change the output vector size to a specific number?

allomy commented 2 years ago

https://github.com/bdqnghi/infercode/blob/8c22a3353aabb2f02ce1e044d439969df7463a0d/infercode/configs/default_config.ini#L4 is where the default config for the output vector size, but didn't find an API to specific it through the InferCode library.