Code

conditional gfn#188

Code

Merged

josephdviviano merged 33 commits intomasterGFNOrg/torchgfn:masterfrom conditional_gfnGFNOrg/torchgfn:conditional_gfnCopy head branch name to clipboardOct 24, 2024

+861-161

Commits

Commits on Sep 25, 2024

• example of conditional GFN computation with TB only (for now)

  josephdviviano

  committed
  6e8dc4d

  Copy full SHA for 6e8dc4d

• should be no change

  josephdviviano

  committed
  39fb5ee

  Copy full SHA for 39fb5ee

• Trajectories objects now have an optional .conditonal field which optionally contains a tensor of conditioning vectors (one per trajectory)

  josephdviviano

  committed
  2bc2263

  Copy full SHA for 2bc2263

• small changes to logz paramater handling, optionally incorporate conditioning into PB and PF computation

  josephdviviano

  committed
  99afaf3

  Copy full SHA for 99afaf3

• logZ is optionally computed using a conditioning vector

  josephdviviano

  committed
  e6d25a0

  Copy full SHA for e6d25a0

• NeuralNets now have input/output dims

  josephdviviano

  committed
  2c72bf9

  Copy full SHA for 2c72bf9

• added a ConditionalDiscretePolicyEstimator, and the forward of GFNModule can now accept raw tensors

  josephdviviano

  committed
  580c455

  Copy full SHA for 580c455

• added conditioning to sampler, which will save the tensor as an attribute of the trajectory

  josephdviviano

  committed
  a74872f

  Copy full SHA for a74872f

• black

  josephdviviano

  committed
  056d935

  Copy full SHA for 056d935

Commits on Oct 1, 2024

• API changes adapted

  josephdviviano

  committed
  96b725c

  Copy full SHA for 96b725c

• added conditioning to all gflownets

  josephdviviano

  committed
  5cd32a7

  Copy full SHA for 5cd32a7

• both trajectories and transitions can now store a conditioning tensor

  josephdviviano

  committed
  877c4a0

  Copy full SHA for 877c4a0

• input_dim setting is now private

  josephdviviano

  committed
  279a313

  Copy full SHA for 279a313

• added exception handling for all estimator calls potentially involving conditioning

  josephdviviano

  committed
  65135c1

  Copy full SHA for 65135c1

• API change -- n vs. n_trajectories

  josephdviviano

  committed
  b4c418c

  Copy full SHA for b4c418c

• change test_box target value

  josephdviviano

  committed
  738b062

  Copy full SHA for 738b062

• API changes

  josephdviviano

  committed
  4434e5f

  Copy full SHA for 4434e5f

• hacky fix for problematic test (added TODO)

  josephdviviano

  committed
  851e03e

  Copy full SHA for 851e03e

Commits on Oct 4, 2024

• working examples for all 4 major losses

  josephdviviano

  committed
  5152295

  Copy full SHA for 5152295

• added conditioning indexing for correct broadcasting

  josephdviviano

  committed
  1d64b55

  Copy full SHA for 1d64b55

• added a ConditionalScalarEstimator which subclasses ConditionalDiscretePolicyEstimator

  josephdviviano

  committed
  348ee82

  Copy full SHA for 348ee82

• added modified DB example

  josephdviviano

  committed
  9120afe

  Copy full SHA for 9120afe

• conditioning added to modified db example

  josephdviviano

  committed
  f59f4de

  Copy full SHA for f59f4de

• black

  josephdviviano

  committed
  c5ef7ea

  Copy full SHA for c5ef7ea

Commits on Oct 9, 2024

• reorganized keyword arguments and fixed some type errors (not all)

  josephdviviano

  committed
  d67dfd5

  Copy full SHA for d67dfd5

• reorganized keyword arguments and fixed some type errors (not all)

  josephdviviano

  committed
  d56a798

  Copy full SHA for d56a798

• added typing and a ConditionalScalarEstimator

  josephdviviano

  committed
  db8844c

  Copy full SHA for db8844c

• added typing

  josephdviviano

  committed
  e03c03a

  Copy full SHA for e03c03a

• typing

  josephdviviano

  committed
  6b47e06

  Copy full SHA for 6b47e06

• typing

  josephdviviano

  committed
  988faf0

  Copy full SHA for 988faf0

• added kwargs

  josephdviviano

  committed
  f2bbce3

  Copy full SHA for f2bbce3

Commits on Oct 24, 2024

• renamed torso to trunk

  josephdviviano

  committed
  eb13a2d

  Copy full SHA for eb13a2d

• renamed torso to trunk

  josephdviviano

  committed
  fd3d9dc

  Copy full SHA for fd3d9dc