Class: NanoGPT::SampleConfig

Inherits:
BaseConfig show all
Defined in:
lib/nano_gpt/train_config.rb

Overview

Configuration for sampling/generation

Constant Summary collapse

DEFAULTS = 
{
  out_dir: "out-shakespeare-char",
  dataset: "shakespeare_char",
  start: "\n",
  num_samples: 5,
  max_new_tokens: 500,
  temperature: 0.8,
  top_k: 200,
  seed: 1337,
  device: "auto"
}.freeze

Instance Attribute Summary

Attributes inherited from BaseConfig

#values

Class Method Summary collapse

Methods inherited from BaseConfig

#[], #[]=, #initialize, load, #load_json, #save_json, #to_h

Constructor Details

This class inherits a constructor from NanoGPT::BaseConfig

Class Method Details

.defaults ⇒ Object 



168
169
170
 
# File 'lib/nano_gpt/train_config.rb', line 168

def self.defaults
  DEFAULTS
end