Skip to content

Instantly share code, notes, and snippets.

View hikmatkhan's full-sized avatar
💭
Always stay foolish and hungry 🍔.

Hikmat Khan hikmatkhan

💭
Always stay foolish and hungry 🍔.
  • New York, United States
View GitHub Profile
@hikmatkhan
hikmatkhan / LossEvalHook.py
Created April 18, 2022 13:57 — 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):
@hikmatkhan
hikmatkhan / LossEvalHook.py
Created April 18, 2022 13:57 — 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):