modify lstm ouput for aggregated model #45
Merged
Add this suggestion to a batch that can be applied as a single commit.
This suggestion is invalid because no changes were made to the code.
Suggestions cannot be applied while the pull request is closed.
Suggestions cannot be applied while viewing a subset of changes.
Only one suggestion per line can be applied in a batch.
Add this suggestion to a batch that can be applied as a single commit.
Applying suggestions on deleted lines is not supported.
You must change the existing code in this line in order to create a valid suggestion.
Outdated suggestions cannot be applied.
This suggestion has been applied or marked resolved.
Suggestions cannot be applied from pending reviews.
Suggestions cannot be applied on multi-line comments.
Suggestions cannot be applied while the pull request is queued to merge.
Suggestion cannot be applied right now. Please check back later.
Using the h value of the lstm top layer as the input for aggregated model is better, and easier to converge.
Train log:
INFO:tensorflow:/job:master/task:0: training step 218206| Hit@1: 0.87 PERR: 0.72 GAP: 0.81 Loss: 5.09105
INFO:tensorflow:/job:master/task:0: training step 218207| Hit@1: 0.85 PERR: 0.73 GAP: 0.81 Loss: 5.16746
INFO:tensorflow:/job:master/task:0: training step 218208| Hit@1: 0.79 PERR: 0.68 GAP: 0.80 Loss: 5.13253
INFO:tensorflow:/job:master/task:0: training step 218209| Hit@1: 0.90 PERR: 0.76 GAP: 0.82 Loss: 5.16415
INFO:tensorflow:/job:master/task:0: training step 218210| Hit@1: 0.88 PERR: 0.76 GAP: 0.80 Loss: 5.36496
INFO:tensorflow:/job:master/task:0: training step 218211| Hit@1: 0.83 PERR: 0.72 GAP: 0.82 Loss: 4.48507
INFO:tensorflow:/job:master/task:0: training step 218212| Hit@1: 0.85 PERR: 0.71 GAP: 0.83 Loss: 4.91971
Eval on frame level validation set,
Avg_Hit@1: 0.838 | Avg_PERR: 0.705 | MAP: 0.367 | GAP: 0.780 | Avg_Loss: 5.795533
The following are the training status
The orange curves: original code using all concatenated state.
The blue curves: modified code using top layer h state.