2015-10-22 40 views
9

Tôi làm việc trên một dự án Torch/Lua, trong đó tôi đã thực hiện một mô hình mạng nơron nhân tạo. Mọi thứ đều hoạt động, nhưng bây giờ tôi muốn sửa đổi mã của mình theo cách sau. Vì dữ liệu đầu vào của tôi rất lớn, tôi muốn chia nó thành N = 20 nhịp.Torch/Lua, làm thế nào để lưu một mô hình mạng thần kinh được đào tạo để tập tin?

Sau đó, tôi muốn đào tạo mạng thần kinh của mình chỉ trên khoảng tập dữ liệu thứ nhất và sau đó kiểm tra trên N-1 = 19 nhịp song song khác.

Để chạy tất cả các công việc song song này, tôi cần lưu chi tiết mô hình mạng nơron của mình vào một tệp và sau đó tải nó cho mỗi 19 công việc.

Có cách nào trong đuốc để ghi chính xác "viết" mô hình mạng nơron nhân tạo để nộp không?

Trả lời

15
-- save the model 
torch.save(filename, model) 

.

-- load the model 
model = torch.load(filename) 
+0

Cảm ơn @smhx, dễ hơn tôi nghĩ! –

+0

Có cách nào để lưu nó và tải nó vào Lua (không có Torch) không? – Adam12344

+0

@ Adam12344 trước khi tải mạng trở lại đuốc, bạn cần 'yêu cầu' các gói được sử dụng trong mô hình được tuần tự hóa, chẳng hạn như' nn', nếu không tải sẽ thất bại. Do đó, điều này chắc chắn cũng sẽ thất bại với 'Lua (không có Torch)' nếu bạn không thể tải, ví dụ: gói 'nn' trước khi tải mô hình. –

Các vấn đề liên quan