diff --git a/main.py b/main.py index a618b92fe485cbada2a1f4880a653d4835cae901..95093345dbe3114566222631aae0d362c86fb763 100755 --- a/main.py +++ b/main.py @@ -62,11 +62,13 @@ if __name__ == "__main__" : if args.mode == "train" : json.dump([t.name for t in transitionSet], open(args.model+"/transitions.json", "w")) json.dump(strategy, open(args.model+"/strategy.json", "w")) + print("Transition Set :", [trans.name for trans in transitionSet], file=sys.stderr) Train.trainMode(args.debug, args.corpus, args.type, transitionSet, strategy, args.model, int(args.iter), int(args.batchSize), args.dev, args.bootstrap, args.incr, args.silent) elif args.mode == "decode" : transNames = json.load(open(args.model+"/transitions.json", "r")) transitionSet = [Transition(elem) for elem in transNames] strategy = json.load(open(args.model+"/strategy.json", "r")) + print("Transition Set :", [trans.name for trans in transitionSet], file=sys.stderr) Decode.decodeMode(args.debug, args.corpus, args.type, transitionSet, strategy, args.model) else : print("ERROR : unknown mode '%s'"%args.mode, file=sys.stderr)