Commit Graph

6 Commits (8ef4b2796eb17cbbdcd24a3b4024aacdf163c49a)

Author SHA1 Message Date
yuchaojie 55ee90e696 fix transformer eval device id
4 years ago
dessyang f27f047f14 Add training and evaluation of Transformer on GPU
4 years ago
hesham 9cee0d2143 Add num_epochs to non-sink training
4 years ago
xiefangqi 9b3c33e157 minddata iterator output ms_tensor
5 years ago
nhussain 92e99ff224 change map calls
5 years ago
yuchaojie df5d162417 move transformer net to new modelzoo directory
5 years ago