[feat] go/syz-manager: extend waiting time

m0ck1ng · Nov 2, 2022 · e3b6b82 · e3b6b82
1 parent 722e78b
commit e3b6b82
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 5 deletions.
diff --git a/prog/sched.go b/prog/sched.go
@@ -38,7 +38,7 @@ func (sch *Sched) Reset() {
 		atomic.StoreUint32(&sch.IntTotal[i], 0)
 		atomic.StoreUint32(&sch.weights[i], 5)
 	}
-	atomic.StoreUint32(&sch.weights[INSERT], 10)
+	// atomic.StoreUint32(&sch.weights[INSERT], 10)
 }
 
 func (sch *Sched) UpdateWeights() {

diff --git a/syz-manager/manager.go b/syz-manager/manager.go
@@ -204,9 +204,9 @@ func RunManager(cfg *mgrconfig.Config) {
 	// dumpSyscalls(mgr.cfg.Target.Syscalls)
 	go func() {
 		for {
-			time.Sleep(60 * time.Minute)
+			time.Sleep(4 * 60 * time.Minute)
 			modelPath := trainModel(mgr.cfg.Syzkaller, mgr.cfg.Workdir)
-			mgr.loadModel(modelPath, 1)
+			mgr.loadModel(modelPath, 0)
 			log.Logf(0, "model trained & updated")
 		}
 	}()

diff --git a/...i/lang_model/data/targetSyscalls_5.10.134 → ...anager/api/lang_model/data/targetSyscalls b/...i/lang_model/data/targetSyscalls_5.10.134 → ...anager/api/lang_model/data/targetSyscalls
diff --git a/tools/model_manager/api/views.py b/tools/model_manager/api/views.py
@@ -42,7 +42,6 @@ def model_train(request):
 
         syscalls = read_syscalls(f"{settings.BASE_DIR}/api/lang_model/data/targetSyscalls")
         corpus = read_corpus(corpus_dir)
-        print(corpus[0])
 
         # syzcorpus_dir = '/data5/corpus/syzkaller_corpus'
         # corpus = read_corpus(syzcorpus_dir)
@@ -60,7 +59,7 @@ def model_train(request):
         train_loader = DataLoader(train_dataset, batch_size=64, shuffle=True)
         test_loader = DataLoader(test_dataset, batch_size=64, shuffle=True)
 
-        device = torch.device("cuda:1" if torch.cuda.is_available() else "cpu")
+        device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
 
         num_epoch = 50
         lr = 0.001