|
@@ -37,7 +37,7 @@ import (
|
|
|
"google.golang.org/grpc/codes"
|
|
|
status "google.golang.org/grpc/status"
|
|
|
|
|
|
- neuralnetworkbase "../neuralnetworkbase"
|
|
|
+ neuralnetwork "../neuralnetwork"
|
|
|
"gonum.org/v1/gonum/mat"
|
|
|
grpc "google.golang.org/grpc"
|
|
|
|
|
@@ -45,7 +45,7 @@ import (
|
|
|
)
|
|
|
|
|
|
type RemoteControl struct {
|
|
|
- nn *neuralnetworkbase.NeuralNetwork
|
|
|
+ nn *neuralnetwork.NeuralNetwork
|
|
|
activationsQueue chan *LayerMatrix
|
|
|
biasesQueue chan *LayerMatrix
|
|
|
weightsQueue chan *LayerMatrix
|
|
@@ -53,7 +53,7 @@ type RemoteControl struct {
|
|
|
mutex sync.Mutex
|
|
|
}
|
|
|
|
|
|
-func (rw *RemoteControl) Init(nn *neuralnetworkbase.NeuralNetwork) {
|
|
|
+func (rw *RemoteControl) Init(nn *neuralnetwork.NeuralNetwork) {
|
|
|
rw.nn = nn
|
|
|
rw.activationsQueue = make(chan *LayerMatrix, 5)
|
|
|
rw.biasesQueue = make(chan *LayerMatrix, 5)
|
|
@@ -203,8 +203,8 @@ func (rw *RemoteControl) DummyStart(context.Context, *None) (*None, error) {
|
|
|
rw.nn.SaveState(outFile)
|
|
|
outFile.Close()
|
|
|
|
|
|
- rw.UpdateState(neuralnetworkbase.StateLearning)
|
|
|
- defer rw.UpdateState(neuralnetworkbase.StateIdle)
|
|
|
+ rw.UpdateState(neuralnetwork.StateLearning)
|
|
|
+ defer rw.UpdateState(neuralnetwork.StateIdle)
|
|
|
failCount := 0
|
|
|
teacher.Reset()
|
|
|
for teacher.NextValidator() {
|
|
@@ -226,7 +226,7 @@ func (rw *RemoteControl) DummyStart(context.Context, *None) (*None, error) {
|
|
|
fmt.Printf("Fail count: %v\n\n", failCount)
|
|
|
failCount = 0
|
|
|
teacher.Reset()
|
|
|
- rw.UpdateState(neuralnetworkbase.StateIdle)
|
|
|
+ rw.UpdateState(neuralnetwork.StateIdle)
|
|
|
}()
|
|
|
|
|
|
return &None{}, nil
|