Load Model set Device - Githubissues

UoA-CARES / cares_reinforcement_learning

CARES Reinforcement Learning Package

11 stars 2 forks source link

Closed beardyFace closed 1 week ago

beardyFace commented 1 month ago

Models are currently loaded into CPU and not added to GPU - update load_model method to handle a device parameter