-
Notifications
You must be signed in to change notification settings - Fork 424
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
using simclr for new downstream task #751
Comments
can u provide your whole command to run on down streaming tasks ? |
sure , my downstream task is for pose estimation not for object detection or segmentation. I create a new directory for benchmarks/train ,and for configs/benchmark/mmpose.
#!/usr/bin/env bash
set -x
CFG=$1
PRETRAIN=$2 # pretrained model
GPUS=$3
PY_ARGS=${@:4}
set work_dir according to config path and pretrained model to distinguish different models
WORK_DIR="$(echo ${CFG%.*} | sed -e "s/configs/work_dirs/g")/$(echo $PRETRAIN | rev | cut -d/ -f 1 | rev)"
PYTHONPATH="$(dirname $0)/..":$PYTHONPATH \
mim train mmpose $CFG \
--launcher pytorch -G $GPUS \
--work-dir $WORK_DIR \
--cfg-options model.backbone.init_cfg.type=Pretrained \
model.backbone.init_cfg.checkpoint=$PRETRAIN \
#model.roi_head.shared_head.init_cfg.type=Pretrained \
#model.roi_head.shared_head.init_cfg.checkpoint=$PRETRAIN \
$PY_ARGS
I run this command: |
Checklist
I am trying to use mmselfsup for another down streaming tasks rather than only object detection and segmentation.
i am using your model of simclr and the existed weight in your website .
and trying to do the downstreaming task on my dataset.
i got this error message as bellow:
any help please .
The text was updated successfully, but these errors were encountered: