Skip to content
New issue

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

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

Already on GitHub? # 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?

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

No branches or pull requests

3 participants