search

Fried-Rice-Lab / FriedRiceLab

Official repository of the Fried Rice Lab, including code resources of the following our works: ESWT [arXiv], etc. This repository also implements many useful features and out-of-the-box image restoration models.
MIT License
205 stars 31 forks source link

About Flops #19

Closed CR7forMadrid closed 1 year ago

CR7forMadrid commented 1 year ago

Hello, when I apply the Analyse integrated with your project to test my model, it prompts Cuda out of memory, but after commenting out Flops it works fine.Is there any other way to test Flops? We noticed that your project only uses one graphics card for testing when performing analyse, what should I type on the command line if I want to use two cards to test flops?

jnpngshiii commented 1 year ago

We do not implement multi-card analysis method :(

But you can modify the parameter img_size of function analyse_pipeline to avoid this error~