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

reduce ram

Browse Source
This commit is contained in:
YannAhlgrim
2026-07-06 11:10:45 +02:00
parent 04585f862f
commit 3e588c616c
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)
executor.update_parameters(
partition=args.partition,
mem_per_gpu='256G',
mem_per_gpu='180G',
time=args.time,
nodes=args.nodes,
ntasks_per_node=args.tasks_per_node,
main_distributed_supervised.py
+1 -1
View File
@@ -77,7 +77,7 @@ def launch():
)
executor.update_parameters(
partition=args.partition,
mem_per_gpu="256G",
mem_per_gpu="180G",
time=args.time,
nodes=args.nodes,
ntasks_per_node=args.tasks_per_node,
tools/run_grid.py
+1 -1
View File
@@ -88,7 +88,7 @@ def main():
)
executor.update_parameters(
partition=args.partition or launch.get("partition"),
mem_per_gpu=launch.get("mem_per_gpu", "256G"),
mem_per_gpu=launch.get("mem_per_gpu", "180G"),
time=args.time or int(launch.get("time", 4300)),
nodes=args.nodes or int(launch.get("nodes", 1)),
ntasks_per_node=args.tasks_per_node or int(launch.get("tasks_per_node", 1)),