search

wubin5 / STTN

paper : <Spatial-Temporal Transformer Networks for Traffic Flow Forecasting>
158 stars 44 forks source link

A Modified Version for Batch Training #4

Closed Vicky-51 closed 3 years ago

Vicky-51 commented 3 years ago

Here is a modified version for batch training. Some other minor modifications have been made.