Git Product home page Git Product logo

Comments (2)

chAwater avatar chAwater commented on May 30, 2024

@theyearis1981 @Gabri95 I think this is a bug, this line of code should be removed.

escnn/escnn/nn/modules/linear.py

Line 305 in 0e8d301

linear.weight.data[:] = self

escnn/escnn/nn/modules/linear.py

Lines 289 to 312 in 0e8d301

def export(self) -> torch.nn.Linear:
r"""
Export this module to a normal PyTorch :class:`torch.nn.Linear` module and set to "eval" mode.
"""
# set to eval mode so the matrix and the bias are updated with the current
# values of the weights
self.eval()
_matrix = self.matrix
_bias = self.expanded_bias
has_bias = self.bias is not None
# build the PyTorch module
linear = torch.nn.Linear(self.in_type.size, self.out_type.size, bias=has_bias)
linear.weight.data[:] = self
# set the weights and the bias
linear.weight.data = _matrix.data
if has_bias:
linear.bias.data = _bias.data
return linear

from escnn.

Gabri95 avatar Gabri95 commented on May 30, 2024

yes, thanks for spotting that.
I've fixed the issue but I haven't had time to push all updates yet, sorry for that

Best,
Gabriele

from escnn.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.