Skip to content

Instantly share code, notes, and snippets.

View hadilou's full-sized avatar
💭
Definitely Future Optimistic

Broski hadilou

💭
Definitely Future Optimistic
View GitHub Profile
@hadilou
hadilou / LossEvalHook.py
Created December 8, 2020 08:40 — forked from ortegatron/LossEvalHook.py
Trainer with Loss on Validation for Detectron2
from detectron2.engine.hooks import HookBase
from detectron2.evaluation import inference_context
from detectron2.utils.logger import log_every_n_seconds
from detectron2.data import DatasetMapper, build_detection_test_loader
import detectron2.utils.comm as comm
import torch
import time
import datetime
class LossEvalHook(HookBase):