Update

zhangjingh · Nov 26, 2022 · 5e64e7f · 5e64e7f
1 parent d86d13c
commit 5e64e7f
Show file tree

Hide file tree

Showing 8 changed files with 2,067 additions and 2,067 deletions.
diff --git a/database/ZData.txt b/database/ZData.txt
diff --git a/decompressor.py b/decompressor.py
@@ -95,7 +95,7 @@
 # Training settings
 nfeatures = X.size(1)
 nlatent = 32
-epochs = -1
+epochs = 10000
 batchsize=32
 window=2
 logFreq = 500

diff --git a/onnx/compressor.onnx b/onnx/compressor.onnx
diff --git a/onnx/decompressor.onnx b/onnx/decompressor.onnx
diff --git a/onnx/projector.onnx b/onnx/projector.onnx
diff --git a/onnx/stepper.onnx b/onnx/stepper.onnx
diff --git a/projector.py b/projector.py
@@ -37,7 +37,7 @@
 nframes = X.size(0)
 nfeatures = X.size(1)
 nlatent = 32
-epochs = 500000
+epochs = 10000
 batchsize = 32
 logFreq = 500
 

diff --git a/stepper.py b/stepper.py
@@ -50,7 +50,7 @@
 # Training settings
 nfeatures = X.size(1)
 nlatent = 32
-epochs = 100000
+epochs = 10000
 batchsize=32
 window=20
 logFreq = 500