safermi

I did these modifications, but it still does not work and gives me this error:C:\Users\sl\Desktop\NRI-master>python train.py Namespace(no_cuda=False, seed=42, epochs=500, batch_size=128, lr=0.0005, encoder_hidden=256, decoder_hidden=256, temp=0.5, num_atoms=5, encoder='mlp', decoder='mlp', no_factor=False, suffix='_springs5', encoder_dropout=0.0,...