distillation

2022-02-23 15:39:51 +08:00
parent a0428c8c73
commit 1f2aa26bd1
2 changed files with 23 additions and 5 deletions
@@ -61,6 +61,7 @@ d_optim_config:
 id_weight: 20.0
 reconstruct_weight: 10.0
 feature_match_weight: 10.0
+distillation_weight: 10.0

 # Log 
 log_step: 300