AutoML for v4.0.2 with Efficientnet_b1_relu

nvidia-smi works now and HTTPConnectionPool(host=‘’, port=31951): Max retries exceeded with url: /api/v1/user/4a268f0c-2b02-4fe3-9ca2-b161c0af7231/dataset also seems to be resolved

About the main issue here, how can I select any model other than resnet? No matter which model I choose in the pre-trained mapping, the specs are always loaded with resnet18. I want to choose efficientnet_b1_relu