search

kacky24 / stylenet

A pytorch implemention of "StyleNet: Generating Attractive Visual Captions with Styles"
63 stars 10 forks source link

About the training. #7

Open Lzhushuai opened 4 years ago

Lzhushuai commented 4 years ago

Hi, It's a great work! I did not run the code. But I suspect there may be a problem. In the paper, the author said that in the multi-task training(ie.training with romantic sentences or humorous sentences) he only update the matrix set {S}. But in your code, the whole LSTM was updated. I didn't find that you frozen other matrixs in somewhere. am I right? Thank you!

kacky24 commented 4 years ago

Hello. I'm sorry for my late reply.

Your advice seems to be right. I really appreciate it!! I'll modify my mistake when I have time.