ding.utils.fake_linklink¶
ding.utils.fake_linklink
¶
Full Source Code
../ding/utils/fake_linklink.py
1from collections import namedtuple 2 3 4class FakeClass: 5 """ 6 Overview: 7 Fake class. 8 """ 9 10 def __init__(self, *args, **kwargs): 11 pass 12 13 14class FakeNN: 15 """ 16 Overview: 17 Fake nn class. 18 """ 19 20 SyncBatchNorm2d = FakeClass 21 22 23class FakeLink: 24 """ 25 Overview: 26 Fake link class. 27 """ 28 29 nn = FakeNN() 30 syncbnVarMode_t = namedtuple("syncbnVarMode_t", "L2")(L2=None) 31 allreduceOp_t = namedtuple("allreduceOp_t", ['Sum', 'Max']) 32 33 34link = FakeLink()