cogdl.tasks.pretrain

Module Contents

Classes

PretrainTask

class cogdl.tasks.pretrain.PretrainTask(args)[source]

Bases: cogdl.tasks.BaseTask

static add_args(parser)[source]

Add task-specific arguments to the parser.

train(self)[source]