bytedance / 1d-tokenizer

This repo contains the code for our paper An Image is Worth 32 Tokens for Reconstruction and Generation
Apache License 2.0
231 stars 7 forks source link

A question about the computation cost #2

Closed speedcell4 closed 1 week ago

speedcell4 commented 1 week ago

Hi, thanks for sharing such a wonderful paper and its code~

I'd like to know the computation cost of training these models. Do you use NVIDIA A100? How many cards and how long does the training take?

Thanks ahead~

cornettoyu commented 1 week ago

Hi,

Please see the appendix A Training and Testing Protocols for the training costs, I attached a screenshot for reference.

image image