зеркало из https://github.com/microsoft/torchgeo.git
Fix more issues with integration tests
This commit is contained in:
Родитель
ca7d48b439
Коммит
3bed0cc015
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 10
|
||||
max_epochs: 40
|
||||
benchmark: True
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 20
|
||||
max_epochs: 100
|
||||
benchmark: True
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 20
|
||||
max_epochs: 100
|
||||
benchmark: True
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 15
|
||||
experiment:
|
||||
task: cowc_counting
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 15
|
||||
experiment:
|
||||
task: "cyclone"
|
||||
|
|
|
@ -1,9 +1,9 @@
|
|||
program:
|
||||
overwrite: True
|
||||
|
||||
|
||||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 5
|
||||
max_epochs: 100
|
||||
benchmark: True
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 20
|
||||
max_epochs: 100
|
||||
benchmark: True
|
||||
|
|
|
@ -2,7 +2,8 @@ program:
|
|||
overwrite: True
|
||||
|
||||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 5
|
||||
max_epochs: 100
|
||||
auto_lr_find: False
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 10
|
||||
max_epochs: 40
|
||||
benchmark: True
|
||||
|
|
|
@ -1,5 +1,6 @@
|
|||
trainer:
|
||||
gpus: 1
|
||||
accelerator: gpu
|
||||
devices: 1
|
||||
min_epochs: 10
|
||||
max_epochs: 40
|
||||
benchmark: True
|
||||
|
|
|
@ -1,7 +1,8 @@
|
|||
program:
|
||||
overwrite: False
|
||||
trainer:
|
||||
gpus: [3]
|
||||
accelerator: gpu
|
||||
devices: 3
|
||||
min_epochs: 50
|
||||
max_epochs: 200
|
||||
benchmark: True
|
||||
|
|
|
@ -93,6 +93,8 @@ filterwarnings = [
|
|||
# https://github.com/scikit-image/scikit-image/issues/6663
|
||||
# https://github.com/scikit-image/scikit-image/pull/6637
|
||||
"ignore:`np.bool8` is a deprecated alias for `np.bool_`.:DeprecationWarning:skimage.util.dtype",
|
||||
# https://github.com/lanpa/tensorboardX/pull/677
|
||||
"ignore:ANTIALIAS is deprecated and will be removed in Pillow 10:DeprecationWarning:tensorboardX.summary",
|
||||
|
||||
# Expected warnings
|
||||
# pytorch-lightning warns us about using num_workers=0, but it's faster on macOS
|
||||
|
|
|
@ -72,7 +72,7 @@ def test_overwrite_experiment_dir(tmp_path: Path) -> None:
|
|||
"experiment.datamodule.root=" + data_dir,
|
||||
"program.overwrite=True",
|
||||
"trainer.fast_dev_run=1",
|
||||
"trainer.gpus=0",
|
||||
"trainer.accelerator=cpu",
|
||||
]
|
||||
ps = subprocess.run(args, capture_output=True, check=True)
|
||||
assert re.search(
|
||||
|
@ -129,7 +129,7 @@ experiment:
|
|||
root: {data_dir}
|
||||
trainer:
|
||||
fast_dev_run: true
|
||||
gpus: 0
|
||||
accelerator: cpu
|
||||
"""
|
||||
)
|
||||
args = [sys.executable, "train.py", "config_file=" + str(config_file)]
|
||||
|
|
Загрузка…
Ссылка в новой задаче