Skip to content
This repository has been archived by the owner on Oct 19, 2024. It is now read-only.

Commit

Permalink
Default values for precision and normalization
Browse files Browse the repository at this point in the history
  • Loading branch information
AjeyPaiK committed Jan 19, 2024
1 parent 3ae4c2c commit a65e85d
Showing 1 changed file with 6 additions and 1 deletion.
7 changes: 6 additions & 1 deletion ahcore/callbacks/h5_callback.py
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,12 @@

class WriteH5Callback(Callback):
def __init__(
self, max_queue_size: int, max_concurrent_writers: int, dump_dir: Path, normalization_type: str, precision: str
self,
max_queue_size: int,
max_concurrent_writers: int,
dump_dir: Path,
normalization_type: str = "logits",
precision: str = "float32",
):
"""
Callback to write predictions to H5 files. This callback is used to write whole-slide predictions to single H5
Expand Down

0 comments on commit a65e85d

Please sign in to comment.