Skip to content

self.device=device #7

@jiayalu123

Description

@jiayalu123

执行命令的时候报错,报错内容如下:
Traceback (most recent call last):
File "train.py", line 207, in
main()
File "train.py", line 166, in main
y_pred = train_problem(current_depth="flat", upper_depth=None, class_num=len(learning_categories), fine_tuning=fine_tuning, embedding_weight=embedding_weight, input_data=input_network_data, model_type=model_type, learning_categories=learning_categories)
File "train.py", line 40, in train_problem
network_output = cnn_train.main(params)
File "E:\▒▒▒▒▒\▒▒▒▒▒▒▒\▒▒▒ܹ▒▒▒\HFT-CNN-master\cnn_train.py", line 121, in main
updater = MyUpdater(train_iter, optimizer, params["output_dimensions"], device=params["gpu"])
File "E:\▒▒▒▒▒\▒▒▒▒▒▒▒\▒▒▒ܹ▒▒▒\HFT-CNN-master\MyUpdater.py", line 36, in init
self.device = device
AttributeError: can't set attribute

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions