Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 1 addition & 1 deletion cm/random_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -97,7 +97,7 @@ def set_seed(self, seed):
class DeterministicIndividualGenerator:
"""
RNG to deterministically sample num_samples samples that does not depend on batch_size or mpi_machines
Uses a separate rng for each sample to reduce memoery usage
Uses a separate rng for each sample to reduce memory usage
"""

def __init__(self, num_samples, seed=0):
Expand Down
2 changes: 1 addition & 1 deletion evaluations/evaluator.py
Original file line number Diff line number Diff line change
Expand Up @@ -338,7 +338,7 @@ def evaluate(
return {
"fraction": float(np.mean(batch_predictions)),
"batch_predictions": batch_predictions,
"max_realisim_score": max_realism_score,
"max_realism_score": max_realism_score,
"nearest_indices": nearest_indices,
}

Expand Down