Browse Source

use all workers for real ^^, cf #1

master
Olivier Courtin 4 months ago
parent
commit
ebe8aa5271
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      neat_eo/tools/rasterize.py

+ 2
- 2
neat_eo/tools/rasterize.py View File

@@ -206,11 +206,11 @@ def main(args):

log.log(
"neo rasterize - rasterizing {} from {} on cover {}, with {} tiles/batch and {} workers".format(
args.type, log_from, args.cover, workers, workers
args.type, log_from, args.cover, args.workers, args.workers
)
)
label_dataset = CreateLabelsDataset(tiles, feature_map, args, config, mode)
loader = DataLoader(label_dataset, batch_size=workers, num_workers=workers)
loader = DataLoader(label_dataset, batch_size=args.workers, num_workers=args.workers)
with open(os.path.join(os.path.expanduser(args.out), args.type.lower() + "_cover.csv"), mode="w") as cover:
for n, x, y, z, in tqdm(loader, desc="Rasterize", unit="batch", ascii=True):
for i in range(len(n)):


Loading…
Cancel
Save