MCPcopy Create free account
hub / github.com/THUDM/GLM / reset

Method reset

mpu/random.py:133–136  ·  view source on GitHub ↗

Set to the initial state (no tracker).

(self)

Source from the content-addressed store, hash-verified

131 self.seeds_ = set()
132
133 def reset(self):
134 """Set to the initial state (no tracker)."""
135 self.states_ = {}
136 self.seeds_ = set()
137
138 def get_states(self):
139 """Get rng states. Copy the dictionary so we have direct

Calls

no outgoing calls

Tested by 2

test_cuda_rng_trackerFunction · 0.36