core.models.retro.utils

Module Contents

Functions

get_config_path	Config copy stored within retro project dir.
get_gpt_data_dir	Get project-relative directory of GPT bin/idx datasets.
get_all_true_mask

API

core.models.retro.utils.get_config_path(project_dir: str) → str

Config copy stored within retro project dir.

core.models.retro.utils.get_gpt_data_dir(project_dir: str) → str

Get project-relative directory of GPT bin/idx datasets.

core.models.retro.utils.get_all_true_mask(size, device)