Learn R Programming

RBERT (version 0.1.11)

.get_model_subdir: Locate a subdir for a BERT checkpoint

Description

Locate a subdir for a BERT checkpoint

Usage

.get_model_subdir(model, dir)

Arguments

model

Character vector. Which model checkpoint to download.

dir

Character vector. Destination directory for checkpoints. Leave NULL to allow RBERT to automatically choose a directory. The path is determined from the dir parameter if supplied, followed by the `RBERT.dir` option (set using set_BERT_dir), followed by an "RBERT" folder in the user cache directory (determined using user_cache_dir). If you provide a dir, the `RBERT.dir` option will be updated to that location. Note that the checkpoint will create a subdirectory inside this dir.

Value

The path to the sub-directory where the checkpoint should be saved.