Skip to content

Instantly share code, notes, and snippets.

Embed
What would you like to do?
from torchbiggraph.config import parse_config
import attr
train_config = parse_config(CONFIG_PATH)
train_path = [convert_path(os.path.join(DATA_DIR, FILENAMES['train']))]
train_config = attr.evolve(train_config, edge_paths=train_path)
from torchbiggraph.train import train
train(train_config)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment