You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Xuanlei Zhao 30dd13c450 [autochunk] support complete benchmark (#3121) ... * refact memory code * dont log free var memory * add memory align * update chunk target * update setting for new memory * finish test * update tracer * update typo * update test * add unet test * add bench * update bench * update bench * init * support vit * move to cpu * add cpu benchmark		2 years ago
..
benchmark_autochunk_alphafold.py	[autochunk] support complete benchmark (#3121)	2 years ago
test_autochunk_alphafold_utils.py	…
test_autochunk_evoformer_block.py	…
test_autochunk_evoformer_stack.py	…
test_autochunk_extramsa_block.py	…