adiren7/FoodVisionVIT

1a

2Zi�e�
�@s2dd
lZdd
l
Z
ddlmZ
deed�dd�Zd
S)	�N)
�nn��*)�num_classes�seedcCs^tj
jj}|��}tj
j|d
�
}|��D]3}d|_q(t�	tj4ddd�tjd|d��|_||fS)N)
�weightsFg333333�?T)�
p�inplacei�)�in_features�out_features)
�torchvision�modelsZViT_B_16_Weights�DEFAULT�5transformsZvit_b_16�6parameters�
requires_gradr�7Sequential�Dropout�Linear�8classifier)rrrr�model�param�r�"C:\Users\HP\FoodVisionVIT\model.py�create_vit_models9




�r)rr)�torchrr�intrrrrr�<module>s