Linear Image classification weights not loading properly: vissl pretrained models simclr or Dcv2

[DC95]

Linear Image classification weights not loading properly: vissl pretrained models simclr or Dcv2

[QuentinDuval]

Hi @DC95,

First of all, sorry for my slow response time, I had a lot on my plate.
Now it's better :) I will have a look at it !

Could you provide me with:

• the command line you are using?
• the dictionary structure of the checkpoint you are trying to evaluate?

For the second point, what I mean by dictionary structure is something like this:

{
  "mode": {
    "state_dict": [name of the weights]
  }
}

Or if you have a public checkpoint that I can use, I can try that as well !

Thank you,
Quentin

[DC95]

Hi @QuentinDuval

Thanks for your response :)

The link for DC_v2 checkpoint is (https://gigamove.rwth-aachen.de/en/download/859065e3c72eabd19c45578abfd17ba0)

The link for simclr checkpoint is (https://gigamove.rwth-aachen.de/en/download/cb9771810632728d4f5885f72ac4e551)

The checkpoint structure looks like this:

phase_idx iteration loss iteration_num train_phase_idx classy_state_dict type

The classy_state_dict looks like this:
dict_keys(['train', 'base_model', 'meters', 'optimizer', 'phase_idx', 'train_phase_idx', 'num_updates', 'losses', 'hooks', 'loss', 'train_dataset_iterator', 'amp'])

The trunk keys looks like this : '_feature_blocks.conv1.weight', '_feature_blocks.bn1.weight'

The command line: python run_distributed_engines.py \config=pretrain/linear_classifier/linear_classifier_k7_g128

Thanks for taking time out,
DC

[DC95]

Hi @QuentinDuval

Wondering if it was in the right direction. Or was I doing something wrong?