i already made some progress Pytorch code class ResidualBlock(nn.Module): def __init__(self, in_channels, channels=32, residual_beta=0.2): super().__init__() self.residual_beta = residual_beta self.blocks = nn.ModuleList() for i in range(5): self.bl…

Would this be of any help? using Flux using Zygote using Functors struct Prelu{T<:AbstractVector} slope::T end Prelu(dim::Int=1; init::Real=0.25f0) = Prelu(fill(float(init), dim)) @functor Prelu function (p::Prelu)(x::AbstractArray) triv = x isa AbstractVector ? () : ntuple(_ -> 1, ndims(…

Need help to translate this Pytorch part to Flux

Specific Domains Machine Learning

Tomas_Pevny August 26, 2021, 4:41pm 5

Can you make it minimal to better see, what is causing a problem? I do not have a GPU, therefore I might not be able to help.

1 Like

Topic		Replies	Views
Attempting to reimplement SRGAN from Pytorch to Flux, any help appreciated Machine Learning	4	1044	June 26, 2022
Translating from torch to Flux Machine Learning	2	305	June 7, 2023
Converting PyTorch to Flux while keeping performance Machine Learning flux , zygote	7	1390	May 29, 2022
Porting a RNN model to Flux from PyTorch Machine Learning	5	2130	October 29, 2018
Flux Transformer Out of Memory Machine Learning	25	1471	March 13, 2023

Need help to translate this Pytorch part to Flux

Related topics