While trying to reproduce the pre-training procedure, I run in the following error:
Traceback (most recent call last):
File "train.py", line 142, in <module>
train(cfg.num_epochs, trainer, args.log_dir, logger, model_save_freq=cfg.model_save_freq,
File "train.py", line 44, in train
avg_loss = trainer.train_epoch()
File "/home/via/python_projects/pretram/run_util/base_runner.py", line 33, in train_epoch
total_loss = self.compute_loss(output, data)
File "/home/via/python_projects/pretram/run_util/trainer.py", line 109, in compute_loss
traj_stateclass_logist = output['traj_stateclass_logist']
KeyError: 'traj_stateclass_logist'
From the looks of it, the 'traj_stateclass_logist' values are not computed anywhere inside the model. Am I doing something wrong? Thank you!
While trying to reproduce the pre-training procedure, I run in the following error:
From the looks of it, the 'traj_stateclass_logist' values are not computed anywhere inside the model. Am I doing something wrong? Thank you!