forked from Alexander-H-Liu/MalConv-Pytorch
-
Notifications
You must be signed in to change notification settings - Fork 1
/
Copy pathexample.yaml
24 lines (21 loc) · 990 Bytes
/
example.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
exp_name: 'example'
### Data path
train_data_path: 'data/train/' # Training data
train_label_path: 'data/example-train-label.csv' # Training label
valid_data_path: 'data/valid/' # Validation Data
valid_label_path: 'data/example-valid-label.csv' # Validation Label
### output path
log_dir: 'log/'
pred_dir: 'pred/'
checkpoint_dir: 'checkpoint/'
### Parameter
use_gpu: True #
use_cpu: 1 # Number of cores to use for data loader
display_step: 2 # Std output update rate during training
test_step: 20 # Test per n step
learning_rate: 0.0001 #
max_step: 1000 # Number of steps to train
batch_size: 20 #
first_n_byte: 2000000 # First N bytes of a PE file as the input of MalConv (defualt: 2 million)
window_size: 500 # Kernel size & stride for Malconv (defualt : 500)
sample_cnt: 1 # Number of data sampled for trianing (defualt 1 = all)