dfgffff

config.yaml

@@ -0,0 +1,66 @@
+global:
+  name: train-abinet-wo-iter
+  phase: train
+  stage: train-super
+  workdir: workdir
+  seed: ~
+ 
+dataset:
+  train: {
+    roots: [
+            'output_tbell_dataset2/',
+            # 'data/training/MJ/MJ_train/', 
+            # 'data/training/MJ/MJ_test/', 
+            # 'data/training/MJ/MJ_valid/', 
+            # 'data/training/ST'
+            ],
+    batch_size: 60
+  }
+  test: {
+    roots: ['output_tbell_dataset2/'],
+    batch_size: 60
+  }
+  data_aug: True
+  multiscales: False
+  num_workers: 60
+
+training:
+  epochs: 1000000
+  show_iters: 50
+  eval_iters: 250
+  # save_iters: 200
+
+optimizer:
+  type: Adam
+  true_wd: False
+  wd: 0.0
+  bn_wd: False
+  clip_grad: 20
+  lr: 0.0001
+  args: {
+    betas: !!python/tuple [0.9, 0.999], # for default Adam 
+  }
+  scheduler: {
+    periods: [6, 4],
+    gamma: 0.1,
+  }
+
+model:
+  name: 'modules.model_abinet.ABINetModel'
+  vision: {
+    checkpoint: workdir/pretrain-vision-model/best-pretrain-vision-model.pth,
+    loss_weight: 1.,
+    attention: 'position',
+    backbone: 'transformer',
+    backbone_ln: 3,
+  }
+  language: {
+    checkpoint:  workdir/pretrain-language-model/pretrain-language-model.pth,
+    num_layers: 4,
+    loss_weight: 1.,
+    detach: True,
+    use_self_attn: False
+  }
+  alignment: {
+    loss_weight: 1.,
+  }

lol.pth

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