Training bị đứng trên RTX 2080 about vietocr HOT 5 CLOSED

hi e,
e thử cài lại pytorch version mới nhất hoặc version trùng với version trên notebook xem sao?

from vietocr.

Em thấy pytorch trên notebook là pytorch 1.9 với cuda 10.2, nhưng mà máy của em hiện chỉ có cuda 10.1 ạ.

from vietocr.

cuda ko cần cài lại. e thử các cách sau:

• cài lại pytorch
• set threading trong dataloader bằng 1
• remove dataset vừa tạo để nó tạo lại nhé.

from vietocr.

Em có chỉnh lại num_workers trong dataloader thành 1, nhưng có vẻ không được ạ, mem GPU vẫn được sử dụng nhưng quá trình training vẫn không chạy ạ.

from vietocr.

mem gpu có là do model đã được khởi tạo. nhưng lúc loop qua dataset của e bị đứng.
sau khi khỏi tạo model e có thể lấy model.train_gen sau đó loop qua để check thử bị sao nhé.

model = ...
data_iter = iter(model.train_gen)
next(data_iter)  # em check xem dòng này chạy được ko nhé. 

from vietocr.