For data custom dataset, Loss function does not converge.

[choibigo]

I succeeded in learning with MVTech data. After that, I want to learn with custom data. I created a file structure like MVTech to train custom data. I'm currently training wall images. I'm going to learn 500 images of the wall. python main_train.py --obj=wallcrack --lr=1e-4 --lambda_value=1e-3 --D=64 --epoch=500 used the command However, the Loss value does not converge as shown below.

I would appreciate it if you could give me a Hyperparameter or some tips.

The picture below is an example of the image I want to learn. Example)

[janerikvw]

In the paper they mention that a larger lambda is preferred for textures, whereas small lambda does well for objects. (Page 12 in the paper, under hyperparameters)