search

tobiascz / TeCNO

GNU General Public License v3.0
52 stars 7 forks source link

what are the model settings to reproduce your best results? #6

Closed wenxixi closed 3 years ago

wenxixi commented 3 years ago

Hi, I wonder what the model settings are in order to reproduce your acc=88.56±0.27 prec=81.64±0.41 rec=85.24±1.06 results?

While the paper states that num_stages=2, what did you use for num_features, and num_layers? Could you please confirm if they are 32 and 8?

The config file and the comments in code don't seem consistent. And I did find it in your paper. Thanks.

image

image

tobiascz commented 3 years ago

for the best performing TeCNO model I used: mstcn_layers: 9 mstcn_f_maps: 64 mstcn_f_dim: 2048 mstcn_stages: 2