microsoft / FocalNet

[NeurIPS 2022] Official code for "Focal Modulation Networks"
MIT License
690 stars 63 forks source link

How to modify in optimizer_config #39

Open svamsip opened 1 year ago

svamsip commented 1 year ago

Hi Team Great work and thanks for your contribution!

I'm working on training the mask_rcnn_focalnet_small_patch4_mstrain_480-800_adamw_3x_coco_lrf.py with custom dataset in coco format.

HostGuest commented 1 year ago

can u help me about the env plz ? which versions do u use , pytorch cuda, ..... i have errors seting up the env ? i have cuda 11.8 , ubunto 20

svamsip commented 1 year ago

@HostGuest I'm using pytorch 1.9.0; cuda11.1; cudnn8; for env setup you can follow the steps I've documented here INSTALL.md