wassname/iris_bigvae
1
0
Fork 0
mirror of https://github.com/wassname/iris_bigvae.git synced 2026-09-09 11:24:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

misc

Browse Source
This commit is contained in:
wassname
2023-11-19 14:50:54 +08:00
parent ecee11fc33
commit 6eb25883f6
4 changed files with 9 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/trainer.py
+1 -1
View File
@@ -200,7 +200,7 @@ class Trainer:
if epoch > cfg_world_model.start_after_epochs:
metrics_world_model = self.eval_component(self.agent.world_model, cfg_world_model.batch_num_samples, sequence_length=self.cfg.common.sequence_length, tokenizer=self.agent.tokenizer)
if epoch > cfg_actor_critic.start_after_epochs:
if epoch > cfg_world_model.start_after_epochs:
self.inspect_imagination(epoch)
if cfg_tokenizer.save_reconstructions: