浏览代码

Correcting the wrong reference (#649)

Co-authored-by: Sebastian Raschka <mail@sebastianraschka.com>
Pratyush Subhadarshi 5 月之前
父节点
当前提交
d56417c34c
共有 1 个文件被更改,包括 2 次插入2 次删除
  1. 2 2
      ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py

+ 2 - 2
ch05/03_bonus_pretraining_on_gutenberg/pretraining_simple.py

@@ -21,8 +21,8 @@ import torch
 # For llms_from_scratch installation instructions, see:
 # https://github.com/rasbt/LLMs-from-scratch/tree/main/pkg
 from llms_from_scratch.ch02 import create_dataloader_v1
-from llms_from_scratch.ch04 import GPTModel, generate_and_print_sample
-from llms_from_scratch.ch05 import calc_loss_batch, evaluate_model, plot_losses
+from llms_from_scratch.ch04 import GPTModel
+from llms_from_scratch.ch05 import calc_loss_batch, evaluate_model, plot_losses, generate_and_print_sample
 
 
 def read_text_file(file_path):