Ngốn VRAM nhanh quá!
DungLeMTA opened this issue · comments
Hello a Cường lại là e đây!
A cho e hỏi là code của file 'loader.py' ấy ạ.
Mình load tất cả ảnh 1 thể, rồi sau đó chia ra làm tập train và tập validation.
Nhưng mà đoạn sau mình lại ko giải phóng đi, do vậy mà lúc train trên colab, VRAM nó bị ngốn quá nhiều a ạ.
Được 2 epoch là out rồi!
A có thể chỉ cho e thêm về cách giải phóng bớt dữ liệu để VRAM có thể tải thêm các đợt tiếp theo được ko ạ?
E cám ơn a nhiều ạ!
Dũng
Hi e,
em hãy sử dụng vietocr thay thế cho repo cũ này nhé https://github.com/pbcquoc/vietocr
bên này train/test dễ hơn và cho kết quả tốt hơn đấy.
Vâng e cám ơn a nhiều ạ!