( callbacks []callback.Callback, event callback.Event, mode callback.Mode, epoch int, batch int, metrics []callback.Log, )
| 929 | } |
| 930 | |
| 931 | func (m *TfkgModel) callCallbacks( |
| 932 | callbacks []callback.Callback, |
| 933 | event callback.Event, |
| 934 | mode callback.Mode, |
| 935 | epoch int, |
| 936 | batch int, |
| 937 | metrics []callback.Log, |
| 938 | ) (halt bool, saved bool) { |
| 939 | for _, call := range callbacks { |
| 940 | actions, e := call.Call(event, mode, epoch, batch, metrics) |
| 941 | if e != nil { |
| 942 | m.errorHandler.Error(e) |
| 943 | continue |
| 944 | } |
| 945 | |
| 946 | for _, action := range actions { |
| 947 | if action == callback.ActionSave { |
| 948 | saveDirGetter, ok := call.(callback.HasSaveDir) |
| 949 | if ok { |
| 950 | e = m.Save(saveDirGetter.GetSaveDir()) |
| 951 | if e != nil { |
| 952 | m.errorHandler.Error(e) |
| 953 | } else { |
| 954 | saved = true |
| 955 | } |
| 956 | } |
| 957 | } else if action == callback.ActionHalt { |
| 958 | halt = true |
| 959 | } |
| 960 | } |
| 961 | } |
| 962 | |
| 963 | return halt, saved |
| 964 | } |
| 965 | |
| 966 | func (m *TfkgModel) Save(dir string) error { |
| 967 | signatureCount := len(m.model.Signatures) |
no test coverage detected