Commit Graph

2604 Commits

Author SHA1 Message Date
Adarsh Kumar 7ac647a7ac
Added loss functions 2020-02-05 22:29:15 +05:30
bors[bot] 60043fa2aa
Merge #1013
1013: Adapt to GPUArrays/CuArrays changes r=dhairyagandhi96 a=maleadt

Changes in response to a29df67184 and https://github.com/JuliaGPU/CuArrays.jl/pull/576. I suppose the next CuArrays release will need to be breaking because of this.

Maybe the `crossentropy` signature needs to be adjusted to support integer vectors, but I'll leave that decision up to Flux developers. This at least is the quick fix to get the tests passing again.

Co-authored-by: Tim Besard <tim.besard@gmail.com>
2020-02-03 16:29:48 +00:00
Dhairya Gandhi ddc2c20e68
Merge pull request #994 from FluxML/ox/doccustomtraining
Add custom training loops to docs
2020-02-01 11:13:54 +05:30
Dhairya Gandhi bc20103ea6 no-op copy 2020-01-31 13:23:33 +05:30
Tim Besard e2c2ec5575 Don't invoke GPU crossentropy with integers.
Broadcasting log on integers does not work.
2020-01-31 08:22:54 +01:00
Tim Besard e66a7f130f Don't compare CPU with GPU arrays. 2020-01-31 08:22:21 +01:00
Dhairya Gandhi b9fbee1ff0 ::typeof(op) -> op 2020-01-31 12:24:36 +05:30
Dhairya Gandhi 620cffc45c
Merge pull request #1008 from FluxML/tb/cuindex
Remove unused imports.
2020-01-29 18:52:53 +05:30
Tim Besard d88f63adb4 Remove unused imports. 2020-01-29 12:15:41 +01:00
Chris Rackauckas 9803826a36 test restructure on the GPU
Requires https://github.com/FluxML/Zygote.jl/pull/474
2020-01-20 13:53:28 -05:00
Lyndon White 7797e31b44
Add custom training loops to docs 2020-01-16 21:57:59 +00:00
bors[bot] d1edd9b16d
Merge #680
680: Added new loss functions. r=thebhatman a=thebhatman

I have added the KL Divergence Loss function, Poisson loss function, Logcosh loss, and Hinge loss function.

Co-authored-by: Manjunath Bhat <manjunathbhat9920@gmail.com>
Co-authored-by: thebhatman <manjunathbhat9920@gmail.com>
2020-01-13 15:46:25 +00:00
Manjunath Bhat 747e01ea02
Test to check for spurious promotions 2020-01-13 18:33:30 +05:30
Dhairya Gandhi 048c31f609 bump Flux version to v0.10.1 2020-01-13 18:16:29 +05:30
bors[bot] f7f0ebbffd
Merge #992
992: Compat bounds for a couple more packages r=dhairyagandhi96 a=dhairyagandhi96

adds compatibility bounds for a few more packages

cc @MikeInnes 

Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
2020-01-13 12:29:05 +00:00
Dhairya Gandhi cd4626d5a7 compat bounds for a couple more packages 2020-01-13 17:38:59 +05:30
bors[bot] 2b222b15fa
Merge #991
991: Update CuArrays + Zygote deps  r=dhairyagandhi96 a=dhairyagandhi96

cc @MikeInnes 

Co-authored-by: Dhairya Gandhi <dhairya@juliacopmuting.com>
2020-01-13 11:14:21 +00:00
Dhairya Gandhi e1698e6617 up cuarrays 2020-01-13 16:18:20 +05:30
Dhairya Gandhi e2a97aec24 up cuda+zygote deps 2020-01-13 16:16:24 +05:30
Dhairya Gandhi de40476beb doc tests on julia 1.3 2020-01-13 14:10:34 +05:30
Dhairya Gandhi d7953ff573 test on julia 1.3+ 2020-01-13 13:45:40 +05:30
Dhairya Gandhi da9f295a8e bump version to 10.1 2020-01-13 13:41:25 +05:30
Dhairya Gandhi 370fd978fa
Merge pull request #986 from FluxML/restructure
Destructure/restructure for models
2020-01-13 13:04:48 +05:30
Dhairya Gandhi 58a7941386 reduce bors timeout 2020-01-13 11:24:04 +05:30
Dhairya Gandhi 0411b9a3e8 rm second slash 2020-01-12 17:35:04 +05:30
Mike Innes f96270c213 free zygote 2020-01-09 17:16:41 +00:00
Mike J Innes 17732e7023 restructure; closes #747 2020-01-06 11:53:47 +00:00
aminya f00b532556 Adding CompatHelper 2020-01-06 03:17:25 +03:30
Dhairya Gandhi e92da0cf85
Merge pull request #973 from FluxML/sf/nnpack_tolerance
Give `NNPACK` a bit of numerical leeway
2019-12-23 15:57:56 +05:30
Elliot Saba 0fdcc00923 Give `NNPACK` a bit of numerical leeway 2019-12-23 01:31:26 -08:00
Viral B. Shah 8a1e2f19d7
Update README.md 2019-12-19 09:44:17 -05:00
Kyle Daruwalla 2f854bdfc0 Recommitting to trigger new build 2019-12-10 09:57:08 -06:00
Dhairya Gandhi ac4c49b63e
Merge pull request #954 from FluxML/decaydocs
[WIP] Decaydocs
2019-12-10 12:11:23 +05:30
Dhairya Gandhi a72ca2b05d fix args 2019-12-09 23:18:01 +05:30
Dhairya Gandhi 894c075b6d rm Zeros setindex 2019-12-09 21:40:58 +05:30
Dhairya Gandhi f39e184814 rm Zeros warning 2019-12-09 21:07:30 +05:30
Manjunath Bhat 8a93be8c6c
Change loss to cost 2019-12-09 20:39:46 +05:30
Kyle Daruwalla 04991d3261 Added entry to docs for outdims 2019-12-07 14:06:11 -06:00
Kyle Daruwalla 0cdd11c0dc Added tests for varying padding, stride, and dilation with outdims. 2019-12-07 14:05:50 -06:00
Kyle Daruwalla a64378b112 Switched to using NNlib for conv.jl outdims. 2019-12-07 13:21:26 -06:00
Kyle Daruwalla 6265b1fa39 Added tests for outdims 2019-12-05 22:54:25 -06:00
Kyle Daruwalla 31dda0ce6c Updated with all basic and conv layers outdims 2019-12-05 21:57:10 -06:00
DrChainsaw 755536bf5e Merge remote-tracking branch 'upstream/master' into samepad 2019-12-04 23:45:03 +01:00
Kyle Daruwalla b4ed16ad9c Added outdims for some basic layers 2019-12-03 22:48:48 -06:00
Kyle Daruwalla 9279d79e63
Merge pull request #1 from FluxML/master
Updating to upstream master
2019-12-03 21:09:35 -06:00
Fredrik Bagge Carlson e67f09c06d Correct some comments in decay docs 2019-12-03 15:32:23 +08:00
Fredrik Bagge Carlson 6e94e59afd Improve docs for decay optimisers 2019-12-03 15:27:44 +08:00
Mike J Innes f46b5243db
Merge pull request #946 from FluxML/pkg-up
compat, pkg up
2019-11-29 12:55:47 +00:00
Mike J Innes 0c99f7f4b7 Merge branch 'dg/news' into pkg-up 2019-11-29 10:42:28 +00:00
Dhairya Gandhi 4b63e69b65 bump version to v0.10 2019-11-29 00:02:59 +05:30