Searched defs:init_model (Results 1 – 2 of 2) sorted by relevance
30 def init_model(): function
392 def init_model(self, device_type, model_parallel_size=2): member in Test2dFSDP1ParallelIntegration