machine-learning-practice-code

machine-learning-practice-code copied to clipboard

抱歉又打扰您，我在用您的DEMO做训练的时候发现数据里面Y/mask/weight都被加载到了GPU中，但是模型和X没有，而且在训练的时候是CPU占用率最高。 正常情况下这种加载方式不是会报错的么，为什么在这里能够正确训练呢？我尝试把X和model都放入GPU中，好像运算时间也没有缩短。求问这是怎么回事。 ~~~python for epoch in range(num_epoch): start = time.time() num, total_loss = 0, 0 # if epoch == 5000: # optimizer.param_groups[0]['lr'] = lr * 0.1 data = tav_data_iterator( corpus_indice,...

jwc19890114