Bất kỳ công cụ pytorch để theo dõi đào tạo mạng lưới thần kinh?


7

Có công cụ nào để giám sát việc đào tạo của mạng trong PyTorch không?

Tôi đang tìm kiếm một tương đương với tenorboard trong tenorflow.

Câu trả lời:


4

Tôi đang sử dụng tenorboardX . Nó hỗ trợ hầu hết các tính năng của TensorBoard. Tôi đang sử dụng vô hướng, hình ảnh, phân phối, biểu đồ và văn bản. Không thử phần còn lại như âm thanh và đồ thị. Nhưng repo cũng chứa các ví dụ cho những usecase đó. Cài đặt có thể được thực hiện dễ dàng với pip. Tất cả đều được giải thích trong readme.

Ngoài ra còn có các phần mềm khác triển khai trình bao bọc cho PyTorch (và các ngôn ngữ / khung khác) của TensorBoard. Theo tôi biết, họ hỗ trợ ít chức năng hơn. Nhưng hãy xem CrayonTensorboard-Logger .


2

Tôi cũng đang sử dụng Tensorboard-PyTorch (TensorboardX). Hơn tất cả tôi khá hài lòng với nó.

Nhưng đừng cố hình dung đồ thị. Ít nhất là không có chút phức tạp nào (ví dụ như resnet50 sẽ không hoạt động). Có một số vấn đề về nó trên trang github của họ .

Nhưng tốt hơn nên kiểm tra diễn đàn Pytorch thường xuyên. Pytorch dường như di chuyển khá nhanh. Và một triển khai tenorboard trực tiếp cũng được lên kế hoạch .


Khi sử dụng trang web của chúng tôi, bạn xác nhận rằng bạn đã đọc và hiểu Chính sách cookieChính sách bảo mật của chúng tôi.
Licensed under cc by-sa 3.0 with attribution required.