* Move model to cpu after testing so that gpu memory is deallocated * Add code comment --------- Co-authored-by: sgwhat <ge.song@intel.com> |
||
|---|---|---|
| .. | ||
| test_optimize_model.py | ||
| test_transformers_api.py | ||
* Move model to cpu after testing so that gpu memory is deallocated * Add code comment --------- Co-authored-by: sgwhat <ge.song@intel.com> |
||
|---|---|---|
| .. | ||
| test_optimize_model.py | ||
| test_transformers_api.py | ||