search

microsoft / IRNet

An algorithm for cross-domain NL2SQL
MIT License
264 stars 81 forks source link

error while training irnet-bert #31

Open jaydeepb-inexture opened 4 years ago

jaydeepb-inexture commented 4 years ago

@JasperGuo can you give me the requirement for my system to train the irnet-bert smoothly? with python ,pytorch version and cuda (9.2 or 10) my system spec are like below: 32gb ram 256gb ssd+1tb hdd 8 gb of 2080 NVIDIA gpu

Firstly i was getting this error [CUDA-OUT-OF-MEMORY] error message

to solve this i have change batch_size to 8. but now i got below error.

Screenshot from 2020-03-12 18-30-26

DevanshChoubey commented 4 years ago

Hey, @jaydeepb-ddit if you are still having this problem I was able to train the model in Kaggle notebooks which have 16gb GPU. Though it took 7 hours,on NVidia K80. and with 8 batch size on google colab