Sadjad Alikhani commited on
Commit
b13ee07
·
verified ·
1 Parent(s): 291d67c

Delete save_model.py

Browse files
Files changed (1) hide show
  1. save_model.py +0 -29
save_model.py DELETED
@@ -1,29 +0,0 @@
1
- # -*- coding: utf-8 -*-
2
- """
3
- Created on Fri Sep 13 19:16:37 2024
4
-
5
- @author: salikha4
6
- """
7
-
8
- import torch
9
- from tokenizer import WirelessChannelTokenizer
10
- from model import WirelessChannelModel, WirelessConfig
11
-
12
- model_name = 'model_weights.pth'
13
- model_path = f'huggingFace/{model_name}'
14
-
15
- # Initialize model config
16
- config = WirelessConfig()
17
-
18
- # Initialize the model
19
- model = WirelessChannelModel(config)
20
-
21
- # Load pretrained weights
22
- model.load_state_dict(torch.load(model_path))
23
-
24
- # Initialize tokenizer (preprocessor)
25
- tokenizer = WirelessChannelTokenizer(patch_size=16, max_len=129)
26
-
27
- # Save the model and tokenizer for Hugging Face
28
- model.save_pretrained("huggingFace/")
29
- tokenizer.save_pretrained("huggingFace/")