Spaces:
Build error
Build error
File size: 526 Bytes
a446b0b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import sys
import os
import argparse
sys.path.append(os.getcwd())
parser = argparse.ArgumentParser()
parser.add_argument("--experiment_type", type=str, default='atomic',
choices=["atomic", "conceptnet"])
parser.add_argument("--experiment_num", type=str, default="0")
args = parser.parse_args()
if args.experiment_type == "atomic":
from main_atomic import main
main(args.experiment_num)
if args.experiment_type == "conceptnet":
from main_conceptnet import main
main(args.experiment_num)
|