params = options.readCommandLine()
manualSeed = 1234
random.seed(manualSeed)
torch.manual_seed(manualSeed)
if params['useGPU']:
torch.cuda.manual_seed_all(manualSeed)
# setup dataloader
dlparams = params.copy()
dlparams['useIm'] = True
dlparams['useHistory'] = True
dlparams['numRounds'] = 10
splits = ['val', 'test']
dataset = VisDialDataset(dlparams, splits)
# Transferring dataset parameters
transfer = ['vocabSize', 'numOptions', 'numRounds']
for key in transfer:
if hasattr(dataset, key):
params[key] = getattr(dataset, key)
if 'numRounds' not in params:
params['numRounds'] = 10
# Always load checkpoint parameters with continue flag
params['continue'] = True
excludeParams = ['batchSize', 'visdomEnv', 'startFrom', 'qstartFrom', 'trainMode', \
'evalModeList', 'inputImg', 'inputQues', 'inputJson', 'evalTitle', 'beamSize', \
# ---------------------------------------------------------------------------
# command options
# ---------------------------------------------------------------------------
params = debug_options.readCommandLine()
print("params:", params)
# Seed rng for reproducibility
random.seed(params['randomSeed'])
torch.manual_seed(params['randomSeed'])
if params['useGPU']:
torch.cuda.manual_seed_all(params['randomSeed'])
# ---------------------------------------------------------------------------
# dataset
# ---------------------------------------------------------------------------
splits = ['train', 'val', 'test']
dataset = VisDialDataset(params, splits)
# extra Params to transfer from dataset
transfer = ['vocabSize', 'numOptions', 'numRounds']
for key in transfer:
if hasattr(dataset, key):
params[key] = getattr(dataset, key)
# Create save path and checkpoints folder
# os.makedirs('/hhd/lvxinyu/visdial-pytorch/checkpoints/', exist_ok=True)
os.mkdir(params['savePath'])
# ---------------------------------------------------------------------------
# Loading Modules
# ---------------------------------------------------------------------------
parameters = [] # save params from abot/qbot
aBot = None
def main(params):
aqmSetting = None
'''
if ("AQMBotRank" in params["evalModeList"]
or "AQMdialog" in params["evalModeList"]
or "AQMdemo" in params["evalModeList"]):
aqmSetting = getAQMSetting(params)
'''
# setup dataloader
dlparams = params.copy()
dlparams['useIm'] = True
dlparams['useHistory'] = True
dlparams['numRounds'] = 10
splits = ['val', 'test']
dataset = VisDialDataset(dlparams, splits)
# Transferring dataset parameters
transfer = ['vocabSize', 'numOptions', 'numRounds']
for key in transfer:
if hasattr(dataset, key):
params[key] = getattr(dataset, key)
if 'numRounds' not in params:
params['numRounds'] = 10
# Always load checkpoint parameters with continue flag
params['continue'] = True
excludeParams = ['batchSize', 'visdomEnv', 'startFrom', 'qstartFrom', 'trainMode', \
'evalModeList', 'inputImg', 'inputQues', 'inputJson', 'evalTitle', 'beamSize', \
'enableVisdom', 'visdomServer', 'visdomServerPort', 'randomCaption', 'zeroCaption',
'numImg', 'numQ', 'numA', 'alpha',
'qbeamSize', 'gamma', 'delta', 'lambda',
'onlyGuesser', 'randQ', 'gen1Q', 'gtQ', 'randA', 'noHistory',
'slGuesser', 'resampleEveryDialog']
aBot = None
qBot = None
aqmBot = None
# load aBot
print('load aBot')
if params['startFrom']:
aBot, loadedParams, _ = utils.loadModel(params, 'abot', overwrite=True)
assert aBot.encoder.vocabSize == dataset.vocabSize, "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
aBot.eval()
# Retaining certain dataloder parameters
for key in excludeParams:
params[key] = dlparams[key]
print('load qBot')
# load qBot
if params['qstartFrom'] and not params['aqmstartFrom']:
qBot, loadedParams, _ = utils.loadModel(params, 'qbot', overwrite=True)
assert qBot.encoder.vocabSize == params[
'vocabSize'], "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
qBot.eval()
# Retaining certain dataloder parameters
for key in excludeParams:
params[key] = dlparams[key]
print('load AQM-Bot')
# load aqmBot
if params['aqmstartFrom']: # abot of AQM
assert params['qstartFrom'] # qbot of AQM
aqmBot, loadedParams, _ = utils.loadModel(params,
'AQM-qbot',
overwrite=True)
assert aqmBot.questioner.encoder.vocabSize == params[
'vocabSize'], "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
aqmBot.eval()
# load qBot
for key in excludeParams:
params[key] = dlparams[key]
aqmQ, loadedParams, _ = utils.loadModel(params, 'qbot', overwrite=True)
assert aqmQ.encoder.vocabSize == params[
'vocabSize'], "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
aqmQ.eval()
for key in excludeParams:
params[key] = dlparams[key]
aqmBot.setQuestioner(aqmQ)
elif params['aqmQStartFrom']:
from visdial.models.aqm_questioner import AQMQuestioner
aqmBot = AQMQuestioner()
aqmBot.eval()
params['qstartFrom'] = params['aqmQStartFrom']
aqmQ, loadedParams, _ = utils.loadModel(params, 'qbot', overwrite=True)
assert aqmQ.encoder.vocabSize == params[
'vocabSize'], "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
aqmQ.eval()
for key in excludeParams:
params[key] = dlparams[key]
aqmBot.setQuestioner(aqmQ)
params['startFrom'] = params['aqmAStartFrom']
aqmA, loadedParams, _ = utils.loadModel(params, 'abot', overwrite=True)
assert aqmA.encoder.vocabSize == dataset.vocabSize, "Vocab size mismatch!"
for key in loadedParams:
params[key] = loadedParams[key]
aqmA.eval()
aqmBot.setAppAnswerer(aqmA)
for key in excludeParams:
params[key] = dlparams[key]
pprint.pprint(params)
#viz.addText(pprint.pformat(params, indent=4))
print("Running evaluation!")
numRounds = params['numRounds']
if 'ckpt_iterid' in params:
iterId = params['ckpt_iterid'] + 1
else:
iterId = -1
if 'test' in splits:
split = 'test'
splitName = 'test - {}'.format(params['evalTitle'])
else:
split = 'val'
splitName = 'full Val - {}'.format(params['evalTitle'])
print("Using split %s" % split)
dataset.split = split
if 'ABotRank' in params['evalModeList']:
if params['aqmstartFrom']:
aBot = aqmBot.appAnswerer
print('evaluating appBot of AQM')
print("Performing ABotRank evaluation")
rankMetrics = rankABot(aBot,
dataset,
split,
scoringFunction=utils.maskedNll,
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'])
print(rankMetrics)
for metric, value in rankMetrics.items():
plotName = splitName + ' - ABot Rank'
#viz.linePlot(iterId, value, plotName, metric, xlabel='Iterations')
if 'QBotRank' in params['evalModeList']:
print("Performing QBotRank evaluation")
rankMetrics, roundRanks = rankQBot(
qBot,
dataset,
split,
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'],
verbose=1)
for metric, value in rankMetrics.items():
plotName = splitName + ' - QBot Rank'
#viz.linePlot(iterId, value, plotName, metric, xlabel='Iterations')
for r in range(numRounds + 1):
for metric, value in roundRanks[r].items():
plotName = '[Iter %d] %s - QABots Rank Roundwise' % \
(iterId, splitName)
#viz.linePlot(r, value, plotName, metric, xlabel='Round')
if 'QABotsRank' in params['evalModeList']:
print("Performing QABotsRank evaluation")
outputPredFile = "/home/lvxinyu/code/aqm_plus/resources/data/visdial/visdial/output_predictions_rollout.h5"
rankMetrics, roundRanks = rankQABots(
qBot,
aBot,
dataset,
split,
beamSize=params['beamSize'],
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'],
zeroCaption=params['zeroCaption'],
randomCaption=params['randomCaption'],
numRounds=params['runRounds'])
for metric, value in rankMetrics.items():
plotName = splitName + ' - QABots Rank'
#viz.linePlot(iterId, value, plotName, metric, xlabel='Iterations')
for r in range(numRounds + 1):
for metric, value in roundRanks[r].items():
plotName = '[Iter %d] %s - QBot All Metrics vs Round'%\
(iterId, splitName)
#viz.linePlot(r, value, plotName, metric, xlabel='Round')
if 'AQMBotRank' in params['evalModeList']:
print("Performing AQMBotRank evaluation")
outputPredFile = "/home/lvxinyu/code/aqm_plus/resources/data/visdial/visdial/output_predictions_rollout.h5"
rankMetrics, roundRanks = AQMRunner(
aqmBot,
aBot,
dataset,
split,
beamSize=params['beamSize'],
realQA=params['aqmRealQA'],
saveLogs=params['saveLogs'],
showQA=params['showQA'],
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'],
selectedBatchIdxs=params['selectedBatchIdxs'],
numRounds=params['runRounds'],
lda=params['lambda'],
onlyGuesser=params['onlyGuesser'],
numQ=params['numQ'],
qbeamSize=params['qbeamSize'],
numImg=params['numImg'],
alpha=params['alpha'],
numA=params['numA'],
randQ=params['randQ'],
randA=params['randA'],
zeroCaption=params['zeroCaption'],
randomCaption=params['randomCaption'],
gamma=params['gamma'],
delta=params['delta'],
gen1Q=params['gen1Q'],
gtQ=params['gtQ'],
noHistory=params['noHistory'],
slGuesser=params['slGuesser'],
resampleEveryDialog=params['resampleEveryDialog'],
aqmSetting=aqmSetting,
).rankQuestioner()
for metric, value in rankMetrics.items():
plotName = splitName + ' - QABots Rank'
#viz.linePlot(iterId, value, plotName, metric, xlabel='Iterations')
for r in range(numRounds + 1):
for metric, value in roundRanks[r].items():
plotName = '[Iter %d] %s - QBot All Metrics vs Round'%\
(iterId, splitName)
#viz.linePlot(r, value, plotName, metric, xlabel='Round')
if 'dialog' in params['evalModeList']:
print("Performing dialog generation...")
split = 'test'
outputFolder = "/hhd/lvxinyu/aqm_plus/dialog_output/results"
os.makedirs(outputFolder, exist_ok=True)
outputPath = os.path.join(outputFolder, "results.json")
dialogDump(params,
dataset,
split,
aBot=aBot,
qBot=qBot,
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'],
beamSize=params['beamSize'],
savePath=outputPath)
if 'AQMdialog' in params['evalModeList']:
print("Performing AQM dialog generation...")
split = 'test'
AQMRunner(
aqmBot,
aBot,
dataset,
split,
beamSize=params['beamSize'],
realQA=params['aqmRealQA'],
saveLogs=params['saveLogs'],
showQA=params['showQA'],
expLowerLimit=params['expLowerLimit'],
expUpperLimit=params['expUpperLimit'],
selectedBatchIdxs=params['selectedBatchIdxs'],
numRounds=params['runRounds'],
lda=params['lambda'],
onlyGuesser=params['onlyGuesser'],
numQ=params['numQ'],
qbeamSize=params['qbeamSize'],
numImg=params['numImg'],
alpha=params['alpha'],
numA=params['numA'],
randQ=params['randQ'],
randA=params['randA'],
zeroCaption=params['zeroCaption'],
randomCaption=params['randomCaption'],
gamma=params['gamma'],
delta=params['delta'],
gen1Q=params['gen1Q'],
gtQ=params['gtQ'],
noHistory=params['noHistory'],
slGuesser=params['slGuesser'],
resampleEveryDialog=params['resampleEveryDialog'],
aqmSetting=aqmSetting,
).dialogDump(params)