svincoff's picture
dependencies and embedding_exploration benchmark
c43fbc6
raw
history blame contribute delete
494 Bytes
# Dictionary: key = run name, values = epochs. (use this option if you've trained your own model)
# # Or, List: item goes to path (use this option if you're using the "best" ckpt from FusOn-pLM paper)
FUSON_PLM_CKPT= "FusOn-pLM"
# Type of dim reduction
PLOT_UMAP = True
PLOT_TSNE = False
# Overwriting configs
PERMISSION_TO_OVERWRITE = False # if False, script will halt if it believes these embeddings have already been made.