Merge branch 'master' of github.com:botkop/scorch

botkop · Dec 27, 2018 · f7100df · f7100df
2 parents 90129ce + 75cb57a
commit f7100df
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 5 deletions.
diff --git a/README.md b/README.md
@@ -1,3 +1,5 @@
+"What I cannot create, I do not understand." - Richard Feynman.
+
 Scorch
 ======
 Scorch is a deep learning framework in Scala inspired by PyTorch.

diff --git a/build.sbt b/build.sbt
@@ -1,3 +1,6 @@
+resolvers +=
+  "Sonatype OSS Snapshots" at "https://oss.sonatype.org/content/repositories/snapshots"
+
 import Dependencies._
 
 lazy val root = (project in file(".")).settings(

diff --git a/project/Dependencies.scala b/project/Dependencies.scala
@@ -2,5 +2,5 @@ import sbt._
 
 object Dependencies {
   lazy val scalaTest = "org.scalatest" %% "scalatest" % "3.0.3"
-  lazy val numsca = "be.botkop" %% "numsca" % "0.1.4"
+  lazy val numsca = "be.botkop" %% "numsca" % "0.1.5-SNAPSHOT"
 }
diff --git a/src/main/scala/scorch/nn/cnn/MaxPool2d.scala b/src/main/scala/scorch/nn/cnn/MaxPool2d.scala
@@ -180,7 +180,6 @@ x.backward(Variable(dx))
         val upd = (window == out(n, c, h, w)) * dOut(n, c, h, w)
         dx(n, c, h1 :> h2, w1 :> w2) := upd
       }
-
       x.backward(Variable(dx))
     }
   }

diff --git a/src/test/scala/scorch/nn/cnn/Conv2dSpec.scala b/src/test/scala/scorch/nn/cnn/Conv2dSpec.scala
@@ -140,9 +140,9 @@ class Conv2dSpec extends FlatSpec with Matchers {
     def fb(a: Variable) =
       Conv2d.Im2colConv2dFunction(x, w, a, pad, stride).forward()
 
-    oneOpGradientCheck(fx, x)
-    oneOpGradientCheck(fw, w.copy())
-    oneOpGradientCheck(fb, b.copy())
+    oneOpGradientCheck(fx, x, 3e-8)
+    oneOpGradientCheck(fw, w.copy(), 3e-8)
+    oneOpGradientCheck(fb, b.copy(), 3e-8)
   }
 
 }