Skip to content
Snippets Groups Projects

`memory_demanding` for TF based transformers

Merged Tiago de Freitas Pereira requested to merge memory_demanding into master
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

Loading
Loading

Activity

Filter activity
  • Approvals
  • Assignees & reviewers
  • Comments (from bots)
  • Comments (from users)
  • Commits & branches
  • Edits
  • Labels
  • Lock status
  • Mentions
  • Merge request status
  • Tracking
7 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
    Loading