Large encoder and decoder model about enet-training HOT 5 OPEN

Hi,

Would you please say how much memory on GPU it is required to run the model?

Thank You

from enet-training.

hi, @Mina1368
about 820MB when I run the trained model
Thanks!

from enet-training.

save model as:

torch.save('filename.t7', model:clearState())

to make model take less disk space for deployment

from enet-training.

hi, @culurciello
I saved model in this way:

torch.save(filename, model:clearState():get(1))

or in this way:

model:clearState()
torch.save(filename, model)

both of these are the same for encoder, about 400M

I also tried "torch.save('filename.t7', model:clearState())"
but it's still the same, about 400M

Do you have any idea?
Thanks!

from enet-training.

When I want to trian ENet in Cityscape Dataset or Camvid Dataset, how much GPU memory should I use

from enet-training.