File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1- import copy
21import logging
32import os
43import threading
5- import time
6- import types
7- from typing import Dict
84
95import torch
106from torch import nn
117
12- import traincheck .config .config as general_config
13- import traincheck .proxy_wrapper .proxy_config as proxy_config # HACK: cannot directly import config variables as then they would be local variables
14- import traincheck .proxy_wrapper .proxy_methods as proxy_methods
158from traincheck .proxy_wrapper .dumper import dump_attributes , get_meta_vars
16- from traincheck .utils import get_timestamp_ns , typename
9+ from traincheck .utils import get_timestamp_ns
1710
1811from .dumper import json_dumper as dumper
19- from .proxy_basics import unproxy_arg , unproxy_args_kwargs
20- from .proxy_handler import PROXY_SUPPORT_OBJ_TYPES
2112
2213# from .proxy_registry import get_global_registry
23- from .utils import print_debug
14+ # from .utils import print_debug
2415
2516
2617def in_dynamo () -> bool :
You can’t perform that action at this time.
0 commit comments