Function get_stats

graphgym/utils/comp_budget.py:18–20 · view source on GitHub ↗

()

Source from the content-addressed store, hash-verified

16
17
18	def get_stats():
19	model = create_model(to_device=False, dim_in=1, dim_out=1)
20	return params_count(model)
21
22
23	def match_computation(stats_baseline, key=['gnn', 'dim_inner'], mode='sqrt'):

match_computationFunction · 0.85

dict_to_statsFunction · 0.85

create_modelFunction · 0.90

params_countFunction · 0.85

no test coverage detected