626_privacy/mia_on_model_distillation
2
0
Fork 1
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

O1: fix epochs

Browse source
This commit is contained in:
Akemi Izuko 2024-12-06 17:12:38 -07:00
parent ebfbd88332
commit 86d16e53d7
Signed by: akemi
GPG key ID: 8DE0764E1809E9FC
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
one_run_audit/audit.py
View file

@ -545,7 +545,7 @@ def main():
model_init, model_trained = train_knowledge_distillation( model_init, model_trained = train_knowledge_distillation(
teacher=teacher_trained, teacher=teacher_trained,
train_dl=train_dl, train_dl=train_dl,
epochs=100, epochs=hp['epochs'],
device=DEVICE, device=DEVICE,
learning_rate=0.001, learning_rate=0.001,
T=2, T=2,