Skip to content
Permalink
Browse files

Train 28D reranker

  • Loading branch information...
Kelym committed Aug 25, 2019
1 parent ea1834d commit 1832e771d917899a7b643716c8da5f06b055ef4a
Showing with 2 additions and 2 deletions.
  1. +2 −2 Training Reranker.ipynb
@@ -118,7 +118,7 @@
" _, world_states, actions, sum_logits, mean_logits, sum_logp, mean_logp, pm, speaker, scorer = candidate\n",
" new_data['candidates'].append(candidate)\n",
" new_data['candidates_path'].append([ws[1] for ws in world_states])\n",
" new_data['reranker_inputs'].append([len(world_states), sum_logits, mean_logits, sum_logp, mean_logp, pm, speaker])\n",
" new_data['reranker_inputs'].append([len(world_states), sum_logits, mean_logits, sum_logp, mean_logp, pm, speaker] * 4)\n",
" new_data['distance'].append(env.distances[scanId][world_states[-1][1]][new_data['goal_viewpointId']])\n",
" my_path = [ws[1] for ws in world_states]\n",
" if my_path == env.gt[path_id]['path']:\n",
@@ -147,7 +147,7 @@
"metadata": {},
"outputs": [],
"source": [
"net = Net(7).cuda()\n",
"net = Net(28).cuda()\n",
"#net.load_state_dict(torch.load('candidates_ranker_{}'.format(.6321)))"
]
},

0 comments on commit 1832e77

Please sign in to comment.
You can’t perform that action at this time.