cosine_metric_learning
cosine_metric_learning copied to clipboard
how to evaluate my own dataset
hello i used
python train_motor.py --mode=eval --dataset_dir=resources/motor-evaluation-master/ --loss_mode=cosine-softmax --log_dir=./output/motor/ --run_id=cosine-softmax --eval_log_dir=./eval_output/motor
i got an error : ValueError: elems must be a 1+ dimensional Tensor, not a scalar
any solution ?
@SpawN1337, did you train the model with the custom dataset? if yes, could you please share folder structure or any articles on how to train the model for custom dataset?