search

mhamilton723 / STEGO

Unsupervised Semantic Segmentation by Distilling Feature Correspondences
MIT License
712 stars 143 forks source link

The number of epochs in train_segmentation #35

Open Benben0506 opened 2 years ago

Benben0506 commented 2 years ago

Hi,

Could you please tell me how to control the number of epochs? I did not see this hyperparameter in tran.yml.

pitlover commented 2 years ago

I have similar problems, too.

  1. Can you give me details of epochs or (iterations, batch size)?
  2. Also, authors said they use batch size of 32 in the paper, but in train_config.yml, batch size is denoted as 16.
  3. In train_config.yml, what is the different between "# cocostuff27 vit small 1/31/22" and "# cocostuff27 10/3 vit_small"? which is right?

Thanks!

Benben0506 commented 2 years ago

I think I know what is the epoch now. You see the maximum steps parameter in the train.yml file. That is the number of mini-batch the network will be trained. You just reduce that to let's say 1000, and the training process will stop fairly quickly.

pitlover commented 2 years ago

But it seems that code does not train even 1 epoch. The number of five crop dataset images is 488510.
5000 steps which is a num of optimizer steps (self.net + self.linear_probe + self.cluster_probe) is extremely less than 488510. update(5000 / 3) * batchsize(32) = 53000

I am curious how "5000" steps come from. There is no mention about epochs or iterations in paper.

Benben0506 commented 2 years ago

But I did not find the max_epochs in train_config.yml. Are you saying I should add one myself?

pitlover commented 2 years ago

Yes! Ex) trainer = Trainer (max_epochs = 10, ....)

Benben0506 commented 2 years ago

Thanks, I will try.

Benben0506 commented 2 years ago

You can manually install the package in anaconda.

gdghds0 commented 2 years ago

You can manually install the package in anaconda. I have also tried to install each package separately, but that still doesn't work. Whether through the official website or the 'conda' command, maybe I'm not familiar with this operation. Thank you for your reply.

KennyChen880127 commented 1 year ago

但仍然不起作用。不管是通過官網還是conda命令，可能我對這個操作不太熟悉。感謝你的

Hello sir, I'm spend a lot of time to set max_epochs by myself but all failed..Would you share your code where you change? thanks a lot!!!