INDArrayDumping.sc

Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
trait INDArrayDumping extends INDArrayWeights { | |
val dumpingPathPrefix: String | |
private var currentDumped: Int = 0 | |
override type INDArrayWeight <: INDArrayWeightApi with Weight | |
trait INDArrayWeightApi extends super.INDArrayWeightApi { | |
this: INDArrayWeight => | |
override protected def backward[SubtypeOfOptimizer]( | |
originalDelta: INDArray)( | |
implicit implicitApplyRest: ImplicitApply.Aux[PartiallyAppliedOptimizer, SubtypeOfOptimizer], | |
asOptimizer: SubtypeOfOptimizer <:< OptimizerApi {type Delta <: INDArray} | |
): Do[Unit] = { | |
super.backward(originalDelta).map { _ => | |
val os = new ObjectOutputStream(new FileOutputStream(dumpingPathPrefix + "/" + currentDumped.toString)) | |
try { | |
os.writeObject(data) | |
} finally { | |
os.close() | |
} | |
currentDumped += 1 | |
} | |
} | |
} | |
} |
MIT License | |
Copyright (c) 2017 ThoughtWorks Inc. | |
Permission is hereby granted, free of charge, to any person obtaining a copy | |
of this software and associated documentation files (the "Software"), to deal | |
in the Software without restriction, including without limitation the rights | |
to use, copy, modify, merge, publish, distribute, sublicense, and/or sell | |
copies of the Software, and to permit persons to whom the Software is | |
furnished to do so, subject to the following conditions: | |
The above copyright notice and this permission notice shall be included in all | |
copies or substantial portions of the Software. | |
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR | |
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, | |
FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE | |
AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER | |
LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, | |
OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE | |
SOFTWARE. |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment