How to tune infer_batch_size smaller in models?

geek-ai / MAgent

A Platform for Many-Agent Reinforcement Learning

MIT License

1.68k stars 332 forks source link

Closed wenshiwei closed 5 years ago

wenshiwei commented 5 years ago

When I run the examples the out-of-memory errors occured,And how to tune infer_batch_size smaller in models?

KornbergFresnel commented 5 years ago

read the code by yourself and more carefully