GNN graph neural net training - I need help debugging the training script

Posted 2 weeks ago
Only freelancers located in the U.S. may apply. U.S. located freelancers only

We are trying to train a Graph Neural Net (GNN) to predict how shipments would move in a theoretical shipping network. We are training this on Vertex AI and we're having a hard time making our training script work. There is a persistent key error that we can't debug. We need some help to fix this. Deliverables will be fixing the script so the model will train. I will share data access to the person we hire for the job. See attached for the current script & output/error response.  Happy to answer any and all questions about how the data is set up.

For what its worth, my hunch is that it has something to do with how the data is being loaded and/or being batched.


  • $700.00

    Fixed-price
  • Expert
    Experience Level
  • Remote Job
  • One-time project
    Project Type
Skills and Expertise
Machine Learning Tools
PyTorch Vertex AI
Other
Machine Learning Python
+ 2 more
Activity on this job
  • Proposals:
    15 to 20
  • Last viewed by client:
    5 days ago
  • Hires:
    3
  • Interviewing:
    13
  • Invites sent:
    44
  • Unanswered invites:
    9
About the client
Member since Jun 4, 2018
  • United States
    New York 9:57 AM
  • 26 jobs posted
    58% hire rate, 1 open job
  • $11K total spent
    18 hires, 5 active
  • Mid-sized company (10-99 people)

Explore similar jobs on Upwork

Machine Learning TensorFlow Artificial Intelligence Neural Network Computer Vision Deep Learning
Need AI engineer
New
Hourly ‐ Posted 15 minutes ago
Artificial Intelligence Python Machine Learning Deep Learning Framework

How it works

  • Post a job icon
    Create your free profile
    Highlight your skills and experience, show your portfolio, and set your ideal pay rate.
  • Talent comes to you icon
    Work the way you want
    Apply for jobs, create easy-to-by projects, or access exclusive opportunities that come to you.
  • Payment simplified icon
    Get paid securely
    From contract to payment, we help you work safely and get paid securely.
Want to get started? Create a profile

About Upwork

  • Rating is 4.9 out of 5.
    4.9/5
    (Average rating of clients by professionals)
  • G2 2021
    #1 freelance platform
  • 49,000+
    Signed contract every week
  • $2.3B
    Freelancers earned on Upwork in 2020

Find the best freelance jobs

Growing your career is as easy as creating a free profile and finding work like this that fits your skills.

Trusted by

  • Microsoft Logo
  • Airbnb Logo
  • Bissell Logo
  • GoDaddy Logo