update env

yiren-jian · Dec 5, 2023 · 7e633e0 · 7e633e0
1 parent 246a832
commit 7e633e0
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 6 deletions.
diff --git a/lavis/projects/blip-T/train/pretrain_stage1.yaml b/lavis/projects/blip-T/train/pretrain_stage1.yaml
@@ -8,8 +8,8 @@ model:
  model_type: pretrain_pformer
  # TODO: support stage 1 pretraining from scratch (load_pretrained=False does not have effect as of now)
  load_pretrained: True
- pretrained: "/home/yiren/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/vq/12m-5/pformer.pth"
- pretrained_stage0: "/home/yiren/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/vq/12m-5/pformer.pth"
+ pretrained: "/home/yiren/new_ssd2/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/pformer.pth"
+ pretrained_stage0: "/home/yiren/new_ssd2/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/pformer.pth"
  freeze_vit: True
  loss_weight: 10.0
 

diff --git a/lavis/projects/blip-T/train/pretrain_stage2.yaml b/lavis/projects/blip-T/train/pretrain_stage2.yaml
@@ -10,8 +10,8 @@ model:
  # model_type: pretrain_pformer_opt2.7b
  load_pretrained: True
  # intialize stage 2 pretraining from stage 1 pretrained model
- pretrained: "/home/yiren/LAVIS/lavis/output/BLIP-T/Pretrain_stage1/20230418231/checkpoint_9.pth"
- pretrained_stage0: "/home/yiren/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/vq/12m-5/pformer.pth"
+ pretrained: "/home/yiren/new_ssd2/LAVIS/lavis/output/BLIP-T/Pretrain_stage1/20230418231/checkpoint_9.pth"
+ pretrained_stage0: "/home/yiren/new_ssd2/LAVIS/lavis/output/BLIP-T/Pretrain_stage0/pformer.pth"
  freeze_vit: True
  loss_weight: 100.0
 

diff --git a/training_logs/pip_freeze.txt b/training_logs/pip_freeze.txt
@@ -1,4 +1,3 @@
-(lavis) yiren@mms-large-2:~/LAVIS$ pip freeze
 altair==4.2.2
 antlr4-python3-runtime==4.9.3
 asttokens==2.2.1
@@ -138,4 +137,3 @@ watchdog==2.2.1
 wcwidth==0.2.6
 webdataset==0.2.33
 zipp==3.14.0
-(lavis) yiren@mms-large-2:~/LAVIS$