1.when i train RNN network in pytorch, the batch size is defined before the training.When i finish training,I want change the batch_size to predict the result of test set.How can i implement it?
2.when i train, i use GPU to train.when i predict the result, i wnat use the GPU, because my cpu’s memory is larger than the GPU’s.How can i implemet it?
I’m confused at them.Can you help me, please!
Thank you.