github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-03 01:19:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename

Browse Source
This commit is contained in:
oahzxl
2023-01-06 17:09:37 +08:00
parent c3d72f7db9
commit da4076846d
6 changed files with 19 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_autochunk/benchmark_autochunk.py
View File

@@ -104,7 +104,7 @@ def benchmark_evoformer():
model = evoformer_base().cuda()
# build autochunk model
# max_memory = 10000 # MB fit memory mode
# max_memory = 1000 # MB fit memory mode
max_memory = None # min memory mode
autochunk = _build_autochunk(evoformer_base().cuda(), max_memory, node, pair)