if not os.path.exists(args.embedding_path):
print("word embedding model file was not found")
exit()
if args.restore is not None and not os.path.exists(args.restore):
print("restore model file was not found")
exit()
After Change
if args.embedding_path is not None and not os.path.exists(args.embedding_path):
print("word embedding model file was not found")
sys.exit(0)
if args.restore is not None and not os.path.exists(args.restore):
print("restore model file was not found")
sys.exit(0)
dataset = SNIPS(sentence_length=args.sentence_length,