Skip to content
This repository has been archived by the owner on Oct 31, 2023. It is now read-only.

Path setting at the training step2 #62

Open
LeeJunHyun opened this issue Mar 9, 2021 · 0 comments
Open

Path setting at the training step2 #62

LeeJunHyun opened this issue Mar 9, 2021 · 0 comments

Comments

@LeeJunHyun
Copy link

LeeJunHyun commented Mar 9, 2021

Hi, I really appreciate your commitment for this remarkable work.

I have a little difficulty to set up LFB path at training.

According to GETTING_STARTED.md, there are two steps for training the model over Charades dataset.

Stage 1:

python2 tools/train_net.py \
  --config_file configs/charades_r50_baseline.yaml \
  CHECKPOINT.DIR /tmp/lfb-nl-stage1-output

Stage 2:

python2 tools/train_net.py \
  --config_file configs/charades_r50_lfb_nl.yaml \
  TRAIN.PARAMS_FILE [path to model weight from Stage 1] \
  LFB.MODEL_PARAMS_FILE [path to model weight for LFB inference] \
  LFB.WRITE_LFB True \
  CHECKPOINT.DIR /tmp/lfb-nl-stage2-output

After Stage 1, I got some files under CHECKPOINT.DIR (/tmp/lfb-nl-stage1-output)

├── checkpoints
│   ├── c2_model_iter12000.pkl
│   ├── c2_model_iter16000.pkl
│   ├── c2_model_iter24000.pkl
│   ├── c2_model_iter4000.pkl
│   ├── c2_model_iter8000.pkl
│   └── converted_model.pkl
├── CNN_1.pbtxt
├── CNN_init_1.pbtxt
├── CNN_init.pbtxt
└── CNN.pbtxt

In here, which path can be proper for TRAIN.PARAMS_FILE [path to model weight from Stage 1] and LFB.MODEL_PARAMS_FILE [path to model weight for LFB inference] at Stage 2?
I thought c2_model_iter24000.pkl can be assigned as TRAIN.PARAMS_FILE, but there was an error (KeyError: u'nonlocal_conv3_1_bn_rm' at remove_spatial_bn_layers(c2cls_weights)) and not sure about LFB.MODEL_PARAMS_FILE.
Could you please give me some specific example filenames for both *PARAMS_FILEs?

Thanks,
Junhyun Lee

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant