gist.github.com
  
  
    https://gist.github.com/dylandjian/f1e56ed65480dbf08b3d024e4e4d1f7f 246
example.py
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.multiprocessing as multiprocessing
import time

DEVICE = torch.device("cuda")

class Model1(nn.Module):
    def __init__(self):
This file has been truncated. show original




  
  
    
    
  
  


I managed to reproduce it.
Any idea on what I’m doing wrong / not understanding ?
Thanks !