Print running average of MACs to personalise.

This commit is contained in:
Agnieszka Slowik 2023-12-12 17:23:17 +00:00
Родитель 10c0b835a9
Коммит 8ef732a28a
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -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",