James YanUsing pre-trained Vision Transformer model and ResNet model as features extractors for image…Transfer learning is “standing on the shoulders of giants”. The “giants” here are those open-sourced models that have been trained millions…Jan 4, 20223Jan 4, 20223
James YanA little thinking on avoiding GPU memory outage during the model training (PyTorch)Long story short, I was facing a problem, when I start to train a model, it can always complete training on only one batch and fail on the…Dec 18, 2021Dec 18, 2021