Act temporal ensembling (#186)

This commit is contained in:
Alexander Soare
2024-05-17 14:57:49 +01:00
committed by GitHub
parent 4d7d41cdee
commit 625f0557ef
3 changed files with 39 additions and 8 deletions

View File

@@ -73,7 +73,7 @@ policy:
n_vae_encoder_layers: 4
# Inference.
use_temporal_aggregation: false
temporal_ensemble_momentum: null
# Training and loss computation.
dropout: 0.1