Print running average of MACs to personalise.
This commit is contained in:
Родитель
10c0b835a9
Коммит
8ef732a28a
|
@ -257,7 +257,7 @@
|
||||||
" if (step+1) % data_queue.num_tasks == 0:\n",
|
" if (step+1) % data_queue.num_tasks == 0:\n",
|
||||||
" evaluator.set_current_user(task[\"task_id\"])\n",
|
" evaluator.set_current_user(task[\"task_id\"])\n",
|
||||||
" _,_,_,current_video_stats = evaluator.get_mean_stats(current_user=True)\n",
|
" _,_,_,current_video_stats = evaluator.get_mean_stats(current_user=True)\n",
|
||||||
" tqdm.write(f\"User {task['task_id']} ({evaluator.current_user+1}/{len(data_queue)}) {get_stats_str(current_video_stats)}, avg # context clips/task: {np.mean(num_context_clips_per_task):.0f}, avg # target clips/task: {np.mean(num_target_clips_per_task):.0f}\")\n",
|
" tqdm.write(f\"User {task['task_id']} ({evaluator.current_user+1}/{len(data_queue)}) {get_stats_str(current_video_stats)}, avg MACs to personalise / task: {np.mean(evaluator.macs_counter)}, avg # context clips/task: {np.mean(num_context_clips_per_task):.0f}, avg # target clips/task: {np.mean(num_target_clips_per_task):.0f}\")\n",
|
||||||
" if (step+1) < num_test_tasks:\n",
|
" if (step+1) < num_test_tasks:\n",
|
||||||
" num_context_clips_per_task = []\n",
|
" num_context_clips_per_task = []\n",
|
||||||
" num_target_clips_per_task = []\n",
|
" num_target_clips_per_task = []\n",
|
||||||
|
|
Загрузка…
Ссылка в новой задаче