[G1][Backbone] add CBAM and export scripts
This commit is contained in:
1 parent
85acef3a28
commit
bc7d124645
17 files changed
+1044
-95
No files matched your search
@@ -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"]
|
||||
|
||||
Reference in new issue
Block a user