Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Models #15

Open
joe-sht opened this issue Jan 17, 2024 · 2 comments
Open

Models #15

joe-sht opened this issue Jan 17, 2024 · 2 comments

Comments

@joe-sht
Copy link

joe-sht commented Jan 17, 2024

Hi, all! I met an issue on running
python main.py --workdir=../out --config=configs/maskgvt_bair_config.py
My goal is training some model on bair test set for frame prediction.
My issue is like no access to gs://magvit/models/bair_3d_base
I checked that this model is needed on evaluation step and tried to switch off evaluation.
But I still have issues and script looks not run without these models.
As I understand these models have not released yet. But is it possible to train some model without these ones
based only on downloaded 30 Gb Bair dataset?

@roadjiang
Copy link
Collaborator

Hi you should be able to train the BAIR model from scratch using a couple of GPUs.

As for the model pretrained weight release gs://magvit/models/bair_3d_base
Please see this note

@shinshiner
Copy link

Hi you should be able to train the BAIR model from scratch using a couple of GPUs.

As for the model pretrained weight release gs://magvit/models/bair_3d_base Please see this note

how long it takes to train the base/large model according to the configurations present in supplement files?

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants