This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
""" MP_TQDM v4.1 by twmicrosheep | |
This module is a wrapper for easy multiprocessing with tqdm | |
""" | |
import math | |
import itertools | |
import multiprocessing | |
from functools import wraps | |
from typing import List, Dict, Any, Callable, Iterable, Optional |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from datetime import datetime | |
from dataclasses import dataclass, field, asdict | |
from typing import Callable, Optional, Dict, Any | |
import torch | |
def transform_dict(config_dict: Dict, expand: bool = True): | |
""" | |
General function to transform any dictionary into wandb config acceptable format |