saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-30 20:18:15 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add tqdm when using CQL

Browse source
This commit is contained in:
Antonin RAFFIN 2020-08-29 20:01:57 +02:00
parent 66f900ead9
commit 73a6d5117e
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
stable_baselines3/sac/sac.py
View file

@ -282,7 +282,9 @@ class SAC(OffPolicyAlgorithm):
ent_coef_losses, ent_coefs = [], [] ent_coef_losses, ent_coefs = [], []
actor_losses, critic_losses = [], [] actor_losses, critic_losses = [], []
for gradient_step in range(gradient_steps): range_ = tqdm(range(gradient_steps)) if self.use_cql else range(gradient_steps)
for gradient_step in range_:
# Sample replay buffer # Sample replay buffer
replay_data = self.replay_buffer.sample(batch_size, env=self._vec_normalize_env) replay_data = self.replay_buffer.sample(batch_size, env=self._vec_normalize_env)