ThemeStation icon indicating copy to clipboard operation
ThemeStation copied to clipboard

Some questions about ThemeStation

Open muses0229 opened this issue 1 year ago • 4 comments

Hi, I'd like to know how much memory is required for ThemeStation? An A100 or just RTX3090 with 24GB will be enough?

muses0229 avatar Jul 06 '24 06:07 muses0229

@yejr0229 Hi, our experiments are conducted on a single A100 GPU, 3090 with 24GB is not sufficient for our current version.

  • For the application of ThemeStation, our main objective is to generate theme-consistent variations of given 3D exemplars. It can also be used to generate a refined mesh with detailed texture if your input is a textured mesh.
  • As our current approach is optimization-based, it may take about 1~2 hours based on how many optimization steps you choose.
  • We are now designing another method with similar applications to ThemeStation, but with less GPU usage (can run on a GPU with 24GB or less) and faster time-consuming (less than 10 seconds). We will release the new model within these two months. You can follow my github/homepage for updates.

zhenwwang avatar Jul 07 '24 05:07 zhenwwang

@yejr0229 Hi, our experiments are conducted on a single A100 GPU, 3090 with 24GB is not sufficient for our current version.

  • For the application of ThemeStation, our main objective is to generate theme-consistent variations of given 3D exemplars. It can also be used to generate a refined mesh with detailed texture if your input is a textured mesh.
  • As our current approach is optimization-based, it may take about 1~2 hours based on how many optimization steps you choose.
  • We are now designing another method with similar applications to ThemeStation, but with less GPU usage (can run on a GPU with 24GB or less) and faster time-consuming (less than 10 seconds). We will release the new model within these two months. You can follow my github/homepage for updates.

@zhenwwang Hi, I tried to run the code on a single A6000 GPU with 48GB VRM but still out of memory. Can you provide some suggestions for this?

hey0chen avatar Jul 15 '24 09:07 hey0chen

@yejr0229 Hi, our experiments are conducted on a single A100 GPU, 3090 with 24GB is not sufficient for our current version.

  • For the application of ThemeStation, our main objective is to generate theme-consistent variations of given 3D exemplars. It can also be used to generate a refined mesh with detailed texture if your input is a textured mesh.
  • As our current approach is optimization-based, it may take about 1~2 hours based on how many optimization steps you choose.
  • We are now designing another method with similar applications to ThemeStation, but with less GPU usage (can run on a GPU with 24GB or less) and faster time-consuming (less than 10 seconds). We will release the new model within these two months. You can follow my github/homepage for updates.

@zhenwwang Hi, I tried to run the code on a single A6000 GPU with 48GB VRM but still out of memory. Can you provide some suggestions for this?

Hi, you may need an A100 GPU with 80GB. We are now working on another project (https://rag-3d.github.io/), which supports fast theme-aware 3D-to-3D generation and requires much less GPU memory. Code will be released once our paper is accepted.

zhenwwang avatar Sep 30 '24 13:09 zhenwwang