Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

use Flux.@layer instead of Flux.@functor #452

Closed
CarloLucibello opened this issue Jul 24, 2024 · 0 comments · Fixed by #484
Closed

use Flux.@layer instead of Flux.@functor #452

CarloLucibello opened this issue Jul 24, 2024 · 0 comments · Fixed by #484
Milestone

Comments

@CarloLucibello
Copy link
Member

CarloLucibello commented Jul 24, 2024

This would get us fancy printing with model parameters count and suppress the warnings

┌ Warning: This type should probably now use `Flux.@layer` instead of `@functor`: WithGraph{SAGEConv{Matrix{Float32}, Vector{Float32}, typeof(identity), typeof(mean)}, GNNGraph{Tuple{Vector{Int64}, Vector{Int64}, Nothing}}}
@CarloLucibello CarloLucibello added this to the v1.0 milestone Aug 5, 2024
This was referenced Aug 7, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant