Skip to content

train #80

Description

@mattat0516

I'm trying to train with reference data, but I'm getting an error like this. Here are my script parameter settings and errors. Hope to get an answer, thanks。
python -m torch.distributed.launch --nproc_per_node=1 --master_port=12233 --use_env run_train.py
--diff_steps 2000
--lr 0.0001
--learning_steps 50000
--save_interval 10000
--seed 102
--noise_schedule sqrt
--hidden_dim 128
--bsz 2048
--dataset CommonsenseConversation
--data_dir ./datasets/CommonsenseConversation
--vocab bert
--seq_len 128
--schedule_sampler lossaware
--notes test-text

error:
train.py: error: argument --dataset: expected one argument
train.sh: line 3: --diff_steps: command not found

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions