AssertionError

[Samridhi Vaid]

@guxd Hi. I am trying to run the Keras version with 2 epochs and the sample dataset. I am able to train the model. But when I running

!python main.py --mode repr_code

I get the below error. Can you please help with this

2021-10-22 21:05:30,164: models: INFO: compiling models
Traceback (most recent call last):
File "main.py", line 272, in
engine.load_model(model, config['training_params']['reload'])
File "main.py", line 42, in load_model
assert os.path.exists(model_path + f"epo{epoch}_code.h5"),f"Weights at epoch {epoch} not found"
AssertionError: Weights at epoch 2 not found

For this code !python main.py --mode repr_code I am changing the reload in the config.py file to 1. For training the model the reload value is -1