error #5
Description
Dear Authors,
I would like to express my admiration for your excellent work. However, I encountered the following issues while attempting to retrain the model:
Traceback (most recent call last):
File "/home/zk/autofragdiff-main/train_frag_diffuser.py", line 236, in
main(args=args)
File "/home/zk/autofragdiff-main/train_frag_diffuser.py", line 69, in main
ddpm = AR_DDPM(
TypeError: AR_DDPM.init() got an unexpected keyword argument 'gaussian_expansion'
After commenting out these two arguments, another error occurred when the program executed the def validation_epoch_end(self, validation_step_outputs) function in the subsequent lightning.py file. The error indicates that the sample_and_analyze() function called within validation_epoch_end is undefined.