[G1][Backbone] add CBAM and export scripts

This commit is contained in:
Zhang Wei committed 2025-12-18 15:40:26 +08:00
1 parent 85acef3a28
commit bc7d124645
17 files changed
+1044 -95

No files matched your search

+1 -1
View File
@@ -200,7 +200,7 @@ def train(hyp, opt, device, callbacks):
# Config
plots = not evolve and not opt.noplots # create plots
cuda = device.type != "cpu"
init_seeds(opt.seed + 1 + RANK, deterministic=True)
init_seeds(opt.seed + 1 + RANK, deterministic=False)
with torch_distributed_zero_first(LOCAL_RANK):
data_dict = data_dict or check_dataset(data) # check if None
train_path, val_path = data_dict["train"], data_dict["val"]