ModelCheckpoint không hoạt động

Tôi đang cố gắng đào tạo một mô hình trong máy ảnh và tôi đang sử dụng ModelCheckpoint để lưu mô hình tốt nhất theo số liệu xác thực được theo dõi (trong trường hợp của tôi là chỉ số Jaccard ).

Mặc dù tôi có thể thấy mô hình được cải thiện trong bảng điều khiển, khi tôi cố gắng tải trọng lượng và đánh giá mô hình thì nó không hoạt động. Hơn nữa, bằng dấu thời gian trên tệp nơi các trọng số được cho là được lưu trữ, tôi có thể nói rằng chúng hoàn toàn không được lưu. Dấu thời gian tương ứng với thời gian tôi bắt đầu đào tạo.

Có ai gặp phải một vấn đề như vậy trước đây?

Bạn có chạy ModelCheckpointtrên các tham số mặc định của nó (bên cạnh monitor)?

ModelCheckpointcó một tham số gọi là modechỉ định loại số liệu sẽ được sử dụng. modecó thể lấy 3 giá trị 'min' 'max'và 'auto'(là mặc định):

• min: có nghĩa là bạn muốn giảm thiểu số liệu (ví dụ: hàm mất).
• max: có nghĩa là bạn muốn tối đa hóa số liệu (ví dụ: độ chính xác).
• auto: cố gắng tự mình tìm ra những việc cần làm. Nếu bạn nhìn vào mã , nó sẽ kiểm tra xem tên của số liệu có chứa 'acc'hoặc nếu nó bắt đầu bằng 'fmeasure'. Nếu có, nó đặt chế độ thành max, nếu không, nó đặt thành chế độ min.

Trong trường hợp của bạn, bạn theo dõi chỉ số jaccard, đây là một số liệu bạn muốn tối đa hóa, vì vậy bạn muốn chế độ được đặt thành max. Thông thường vì "jaccard" chứa chuỗi "acc", ngay cả khi chế độ được đặt thành autonó sẽ hoạt động tốt.

Tuy nhiên, nếu bạn đặt tên cho số liệu của mình một cái gì đó tùy ý (ví dụ my_metric), chế độ mặc định sẽ được đặt thành min, điều đó có nghĩa là nó sẽ lưu trữ các trọng số đạt được hiệu suất thấp nhất trên số liệu của bạn, phải là trọng số của kỷ nguyên đầu tiên.

Gợi ý : lần sau hãy thử với mode='max'để chắc chắn.