MCPcopy Create free account
hub / github.com/CodingBeard/tfkg / callCallbacks

Method callCallbacks

model/model.go:931–964  ·  view source on GitHub ↗
(
	callbacks []callback.Callback,
	event callback.Event,
	mode callback.Mode,
	epoch int,
	batch int,
	metrics []callback.Log,
)

Source from the content-addressed store, hash-verified

929}
930
931func (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
966func (m *TfkgModel) Save(dir string) error {
967 signatureCount := len(m.model.Signatures)

Callers 1

processCallbacksMethod · 0.95

Calls 4

SaveMethod · 0.95
CallMethod · 0.65
ErrorMethod · 0.65
GetSaveDirMethod · 0.65

Tested by

no test coverage detected