search

ViTAE-Transformer / SAMRS

The official repo for [NeurIPS'23] "SAMRS: Scaling-up Remote Sensing Segmentation Dataset with Segment Anything Model"
288 stars 14 forks source link

环境配置 #11

Open 15926273249 opened 11 months ago

15926273249 commented 11 months ago

![Uploading 1.png…]() cuda12.2 在conda里面配置相关环境一直不成功，需要对应什么版本呀

DotWang commented 11 months ago

@15926273249 我这边cuda11.7可以

15926273249 commented 11 months ago

finetune的配置里面backbone好像不包含vit_b

DotWang commented 11 months ago

@15926273249 包含啊

https://github.com/ViTAE-Transformer/SAMRS/blob/main/Pretraining%20and%20Finetuning/Encoder_Decoder/models.py

line 503