Giter Site home page Giter Site logo

change device seems not work about tabnet HOT 8 CLOSED

SH-2093 avatar SH-2093 commented on July 24, 2024
change device seems not work

from tabnet.

Comments (8)

Optimox avatar Optimox commented on July 24, 2024

Does the command torch.cuda.is_available() returns True in your environment ?

from tabnet.

SH-2093 avatar SH-2093 commented on July 24, 2024

Does the command torch.cuda.is_available() returns True in your environment ?

yes, the cuda environment is prepared. I have no idea about how to pinpoint the issue.

tabnet1

this is the tabnet related part in the code,

tabnet2

from tabnet.

Optimox avatar Optimox commented on July 24, 2024

Why did you close this issue ? Did you solve your problem?

from tabnet.

SH-2093 avatar SH-2093 commented on July 24, 2024

Why did you close this issue ? Did you solve your problem?

sorry, i didn't solve it. close the issue by mistake click.

from tabnet.

Optimox avatar Optimox commented on July 24, 2024

what is the size of your dataset? what batch size are you using? the smaller the dataset and the smaller the batch the smaller the impact of your GPU on computational speed.

from tabnet.

SH-2093 avatar SH-2093 commented on July 24, 2024

what is the size of your dataset? what batch size are you using? the smaller the dataset and the smaller the batch the smaller the impact of your GPU on computational speed.

The size of the training dataset is approximately 600,000 samples, with each sample containing 256 features. the batch size is the default value (1024). In actually, I did not perform the operation of transferring the dataset to the GPU , according to the input parameter of the fit function in document is a numpy-type variable. I am not sure if this is the cause of the problem.

from tabnet.

Optimox avatar Optimox commented on July 24, 2024

no everything should be done internally, you may want to increase the batch size to see if you see an imrpovement in speed with gpu against cpu. It's diffocult to help you as the cpu vs gpu version has been here for a long time and is working for sure, so I don't know what is going wrong with your setting.

from tabnet.

SH-2093 avatar SH-2093 commented on July 24, 2024

no everything should be done internally, you may want to increase the batch size to see if you see an imrpovement in speed with gpu against cpu. It's diffocult to help you as the cpu vs gpu version has been here for a long time and is working for sure, so I don't know what is going wrong with your setting.

thank u very much for the help. I will comment here if i make any progress.

from tabnet.

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.