Spaces:
Sleeping
Sleeping
File size: 2,155 Bytes
abd15df |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 |
defaults:
- model: default
- computer: v100
- dataset: osv5m
- _self_
- exp: ???
model:
val_metrics:
_target_: metrics.distance_based.HaversineMetrics
acc_radiuses:
- 1
- 25
- 200
- 750
- 2500
acc_area: []
aux_data: ${aux_data}
test_metrics:
_target_: metrics.distance_based.HaversineMetrics
acc_radiuses:
- 1
- 25
- 200
- 750
- 2500
acc_area: ${areas}
aux_data: ${aux_data}
datamodule:
_target_: data.datamodule.ImageDataModule
train_dataset: ${dataset.train_dataset}
val_dataset: ${dataset.val_dataset}
test_dataset: ${dataset.test_dataset}
global_batch_size: ${dataset.global_batch_size}
num_workers: ${computer.num_workers}
num_nodes: ${computer.num_nodes}
num_devices: ${computer.devices}
val_proportion: 0.1
trainer:
_target_: pytorch_lightning.Trainer
devices: ${computer.devices}
accelerator: ${computer.accelerator}
strategy: ${computer.strategy}
num_nodes: ${computer.num_nodes}
precision: ${computer.precision}
max_epochs: ${max_epochs}
logger:
_target_: pytorch_lightning.loggers.WandbLogger
save_dir: ${root_dir}
name: ${experiment_name}
project: plonk
log_model: False
offline: False
entity: imaginelab
checkpoints:
_target_: pytorch_lightning.callbacks.ModelCheckpoint
dirpath: ${root_dir}/checkpoints/${experiment_name}
filename: 'epoch_{epoch}'
monitor: val/loss
save_last: True
save_top_k: 0
every_n_epochs: 1
progress_bar:
_target_: pytorch_lightning.callbacks.TQDMProgressBar
refresh_rate: ${computer.progress_bar_refresh_rate}
aux_data: []
max_epochs: 100
data_dir: ${root_dir}/datasets
root_dir: ${hydra:runtime.cwd}
experiment_name: ${dataset.name}__${model.name}
mode: train # change that to eval to do the testing
num_classes: 0
areas: ['country', 'region', 'sub-region', 'city']
class_name: null
streetclip: False
blur: False
text_tuning: False
hydra:
run:
dir: outputs/${hydra.job.name}/${now:%Y-%m-%d_%H-%M-%S}/${experiment_name}
job:
chdir: true
|