microsoft / archai

Accelerate your Neural Architecture Search (NAS) through fast, reproducible and modular research.

https://microsoft.github.io/archai

MIT License

467 stars 92 forks source link

fix(lm_eval_harness): Fixes inputs not being placed on device. #161

Closed gugarosa closed 1 year ago

gugarosa commented 1 year ago

Pull Request Template

Description

Fixes inputs and model not being placed on proper device in lm_eval_harness.

Changes

[X] Bug fix (non-breaking change which fixes an issue)

Configuration:

Archai version: 1.0.0
Environment: Python 3.7
OS: Windows 11

Final checks:

[X] Follows guidelines of project
[X] Self-review of code
[X] Commented code in hard-to-understand areas
[X] Corresponding changes to the documentation
[X] Changes generate no new warnings
[X] Dependent changes have already been merged
[X] Checked code correction any misspellings