itakura-2023-12-22 進捗報告
やったこと
train_net.py
クラス分類ではないのでエラー率を計算する部分は削除
code:def train_epoch
# Compute the errors.
num_topks_correct = metrics.topks_correct(preds, labels, (1, 5))
top1_err, top5_err = [
(1.0 - x / preds.size(0)) * 100.0 for x in num_topks_correct
・・・
# Copy the stats from GPU to CPU (sync point).
loss, top1_err, top5_err = (
loss.item(),
top1_err.item(),
top5_err.item(),
)
code:def eval_epoch
# Compute the errors.
num_topks_correct = metrics.topks_correct(preds, labels, (1, 5))
# Combine the errors across the GPUs.
top1_err, top5_err = [
(1.0 - x / preds.size(0)) * 100.0 for x in num_topks_correct
]
・・・
# Copy the errors from GPU to CPU (sync point).
top1_err, top5_err = top1_err.item(), top5_err.item()
学習の実行
python tools/run_net.py --cfg configs/DReyeVE/DReyeVEformer_divST_8x32_224.yaml \
→学習率を変えてもlossが下がらない。
画像のスケーリングが問題?
評価の方法はどうすればよいか