Git Product home page Git Product logo

Comments (4)

MisaOgura avatar MisaOgura commented on May 22, 2024

Hi @rotobash thanks for submitting an issue.

I have reproduced the error - it is due to target tensor not being transferred to GPU.

Also, perhaps a better API design would be to allow users to explicitly set a device to use, rather than automatically using GPU if available.

I'm working on it and will open a PR soon :)

from flashtorch.

MisaOgura avatar MisaOgura commented on May 22, 2024

Hi again @rotobashm, the fix has been merged and the updated package is published as v0.0.8 (release note).

Let me know whether the issue gets solved, so I can close this :)

from flashtorch.

rotobash avatar rotobash commented on May 22, 2024

Hi @MisaOgura, so sorry for the delay I hadn't had a chance to look at this till now. This seems to solve the cuda transfer issue for the one model I tried, I'll try some other models when I get a chance and let you know if I run into any issues. Thanks for the fix!

from flashtorch.

MisaOgura avatar MisaOgura commented on May 22, 2024

Brilliant @rotobash - I would love to see flashtorch being used in other projects, and would love to list them on README in the future. So please do let me know how yours goes :)

from flashtorch.

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.