`memory_demanding` for TF based transformers
2 unresolved threads
2 unresolved threads
There are some situations where you have SampleSets
with more than 1000 samples and we can't just TF forward
them in one shot without OOM.
In this MR I introduce the argument memory_demanding
, where, once it's set to true, will `forward one sample at a time.
Edited by Tiago de Freitas Pereira
Merge request reports
Activity
added 1 commit
- a079dc70 - Add a memory demanding feature in the TF based transformers
ping @amohammadi @ydayer @lcolbois
This is not good @tiago.pereira, you should use dask to split your data.
Edited by Amir MOHAMMADIsorry I thought this is related to bob.pipelines#27 (closed)
sorry I thought this is related to bob.pipelines#27 (closed)
No, no. The 2 things are not related
I can't break the API just to dask this thing again.
Furthermore,
memory_demanding
isFalse
by defaultEdited by Tiago de Freitas Pereira7 7 VanillaBiometricsPipeline, 8 8 ) 9 9 10 10 memory_demanding = False 11 11 if "database" in locals(): 12 12 annotation_type = database.annotation_type 13 13 fixed_positions = database.fixed_positions 14 memory_demanding = ( 15 database.memory_demanding if hasattr(database, "memory_demanding") else False 16 ) mentioned in commit b385d51c
Please register or sign in to reply