Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion examples/multimodal/evaluation/evaluate_mmmu.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,4 @@
# Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
import argparse
import glob
import json
Expand Down Expand Up @@ -104,7 +105,7 @@ def mmmu_eval(input_path, groundtruth_path):
print(output.stderr)
print(output.stdout)

m = re.search("'Overall': {'num': \d+, 'acc': (\d.\d+)}", output.stdout)
m = re.search(r"'Overall': {'num': \d+, 'acc': (\d\.\d+)}", output.stdout)

return float(m.group(1)) * 100.0

Expand Down
6 changes: 3 additions & 3 deletions megatron/legacy/model/vision/esvit_swin_backbone.py
Original file line number Diff line number Diff line change
Expand Up @@ -726,7 +726,7 @@ def init_weights(self, pretrained='', pretrained_layers=[], verbose=True):
for k, v in pretrained_dict.items():
need_init = (
k.split('.')[0] in pretrained_layers
or pretrained_layers[0] is '*'
or pretrained_layers[0] == '*'
or 'relative_position_index' not in k
or 'attn_mask' not in k
Comment thread
HollowMan6 marked this conversation as resolved.
)
Expand Down Expand Up @@ -785,7 +785,7 @@ def freeze_pretrained_layers(self, frozen_layers=[]):
if (
name.split('.')[0] in frozen_layers
or '.'.join(name.split('.')[0:2]) in frozen_layers
or (len(frozen_layers) > 0 and frozen_layers[0] is '*')
or (len(frozen_layers) > 0 and frozen_layers[0] == '*')
):
for _name, param in module.named_parameters():
param.requires_grad = False
Expand All @@ -796,7 +796,7 @@ def freeze_pretrained_layers(self, frozen_layers=[]):
for name, param in self.named_parameters():
if (
name.split('.')[0] in frozen_layers
or (len(frozen_layers) > 0 and frozen_layers[0] is '*')
or (len(frozen_layers) > 0 and frozen_layers[0] == '*')
and param.requires_grad is True
):
Comment thread
HollowMan6 marked this conversation as resolved.
param.requires_grad = False
Expand Down
4 changes: 2 additions & 2 deletions tests/test_utils/python_scripts/launch_nemo_run_workload.py
Original file line number Diff line number Diff line change
Expand Up @@ -140,10 +140,10 @@ def main(
succeeded = str(job_dict["status"]) == "SUCCEEDED"

if succeeded:
logger.info(f"Job succeeded with status: {job_dict["status"]}")
logger.info(f"Job succeeded with status: {job_dict['status']}")
sys.exit(0)

logger.error(f"Job failed with status: {job_dict["status"]}")
logger.error(f"Job failed with status: {job_dict['status']}")
log_file_paths = pathlib.Path(os.getcwd()).glob("assets_dir/logs/*/*/attempt_0/*/std*.log")
all_ranks_all_logs = []
for log_file_path in log_file_paths:
Expand Down
Loading