ShenZhang-Shin comments

Repositories
Issues
Comments

Results 2 comments of


                                            ShenZhang-Shin

[BUG]: The error of running diffusion: ImportError: Please install colossalai from source code to use HybridAdam

Hi, I try to change the version of CUDA, pytorch and colossalai and succeed. Here is my version: CUDA : 11.2 CUDNN: 8.1.0 pytorch: 1.11.0+cu113 colossalai: 0.1.10+torch1.11cu11.3 pytorch-lightning: 1.8.6 But...

Train with the provided hyperparameters and cannot get a small model with 78.8% acc1 and 3.6G FLOPs

> Hello, I want to ask you, how can you get the FLOPs? I get the FLOPs of each sample on the ImageNet Val dataset and average the FLOPs.