jiangzhs/OpenCompass
1
0
Fork 0
mirror of https://github.com/open-compass/opencompass.git synced 2025-05-30 16:03:24 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

change max_task_size to dynamic value

Browse Source
This commit is contained in:
disperaller 2024-05-22 13:55:05 +08:00
parent db8d9a9798
commit 78e89ce8b5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
opencompass/partitioners/size.py
View File

@ -1,6 +1,7 @@
import copy
import math
import os.path as osp
import torch
from fnmatch import fnmatch
from typing import Dict, List, Optional, Tuple, Union