From 51b9c7036bf5047232cbd60aa633846f7434985f Mon Sep 17 00:00:00 2001 From: borg323 Date: Fri, 5 Mar 2021 20:13:25 +0200 Subject: [PATCH] flag to keep training data --- lc0_main.go | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/lc0_main.go b/lc0_main.go index 4083bb6..2cdeee1 100644 --- a/lc0_main.go +++ b/lc0_main.go @@ -72,6 +72,7 @@ var ( report_gpu = flag.Bool("report-gpu", false, "Send gpu info to server for more fine-grained statistics") cudnn = flag.Bool("cudnn", true, "Prefer the cudnn backend (if available)") settingsPath = flag.String("config", "", "JSON configuration file to use") + keepData = flag.Bool("keep-data", false, "Do not delete training data") ) // Settings holds username and password. @@ -199,10 +200,11 @@ func uploadGame(httpClient *http.Client, path string, pgn string, var duration = time.Since(startTime) var speed = int(float64(totalGames) / duration.Hours() * 24) log.Printf("Completed %d games in %s time (%d games/day)", totalGames, duration, speed) - - err := os.Remove(path) - if err != nil { - log.Printf("Failed to remove training file: %v", err) + if !*keepData { + err := os.Remove(path) + if err != nil { + log.Printf("Failed to remove training file: %v", err) + } } return nil @@ -700,7 +702,7 @@ func train(httpClient *http.Client, ngr client.NextGameResponse, defer func() { // Remove the training dir when we're done training. trainDir := trainDirHolder[0] - if trainDir != "" { + if trainDir != "" && !*keepData { log.Printf("Removing traindir: %s", trainDir) err := os.RemoveAll(trainDir) if err != nil {