search

TACJu / TransFG

This is the official PyTorch implementation of the paper "TransFG: A Transformer Architecture for Fine-grained Recognition" (Ju He, Jie-Neng Chen, Shuai Liu, Adam Kortylewski, Cheng Yang, Yutong Bai, Changhu Wang, Alan Yuille).
MIT License
382 stars 88 forks source link

memory error #25

Closed Manigoat closed 2 years ago

Manigoat commented 2 years ago

Dear author, why I always face memory error when I start training. My memory is 16G. Is there any problem in dataset pipline?