search

HUANGLIZI / LViT

[IEEE Transactions on Medical Imaging/TMI] This repo is the official implementation of "LViT: Language meets Vision Transformer in Medical Image Segmentation"
MIT License
298 stars 26 forks source link

Training on 1 Gpu #32

Closed uurcelikk closed 11 months ago

uurcelikk commented 11 months ago

Hello, What kind of optimisation can I do to train on 1 gpu?

HUANGLIZI commented 11 months ago

You should reduce the batch size and also the learning rate to get optimal performance on 1 GPU.