Change based on review feedback.
This commit is contained in:
Родитель
f5e34f84b2
Коммит
7c0c1b0dc0
|
@ -41,7 +41,7 @@ trainNetwork = {
|
|||
# loss and error computation
|
||||
diff = labels - z
|
||||
sqerr = ReduceSum (diff.*diff, axis=1)
|
||||
rmse = SumElements (Sqrt (sqerr / labelDim))
|
||||
rmse = Sqrt (sqerr / labelDim)
|
||||
|
||||
# declare special nodes
|
||||
featureNodes = (features)
|
||||
|
|
|
@ -43,7 +43,7 @@ trainNetwork = {
|
|||
# loss and error computation
|
||||
diff = labels - z
|
||||
sqerr = ReduceSum (diff.*diff, axis=1)
|
||||
rmse = SumElements (Sqrt (sqerr / labelDim))
|
||||
rmse = Sqrt (sqerr / labelDim)
|
||||
|
||||
# declare special nodes
|
||||
featureNodes = (features)
|
||||
|
|
|
@ -38,7 +38,7 @@ TrainConvNet = {
|
|||
# define regression loss
|
||||
diff = regrLabels - ol
|
||||
sqerr = ReduceSum (diff.*diff, axis=1)
|
||||
rmse = SumElements (Sqrt (sqerr / labelDim))
|
||||
rmse = Sqrt (sqerr / labelDim)
|
||||
|
||||
featureNodes = (features)
|
||||
labelNodes = (regrLabels)
|
||||
|
|
Загрузка…
Ссылка в новой задаче