ADD exception handler for efficientnet model#6

jerogar wants to merge 1 commit intoc-gabri:masterc-gabri/Federated-Learning-PyTorch:masterfrom

jerogar:masterjerogar/Federated-Learning-PyTorch:masterCopy head branch name to clipboard