vision
classification
uncertainty
pe_resnet50_in1k / config.yaml
o-laurent's picture
Add checkpoint :sparkles:
7c063d0
raw
history blame
1.43 kB
seed: null
multi_gpu: true
logger: true
checkpoint_callback: null
enable_checkpointing: true
default_root_dir: null
gradient_clip_val: null
gradient_clip_algorithm: null
process_position: 0
num_nodes: 1
num_processes: null
devices: null
gpus: -1
auto_select_gpus: false
tpu_cores: null
ipus: null
log_gpu_memory: null
progress_bar_refresh_rate: null
enable_progress_bar: true
overfit_batches: 0.0
track_grad_norm: -1
check_val_every_n_epoch: 1
fast_dev_run: false
accumulate_grad_batches: null
max_epochs: 105
min_epochs: null
max_steps: -1
min_steps: null
max_time: null
limit_train_batches: null
limit_val_batches: null
limit_test_batches: null
limit_predict_batches: null
val_check_interval: null
flush_logs_every_n_steps: null
log_every_n_steps: 50
accelerator: gpu
strategy: "A3"
sync_batchnorm: true
precision: 16
enable_model_summary: true
weights_summary: top
weights_save_path: null
num_sanity_val_steps: 2
resume_from_checkpoint: null
profiler: null
benchmark: true
deterministic: false
reload_dataloaders_every_n_epochs: 0
auto_lr_find: false
replace_sampler_ddp: true
detect_anomaly: false
auto_scale_batch_size: false
prepare_data_per_node: null
plugins: null
amp_backend: native
amp_level: null
move_metrics_to_cpu: false
multiple_trainloader_mode: max_size_cycle
stochastic_weight_avg: false
terminate_on_nan: null
batch_size: 1024
val_split: 0
num_workers: 12
num_estimators: 4
augmentation: 3
gamma: 1
resnet: '50'