Vivswan

import pickle
import sqlite3
import zlib
from typing import TypeVar, MutableMapping, Iterator

import numpy as np
from deduplicationdict import DeDuplicationDict
from sqlitedict import SqliteDict

KT = TypeVar('KT')  # Key type.

Vivswan

from timeit import default_timer as timer

import numpy as np
import torch
import torch.nn as nn
import torch.utils.benchmark as benchmark
from torch.profiler import profile, ProfilerActivity


@torch.no_grad()