Closed hobofan closed 8 years ago
Yes, we should do that.
commit message: "implace" -> "in-place"
The in-place activation layers bring the memory usage down. For the Alexnet from 3GB to 2.5GB and for the Overfeat Network from 8.2GB to 7.4GB.
The .to_string()
method calls couldn't be removed, because that caused unresolvable issues with the collection String from env
.
@homu r+
:pushpin: Commit 223eaa4 has been approved by MichaelHirn
Activations can now be calculated in-place, requiring less memory. To use it, the same blob name should be supplied as input and output to a activation layer.
Example: