English
VLKEB-models / ke /minigpt4 /hparams.yaml
HymanH's picture
Upload 13 files
3985c47 verified
raw
history blame
1.76 kB
accelerator: null
accumulate_grad_batches: 1
amp_backend: native
amp_level: O2
auto_lr_find: false
auto_scale_batch_size: false
auto_select_gpus: false
automatic_optimization: null
batch_size: 1
benchmark: false
check_val_every_n_epoch: 1
checkpoint_callback: true
default_root_dir: null
deterministic: false
dev_data_path: /data0/huanghan/datasets/mmkb/mmkb_edits_eval_alter_entity.json
dirpath: models/
distributed_backend: null
divergences: lp
enable_pl_optimizer: null
eps: 0.1
fast_dev_run: false
flush_logs_every_n_steps: 100
gpus: 1
gradient_clip_val: 0
limit_predict_batches: 1.0
limit_test_batches: 1.0
limit_train_batches: 1.0
limit_val_batches: 1.0
log_every_n_steps: 50
log_gpu_memory: null
logger: true
lr: 0.0003
lr_alpha: 0.1
margin_lp_max: 0.001
margin_lp_min: 1.0e-07
max_epochs: null
max_length: 32
max_scale: 1
max_steps: 30000
min_epochs: null
min_steps: null
model_checkpoint: ''
model_name: minigpt4
move_metrics_to_cpu: false
multiple_trainloader_mode: max_size_cycle
num_nodes: 1
num_processes: 1
num_sanity_val_steps: 2
num_workers: 0
overfit_batches: 0.0
p: 2
plugins: null
precision: 32
prepare_data_per_node: true
process_position: 0
profiler: null
progress_bar_refresh_rate: null
reload_dataloaders_every_epoch: false
replace_sampler_ddp: true
resume_from_checkpoint: null
save_top_k: 2
seed: 0
stochastic_weight_avg: false
sync_batchnorm: false
terminate_on_nan: false
total_num_updates: 30000
tpu_cores: !!python/name:pytorch_lightning.utilities.argparse._gpus_arg_default ''
track_grad_norm: -1
train_data_path: /data0/huanghan/datasets/mmkb/mmkb_edits_train_alter_entity.json
truncated_bptt_steps: null
use_views: false
val_check_interval: 1.0
warmup_updates: 1000
weight_decay: 0.01
weights_save_path: null
weights_summary: top