NVIDIA / semantic-segmentation

Nvidia Semantic Segmentation monorepo
BSD 3-Clause "New" or "Revised" License
1.76k stars 388 forks source link

How to define optimizer other than SGD? Like Adam etc. #196

Open hamzagorgulu opened 10 months ago

hamzagorgulu commented 10 months ago

I wonder how can I use optimizer as my loss function. I change the optimizer type to adam but the adam optimizer requires beta values. When I define beta values as "betas", the runx gives error as: train.py: error: unrecognized arguments: --betas 0.9

Any help about this?