[pytorch] DataParallel ๋กœ ํ•™์Šตํ•œ ๋ชจ๋ธ load

2021. 2. 17. 01:02ยท๐Ÿ’ป Programming/AI & ML
๋ฐ˜์‘ํ˜•
model = custom_LSTM()

model = torch.nn.DataParallel(model)
with open(os.path.join('C:/Users/' + 'model_1.pt'), 'rb') as f:
    model.load_state_dict(torch.load(f))

DataParallel ๋กœ ํ•™์Šต์‹œํ‚จ ๋ชจ๋ธ์„ loadํ•ด์„œ ์‚ฌ์šฉํ•  ๋•Œ๋Š” ์œ„์™€ ๊ฐ™์ด torch.nn.DataParallel(model) ์ฝ”๋“œ๋ฅผ ์จ์ค˜์•ผ error ์—†์ด ์‚ฌ์šฉ ๊ฐ€๋Šฅํ•˜๋‹ค.

 

๋ฐ˜์‘ํ˜•

'๐Ÿ’ป Programming > AI & ML' ์นดํ…Œ๊ณ ๋ฆฌ์˜ ๋‹ค๋ฅธ ๊ธ€

[pytorch] pytorch ๋ชจ๋ธ ๋กœ๋“œ ์ค‘ Missing key(s) in state_dict ์—๋Ÿฌ  (0) 2022.12.15
[pytorch] COCO Data Format ์ „์šฉ Custom Dataset ์ƒ์„ฑ  (1) 2022.06.04
[pytorch] model ์— ์ ‘๊ทผํ•˜๊ธฐ, ํŠน์ • layer ๋ณ€๊ฒฝํ•˜๊ธฐ  (0) 2022.01.05
[pytorch] Custom dataset, dataloader ๋งŒ๋“ค๊ธฐ  (0) 2022.01.02
[pytorch] ๋ชจ๋ธ์˜ ์ผ๋ถ€ ๋ ˆ์ด์–ด ์›จ์ดํŠธ ์—…๋ฐ์ดํŠธ ๋ง‰๊ธฐ | model freezing (๋ชจ๋ธ ํ”„๋ฆฌ์ง•)  (0) 2021.02.17
'๐Ÿ’ป Programming/AI & ML' ์นดํ…Œ๊ณ ๋ฆฌ์˜ ๋‹ค๋ฅธ ๊ธ€
  • [pytorch] COCO Data Format ์ „์šฉ Custom Dataset ์ƒ์„ฑ
  • [pytorch] model ์— ์ ‘๊ทผํ•˜๊ธฐ, ํŠน์ • layer ๋ณ€๊ฒฝํ•˜๊ธฐ
  • [pytorch] Custom dataset, dataloader ๋งŒ๋“ค๊ธฐ
  • [pytorch] ๋ชจ๋ธ์˜ ์ผ๋ถ€ ๋ ˆ์ด์–ด ์›จ์ดํŠธ ์—…๋ฐ์ดํŠธ ๋ง‰๊ธฐ | model freezing (๋ชจ๋ธ ํ”„๋ฆฌ์ง•)
๋ญ…์ฆค
๋ญ…์ฆค
AI ๊ธฐ์ˆ  ๋ธ”๋กœ๊ทธ
    ๋ฐ˜์‘ํ˜•
  • ๋ญ…์ฆค
    CV DOODLE
    ๋ญ…์ฆค
  • ์ „์ฒด
    ์˜ค๋Š˜
    ์–ด์ œ
  • ๊ณต์ง€์‚ฌํ•ญ

    • โœจ About Me
    • ๋ถ„๋ฅ˜ ์ „์ฒด๋ณด๊ธฐ (198)
      • ๐Ÿ“– Fundamentals (33)
        • Computer Vision (9)
        • 3D vision & Graphics (6)
        • AI & ML (15)
        • NLP (2)
        • etc. (1)
      • ๐Ÿ› Research (64)
        • Deep Learning (7)
        • Image Classification (2)
        • Detection & Segmentation (17)
        • OCR (7)
        • Multi-modal (4)
        • Generative AI (6)
        • 3D Vision (2)
        • Material & Texture Recognit.. (8)
        • NLP & LLM (11)
        • etc. (0)
      • ๐ŸŒŸ AI & ML Tech (7)
        • AI & ML ์ธ์‚ฌ์ดํŠธ (7)
      • ๐Ÿ’ป Programming (85)
        • Python (18)
        • Computer Vision (12)
        • LLM (4)
        • AI & ML (17)
        • Database (3)
        • Apache Airflow (6)
        • Docker & Kubernetes (14)
        • ์ฝ”๋”ฉ ํ…Œ์ŠคํŠธ (4)
        • C++ (1)
        • etc. (6)
      • ๐Ÿ’ฌ ETC (3)
        • ์ฑ… ๋ฆฌ๋ทฐ (3)
  • ๋งํฌ

  • ์ธ๊ธฐ ๊ธ€

  • ํƒœ๊ทธ

    Text recognition
    pytorch
    multi-modal
    nlp
    object detection
    CNN
    LLM
    GPT
    OCR
    3D Vision
    Python
    AI
    VLP
    ChatGPT
    pandas
    ํŒŒ์ด์ฌ
    OpenCV
    ์ปดํ“จํ„ฐ๋น„์ „
    airflow
    ๊ฐ์ฒด ๊ฒ€์ถœ
    segmentation
    ๊ฐ์ฒด๊ฒ€์ถœ
    ๋„์ปค
    Computer Vision
    material recognition
    Image Classification
    deep learning
    ํ”„๋กฌํ”„ํŠธ์—”์ง€๋‹ˆ์–ด๋ง
    OpenAI
    ๋”ฅ๋Ÿฌ๋‹
  • ์ตœ๊ทผ ๋Œ“๊ธ€

  • ์ตœ๊ทผ ๊ธ€

  • hELLOยท Designed By์ •์ƒ์šฐ.v4.10.3
๋ญ…์ฆค
[pytorch] DataParallel ๋กœ ํ•™์Šตํ•œ ๋ชจ๋ธ load
์ƒ๋‹จ์œผ๋กœ

ํ‹ฐ์Šคํ† ๋ฆฌํˆด๋ฐ”