mirror of
https://github.com/emadeldeen24/TSLANet.git
synced 2024-11-27 00:50:10 +08:00
Update TSLANet_classification.py
This commit is contained in:
parent
9d191631f9
commit
be1b18e506
@ -388,7 +388,7 @@ if __name__ == '__main__':
|
|||||||
|
|
||||||
parser = argparse.ArgumentParser()
|
parser = argparse.ArgumentParser()
|
||||||
parser.add_argument('--model_id', type=str, default='TEST')
|
parser.add_argument('--model_id', type=str, default='TEST')
|
||||||
parser.add_argument('--data_path', type=str, default=r'C:\Emad\datasets\UCR_2015\UCR_2015_preprocessed\OSULeaf')
|
parser.add_argument('--data_path', type=str, default=r'data/hhar')
|
||||||
|
|
||||||
# Training parameters:
|
# Training parameters:
|
||||||
parser.add_argument('--num_epochs', type=int, default=100)
|
parser.add_argument('--num_epochs', type=int, default=100)
|
||||||
|
Loading…
Reference in New Issue
Block a user