YannAhlgrim/wilds-ijepa
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

higher ram size

Browse Source
This commit is contained in:
YannAhlgrim
2026-07-06 11:00:32 +02:00
parent e2b38eabbd
commit e2e558f168
3 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
main_distributed.py
+1 -1
View File
@@ -77,7 +77,7 @@ def launch():
max_num_timeout=20) max_num_timeout=20)
executor.update_parameters( executor.update_parameters(
partition=args.partition, partition=args.partition,
mem_per_gpu='55G', mem_per_gpu='256G',
time=args.time, time=args.time,
nodes=args.nodes, nodes=args.nodes,
ntasks_per_node=args.tasks_per_node, ntasks_per_node=args.tasks_per_node,
main_distributed_supervised.py
+1 -1
View File
@@ -77,7 +77,7 @@ def launch():
) )
executor.update_parameters( executor.update_parameters(
partition=args.partition, partition=args.partition,
mem_per_gpu="55G", mem_per_gpu="256G",
time=args.time, time=args.time,
nodes=args.nodes, nodes=args.nodes,
ntasks_per_node=args.tasks_per_node, ntasks_per_node=args.tasks_per_node,
tools/run_grid.py
+1 -1
View File
@@ -88,7 +88,7 @@ def main():
) )
executor.update_parameters( executor.update_parameters(
partition=args.partition or launch.get("partition"), partition=args.partition or launch.get("partition"),
mem_per_gpu=launch.get("mem_per_gpu", "55G"), mem_per_gpu=launch.get("mem_per_gpu", "256G"),
time=args.time or int(launch.get("time", 4300)), time=args.time or int(launch.get("time", 4300)),
nodes=args.nodes or int(launch.get("nodes", 1)), nodes=args.nodes or int(launch.get("nodes", 1)),
ntasks_per_node=args.tasks_per_node or int(launch.get("tasks_per_node", 1)), ntasks_per_node=args.tasks_per_node or int(launch.get("tasks_per_node", 1)),