James Bradbury
|
af12f006f2
|
Use broadcast for dropout
Should be fast enough on GPU now that it's not going to be an optimization target again for a while. Hopefully isn't meaningfully slower on CPU?
|
2018-05-20 04:04:33 -07:00 |
|
Mike J Innes
|
baff20514d
|
gpu broadcast fix
|
2018-04-17 18:05:58 +01:00 |
|
Mike Innes
|
d12fb98f2a
|
nicer batchnorm shape error
|
2018-04-15 20:29:25 +01:00 |
|
Mike J Innes
|
5fd240f525
|
interface tweaks
|
2018-04-15 20:04:42 +01:00 |
|
Mike J Innes
|
73a0be3e04
|
Merge branch 'master' into pull-request/07b0f95d
|
2018-04-15 17:10:29 +01:00 |
|
Mike J Innes
|
cb3ae8df6a
|
rename normalise.jl
|
2018-04-15 15:45:46 +01:00 |
|