Testing code and wandb integration

[shubhraneel]

python run_attacker.py \
  --train_input_dir data/ \
  --train_logits_file logits.pkl \
  --test_input_dir test_data/ \
  --test_logits_file test_logits.pkl \
  --attacker_model_name resnet-lstm \
  --victim_model_name swin-t \
  --epochs 20 \
  --resnet_lstm_trainable_layers 1 \
  --learning_rate 1e-3 \
  --load_test_from_disk

[KaLiMaLi555]

@shubhraneel Can you add a section in the README.md for cnn_lstm and paste the command over there?

[shubhraneel]

@shubhraneel Can you add a section in the README.md for cnn_lstm and paste the command over there?

done

[KaLiMaLi555]

The get_embeddings.py is specific to cnn_lstm, right? If yes, can you just rename it to get_embeddings_cnn_lstm.py. Since there is another get_embeddings file from another PR. @shubhraneel

[shubhraneel]

get_embeddings.py is the same as the one @Sohanpatnaik106 used, except I have just put the list of logits to be stored in the pickle instead of the zip of logits and labels in the pickle (since zips can be read only once)

Even for the purpose that Sohan was using, I discussed with him that we will be using list after this.