标签：training GMVSE 训练 int number 参数 str model 源代码
一、参数配置
Parameters:
| Name |
Type |
Description |
Note |
| mode |
enum(str) |
pretrain, train or evaluate. |
模式：预训练、训练、评估 |
| data_filename |
str |
data file (e.g., ./data/processed_porto.csv). |
数据及文件名 |
| map_size |
(int, int) |
size of the grid map. |
网格图大小 |
| model |
str |
choose a model |
GMVSAE |
| token_dim |
int |
dimensionality of grid token. |
网格标签的维度 |
| rnn_dim |
int |
dimensionality of rnn hidden state. |
RNN隐藏层神经元个数 |
| cluster_num |
int |
number of Gaussian components. |
高斯分量数目 |
| model_dir |
str |
directory to save/load a model during training or eval. |
在训练或者评估期间保存/加载模型的文件夹 |
| pretrain_dir |
str |
directory to save/load a model during pretraining. |
在预训练期间保存/加载模型的文件夹 |
| num_negs |
int |
number of negative samples during training. |
在训练期间负样本的数目 |
| optimizer |
enum(str) |
training optimizer (e.g., adam or sgd). |
训练优化器 |
| learning_rate |
float |
learning rate for training. |
训练时的学习率 |
| batch_size |
int |
minibatch size |
批量数 |
| num_epochs |
int |
number of passes over the training data. |
训练轮数 |
| partial_ratio |
float |
partial trajectory evaluation |
|
| log_steps |
int |
number of batches to print the log info. |
输出日志信息的批量数 |
| gpu_id |
str |
0 |
GPU的id号 |
标签：training,
GMVSE,
训练,
int,
number,
参数,
str,
model,
源代码
From： https://www.cnblogs.com/sgqlog/p/17092105.html