Giters

justchenhao / STANet

official implementation of the spatial-temporal attention neural network (STANet) for remote sensing image change detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to train with my own dataset?

Daisy-7 opened this issue · comments

commented

Thanks for your work.
I want to use your model for training, and at the same time change the learning rate drop method and data augmentation method during training. What should I do?

commented

Thanks for your attention. You can modify the optimizer in the model file and the augmentation functions in base_dataset.py.

commented

Close for now.