Reproducing standard benchmarks using Transolver++

Open
#8 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Assessment

Difficulty
4/5
Estimated time
3-5 days
Newbie friendliness
35/100
Issue type
Bug
Clarity
Needs clarification
Activity status
Quiet
Tech stack
python

Research direction

No files, tests, or entry points are named. Start by locating the training scripts and configs for the six standard benchmarks, then compare them with the paper’s hyperparameters and the stated in_project_fx and Gumbel-Softmax changes. Done means either reproducing the reported results or documenting the additional changes needed.

Written by the indexing model from the issue text.

Description

Hi, and thanks for the great work!
I’m having trouble reproducing the reported Transolver++ results on the six standard benchmarks. My understanding is that Transolver++ (a) reduces in_project_fx and (b) replaces the softmax with Gumbel-Softmax. Using the training hyperparameters from the paper, my runs do not outperform the original Transolver.

  1. Are there additional architectural or training changes beyond reducing in_project_fx and using Gumbel-Softmax?
  2. Could the reduced parameter count require capacity to be added elsewhere?
  3. Has anyone successfully matched the paper’s numbers, and if so, could you share the training scripts and configs?

Thanks in advance!

Dominant language
Python
Stars
90
Forks
12
PR merge metrics
No merged PRs in 30d

Getting set up

This project ships no dev container, Dockerfile or contributing guide, so setting up is up to you: start from its README, and see our first-contribution guide for the general steps.

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

More from thuml/Transolver_plus

All issues in thuml/Transolver_plus

Similar issues

More Python issues

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.