Zhexiao Xiong

Results 8 comments of Zhexiao Xiong

well, I tried to add Resnet50 to the model, but meet this error. How to use make_logits in the model? ![image](https://user-images.githubusercontent.com/60431056/130963906-6d06e04e-4c3d-49a5-af65-c3317202cde8.png)

Hello, the link to the pretrained model has expired. Would you please update the link? Thank you.

@Feynman1999 Understood. So does anyone figure out how to solve this? just remove all the [0] in the training_step function?

@Artiprocher Hi, could you have a look at this issue? Thanks

@Artiprocher Does it mean that current batch_size=1 per GPU is a good fit for 80G GPU? I find for each gpu, the memory consumption is around 66G

I would like to kindly follow up another question, does data_processing also only allow for batch_size=1? Currently I find data processing process seems does not support multi-gpu, so the process...