Skip to content

Commit 0c96140

Browse files
update default values
Signed-off-by: Youngeun Kwon <youngeunk@nvidia.com>
1 parent 1e922a6 commit 0c96140

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

nemo_rl/utils/packed_tensor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
@lru_cache(maxsize=1)
2424
def get_target_packed_tensor_size():
25-
memory_ratio = os.getenv("NRL_REFIT_BUFFER_MEMORY_RATIO", "0.01")
25+
memory_ratio = os.getenv("NRL_REFIT_BUFFER_MEMORY_RATIO", "0.02")
2626
device = torch.device("cuda")
2727
props = torch.cuda.get_device_properties(device)
2828
total_memory_bytes = props.total_memory
@@ -33,7 +33,7 @@ def get_target_packed_tensor_size():
3333

3434
@lru_cache(maxsize=1)
3535
def get_num_buffers():
36-
return int(os.getenv("NRL_REFIT_NUM_BUFFERS", "2"))
36+
return int(os.getenv("NRL_REFIT_NUM_BUFFERS", "3"))
3737

3838

3939
def packed_broadcast_producer(iterator, group, src, post_iter_func):

0 commit comments

Comments
 (0)