Skip to content

Instantly share code, notes, and snippets.

View malkin1729's full-sized avatar

malkin1729

  • Mila and Université de Montréal
  • Montréal, Québec, Canada
  • 08:51 (UTC -04:00)
View GitHub Profile
Loading
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@malkin1729
malkin1729 / grid.py
Created January 30, 2022 19:28
GFlowNet trajectory balance: grid environment example
import torch as T
import numpy as np
import tqdm
import pickle
device = T.device('cpu')
horizon = 8
ndim = 2