def profile_interactive(opt):
# Create model and assign it to the specified task
agent = create_agent(opt, requireModelExists=True)
human_agent = RepeatQueryAgent(opt)
world = create_task(opt, [human_agent, agent])
agent.opt.log()
pr = cProfile.Profile()
pr.enable()
# Run
cnt = 0
while True:
world.parley()
if opt.get('display_examples'):
print("---")
print(world.display())
cnt += 1
if cnt >= opt.get('num_examples', 100):
break
if world.epoch_done():
logging.info("epoch done")
break
pr.disable()
s = io.StringIO()
sortby = 'cumulative'
ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
ps.print_stats()
print(s.getvalue())
def profile_interactive(opt, print_parser=None):
if print_parser is not None:
if print_parser is True and isinstance(opt, ParlaiParser):
print_parser = opt
elif print_parser is False:
print_parser = None
if isinstance(opt, ParlaiParser):
print(
'[ Deprecated Warning: interactive should be passed opt not Parser ]'
)
opt = opt.parse_args()
# Create model and assign it to the specified task
agent = create_agent(opt, requireModelExists=True)
human_agent = RepeatQueryAgent(opt)
world = create_task(opt, [human_agent, agent])
if print_parser:
# Show arguments after loading model
print_parser.opt = agent.opt
print_parser.print_args()
pr = cProfile.Profile()
pr.enable()
# Run
cnt = 0
while True:
world.parley()
if opt.get('display_examples'):
print("---")
print(world.display())
cnt += 1
if cnt >= opt.get('num_examples', 100):
break
if world.epoch_done():
print("EPOCH DONE")
break
pr.disable()
s = io.StringIO()
sortby = 'cumulative'
ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
ps.print_stats()
print(s.getvalue())