Closed lhnguyen102 closed 1 month ago
@van-dai-vuong It is fixed for batch size of 1. In general, you dont need to approve if the issue hasn't fixed otherwise someone might accidentally merge to main. @davidwardan Could you please test it to see if it is fixed? Thanks
@lhnguyen102 batch_size = 1 for cuda is now working 👍🏻
@davidwardan Awesome! could you approve my PR to merge?
Description
This PR fixes the bugs in cuda version for LSTM that raised in issue #67.
Changes Made
Related Issues
close #67 #66
Note for Reviewers
pytagi
folder, we need to test and release new version which can be bloated. Based on the specific datasets, you should have your own dataloader saved inexamples
folder. You can use the Pytorch's dataloader likeexamples/mnist_bench.py
for time series.