목록pytorch (16)
rueki
1. OpenVINO 설치 (PIP 기준) - Windows python -m venv openvino_env openvino_env\Scripts\activate python -m pip install --upgrade pip pip install openvino-dev==2023.0.1 - Linux python3 -m venv openvino_env source openvino_env/bin/activate python -m pip install --upgrade pip pip install openvino-dev==2023.0.1 2. pytorch -> ONNX 변환 pytorch 모델을 onnx float32 type의 모델로 변환 먼저 적용 torch.onnx.export(net, x, on..
로컬에서 학습하는 것 모니터링 해당 X 리눅스 서버를 통해서 학습하는 내용 monitoring 확인 시 내 로컬에서 확인하기 위함. ssh -NfL localhost:6006:localhost:6006 "서버 접속자 id"@"서버 주소" 맨 뒤에서 사용하는 서버의 정보에 대한 것만 개별 설정하면 끝. localhost 번호 변경은 사용자 마음. 이 전에 tensorboard를 서버에서 실행시켜줘야한다. tensorboard --logdir="텐서보드 log기록 경로" --port=6006 train 코드에서 tensorboardx를 사용하게 되면 본 폴더 내에 runs라는 기록 저장 폴더가 생긴다. 그 폴더 경로를 logdir에 넣어주면 해당 폴더 내의 실험 기록들이 전부 올라가며, 학습 중인 것도 실시..
1. conda 환경 및 도커 환경 구축 - 토치 환경 체크 후 다 맞춰서 설치할 것 2. mmdetection 설치 및 추가 라이브러리 설치 git clone https://github.com/open-mmlab/mmdetection.git cd mmdetection pip install -v -e .
isinstance로 모델 타입 확인하기 -save if isinstance(model, (DataParallel, DistributedDataParallel)): torch.save(model.module.state_dict(), model_save_name) else: torch.save(model.state_dict(), model_save_name) -load state_dict = torch.load(model_name, map_location=current_gpu_device) if isinstance(model, (DataParallel, DistributedDataParallel)): model.module.load_state_dict(state_dict) else: model.load_s..