_BASE_: "../base-prompt.yaml" RUN_N_TIMES: 1 DATA: NAME: "StanfordDogs" DATAPATH: "" #TODO: need to specify here NUMBER_CLASSES: 120 MULTILABEL: False MODEL: TYPE: "swin" SOLVER: BASE_LR: 0.001 WEIGHT_DECAY: 0.0001
时间: 2023-06-20 11:10:36 浏览: 119
根据这段 YAML 配置文件来看,你们正在使用一个叫做 "StanfordDogs" 的数据集,但是 DATAPATH 还没有被指定。同时,这个数据集有 120 个类别,不是多标签的数据集。模型选用了 swin,优化器使用了基础学习率为 0.001 和权重衰减为 0.0001 的设置。最后,这个配置文件只会运行一次。
阅读全文