首页
学习
活动
专区
工具
TVP
发布
社区首页 >问答首页 >RNN 中的time_step到底如何设置?

RNN 中的time_step到底如何设置?

提问于 2020-06-01 09:52:02
回答 0关注 0查看 267

for epoch in range(EPOCH): for step, (x, b_y) in enumerate(train_loader): # gives batch data b_x = x.view(-1, ?, ?) #

网店转让这里的reshape我应该设置为多少? 按照教程,应该设置为(batch, time_step, input_size)的形式,因为我想让网络记住每个sample的feature的sequence,所以我把time_step设置成了20(一个sample有20个featur,最后我得到了(-1,20,20),但是会报错,然后我又改成了(-1,1,20), 这就没问题了,但是改成这样,是不是就没有记住feature的sequence? output = rnn(b_x) 网店转让

loss = loss_func(output, b_y) optimizer.zero_grad() loss.backward() optimizer.step(www.alwdzr.com)

回答

和开发者交流更多问题细节吧,去 写回答
相关文章

相似问题

相关问答用户
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档