import os
import frcnn_utils
import init_frcnn_utils
from experiments.exp_utils import get_config_var, LoggerForSacred, Args
from init_frcnn_utils import init_dataloaders_1s_1t, init_val_dataloaders_mt, init_val_dataloaders_1t, \
init_htcn_model_optimizer
from sacred import Experiment
ex = Experiment()
from sacred.observers import MongoObserver
enable_mongo_observer = False
if enable_mongo_observer:
vars = get_config_var()
ex.observers.append(
MongoObserver(
url='mongodb://{}:{}@{}/admin?authMechanism=SCRAM-SHA-1'.format(
vars["SACRED_USER"], vars["SACRED_PWD"], vars["SACRED_URL"]),
db_name=vars["SACRED_DB"]))
ex.captured_out_filter = lambda text: 'Output capturing turned off.'
from dataclasses import dataclass
import numpy as np
import torch
import torch.nn as nn
from model.utils.config import cfg, cfg_from_file, cfg_from_list
from model.utils.net_utils import adjust_learning_rate, save_checkpoint, FocalLoss, EFocalLoss
import os
import frcnn_utils
import distill_frcnn_utils
from experiments.exp_utils import get_config_var, LoggerForSacred, Args
import init_frcnn_utils
from sacred import Experiment
ex = Experiment()
from sacred.observers import MongoObserver
enable_mongo_observer = False
if enable_mongo_observer:
vars = get_config_var(is_eval=True)
ex.observers.append(MongoObserver(url='mongodb://{}:{}@{}/admin?authMechanism=SCRAM-SHA-1'.format(vars["SACRED_USER"],
vars["SACRED_PWD"],
vars["SACRED_URL"]),
db_name=vars["SACRED_DB"]))
ex.captured_out_filter = lambda text: 'Output capturing turned off.'
from dataclasses import dataclass
import numpy as np
import torch
import torch.nn as nn
from model.utils.config import cfg, cfg_from_file, cfg_from_list
from model.utils.net_utils import adjust_learning_rate, save_checkpoint, FocalLoss, EFocalLoss
from model.faster_rcnn.resnet import resnet as n_resnet