I am now using the pre-trained model to predict transformation.
My data are 8000 and 7000 respectively, but when using the model for inference, but it shows that the memory is out of bounds. How should I solve this problem in the code or data.?
My graphics card is RTX3090, changing the graphics card is too expensive for me.