NameError: name 'mobilenet_v3_large' is not defined

[Abhijeet241093]

python3 main.py --gpu 0 --pro -enc mobilenet_v3_large --dataset mvtec --action-type norm-test -inp 256
--class-name metal_nut --checkpoint mvtec_mobilenet_v3_large_freia-cflow_pl3_cb8_inp512_run0_metal_nut.pt
LR schedule: [12, 18, 22]
Number of pool layers = 3
Traceback (most recent call last):
File "main.py", line 91, in
main(c)
File "main.py", line 84, in main
train(c)
File "/home/radhakrishna/cflow-ad/train.py", line 262, in train
encoder, pool_layers, pool_dims = load_encoder_arch(c, L)
File "/home/radhakrishna/cflow-ad/model.py", line 158, in load_encoder_arch
encoder = mobilenet_v3_large(pretrained=True, progress=True).features
NameError: name 'mobilenet_v3_large' is not defined