saving results

alicebizeul · Apr 16, 2023 · 8392ba2 · 8392ba2
1 parent d61782b
commit 8392ba2
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 0 deletions.
diff --git a/bash_all.sh b/bash_all.sh
@@ -110,6 +110,22 @@ python main.py \
  --learn_std=${LEARNSTD} \
  --std_epochs=${LSTD_EPOCHS} \
  --epochs=${EPOCHS} \
+ --epoch_num=${EPOCHS} \
+ --classes=${CLASSES} \
+ --normalize=${NORM} \
+ --lr_change=${LR_CHANGE} \
+ --lr=${LR} \
+ --bound=${BOUND} \
+ --subsample=${SUBSAMPLE} \
+ --model_path=${MODEL_PATH}
+
+
+python linear_evaluation.py \
+ --custom=True \
+ --learn_std=${LEARNSTD} \
+ --std_epochs=${LSTD_EPOCHS} \
+ --epochs=${EPOCHS} \
+ --epoch_num=${EPOCHS} \
  --classes=${CLASSES} \
  --normalize=${NORM} \
  --lr_change=${LR_CHANGE} \

diff --git a/linear_evaluation.py b/linear_evaluation.py
@@ -207,3 +207,7 @@ def test(args, loader, simclr_model, model, criterion, optimizer):
  print(
  f"[FINAL]\t Loss: {loss_epoch / len(arr_test_loader)}\t Accuracy: {accuracy_epoch / len(arr_test_loader)}"
  )
+
+ with open(args.model_path+'/accuracy.npy', 'wb') as f:
+ np.save(f, np.array([accuracy_epoch / len(arr_test_loader)]))
+