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 copy
import math import math
import os.path as osp import os.path as osp
import torch
from fnmatch import fnmatch from fnmatch import fnmatch
from typing import Dict, List, Optional, Tuple, Union from typing import Dict, List, Optional, Tuple, Union