Using trained models with CPU

facebookresearch / hanabi_SAD

Simplified Action Decoder for Deep Multi-Agent Reinforcement Learning

Other

96 stars 35 forks source link

Using trained models with CPU #19

Closed ingo-ch closed 3 years ago

ingo-ch commented 3 years ago

Hello

I don't have access to a GPU and was hoping to use the trained models on a mac cpu. I tried to change the map_location to 'cpu' device in eval_model.py but it won't work. Is there a way to use the trained models with a cpu machine?

hengyuan-hu commented 3 years ago

I pushed a patch for evaluation with CPU. Hope it helps.

ingo-ch commented 3 years ago

Thank you very much! It works now.