Anchor-x/fed-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

移除训练过程中的调试打印语句

Browse Source
This commit is contained in:
Yunhao Meng
2025-10-23 13:06:13 +08:00
parent 76d7149512
commit 52382e460d
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
fed_algo_cs/client_base.py
View File

@@ -120,7 +120,7 @@ class FedYoloClient(object):
# track_model = self.model.module if is_ddp else self.model # track_model = self.model.module if is_ddp else self.model
ema = util.EMA(self.model) if args.local_rank == 0 else None ema = util.EMA(self.model) if args.local_rank == 0 else None
print(type(self.train_dataset)) # print(type(self.train_dataset))
# ---- Data ---- # ---- Data ----
dataset = Dataset( dataset = Dataset(
@@ -188,7 +188,7 @@ class FedYoloClient(object):
loss_dfl_meter = util.AverageMeter() loss_dfl_meter = util.AverageMeter()
for i, (images, targets) in enumerate(loader): for i, (images, targets) in enumerate(loader):
print(f"Client {self.name} - Epoch {epoch + 1}/{args.epochs} - Step {i + 1}/{num_steps}") # print(f"Client {self.name} - Epoch {epoch + 1}/{args.epochs} - Step {i + 1}/{num_steps}")
step = i + epoch * num_steps step = i + epoch * num_steps
# scheduler per-step (your util.LinearLR expects step) # scheduler per-step (your util.LinearLR expects step)
@@ -257,9 +257,9 @@ class FedYoloClient(object):
else self.model else self.model
) )
# print loss to test # print loss to test
print( # print(
f"loss: {total_loss.item() * accumulate:.4f}, box: {box_loss.item():.4f}, cls: {cls_loss.item():.4f}, dfl: {dfl_loss.item():.4f}" # f"loss: {total_loss.item() * accumulate:.4f}, box: {box_loss.item():.4f}, cls: {cls_loss.item():.4f}, dfl: {dfl_loss.item():.4f}"
) # )
torch.cuda.synchronize() torch.cuda.synchronize()
# ---- Final average loss (per image) over the whole epoch span ---- # ---- Final average loss (per image) over the whole epoch span ----