Upload 2 files

config.yaml

@@ -0,0 +1,66 @@
+mode: training
+job_name: bart_large
+job_dir: /poster/jobs/bart_large
+seed: 1
+cuda_visible_gpu: '0'
+ckpt_dir: null
+checkpoint_name: best
+debug: false
+use_wandb: false
+wandb_project_name: SciPosterLayout
+write_preidiction_epoch: 10
+backbone:
+  _target_: trainer.models.transformer_utils.TransformerEncoder
+  encoder_layer:
+    _target_: trainer.models.transformer_utils.Block
+    d_model: 512
+    nhead: 8
+    dim_feedforward: 2048
+    dropout: 0.1
+    batch_first: true
+    norm_first: true
+  num_layers: 12
+dataset:
+  _target_: trainer.datasets.posterGlobal.posterGlobalDataset
+  _partial_: true
+  dir: /poster/data
+  max_seq_length: 10
+model:
+  _target_: trainer.models.bart.BART
+  _partial_: true
+extra_model_params:
+  use_size_position_decoders: false
+  category_loss: true
+  multi_decoder: false
+data:
+  batch_size: 16
+  num_workers: 4
+  num_bin_bboxes: 32
+  bbox_quantization: linear
+  shared_bbox_vocab: xywh
+  pad_until_max: true
+  special_tokens:
+  - pad
+  - bos
+  - eos
+  - mask
+  transforms: []
+  var_order: c-w-h-x-y
+training:
+  epochs: 3000
+  grad_norm_clip: 1.0
+  weight_decay: 0.1
+  loss_plot_iter_interval: 50
+optimizer:
+  _target_: torch.optim.AdamW
+  _partial_: true
+  lr: 0.001
+  betas:
+  - 0.9
+  - 0.98
+scheduler:
+  _target_: trainer.helpers.scheduler.VoidScheduler
+  _partial_: true
+sampling:
+  name: random
+  temperature: 1.0

pytorch_model.bin

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:165bbb78d802f66dd612bbcd36cf945806a6a90f13bfc0af603712f08f9724c4
+size 416938572