diff --git a/FID_caption.txt b/FID_caption.txt
new file mode 100644
index 00000000..44827858
--- /dev/null
+++ b/FID_caption.txt
@@ -0,0 +1,10000 @@
+000000029984 A group of people wading around a large body of water.
+000000370818 a cabinet towels a toilet and a sink
+000000073533 A few boys read comic books together outside
+000000036539 There is a man and a young girl snowboarding 
+000000193429 A male snowboarder sits with his board on a snowy hill.
+000000337055 A woman wearing a skirt shows off her tatttoos
+000000224200 A large dog is tied up to a fire hydrant.
+000000322163 A man making a pizza in a kitchen. 
+000000133819 A blue bus makes its way down a narrow city street.
+000000080153 A person on some skis and a dog in the snow.
+000000171050 A fire hydrant sits in front of a brick building
+000000560911 a man sitting on a black leather couch messing with a phone.
+000000229753 Two giraffes are standing by a palm tree.  
+000000528980 a red umbrella is inside out in a city
+000000212559 A stable full of lambs standing and laying around.
+000000309964 A pair of friends sit beneath a massive blue umbrella.
+000000067534 A young man is taking a "selfie" of himself. 
+000000566758 a school bus on a city street 
+000000178028 A bathroom sink with a towel rack, hair dryer and a mirror.
+000000418959 A duck standing on a pile of debris in water
+000000081738 A person has a knife in his hand above the cake. 
+000000462576 Breakfast sits on the table, with eggs, hash brown and roll.
+000000280325 Graffiti is on the wall next to the parking meter
+000000530820 A man standing on a tennis court holding a racket
+000000523241 A busy street with several lanes and a crossing sign in the foreground.
+000000466125 A little girl smiling for the camera with an umbrella behind her.
+000000281032 A man with his hair tied back prepares food. 
+000000501023 a street sign with a sky back ground
+000000563470 A woman is standing outside with her head on a bat.
+000000491725 Small sailboats are lit up on the water.
+000000130586 Kid is over joyed as he flies his multi colored kite
+000000414510 A few people sitting by a busy street
+000000038576 A desktop computer with it's monitor sitting on it's side.
+000000547336 a woman sitting next to a fire hydrant with a notebook in hand
+000000052507 A person that is going out in the water.
+000000061108 A dog sits in the basket of a bicycle leaning against bleachers.
+000000050149 Bunches of bananas are hanging from a fruit stand.
+000000398028 A person practices batting during a baseball game.
+000000338560 A blue and yellow fire hydrant on the side of a road.
+000000474170 Two elephants outdoors walking on green grass with trees near by.
+000000323709 An airplane is parked next to a domed tower.
+000000367569 a fireplace a table couch piano lights and a picture
+000000287959 a sandwich that has something green and red in it
+000000287527 two people standing side by side smiling and one wearing a tie
+000000426372 A truck is shown with another car in the back of it.
+000000219440 A group of cows standing in the grass.
+000000059044 A group of people looking at some kind of show or exhibit 
+000000450559 A skateboarder executing a jump trick at a park.
+000000043581 A small pizza sitting on a blue table.
+000000514376 An electric trolley bus on the street in Seattle
+000000445834 A group of people begin boarding a touring bus.
+000000440475 The dark wood and marble pillars give this living space a warm feel. 
+000000564133 Pack of 6 elephants walking through the native brush.
+000000403817 a white and black cat looking up in the air in front of a desktop computer.
+000000451155 a woman sitting on the bed of a dark room with the laptop screen lighting on her face
+000000070158 A group of zebras walk along the sand.
+000000006460 A man is surfing and a wave is crashing
+000000394328 A toilet with a poster above it in a bathroom
+000000463918 Two guys standing in a room playing video games.
+000000543047 Living area with purple couches, looking into the dining area.
+000000103723 an elephant in a zoo walks around slowly
+000000413247 A laptop and a computer mouse on a desk.
+000000375430 A pile of pancakes sitting on a plate of food.
+000000261097 A man in black shirt and white shorts on a tennis court.
+000000505638 A locomotive in the rear of a train pushing it. 
+000000415194 An old fashioned kitchen is set up for display.
+000000250766 A plate with salad and a cut of meat and silverware laid on top.
+000000335081 Stuffed bears and ducks are lined up on the floor.
+000000398028 The man has a bat in his hands.
+000000545826 A cat on a stool with something on its head
+000000085772  a blonde haired teenage man is swinging a tennis racquet at a ball.
+000000274687 A black road bicycle leaning against a bed.
+000000180487 A man sits at a table outside wearing an umbrella hat.
+000000055167 A large bus driving down a road. 
+000000455872 A black and white image of a shipyard with some boats. 
+000000002299 A vintage school picture of grade school aged children.
+000000363072 The commuter train is headed toward the tall building.
+000000071711 A group of people walk around old planes in a hanger.
+000000529762 A man at a table taking a picture of his food.
+000000186345 4 different colored sea horses flying with 4 birds.
+000000161397 Several displays of flowers are on a table.
+000000349678 two flags flying in the wind while atop a train station clock.
+000000507223 A baseball players is ealking toward the opposing team.
+000000297698 The skier takes off down the steep hill
+000000516601 People are skiing and snowboarding on a high mountain. 
+000000222825 an image of a kitchen setting with white appliances
+000000161879 there is a man riding a skateboard down a rail
+000000509014 a bunch of people that are walking across the snow
+000000151629 A man takes a swing at a tennis ball on a grass court.
+000000529105 A horse laying down on top of a lush green field.
+000000473821 THIS IS A PICTURE OF A WELL DECORATED ROOM
+000000557884 A green leafy type plant is located on a wall.
+000000422998 A person makes a sandwich on a paper plate.
+000000366199 I am not sure what this image is.
+000000108440 A train on top of an overpass bridge set of train tracks with fields on both side.
+000000121031 Four men ride horses across a body of water. 
+000000039484 A street full of cars that are driving and parked and a group of people sitting outside eating.
+000000531134 A traffic light with a frowning face on the red light and a city street and buildings behind it.
+000000009400 A group of people sitting around a table using computers.
+000000458045 Little boys playing on a baseball field in baseball uniforms. 
+000000331569 The strawberries was sitting beside the tall glass of milkshake
+000000331075 a white brown and black dog is by some water 
+000000532690 This blonde woman is getting ready to play a game.
+000000051976 A person on a surf board riding a wave.
+000000022969 Two giraffe stand in an enclosure, on giraffe leans over the wall.
+000000015517 The view of a metropolitan area behind many buses.
+000000097337 This is a nice living room set up with two couches and a television.
+000000410510 A young boy on a tennis court prepares to hit a tennis ball with his racquet.
+000000184611 Guy pedals a bike through a small village.
+000000468632 some males are playing frisbee and some trees
+000000271116 Two men dressed in suits holding their cellphones.
+000000328286 Boats docked at a harbor facing the city.
+000000209753 The young girl using her cell phone is wearing a glowing belt.
+000000338428 an antique green pickup truck is parked in the grass.
+000000493905 Man in white outfit swinging a racket at a tennis ball.
+000000363666 Young men having a meal at a table.
+000000089648 A crowd of people sitting down next to each other.
+000000239717 A man holding two plates of food while smiling.
+000000536038 The woman is reading something while laying down in bed. 
+000000209142 Green salad with broccoli and peas with fork and bowl
+000000465585 A silver container with cheese and broccoli in it.
+000000367228 A man is flying a kite in the middle of a field next to the ocean.
+000000180878 A birthday cake made to look like a bag, shoe, car, and iPhone.
+000000025057 A man without shirt on is throwing a frisbee.
+000000242934 A large living room is seen in this image.
+000000041635 Three cows stand in a field behind a barbed wire fence.
+000000271402 Two young girls on a tennis court with a racquet and a scooter
+000000104198 A stop sign with a pagoda in the background.
+000000163682 A man in an office, wearing a shirt and tie, smiles at the camera.
+000000080274 Two elephants that are standing next to each other.
+000000455157 Man sitting at a picnic table using a laptop shaded by an umbrella.
+000000023359 A man flying through the air while riding a snowboard.
+000000467776 A mother and it's calf standing in a field.
+000000084241 A group of women in kitchen next to pot.
+000000053626 Four people on skis below a mountain taking a picture.
+000000265777 A person waiting for her meal to be served.
+000000151662 a giraffe behind a fence eating grass from a feeder
+000000074256 Two women walking on  a train platform
+000000133778 A brown calf drinking milk from its mother.
+000000013201 A young man with long hair, wearing all black, doing stunts on a skateboard.
+000000568439 two public transit buses on a city street 
+000000240767 A red brick building with a street sign sitting outside of it.
+000000549738 A large, colorful kite flown over an urban area.
+000000256192 a bright yellow police motorcycle parkd in the road
+000000082812 A group of commuters standing around a train station.
+000000353180 Over ten men pushing a charter bus with a woman watching.
+000000193245 A crowd of people standing on a field flying kites.
+000000297353 The man is standing on a bed playing guitar.
+000000172946 A group of people standing around holding Nintendo Wii controllers.
+000000105249 White toilet and sink in white tiled walled bathroom.
+000000458768 A KITCHEN IN THE ROOM WITH WHITE APPLIANCES 
+000000088432 A road between the buildings where a big car is parked.A water hydrant is also seen.
+000000242724 Two women ride horses in the water at the beach.
+000000052507 a person holding a surf board in a body of water
+000000015517 Buses parked on a road outside a large bus station. 
+000000124975 Two zebra standing next to each other on a field.
+000000346703 A person that is putting fire on a cake.
+000000235857 Three cows two black and one brown grazing in a green field.
+000000535156 An elephant rubbing its head against a large metal pole.
+000000552371 Two horses in an enclosed area during the day.
+000000016439 A lap top computer is open and on and is sitting on a desk next to a glass of liquid and a lamp.
+000000521052 a brightly painted truck on a city street
+000000541123 A mens doubles tennis tournament on a grass court.
+000000557172 A bathroom with a sink, tub, shower head and mirror.
+000000405205 Buses and tractor trailers are parked together in different directions.
+000000024144 A cheese pizza from a store in a box. 
+000000303893 A boy and a girl sitting on a sidewalk eating.
+000000137576 A group of cows in a fenced in area.
+000000308430 A pan of stir fried vegetables being cooked. 
+000000463618 A woman holding a baby next to a caged bird.
+000000159311 Two zebra stand near bushes and tall grass.
+000000062025 A bathroom contains a toilet and sink below a vanity.
+000000144706 The window showing a traffic signal is covered in droplets of rainwater. 
+000000095843 A blue and white double decker bus on side of street.
+000000579091 Some green broccoli and hams are on the plate. 
+000000450559 A boy skate boarding high into the air.
+000000443969 A small child stands in a shopping cart with an umbrella.
+000000472623 A man balances his bicycle on one wheel
+000000164883 A man has his face up to a video game monitor.
+000000181969 An animal, curled into a ball, takes a nap.
+000000457848 A line of parked motorcycles with the edges of image blurred.
+000000365766 An unadorned kitchen with oven, sink, cabinets, microwave, wood floor, and a window.
+000000526751 A small boat is beached on the shore.
+000000183709 A group of people standing next to a crosswalk.
+000000534270 Man and woman with umbrella hats sitting on top of a bridge. 
+000000110972 A gigantic black bear roams around with his head hanging low. 
+000000018737 a motorcycle parked on the road in a desert
+000000395343 A beautiful picture of a table full of flower vases. 
+000000289594 An old man walking down a sidewalk in black and white.
+000000030494 A black and white dog examines something on the ground.
+000000350002 A woman poses in a dress next to a hair dryer. 
+000000233370 Someone who is laying down under a suitcase.
+000000153510 Food is shown with rice, broccoli and other things on a plate. 
+000000047571 A group of people that are on a horse.
+000000203629 The man is giving a thumbs up while on his phone.
+000000084431 a single slice of pizza with ham slices on it  on a plate
+000000208901 A modern jet airplane flying up in the sky
+000000222299 A floor filled with the contents of a woman's bag.
+000000112634 a close up of a giraffe on a dirt ground 
+000000307145 A KITCHEN WITH A SINK AND TABLE IN IT 
+000000107339 A pretty young lady standing in front of a couch behind a wooden table.
+000000412286 A man running to hit a tennis ball with his racquet. 
+000000079229 a dog following a man on his horse in a field
+000000138856 A loaf of bread hits on top of an oven.
+000000261732 A man jumping up on a blue tennis court with a black tennis racket in his hand. 
+000000418961 A clock mounted to the side of a building.
+000000359540 Baseball batter swinging at a pitch and the catcher trying to catch the ball.
+000000428454 A man on a snowboard para sailing in the snow.
+000000473237 A young girl eating a slice of pizza.
+000000204329 A colorful surfboard stands erected on the sandy beach.
+000000213445 The cat is sitting inside of a bowl on a table.
+000000060835 A dog that is sitting down inside of a cage.
+000000419379 A wooden table topped with carrot cake donuts.
+000000045596 a bunch of parked bikes sit between two buildings 
+000000074457 A man holding a surfboard on a beach with a pier in the background.
+000000397351 A man holding a bunch of carrots behind a display of carrots.
+000000559160 A kid walking with his skateboard in a skateboard park 
+000000341973 Two girls pose for a photo holding doughnuts.
+000000398237 A baby is holding some poles in the middle of a grass field. 
+000000167240 A vase with red flowers on it next to a map.
+000000371749 Two girls holding Wii controllers, playing video games in a living room.
+000000006471 A man at bat waiting for a pitch with a catcher and umpire behind him and players from the opposing team in the dugout.
+000000488673 a male and female a cake and a candle
+000000567640 A group of men running across a soccer field.
+000000193494 a bed surrounded with plastic for walls and ceiling 
+000000082085 a train that is on a rail road track
+000000343937 A woman wearing a white hat and white coat snowboarding down a slope.
+000000400161 four different controllers in a couches controller case and a person
+000000128372 A transit bus with its doors opened as passengers board.
+000000151051 A man on skis sliding down a snowy mountain.
+000000248752 A baseball player is getting ready for a pitch.
+000000311928 A little boy gets ready to hit the ball with his bat.
+000000359855 Beat up blue suitcase lying on side of road.
+000000573094 A room with chair, bicycle, television and a Christmas tree.
+000000042888 A corner street sign with a tow sign and a art piece
+000000194216 a couple of bears stand on a tree log 
+000000061584 A man and woman are sitting on a statue of a horse.
+000000394611 Two tall giraffes graze on bushes in an open field.
+000000297562 A baby sitting in a chair getting a haircut at a salon.
+000000054592 Two people hiking up a mountain with their ski's.
+000000000785 A woman posing for the camera standing on skis.
+000000286708 Cat comically wearing a beanie hat looking agitated.
+000000214200 People don't seem to be able to pass a Stop sign without defacing them.
+000000328117 Donuts in a staging area with a pair of tongs near them.
+000000524456 A man sitting at a table using a laptop computer.
+000000142620 A man eating out on a table and bench.
+000000067310 A skateboarder is doing tricks on a ramp.
+000000172595 A purple chair is next to a desk with a computer and a laptop.
+000000544052 A couple of storm troopers standing next to each other.
+000000545129 Three zebras standing in a sandy desert area.
+000000114049 Passengers with luggage in an airport "limo" bus.
+000000507037 A group of people visit and eat food outside city hall.
+000000322724 A man is standing on a snowy hills with his skiis.
+000000290843 A person sitting on bed behind an open laptop computer and a cat sitting beside and looking at the laptop screen area.
+000000403817 The grey and white cat stares up near a laptop.
+000000177861 two people walking with umbrellas down the sidewalk in the rain
+000000288685 A dog is herding sheep in a field.
+000000119516 A beautiful woman standing on the side of a rad next to a street.
+000000270677 A boy posing for a picture with his skateboard.
+000000361730 A park with kites flying in the bright blue sky.
+000000338219 A man looks at a motorcycle amongst others in a field.
+000000334399 A man gazes at an intricately designed clock with roman numerals.
+000000255912 Meat sandwich on a roll with greens on a plate.
+000000545594 The person with the tye dye shirt is choosing a seat.
+000000018193 A person eating from a plate on their lap while sitting in a camping chair.
+000000404601 A sign advertising The Cave Restaurant and Lounge
+000000282296 a kitchen with a refrigerator near a pair of shoes
+000000095707 A table topped with a train styled cake.
+000000529528 A mixing bowl and beaters with batter on them
+000000255912 There is a sandwich and lettuce on a plate
+000000240754 a black sheep and others in a corral together
+000000396518 A woman sitting on the bow of a boat using a pole to pilot the boat.
+000000335954 this is a salad in some glass bowls
+000000257084 A guy on a skateboard about to go down a ramp.
+000000229747 A airplane that is flying in the sky.
+000000229553 A man sitting on top of a skateboard on a street.
+000000563267 Two guys in an office walking towards the chalk board.
+000000268378 A group of people sitting at a table enjoying food.
+000000273551 A row of motorcycles parked on the side of a road.
+000000512476 A bowl of food sitting on the edge of a sink.
+000000158660 White bowls of healthy fresh fruit on a counter. 
+000000290248 A crowd of people standing in the middle of a street.
+000000005586 a tennis player with a racket on a court 
+000000514586 A young boy holding a baseball bat near a batting cage.
+000000383921 A herd of elephants in rocky area next to water.
+000000448365 Young skateboarder displaying skills on sidewalk near field.
+000000529966 A couple of glass bowls or oranges sitting on a metal counter.
+000000258883 Someone is working on grilling some food on a grill.
+000000463647 Two signs with arrows pointing to each other for detour
+000000370999 A coin with a portrait and some rings on the table
+000000089556 Several cars parked in a lot in a city.
+000000453708 A woman carries a box of bananas down a crowded street.
+000000319184 Two people playing a game of frisbee in a field.
+000000526751 A boat stowed up on the beaches on sand.
+000000296634 Three white bowls of food that include soup and vegetables.
+000000488710 A smiling girl standing beside a sign that says "Princess Parking Only".
+000000455085 A bus stopped on the side of the road.
+000000180792 A small bathroom features a toilet, window and piping.
+000000445999 A woman brushes her teeth with foamy toothpaste.
+000000404128 An old plane sitting on the grass beside the runway.
+000000226171 A computer monitor sitting above a computer keyboard.
+000000189078 An assortment of different fruits laying in two rows on a tray
+000000216497 a kitchen with a stove a table some chairs and a sink
+000000082821 A group of boats sitting next to a dock near a boat house.
+000000205333 A pair of skiers on the top of a ski slope with clouds and large mountain in the background.
+000000331352 A tan toilet and sink combination in a small room.
+000000052891 There is a dog holding a Frisbee in its mouth.
+000000327890 photo of three benches for sitting facing each other
+000000462614 A bathroom filled with fixtures and a shower.
+000000364126 The man is riding the surfboard on the water. 
+000000551350 The picnic table under the tree is empty of people.
+000000152686 Tennis player serving  a tennis ball on a clay court
+000000030494 an image of a dog next to a tree playing with a shoe
+000000185157 A one way sign with a yellow blinking light on top.
+000000016228 A trolley that is being pulled by a white horse.
+000000480212 A soda, cell phone and pen sit on a desk near a computer keyboard.
+000000108253 Some cinnamon toast and eggs that are half eaten on a table.
+000000349302 The two giraffes are standing in the dirt.
+000000334371 Double-Decker buses line up at a bus stop.
+000000125405 Two dogs running and playing in the sun.
+000000142585 A couple of people sit on a motorcycle next to a white car. 
+000000534639 The metro train is painted white with blue lettering.
+000000528314 A person on skis with one standing vertically
+000000233370 A person lying on the ground with a suit case on top of them.
+000000462576 a plate of breakfast food sits on a table
+000000122166 The cars are travelling behind the guy on the bicycle
+000000473974 A group of zebra standing in the tall grass.
+000000293794 Bunches of bananas are piled up at a market with some shoppers looking around.
+000000146457 two children being entertained by preparing pizza at home.
+000000318908 A small white dog looking into a camera.
+000000514508  A man is holdng a mic while looking onto a screen.  
+000000013177 a man in a green shirt a rim and motorcycle
+000000016010 African animals placidly grazing in a park enclosure.
+000000205105 A man holding a tennis racquet on a tennis court.
+000000246968 A young woman standing in the kitchen pours from a large measuring cup.
+000000027620 a desk with a computer sitting on top of it 
+000000577735 a small copper vase with some flowers in it
+000000489305 a small bowl and a very big bowl full of fruit salad
+000000098287 A young batter is awaiting a pitch during a game.
+000000535608 THERE IS A PERSON ON THE BEACH 
+000000249643 A train that is riding on tracks through a station.
+000000107226 Men race bicycles on grass pulled by sleigh dogs.
+000000351331 a small fully cooked pizza resting on plate next to a fork
+000000284296 an image of a herd of giraffes in the water
+000000190637 A small child lounges with a remote in his hand.
+000000393226 A busy intersection with an ice cream truck driving by.
+000000456143 A few baskets of food, including a pizza, some vegetables and dip, and a sandwich
+000000074457 A  man standing on a beach while holding a white surfboard.
+000000306139 A group of people sitting at a table using computers.
+000000332570 A person holding a cell phone showing a text.
+000000007795 Two beds are in a hotel room type setting.
+000000400082 a close up of a paper box holding hot dogs and french fries
+000000013659 A man sitting in front of a laptop computer in an office.
+000000214869 The woman is pointing to a phone in a cover.
+000000180135 A young skateboarder talking into a microphone 
+000000322895 A room with a full wall of windows with multiple sofas and chairs.
+000000170955 A president in a nice suit walking near a building.
+000000131556 a man wearing skis is posing for the camera
+000000304812 A person walking across a beach next to the ocean.
+000000127517 A row of surfboards sticking out of the sand sitting next to each other.
+000000568439 a couple of buses that are parked next to each other
+000000312406 a table that has some bananas and a cup on it
+000000162035 A person walking down a street past a teddy bear sign.
+000000032861 A man holding an umbrella in a  hallway.
+000000364166 Two zebras standing side by side in a field
+000000114770 A large gun, a jeep, a motorcycle, and a warplane are in front of a war backdrop. 
+000000459437 Four sheep standing against a fence looking at a man with a red bowl.
+000000132544 A girl attempts to hit a ball with her bat while playing baseball.
+000000580410 A small den with a  couch, a bag on the floor, a book shelf, and a wire going from the wall to near the window.
+000000020333 Nicely dressed little boy gets ready to play in the dirt.
+000000301061 A man pulls an elephant out of a truck.
+000000368294 A woman wearing a pink hair towel working in a factory.
+000000373705 A red and yellow fire hydrant sitting on the side of a road.
+000000232684 A person sitting next to a laptop in a dark room.
+000000443969 A little girl is riding in a shopping cart while holding her umbrella.
+000000426268 A large long train with man inside on the track.
+000000569030 Girl sitting on a bench next to a fire hydrant that is being used.
+000000221693 A dog is in the grass with a frisbee in its mouth.
+000000500464 The bathroom with the white tile has been cleaned. 
+000000393226 The man is walking behind the concession bus.
+000000186624 Someone is standing near a train that is parked. 
+000000313182 A bus sits diagonal from the curb next to a building and people.
+000000255165 a desk filled with some paperwork, a laptop and a computer with two monitors
+000000169169 A sign announcing the season a people walk on the sidewalk.
+000000404484 A woman in a living room with a dog. 
+000000194940 Three bowls filled with vegetables on top of a table.
+000000335177 Suburban street scene showing a curve, houses, and cars in driveway.
+000000187249 A television set is on next to a bottle of water.
+000000243148 A young child is sitting in the snow 
+000000356094 A man is up to bat on a baseball field.
+000000452321 A bus parked in front of other buses.
+000000197870 a small chicken is standing by a bench
+000000013546 A young man riding a skateboard on a stone wall.
+000000283318 A store on the corner with an American flag flying from it's side.
+000000468124 A double-decker bus turned in to a traveling showcase for a University
+000000501243 Three zebras are standing together in a line. 
+000000058029 A small clean bathroom that is well lit. 
+000000468965 some young people are flying a kite in the air
+000000369503 A woman standing in an un-lived in kitchen
+000000531135 A baseball pitcher throws the ball as the batter prepares to swing.
+000000102331 A boy on a motocross motorcycle flying through the air.
+000000538236 A tray filled with croissants with hotdogs in the middle
+000000229997 A bear sitting in the grass surrounded by five vultures.
+000000362520 Small child and protective gear jumping on a skateboard. 
+000000430048 A business window with an "ATM INSIDE" sign,  reflecting a clock tower.
+000000388215 A man reaches into the air with a tennis racket, in an attempt to hit a flying tennis ball.
+000000390301 A white and green rusted fire hydrant spewing out frozen water.
+000000359135 a close up of some elephants with tusks
+000000338560 A fire hydrant on a sidewalk of a city 
+000000250137 A young redheaded woman in sunglasses and black tank top holds a black leather purse and a white umbrella.
+000000504074 Two women sitting together in the back yard
+000000003255 Skiers look at the snow that they skied down.
+000000465179 Woman dressed in black, smiling and brushing her teeth
+000000184400 a colorful train going along an elevated track 
+000000233139 Two clocks are hung on a wall behind a glass sign.
+000000532071 A big black bear laying in the grass surrounded by trees.
+000000549930 A man and a women walking under an umbrella.
+000000516601 Many people skiing down a snow covered slope.
+000000460683 A man wearing a blue tie with the ten commandments on it.
+000000347930 A dog is sitting on a couch with a ball and two cushions in back.
+000000005477 A large golden airplane is on the runway.
+000000279730 A sandwich is being eaten and sodas are on the table.
+000000046872 a man standing next to a truck with some stuff on it
+000000215723 Two men smiling and wearing yellow jackets behind construction markers.
+000000459500 The statue of two pelicans has a clock above it.
+000000323202 Small white toilet with seashells sitting on top of it. 
+000000170955 A man in suit and tie walking in front of a white building.
+000000163117 A group of people are flying kites on a crowded field.
+000000008021 A presenter projected on a large screen at a conference.
+000000145020 A person who is in a duck costume.
+000000044195 Two people standing atop a snowy hill with snow boards in hand ready to head down the slope
+000000157365 There is no image here to provide a caption for.
+000000046463 A sandwich appears to have a mouth like creation.
+000000576566 A man flying through the air on top of a skateboard.
+000000183104 Two zebras stare at the camera in a zoo.
+000000464251 a nice bird standing on a bench gazing at
+000000423617 a wet road busy with traffic like a taxi and bus
+000000377000 A cat sitting on a sill looking out a window.
+000000326462 A person holding a donut with a napkin.
+000000144932 A tugboat is pulling another boat in the middle of a water.
+000000248111 the kitchen is white with brown counter tops and a white fridge
+000000052565 A cow standing on the beach by a body of water.
+000000448448 GIRAFFES EATING THE SHRUBBERY OR THE TOP OF TREES
+000000187745 A toy plane flies through a cloudless sky. 
+000000504635 Three zebras are standing in the grass by the trees.
+000000084752 A bred and silver plane resting on stands outside.
+000000174231 A cat laying with a laptop on a table.
+000000388927 Hand holding an electronic component with a clock on it.
+000000546826 A close up of a red pair of scissors and a green sharpie marker. 
+000000554291 A kitten climbing out of a blue bowl.
+000000068409 Black and white photograph of young schoolboys. 
+000000419379 Four donuts topped with icing displayed on metal trays.
+000000076416 A double decker bus that is parked in front of a big building. 
+000000374551 There is a tiny bird on the branch.
+000000051326 Pumpkins are made with different faces and one has flowers. 
+000000080022 A tennis player returns the ball using a backhand.
+000000197022 A cooked pizza pie on a tray with another that is uncooked.
+000000301718 two people seated at a dining table with a plate of food before them
+000000321333 two children one holding a bear and one holding a cell phone
+000000275727 Yellow and grey train near platform at railway station.
+000000290592 a bunch of sheep are standing behind a fence
+000000087742 A glass vase full of feathers on a table in front of a window.
+000000204871 A red fire hydrant with graphic art decal.
+000000359781 A giraffe standing in the dirt near trees and rocks
+000000378605 A donut is on the table on a white saucer.
+000000480842 Two giraffes are towering over the plants under them.
+000000574315 Cat in front of a laptop looking up from the screen
+000000325527 a close up of a stuffed animal on a cardboard box
+000000193162 a cow walks as a man on a quad follows it 
+000000512836 A woman is walking two dogs in the snow. 
+000000333402 Young male boarding a city bus at a bus stop.
+000000023023 A collection of suitcases lined up in front of hotel curtains.
+000000314251 A group of motorcyclists drive down a tree lined street.
+000000340272 Three sheep eating grass near a water source.
+000000512476 A bow of food set next to a sink inside of a kitchen
+000000513041 A peeson at a table is eating a small pizza
+000000182021 A man holding a baseball bat while standing on a baseball field.
+000000163611 A pizza sitting on top of a metal pan on a table.
+000000097585 A wall with different types of decorations of art pieces. 
+000000525322 two stunt airplanes touching their wings as the fly
+000000469246 Person waiting on a train platform near a train.
+000000214720 A smiling guy sitting at a table with some food.
+000000286908 four different pictures of a variety of foods
+000000402992 Cattle lie in the grass to chew their cud.
+000000420916 A group of people sitting on a bench near a lake.
+000000082688 Some people are enjoying themselves playing game. 
+000000523241 Street signs on a pole on a busy intersection. 
+000000449661 A room with many medical supplies in it. 
+000000111609 A man is sleeping with the covers pulled up high.
+000000068765 A keyboard, mouse, and wires on a desk.
+000000122606 A front shot of a Crosstown bus is parked on the side of the road. 
+000000478393 A bed with a cat sleeping on it next to a teddy bear and another cat.
+000000187745 a small yellow Cessna plane flying on a clear day
+000000439715 A man in uniform riding a horse by a fence
+000000173302 A restaurant with red stools next to a bar  and a blue table.
+000000110042 A game at an outdoor fair called "Ball In The Bog"
+000000272136 A large propeller plane mounted to the ceiling of a building.
+000000051314 A man that is surfing in the water.
+000000113403 Three teddy bears laying in bed under the covers.
+000000163057 Many people walk through a park as few kites fly in the air.
+000000190841 A man attempts a trick after riding off of a low ramp.
+000000440507 a blue suitcase some shoes and a wooden table
+000000562443 herd of zebras grazing on some green vegetation 
+000000041990 Three young people on skis pose for a picture.
+000000133567 A train is passing by while an onlooker is standing next to track.
+000000161044 Two airplanes are flying in the cloudy sky.
+000000570169 a teddy bear holding a stick next to  a block toy
+000000528705 A brown bear sitting on top of a persons back.
+000000356612 A group of cows walking in front of several cars.
+000000042889 A teddy bear sits by a keyboards and microphone.
+000000231508 A baseball pitcher who is in mid pitch.
+000000433515 A large colorful kite is being flown over a field.
+000000271457  A bench sitting on to of a field of tall grass near water.
+000000119088 A man in shallow water bending over to pick up a frisbee.
+000000041872 A room with a bed, a clock, a lamp, a fireplace and a television.
+000000236599 A fence with a bunch of kites attached to it
+000000369310 A toddler boy is on a white pad playing a video game.
+000000530099 A cat is on top of a car in the dark.
+000000217948 Three bears stand together near a fence. 
+000000565469 A tall lighted clock on top of a brick tower
+000000229753 Two giraffes are standing next to a tree trunk.
+000000448810 Street signs going in every direction in front of a yellow building.
+000000065074 A bed covered in blankets and pillows next to a window.
+000000497628 A wooden bed with white bedding and pillows
+000000524108 There is a group of motorcycles that are parked together.
+000000435205 A hand that is holding a stuffed animal and a pencil.
+000000253452 A hearty fish dinner, with a side of onion rings and coffee.
+000000279278 a person riding a skate board on a city street
+000000122217 Two longboarders coasting, crouched, across a downhill road
+000000327592 A woman with glasses eating a piece of cake.
+000000287714 A clean hotel bathroom is pictured in this image.
+000000022755 A rear view mirror sitting on the side of a yellow bus.
+000000419098 A silver ball chain is looped through pink rings.
+000000058705 A man and woman at a party dressed in costumes.
+000000294162 A view of a small laptop with a mouse next to it.
+000000038048 The red and white fire hydrant is embedded in the street.
+000000154358 A woman sits in one of the easy chairs in the big living room. 
+000000450488 Small bedroom with a bed and dresser with a small monitor on top of it. 
+000000179642 A player attempting to cool off during a tennis match.
+000000255401 A bathroom filled with toilets and a tub next to a sink.
+000000570688 Kites flying over a busy beach area on clear day.
+000000278006 A street sign marks the intersection of 3 streets, while three housed stand in the background.
+000000222991 An umbrella and rain boots in a corner 
+000000070229 a park bench that has a teddy bear on it
+000000156071 A young child mutton busting at a rodeo event
+000000261116 a table with a large cake and small plates of cake 
+000000412894 People walk down a busy city street, with traffic light.
+000000226147 A woman is taking a picture of food with a mobile device.
+000000102356 A woman riding a green motorcycle with a side car.
+000000094157 A group of people in a park with a tall clock tower in the background.
+000000545219 A double decker bus with neon lights at night.
+000000389532 A man is standing up, taking a shot of the water, while a pigeon looks on. 
+000000399560 A cat is sitting in a partially closed laptop.
+000000378673 A man riding a skateboard down the side of a ramp.
+000000414261 A bear cub struggles to scale a boulder.
+000000509824 A very tidy living room with a white couch with pillows on it.
+000000359540 A baseball player is swinging a bat at a ball while the catcher tries to catch it.
+000000201646 A young man playing with video game controllers.
+000000544306 Big Ben set against a grey cloudy sky.
+000000377113 A person on some skis in the water.
+000000047801 A bald headed man eating a slice of cake.
+000000021167 some people standing in a room in formal wear
+000000210708 A mother elephant gives her baby a bath.
+000000200152 Dark surrounding a lit street sign and American flags.
+000000130586 A boy running while holding onto a kite string.
+000000554735 A person grabbing a slice of pizza from a pizza box.
+000000129054 A herd of zebra are grazing in a grassy meadow.
+000000365095 The man is playing soccer in the field. 
+000000043314 Skier in mid air following loss of balance on mountaintop.
+000000376310 A small bathroom with a mirror and sinks.
+000000147725 there is a bus coming up the street next to a car
+000000295231 The sheep with most of it's wool shaved off stands near the herd.
+000000530836 An old ad is showing a retro kitchen.
+000000410456 a surfing instructor teaches students with surfboards on a beach in front of large hotel buildings.
+000000251140 The book is about fixing your bike. 
+000000427055 A pole that has a clock on the top of it.
+000000426297 A zebra looking down at a grass and dirt filled ground.
+000000345941 A dessert topped with ice cream, and strawberries sits on a plate.
+000000384136 A woman points a hair drier like it is a gun.
+000000096549 An Egyptian airlines plane landing at an airport.
+000000488710 A woman that is standing near signs smiling.
+000000096960 A polar bear swimming in water near a rock wall.
+000000462629 Old appliances outside the back door of a repair shop. 
+000000445834 Some guys are getting on a bus wearing blue.
+000000086755 a couple of people skiing down a snowy slope
+000000427160 This is a picture of a boy getting ready to make a play.
+000000191614 A horse standing in a lush green grass field.
+000000549738 A photo of a kite is a town in the background.
+000000368684 A large flat screen TV sitting in a living room surrounded by furniture.
+000000231088 6 open umbrellas of various colors hanging on a line
+000000308328 A group of men loading luggage onto an airplane.
+000000235064 A bear doing a tightrope act on a stone wall.
+000000416269 A passenger sign on the tracks at a station.
+000000572678 An upscale living area containing white and glass furniture with brown accents.
+000000051008 A cat looking like it is using a laptop 
+000000227511 Rows of cars parked on the side of a city street going uphill. 
+000000511647 an image of a boy that is flying a remote control plane in the air
+000000236721 A breakfast was delivered to a hotel room on a tray.
+000000211069 Two giraffes walking in long grass near short trees.
+000000565045 A shower in a person's bathroom in their home.
+000000148719 A red truck parked in grass next to other trucks.
+000000389684 Guy walking pass a bus sitting at the curb 
+000000485237 Two jumbo jet air planes on a runway facing each other.
+000000376307 there are two people that are standing in a farmers market
+000000537270 Three students check their cell phones in a classroom.
+000000500826 A view of the street signs "W 122 St.", "Seminary Row", and "Broadway" in front of an old red brick building. 
+000000552883 A view of a kitchen with cabinets, a sink and a clothes washer.
+000000310200 A pastry with sugar on the outside that is cut in half
+000000354307 A person with several severe cuts on their face and head.
+000000335954 a close up of two cups of fruits on a counter 
+000000221502 an image of a park bench with flowers next to it
+000000120777 A tray a drink, a salt shaker and two plates that contain sandwiches and potato chips.
+000000286907 A bathroom has some blue walls and toilet.
+000000001353 Several children on a small  indoor kiddie train.
+000000044195 A couple of people standing on top of  a snow covered slope.
+000000321118 A pair of  hairy objects sitting on a laptop next to a mouse.
+000000572956 A row of motorcycles parked on the side of a street.
+000000376856 a tall church tower with a clock at the top
+000000283785 A man is wearing a costume and speaking to children.
+000000292908 there is a surfer at the beach riding waves in the ocean
+000000455981 A toilet with some toilet paper and a plunger
+000000043581 A restaurant table with a plate of vegetable pizza and garnishment
+000000100238 A family standing next to each other while holding frisbees.
+000000151051 A man on skies is cutting through the snow
+000000170278 Dog smiles while sleeping in a bed room.
+000000370900 Four stuffed bears have crowns on their heads.
+000000121497 Several people stand at a table that appears to hold some machinery.
+000000364884 A person in a snow sporting event  is going rapidly down this show chute-type course as people watch from the sidelines. 
+000000428218 THERE ARE PEOPLE THAT ARE STANDING IN THE GRASS
+000000493284 The family of elephants is looking for some good food.
+000000286507 a train in a rural area covered with heavy snow
+000000303713 A man swinging a tennis racquet on top of a tennis court.
+000000476491 an alley between two buildings with a sky in the background
+000000391144 A group of adult elephants and one baby elephant drinking water.
+000000471991 A church with pews running along the sides with a clear center isle.
+000000551304 A white and gold fancy toilet and stainless trash can.
+000000357816 Cole Ballew at the bat for his baseball team
+000000399764 An old man walking a big brown cow around.
+000000031050 a vase with flowers on a display near a wall
+000000482275 A newly married couple cutting a cake in a park.
+000000430073 In the Asian market, one can find residents, tourists, and domestic animals. 
+000000361147 A tennis player is standing on a tennis court with a racquet in his hand.
+000000499775 A bus is stopped along a street with no passengers waiting to board.
+000000361621 A yellow cat sits in a small sink.
+000000163257 A man holding and dancing with an umbrella.
+000000244592 Three zebras outside with two grazing next to each other.
+000000485844 a lady that has a bowl in her hand
+000000479030 A long red and white train travels near many other tracks.
+000000267903 A dirty bathroom that has broken wood and pipes.
+000000231879 A woman looks at a child being held by a man.
+000000129135 A train that is on the tracks sitting in front of a bench.
+000000219485 A grey cat standing in a window with grey lining.
+000000257169 A person is holding up a hair dryer in the bathroom. 
+000000562197 Three pictures of plates with salads, including beans, fish and vegetables.
+000000109827 A teddy bear leaning against a tree next to the road.
+000000142971 A man teaching kids how to ride a surf board in the water
+000000402774 A woman holding an umbrella while walking in the rain.
+000000454750 A larger zebra resting its head on a smaller zebra.
+000000273712 Large black two door refrigerator in an apartment under renovation.
+000000026564 There is a desktop computer and a laptop sitting on an organize desk in an office .
+000000237928 A small kitchen is shown it has many appliances.
+000000518213 a large bull with horns standing in a grassy field
+000000131386 A row of four trick Navy Planes flying in sequence.
+000000357816 A boy swinging a baseball bat at a ball.
+000000091921 a grassy field with two zebras in it
+000000126592 a red stop sign with a street sign attached
+000000112997 A person holding a walkie talkie in their left hand.
+000000311789 A cat that is laying down on a desk.
+000000109798 A variety of doughnuts sitting on a paper plate.
+000000260261 A man cutting a cold coconut on the back of a truck
+000000378139 A large passenger boat is floating down a canal.
+000000095899 A red stop sign next to a blue street sign.
+000000064084 a plate with some pieces of bread on it 
+000000304396 Salmon-colored clock tower with Florida architectural features including a window's walk.
+000000555050 A car driving in an intersection, past a furniture shop. 
+000000000885 a male tennis player in white shorts is playing tennis
+000000079588 Some people standing in front of a large building.
+000000370677 Some girls in colorful shirts standing by some pastries.
+000000271402 Two young girls with a tennis racket and a scooter.
+000000479732 A gourmet sandwich with fresh vegetables in a takeout box
+000000099242 An older man downhill skiing down a slope.
+000000520659 A white tent with a row of surfboards sitting under it.
+000000182155 Two girls with brown hair play with Wii remotes.
+000000343466 An empty kitchen in a house with a window in the corner.
+000000127263 Two men on opposite teams jumping for a Frisbee.
+000000288762 A salad made with yellow pepper strips and green sprouts sits on a square white plate.
+000000541123 Four men on a tennis court playing a doubles tennis match.
+000000250766 Meat and a salad with knife and fork on a plate.
+000000227187 A small black and orange bird is perched on a table with banana peels in it.
+000000062692 a woman is out on a court playing tennis
+000000112798 a close up of a cat laying on a desk
+000000123585 A pigeon outside with his head in a piece of bread. 
+000000304180 A woman in a short red dress is holding her skateboard while getting food from a take out window.
+000000317999 A baby is looking at a book while sucking a pacifier.
+000000140203 a woman holding a dog and a truck
+000000277005 Several people are leaving a bus station with a bus next to it as a person rides by on a bicycle
+000000507037 a gathering outside on the lawn of city hall
+000000347456 Mother and child elephant in a fenced pastoral field.
+000000513484 A brown bear enjoying the sun and a little water at the zoo
+000000519338 An old rusted train sitting on abandoned train tracks.
+000000410878 a close up of a motorcycle parked on a dirt road
+000000111036 A kitchen stove laying in a dumpster. 
+000000312340 A black and white cat relaxes atop a bunch of leaves
+000000347664 A herd of cattle walking across a sandy beach near the ocean.
+000000156643 Person cutting cake at a theme restaurant characters in background.
+000000360097 A pile of luggage sitting up against a white fence.
+000000090155 This is a extremely long freight train heading south.
+000000534639 A white train traveling down the tracks with expo center on a sign on the front. 
+000000192699 Two people wearing helmets while riding their motorcycles.
+000000094751 A view from the street overlooks the ocean.
+000000325031 A sculpture of an elephant, decorated with images of other animals.
+000000100624 a woman standing on a sidewalk while using a cell phone 
+000000234660 Two white passenger trains are parked on the tracks, side by side.
+000000033707 Two giraffes standing outside near a building.  
+000000134882 A cat laying on a bed in front of a book shelf.
+000000516316 Several zebras eat the green grass in the pasture.
+000000485844 The woman is placing a turkey into a black pan.
+000000496722  a mail box standing in the street outside apartment buildings
+000000540962 A grey and white living room view from above
+000000178982 there are two people on motorcycles approaching a tunnel
+000000037740 Not the biggest workspace in the world, but it works
+000000291619 There are two men playing frisbee on a field 
+000000042888 A street sign stating no parking and tow away zone.
+000000349184 A person sits on a wooden bench facing blooming trees.
+000000200667 Cow looking into the bottom of a machine.
+000000566758 The yellow school bus has stopped at the bus stop.
+000000186929 Lots of oranges are sliced iin several sizes..
+000000559160 Teenager returning with skateboard after skate run at the skate park.
+000000237118 A man taking a picture of himself in a mirror 
+000000036660 The cell phone is broken at the feet of someone in flip flops.
+000000257084 A person standing on a skateboard on top of a ramp.
+000000215259 Two women playing a game with a Nintendo Wii controller.
+000000435003 A computer workstation with a laptop and two additional monitors.
+000000229753 Two tall giraffes are next to bare trees.
+000000451879 A baseball player throws the ball to his team mate as they try to get another player out.
+000000482319 a man holding a kid and a mother smilling
+000000353027 A person picking up an under cooked, floppy slice of pizza.
+000000248980 A blue plate with a green salad on top of it.
+000000190676 Men lined up an a runway in a desert greet an arriving jet plane.
+000000357567 A toilet, counter and shower in a bathroom.
+000000428562 A sheep stands near other sheep in a closed off pen.
+000000088462 A large yellow school bus stopping on a road.
+000000434479 a slice of food with two doll figurines standing next to it 
+000000468124 A double decker bus sitting on a green field.
+000000003156 A man attempting to lift up a toilet off the floor.
+000000011511 A sculpture of two women stting on a bench with their purses on the ground while people standing in a line behind them. 
+000000329041 A group of brief cases are piled up on the sidewalks.
+000000368752 A little boy that is sitting in front of a table.
+000000555597 Many cars parked on a city street with tall buildings in the background.
+000000031296 A group of people, some in leftover US army clothing, sit at tables in a large dining room.
+000000117908 There is a cat looking in a refrigerator with it's human.
+000000158227 A young boy in a baseball uniform getting ready to swing at a pitch.
+000000162130 A park bench next to fence and trees by grassy field.
+000000460682 A horse that is sitting down in an open field.
+000000101068 a man is swinging a baseball bat outside
+000000131444 a man that is in a suite and a tie
+000000332351 so many people on the sea skating and swimming
+000000526728 A man and a woman are standing behind some suitcases.
+000000245576 A yellow cat is on the desk sniffing a computer keyboard.
+000000256518 A white plate topped with food next to a cup of coffee.
+000000470779 Men and woman posing for a picture on skiis and a snowboard. 
+000000430961 A baseball player lunges and reaches back with the ball.
+000000053909 A man holding a phone that has a picture on it of a man holding a phone.
+000000050149 A group of bunches of bananas are shown hanging together.
+000000215114 a man is hiding inside of a refrigerator 
+000000475387 A train traveling down tracks near a train station.
+000000419379 A carrot cake doughnut displayed on a counter. 
+000000166563 An eyecatching sign shaped like an elephant holding a billboard.
+000000009590 A group of people sitting around a wooden table.
+000000357888 A man holding a yellow kite while standing in a room.
+000000279541 A buffet with many pizzas lined up behind glass.
+000000376284 fire hydrant on city street with several parked cars.
+000000292005 A bathroom with white spots on the wall.
+000000163117 A large group of people in a field with kites.
+000000275749 small gathering of people at a friends house for drinks
+000000131938 Cat sitting up with a fake tie around it's neck.
+000000274687 A bicycle leaning against a bed in a bedroom.
+000000409542 A clock that is at the top of a pole.
+000000492992 a cow that has a small white bird on it
+000000257566 A boat traveling down a narrow canal next to a forest.
+000000027186 The little girl is intently playing the video game.
+000000068765 A black and white photo of a computer keyboard, flash drive, and mouse on a marble tabletop.
+000000499313 A pizza on tray cut into six slices.
+000000258541 A man dressed in a shirt and tie standing in a lobby.
+000000019742 A brown glass on the table for display art
+000000389316 Two elephants outside, one being fed, one standing.
+000000098497 A traffic sign that has a picture of a man holding a surfboard on it.
+000000046031 A laptop and a desktop computer sitting on a table.
+000000402774 a person walking in the rain holding an open umbrella
+000000389812 A man is holding a bunch of bananas.
+000000415990 Two men are standing near the pasture of cows.
+000000166478 A man in brown shirt on black sofa with phone.
+000000007818 A table topped with two wine glasses and a centerpiece.
+000000361238 A woman sitting in front of a giant pizza.
+000000047585 A man holds the umbrella over the bride
+000000027982 a toilet with the toilet seat left up and the sink next to it
+000000060449 A very nice looking deck area with a small table and a laptop.
+000000067616 cars parked on the side of the road while people eat at tables outside
+000000263425 A couple of people standing in a yard.
+000000036936 A man and woman sitting on a sofa watching tv.
+000000193245 Many people are flying different kites in a field on a cloudy day.
+000000181859 a cat sitting in the sink in the bathroom
+000000246522 an ornate clock and watchtower on top of a building
+000000031217 Person on tennis court preparing to return a volley.
+000000163290 A giraffe eats a hanging tree limb while a zebra grazes nearby. 
+000000225757 This workstation features three desktop monitors with a single keyboard as well as a laptop computer.
+000000494188 a black bird standing on the floor in the street
+000000249219 A double decker bus driving down the road.
+000000343706 It is winter, and this young woman wearing her cold weaterh coat also has her Smartphone in a handknitted sleeve.
+000000162035 A sign on a sidewalk has a teddy bear on it.
+000000001268 A waterway under a bridge with people sitting down and a woman taking a photo.
+000000501243 Three zebras standing next to each other in a field.
+000000475365 A blue and white train traveling down tracks near a building.
+000000079034 A fire hydrant sits in the middle of the park
+000000119641 Many tourists riding on elephants in a river
+000000079188 Two Giraffe are out in the wilderness looking at the camera. 
+000000245651 A cake with blue, yellow and green fondant stars on it.
+000000530162 Several people standing on a sidewalk under an umbrella.
+000000357081 two cows with spots are grazing in some grass
+000000030828 A person covered with an orange sleeping bag rests his head on a blue back pack as he sleeps on a park bench. 
+000000015597 a male shirtless skateboarder is doing a trick
+000000203931 A professional baseball player takes a swing in front of fans in a crowded stadium.
+000000125936 People in a store that is selling bananas and apples.
+000000128051 A bus that is sitting in the grass near a street.
+000000051961 a coin meter that has paint all over it
+000000519611 A polar bear standing on a large rock with its mouth wide open.
+000000516318 A man on a surfboard riding a wave.
+000000192047 A bathroom with sink, mirror, and lights in it.
+000000416837 A cow is standing on a grassy sloped hill.
+000000477441 A large passenger jet sitting on top of an airport tarmac.
+000000423229 An old fashioned steam engine train traveling down railroad tracks.
+000000042070 A very big city bus on a big street.
+000000488664 A long passenger train traveling over a rail bridge.
+000000394510 A young boy riding a skateboard on a sidewalk
+000000334555 There is a cow in the middle of a busy road
+000000389109 The personal sized pizza is on a table in the restaurant full of people. 
+000000458755 A person petting the head of a cute fluffy sheep.
+000000458768 A brightly lit kitchen with lots of natural light.
+000000272148 A group of surfers on their boards in the ocean.
+000000414034 The little boy is drinking from a sippy cup and holding the remote control. 
+000000343466 A kitchen with a window and bunch of kitchen furniture in it
+000000537964 A red fire hydrant is sitting in the woods near fallen leaves.
+000000389197 A boy is riding the wave on a boogie board.
+000000170099 A man wearing a red tie with his long sleeve shirt and pants
+000000383676 Giraffes inside of their habitat at a zoo 
+000000136915 a guy showing another guy a base ball bat
+000000127624 View of a highway near a city at sunset
+000000019109 A group of motorcycles are parked on the street.
+000000362520 A skateboarder performing a trick wearing safety equipment.
+000000210032 A bird attempting to bite a piece of sandwich bread.
+000000146155 Two  men standing near a bar drinking together
+000000111086 A row of cars parked next to a large boat
+000000109055 A cat sitting between two bikes  sitting next to each other.
+000000377393 a clock on a small tower near a building 
+000000252294 A woman sitting on top of a bed with curtains hanging over it.
+000000565045 Blue bathroom with two white towels hanging by the shower.
+000000007784 A large orange and white kite flying in a blue sky.
+000000427338 A city made out of stone brick with large arches.
+000000423971 a bathroom with a fancy toilet in it
+000000503855 A grass yard area that has several open umbrellas laid on the grass and various people around the roadway area.
+000000082807 a dog that is sitting at a table with some cake
+000000402118 A person performs a jump in the air on a snowboard.
+000000451571 A HALD EATEN CORN AND A SANDWICH ON THE PLAE
+000000328238 a man in a blue shirt playing with a white ball
+000000181303 A child standing outdoors in the yard playing with a kite
+000000467176 The friends are playing a game together in a living room.
+000000222094 Lots of junk on a field of grass next to a stop sign.
+000000297396 A set up small chocolate cakes decorated like trains.
+000000141671 A group of horses are crossing a finish line
+000000007281 Two men smile as they ride horses on the beach.
+000000252701 A young man rides a surf board on a small wave while a man swims in the background.
+000000374369 a person riding skis on a snowy surface 
+000000333237 A bedroom with dresser, desk, and bed in it.
+000000396200 Young adults riding on city sidewalk with skateboards.
+000000471869 a tan colored teddy bear sits on a crochet covered chair.
+000000000776 three teddy bears giving each other a hug
+000000058655 A clock on top of a signpost stands in front of a building.
+000000063047 A man with glasses and earbuds holding a cup and a paper bag.
+000000212166 A small child eating a plate of food with their hands. 
+000000259097 A young man in a red jacket jumping for a Frizbee in a field.
+000000529528 A bowl that has some kind of a batter in it near a mixer.
+000000578500 The room has red walls with a blue chair
+000000074092 A woman hitting a tennis ball with a racquet.
+000000331799 A woman that is handing fruit to someone else.
+000000573391 A bear is seen walking in a forest in a blurry photo.
+000000268375 A goup of elephants in an enclosed area with a lot of trees behind them.
+000000554291 A kitty cat jumping out of a blue bowl 
+000000423617 a number of vehicles driving down a road
+000000382111 Two people ride matching cream-colored horses at a gas station.
+000000156292 A person is looking at a display of clocks
+000000502910 An orange sign pointing to the right sitting on a sidewalk.
+000000050145 Shirtless man pushing his bike on a city street. 
+000000134689 A giraffe standing in dry dead brush on the savannah.
+000000157767 Several people at a gathering where there is food present. 
+000000289417 A boy standing on skis in the snow.
+000000382125 a person holding a slice of pizza next to someone else posing for a photo
+000000011051 A girl puts a flower on the lapel of a boys coat. 
+000000374982 A young girl standing in front of a pizza.
+000000128699 A man that has a skateboard in the air.
+000000140439 A. clear vase containing bouquet of pink with white tipped  tulips 
+000000395801 People are walking down the sidewalk in a city. 
+000000492362 A man on a skateboard riding by food vendors.
+000000295797 A rooftop next to a clock tower with a wind indicator on top.
+000000014380 A very large bridge over lots of train tracks.
+000000286907 A simple blue bathroom with only one toilet.
+000000039785 There is a man surfing on some waves.
+000000127624 grainy image of two trains on tracks moving through a city
+000000404678 A group of people sitting in a restaurant booth eating food.
+000000010707 A man playing a video game while two men sit on a couch. 
+000000260470 There is a window with a cake and other baked goods showing.
+000000560011 A mouse is next to a keyboard on a desk
+000000481386 A woman that is bent over on a counter.
+000000363840 Several laptop computers and electronic devices on a desk.
+000000279887 A group of motorcycles are on display at a museum.
+000000413247 Computer desk with laptop open on it with pictures on wall
+000000418281 A cow standing in a grassy open field.
+000000001993 A simple modern bedroom sheets on the bed
+000000566282 Two men kicking a ball on a city street.
+000000222299 A assortment of things lined up on a bed including writing utensils.
+000000055528 A man is holding a toothbrush and a paper bag while seated.
+000000272416 The toilet is very low to the ground.
+000000369370 A sandwich on a plate served with garnish.
+000000437351 A girl sitting next to three old suitcases.
+000000569972 a man on a surf board stands in shallow water 
+000000125936 A man is at the counter by a bowl of fruit.
+000000055299 A bird is perched on a large rock near the shore.
+000000533958 A plate of food with a fried egg and colorful vegetables.
+000000210789 The two people are standing under an umbrella in the rain.
+000000060052 A black and white view of a woman holding an umbrella.
+000000525083 a wood floor in a large home kitchen
+000000104612 A broccoli and pasta salad with oil and vinegar dressing.
+000000471450 Three bears standing in a field outside. 
+000000286503 An elephant eating something from a hanging net.
+000000050679 A close-up of an orange on the side of the road.
+000000366178 A road signage giving directions on a road covered with snow
+000000190140 A man in a hat and glasses is sitting on a boat with a dog.
+000000310980 A bear is sitting in front of a computer.
+000000251537 There is a carrot next to some greens on the red cutting board.
+000000151051 A skier sliding down a snow covered hill.
+000000195918 a desk with a monitor and a keyboard
+000000455716 Motorcycles lined-up in an empty, grassy field, all in front of a house.
+000000242287 Bicycle leaning up against a dirty board, near outdoor market.
+000000248112 A man playing tennis is getting ready to swing at the ball.
+000000189820 a desk in a room next to a couple of windows
+000000314177 Someone sitting on the toilet in the bathroom holding their camera at the mirror. 
+000000085576 A commercial bathroom stall with empty toilet paper holders.
+000000011051 A young girl wearing a formal dress, and young man in a tie.
+000000260925 A black and white cat lays on the hood of a car.
+000000140987 A girl sitting on a chair in front of a pepperoni pizza.
+000000402720 A group of three men at a restaurant drinking wine. 
+000000357903 A pizza topped with cheese, veggies and an egg.
+000000488251 a man in a black hat standing next to and  holding the reigns of a horse
+000000292415 A guy rides his skate board in a half pipe. 
+000000235399 a white dog is sitting in the back of a truck and a mattress
+000000426203 A man riding a skateboard down a metal hand rail next to some buildings.
+000000255749 A man stands beside a bus in a snowy forest at night.
+000000254368 A man laying in bed holding a plate of food.
+000000135604 a young boy is playing soccer on a field 
+000000515982 a man is sliding up to a base at a baseball game
+000000311909 A white bus that has a 41 Crosstown sign on the front.
+000000460147 Semi carrying cars on busy road in city near white car. 
+000000163258 A white toilet sitting in a bathroom next to a wall.
+000000180296 Packing everything for a trip including a childs Barbie doll.
+000000323895 A man in a tennis tournament getting ready to hit the ball
+000000439854 A person does a trick on a skate board by a beach.
+000000233825 A basic living with a TV, couch, and rack, 
+000000482917 A man relaxing at home, watching television with his dog.
+000000130386 A large sign with neon green wording sitting on the side of a road.
+000000343524 A male tennis player is preparing to serve the ball tennis court
+000000150726 A pair of giraffes walking in an area between a rock and a tree.
+000000528980 An umbrella sculpture is displayed on a post.
+000000473406 A little boy sitting on a kitchen counter top next to a plate of sliced apples.
+000000516038 Pitcher in the baseball game getting ready to throw the ball
+000000082807 A very cute looking small dog by some food.
+000000085089 A bunch of people that are holding wine glasses.
+000000304291 a horse walks along a dirt path alone 
+000000537964 Red fire hydrant with Leizan trees around it. 
+000000236308 Two people dressed in red skiing across a snowy landscape.
+000000491470 A cleared crosswalk in a city on a snowy, slushy day
+000000185473 Horses lined up on a grassy field, some with riders. 
+000000473118 A young man balances himself as he tilts his skateboard up at the edge of an obstacle.
+000000166478 A young guy using his computer and talking on his phone.
+000000100624 A girl is standing by a storefront while talking on her phone.
+000000147498 A flock of birds flying over a beach.
+000000203488 Parketing meter next to a tree with a warning sign on it. 
+000000519688 a white woman wearing a blue jacket holding a animal
+000000177213 A pizza sitting on top of a white plate on a kitchen table.
+000000084431 A ham pizza with grated parmesan is on a plate in front of a TV.
+000000303863 A railroad train rides down the tracks near buildings. 
+000000074200 Three surfboards in a display with awards and pictures
+000000482970 An open laptop computer sitting on top of a wooden desk.
+000000426268 Train cars sit on a track next to a roadway. 
+000000191288 A woman riding on the back of a brown horse over an obstacle.
+000000366141 A living room with a chair, couch, and a TV
+000000528862 Several giraffes and other animals in a fenced enclosure.
+000000475484 A yellow parking meter on the side of the street.
+000000099428 a laptop and three cellphones on a yellow table
+000000221017 A rooster walking on a beach near the ocean.
+000000110784 A lady is riding the back of a trolley train.
+000000195842 a man plays Wii Sports boxing in a very messy living room
+000000482585 Two trains at station near group of people on platform.
+000000405205 A number of  vehicles stops in a traffic jam
+000000481159 A horse standing in a snowy pasture looking over a fence.
+000000384666  A picture of skiers skiing on the snow
+000000375015 a black bear walking toward a rock 
+000000521259 A man is playing frisbee with a school child.
+000000250282 A group of young children sitting next to each other.
+000000470924 Four people enjoying a meal are posing for their picture. 
+000000168619 A road traveling along a field under a cloudy sky.
+000000404923 Baseball pitcher in the middle feel about to practice
+000000180101 A boy getting ready to blow out candles on a birthday cake.
+000000465549 A pair of smiling young people play a Wii game.
+000000220310 A teddy bear with a heart is in a green bin.
+000000085376 A single green motorcycle Ina full parking lot
+000000217285 A batter, catcher and umpire during a baseball game.
+000000257169 a woman holds a dryer in a bathroom 
+000000240767 Street signs sitting beside a large red brick building. 
+000000410934 A bunch of green leaves on a broccoli plant.
+000000161875 A woman standing under trees by a field.
+000000567011 The man is riding the waves on the water on his board. 
+000000528314 A man who is skiing performs a trick. 
+000000407650 A person is snowboarding off of a rock formation.
+000000149375 A young person riding a skateboard at a skate park.
+000000521819 a group of people are playing frisbee in a park
+000000139260 There is a banana split with whipped cream and a cherry.
+000000423971 A toilet with an illuminated seat in a restroom.
+000000026941 A pile of suitcases that are stacked on a green roller.
+000000051008 A cat that is checking into its Twitter account to post a tweet.
+000000226662 Several custom made motorcycles on display while riders chat.
+000000378605 A chocolate frosted donut on a plate with a cup of coffee and a penguin napkin holder. 
+000000341058 Advertising sign above a patio umbrella on sunny day.
+000000409630 Laptop computer with wireless keyboard and mouse on table.
+000000165831 A big shiny metal pan filled with some veggies.
+000000565624 Two elephants walking around inside of a grassy enclosure.
+000000383339 A man surfing in the ocean at sunset.
+000000431568 A whole pizza on a red checked table cloth.
+000000378139 There is a water shuttle going down a canal.
+000000140556 A group of men standing near a fire hydrant.
+000000147205 A piece of roast meat with walnuts and mushrooms on a serving plate
+000000131386 a formation of fighter jets flying by in the air
+000000066561 Couple of sheep relaxing in a fielder brown
+000000115946 The people are parked along the street by the tall buildings.
+000000500716 A man peering through the handle of folding scissors, as if they were eyeglasses.
+000000459437 A person feeding sheep behind a white picket fence,
+000000033109 A blue semi truck drives down the street.
+000000176446 A yellow not book sitting on top of a desk next to a laptop.
+000000154087 A group of people standing on a snow covered hill.
+000000083540 A skiier on a snowy path at sunset.
+000000521601 A small donut at the bottom of a ceramic mug.
+000000115245 Luggage and coats lined up at portable tables
+000000192191 A frozen pizza is cooked in an oven.
+000000492110 A man sitting at a desk on a laptop computer.
+000000374083 A man and a girl standing next to two cakes in the kitchen. 
+000000290771 A large glass window in a living room.
+000000161642 a wall clock shaped like a birdhouse with tassels  hanging from it.
+000000258541 A man wearing a neck tie and a white shirt.
+000000484893 Several sheep are in a pen as some kids stand next to it while one boy has his hand on one of the sheep's head.
+000000115245 A table has suitcases leaning on it and clothes on top.
+000000076468 Twp people shaking hands on a tennis court.
+000000308793 A person walking in the snow with skis on their back.
+000000173371 this is a table with a pizza on it
+000000529122 A woman and her son picking out sweets at a bakery.
+000000523957 Man on long board skating down steep road
+000000190841 A man on a skateboard jumps over a ramp.
+000000561335 A green coat flipped upside down on top of a beach.
+000000480936 A barefoot woman eating food outside from a plate on her lap.
+000000032887 two men camping in the woods near rocks
+000000017627 A green car has parked on the curb in a parking lot
+000000104669 A plate with a meal of meat, potatoes, carrots, and broccoli. 
+000000556498 a couple of signs are hanging on a black pole
+000000500478 A man in white baseball uniform throwing a ball.
+000000528705 A stuffed bear is on someone's red jacket
+000000426372 A white truck carrying a small car in the back.
+000000105264 Young men playing on the beach with a cow in the foreground.
+000000344268 The president making a speech with a communist/american flag combo behind him.
+000000044652 An airplane is flying high in the cloudy sky 
+000000019742 A picture of a cup on the table.
+000000248980 A plate of food sitting on a table, it appears to be a salad.
+000000361180 A bear is standing on a rock face.
+000000073946 Two men fake biting a surfboard for a picture.
+000000459437 A row of sheep by a wooden fence with a guy holding a bowl. 
+000000124975 Two zebras stand together on a dirt area near short grass.
+000000311392 A farmer inspecting cattle at a livestock auction.
+000000254368 A person with a piece of pizza over a plate.
+000000018519 A man doing a trick on roller-skates in a skate park.
+000000369541 A dog in a vest catching a fristbee.
+000000565469 there is a very tall tower with a clock at the top
+000000334555 A cow is standing on a street corner.
+000000328030 A scene with a teddy bear playing hockey.
+000000454978 a motor bike sits parked next to a street sign 
+000000182202 A Vodafone cell phone sitting next to a laptop.
+000000218997 A child sits with a ball and bats around them.
+000000530162 Some people standing on a dark street with an umbrella.
+000000154213 A tin with pie crust in the bottom 
+000000308430 A skillet on a stove with vegetables in it 
+000000553511 Cars passing through a street intersection with office buildings in the background.
+000000276018 Kids walking while holding their stuffed animals. 
+000000156071 We are looking through the bars of a fence into a rodeo arena.
+000000254814 A large group of people on some bikes on a street.
+000000450075 A person is tying a tie around a girls .neck.
+000000410612 A boat passes by waterfront houses flanked by trees.
+000000144300 A motorcycle that is sitting in the street on a stand.
+000000177935 A large white kitchen range with gold towels on the front.
+000000356531 a woman with the meal on a green plate
+000000455555 A boy riding a skateboard on a sidewalk as another laughs in the background.
+000000402720 Three men pose and smile at a table.
+000000039405 A member of a tennis doubles team just returned a volley.
+000000516916 A table with laptops, monitor screens, keyboards, mice, phone handsets and    chairs
+000000323799 A group holding frisbees pose for a picture.
+000000526706 A cow breastfeeding its calf  while other cows stands around in a grass area.
+000000033221 Two women at a beach with tennis rackets in their hands and one girl looking very excited with her tennis racket in the air. 
+000000022755 In a rear view mirror, a school bus can be seen.
+000000470173 A female statue in a sea of green grass next to plant
+000000480944 A traffic light and sign on a street corner.
+000000055150 Two children in a stroller among cars in a rental lot.
+000000302760 An image of a bathroom with a sink and mirror 
+000000211069 There are two giraffes walking near some tall grasses.
+000000506178 Blurry photo of a young man playing a video game.
+000000086220 A silver car in the street next to a metal railing.
+000000066817 This plate consists of rice with sauce and peas and meat patties on the side.
+000000345027 A scooter and tiny truck at a beach dry dock.
+000000267903 A filthy bathroom has a dirt covered sink and a rusty toilet.
+000000079014 This is a blurry image of a man looking at something yellow and roundish.
+000000005600 A plate has two bowls on it with two different types of food, one looks like pickled onions and the other looks like cooked meat. 
+000000510095 A man is holding a bat in a field.
+000000398377 three females impatiently waiting. Two of them are on their cell phones
+000000149375 a person on a skateboard is doing a jump
+000000018150 A man feeding pizza to a reluctant toddler.
+000000269942 A green traffic light hovering over a street.
+000000026690 A LARGE GROUP OF PEOPLE WATCH AS A SKATER DOES HIS TRICKS
+000000257566 A narrow boat coming down a small water way next to a sidewalk.
+000000565853 A man playing a game with a remote controller.
+000000507575 a white desk a keyboard two laptops and a printer
+000000578545 A pregnant woman laying in bed reading a book.
+000000212072 A stop sign leans to the right at a small town intersection
+000000228981 A close up of a PDA, digital camera and a music player.
+000000011511 A metal statue of two women sitting on a bench.
+000000546325 A living room with a couch table and opening to another room.
+000000313182 A purple and white city bus pulling up to the curb
+000000464522 A dog with a collar standing on the grass
+000000372718 Some flowers in a clear vase filled with water.
+000000209753 a woman standing while looking at her cell phone 
+000000125245 Broccoli is tied on a piece of plastic in an aquarium.
+000000147338 a fire hydrant with a face drawn on the front
+000000472375 a brown black and white dog wearing a helmet and a black scooter
+000000445722 duffel bags and other equipment on a long boat
+000000253452 A white plate topped with food next to a plate of onion rings.
+000000360564 A toilet with a wood seat and a tiled floor
+000000283412 a white dog sitting on a table next to a newspaper and a cup.
+000000231339 A little girl is kneeling in front of an open refrigerator. 
+000000341196 The person is riding the waves in the water.
+000000567640 two soccer teams going after the soccer ball
+000000031118 A clock tower and a large Christmas tree in the center of a plaza.
+000000002532 A person is skiing on a snowy hill top.
+000000418696 The bright lights are shining on the street.
+000000334483 Women and children are gathered around a blue cake.
+000000290619 A banana sitting on a counter next to a pitcher.
+000000327890 Three empty benches face each other beneath a canopy.
+000000438907 A man is doing a trick on a skateboard.
+000000452122 The large jumbo jet has it's landing gear lowered.
+000000529148 Two children use their laptops in a room.
+000000439426 A left hand holding a glazed doughnut over a tiled floor.
+000000324927 Slices of meat pizza with a shot glass of mustard
+000000108503 Two people holding surfboards walk down a beach.
+000000089045 a living room with a couch and piano 
+000000440184 Four people play a game of mixed doubles tennis while another man looks on.
+000000401991 A cat and a dog both sleeping on a bed.
+000000498747 Guy sitting down and using his cell during the party
+000000334521 Two giraffes that are together in an enclosure. 
+000000385997 A living room filled with furniture and a large window.
+000000386879 A woman wearing a red tennis outfit holding onto a racquet.
+000000197388 Four men ski down a snowy cliff in team clothing.
+000000380711 Blurry photograph of a surfer emerging from the ocean with his board
+000000459954 Many old appliances in trash head with woman in center.
+000000428111 a man is in the air on a skateboard
+000000261712 Two giraffes wander the open park in the same direction.
+000000276284 A motorcycle in a parking lot at night.
+000000578967 A monorail is going down the track with people on it. 
+000000493799 A white plate full of cooked meat and vegetables. 
+000000376900 A tennis player prepares to serve a tennis ball.
+000000025986 Food is on the counter including a glass of water, rice, vegetables, and more
+000000236308 The people are skiing down the snow covered hill. 
+000000479596 a shelf with a lot of assorted colorful fruit for sale 
+000000311392 A cow licking its side in an enclosure
+000000429718 A black and green train locomotive engine traveling along a railroad track.
+000000027620 A laptop sitting on a desk next to a chair and another computer.
+000000531135 Baseball players on the field are competing during a game.
+000000173033 A replica of a bear and her cub in a glass case in an exhibit.
+000000491613 A standing zebra and its reflection in water.
+000000457559 A soccer game with players from both teams going after the ball and other players close behind.
+000000344268 A president standing at the microphone by a big flag.
+000000138954 Slices of food are arranged around food piled on a plate.
+000000194216 A group of forest creatures near a downed tree in the woods.
+000000458410 A 70s themed living room displays an old tv.
+000000309964 Two people smile as they sit on coolers under a huge umbrella.
+000000104603 A black bear cub walking down a small hillside.
+000000412894 People are walking across the street at an intersection.
+000000424349 A chef is pulling a pizza out of a pizza oven
+000000125806 a couple of  black cows in the woods
+000000213605 Van drives by a bus sitting at the curb
+000000519491 A large clock tower with a statue on the top.
+000000323709 A large jet airplane parked next to a domed tower.
+000000067896 an old clock hanging up against the side of a wall
+000000507081 A kitchen has an oven, refrigerator, and a door with a window in it.
+000000530836 There are appliances and cookware in a kitchen.
+000000402765 a closeup of buttons on a nintendo wiiremote 
+000000019221 Two fingers holding a  small piece of broccoli 
+000000278705 Two young men skate down an urban road.
+000000221291 A BOY IS OUT ON THE PARK FLYING A KITE
+000000485480 A baseball player standing next to home plate.
+000000193717 a fire hyrdant sitting on the sidewalk next to the road 
+000000481404 A church with a stain glassed window of the virgin mary.
+000000335658 A keyboard and mouse sitting on a desk.
+000000325991 A man holding a frisbee with one hand in the grass.
+000000275749 Group of people standing around computer monitors. 
+000000484351 A classroom with kids and various men in suits.
+000000203317 Bicycle chained to a building with graffiti on the door.
+000000289594 An old man that is leaning against a pole right by the corner.
+000000407825 A clock shaped like an apple sitting on a counter.
+000000491464 A boy with a baseball bat in the ground.
+000000304560 A black cat sitting in a field of grass.
+000000231125 Pans of potatoes and a  person wearing a red apron in a commercial kitchen.
+000000388927 An old cellphone that doesn't flip or have a touch screen. 
+000000032735 A person in the air with skiing equipment. 
+000000462643 A person holding a broken up cell phone.
+000000080949 A kitten lies down on the keyboard of a laptop.
+000000289229 The male doctor is holding a medical saw.
+000000039951 A boy hitting a tennis all with a racquet.
+000000074256 Two women standing in a train station looking at their phones. 
+000000064523 An all female tennis team taking a group photo on a tennis court
+000000084650 The cat is laying on a luggage bag
+000000532530 A map of the town in the middle of the street with buildings in the background. 
+000000078843 The young man is kissing the young woman on top of her head.
+000000466085 an unfinished bathroom with a sink and pipes on the floor
+000000065288 A bird is perched on the back of a chair.
+000000157124 A white sink placed in the middle of a constructed kitchen.
+000000113354 Three zebras munching on grass in their natural habitat.
+000000523241 A busy city intersection on a cloudy day.
+000000157928 A cooler attached to a sled near some skiers 
+000000323151 A pizza sitting on top of a white plate next to a glass of beer.
+000000155145 Three people preparing to launch a small boat in a river.
+000000131431 An clock is encased in the wall of an older building.
+000000460967 A colorful bus parked in a lot beside a tall building with many windows.
+000000090956 a skier in a green shirt is coming down a mountain
+000000439522 A woman riding skis across snow covered ground.
+000000030494 A black and brown dog digging at an object on a dirt ground.
+000000065074 A view of a bedroom with the bed unmade.
+000000551660 A bowl with broccoli and apple slices on a placemat.
+000000409211 A man taking a swing at a tennis ball
+000000541291 The old, dilapidated  bathroom has fallen into disrepair.
+000000247838 A herd of bison stroll by the snow covered banks of a stream.
+000000562581 A teenager getting ready to swing at a tennis ball.
+000000530470 A park with several vehicles parked on the grass.
+000000416885 Three pastry desserts and two glasses of wine are on a table. 
+000000008277 A white plate topped wti chicken and a cup of guacamole.
+000000209972 A sail boat sitting on the sand at the beach
+000000207844 People are sitting in lounge chairs on the beach.
+000000543581 a dog on a couch in a living room 
+000000289594 A man in a hat leaning against a pole
+000000351096 A hand holding a pair of giant scissors above boxes.
+000000379533 A zebra grazing on a dry grass field next to elephants.
+000000005060 a man sitting in a room taking a selfie in a mirror 
+000000002153 A baseball game where the batter is waiting for the pitch.
+000000359781 The tall adult giraffe is standing next to a boulder.
+000000312552 A woman holding a plate and eating pizza.
+000000092091 A baby girl chews on a stick with a teddy bear in hand.
+000000318138 A girl playing Wii holding two remotes 
+000000343149 Outside view of people walking in front of a store
+000000157767 A couple of people that are eating some food.
+000000245173 A tower building with a clock up high.
+000000021839 A person crossing a street next to a crosswalk.
+000000004765 A man riding a surfboard on a wave in the ocean.
+000000471450 Some very big furry brown bears in a big grass field.
+000000344059 Two giraffes nudging over a pole in a forested area
+000000333697 Three traffic signs sit on on a corner in front of a brick wall.
+000000384850 A bathroom with a shower curtain that has four palm trees on it and a white tile floor.
+000000350023 a long road with cars on both sides of it 
+000000562197 Different types of foods and vegetables side by side.
+000000511599 some people and a long thin boat water and houses
+000000482735 A male surfer on a white board riding a wave.
+000000014473 A very nice looking toy train and some workers on the tracks.
+000000226154 The large green bus drives near two people on a motorcycle.
+000000383838 A woman holding a Nintendo Wii game controller near a man.
+000000560256 Assorted vegetables are displayed at a fruit stand including bananas, oranges, apples, and cherries.
+000000236784 Two black dogs on a couch sleeping peacefully.
+000000463522 A horse stands on the side of the road with a harness on.
+000000176037 A very big city bus on wet city street at night.
+000000567640 A group of young boys playing soccer on s field. 
+000000219440 A herd of cattle standing next to a tree together.
+000000537355 A red fire hydrant with a building in the background.
+000000460927 A black and white photo of a bear in the wilderness.
+000000356424 a man smiling while looking at his plate of food
+000000333402 A man standing in front of the open door of a bus.
+000000253386 A big dog next to a person's leg.
+000000553094 Fighter Jet with afterburners on flying almost vertically.
+000000035197 Person riding on edge of skateboard about to do a trick
+000000082807 There is a dog that is seated at the table with a frosted cup cake in front of the dog
+000000047121 A black cat drinking water out of a water faucet.
+000000341681 A tennis player trying to hit a tennis ball.
+000000004795 Gray cat looking at electronic monitor next to phone.
+000000210230 The workers are busy serving meals under the tent.
+000000424521 This is a painting of a skateboarder in the midst of a jump, and his  surroundings.
+000000162581  A man playing tennis in a tennis court runs to hit he ball 
+000000341973 The two woman are eating donuts outside. 
+000000243626 A fork, a pork chop, green beans, an egg and parsley on a white plate.
+000000192670 A boy swinging a bat with a catcher in place behind the batter.
+000000060102 a man hitting a soccer ball with his head 
+000000476770 A baseball player is looking at another player as he runs.
+000000425221 A large terminal in a quiet airport hall.
+000000168458 A vase with purple iris's sitting next to a ceramic pitcher.
+000000547383 two sheep laying in a field of grass
+000000203488 A silver and gray parking meter sitting by a tree.
+000000553669 a male in a brown shirt sitting on a bench with a laptop
+000000555597 A old photo in black and white of a city street.
+000000167486 A girl is onstage singing while the guest on the bike enjoys himself.
+000000521052 A black and red truck with artwork is parked near a sidewalk on a street.
+000000088485 A man outside playing frisbee in front of a building.
+000000493905 A man in white shirt and shorts playing a game of tennis.
+000000269113 Several dogs in a grassy field playing. 
+000000309173 A stop sign on a pole on a street corner.
+000000128699 A man makes a jump on his skateboard before a crowd.
+000000426372 A white truck with a red car sitting in it's back.
+000000005477 large mustard yellow commercial airplane parked in the airport
+000000187734 Man and women gather around to play a video game.
+000000091779 Three hotdogs are prepared and loaded in different ways.
+000000294695 Two women leaving a beach with kites flying in the background.
+000000578236 A lizard and some bird sitting on some steps.
+000000422886 A man eating a hot dog with cheese and onions.
+000000022892 A cats sits on the table with a dog outside.
+000000268831 A black and white tiled floor in a bathroom
+000000195165 A sink is pictured with a small shelf full of products.
+000000277020 A man sitting at a bar next to another man near a waitress.
+000000218091 A hotel suite with a desk sofa and bed
+000000291664 a dog sniffing a red fire hydrant near grass
+000000011760 Three zebras look towards the camera in their habitat.
+000000321333 A picture of two young kids posing or a picture.
+000000189451 A part of a cake on a plate with a knife and cake server next to it.
+000000130599 A man pretending to punch a giraffe sculpture.
+000000286182 A man sitting by a table with food in bowls and pans 
+000000253386 A medium sized dog is standing with some people.
+000000098633 a white square cake with five strawberries on the top
+000000134856 Three people on horseback ride along the beach while waves come in.
+000000338428 A truck painted green with designs parked by other vehicles.
+000000060855 The two apples are surround by a bunch of oranges.
+000000442480 a twin propeller plane sits parked on a run way 
+000000501005 baseball players with a bat and glove at the pitch
+000000098018 Group of old ornaments sitting on the ground of an old building. 
+000000551439 A woman holding a book about the topic fun.
+000000389381 A colorful assortment of different fruits and vegetables. 
+000000570456 A view down into a living room and kitchen.
+000000392722 a double decked bus sits parked next to a side walk 
+000000347265 A person skiing through a snowy forest with tall trees.
+000000116362 Grainy photograph of thinly sliced fruit in a bowl
+000000557172 A CONTEMPORARY BATHROOM THAT HAS A VERY LARGE MIRROR
+000000163746 Girls playing on a beach with an airplane flying above them. 
+000000470952 a person riding skis on a snowy surface
+000000094751 A street light attached to a pole next to a street and a sidewalk railing.
+000000099039 a table thath as some pizza and wine on it
+000000252294 A pair of people lean out from behind curtains and smile.
+000000329219 A man is at a kitchen counter by a dog.
+000000161875 A little girl stands, hand to face on the edge of a field.
+000000272566 two people siting on the ground near many different fruits
+000000173383 The cake is prepared and ready to be eaten.
+000000473974 A herd of zebra are grazing on tall dry grass.
+000000425925 A busy city with a building that has a clock on it.
+000000002685 Many people line up to taste some wine. 
+000000148719 A red Dodge truck is parked near another Dodge. 
+000000047769 The small room has a couch, a television, and a table with two stools.
+000000372260 A car is parked next to a decorative clock.
+000000020107 A rusted fire hydrant sits off to the side
+000000152870 A giraffe stands in an area of dead shrubbery.
+000000423944 Two headless mannequins display fashionable clothing in front of a pink background.
+000000139684 A living room with yellow walls and a black and white area rug.
+000000052017 I am unable to see an image above.
+000000284623 Black cat with green eyes sitting in a bathroom sink. 
+000000082812 Pedestrians standing outside a train station waiting for the train. 
+000000415990 The men are checking in on the herd of cows.
+000000179285 A tennis ball sitting on a tennis racket.
+000000135890 A building in shadow with a clock tower.
+000000210273 A highway of traffic with several transit buses riding down it.
+000000460929 A hot dog with mustard and an unopened bottle of Bubble Up
+000000382125 Two men are posing for a photo, one man is holding a slice of pizza on a plate, and they are surrounded by other people sitting at tables.
+000000522156 a close up of a plate with food 
+000000167486 A person performing on the stage and a man sitting on the bike 
+000000050331 Two clay pots with leafy green plants in them.
+000000185473 a large group of horse with men riding on a few of them
+000000317999 A  girl reading a book on a bed next to a man.
+000000022479 The skateboarder wears headphones while jumping to grab his board.
+000000456015 People are riding horses down a beach next to the water.
+000000455301 a child and a woman in bed with a book
+000000163117 A crowd of people standing on a field flying kites.
+000000446207 A digital SLR camera next to a smart phone camera
+000000116825 A black and white cat pawing at an object on a table.
+000000446005 An empty kitchen is photographed as if for a real estate listing.
+000000347265 A person in the snow on some skis.
+000000217948 Three brown bears are looking outside their enclosure.
+000000094944 People are standing on top of a snowy mountain.
+000000162415 A YOUNG GIRL PRACTICING BALL IS HOLDING HER MIT
+000000515579 A man holding a ball while standing next to a gate.
+000000066841 A toaster is decorated with kitchen utensils to have a sad expression. 
+000000180487 GUY WITH UMBRELLA HAT SITTING A TABLE WITH ANOTHER PERSON
+000000569059 A wooden desk sitting in a living room with a monitor on top of it.
+000000149568 A man and two dogs playing with an object.
+000000284762 Stop light in a city next to a street sign on during the day. 
+000000227478 A pair of girls sit on a bench outside.
+000000128748 An umpire is talking with two players on different teams.
+000000360393 A raw pizza with red peppers and cheese is on the counter.
+000000498286 a vety cute furry dog posing near a big sign.
+000000298904 An old fashion vase with flowers and rocks inside. 
+000000446207 A cell phone sits next to a black camera.
+000000133000 A city view of an E-train on a track
+000000540414 Group of people enjoying food at a market.
+000000515025 A woman looking at a cat licking pizza grease
+000000229858 A baby deer and a large deer are standing in the woods. 
+000000516804 A close shot of a Bus stop sign near the building. 
+000000346968 A woman in uniform is talking on a cell phone.
+000000492282 Oman and group of people on horseback moving down cement path
+000000090208 a green double decker bus sits at the front of a row of buses
+000000492284 A woman pets the cat that is standing on the rocks.
+000000404922 A woman, little girl and boy, walking with tennis rackets.
+000000159791 A refrigerator with a variety of magnets and pictures on it.
+000000146667 A woman digging through the storage compartment on her moped.
+000000414340 An empty street and steeple with a clock on it
+000000399462 a little girl is running on the grass with a kite
+000000187745 A small plane flying through a blue sky.
+000000141328 A pasta with some different vegetables mixed in
+000000236412 A cheese pizza pie is in the serving dish on the counter.
+000000169076 A dog laying in a room near a television and dresser.
+000000166768 An elderly woman is holding onto the arm of an elderly man.
+000000179898 A boy standing in front of his class holding a hot dog in each hand.
+000000577864 Three people are playing with a red kick ball.
+000000227511 There are plenty of cars parked on the side of the street. 
+000000017905 a man that is standing next to a traffic light
+000000329827 A person in a blue coat is skiing down a slope.
+000000085911 Something sticking out of the body of water.
+000000341094 A statue of a bear on a car used as a warning about the bears.
+000000139260 A banana split is sitting on top of a table.
+000000162543 Some elephants are roaming around their habitat in the zoo.
+000000553731 A man dressed up as a clown is holding a cellphone to his ear. 
+000000537153 Two people sitting on a bench looking out over the ocean 
+000000086956 a couple of people having a few laughs with each other.
+000000069795 a watering can for a vase of purple flowers
+000000079837 The boats are in the water for a peaceful night 
+000000488075 The view of the door and kitchen area of an apartment from the living room.
+000000365745 Several police vehicles at an urban crime scene.
+000000191471 A close up view of a stop sign in a neighborhood.
+000000262440 A white bathroom with all of the necessities.
+000000170191 A man in a leopard robe stands next to a bed
+000000240940 A cat sitting on the floor watching television.
+000000361180 The bear is on all fours, facing the photographer.
+000000479099 A green yard has a little bench and brick fence.
+000000016010 Some Zebras and cattle are standing together, some eating. 
+000000131131 A kitten is trying to interact with a kitten that is on television.
+000000441286 The surfer is getting ready to ride the wave.
+000000565624 Two large elephants walking behind a wire fence on green grass.
+000000345027 A beach dock area lined with boats, and a small shack with a scooter and small cart near it.
+000000301718 A table with three plates with food and a man and a woman both holding utensil near plates.
+000000246522 A large building can be seen in the background with a clock on it's face.
+000000125405 two brown dogs walking in the field with their mouth open
+000000533855 A white frosted donut with sprinkles and a jump rope with leather roping next to it.
+000000335177 a nice neighborhood with some green grass in it
+000000024144 A cheese pizza cut into eight pieces in the box. 
+000000212226 A person in a the reflection of a truck mirror.
+000000035770 a bathroom with three toilets and a big sink
+000000052412 View from an airplane looking looking out over large parking lot.
+000000544444 A lone man sliding sideways downhill while on skis.
+000000002473 A skier is jumping while other skiers watch.
+000000566042 Six giraffe are standing looking in the same direction.
+000000267434 A herd of cows are standing in a green field.
+000000203546 The bear is inside of the river stream. 
+000000547336 A young woman sitting on a curb next to a fire hydrant writing on a notepad.
+000000531134 A traffic light with a sad face on the red light. 
+000000327780 A piece of cake sitting on top of a blue and white paper plate.
+000000361551 A group of people and luggage on a airport tarmac.
+000000308193 a clock tower behind a building with a sky background
+000000533145 A video game controller being pointed at a television.
+000000407403 A pitcher filled with yellow tulips on a black and white background.
+000000087144 a couple sitting on a bench with a little girl
+000000417043 A bridge,a bench and lots of trees and branches.
+000000553990 A man riding on a horse as it jumps over a pole. 
+000000203317 A red bike is parked outside of a barred window.
+000000106881 This is a picture of a bird on a limb of a tree.
+000000063552 A cat that is standing looking through a glass.
+000000131131 A cat sitting in front of a monitor that is displaying a picture of another cat.
+000000186449 An elderly woman with a kerchief on her head sits on the bench.
+000000327617 A woman holds a racket in front of a net.
+000000114907 Woman demonstrating equipment used in conjunction with cow.
+000000241668 A woman in a wedding dress with another lady holding a piece of cake.
+000000480842 Two giraffes are peering up above some bushes.
+000000050896 This is a clear bowl that is filled with oranges. 
+000000216636 A partially eaten bundt cake next to a lit candle.
+000000357742 Person skiing on snowy slopes with wires above
+000000296657 Six men jumping in the air on the lawn
+000000503841 Green stop lights are set against a tall building.
+000000110211 Zebras and rhinos out in the wild on a sunny day 
+000000451879 A group of men playing baseball, one is running, while another is throwing a ball to another player.
+000000211674 Red double decker bus with people standing on top.
+000000430871 A parking meter next to a handicap parking space.
+000000177539 a person in a stripped dress leaning against a surf board
+000000010092 A bed in a hut covered in a bug net.
+000000520871 a pizza sitting on a plate next to a little jar on a table 
+000000556765 A young elephant in a river with rappids
+000000301867 Three young women smile brightly together under a large umbrella.
+000000141328 The bowl is full of pasta and vegetables. 
+000000246308 THERE ARE DIFFRENT TROPHIES ON DISPLAY ON THE WALL 
+000000009378 A man holding onto a yellow frisbee while having long hair.
+000000235778 A little kid that is playing with stuff.
+000000450399 Several people are working behind the counter of a bakery.
+000000506933 A man with glasses is holding up an iPhone. 
+000000422706 People watching off the side of a cruise ship as a life boat is lowered.
+000000133645 A view of a lake with boats in it. 
+000000474167 two plates of pizza and a drink on a wooden table
+000000228771 A street corner sign is on the corner in Little Italy.
+000000447611 People sticking knives into the back side of a laptop
+000000356387 A row of parked motorcycles sitting in front of a tall building.
+000000231879 a person holding up a child with a cake 
+000000198960 A black and white picture of three lambs. 
+000000559956 Two women have stopped to pet the sheep.
+000000315492 A girl is playing with a teddy bear in a bathroom.
+000000281929 A man in a suit standing beside his bicycle. 
+000000434996 A kitten sleeps with many stuffed animals on the bed.
+000000576031 A woman on skis stands on a flat snowy surface.
+000000210273 a bunch of vehicles are coming down the road
+000000007977 A boy is doing a trick on his skateboard. 
+000000047010 A pond of water with three giraffe walking in the dirt.
+000000275749 A group of people are socializing with each other.
+000000575081 A young child standing in front of a TV holding a Wii controller.
+000000345385 A couple is sleeping in a bed with red sheets.
+000000378873 A produce stand in an open air market with different veggies.
+000000019109 Many small motorbikes are parked along the street. 
+000000176232 A vase with beautiful purple flowers decorated in it.
+000000297085 a cat is standing in front of a television
+000000139684 A cat is perched on the back of a sofa in a multi colored room.
+000000304365 A large long train on a steel track.
+000000079837 a lake with a lot of boats on it
+000000567886 A person sits on the floor beside flip flops and a teddy bear.
+000000153217 A black cat meanders up the walkway in front of a motorcycle in front of a house.
+000000263594 A white care is sitting in a handicapped parking spot.
+000000308587 The photo shows the many ways to enjoy the beach--kite flying, bathing, lying in the sun, walking along the sand.
+000000037777 The dining table near the kitchen has a bowl of fruit on it.
+000000295713 A statue is sitting on a bench and a woman sits on a cement block.
+000000008844 A woman smiling as she walks toward bunches of bananas. 
+000000018575 a table top with some food on top of it 
+000000454404 A lady is sitting on a bench barefooted while using her phone.
+000000096960 A large white polar bear swimming in a pool of blue water.
+000000050679 an orange is on the white line on a street and some cars
+000000268000 A passenger train on the tracks going under a bridge.
+000000402334 a building that has a small clock at top
+000000434459 A pair of military personnel cutting and serving a cake.
+000000372307 A horse standing in the grass in a fenced in area.
+000000311081 A bathroom is adorned with beige and white tiles.
+000000378116 a person riding a surf board on a wave
+000000036494 the people are all in a restaurant some are sitting 
+000000374727 a polewith many street signs standing in front of a very tall building
+000000031269 three zebras in a field near bushes 
+000000086582 A person watering plants with a computer in the foreground.
+000000044068 teddy bear sitting on wicker chair with red cushion
+000000020059 Two zebras grazing on the green grass in an enclosure.
+000000007816 A person on a motor bike on a road.
+000000338219 a close up of some motorcycles parked on a grassy field
+000000383443 A bathroom with a tub, sinks, lights and a television.
+000000303818 A group of people cross the curved street.
+000000189820 An unattended office containing several computers and a chair.
+000000427500 A broken fire hydrant leaking water on the sidewalk.
+000000396205 Two brown cows on a pasture of green grass.
+000000508370 A couple of people standing next to a bike.
+000000023899 Some people chilling on the couch playing with a Nintendo Wii
+000000579070 A group of people gathered around the table.
+000000156372 A chair in the corner on a boat.
+000000269196 A knot of sheep are huddled close and looking out of a barbed wire fence with a frost-covered landscape behind them. 
+000000101420 A cat that is laying on a couch.
+000000175387 A toilet in a public bathroom with the stall door open 
+000000170670 An outdoor table containing assorted bowls of food and beer.
+000000405972 a family of elephants walking next to each other 
+000000101884 Ivy is growing up the walls of a building.
+000000057672 The boat is full of many rowers as one person stands steering.
+000000042528 a person lifting a little stool with their feet and a person holding a cell phone 
+000000409358 The corner of a bathroom, showing the sink and toilet.
+000000480122 A bowl of fruit is placed on a table in a kitchen.
+000000411665 A cat playing with it's reflection in a mirror.
+000000334521 Two giraffes inside of a cage at the zoo.
+000000548780 Two women sitting on a bench with one smoking a cigarette.
+000000026564 A desktop computer next to a laptop computer.
+000000498747 A young guy on a cell phone by a crowd of people.
+000000100582 There are meats and vegetables on a flatbread
+000000397681 A group of twenty-five squares of different photos of carrots.
+000000531771 three types of microwaves on top of each other.
+000000459467 A black plane taking off from an airport runway.
+000000409542 People walking by a clock on the street.
+000000098018 a two burner stove a door a window and a table
+000000302990 A man riding a wave on a surfboard at the beach.
+000000092839 The polar bear has his chin up as he lies on his chest
+000000530146 A plate filled with various kinds of food.
+000000002473 A man flying through the air while riding skis.
+000000072813 A cute brown puppy is snuggled on a rumpled bed.
+000000203488 A sign stating tree trimming is on a tree by a parking meter.
+000000519764 A cat that is laying down on a chair.
+000000261796 an older home with multiple windows, shot from the street. 
+000000322574 a hot dog sitting on top of a mound of french fries 
+000000565778 A train with people inside is about to go into a tunnel.
+000000301563 a big hill that has a bunch of snow on it
+000000554328 Two old time baseball players talk to a man.
+000000052413 A person holding up a toy cell phone.
+000000529762 A man sitting at a table eating a hamburger and fries.
+000000144333 a girl is sitting on a bicycle outside
+000000450488 A basic motel room with a bed and television.
+000000196843 A painted school bus reads "Van Gogh Museum Bus".
+000000386210 A window display with lots of fruit and grass. 
+000000071877 A guy on a skate board up in the air by a ledge.
+000000182417 A beautiful dessert waiting to be shared by two people
+000000430377 A woman on ski's standing in the snow.
+000000147729 A man in a crowd taking pictures with a cell phone.
+000000145620 a black and white photo of a person riding a dirt bike
+000000031269 Three adult zebra stand near the upturned dirt.
+000000576955 A rider takes their horse through some water at speed.
+000000451435 A tennis player preparing to swing at a ball.
+000000512194 Some flowers are sitting in the white vase in front of the window. 
+000000000285 A close up picture of a brown bear's face.
+000000037988 A female tennis player tossing a tennis ball up.
+000000480021 A painting of men riding on motorcycles next to a gas station.
+000000031050 A vase with various flowers in it on a display case.
+000000203629 A man giving a thumbs up while on a cell phone.
+000000465129 A person adjusting a  metal object hanging from a ceiling.
+000000033759 a boy swinging his baseball bat at a baseball
+000000326128 A person standing in the snow near a snow board.
+000000110999 A painting of a bowl of fruit sitting in front of another painting,
+000000023272 A calico cat sits on  the hood of a black car.
+000000145781 A dog licking the top of a  beer bottle
+000000166391 A number of red and green traffic lights on a wide highway.
+000000407403 a white vase with yellow tulips against a grey background
+000000147415 A young woman writes in a notebook beside an open notebook computer.
+000000153011 Several food trucks parked in an area that has tables and some chairs next to it.
+000000474039 A baby sitting between big stuffed animals. 
+000000429011 Two toy vehicles and a police box are set up in a street scene.
+000000516318 A man surfing beside a bird on a cloudy day
+000000541952 A clock tower has a white pole on top.
+000000513041 A pizza on a wooden plate near glasses of wine.
+000000163290 Image at a zoo of a giraffe near a zebra pen with both animals eating.
+000000189775 A man playing a tennis game in front of a crowd.
+000000419312 Full course dinner served on large plate including drinks and dessert.
+000000018150 A man with a receding hairline offers a slice of pizza to a toddler.
+000000355677 Lifesaving devices are stowed along a beach while a lifeguard is elsewhere.
+000000180383 A TV mounted in a cow on a wall.
+000000400161 a remote control holder attached to living room furniture that is full of controllers
+000000094336 A large cat sitting in a bathroom sink under a mirror.
+000000032334 Man and woman doing a toast with a glass of wine.
+000000007278 a man on a surf board rides a wave 
+000000152686 tennis player in mid-jump with racquet overhead and tennis ball at edge of photo
+000000423944 Two fashion mannequins dressed in garments on a runway.
+000000168619 A paved path stretches through the grass under the cloudy sky.
+000000562059 A table topped with a bowl of granola and two apples.
+000000047828 A bridge is covered up in bright lights at night.
+000000438907 a teenager attempting a jump on his skateboard
+000000398810 Tow cats laying down looking out of the windows.
+000000122606 A white bus with bicycle rack parked on the side of a street.
+000000060886 The entire baseball team has gathered on the field for a celebration. 
+000000101787 A plane is flying in a blue cloudy sky.
+000000340015 Full pizza with tomatoes, mushrooms, sausage, pepperoni, and peppers.
+000000500423 a train at a station near a platform
+000000542776 a small bird sits on top of a book page 
+000000465836 A group of men standing next to each other.
+000000545100 The man in the business suit takes a video of city buildings.
+000000068078 A small dug stands on a white rug in the bathroom.
+000000239041 A bathroom with purple walls and two mirrors.
+000000179392 Cat staring into the camera in the day time.
+000000576654 Man standing beneath large sail on beach with man controlling from distance.
+000000462643 A person is holding a small electronic device.
+000000456143 A table full of different foods, such as pizza, fries, and a sandwich. 
+000000511760 a man and achild are standing in a field
+000000160556 Individual is doing something at the moment that is intriguing.
+000000023781 Various fruits and vegetables are on display close together.
+000000078915 A woman tennis player hitting a tennis ball.
+000000014473 A red and black train is traveling down the tracks. 
+000000343149 People walking and standing in front of a building with a clock on it.
+000000429761 Two tennis players on tennis court at net shaking hands.
+000000322895 A living room that has a bunch of different couches.
+000000246522 A spire over a beautiful brick building behind red tiled roofs.
+000000186282 An Apple computer monitor sits on a desk.
+000000338986 a blue and yellow train a building and some cars
+000000418062 Two men holding wine glasses posing for a photo
+000000409542 a group of people walk on a side walk next to a clock 
+000000195754 A living room with the tv on and a wheel on the wall.
+000000505942 A sign with an arrow point upwards is seen on a street. 
+000000210394 an old bus on the road and people boarding it
+000000430048 a photo of a window with an atm inside sign
+000000294783 Large open floor space with makeshift bed and shelving with books.
+000000400573 there is a man wearing glasses and eating a donut
+000000396274 A row of plants growing in the ground by a building.
+000000464089 A little league baseball player is getting ready hit the ball with his bat. 
+000000104572 a bath room with sinks and mirrors 
+000000233727 A bus drives on a paved road with markings.
+000000021604 A man in a suit is wearing a lit up tie.
+000000491090 A person sits on top of a motorcycle with others.
+000000032285 A bathroom with brown tile and a bare pantry.
+000000483531 Two twin beds in a room with a picture of flowers above it.
+000000085329 Painting of girl that has multiple piercings on face
+000000000872 A couple of men play baseball and the batter runs for base.
+000000295809 Street sign painted on a road that looks like a bicycle. 
+000000166166 A small vase of flowers sit on a wooden table near magazines.
+000000460682 A horse that is laying down in a grassy field. 
+000000168458 A vase full of irises with a pitcher on an end table
+000000140640 A group of women standing next to a cake.
+000000455624 People standing behind bales of straw as a protection from motorcycles on a track.
+000000297396 A plate of desserts with frosting and sprinkles.
+000000187585 a young man is on his skateboard doing a trick 
+000000270908 There is a tennis player laying down on the court
+000000190676 The people was outside about to aboard the plane
+000000363666 a couple of people sitting around a table 
+000000154339 A close shot of some type of  machine, maybe in a factory. 
+000000356094 a man in a baseball uniform holding a bat
+000000349594 A young boy holding a doughnut with a bite out of it.
+000000407298 A young ball player poses with his baseball glove.
+000000166747 a horse walking on a small platform in front of a group of people
+000000364322 A flock of very cute sheep in a grassy field.
+000000304404 A man swinging a baseball bat on a baseball field.
+000000358427 A piece of broccoli on a metal fork.
+000000523033 A few people are riding a wave on a surfboard.
+000000158956 The woman is getting ready to cut a piece of the desert.
+000000261888 A bicyclist waiting in the roadway until two horses pass.
+000000153669 A man taking a swing of a bat at a baseball game
+000000100624 A woman on her cellphone standing on the sidewalk in a city. 
+000000029640 a recipe consisting of ham broccoli and carrots
+000000369503 A kitchen counter and cabinetry with a large window.
+000000239347 The picture of the couple in bed is displayed in the large window.
+000000450303 Several students sit at a conference table with their laptops.
+000000286503 an elephan standing around next to some people 
+000000092939 A couple holding a knife and cutting their wedding cake together.
+000000256192 A yellow police motorcycle parked on a road near a large crowd behind a barrier.
+000000243495 A bathroom with a soap dispenser, toilet and rolls of toilet paper.
+000000415990 A bunch of cows that are standing in the grass.
+000000411817 A group of people playing with interactive gaming units.
+000000448263 A man and a boy in red jerseys on a baseball field,
+000000338304 A herd of sheep that are walking through a large group of people.
+000000532530 The subway stop Square Victoria entrance and the map of the neighborhood
+000000464824 A man getting water from a fire hydrant on a sidewalk.
+000000023272 a cat is sleeping on the hood of a car
+000000051712 a man standing on a pair of skis and coming down a snow covered hill
+000000249550 A bed is shown with a rug and side table.
+000000220858 Row boat at the beach's edge ready to go.
+000000215259 Two women play a game in a living room.
+000000234779 A sandwich is sitting in a white plate on top of a wood table.
+000000193494 A makeshift bedroom is decorated in plastic and cardboard.
+000000467176 A group of people are gathered around at television playing WII.
+000000205282 A group of giraffes that are standing in the grass.
+000000249550 A spare bedroom with a bed with a flowered blanket.
+000000302882 a white male taking pictures with his cellphone
+000000005600 A pan filled with onions sitting next to a pan of stew.
+000000383621 A white plane flies in the cloudy sky.
+000000414385 Man riding a skate board towards the traffic on a two lane roadway.
+000000491683 horse grazing in a large pasture with a train in the background
+000000002923 ships in the distance behind a grassy marsh with egrets
+000000324715 A white Macintosh next to duct tape and red scissors.
+000000365886 A picture of a man giving a thumps up with a cell phone in hand. 
+000000378454 A person jumps in the air and catches a Frisbee at the beach.
+000000577959 an image of a man walking on the sand
+000000014226 A person is sitting in a chair at a table with a laptop computer.
+000000087875 A fire hydrant beside a leafy green bush on a green lawn.
+000000162130 A wooden bench in a grass field with a tree.
+000000017436 A man sitting on a bench with a church behind him.
+000000221213 A long train filled with tanker cars on rail road tracks.
+000000253695 a close up of a baseball player catching a ball
+000000001532 Some cars on the freeway are exiting onto Sunset Blvd.
+000000369370 A table with plates of sandwiches and fruit.
+000000166287 A group of cows standing on top of a shore near water.
+000000172649 A person hitting a half pipe on a skateboard.
+000000390246 A man in wetsuit on surfboard riding a wave.
+000000427160 a bunch of little boys running around on a baseball field
+000000308476 A large white bear standing near a rock.
+000000302882 The man is taking a photo of him self with his cellphone. 
+000000292024 A woman holds a fork and looks at a pizza.
+000000061268 A train passing through a station, in the middle of the day.
+000000342128 A man with shorts and long tube socks holds a racket
+000000364884 A person riding a snowboard down a snow covered ramp.
+000000435880 A kitchen has a refrigerator and ice chest. 
+000000493613 The upturned skateboard is below a floating pair of shoes.
+000000281414 A girl is sitting on a wall and looking down at her cell phone.
+000000450758 This is a field of giraffes roaming together.
+000000145020 Donald Duck talking to Minney Mouse at Disney World at Christmas.
+000000360951 this si a desk with two computers on it
+000000538236 A pile of a pigs in a blanket.
+000000482477 blue and white bird standing on a branch
+000000134096 A cat drinking water from a bathroom faucet.
+000000533493 Three men playing frisbee with a blue sky in background.
+000000134689 A giraffe is standing by some brush in a field.
+000000462031 a black and white photo of a baseball player
+000000526392 Cars parked along a street with a street light that is green.
+000000478721 Two men carrying plastic containers walking barefoot in the sand.
+000000507575 A computer desk with electronics on it and other desk accessories 
+000000530975 a close up of a hot dog on a table 
+000000408774 a man catching a ball in the outfield during a baseball game.
+000000102411 Two people riding a motorcycle to the beach.
+000000473118 A skateboarder is turning his board at the top of a ramp
+000000046804 A lonely sheep standing in a field in front of a rock wall.
+000000563702 A fire fire truck parked in a parking lot.
+000000026564 An office desk with a row of books and a desktop computer and laptop computer on the desk.
+000000466125 The child smiles brightly while being handed a pink umbrella. 
+000000025393 Two men standing on the street wearing a suit and tie
+000000579902 and man with a skateboard on a motorcycle in traffic
+000000579655 A woman holding a phone in one hand up to her ear. 
+000000253386 A black and brown dog is wearing a heavy metal collar.
+000000463918 Two boys standing up playing a video game.
+000000327701 A woman hitting a tennis ball with a racquet
+000000538236 A pastry displayed on a wood table in a store setting.
+000000263474 A bathroom area with a toilet, trashcan and dresser.
+000000563604 A sign outside of a closed road redirecting traffic
+000000213830 A couple of men standing on top of a tennis court.
+000000349184 A woman sitting on a wooden bench viewing some beautiful trees.
+000000272136 Some type of vehicle with a number on its side is pared inside a garage.
+000000353027 A person grabbing a slice of pizza with cheese and mushrooms.
+000000328286 A city ferry sits docked with the city in the background. 
+000000324818 Two multicolored exotic birds perched on a wooden branch with trees and flowers behind them.
+000000414638 A white plate with a sliced up piece of food on top of it.
+000000116362 A small bowl filled with chopped vegetables and spoon
+000000556765 An elephant with red on it's face in the water. 
+000000260261 A man standing behind a truck filled with produce.
+000000555412 A pizza is shown with various toppings on it.
+000000010977 A toilet lid is closed next to a bathtub with a white curtain and pink tile wall.
+000000078170 People sitting by the windows in a train.
+000000407524 A pasta dish with mushrooms and broccoli in a cream sauce
+000000263796 An old toilet painted red with a wooden lid. 
+000000527616 A picture of someone in the TV on the wall.
+000000405249 A group of kids at a table with a cake.
+000000053529 A dog with a hat and a person in a truck.
+000000405279 A filed full of people flying colorful kites.
+000000084362 A toddler and cat are in a living room with open curtains.
+000000131131 A cat sitting in front of a large computer monitor.
+000000003553 A person wearing sneakers balancing on a skateboard on a wooden platform.
+000000468954 A man and girl playing with an interactive video game system.
+000000210273 A highway filled with lots of traffic and buses.
+000000576031 a woman attempting to ski on a flat hill
+000000206994 A yellow and red train riding into a station.
+000000445602 A skier on a snow covered mountain resort.
+000000377239 A woman holding a cigarette smiling at the camera on a sunny day.
+000000397639 A ewe and her kid in a field.
+000000209753 A woman looks down at her phone in her hands.
+000000081988 A group of kids being instructed on how to boogie board.
+000000044068 A teddy bear sitting on a chair inside a room. 
+000000540280 A light that is sitting underneath a umbrella.
+000000134856 Three people on horses riding on the beach.
+000000173302 An empty restaurant with two employees working behind the counter.
+000000438269 A tall white bird standing on a lake in a swamp.
+000000426376 The person holds both hands in the air while snowboarding.
+000000091615 A kitchen is shown with stainless steel appliances.
+000000579158 A red yellow and blue airplane is sitting on the runway. 
+000000309467 Two people cross country skiing on a snow covered hill.
+000000291664 A dog standing next to a fire hydrant on a green grass covered park.
+000000026564 a desk with a laptop some keyboards and some books
+000000562243 A man wearing red shirt, tie and glasses.
+000000446522 A dog is sitting on an armchair next to a fridge.
+000000417043 A bench at the end of a bridge
+000000245320 The skateboarder is jumping near a metal fence. 
+000000349594 A child holding chocolate donut with both hands.
+000000526197 THERE IS A PASTRY THAT IS SITTING ON A PLATE
+000000222317 A dog lying on a couch with blankets and pillows.
+000000424776 A jet fighter flying through the sky above a snow covered mountain.
+000000265816 A horse, buggy and driver sitting in front of a building.
+000000094157 A large brick clock tower stands high above the trees.
+000000025593 A street sign in an asian language in front of a blue sky.
+000000559956 Two women are squatting down and petting brown and white long haired goats. 
+000000508101 People and their stuff on a little boat in the ocean
+000000222317 A dog curled up on a couch with blankets and pillows.
+000000160556 a person sitting at a desk with a monitor and keyboard
+000000088040 Plates of shrimp, vegetables, and noodles served with tea.
+000000370999 A dime surrounded by a bunch of stones with holes in them
+000000482487 Two small clocks sit behind a glass window.
+000000400367 A city at night with cars driving by on a street. 
+000000414673 A boy prepares to hit a ball at a baseball game.
+000000569565 some construction stucks pared on the side of the road
+000000012639 A little-league baseball game with a kid up to bat.
+000000515982 A baseball player sliding into home plate, in a game.
+000000482719 Plums and bananas are in a glass bowl. 
+000000385205 A cat sitting on a rug scratching its ear
+000000051326 A group of Halloween Jack o' lantern sitting on a floor with flowers and paper cut outs. 
+000000553788 there is a woman laying in a bed using a lap top
+000000217285 A baseball player holding a baseball bat during a game.
+000000188906 Food that has been neatly prepared for a presentation. 
+000000228214 A Marine that is looking at his cell phone.
+000000189828 Grey plane taking off above some green vegetation.
+000000399205 All of these men are using their cell phones.
+000000230008 A police officer is outside on his bike.
+000000533816 The people are standing and talking on the tennis court.
+000000361621 close up of a cat sitting in a bathroom sink
+000000217957 An beautifully designed clock tower with gold and blue. 
+000000429761 Two tennis players shaking hands over the net
+000000186282 A desktop computer monitor sitting on top of a desk next to a mouse.
+000000423798 Beams of light coming off of different objects on a city street. 
+000000490125 White birds are gathered around a water puddle.
+000000274708 Three skiers skiing down a steep snowy slope.
+000000178028 A single basin bathroom sink with a granite vanity.
+000000378116 A person riding a wave on top of a surfboard.
+000000068409 School class photo from the boys school of Standard South Carolina
+000000038576 A computer, mouse and keyboard on a desk.
+000000341719 A group of people with skis and snowboards stand on a mountain.
+000000033109 A large semi-trailer truck driving through a town area
+000000253695 outfield baseball player looking up to catch fly ball
+000000172396 A colorful square shaped pizza in an oven.
+000000561009 A brown bird has a small yellow head.
+000000303566 Boxes of fruit displayed at an open air market.
+000000129113 a burnt tea spoon sits on a wooden surface 
+000000011615 a stack of traffic signs on a pole next to a street.
+000000409424 A basket of fruit is sitting on a table.
+000000059920 A bathroom with a toilet and a drain in it. 
+000000222863 An animal with horns wears a rope from its nose and ear to its neck.
+000000155179 Closeup of a cake that is sitting on a white plate.
+000000164115 A surfboard is stowed in a rack alongside a walkway.
+000000167540 A small engine plane flying over in the blue sky.
+000000242946 Three woman looking over a table full of food.
+000000176901 Three skiers climb up a snow covered hill.
+000000489764 A woman and a man playing a video game in the living room
+000000222094 this is a stop sign in a rocky field
+000000458410 A room with an old sofa, coffee table, lamp and t.v.
+000000114884 A group of people standing in a large building.
+000000005529 A person with green skis skiing down a big hill.
+000000470121 A table topped with red trays of food.
+000000383838 A woman that is looking at a white remote.
+000000414170 Three people on packed snow trail, two skiing, one walking.
+000000519338 a train sitting on the tracks next to a mountain side
+000000302030 Laptop computer on long desk with assorted paperwork.
+000000154718 A white toilet filled with crap with two pink roles of toilet paper.
+000000407403 A white vase holds some pretty yellow tulips in this still life study.
+000000177893 A red and white bus parked in front of a sidewalk.
+000000195918 An office work station with two computer monitors, and laptop.
+000000092416 A male with a bottled beverage wearing a flower in his lapel.
+000000292908 A person on a surfboard riding a wave in the ocean.
+000000516677 A man is lacing his boots while several others are ready to ski.
+000000481573 A person that is on top of a skateboard.
+000000542127 Two skiers on a ski slope doing a flip in mid air.
+000000033638 A woman is cooking on a large black stove
+000000098520 An airplane makes its way down a snowy runway.
+000000262048 a couple of people lay under an umbrella on a beach 
+000000397303 Two men at a table under a tent selling neckties.
+000000167353 An old clock hangs on the wall and gives the time.
+000000356125 A elephant reaching out with some thing with its nose. 
+000000321887 A sheep stands in front of a box of colorful objects.
+000000476258 A man on a long exposure picture riding an electric skateboard.
+000000284279 The bird is standing on the rim of the bird feeder.
+000000340175 The large room has a wooden table with chairs and a couch.
+000000314541 A bench that has been placed in a field.
+000000521601 Closeup of a white and yellow vase with a red circle at the bottom.
+000000480842 two giraffes and some green trees and yellow flowers
+000000243495 a toilet with writing under the lid 
+000000269682 Traffic lights and camera under a stormy sky.
+000000066771 A man and women sitting at a booth photographing tea.
+000000338191 Different fire hydrants have different looks including bright colors and stand in dirt and grass and the sidewalk.
+000000140203 A vehicle displays a customized license plate and milk advertisements.
+000000151480 The baseball team getting ready to walk off the field.
+000000508730 Two children are practicing using the potty next to each other.
+000000258541 A man posing for the picture in a building 
+000000440336 a group of people throwing around a freez be
+000000360960 A person walking while carrying a rainbow umbrella
+000000061171 A herd of black and white cows eating dry grass.
+000000377239 a beautiful lady walking with a pink umbrella holding bottled water and a cigarette.
+000000170739 a pack of elephants standing next to each other in a pen
+000000219271 A tow truck pulls a large crate down a mountain pass as crowds watch.
+000000235857 Some cows eat grass together in a field. 
+000000067213 A dog competing in a long jump in to water contest
+000000275749 A lot of people that are in a room together.
+000000163951 a small stereo a banana and a flashlight
+000000077396 A cat being lazy and a cat being nozy in a living room with tv and a laptop displaying the same things.
+000000061171 Cows and ponies are eating hay in the barn
+000000139077 a little kid playing a game with the wii driving remote
+000000108026 A gamer holding a controller and using a keyboard
+000000468245 The unmade bed has three pillows on it.
+000000350488 A single zebra grazes on a ledge near the water,
+000000572620 a man with a hat jumping a barrier with a skate board
+000000184611 A man drives his bike taxi with luggage in the back.
+000000023937 Several cows sitting out in the grass of a green pasture.
+000000414385 View from inside of a car of cars on road and a man on the side of the road.
+000000128476 Two women are looking at many colorful, heart shaped cakes. 
+000000153632 A plate of food at the table with meat
+000000185157 A blue traffic sign sitting on top of a pole.
+000000411665 The kitten is pawing its own image while gazing into the hand mirror.
+000000491216 A cat is walking in a kitchen that has cookware hanging on the wall.
+000000020059 Two zebras graze on grass inside of an enclosure. 
+000000486040 A computer on a desk with a splatter screen saver pattern.
+000000186632 A flat screen TV sitting in a living room in front of a dining room table.
+000000281414 A beautiful woman sitting on a brick wall.
+000000086483 A sign that says "Abbey Road NW8 City of Westminster".
+000000377723 People line up to take a city bus.
+000000360960 A person walking in a square carrying a rainbow colored umbrella.
+000000393469 A person is standing in the white snow.
+000000488592 A very tall and big room with some people.
+000000356248  a man holding onto a little blond girl 
+000000336053 A young boy and young girl sitting at a table eating food.
+000000311928 A young boy swinging a baseball bat towards a ball.
+000000349152 A few boxes of glazed donuts on a table.
+000000023781 a variety of different vegetables and some strawberries
+000000166918 Long dining room table filled with people in dress clothing. 
+000000303893 Two kids witting on a sidewalk having a snack.
+000000158956 A woman cutting a cake with a long knife.
+000000458054 There is a row of used and broken toilets in a field.
+000000324715 An apple computer is on a table with supplies.
+000000025986 Salad, rolls and entrees are served on a stainless steel cafeteria counter. 
+000000558213 A man in black shirt doing a trick on a skateboard.
+000000300039 A person holds a smartphone taking a picture of a bottle and glasses on the table
+000000224222 Three surfers are close together on the waves.
+000000017714 A table that has plates of food, cups, and silver ware on it.
+000000452793 View of kitchen sink, stainless refrigerator and breakfast room
+000000434247 A couple of people are riding horses on the beach
+000000147745 Two trucks parked next to each other on a dirt road.
+000000509014 This is a picture of skiers enjoying their day on a mountain.
+000000419379 The carrot cakes look like doughnut on display.
+000000415238 A painting that shows a vase with flowers and a table.
+000000453634 A bathroom scene with focus on the toilet.
+000000284445 A light post leans towards a crowd on a busy city street.
+000000385719 Three computers are shown at the wooden desk.
+000000119038 Several bovine look out over the hills on a cloudless sunny day.
+000000482917 A dog and a person are watching television together.
+000000073118 Two large identical teddy bears are piled against the wall.
+000000328238 A man catching a frisbee in a park 
+000000210299 A little boy wearing a helmet and riding a bike.
+000000480275 A wooden wagon with wheels topped with bunches of bananas.
+000000352584 The toilet is covered in sparkles with a red object in front of it.
+000000051712 A man with goggles on snow skiing on a mountain
+000000138241 Plates of food have a tortillas and meat on them.
+000000546964 A room filled with furniture and very large rugs.
+000000061960 A PERSON LAYING ON A BED WITH THIGH HIGHS AND HEELS
+000000296231 A television and some books in a room.
+000000354072 A bathroom sink under a mirror next to a plant.
+000000297085 A cat sitting in front of a flat screen TV.
+000000465179 A woman is brushing her teeth and posing for a picture.
+000000361142 Cars driving by an old building and clock tower.
+000000490515 A man is falling on his back while snowboarding.
+000000361268 A brown and white cow grazing in yard next to street.
+000000191761 Large moths atop a plate of bananas and citrus.
+000000467511 A biker chic in high heel boots is talking to a man at a gas pump.
+000000251065 The small bathroom has a toilet with a black seat.
+000000088269 A plate holding a grilled cheese sandwich and bowl of soup.
+000000302030 The desk with a laptop has papers strewn all over it.
+000000245026 A woman in glasses holding up a cake with frosting.
+000000187243 A woman laying on a bathroom floor next to a toilet.
+000000500663 Cattle are grazing in a bright green pasture along a river and mountains.
+000000224222 a surfer is riding a wave in the ocean with some swimmers
+000000227491 A woman that is sitting outside on a bench in the snow.
+000000242060 A dish of Flan sits on a table next two drink glasses.
+000000557884 A plant hanging from a rack on a white wall.
+000000372317 A bus that is parked along side a curb
+000000165257 a clean kitchen counter with a clean sink
+000000025057 a man holding a drink with a frizbee in front of him
+000000220764 A person cross country skiing across a snow filled plain.
+000000270474 A blond girl holding a baseball bat and smiling with her hand over her mouth.
+000000553990 A woman riding a horse jumps over an obstacle.
+000000526751 A boat is sitting in the sand near the water.
+000000183716 Two young girls in uniforms sitting closely together.
+000000507081 A kitchen with a stove, sink and coffee maker.
+000000500211 A row of closed brown umbrellas next to a building.
+000000519039 A white service truck passes a neighborhood intersection.
+000000214224 a home bar with different drink ingredients under a large decorative sign saying, "PUB".
+000000436617 THERE IS A GROUP OF PEOPLE SITTING AT A TABLE 
+000000157847 a bird that is flying high over the sky
+000000475064 A person holding a piece of food in front of a sign.
+000000561465 The table has meat and donuts sitting on it.
+000000070048 a table that has some food on top of it
+000000163257 A man in a florescent green hoodie holding a fuchsia umbrella.  
+000000424545 A tabby cat lying in a window sill looking up at the photographer.
+000000067534 A man wearing a collar shirt with a blue tie
+000000544052 Action figures put on skateboards on the street
+000000025393 Two formally dressed guys are crossing the street. 
+000000567011 A person riding a beautiful wave very smooth
+000000028285 The view of a top of a buildings spire showing a small clock on it.
+000000465585 A large dish of chopped broccoli and cheese.
+000000414795 A herd of elephants  being herded into a river while tourists take photos.
+000000533145 A hand holding a Nintendo Wii game controller.
+000000465822 Several people in a room having a meeting while someone is recording it.
+000000151820 a group of people sit at a table eating
+000000576955 A man is guiding a horse through a body of water.
+000000216277 A half-eaten apple sits next to a cell phone.
+000000438876 A man swinging a tennis racquet at a tennis ball.
+000000255749 City bus stopped, taking on passengers, at night.
+000000529966 Two clear glass bowls full of oranges on a bar.
+000000095155 Four persons are skating on the skate board on snow.
+000000501368 A person wearing a towel is drying their hair with a hair dryer.
+000000062554 a plate of broccoli on a table next to some other plates of food 
+000000089761 A toilet missing the back section sitting next to a tree.
+000000403122 a girl playing tennis with a bad knee
+000000035062 A person in a bed with a dark bedspread with flowers.
+000000191471 A Stop sign with rain drops on it near some buildings.
+000000140987 A cute little blond girl sitting next to a pizza.
+000000229111 a boy holding a lego wii mote to a real wii mote
+000000473118 A picture of a person doing a skateboarding trick.
+000000060932 a couple of people that are dressed up dancing
+000000287649 A cat sits on a desk next to a keyboard. 
+000000506178 The men are playing video games on the wii together
+000000209757 A man in a baseball cap riding a skateboard grinding an edge.
+000000021604 a man in a suits fixes his tie 
+000000195754 A living room with furniture and a television.
+000000520707 Several people are seen walking through the airport while waiting for their bags.
+000000488673 A couple celebrating a birthday in a kitchen.
+000000025386 two people on a train with trays of food 
+000000263966 Looking through a circle at a horse as it stands near a fence.
+000000405195 A display of a variety of donuts at a store. 
+000000266082 A view of a modern building with skylight and a fire hydrant.
+000000024919 A couple of elephants standing in the grass.
+000000567740 Two people with skis on at the top of a mountain. 
+000000323751 A train one track away from a platform at a stop.
+000000546626 An apple, a phone, and a cup sitting on a table. 
+000000425226 A man kneeling down in front of a refrigerator.
+000000484296 A bunch of zebras in a park somewhere.
+000000520707 A cop standing next to a crowd of people.
+000000223090 A knife, carrots, and cabbage on a white cutting board.
+000000551804 A woman holding a tennis racquet on a  tennis court.
+000000504580 Two giraffes standing one behind the other among brush. 
+000000550349 a group of people waiting for the metro bus
+000000116206 A cutting board and knife, with chopped carrots, celery, and onion.
+000000146358 A couple that is posing for a picture.
+000000148783 Two zebra grazing on grass next to each other.
+000000320490 A boy swinging a baseball bat at a ball.
+000000072813 Brown dog lying on unmade bed in bedroom of home.
+000000193717 there is a red fire hydrant with a chain that is attached to it
+000000039956 A bedroom in an old house that is being renovated 
+000000025096 A boy cutting a cake in the shape of a skateboard.
+000000187055 An older man catching a fast moving tennis ball.
+000000532129 A delicious pizza sets on top of a plate. 
+000000263403 A surfer goes airborne on top of a wave in the ocean.
+000000023230 ducks in a body of water with trees in the background
+000000426795 A young man with long hair taking a bite out of a full loaf of bread.
+000000080340 A kid standing with a small suitcase on a street.
+000000472046 A large modern kitchen has pineapples on the counter.
+000000269682 An intersection with many traffic lights  and a traffic camera visible
+000000302882 A man is taking a selfie using a mirror.
+000000087875 a fire hydrant near a tall bush in a field 
+000000187585 Grabbing the front of the skateboard maintains control over it.
+000000423123 A young oerson is raiding a small fridge in their room.
+000000357816 a little boy holding a baseball bat by a fence
+000000452784 The cook is rinsing a draining the greens at her kitchen sink.
+000000221754 a city street with multiple bildings and a street light
+000000407650 A person sailing in the air on a snow board.
+000000153510 A green plate filled with rice and a mixture of sauce on top of it.
+000000580757 A red, white and blue fire hydrant on a city sidewalk.
+000000216516 The snow skier is happy to pose for a picture.
+000000313562 A woman standing in a kitchen touching a pile of carrots.
+000000468577 a girl toddler laying on a day bed 
+000000338901 A large brown dog laying on top of a blue couch.
+000000198915 a close up of a street sign on a pole 
+000000457262 Two very small yellow bananas on top of a wooden table next to a Malaysian coin. 
+000000362682 A Japanese community bus with a cartoon dog on the side
+000000084241 They are all working in the kitchen together.
+000000572303 This is a train on the tracks that is filled with doors for houses. 
+000000145665 A person lies down along the lines in a parking space.
+000000198928 A woman stands next to a traffic light.
+000000195754 A living room with comfortable furniture and large tv
+000000002587 A banana and donut in the same plastic bag.
+000000309964 a woman and man sit under what appears to be a large blue umbrella on a grassy lot 
+000000338905 People sitting around a table eating food outside.
+000000322864 Two guiding signs are pictured in front of a busy street.
+000000297427 A cheeseburger is topped with a glazed doughnut.
+000000271997 A portrait picture of a young man in a suit and tie.
+000000179141 two women and two men smiling over a pizza 
+000000163117 Large crowd gathered to watch decorative a fleet of decorative kites.
+000000347693 A couple of windows sitting inside of a room.
+000000273617 A light shining on the street at night time
+000000581615 A wall urinal in a stall with an auto sensor flushing system.
+000000405432 Donuts and a bag of beans on a table with remote.
+000000328430 Woman tennis player returning a tennis ball. 
+000000352684 A man in shirt and tie standing in lobby next to other people.
+000000115245 A table topped with bags of luggage and purses.
+000000407002 A person wearing a red coat and skis posing for a picture.
+000000495732 Two men standing in a living room playing games.
+000000341094 A statue of a large brown bear tearing off a cars door.
+000000264441 A cat is sleeping in a chair with its leg curled around its head. 
+000000574297 A man in a helmet is riding a horse across a dirt road
+000000051938 A man skiing on top of a pipe.
+000000345361 Four children at an outdoor party wearing festive hats.
+000000100428 two men with ties taking a creepy looking photo
+000000189475 A woman and child sitting at table with food and drinks.
+000000548267 A herd of sheep grazing on a lush green field.
+000000514914 A modern bathroom with a large bathtub and walk in shower.
+000000570782 a desk with two laptops and a monitor
+000000502136 a yellow and brown wall a gray door and a sign 
+000000135670 A bike behind a rail for safety on a train ride.
+000000577864 3 People in a dirt field playing soccer.
+000000467848 A truck transports another tiny truck on its tow.
+000000011699 Two girls standing with their bags and in front of a suitcase. 
+000000325838 this woman is at a desk using a computer
+000000334006 Garbage and police trucks on a city street
+000000350405 A man is snowboarding past blue markers on a mountain.
+000000185950 A boy performs a skateboarding stunt in a skate park
+000000444142 A man holding a tennis racquet on top of a tennis court.
+000000382088 A white horse standing on top of a dirt field.
+000000154705 A computer sitting on top of a white table with another computer.
+000000200839 The library bus has many colorful books painted on it. 
+000000398377 Commuters using their cell phones during a train ride
+000000451308 A stack of many different types of suitcases.
+000000317433 A man behind farm equipment pulled by horses
+000000189213 A pile of junk sitting next to a curb on green grass.
+000000085329 An image of a very cute girl with face piercings.
+000000442456 Man in red shirt and black pants standing next to white flatbed truck cab next to a set of new tires.
+000000569565 Large work trucks sitting on the side of the street.
+000000233238 Gathering of people dishing out food set on a long table with checkered tablecloth.
+000000345356 A man in a gray shirt holds his phone as he sits by a man in a white t-shirt.
+000000059044 A group of young and older people standing around.
+000000549055 A man on a surfboard in the water.
+000000007511 A man is flying a kite at the beach.
+000000500477 A person in the process of sheering a woolen animal.
+000000098392 A downtown clock tower at the intersection of a busy street.
+000000155051 A clock tower at night time lit up near a view of the moon.
+000000255536 some abandoned chairs sitting by and umbrella by a fence 
+000000309964 two people sitting in a field with open umbrellas 
+000000489764 A woman playing a video game with the Nintendo Wii.
+000000537672 A black refridgerator has several things hanging on it.
+000000534041 A woman feeding a little boy french fries at a restaurant. 
+000000015440 a street sign with buildings in the background
+000000581357 A boy doing a mid air trick on his skateboard.
+000000467315 A bouquet of flowers on a marble table in the lobby of a business location with an escalator in the background.
+000000551215 a tennis player running to get to the ball
+000000378244 A person in a ski suit skiing down a slope
+000000104424 Lady standing on the tennis court posing with the racquet
+000000463618 A woman holds a baby while sitting next to a cage with a bird in it. 
+000000117719 Bottles, and glasses are lined up on shelves in a bar.
+000000284698 The image of a man holding a camera is in this motorcycle side mirror. 
+000000256941 a bicycle with a basket and a blue and pink umbrella 
+000000439180 A group of people are riding horses at a park.
+000000279145 A cat sits near a bench in a garden with many potted plants.
+000000213035 Two men are holding phones together to create a linking .
+000000369037 A man standing with an elephant that has painted markings on it.
+000000379476 The long table is set for a formal dinner with  glassware for wine.
+000000267933 Group of white horses walking next to a river.
+000000013729 Three men and a woman playing a wii video game
+000000088345 A woman posing in front of a sculpture at a bar.
+000000513688 A chair next to a desk with a laptop and phone sitting on it.
+000000509403 A man is playing catch with two children and a dog. 
+000000140420 A motorcycle is parked on a dirt road in a forest.
+000000506707 A man in a white shirt and gray pants walks toward a grassy area as kids in baseball uniforms and an umpire are near him.
+000000553511 A city road with cars, traffic lights and road signs.
+000000088951 A man and his dog sit at the park.
+000000508370 A bicycle parked on top of a baby dolls head.
+000000233567 The goats are standing together on the straw.
+000000475223 a large bird catching a gray from in shallow water near grass
+000000356428 The people are walking down the aisles at the super market.
+000000523807 A hot dog that has some ketchup and mustard on it.
+000000350003 A large red and yellow fire truck passed by a white car,
+000000459500 A clock sitting on a post above the statues of two birds.
+000000183500 White and red biplane flying through the air. 
+000000237984 a bench sits in front of a chair 
+000000511599 People are standing on a dock with their baggage.
+000000263679 Two people on a beach flying  akite
+000000440507 A turquoise suitcase and two pairs of shoes sit by the door.
+000000013177 A man sitting on the ground, fixing a motorcycle wheel.
+000000186282 A computer screen has many colors, with a mouse and speakers next to it on the table.
+000000170613 Surfers holding boards standing in water at ocean.
+000000411953 A man with a guitar in front of a microphone.
+000000495054 a plane lands on a big wide run way
+000000491867 A young bald man dressed sharply smiles happily
+000000311909 The #41 bus parked at the curb waiting for passengers to board
+000000349480 A nice setup of stuffed bears having a picnic. 
+000000283268 A white plate topped with a toy hot dog in a toy bun.
+000000247806 We are looking up at an old clock tower.
+000000470924 Three men and one woman sit at a table eating pizza. 
+000000475779 An elephant is walking through the green shrubs.
+000000223955 A person with long hair about to throw a frisbee
+000000072281 A man that is in the air with a surfboard.
+000000017031 A giraffe is coming up close to people
+000000350488 A zebra near a waterhole grazing on some grass.
+000000499313 A pizza with cheese and spinach on a plate.
+000000391648 A tower with a clock on it during the night.
+000000402615 A tennis match takes place on an orange court with full stands.
+000000565563 A narrow sidewalk with cars parked on both sides of the street.
+000000227187 a table that has some bananas on it
+000000337498 A sliced of pizza is served on a white plate.
+000000146155 Two men that are standing together holding glasses.
+000000184400 a train on a train track on an elevated bridge
+000000502136 A building wall and pair of doors that are open, along with vases of flowers on the outside of the building.
+000000410487 A view of a oven with the food flipped over in it.
+000000039914 A lady and her child outside playing with a kite.
+000000062808 A table topped with a pizza surrounded by men.
+000000006954 Five small boys sitting on the grass holding two frisbees
+000000143998 A bunch of carrots sitting on top of a cutting board.
+000000097337 A living room with a chair, couch, coffee table and television.
+000000455872 Several old boats sitting in an old ship yard.
+000000365642 A woman sitting on a bucket on the side of a street. 
+000000385997 A DOG QUIETLY SLEEPING IN HIS BED ENJOYING THE SUN.
+000000213605 A congested road where different vehicles are running.
+000000012748 A man holding a baby whose petting a horse.
+000000093437 A shot of an elderly man inside a kitchen.
+000000146831 a person that is standing on a skateboard
+000000341828 A man holds phone up to his ear while eating.
+000000129135 Yellow Train stopped at the station in front of a bench.  
+000000256407 Ornate cup and saucer clock displayed against red background.
+000000363188 A group of people walk down the street dressed as zombies.
+000000211674 A double decker bus driving down a street.
+000000474170 A couple of elephants standing on a lush green forest.
+000000227044 A cat stands in a white bathroom sink.
+000000376365 A good luck plant is in a round vase.
+000000269316 Four men playing Ultimate Frisbee in a park.
+000000462728 A dog surfing a small wave with his owner swimming behind him.
+000000551820 A tennis plare takes a swing very close to the net.
+000000146831 The kid is on a skateboard and on a ramp. 
+000000508586 Group of zebras walking through tall grass together. 
+000000516143 a green and white bus is on the street
+000000312263 The clock tower is on the first floor of the tall building. 
+000000484760 Statue and old buildings shot at an angle.
+000000206487 An old bus makes its way through an intersection of traffic.
+000000518770 A group of young children standing in a room next to each other.
+000000413552 A woman holds her small child in her lap.
+000000267300 A dog eating cookies from a plate on the floor laying on a dog bed.
+000000379476 many tables next to each other in a row
+000000417043 A bench sitting on the side of a road.
+000000016502 A sheep standing on top of a rock. 
+000000125778 a living room with a couch and a chair 
+000000436551 two black birds are swimming  in some water
+000000366178 There is a street sign on a pole on the side of a snowy highway.
+000000000139 A woman stands in the dining area at the table.
+000000031749 A painting of a doll sitting down and a clock close by.
+000000368900 A small apartment has many belongings in it
+000000216516 A man standing on skis at the bottom of a snow covered hill.
+000000397354 A group of friends hanging out in a kitchen together.
+000000242411 A clock on a white building in a city
+000000198805 A group of trucks and cars parked in a parking lot.
+000000338986 A train pulls up along some buildings. 
+000000125072 Cattle grazing on grass near a lake surrounded by mountain..
+000000405970 A very nice looking room with some nice windows.
+000000128051 A van is on the roof of a bus of the same door.
+000000283268 A person holds a plastic hot dog in a plastic bun.
+000000193926 multiple shots of a banana and a banana guard
+000000481480 Two jockeys ride their horses in a race at the track. 
+000000344888 A group of horses standing next to each other.
+000000267300 A tan dog eating food scraps from a plate.
+000000433243 a zebra and a giraffe stand next to each other.
+000000541773 A man standing in front of a wooden table covered in glasses of wine.
+000000546829 A small white dog on a wooden bench.
+000000383676 Two giraffes stand near rocks in a zoo.
+000000173183 Two street signs atop a post beneath several trees.
+000000281929 A man that is standing next to a bicycle in the grass.
+000000486104 A baseball player holding a bat while standing on a field.
+000000351559 Green traffic light shown against a tall sky scraper at night.
+000000460347 A bus and other cars driving down a multi-laned street .
+000000350019 A little boy is playing with a suitcase.
+000000025424 A tennis player hits the ball up from the racket.
+000000190236 The home office space seems to be very cluttered.
+000000487583 A white toilet and cup in a room.
+000000000785 A young woman is skiing down the mountain slope. 
+000000231088 Six umbrellas floating in the air near a building.
+000000391722 A man presents a cake with lit candles on it to a seated man.
+000000296284 A display case in a bakery filled with donuts.
+000000191471 A red stop sign sitting on top of a metal pole.
+000000329614 A busy street by large Gothic style buildings.
+000000500565 A woman is holding a baby who is wrapped in a towel and holding a toothbrush
+000000315219 A lan line phone is hanging on the wall
+000000132587 A man sitting down on a bench beside a water bottle
+000000063552 A cat looking directly into a magnifying glass on a table.
+000000481582 Three people are talking and laughing while in the stable with a horse.
+000000103723 a person standing close to an elephant 
+000000110282 A stop sign that is on the side of a pole.
+000000337987 A feathered bird is sitting on a tree branch.
+000000359937 A yellow, white, and brown double decker bus parked in a lot.
+000000172083 Counter top inside a house designed with different colors. 
+000000097994 Four computers and keyboards arranged on a desk.
+000000182923 A woman jumping in the air on the tennis court
+000000442822 Several rows of boats docked on a body of water.
+000000473219 A man and woman look into each others eyes while getting married.
+000000185472 A train approaching a station with a man and a bike on it. 
+000000210099 A cat sitting on a chair outside in the sun.
+000000214224 A sign with the word "Pub" hangs above several liquor bottles.
+000000441442 A professional rider is riding a horse outside
+000000486438 A plate of donuts with a person in the background.
+000000562843 A close up of a cell phone, scissors and a cup.
+000000036494 People sitting at tables inside a large clock tower.
+000000226802 An Asian city square, with people, buses, and a McDonald's.
+000000404805 A woman on a boogie board riding a wave.
+000000368038 A row of train colors of various colors next to trees.
+000000537672 A black fridge that is reflecting the room.
+000000286907 A blue toilet sitting in a corner of a bathroom.
+000000531495 a dock with some boats parked there 
+000000060855 Five oranges and two applies sitting on a white surface.
+000000328286 A city's downtown skyline at dusk on the waterfront.
+000000494759 Two people on a beach flying a kite in the air.
+000000097278 A young person is on their ski board in the snow. 
+000000102411 A couple sit on a motorcycle at the beach.
+000000434548 a baseball player throwing a pitch from the mound
+000000051598 a garbage bag in a white lighted bathroom
+000000056127 Fish eye angle view of small kitchen with fire extinguisher at far end.
+000000483531 The room is equipped with two seperate twin beds.
+000000014380 The white bridge stretches out over the horizon as the transport train travels below it.
+000000525155 A man carrying a sheep on his shoulders. 
+000000292415 A person on a skateboard on a skate ramp.
+000000126226 Many caps and gloves sit in front of the players in a baseball dugout.
+000000563281 A little boy in pajamas playing with his toys.
+000000369812 A city street with cars and street signs giving directions
+000000297595 A person in a black wetsuit surfing on a wave alone
+000000074209 A black and stainless steel kitchen with a snack bar and bar stools. 
+000000210502 A train with a striped door waiting on a train track.
+000000484893 A couple of kids petting sheep inside of a corral
+000000154358 A view of a living room that appears to be very elegant.
+000000440171 Man riding an elephant up a hill near a field.
+000000475191 a tennis player with a racket on a court 
+000000013659 Adults working on projects and computers in office environment.
+000000446574 A toilet sits next to a bathtub in an empty bathroom. 
+000000475191 A women looks toward the tennis ball she just hit.  
+000000148739 Woman in black body suit surfing on a small wave.
+000000369442 A group of people standing around a herd of animals in a city.
+000000350607 A bench on a road side with a quilt covering it.
+000000008762 A dusky evening with a green street light shining.
+000000082765 A bed and window in a small room.
+000000456559 Man posed in flashy clothes, with hat and sunglasses.
+000000579070 A group of people sitting at a table with many small sip cups.
+000000398905 A person is sitting at a keyboard near a microphone.
+000000576654 A person in a dry area with a sail high in the sky
+000000393838 A pan filled with celery, onions, and carrots.
+000000359540 Several baseball players in the middle of a game.
+000000579635 A man surfing in the ocean with a sail boat behind him.
+000000197388 Five skiers race each other on the snow.
+000000510095 A man in casual wear holding a baseball type bat.
+000000518770 Some kids standing up playing wii with each other
+000000220732 Many trucks and city buses drive down the crowded street.
+000000065798 A baseball game with a bunch of people in the stands. 
+000000521282 A vase filled with yellow and white flowers.
+000000464251 A small red bird perched by a wooden flower box
+000000235252 A herd of giraffe walking along a grassy tree covered plain.
+000000348243 A large black horse standing on a field filled with green and brown grass.
+000000179642 a tennis player sitting in a chair shirtless
+000000168593 A small refrigerator in a small kitchen with a window.
+000000461009 A man holding an apple between his fingers.
+000000226802 A red bus driving down a busy city street surrounded by tall buildings.
+000000464089 A catcher is ready to catch the ball after it crosses the plate.
+000000118594 the cog is grazing on the grass, he has big horns
+000000460494 A red plate topped with broccoli, meat and veggies.
+000000455597 A woman is standing in a professional kitchen.
+000000333402 A man is standing outside of a bus on the curb.
+000000435206 People and cattle standing at the waters edge on a bright sunny day.
+000000351823 A man taking a swing at a tennis ball
+000000002685 The group of people is standing at a wine bar. 
+000000498709 A view of a motorcycle. The picture is very close.
+000000033368 A tennis player is standing on the court.
+000000481386 A young girl that is in a very clean and well stocked kitchen. 
+000000176847 there is a black bird sitting in a berry tree
+000000019432 A tennis player with a racket hitting the ball
+000000132703 A guy doing tricks on his kate board
+000000443969 A GIRL IS IN A GROCERY CART 
+000000532575 A large brown and black dog laying on the back of a boat.
+000000092124 There is a tub, a toilet, and a sink in the bathroom.
+000000082986 Two men watch another throw a Frisbee in a field. 
+000000227898 A horse pulling a large carriage down the street with several people on the carriage.
+000000122962 A group of children sitting at a long table eating pizza.
+000000570736 A small bathroom is featured in this photo.
+000000451150 The six donuts are made with various seasonings.
+000000437205 a close up of a child holding a banana
+000000056344 a desk with a keyboard, mouse and two monitors on it 
+000000284725 A bus in China is parked along a curb. 
+000000020571 a group of boats parked next to a dock in the water
+000000265518 A plate with food on it, a fork and some kind of drink.
+000000091654 The asian dish is filled with several different ingredients.
+000000431693 A black man running towards the tennis ball with his racket.
+000000534601 A stuffed, white bear is on the bed with a person.
+000000149222 The computer monitor has a framed picture near it.
+000000073533 a couple of kids in yellow shirts are sitting together
+000000002923 An overcast day at dock with sea birds resting in the nearby marsh. 
+000000024021 A group of young men pose for a picture with some adults.
+000000552612 A bunch of sheep eating grass next to the forest. 
+000000141671 A group of people riding motorcycles and horses on a race track.
+000000155451 The city street is quiet during this time of night. 
+000000185250 a man watches as a dig looks at a toy in the air 
+000000114871 A banana is couple with sliced cheese on a plate.
+000000538236 There are pasteries with winnies inside of them on display for sale 
+000000184324 A fish eye lens shows the corner of a busy city street with bikes, people and buildings.
+000000060823 a plaza filled with a lot of birds and some cows
+000000029393 A large brown dog standing on the side of a white wood fence.
+000000475678 Several suit cases lined in rows with luggage tags on them.
+000000343496 The Pleasant Lane street corner and a woman carrying bags
+000000057672 A long canoe paddled along by a large group of people. 
+000000366225 A white computer monitor sitting above a keyboard.
+000000034257 A large number of fresh vegetables have been stacked together.
+000000106912 A boy skateboards on concrete behind a backyard.
+000000257370 Two people playing a game on a TV.
+000000200162 A train traveling under a signal lights on top of tracks.
+000000022705 A woman standing in front of a refrigerator freezer in a black dress.
+000000085195 A plate with an orange cake dessert and a fork.
+000000250619 a woman in glasses bursking under an umbrella
+000000257624 The young child is standing beside two soccer balls. 
+000000342128 The man is the sport jacket is holding a racket.
+000000210915 a man is riding a wave on a surfboard
+000000256195 A metro style train engine passing under a bridge.
+000000400367 Several cars parked on a street near trees.
+000000542423 A woman sits by herself on a park bench.
+000000569273 Fire hydrant sitting in the middle of a hiking area on the side of a gorge. 
+000000565776 white cabinets silver oven sink and  refrigerator 
+000000166521 A room that has two televisions in it.
+000000423123 A woman bending over and looking inside of a mini fridge.
+000000157098 Two giraffes make contact while a third eats from a tree.
+000000002261 The boy is looking back at a wave in the ocean. 
+000000199977 The airplane in the sky is doing tricks while spitting out smoke. 
+000000357567 A bathroom with matching shower curtain and rug
+000000006471 A hitter is waiting for the pitch to be thrown
+000000095862 A man pitching a baseball from a mound on a field.
+000000383337 A hand reaching into a box of donuts on a counter.
+000000297147 a black and silver motor cycle parked by a shed
+000000343803 A young man holding a Frisbee in a throwing position.
+000000091779 Three sausages covered in various condiments side by side
+000000112298 A young man standing below a shelf full of odds and ends.
+000000445722 a big boat with some luggage inside of it 
+000000380706 A person is riding a motorcycle around a corner.
+000000331352 A small closed toilet in a cramped space.
+000000042276 The bathroom is a mess with sports gear. 
+000000377946 A yellow food truck parked close to a car
+000000372349 A man and a woman in a river.
+000000255718 A rusty old fire hydrant on an inner city street
+000000439426 A person's hand holding a bitten into doughnut.
+000000258793 I love the way the sun is creeping behind those two buidings
+000000422886 a man in a hat is biting into a hotdog
+000000532901 A daybed in a sitting rooms with a fireplace.
+000000017436 A man is on a bench in front of a steple tower.
+000000294163 Flags wave beneath the illuminated clock and tower.
+000000041635 Three cows with hairy heads laying down in a field.
+000000080671 A person is trying to ski down a snowy slope.
+000000339442 A sandwich with meat sitting next to grapes and an apple.
+000000374727 A pole with multiple street signs in front of a tall building.
+000000104803 A restroom toilet and lid covered in silver glitter.
+000000281447 a blonde horse is standing in a field
+000000383384 A batter, catcher and umpire in a baseball game with man recording.
+000000106048 a black and white bus some bushes and building
+000000231747 Several museum displays of baseball uniforms and bats.
+000000098018 This is a picture of a kitchen in a poverty area of a town.
+000000496722 Some big red metal boxes near a road.
+000000021839 a person walking across the street with a traffic light above
+000000362682 A bus that is sitting in the street.
+000000341058 An umbrella is opened over an outdoor table.
+000000025139 Two zebras relaxing in the fenced area of a zoo
+000000460841 A hand holding a water bottle in front of a cat.
+000000272212 Cows grazing on the grass in front of a building.
+000000336356 a girl sitting at a table in front of two pizzas
+000000321333 A couple of children sitting in front of a blanket.
+000000041635 Three cows laying in a field behind a barb wired fence.
+000000036539 A man and a little girl wearing snowboards. 
+000000018491 A boy catches a ball as a player slides to the base.
+000000111609 A man asleep under a blanket on a bed 
+000000260925 A cat laying on a white car outside. 
+000000192871 A glass bowl full of oranges and apples.
+000000289229 A man wearing a white lab coat holding a bone saw.
+000000389316 Two young elephants stand next to a wooden fence.
+000000482436 A woman sits at a counter while a man looks at her through a window.
+000000082821 Paddle boats parked at the edge of a river or lake 
+000000113403 Three stuffed animals lay in a bed cuddled together.
+000000034873 A sink in a peninsula in a kitchen.
+000000476215 A man standing behind two brown horses on a dirt field.
+000000052017 A yellow and blue plant flying in a gray sky.
+000000088485 A man is holding a yellow frisbee outside
+000000012670 A group of men and women gathered together outside. 
+000000242678 A person jumping a horse over an obstacle on a course.
+000000185599 A handful of apples and oranges sit in a bowl.
+000000007511 Man flying a kite on the beach while others run along the sand.
+000000430056  A cut in half sandwich sitting inside of a microwave.
+000000250758 a couple of zebras stand on a grass land 
+000000489305 A person shows two bowls full of different fruits
+000000298904 a glass fish bowl with rocks and weeds in it.
+000000191845 A crowd of people with umbrellas on a street.
+000000123480 A man taking a selfie while brushing his teeth and looking in the mirror 
+000000248810 a large giraffe that is next to a fence
+000000446522 A dog relaxes on an armchair in a living room.
+000000550426 A tall vase with red and white tulips in water.
+000000561335 Two row boats are laying upside down in the woods.
+000000095155 Four picture collage of a snowboarder wearing a red jacket and brown pants going down a snowy mountain side.
+000000237984 A green chair is next to a long bench.
+000000235836 A toddler hitting the ball with a baseball bat in his backyard.
+000000035963 A picture of a little statue in some dirt.
+000000513567 The woman are eating their hot dogs while walking.
+000000144984 A young boy is playing Wii on the couch.
+000000560279 White toilet in a small tiled room rainbows on toilet
+000000060770 A couple of zebras are grazing next to some trees.
+000000041990 a couple of people that are skiing posing for pictures
+000000208901 A plane is flying in a clear sky
+000000546556 a few goats in a big grassy field out doors
+000000189436 A teddy bear that appears to be made out of wood.
+000000267670 a mirror a sink a shower and a rug
+000000542127 A couple of men flying through the air while riding skis.
+000000303893 A beautiful little girl sitting next to a little boy sharing a meal.
+000000515828 A woman holding up a tennis racket during a match.
+000000376284 A fire hydrant and some cars on a street.
+000000551815 Two cats rest on each other to take a nap.
+000000420472 Two giraffes are eating a leafy branch together.
+000000182021 A MAN IS HOLDING A BASE BALL BAT STANDING AT THE DIRT
+000000293474 A yellow fire hydrant on a sidewalk by a storefront.
+000000166277 A grey colored cat that is drinking from a glass of water.
+000000250127 The young woman is standing in the rain under her umbrella 
+000000262227 Part of a computer screen in front of a laptop.
+000000228771 Street signs on a pole painted in the colors of the Italian flag.
+000000406997 Families in a cage bedded with hay petting a sheep.
+000000242946 A group of women sitting around a table sharing a meal.
+000000178469 A laptop computer sits on a cluttered desk.
+000000470773 Pastry items and toppings on display on a table.
+000000071756 A pair of black bears are shown through nightvision.
+000000581482 A shot of a clock in the train station.
+000000529939 a man standing by a display of surfboards and wetsuits
+000000431693 there is a male tennis player trying to hit the ball
+000000424551 A skier is standing at the top of ski slope. 
+000000491366 Two men playing a video game in the house. 
+000000009400 A group of people gaze at laptop screens.
+000000540414 The people are sitting and standing at the event in the street.
+000000193884 Lone stop sign at small intersection in a cemetery of castes.
+000000436883 A green bus with signage on a city street.
+000000325483 Two young men are playing a WII video game.
+000000209747 The tiny kitten looks sad sitting in the bathroom sink with a scrub brush on the rim.
+000000277051 Two birds perched on a table near a plate of food.
+000000141597 A bus and a smart car parked outside of a building.
+000000005477 A brown jetliner sitting on top of an airport runway.
+000000468577 Child playing and relaxing within a crib of white walls
+000000409268 A teddy bear that has been buried in the sand.
+000000438774 A group of people standing in a kitchen having conversation.
+000000328430 a woman reaching up while jumping to hit a tennis ball 
+000000277005 A young man walking across a street at a crosswalk.
+000000426166 A bicycle is locked to a lamp post in front of an old building
+000000025181 A International train station with a passenger train passing by.
+000000473121 A person leaning hard into a turn on a downhill ski course.
+000000120420 Individual is doing something at the moment that is intriguing.
+000000001761 two jets are flying over a bridge and some water
+000000015335 A man, woman and child eating together at a restaurant.
+000000004495 a room with a tv and some different types of couches
+000000287714 a tub and a sink in a rest room
+000000224200 A dog anchored to a fire hydrant by his leash.
+000000456015 A group of men riding horses on a beach.
+000000268831 A bathroom with bidet, toilet, tub, and a checkerboard floor.
+000000308587 A few para-surfers on the water near a beach.
+000000014007 A brown colored cat is on top of a tall gray cabinet.
+000000280779 A young man riding skis across a snow covered slope.
+000000070774 A motorbike parked, with several bicycles stacked on on the back of it.
+000000233238 a long table with many containers of food on it and many people around it
+000000311394 A infant holding a baby toothbrush in his hand looking at it.
+000000155341 A red car on street with utility vehicles in background.
+000000482100 These toilets are being used outside as decor
+000000065074 an unmade bed next to a sunny window
+000000021465 a cabinet with some paint on top of it 
+000000242060 A plate of cupcakes on a napkin with spoons and drink glasses.
+000000245764 a black and gray cat is sitting on a toilet
+000000372349 A man brushing his teeth in a river while a woman looks on.
+000000269942 a couple of street lights that is next to the road.
+000000544811 A couple of birds flying over a body of water.
+000000031735 A living room painted red with black couches and coffee tables
+000000153343 two teddy bears that are large sitting in a small garden
+000000110999 a bunch of fruit sits in front of a portrait 
+000000119088 two men playing frisbee in the water. One man walking away
+000000235252 A small herd of giraffe standing in a plain.
+000000526392 there are many cars that are along this street
+000000537964 a black blue and red fire hydrant and some leaves and trees
+000000189213 Garbage and old chairs are thrown onto a curb.
+000000122672 A guy on a surf board in the water.
+000000134856 Three people in the distance riding horses along the beach.
+000000437351 A woman is sitting and posing next to luggage.
+000000447465 a person on skis with ski poles standing at the top of a hill
+000000063154 A surfer is riding a wave in light blue water.
+000000260106 A man on a horse during  a race jumps over a hurdle
+000000093965 Two zebras are in a field and one is eating the grass. 
+000000172935 A water skier performing a trick while flying through the air.
+000000459809 A kite is flying over the water of a bay.
+000000428562 The wholly sheep stands besides others in a cage. 
+000000122217 Two people on skateboards riding down a slanted road.
+000000581206 Two chili dogs in buns with peppers in a cardboard container.
+000000328030 a stuffed animal holding a hockey stick 
+000000046031 A laptop and computer monitor with the same screen.
+000000553990 A young person ridding a horse jumps a gate in a competition.
+000000226058 teddy bears dressed up in clothing sitting on a loveseat together
+000000287959 This is a meatball sandwich sitting on a plate. 
+000000572517 A polar bear and a bird near a small pond.
+000000478721 Two people that are carrying a clear box kite. 
+000000325114 a white urinal white tiles a red table with a dish
+000000579902 a man on a motor bike riding in the high way
+000000394328 There is an indoor toilet underneath a sign that says please flush. 
+000000483667 A man with a top hat and a red tie.
+000000565778 A train going back to its coarse filled with people.
+000000309484 A dog having his teeth brushed with a novelty electric toothbrush.
+000000475365 A blue and grey Amtrak train is displayed.
+000000367680 A horse and cart are stopped on a busy street.
+000000106563 A clock that is hanging underneath a glass arch.
+000000119641 A herd of elephants walking down a river with people riding on their backs.
+000000492878 A sink with toiletry items setting on the edge.
+000000058636 A wooden utility pole sports several street signs.
+000000577539 A bowl full of marshmallows, chocolate, and other delicious treats.
+000000560312 A window sitting above a seating area with a TV.
+000000107851 A group of people walking through a field under the sun.
+000000399296 The hot dogs are being roasted on the camp fire.
+000000340894 Two computers are sitting on top of the desk.
+000000162035 a sign with teddy bears and some people walking on a wet street
+000000088345 A woman smiling for the camera in front of a wall clock
+000000304984 A white plate with a sandwich and sides on it
+000000500270 A young woman on a skateboard in a skate park
+000000232563 A man on sidewalk holding a zebra strip umbrella.
+000000514540 Many signs are posted on the light post near a skyscraper.
+000000272136 An air hanger with an old fighter plane inside.
+000000294163 A tree branch in the foreground with a clock tower set against a blue sky in the background.
+000000155145 a boat with a rainbow umbrella sitting in water 
+000000023034 a man standing on rocks near a man riding horses on a trail in the woods
+000000291619 Two young men playing frisbee at night on a sports field
+000000114884 A crowd of people outside a terminal boarding buses.
+000000435205 A pregnant with writing on her stomach while holding a brown teddy bear.
+000000427338 A person riding a motorcycle in an abandoned stone building
+000000312278 A rack filled with lots of different bags of luggage.
+000000447187 A man is swinging a bat a ball being pitched at him.
+000000307145 A country-style kitchen with red tile and white cabinets.
+000000169356 A man flying through the air while riding a skateboard.
+000000514914 Modern looking bathroom featuring a bathtub, shower, sink and toilet
+000000070048 A glass of wine sits next to a bowl of soup, salad, and a plate with a sandwich.
+000000063602 A laptop computer is displaying a screen with words.
+000000329319 a cat sitting on a chair looking at the ground 
+000000351331 a small pepperoni pizza next to a fork
+000000534664 Vintage suitcases with price tags on their handles. 
+000000250282 A group of children pose for a class picture. 
+000000185890 two women playing the wii game together .
+000000065350 A skateboarder wearing a helmet doing a trick at a skate park.
+000000172935 The person is doing tricks on the water skiing. 
+000000116479 A bed in a bedroom with a white covering next to a doorway.
+000000017207 A bus stuck in traffic on a busy highway.
+000000249786 Person high up in the air doing ski tricks
+000000196185 A jetliner taking off from an airport runway.
+000000392481 A fighter jet gaining altitude in a cloudy sky.
+000000123633 Woman watching four children play in a house.
+000000115898 A man surfing rocking waves in the water.
+000000184978 a man that is snowboarding down a snowy hill
+000000489842 A black and white image of people sitting near a santa
+000000516601 A group of people are skiing and snowboarding down a mountain.
+000000541055 A group of people on skis at a peak of a mountain.
+000000147518 Compact bathroom area, tub, toilet, magazine area and sink.
+000000137950 A tri-motor, vintage airplane on final approach to land.
+000000254516 A baseball player practicing his swing at a game.
+000000441491 Two girls that appear to be sharing food while another looks on.
+000000162415 The girl with the baseball mitt is smiling. 
+000000311392 A black and white cow in a barn next to a man.
+000000245173 A tall red bricked clock tower with three windows.
+000000210502 The side of a passenger rail train with a stripped door.
+000000478474 A truck with two trailers moves down a road. 
+000000179174 There are lots and lots of suitcases stacked up together.
+000000379441 Somebody left a lap top plugged into the wall on a bench. 
+000000165351 odd meal on a plate with bananas and gravy
+000000105264 A bull is running freely on the beach with other boys.
+000000015751 A bird flies away from three others that are sitting.
+000000458755 A child is petting a sheep while another sheep watches.
+000000261982 a person on a skate board does a trick 
+000000556193 Two men are holding video game controllers preparing to play.
+000000425226 A man wearing an apron peering into the bottom of an open fridge
+000000571804 A collection of soap dishes, bowls, and small trinkets. 
+000000395801 There are people walking down the street that is crowd
+000000429281 Bananas and apples are being sold at the market.
+000000418961 A decorative clock has roman numerals on its face.
+000000207844 a large beach filled with many people and umbrellas
+000000231549 A bed flanked on both sides by two room lights and desks.
+000000376093 The teen group is serving pizza and drinks to the younger children.
+000000504711 A closeup on some chocolate cake has a fork by it.
+000000488673 A man and woman are in a kitchen.
+000000276804 A baseball player standing next to home base on a baseball field.
+000000519764 A cat sitting on top of a black leather chair.
+000000057725 A large brick building has a clock tower on it.
+000000015956 A woman training her beautiful brown horse. 
+000000015272 A traffic light in front of trees with white blossoms.
+000000213547 Woman holding a small birthday cake rides a coin-operated children's ride in a convenience store.
+000000209747 A kitten sitting in a skin with a green brush with green bristles.
+000000437205 A young boy who is eating a banana.
+000000201418 A young man standing in front of a luggage carousel at an airport.
+000000474452 A red sign notifying drivers that there is no parking allowed at any time.
+000000006040 A close-up of an electric train going by on the tracks.
+000000552842 A baseball pitcher throwing a baseball during a baseball game.
+000000316054 A train blowing smoke is coming down the tracks.
+000000180792 A small bathroom has an army green toilet.
+000000025181 A train pulling into a station beside a platform.
+000000274272 A bus driving in the middle of traffic.
+000000227491 A skier sitting on a picnic table and eating a doughnut. 
+000000434297 A white plate topped with food on a counter.
+000000512657 A couple of people playing a game with remote controllers.
+000000542073 A white building has a street lamp with a sign.
+000000237864 Elephant going after a small animal of prey.
+000000479732 A sandwich of meat, carrots, cilantro and cucumber. 
+000000564336 A man sitting in front of a laptop computer.
+000000302882 A man takes a black and white selfie with a phone. 
+000000257624 a little girl is standing outside with a soccer ball
+000000079565 A baby giraffe stands next to a adult giraffe on staring out beyond them.
+000000124798 a yellow and blue bus is on a busy street
+000000026465 The opened  laptop is sitting on the table along with other technological tools.
+000000122927 Two birds walking down the road in the daytime.
+000000500477 An adult sheep is getting shorn by a male farmer.
+000000154004 A group of people on a beach looking towards the water. 
+000000105912 A hydrant is shown on the side of the walkway.
+000000010977 A bathroom with blue walls and pink tile.
+000000475678 A pile of different colored bags of luggage sitting on a floor.
+000000144932 a couple of boats sitting on top of a body of water.
+000000203546 A brown bear shakes water off while swimming.
+000000186938 A grey and white bird on sandy beach next to rocks.
+000000263068 A young man riding a skateboard on top of a cement block.
+000000449579 A strong woman is playing tennis on a court.
+000000347930 A large brown dog sitting on top of a couch.
+000000249025 A yellow rose in a clear vase filled with water.
+000000310200 A piece of food sits on the plate on the table.
+000000496722 Two red mail boxes along a city street.
+000000152120 a person riding a motorcycle around a curve in the road.
+000000268378 A group of people eating a meal at a table.
+000000322895 A room with a couch, chairs, television and a table. 
+000000105249 Doorway view into a bathroom with a sink and toilet.
+000000214703 Two young people are body boarding in the open water.
+000000163562 a close up of a person standing holding a frisbee
+000000119038 Three cows stand in a pasture on a mountain.
+000000243075 A polar bear is sitting next to a sheet of ice.
+000000139883 some kind of cage that is next to a tree
+000000082821 Paddleboats are lined up and ready for tourists this morning
+000000466416  A downtown city view from a high vantage point at night.
+000000491008 A man and a woman holding up a pizza in a seating area.
+000000302760 A bathroom sink with a bottle of hand soap 
+000000351589 A small bench in front of a building
+000000520264 a man and woman standing side by side in a kitchen next to a window
+000000313454 A man is speaking at a panel in front of a crowd.
+000000160666 A close up of a toothbrush with some Crest paste on it. 
+000000246454 A black and white photo of a girl petting three dogs.
+000000117197 A woman sitting on the sidewalk talking on her cell phone. 
+000000570169 a stuffed bear is sitting with a toy
+000000220764 A person riding skis across snow covered ground.
+000000439290 A bunch of bananas hanging on the tree
+000000202445 A brown cat is curled up on some bed sheets.
+000000183716 two children wearing sweaters shirts and ties 
+000000212559 A herd of sheep grazing on a pile of hay.
+000000495054 A plane that is sitting on a tarmac.
+000000571804 A collection of sinks, and basins are on display.
+000000421757 Two boats are floating on the river near the shore.
+000000579091 a close up of a plate of food with broccoli
+000000147223 Advertisements covered bus stopped on a city street.
+000000236166 A man holding a dog wearing a costume in his arms.
+000000434230 Various pots and plates displayed next to a woman.
+000000223789 A bathroom sink decorated with candles and rose petals.
+000000460160 A white bird sitting on top of a lake.
+000000221872 A serving of rice, carrots, green pod beans and an egg each in their own small plastic containers.
+000000127270 A tennis player is swinging at a tennis ball.
+000000411953 Man wearing a white shirt and black tie playing a guitar. 
+000000391140 The boy is playing a video game and making a funny face.
+000000056344 Looking down at a desktop with two monitors, a key board, mouse and cell phones on it.
+000000441286 Surfer riding large breaking wave on open ocean.
+000000516318 A surfer is riding a wave in the ocean.
+000000107094 A man stands on his skis on a flat patch of snow near a fence.
+000000046031 A laptop screen displays an image of bright green leaves.
+000000546626 A smart phone sitting on top sitting on top of a wooden table.
+000000186624 a woman stands next to a parked train 
+000000213593 A red stop sign sitting on the side of a tall building.
+000000458790 A view of a wet day from indoors.
+000000566524 A trolley train is traveling on a track along the beach. 
+000000390555 A group of students are posing for a class picture. 
+000000288685 A group of sheep are being herded by a dog as people watch.
+000000171190 a party that is full of wine tasters seems pumping.
+000000229311 a kitchen with a sink near a window 
+000000473121 Man skiing down a turn on  an official ski slope.
+000000193162 A black and white cow walking across a lush green field.
+000000462614 A template example of a bathroom is shown in bright colors.
+000000357430 A large pizza sitting on a pizza pan with a slice missing.
+000000368294 A woman sitting with a washcloth on her head and scissors in her hands. 
+000000356248 A man with glasses holds a small child at a restaurant.
+000000032817 A man climbing on a toilet bowl in a closet
+000000436315 A homemade pizza with a funny shape on a pan.
+000000292997 A car that has slid into a light pole on a snow covered road.
+000000352584 A toilet covered in glitter sitting inside of a room.
+000000193926 The various pictures are featuring bananas and holders.
+000000570756 People on a beach prepare to use their glides
+000000031322 A flock of geese are sitting on the water with boats in the background.
+000000212453 A white toilet in a trendy wall papered bathroom.
+000000353180 There are many people trying to push the school bus. 
+000000296222 A man is sitting on a chair with some pizza.
+000000064495 a double parking meter in front of a building 
+000000270677 A young man holding his skateboard, is posing for the camera. 
+000000465180 People with elephants doing various activities in a river.
+000000059920 A tiny bathroom area features a patterned wall, bidet and toilet. 
+000000436617 A large group of people at a table.
+000000150224 People lay on blankets and sit in chairs on the beach under an umbrella. 
+000000159791 A white refrigerator freezer covered in magnets and pictures.
+000000166277 A cat drinking out of a glass on top of a table.
+000000240940 A cat sitting on the floor watching a television.
+000000400044 an image of a man on skis' doing tricks
+000000568439 Two double-decker buses going in opposite directions on a road.
+000000478474 a white and red truck is driving down the road
+000000430973 The deep dish pizza is topped with sausage and olives.
+000000134722 Fast commuter train moving past an outdoor platform.
+000000241668 A woman in a wedding dress with another woman in a suit behind
+000000333956 A parking meter at the edge of the ocean with a boat behind it.
+000000275791 A large trains travels past a river that contains several different boats
+000000507235 well cooked food in a big bowl on a table
+000000389316 Two elephants in an enclosure at a zoo.
+000000463849 a couple of buildings surrounding a pond with boats
+000000504635 Three zebras in the field with trees and grass
+000000350002 A model in a blue and white dress sits for a photo. 
+000000454661 Three similar cars are stopped at traffic lights.
+000000297353 A guy with a nice guitar standing on a bed.
+000000301867 three people standing near one another with an umbrella
+000000447522 Add cooked vegetables to plain white rice for a healthy dinner.
+000000163118 Two men playing frisbe on an indoor court.
+000000413395 a man sits on a couch with a couple of cats 
+000000004134 Two men are shaking hands at a social gathering.
+000000376442 This is a stuffed animal on a toilet seat.
+000000223188 An Elvis look-alike is seated on the  back of a statue of what appears to be a bull, or water buffalo, against a backdrop of green foliage. 
+000000438017 a street light on a corner with a red light on it 
+000000020571 Two tug boats tied up to the dock.
+000000003255 a group of people riding skis on a snowy surface
+000000565989 a very tall building with sculpture and a clock
+000000415748 A man riding on the back of an elephant through a city street.
+000000190923 A One Way sign is located on the traffic light.
+000000470173 a statue of a lady carrying a water pitcher
+000000039477 Someones living room with a sewing machine and plants. 
+000000490936 A wow of motorcycles pared along a street some with covers.
+000000021604 A man with glasses and sideburns straightening his necktie that has lights on it.
+000000343076 A cat sitting on top of a D.V.D player. 
+000000442306 A little girl standing in a forest holding a black umbrella.
+000000491090 A middle-aged man leans on a sports bike, smiling
+000000032038 A close-up of a cheese pizza on a wooden table.
+000000386134 A to go box filled with rice, onions, and spinach. 
+000000141821 An oddly shaped pizza resting on a picnic table near several beverages.
+000000569030 A city crosswalk has a leaky fire hydrant on the street.
+000000501243 Three zebras standing in the sand in a zoo.
+000000334521 Giraffes and a bird behind a chain link fence at a zoo
+000000093717 Two boys skateboarding on a ramp at a skate park.
+000000291551 a boy that is on a skateboard doing a trick
+000000404923 A baseball player holding a baseball bat is getting filmed by a camera man.
+000000407646 A man on a tennis court who has just hit the tennis ball
+000000570756 Several people are flying kites on the beach.
+000000116206 Vegetable pieces and knife are kept on cutting board.Cabbage is on the table.
+000000078823 a dog with a bandana on the trunk of a car
+000000128598 A group of red, yellow and orange fruit  mixed together.
+000000213224 A vase filled with flowers next to a smaller vase filled with even more flowers.
+000000481404 A golden obelisk stands in the middle of an ornate building. 
+000000512985 A man walking across a beach holding a white surfboard.
+000000408696 A man and woman in attic with vacuums and a toilet seat.
+000000330818 A man prepare food in the small kitchen.
+000000515025 A woman in black jacket watching a cat eating from pizza box.
+000000142971 A man helping a boy on a paddle board in the water.
+000000400044 Someone who is on a snowboard midair doing a trick.
+000000232489 Two slices of pizza are sitting on a paper plate.
+000000155179 a close up of a cake on a plate on a table
+000000012280 A woman in a coat and cat is towing luggage through an airport.
+000000013348 The humongous twin engine airliner is being loaded. 
+000000099182 a man is flipping through a book on a bed
+000000172595 A computer desk with a large purple office chair
+000000226130 Two hot dogs on a white plate with piece of parsley.
+000000180487 A person they sitting down in a chair.
+000000485895 A giraffe stands underneath a tree on grass.
+000000308328 people at the airport offloading the baggage with their names on
+000000475484 Something that looks like a parking meter painted yellow on a sidewalk.
+000000227765 The meal is being prepared in the big pot. 
+000000210394 A city passenger bus stopping to take on passengers.
+000000471869 A Teddy bear sits on a floral patterned chair.
+000000359135 Two elephants with their heads close together. 
+000000066886 A girl grabs her hair and holds onto her hairbrush. 
+000000291551 Kid on a skate board in a skate park preforming a stunt.
+000000283268 Parent serving a plastic hot dog in a bun to their child during play
+000000261888 a couple of people that are biking down a road
+000000361238 a woman in a black jacket and a pizza 
+000000376478 A vase containing water as well as a flower in it.
+000000551660 A small bowl of broccoli with some slices of apple and a lemon wedge.
+000000371699 A room full of shelving and many supplies
+000000546219 The diners are enjoying their various beverages with their meals..
+000000121497 a person operating a blender hooked up to a small motor
+000000494869 a yellow lap laying in a kitchen and looking at a little kid 
+000000546976 A man riding a scooter in the kitchen of a home. 
+000000433192 A laptop computer with power cube sitting on a twin sized bed.
+000000474881 The mountain goats are eating the grass on the slope.
+000000033368 A man that is on a tennis court with a racquet.
+000000361586 People are standing near an airport ticket kiosk.
+000000084241 two women are talking while they cook in a kitchen
+000000304365 A train at a rest in a small train station
+000000128148 A living room filled with two leather couches on a hard wood floor.
+000000308531 a tall clock tower with bushes and trees in the foreground
+000000164637 I am unable to see an image above.
+000000533855 a frosted donut sitting on top of a jump rope 
+000000507975 Some very pretty horses with riders in the grass.
+000000140076 The table has been decorated with a very modern look.
+000000313562 A woman has an apron and head scarf while touching carrots at a produce market.
+000000035770 There is a sink and a black and white toilet
+000000046048 A little girl sitting on top of a bed next to a lamp.
+000000453302 A break room kitchen is shown with two microwaves.
+000000282296 A fridge stands in the corner of a room. 
+000000398438 A man and woman making a cut unto a cake
+000000468501 Two players compete in a video game as another person watches. 
+000000222735 A remote control on a table for the television. 
+000000444879 A train on tracks while its snowing outside.
+000000266400 a number of motorcycles parked near each other
+000000272364 These giraffes are feeding on hanging plants in a tree
+000000546823 The city is very busy today with foot traffic. 
+000000528977 Street signs, graffiti, old buildings and a bridge.
+000000030504 A person riding skis down a snow covered slope.
+000000568213 A couple of men standing on top of a field together.
+000000459396 A brown cow grazing on a patch of lush green grass.
+000000488710 A girl in a blue bowtie standing next to a pink Princess Parking Only sign.
+000000153529 a woman with her luggage at a train station
+000000393282 Two giraffes standing around in the middle of a field.
+000000147740 A lady walks down the street with bananas on her head
+000000233370 A person laying crushed underneath a black piece of luggage.
+000000498032 A woman holding a Wii controller in the air. 
+000000573626 Two horses stand on rocks near a river. 
+000000023899 Three young men playing Wii on a projection television.
+000000049091 Signs displayed in a crowded urban alleyway 
+000000260925 A cat sitting on the hood of a car.
+000000200839 A library book mobile parked near a sidewalk.
+000000044279 Two men cooking in a stainless steel restaurant kitchen
+000000424721 A frying pan filled with vegetables and apples.
+000000390826 A person holding an object in front of a panda bear.
+000000063552 a beige cat looking into a big mirror 
+000000270705 A large bird perched on top of a stick near a window.
+000000066706 A cappuccino with a plate of banana slices on bread. 
+000000451693 A pizza that is on a stack of plates.
+000000559348 A baseball player at bat during a baseball game.
+000000286553 Two people sitting at a table with two pizzas.
+000000376206 A man crouched on a surfboard as he rides under a wave.
+000000124975 Two zebras standing side by side in a field. 
+000000142790 two people on skis falling in the snow
+000000193162 A man is right behind a cow on green grass.
+000000310862 A man wearing a gray vest and a red stripe tie.
+000000166642 an elephant in captivity with trees in the background
+000000134096 A calico cat drinking from a sink faucet.
+000000244496 A man in a suit with a red tie that has a robot on it 
+000000177213 A close up of a cheese pizza on a plate.
+000000523229 a handicap toilet with a foreign language written above it
+000000498463 A stainless steel refrigerator is next to the kitchen cabinets.
+000000355325 A table with food and a glass of red wine.
+000000276720 A stone building with a blue door on it
+000000001490 A man in a wet suit stands on a surfboard and rows with a paddle.
+000000031817 A man leans forward on his motor bike as he sits in traffic.
+000000045472 A table with some oranges and some cups.
+000000292155 A large open multi- colored umbrella and tree branches.
+000000190756 a person jumping in the air on  a motorcycle
+000000509014 a bunch of people on skis are in the snow 
+000000411817 A group of people playing games while holding remotes
+000000223130 A tall giraffe stretches to reach high leaves on a tree.
+000000534394 The sheep follow each other over the short block wall.
+000000467315 The table has a flower arrangement in an ornate vase. 
+000000161032 Woman in high heels in a crumbling room.
+000000237071 The young man is playing a game of tennis.
+000000476258 A skateboarder demonstrates on his skateboard that has a large wheel in the back.
+000000157098 Three giraffe's in a green grassy tree filled area.
+000000276284 A lone motorcycle parked in parking space at night.
+000000274460 a bunch of people sitting on a beach 
+000000374545 A grassy field with pink flowers and a flag in the middle
+000000438017 The road is wet because it has rained. 
+000000184324 A group of people walking across a busy city street.
+000000442009 A bathroom with a bathtub and a sink overlooking a blue ocean.
+000000496409 A skier slaloms down a snowy hill in low visibility
+000000350019 A young boy pulls a large brown suitcase.
+000000290081 A pile of orange sitting in a blue bowl.
+000000153527 A row of wooden raid with various tools on them.
+000000405279 People flying kites in a park next to a lake.
+000000297022 a man and his friends are standing outside next to a truck
+000000212226 The reflection of a women in a mirror.
+000000503855 Black and yellow umbrellas open on the grass of a park.
+000000383386 Stuffed toy bear with rubber green turtle displayed next to chair.
+000000101068 a baseball player is swinging a bat and some grass
+000000197870 a close up of a chicken under a bench near a field
+000000410878 A parked motorcycle sitting on a dirty road.
+000000128654 A man in mid air attempting to catch a frisbee.
+000000088265 A horse pulling a carriage on a narrow path.
+000000481573 There is a young man standing on a skate board.
+000000458255 A little girl laying in bed holding a book next to a black cat.
+000000084664 Plate of food with meats, potatoes, eggs, and fruit.
+000000173044 A building with a clock that is in front of a building.
+000000274272 a public transit bus with traffic lights 
+000000578967 A three car passenger train on the train tracks.
+000000062554 A picture of a bunch food sitting on a table.
+000000104119 A bird in a forest looking down towards the ground.
+000000393115 A picture of a person on a skateboard.
+000000572388 A black cat licks a doughnut in front of a yellow toy.
+000000565045 Two towels are hanging on the wall beside a shower door.
+000000364557 The sun setting on the beach, and surfers walking the wet beach.
+000000180487 A guy sits at a table with an umbrella hat. 
+000000009483 Man in suit smiling next to electronics and monitors
+000000440336 A group of people in a field playing frisbee.
+000000309655 A woman stands on skis on a snowy hill.
+000000436551 A black swan has some wet grasses in its mouth. 
+000000236412 A pizza pie sits in a dish on a table. 
+000000004134 Two men shake hands at a formal dinner gathering.
+000000292488 A person I. A hood jacket capturing a photo using a flip phone 
+000000312278 A luggage rack that has various pieces of luggage and containers on it.
+000000048504 Elephants are in a ring with a person near bleachers.
+000000182162 A living room with hard wood flooring and black furniture.
+000000001000 A group of people that are standing near a tennis net.
+000000456559 A smiling man in a hat is wearing sunglasses.
+000000272364 Four tall giraffe are looking up to eat from a feeder.
+000000024027 A colorful object is flying through the blue sky
+000000316015 A cat sits on a desk in front of a computer. 
+000000443303 A cat laying on clothes that are in a suitcase. 
+000000428280 A laptop on a table with some chairs.
+000000217400 A train is pulling into a subway stop. 
+000000561335 Two small green boats on a platform in the woods.
+000000384350 A red and white plane sitting on a runway.
+000000497867 A white Ecohopper bus driving down a street.
+000000205105 a tennis player is serving the ball to his opponent
+000000549136 A chain link fence is overgrown with vines.
+000000451150 A variety of different donuts in a box.
+000000042628 A group of people walking across very dirty snow.
+000000193884 stop sign on a street of a cemetary
+000000493334 A sheet cake with a lake scene displays the words, "What is the Pickup Line Today?"
+000000066635 A computer monitor with a keyboard and a mouse.
+000000209747 A gray and white kitten sitting in a bathroom sink.
+000000369323 A woman holding a tennis racket on a court.
+000000131386 The planes are all flying in the same direction.
+000000200961 A man is charging through a field of grass.
+000000035062 A little boy asleep under a sunflower print blanket
+000000442161 A group of people tasting wine at a bar. 
+000000323751 A man waiting for the next train to come.
+000000241668 A bride is with a long red haired person with cake.
+000000372307 Horse in a large corral eating grass and trees in the back. 
+000000209222 an image of a man sitting on a park bench
+000000499266 A smart phone displays some text and a picture.
+000000058636 street signs on the corner of Gladys and Detroit
+000000381587 Several pictures of food on a table at a restaurant. 
+000000224093 A here of cattle grazing a field near trees.
+000000255824 A half eaten eclaire on a white plate
+000000056545 A bird is perched on a twig in the trees.
+000000058029 An undecorated bath room with shower, toilet and sink
+000000544306 The Big Ben clock tower towering over the city of London.
+000000023937 a small sheep is laying in a field
+000000107339 A young girl is using a video game remote control.
+000000135561 A small, white formica kitchen with a refrigerator, sink and small electrical appliances
+000000217425 A tall clock tower with a clock sitting on it's face.
+000000180188 The freightliner sits on the tracks with graffiti spray-painted on the side.
+000000245102 The entrance and exit to a snowy parking garage.
+000000015751 three birds perched and one flying towards the viewer
+000000065455 A large group of people holding their arms out to feed giraffes.
+000000549136 A barbed wire fence has a teddy bear in it.
+000000498463 A metallic refrigerator freezer sitting in a kitchen.
+000000080671 A woman touching her skis going down a ski hill.
+000000324818 There are two parrots perched on the branch. 
+000000113589 A container holds apples, a banana, snap peas and triangle finger foods.
+000000184321 A train coming down the tracks arriving at a station.
+000000186980 A table and chairs in a small kitchen.
+000000341469 A man standing next to suitcases and bags.
+000000048924 a green tent and a parked motorcycle and some trees
+000000297698 a man on skis fly through the sky 
+000000414340 Tilted shot of a clock tower attached to a building.
+000000405972 An adult giraffe standing next to a small elephant.
+000000467176 A group of friends play a game on the Nintendo Wii.
+000000422886 A man eagerly stuffing a cheesy hot dog into his mouth
+000000264968 A man with blue jersey holding a baseball bat.
+000000104669 The meal features steak and potatoes on the plate..
+000000350023 A hill of a street at a stoplight with traffic 
+000000003255 A group of people stand at the bottom of a snowy hill.
+000000000872 Two men playing baseball in a field on a sunny day.
+000000398377 Three women sitting on a train seat on cellphones.
+000000081594 a woman is holding a black umbrella and smiling
+000000486573 White flowers sit in a vase by the window.
+000000233825 A glass table that is next to a window.
+000000116439 A person sits in a small boat on the water.
+000000077460 Many people flying kits under a party cloudy sky.
+000000073946 Two men biting a surfboard with a shark on it. 
+000000169169 A tall building with glass and lots of people walking in front of it and there is a green banner that says Spring in the city.
+000000571008 A red stop sign with the word " Hammer Time ". written on it.
+000000437898 A picture of a small-sized kitchen with wood cabinets.
+000000099810 A little boy smiling for a picture as he eats.
+000000287874 A train traveling down train tracks under parking lights.
+000000328238 A man wearing a blue shirt catching a white frisbee.
+000000248752 a baseball player on a base ball field holding a bat
+000000130613 A tray of fish, with a side of vegetables.
+000000006614 A black and white image of a lot of round objects.
+000000296317 Dirt roads on a hilly landscape with giraffes and antelope in the foreground.
+000000033368 a person standing on a  tennis court holding a racquet.
+000000485424 A room in a dusty building with cleaning supplies in it.
+000000176847 A black bird on top of berry branches
+000000455716 a motorcycle is parked in a grassy field
+000000357430 A pizza cut in half on a plate. 
+000000578792 A person is leaning low on their motorcycle on the tracks.
+000000187249 Television showing a movie with one man threatening another with a razor blade.
+000000226883 A horse is eating some grass next to a building and is surrounded by a fence.
+000000425906 a person jumping a snow board in the air 
+000000061960 A man lays on a bed wearing women's clothing complete with stockings and high heels.
+000000540466 A woman walking by a fence holding an orange umbrella over her
+000000444275 Microwave and other appliance sitting outdoors on pavement against a fence.
+000000399560 There is a cat lying down on something. 
+000000183127 A man on a surfboard surfing a large wave.
+000000239318 A laptop computer sitting on top of a wooden desk.
+000000068387 A batter up to bat at a baseball game.
+000000224119 A view from the seat of an airplane looking down rows of other seats. 
+000000500477 A sheep's wool is being shaved off with shears. 
+000000094944 The snowboarder is posing for a picture on the mountain. 
+000000190756 A person who is on their motorcycle in the air.
+000000006460 a person riding a surf board on a wave
+000000190140 A man and a dog on a small boat.
+000000449406 A zebra is standing on some rocky brown land.
+000000016598 A woman who dyed her hair blue wearing all blue. 
+000000182155 Two teenage girls paying a video game together.
+000000129492 Two small children holding remote controls while sitting on a floor.
+000000332318 Several herd animals are on the grass by a mountain.
+000000090062 A black and white cow stands in a paddock with firewood in the background.
+000000069106 Four zebras stand together in the mud against a wall.
+000000293300 Looking up at elephants in an enclosure at the zoo
+000000025139 Two zebras are stranding next to each other by a fence.
+000000066926 Several rows of donuts sitting on a table.
+000000289229 Is he going to saw somebody's arm off with that thing?
+000000264968 a baseball player standing in the grass with a bat in his hand 
+000000202001 A WOMAN IS LOOKING AT A CELL PHONE 
+000000113589 Two containers containing quiche, a salad, apples and a banana on the side.
+000000438304 An ad for Adidas featuring a man and woman playing tennis together.
+000000370677 Three workers stand next to each other with their baked goods behind them.
+000000086220 A car and a public transit vehicle on a road.
+000000267537 Black and white photograph of group sitting under umbrellas.
+000000008899 The building in the city has been vandalized.
+000000253386 A dog that is wearing a dog collar smiling
+000000278463 A wooden desk that is very cluttered with items.
+000000354307 A man badly beating laying unconscious near a nurse.
+000000189698 A green street sign that says Bodacious Drive.
+000000286458 an open luggage bag on the ground next to a back pack
+000000555972 Two large, purple flowers are in a large vase.
+000000297353 A man standing on a purple bed playing an electric guitar
+000000005060 A man sitting cross legged in front of a mirror on the floor taking a self portrait 
+000000237517 A dinner table with two plates and two bowls of food.
+000000194716 A woman holding a baby while she has something in her mouth.
+000000122606 A bus that is on the side of the road.
+000000065455 Zoo scene of children at zoo near giraffes, attempting to pet or feed them.  
+000000358923 A woman with an umbrella hat holds a cellphone up to her ear.
+000000164885 The skiier easily and skillfully jumps the small mogul.
+000000361571 A white colored dog with it's mouth open and wearing a person's baseball style cap on it's head.
+000000244592 Three zebras standing close together behind a fence.
+000000445602 A cross country skier move across a snow covered ground.
+000000576052 A female horse rider mounted near a body of water and mountains.
+000000042102 The man in the picture is wearing a stylish kilt.
+000000033221 A woman holding a tennis racquet in the air
+000000307598 A green and yellow train is driving on the track.
+000000385190 Street art painted on the wall in an asian country. 
+000000311518 A small green boat in a pond outside an office building.
+000000552883 A counter top in a laundry room with a tile back splash
+000000143556 A group of bikers riding motorcycles across a bridge.
+000000356261 A small buckskin horse walking slowly down a hill.
+000000182021 Blurry shot of batter and catcher in game of baseball.
+000000322844 A teddy bear and a stuffed fish set next to each other.
+000000202001 A woman with a nose ring using a phone
+000000370270 A women is playing stick ball between counters.
+000000195165 A bathroom with toilet and sink with lots of items on the sink counter.
+000000398905 A man with a beard playing a keyboard on stage.
+000000469067 A women in a blue shirt cuddles up with her cat
+000000270908 This is an image of a man laying down with a tennis racket.
+000000263969 A couple cut their wedding cake on their wedding day.
+000000067406 A pair of men fly a kite in a grassy field.
+000000361586 several people are purchasing tickets at a bus station
+000000149222 A view of a desktop computer that is online.
+000000249180 Two men passing each other on the stair.s
+000000520871 A pizza with burned edges is sitting on the table.
+000000302030 In home desk, covered with papers and two computers.
+000000291490 A kitten on a bad with its arm stretched out toward the camera
+000000315257 A blue bird sitting on top of a bowl of seed.
+000000404839 A couple standing together holding Wii controllers next to a building.
+000000389684 A transit bus passing a man on a sidewalk.
+000000459757 A giraffe standing by a pair of skinny trees.
+000000122927 Two small birds walking on a sidewalk next to patches of grass.
+000000326970 A man and two women sitting on a tan ledge.
+000000482719 The fruit is placed in a bowl out in the open to be eaten
+000000194746 A pizza in an oven that had a lot of cheese melted off of it.
+000000568814 A man with a beard in a tie sits with his arms crossed.
+000000412531 A parking meter in front of a fence and parking lot.
+000000492878 A bathroom sink and its reflection in the mirror.
+000000427034 A brown and black dog standing next to a man sitting on a laptop computer.
+000000512985 The man stands holding a surfboard as he looks out at the sea.
+000000241677 The pack mules carrying loads on their backs are ascending the stairway. 
+000000516316 Three Zebra's eating grass as they walk. 
+000000482477 there is a bird that is sitting at the top of a branch
+000000393014 A display case holding a sandwich with meat and cheese.
+000000476119 a man skating on the road very fast
+000000527784 Two plates that have food on a counter.
+000000115245 Several suitcases on wheels leaning up against one long table with chairs.
+000000094185 An enormous black teddy bear in front of a teddy bear shop.
+000000442746 a woman on some grass is throwing a green frisbee
+000000014888 Small cow sitting on a milking machine in a dirty area.
+000000360960 a person walking down the street carrying a rainbow colored umbrella
+000000430961 A baseball player pitching a baseball on top of a field.
+000000300341 The two men are playing video games while others watch
+000000359219 Two pizzas on tin plates on an oven with a fork and knife.
+000000298697 Horses are standing in an open field in the grass.
+000000023034 A man smiles watching a rider approach with two horses.
+000000480944 School buses, cars and trucks sit at a red light at a busy corner
+000000181421 Three people sitting side by side in a canoe with tow paddling and the middle person holding up an umbrella.
+000000539143 A train riding on tracks underneath a bridge.
+000000186296 A young cat on a mat with a flip flop shoe.
+000000213422 A bed has white and brown pillows and sheets on it.
+000000060770 A couple of zebra standing next to a tree.
+000000059920 The toilet is across from a bidet in the small bathroom. 
+000000439522 She is cross-country skiing for the exercise it provides for her.
+000000069224 A man and woman eating doughnuts out of a box at a coffee shop.
+000000350002 a woman in a dress and pale makeup sitting by a table
+000000449312 A woman sitting in front of a case filled with sweets.
+000000521540 The plate has a banana being sliced near cookies.
+000000319100 A couple sitting on a couch holding Wii remotes.
+000000233033 An outdoor with a patio with chairs and a wooden deck. 
+000000193162 A man on a vehicle in a field and a few animals.
+000000265816 A man in a horse drawn carriage parked in front of a stone building.
+000000100274 a train on a train track next to a parking lot
+000000050679 An orange sitting in the middle of a road
+000000099114 People walking across a pedestrian cross walk 
+000000416343 Their underbelly of the surfboard shows that it has been used frequently.
+000000243199 A laptop computer sitting on a quilt comforter on a bed.
+000000047769 A small living room with seating and a television.
+000000455352 a white clock is on top of a tower
+000000193181 there is a baseball player that is at base
+000000508602 a close up of a bird on top of a parked car 
+000000157807 Cat standing on a toilet seat looking at the person taking the photo.
+000000331799 A blond lady with a flowered shirt passes out watermelon.
+000000288042 A man walking with an umbrella near a railing above a river.
+000000376278 A zebra lays down in a leafy field.
+000000506933 A man with a bald head wearing a pair of glases.
+000000561256 A woman taking a picture of herself in a bathroom.
+000000274460 A group of young people gather with surfboards on a tropical beach.
+000000458992 A woman poses on front of a large pizza
+000000420840 A female toddler who just finished eating with a bow in her hair
+000000491613 The zebra is standing by the local watering hole.
+000000515266 A white truck parked next to a white trailer.
+000000409358 A modern bathroom with square sink and backlit mirror.
+000000074733 Two small flat breads with beer samplers on each side.
+000000288762 A plate of freshly made vegetables served on a plate.
+000000172330 A cat at attention between two parked cars.
+000000463842 a bunch of bottle sitting under a little bed 
+000000031093 A man on a skateboard wearing a helmet and safety pads while riding in an empty swimming pool.
+000000490125 A flock of white birds stands in a parking lot puddle.
+000000027620 The laptop was left open on the desk. 
+000000484978 A woman sitting at a table while eating food.
+000000217400 Subway car leaving the subway station at night.
+000000107339 Two girls in a room are looking at something.
+000000293804 THERE IS A FIRE BURNING IN THE FIREPLACE IN A SITTING ROOM
+000000096960 A polar bear swimming underwater, approaching some rocks.
+000000095899 Two cranes behind a stop sign and a street sign indicating Fort York Blvd.
+000000050165 A short train traveling through a rustic contryside.
+000000320554 A cat sits under a park bench outside.
+000000269932 A plastic horse standing on top of a chair.
+000000392818 A dog is wearing a harness that reads "Outward Hound".
+000000401862 a line of food trucks and a group of people stand by
+000000239857 There is a green Frisbee in the air with red chairs in the background.
+000000521259 A couple of people that are playing with a frisbee.
+000000006040 A large white and blue train on a train track.
+000000437514 The empty bench is sitting next to the potted tree.
+000000066135 A train comes to a stop on the tracks next to the sidewalk.
+000000369037 A person pets an elephant with markings on it. 
+000000460494 A red plate holds on it broccoli, carrots, muffins and meat.
+000000201934 A white bus stopped alongside a sidewalk in the street.
+000000384468 A group of people riding a large gray elephant.
+000000351810 many backpacks and luggage bags piled up on top of each other
+000000343561 there are many bike riders racing in a street race
+000000138639 A group of men standing on the side of a street.
+000000286523 The man is listening on his cell phone
+000000408774 A player running the bases of a base ball game while an opposing player goes for the ball.
+000000483050 A neat and tidy bed in a hotel room
+000000109798 Frosted and sprinkled donuts stacked on a paper plate.
+000000475904 A white swan is swimming near a docked boat.
+000000426297 The zebra is wondering about the shadow that he sees.
+000000331075 A wet dog standing on a beach next to the ocean.
+000000200421 A child holds a racket with a hat on his head.
+000000256916 Four metal plates each holding a pizza on a table.
+000000427034 A dog is resting it's head on a laptop while watching the screen.
+000000026204 Cars and a bus are stopped at an intersection.
+000000267191 A man that is on the back of a elephant.
+000000036861 A parking meter on the side of the street with a ring for attaching bicycles.
+000000144798 Bathroom tub with children's toys and books on floor and tub.
+000000143998 A pie of carrots and a knife on a cutting board. 
+000000379453 The bottom of a jumbo jet is flying through the air.  
+000000528862 Several giraffes spending time in an enclosed, grassy area
+000000537506 A group of people standing on the side of a road.
+000000104669 A white plate served on a multi colored table cloth
+000000255912 A white plate topped with a cut in half sandwich and a salad.
+000000087244 A fire hydrant sits by itself in front of a business
+000000266892 Woman waiting for a serve in a tennis match.
+000000335800 A green pole with street signs attached to it.
+000000022589 Two sheep in a pasture behind a wire fence
+000000534673 A white and blue buss driving on a cobblestone road.
+000000029675 an image of three hot dogs with mustard
+000000509403 A man, two kids and a dog are playing Frisbee.
+000000581100 two giraffe are standing  and the other animals are walking
+000000070739 a couple of women smile as they hold wine glasses
+000000160012 A close up of a pizza pie sitting on a table.
+000000434230 A family looking at vases and plates in a store.
+000000410221 A plane that is flying in the sky above wires.
+000000314034 many cows next to a body of water in a field
+000000256775 two children playing near a kite flying in the sky
+000000226111 A street sign mounted to the side of a pole.
+000000189310 A living area with a red couch, mirror, window and painting.
+000000569030 A woman is standing at a crosswalk at a traffic intersection.
+000000341058 This is an empty table at a restaurant with ships in the background.
+000000358923 A woman with an umbrella hat talks on her cell phone. 
+000000301061 A person is moving green hay towards an elephant that is inside the back of a white truck.
+000000293245 A motorcycle parked outside the doors of a building 
+000000138241 Three plates of food showing a meal in the shape of log covered with vegetables. 
+000000541952 Far shot of the clock and flag pole of a building.
+000000253835 A woman and a small child watch a train as it passes.
+000000523100 Oranges that have been peeled are placed into a larger bowl.
+000000449312 A very nice cake dish with some pastries in it.
+000000132408 A farmer with draught horses uses a plow to furrow
+000000355325 A cutting board with pizza and a glass of wine
+000000365208 Black and white photograph of a person under a tarp.
+000000226592 A bedroom done in animal prints with red and black pillows.
+000000286994 One elephant standing away from the rest of the herd, drinking water.
+000000126107 a white bird is flying over a beach
+000000169356 a person jumping a skate board in the air
+000000095862 Male professional baseball players playing a baseball game. 
+000000558114 A group of basketball players walking on the court
+000000520324 a small air plane on a small run way
+000000166509 A road in the middle of a beautiful green forest
+000000347163 A stop sign with graffiti that reads "gentrifying georgetown."
+000000051326 Three decorated, carved jack-o-lanterns, one with pink flowers and a vase inside it
+000000506004 The ferris wheel is located in the middle of the water.
+000000322724 A man in a neon snowsuit is standing on skis. 
+000000415741 A room in a home with a chair set up against the wall. 
+000000089296 A man in red jacket and shorts in a batting cage.
+000000109827 A teddy bear is leaning on a tree 
+000000203294 A large bus parked in a parking lot next to a group of women.
+000000042178 Smoggy picture of many trains in a busy train station.
+000000236166 Man cradling dog dressed in uniform with it's head resting on his shoulder.
+000000446574 A bathroom with curtains that have floral decor on them.
+000000005060 A man in green shirt with cellphone reflecting in a mirror.
+000000187745 A yellow airplane flying in the clear sky.
+000000493019 A group of zebras are bending down and eating a box full of grass. 
+000000513283 this is an image of a plate of pizza with a toy bunny.
+000000113051 A motorboat docked on a pier under a tunnel.
+000000288762 A fresh-looking salad on a square plate. 
+000000314177 A picture of the photographer sitting on the toilet 
+000000398652 A lot of potted plants sitting on a glass table.
+000000050006 A row of boats parked next to a shore line.
+000000052412 a big parking with a bunch of cars parked in it
+000000150930 A clock protruding from the side of a building.  
+000000152120 Motorcyclist on chromed motorcycle rounding a curve roadway.
+000000233139 Two clocks with different times and a sign in an foreign language. 
+000000414133 A stop sign saying whoa instead of stop.
+000000292060 A kitchen with ceiling fan, stove, and sink.
+000000217753 A large display of fruit: applies, grapes, oranges, lemons, limes and grapefruit
+000000017899 A person at a table in a room with some food.
+000000296969 Family of giraffes standing out in the open.
+000000331317 A wall mounted golden clock on the side of a building.
+000000129812 A man riding skis while flying through the air.
+000000134689 there is a very tall giraffe standing in the wild
+000000505942 Three people stand near a street sign and a traffic light.
+000000562561 Two dogs are in a backyard playground area equipped with a kiddie pool.
+000000491867 a man that has a green tie and white shirt
+000000500826 A very tall brown brick building next to a street sign.
+000000127394 A group of people sitting around a table filled with food.
+000000356432 a couple of couches in a living room
+000000340930 Closeup of a cup holding scissors and other pens and pencils.
+000000459809 A shot of the blue water with people flying a kite. 
+000000133343 A man with thick sunglasses and multiple piercings sits with his mouth open.
+000000230166 A herd of sheep grazing on a green grass covered piece of land next to a forest.
+000000562229 A black and white photo of a child on a skateboard
+000000209757 A man on a skateboard on a ledge.
+000000556498 A street light with street names attached in front of a brick building.
+000000006614 A pile of nuts in front of some assorted fruit.
+000000446206 A passenger bus on the road with a train on the overpass
+000000350054 Cat sitting on cabinet in front of large screen television.
+000000379453 A large airplane flying in the blue sky.
+000000135673 Two airplanes having a close call in the sky.
+000000183500 A person is flying a biplane in the sky.
+000000562197 three different pictures of bean salad on a plate
+000000055299 The animal is sitting on the rocks looking at the water
+000000421757 two boats are idly floating on a lake.
+000000307658 A brown fuzzy bear standing in front of a leaf filled tree.
+000000038210 A man on skis riding through a snowy trail.
+000000198805 A yellow pickup truck parked in a parking lot.
+000000006723 A bike and walking path running side by side near a city street.
+000000338325 A striped plane flying up into the sky as the sun shines behind it.
+000000338532 THERE IS A GIRAFFE THAT IS STANDING IN THE GROUND 
+000000350679 a newly married couple cutting their wedding cake
+000000043435 Two surfers are riding two large ocean waves.
+000000014439 a colorful kite flying by some people in the park 
+000000369442 a bunch of animals by a nasty street
+000000571008 a stop sign with hammertime written on it
+000000383337 Hand reaching into a box to pick up a glazed donut
+000000104424 A woman standing on a tennis court in the sun.
+000000304396 a big building with a clock at the top of it 
+000000533493 Three men celebrating some kind of personal victory.
+000000578093 A train is shown on the inside of a station.
+000000428111 A man riding a skateboard up the side of a ramp.
+000000093154 Signs along a street on a rainy day.
+000000337987 A brown, black and white bird resting on a tree branch.
+000000186938 a small bird standing on a rocky beach
+000000441586 A man on a piece of equipment resembling a bicycle that has very large wheels.
+000000134722 A train on the track pulling by a train station.
+000000278006 Street signs in front of a hree-story grey frame house with bay windows, dormers, and a porch.
+000000121586 Office space with cat on the television and work.
+000000262048 A woman in a bikini laying under a red umbrella.
+000000520871 there is a large pizza pie on a white plate
+000000456559 A picture of a man wearing a suit, hat, and sunglasses.
+000000220732 A bus with its serial number that can be seen from the air overlooks a panel truck and another bus.
+000000222299 Shoes, a cap, an iPod, a name tag and other items
+000000293390 A bathroom that is covered from bottom to top in stone.
+000000351362 A white bath tub sitting next to a toilet.
+000000552842 A baseball pitcher throws a baseball from a pitcher's mound.
+000000232646 A stop sign is standing next to a road in the snow.
+000000331799 A woman standing in front of a box handing a woman a bag of food.
+000000214753 Two people riding horses on the beach with waves coming in across the rocks on the shoreline.
+000000508312 A stove that has a light above it.
+000000106048 a large white bus that is by a building
+000000441247 Man and woman in the kitchen of a white house.
+000000157390 a close up of a bowl of carrots and broccoli
+000000468954 A man and woman are playing a video game.
+000000059044 A group of people are standing around holding video game controllers.
+000000310072 A park bench has a persons hat sitting on it.
+000000187144 A white bus is pulled up to a sidewalk.
+000000369757 Three different types of pizzas and other items on display at a pizzeria.
+000000256941 A bike with a basket attached to it is parked.
+000000219578 A dog and cat are sleeping together on an orange couch.
+000000212559 a bunch of goats on grass in an area
+000000207728 Two mountain goats standing outside by the snow.
+000000365655 A bus travels down the road next to a mountain.
+000000331604 The man is para sailing on the water. 
+000000128699 A boy jumping on a brick patterned ramp on his skateboard. 
+000000278463 A quite cluttered desk with a lap and many other office material, plus a cat. 
+000000500270 A man riding a skateboard at a skate park.
+000000388056 A batter getting ready to bat with the catcher and umpire behind him.
+000000050149 Many bunches of bright yellow bananas hanging on display.
+000000225405 Two boys are playing with a soccer ball.
+000000050638 A boy carries a stuffed animal as he walks beside some flowers.
+000000263860 Two African elephants greeting each other with their trunks.
+000000160864 three baseball players holding up bats on a baseball diamond
+000000355677 lifeguard station on a beach overlooking a swimming area
+000000093353 Two hot dogs are sandwiched between two slices of bread
+000000219578 A cat and dog napping together on the couch.
+000000432553 A bike, two dogs and a man sitting on public transportation. 
+000000152686 A man dressed in orange is playing tennis.
+000000157756 A large tower with a clock on it sitting near a road.
+000000162581 A man on a court with a tennis racket.
+000000327769 A cat lying inside a sink with a hair brush, sanitizer and toothpaste on top
+000000378873 A lot of food that are in some baskets.
+000000563267 Two men are in a building with brick walls.
+000000320490 a boy is swinging a baseball bat at a game
+000000474039 A baby sitting in the middle of a bunch of teddy bears.
+000000541634 Soup with meat, pasta and greens in a white bowl.
+000000578093 A train is scene in the station with no people around.
+000000519569 A kitchen is shown with chairs and an oven.
+000000054123 A group of zebra grazing grass with a backdrop of bushes.
+000000446574 a toilet a bathtub a rack bottles and a shower curtain
+000000166166 A TV sitting in a living room on a hard wood floor.
+000000488166 A man is standing at the railing of his balcony.
+000000296222 A man eating food off of a plate 
+000000578489 The girl in a pink sweater is playing a video game.
+000000360661 People riding on horses as the run on a beach. 
+000000548339 A boy swings his bat at a baseball game as people watch on the bleachers behind him.
+000000523100 A bowl of peeled oranges sit next to several jars of peeled oranges.
+000000181969 A small brown dog curled up in sleep.
+000000076417 A dog is sticking it head outside of a car window 
+000000396205 two cows grazing in a field near some trees
+000000558558 A colorful but blurry street scene at night features a car driving through an intersection.
+000000269632 A bright yellow transit bus is making it's way down a street in the dark.
+000000314177 The person is sitting on the unusual toilet in the bathroom.
+000000015254 A lunch tray with multiple compartments filled with food.
+000000078748 A couple of motorcycles in front of sign, with people standing around.
+000000045090 People on a wave using a surf board in the ocean.
+000000001490 A man is holding a long racquet on a surfboard in the middle of the sea.
+000000372577 two people playing a match of tennis on a court.
+000000339442 A sandwich with a pile a olives and an apple on a tablecloth. 
+000000482917 A person is sitting with their dog watching tv.
+000000573943 Yellow, white and blue trucks are parked in a parking lot.
+000000323151 A beer sits untouched on a table with other food
+000000209829 A person standing on a surfboard in the water.
+000000571893 A kitchen with a teapot on the stove and books on a shelf.
+000000135902 A train sits running in a modern station.
+000000088432 Truck driving through the intersection of Biltmore and Short Coxe Ave.
+000000208363 A large group of people are eating pizza in an office.
+000000209613 A group of sheep surrounded by three dogs.
+000000178028 a very clean sink in a bathroom and  a towel
+000000482719 A glass bowl filled with prunes and bananas.
+000000505789 Two children standing near wrought iron fence near brick building.
+000000532071 A black bear lying down in the grass next to trees. 
+000000212226 A girl sitting in a truck is reflected in a mirror. 
+000000491470 A city with snow and slush filling the ground.
+000000425702 A person walking along the beach with a surfboard.
+000000157767 People dressed in business casual clothing stand around a table of hors d'oeurves.
+000000338325 A plane flies through the sky at an angle. 
+000000507575 This computer desk has several laptops and a printer sitting on it.
+000000486046 a close up of a giraffe with a zebra in the background
+000000098520 A jet is towed down the tarmac on a dreary gray day.
+000000577932 A woman walking next to a man riding a pink bike.
+000000120572 a old looking tower that has a clock on it
+000000504711 a piece of chocolate cake and a fork on a plate
+000000534394 A herd of sheep walking along a road.
+000000276024 Two men on horses drive cattle down the road.
+000000065736 A tennis player is hitting a ball on the court.
+000000464824 The woman is bending over to pick something up off of the ground.
+000000352618 A pear, orange, and two bananas in a wooden bowl.
+000000237071 Tennis player running with ball in mid air.
+000000278353 A sheep with short hair stands under a fence.
+000000376206 a man riding a surfboard under the swell of a wave 
+000000175364 a room showing a cooker and an oven
+000000373353 People crossing a street in front of a taxi and tour bus.
+000000554002 A black dog standing between people in line
+000000512403 Vase full of feathers sitting on a table next to a floral drape.
+000000245651 A large white multi layered cake sitting on top of a table.
+000000386277 A photo of an orange growing on a tree.
+000000458325 A man in blue shirt on skateboard next to cars.
+000000434996 A pile of stuffed animals sitting on top of a bed.
+000000553664 A red laptop with earbuds sitting on a table
+000000074256 Two women in winter types clothes sitting at a train station
+000000281179 A rain covered terrain after a night of rain. 
+000000459887 three yellow  bananas sitting inside of a basket
+000000518213 a very ugly cow with long horns  staring
+000000523100 A person fills jars with orange slices from a large bowl.
+000000179141 Smiling people indoors posing with a delivery pizza.
+000000385997 A room with filtered light coming through vertical blinds on a doorwall,  with a sleeping dog lying on a dog bed next to a rocking chair
+000000505451 A person on surfboard riding on a small wave.
+000000099054 some planes on an air port run way
+000000033368 there is a male tennis player wearing a blue shirt playing on the court
+000000013729 A group of people are playing Wii in their living room. 
+000000066561 A view of a baby goat standing and a mother goat sitting in a field.
+000000013201 A man is airborne on a skateboard pointing to the sky behind him.
+000000475387 A passanger train stopped to pick up passangers at a station.
+000000553776 A man riding a motorcycle in overcast weather
+000000256192 A big, yellow police motorcycle parked on a city street.
+000000213445 A cutting sitting upright in a pottery bowl on a coffee table
+000000391648 a clocktower standing high with lights on 
+000000295138 a young man is doing tricks on his skateboard 
+000000101780 A giraffe eating some leaves from a tree at the zoo
+000000267670 A white sink sitting under a bathroom mirror.
+000000042563 A yellow bus travels through a snowy, tree lined street.
+000000151657 A man and a woman dressed in a gangster costume and ball dress.
+000000085682 The people are waiting to ski on the snowy mountain.
+000000309173 there is a red stop sign on this street
+000000204186 Spectators enjoying a show of two men on four-wheelers.
+000000557501 A colorful public restroom with striped walls and yellow counter.
+000000432898 A man standing on a lush green field under a kite.
+000000085682 A line of skiers waiting for the ski lift.
+000000221708 Lights shine on a wooden dining room table.
+000000419408 A skateboarder is attempting a trick mid air.
+000000117197 a lady on her phone sitting on a curb
+000000027696 there is a small pizza that is on a white plate
+000000320232 A Public Parking Payment booth in a lot.
+000000362520 a person holding onto a skateboard during a jump
+000000460967 A big bus in a parking lot next to a big building 
+000000532855 A young man doing an axle grind on a piece of pipe in a park.
+000000396274 A kale and sweet pea home garden getting the last rays of sunlight. 
+000000083172 A woman holding a racquet on top of a tennis court.
+000000069213 Two men who are wearing suits and hats standing next to each other.
+000000211825 A man blowing out candles on a cake.
+000000543300 A white double decker boat n water next to buildings.
+000000167353 A large brown wooden clock mounted to a wall.
+000000425227 A being is doing something as of right now that is splendid. 
+000000506279 Stemmed glass of amber liquid on a counter
+000000106330 an image of a stuffed teddy bear sitting up
+000000078404 3 ladies sitting together on a bench in the park. 
+000000419653 A kitchen area with a table full of various foods.
+000000161875 A girl standing in a field in a dress.
+000000430871 A parking meter that is placed right next to a parking lot.
+000000197658 Cather preparing to catch ball in an empty field 
+000000243867 a bus and a car on a city street
+000000512564 A city street lined with very tall buildings.
+000000509258 A street sign shows directions to highways and towns.
+000000052413 A child holding a pink toy cell phone with "Princess Aurora" inside it.
+000000151662 A giraffe at a zoo eating some grass.
+000000100582 Some meat and greens on a pizza crust.
+000000504580 A couple of giraffes looking at the camera with alert looks.
+000000068078 A dog in a small bathroom with an orange shower curtain.
+000000562818 A group of people in the snow with skis.
+000000334977 Various angle shots of the Nokia Windows cell phone
+000000116208 pizza on a serving tray with empty plates next to it
+000000138115 A man puckering his lips holding an orange
+000000027932 A catchers mitt sitting inside of a blue baseball hat.
+000000127394 A group of people sit at a table laden with food.
+000000207585 Four stuffed animals, a leopard and three teddy bears, in a row sitting on a stone ledge with grass and trees behind.
+000000442993 A group of buildings with clock tower and trees in night sky.
+000000352582 A man in a suit tossing a frisbee
+000000074733 A table topped with three mugs of beer and two pizza's.
+000000423798 A street at night is lit up by streetlights.
+000000091921 The zebras are behind the fence in the zoo
+000000055528 Young man wearing glasses holding up a toothbrush.
+000000173004 pizza with olives on black plate setting on table
+000000471567 Giraffe from the zoo staring into the distance.
+000000269866 the microwave is closed the clock on it is at 10:14
+000000239857 A woman is about to throw a frisbee in the dirt.
+000000162092 A lady is posing for the camera while many others look on. 
+000000014831 White and orange fur lays on a white blanket.
+000000270883 A woman in lingerie is bound to a bed.
+000000302990 A man is on a surfboard riding a wave. 
+000000182417 An extravagant dessert on a plate overlooking the water.
+000000060052 A woman twirling an umbrella with flowers on it.
+000000506004 The London eye stands overlooking the Thames river, various buildings, and Big Ben.
+000000259382 A giraffe stands with it's front legs spread apart.
+000000492992 Bird sits atop cattle on snow covered field.
+000000064462 A woman smiles as she stands in skis on a snowy hill.
+000000082180 Two little bears sitting on a weathered wood chair.
+000000144300 A motorcycle parked behind two vans in a parking lot.
+000000481582 There are people standing next to a horse.
+000000181753 A living room with a loveseat and chairs surrounding a table.
+000000262048 two people on the beach laying under a red umbrella
+000000301061 An elephant is being fed while in a truck.
+000000575243 A person is walking down a sidewalk holding an umbrella.
+000000546325 A couch and table sit in a living room.
+000000005477 A brown LOT airliner sitting on the tarmac
+000000540928 A cat that is standing on a wood floor.
+000000261796 A black and white picture of a large house is shown.
+000000466416 The view of a clock tower near a body of water at night.
+000000344909 A fire hydrant on the side of a street.
+000000209530 A group of people sitting around a table.
+000000201426 two young boys are enjoying pizza and pepsi
+000000122217 Two male skateboarders rolling down a paved slope.
+000000569565 A yellow dump truck driving past a forest.
+000000156924 Two people are playing on a Wii in a house.
+000000343076 A fluffy cat laying on an electronic device
+000000555012 The roll of toilet paper beside the toilet is empty. 
+000000474170 Two elephants walk through a grassy area with trees.
+000000479953  a girl wearing all green playing soccer on a soccer field
+000000580294 A lady is preparing pancakes in a charming white kitchen.
+000000460333 a kid is sitting on a small bed in a room
+000000263796 an extremely nasty toilet in front of pipes and cable
+000000084674 a girl holds a baby eating a bagel with a boy standing next to them
+000000382030 A pair of sandals, an umbrellas and other items are piled onto a table.
+000000385719 Laptop with computer monitors displayed on wooden desk.
+000000185409 A herd of zebras playing in an open field.
+000000234757 there is a street that has been very flooded 
+000000574297 A man dressed as a jockey riding a horse along a path.
+000000035062 a sleeping child in a bed with a black and white cover
+000000012748 a man letting his baby pet a horse
+000000262587 a man riding the waves on a surfboard
+000000109992 A woman on a snow board riding down a hill.
+000000526256 Clocks may be seen on all sides of the tower.
+000000566524 Two people walking along a side walk next to a train on the tracks.
+000000407614 Small family kitchen with picture covered refrigerator next to dining room.
+000000223789 This is someones sink in their bathroom in their home.
+000000155291 A cat is sleeping in a window sill. 
+000000255165 The computer desk has two monitors and two keyboards near a laptop. 
+000000563281 A little boy standing in a room next to a chicken sculpture.
+000000224807 The large crowd has gathered and is eating. 
+000000379453 An Air Force jet flying in a deep blue sky.
+000000189451 A piece of cake left on a plate
+000000074209 A counter in a kitchen line with black metal stools.
+000000142324 A group of people riding motorcycles and bikes down a city street.
+000000573626 Two large animals walking on rocks along the side of river. 
+000000447789 A street sign is above a do not enter sign.
+000000531134 A traffic light with little faces to help people know what to do.
+000000130579 A baseball pitcher standing on the pitcher's mound palming a ball.
+000000333697 a alley way with graffiti and some signs
+000000291791 A World War Military Motocycle on display at an event.
+000000090284 A woman is standing on a tennis court with a racket.
+000000386210 A tall glass case with some fruits in it and some grass growing
+000000475064 The front of a restaurant with someone holding up food 
+000000226058 A boy and girl teddy bear are posed next to each other on a couch.
+000000214703 Two people lying on belly on surfboards close to each other, in large watery expanse.
+000000564133 A herd of elephants are marching triumphantly through a field.
+000000561958 Many people attend the ski event dressed in warm clothing.
+000000496954 A slice of cake with frosting on a plate
+000000042070 Blue, white, and green passenger bus parked at a stop. 
+000000298251 A panoramic view of zebras in a dry field.
+000000284725 A red bus sitting on the side of a road.
+000000217614 A giraffe standing in some tall grass outside.
+000000092124 Bathroom with orange shower and mirror/cabinet curtains 
+000000046804 A sheep that is standing in a field.
+000000500464 A sink in a bathroom is shown with other items.
+000000328601 A tennis player gets ready to hit a ball.
+000000213255 A man cooking hot dogs with a light colored jacket on
+000000058111 An orange and white cat laying on top of a sink.
+000000521719 A porch with several wind catchers hung from hooks. 
+000000222559 a number of people in a bod of water
+000000391648 A large clock tower lit up at night time near another building.
+000000174123 A pizza topped with cheese, tomato sauce, and mushrooms being sliced on a plate.
+000000270474 A girl with a baseball bat in the backyard smiles. 
+000000543043 A nice red car parked in front of a roadside junk sale.
+000000072281 A man on a skateboard with a hat. 
+000000003934 A young girl holding a controller playing a video game.
+000000392933 a single giraffe walking towards a group trees
+000000345361 Children at a birthday party sitting at a picnic table.
+000000179487 A woman tennis player is about to return the ball. 
+000000110784 A trolley with people drives down a street by a building.
+000000294163 A clock tower on a late eveining day
+000000267351 A large clock tower is yellow and white.
+000000022396 a plane flying by a bright moon in the sky
+000000240940 a cat siting on the ground watching a tv
+000000332845 The man is sitting next to his bed on the laptop. 
+000000024144 The pizza looks like it was ordered with extra cheese.
+000000192670 A young boys is hitting a baseball while another young boy is playing catcher.
+000000125472 A man flying through the air while riding a skateboard.
+000000147745 A couple of garbage trucks sitting in a landfill.
+000000371042 Three people playing tennis on a grass court. 
+000000396568 a black and red train going down a track by some trees 
+000000290843 A cat is sitting on the bed near a person using a laptop.
+000000573094 A bike in view in a living room with a Christmas Tree in the background.
+000000224222 three guys are catching waves and one is on a surfboard
+000000011615 Many large signs stacked up on a pole.
+000000020247 there are two brown bears together in the wild
+000000094751 A parking lot is on the side of the ocean.
+000000273760 A man is dressed in blue playing tennis.
+000000546011 A group of zebras that are standing in the dirt.
+000000145597 A bunch of girls sit on the grass with several things on a blanket.
+000000504711 A fork rests on a plate next to a piece of cake.
+000000105264 a few people running around some kind of animal
+000000266082 A fire hydrant waits in front of a tall building.
+000000082715 A man riding a wave on a surfboard.
+000000579900 A pizza topped with an assortment of vegetables on a wooden table.
+000000365642 A woman in a coat, sitting outside on a toilet.
+000000343076 Cat hangs it's paw over the edge of the laptop 
+000000107226 Two men being drug on buggies by dogs.
+000000291619 A Frisbee player is about to catch a thrown Frisbee.
+000000560312 An indoor station has a television on the wall.
+000000180135 A man holding a microphone in one hand while he holds a skateboard in the other.
+000000151938 A long wooden table with a woman sitting under an umbrella.
+000000535858 Boogie boarder about to catch a wave about to break on the ocean.
+000000085089 People are holding wine glasses next to each other.
+000000142472 A car stops while people cross in front of it. 
+000000172547 A scenic view of a valley with mountains in the distance.
+000000311392 A black and white cow at market in her stall.
+000000300039 A person holding up a cell phone by a lit candle.
+000000027982 The black and white photo shows a toilet and a bathroom sink.
+000000351530 The two women sit on a bench together.
+000000097679 A couple of cars parked in a lot.
+000000555012 a bathroom done in almost total white 
+000000511398 A multi-colored black dog laying on top of the sand.
+000000107094 A man standing on his skiesat night with a light  on his head. a city is close behind him.
+000000025096 A young boy cutting a skate board shaped cake.
+000000534827 People are riding on motorcycles together down a street.
+000000521405 a stuffed animal that has a tooth brush
+000000144784 A hand poised to push a button on a liquid filled blender.
+000000338901 A dog looks up while laying on a blue couch.
+000000425221 A person at an airport terminal with planed in view outside of the windows.
+000000148783 an image of two zebras grazing in the grass
+000000073118 Two stuffed brown teddy bears sitting next to a wall.
+000000127092 A city street with a road sign next to some buildings
+000000142620 a man that is sitting in front of a picnic table
+000000382743 Two kites, one with a smily face fly high in the sunny sky.
+000000026926 An orange fire hydrant with a face and bow tie drawn on it. 
+000000486046 A giraffe is close up next to the camera.
+000000088218 A traffic sign is shown with traffic signs above it.
+000000110638 A baby elephant is walking under a larger elephant.
+000000226417 Most people negotiate the city streets with motorbikes in this Asian city.
+000000227765 Overview of a pot of vegetable soup with wooden spoon on a stainless steel stove top.
+000000007795 A hotel room with two beds side by side  .
+000000288882 Two giraffes standing next to each other at a zoo
+000000129135 Blue and yellow train at a train station in front of a bench. 
+000000395903 a man that is holding a surfboard in hand
+000000270402 Two gray elephants standing next to each other on a road.
+000000170099 A man is standing talking in front of a picture.
+000000475223 Tan and white wading bird standing in shallow water, perhaps a swamp. 
+000000081394 A decoratively painted business truck for a motor company
+000000197658 a baseball player on a baseball field fielding a baseball
+000000331817 A person is eating food at the table.
+000000287291 The cars at the intersection are waiting for the pedestrians to cross.
+000000032861 A young man stands inside with an umbrella pouring rain.
+000000335450 a small horse and a larger horse standing behind a fence
+000000148730 a boat is sitting next to a dock
+000000283037 A speed detector displays the current speed of a car
+000000157756 A clock that is sitting on the side of a tower.
+000000577735 A vase full of purple and white flowers.
+000000153669 A baseball player hitting a ball at a game 
+000000080274 Elephant mama and baby in their zoo enclosure.
+000000402433 A panned sliced pizza and a bike shaped cutter.
+000000273711 A table covered in different kinds of baked goods and fruit.
+000000187144 A group of people stand on a path outside a parked bus. 
+000000410496 A man plays tennis during a competition while a crowd looks on. 
+000000189213 A pile of old furnishings on the curb for trash pickup.
+000000047828 People walk by a boat near a lighted bridge
+000000345027 A vehicle and a motor bike on the beach.
+000000324927 a plate with a couple pieces of pizza sitting on it 
+000000197528 A cat is looking out a window in front of some blinds.
+000000155179 an upclose photo of a cake with chocolate pieces
+000000525155 A sheep is being carried up a hill.
+000000439623 An old white fire hydrant on a sidewalk in a town.
+000000361506 This skateboarder is riding the cityscape at night when traffic is minimal.
+000000199551 Boy sleeping with no blanket in a large bed. 
+000000138856 some kind of bread that is on a pan
+000000366611 A dog chewing on top of a large white ball.
+000000366178 a road that has some snow all around it
+000000303908 A wooden bench sitting next to a large body of water.
+000000170739 a couple of elephants are standing behind a fence
+000000481413 A toddler boy stands at the ocean's edge
+000000074058 A bunch of bikes that are sitting on the sidewalk.
+000000181796 A close-up of a plate of food containing meat and beans.
+000000114884 A woman with a cigarette standing, while a group of people with luggage walk towards a bus.
+000000557884 A green plant with hanging branches on a wall.
+000000156643 A man cuting a piece of cake while a person in costume stands behind him.
+000000157601 Woman sitting at table with beverages consuming sandwich. 
+000000062692 a tennis player with a racket on the court
+000000482487 A clock in a metal frame mounted on a wall.
+000000314264 A man putting flowers into a white vase on a grave.
+000000349480 some teddy bears dressed and posed for a picnic 
+000000255965 A cat pausing as it's picture is taken.
+000000168337 a woman is pulling luggage on to a sidewalk near a fire hydrant
+000000256916 Several square pizzas are sitting on round plates. 
+000000389197 a small boy in an orange shirt on a blue and white belly board
+000000034760 A full view of a bathroom with sink and toilet in the back. 
+000000018737 A motorcycle parked near a fence by a beach.
+000000474167 A woven basket with slices of pizza in it
+000000344029 A bus that is driving down the road.
+000000315450 some cars a red bus and a blue bus
+000000043314 A downhill skier doing a trick in front of the camera.
+000000456303 a person jumping a snow board in the air
+000000299887 Man and woman sitting on the back of a motorcycle.
+000000090062 A cow is standing next to chopped wood looking at the camera.
+000000370486 A girl stands with her umbrella in hand in a crowd. 
+000000392481 A fighter jet plane ascending into the sky.
+000000106281 Two trains on the track at a railway.
+000000491867 Man with green tie with white polka dots stands at outdoor event.
+000000578489 Group of children and adults playing a video game.
+000000406997 Small children standing around a small white sheep. 
+000000035279 A man laying his head down next to electronics, a butterfly, and a book. 
+000000549930 Two people with an umbrella walking on street in rain.
+000000114871 A plate with a banana and slices of cheese.
+000000166426 A white plate filled with meat and vegetables.
+000000237316 A small bathroom still has all of the necessary amenities.
+000000320743 Boys with wide hats are riding elephants on the dirt.
+000000131556 A skier stands on a small ledge in the snow.
+000000492905 A desk sitting on saw horses with a monitor, keyboard and mouse.
+000000419882 a lot of trains that are parked next to each other
+000000032811 A bird sits by a window on the water.
+000000509260 A game is displayed on a television in a living room.
+000000183675 an old man sitting on top of a horse next to the mountains
+000000094944 man on his knees with his snow board still attached, hands on hips
+000000038210 A man in a racing bib in his skis and a beanie. 
+000000463522 A horse is pulling something down a residential street.
+000000528314 The person was standing on the snow skiing course in his skis.
+000000186637 A giraffe that is eating some of his food from a high up feeder.
+000000226984 A kitchen with white appliances, countertop and oak cabinets.
+000000553339 A sign that gives directions to drivers driving down the road
+000000394677 A woman that has fallen asleep next to a man wearing a headset. 
+000000247806 A church tower with a clock in the centre and arched windows
+000000311883 A yellow train engine pulling train cars over road.
+000000395180 A person is standing with a bicycle while people row by in a boat.
+000000364636 a black brown and white dog with a red collar
+000000194724 A small pizza and beverage sitting on a wooden table.
+000000223090 Salad ingredients,  carrots, onions and some leafy green
+000000248400 a kid is showing a pizza with a joke written inside the box cover
+000000477227 The shore line of a city on water 
+000000304812 A person carrying a surfboard on the beach at sunset.
+000000374369 The skier is on the flat   with front ski edges together and holding ski poles.
+000000508312 This is a kitchen counter, cooktop and double range.
+000000095786 Three hand formed clay pots with dripping paint
+000000058350 a woman holding onto a piece of pizza with her mouth 
+000000571264 Man in black jacket standing in front of a black motorcycle. 
+000000277020 A man sitting at a bar being served by a woman
+000000053994 A parking meter is on a sidewalk next to some green bushes.
+000000042102 A young man that is wearing a nice suite coat with a skirt and a purse.
+000000139077 A young boy holding a Nintendo Wii game controller.
+000000240754 A small cow in a enclosure with straw on the floor. 
+000000122745 a vandalized stop sign in the dark with a sky background
+000000015660 a man out in the water next to his surfboard watching some kites
+000000152771 An empty parking lot features green trees and a single white bicycle.
+000000119365 A couple of people with a laptop at a table.
+000000093965 Two fake zebras are composed of materials and placed on grass. 
+000000452321 Red bus sitting in a bus station parking spot.
+000000037777 The kitchen is clean and ready for us to see.
+000000246436 A person slices a pizza fresh out of the oven.
+000000507975 Three jockeys and three horses competing in a race against each other
+000000307172 A pizza pie being enjoyed at a cafe of some sort.
+000000559547 Three vintage baseball players posing together in uniform.
+000000447522 A dish on a blue plate contains carrots, onions, broccoli and rice.
+000000032735 a person jumping a pair of skis in the air
+000000189820 A view of a large, cluttered office desk area.
+000000232088 this living room has all white couches and a flat screen tv
+000000090108 a bathroom with blue and white walls 
+000000306893 Red and white train at the end of a train station.
+000000081394 A tan utility truck with a blue interior.
+000000485027 Professional tennis player on clay serving the ball
+000000257084 a person on a skate board ride down a ramp 
+000000310072 A bench on cement in a park 
+000000010363 A gray cat standing on top of a black car.
+000000079229 a dog follows closely behind a man on a horse.
+000000530099 A orange calico cat on top of a vehicle.
+000000137106 a hand of a person wearing a pink and white shirt 
+000000127182 A KITCHEN WITH A SINK AND APPLIANCES 
+000000220584 A surfer loses his balance at the end of a ride
+000000273642 A dog lies down holding a remote control in its paws.
+000000445792 there is a woman sitting on a couch playing a video game
+000000084431 A slice of pizza with toppings on a plate. 
+000000450303 Small group using laptops and headphones in a classroom.
+000000508370 two men standing next to a bike parked on a street corner
+000000293044 A lunch box with a sandwich, carrots, salad and a muffin.
+000000005037 A passenger bus pulling up to the side of a street.
+000000296231 This small room is packed with personal memorabilia.
+000000231097 A half eaten plate of food sits on the table.  
+000000055002 Metal commode from above, photographer's shoes visible. 
+000000156278 Busy kitchen in residential home with cabinets and wooden table.
+000000394677 A person bent over sleeping in a chair next to others.
+000000571008 A stop sign that is displaying a different meaning than the original one.
+000000117525 A man in front of a Christmas tree with his dog.
+000000007281 two guys horseback riding and playing on the beach.
+000000048153 A foot wearing a pointy pink flat shoe rested upon a bench.
+000000426203 A skateboarder is riding along the top of a rail.
+000000171382 a person riding a skate board on a city side walk
+000000436738 an old photo of cars in a city street
+000000201148 A train approaching a railroad crossing during the day.
+000000088462 A school bus goes down the busy road.
+000000372349 A man and women in the river washing and eating.
+000000248111 some cans sitting on the kitchen floor 
+000000373315 A skateboarder doing tricks on the rim of a pit
+000000140076 A made up dinner table with a flower vase on the table. 
+000000457884 a child catching a ball at the ball park 
+000000106389 A cat sitting in an orange chair in a bedroom
+000000326082 A big, open room with large windows and wooden floors.
+000000560880 A black and white photo of a man working a field with a team of horses.
+000000313588 A woman stands on a cobblestone walkway carrying an umbrella. 
+000000136772 a small bowl of fruit on a table 
+000000442746 A girl on the ground ready to throw her frisby.
+000000035326 a black stove top built in to a kitchen counter
+000000414676 Clouds soar above a tall building on a sunny day.
+000000020107 An old rusted fire hydrant is on the ground near a painted wall.
+000000267670 A toilet, mirror and sink in a toilet
+000000083531 There are many links of sausage on the grill.
+000000275727 A freight train riding on tracks through a forested area.
+000000342006 A barge floating down a river with the skyline in the background.
+000000427997 a woman leaning over a bit near the end of a court
+000000097924 Three people try to guide a horse into a trailer. 
+000000149568 A man and two dogs on the beach.
+000000387148 A herd of cattle grazing in a field 
+000000505565 A flock of birds standing on top of a lush green field.
+000000540466 A woman stands in a garden holding an umbrella.
+000000425906 People look on as an airborne snowboarder competes.
+000000287347 A person making thin crust pizza in a kitchen
+000000050943 Man surfing a wave on a dull day.
+000000562843 A cup of coffee, cell phone and scissor sitting on a desk.
+000000024243 A woman with a beer sits inside a refrigerator on the footpath
+000000323496 A boy swinging a tennis racket at a ball on a court.
+000000248752 There is a man holding a baseball bat in his hands.
+000000338428 Truck viewed from the back with vans and trucks visible in the background of the image.
+000000387387 A yellow bus that is sitting in the grass.
+000000579818 A train driving over a small bridge on a green hillside.
+000000342971 A man reaching out to catch a frisbee.
+000000005529 A person on skis skiing down a mountain slope.
+000000162581 A man playing tennis is swinging his racket.
+000000296224 A close up view of a very big pretty city bus.
+000000000872 two baseball players are playing baseball on a field
+000000152214 a close up of a person standing wearing a suit and tie
+000000540932 Beach scene with walkers on beach, boaters and swimmers.
+000000010583 Plate of food with gravy on mesh table with knife
+000000010707 A man standing in a living room holding a Nintendo Wii game controller.
+000000344100 A white plate on a table topped with a sandwich.
+000000474786 A bathroom with a sink and television inside of the mirror.
+000000019432 A man with a racquet standing on a tennis court.
+000000034139 Two people posing next to a giant suitcase in front of a building.
+000000009914 A grilled chicken sandwich sits beside french fries made with real potatoes. 
+000000528524 Some cattle are walking on a dirt trail
+000000397133 A baker is working in the kitchen rolling dough.
+000000391290 two teenagers playing in a grassy field while a fribee flies their way
+000000255917 Traffic stopped at an intersection waiting for the light to turn green.
+000000052891 A very cute looking dog with a disc on the beach.
+000000502732 Side by Side stainless refrigerator in a kitchen.
+000000092177 A birthday cake with a island design and theme.
+000000416451 Two men in living room playing a game with a Nintendo Wii controller.
+000000572408 A bull standing over a calf in a large open pasture.
+000000038825 Two black and white zebras standing next to each other while grazing on grass. 
+000000292060 A white kitchen with stove, sink, and refrigerator.
+000000496954 A slice of brown cake possibly a carrot cake next to a slice of grapefruit.
+000000408112 A P-51 Mustang and B-25 Mitchell on the airport tarmac. 
+000000226058 A teddy bear couple sit on a couch together.
+000000248810 A giraffe is looking over a metal fence.
+000000016249 there are many people that are sitting on the benches
+000000578871 there is a young girl sitting at a table eating
+000000360943 A fat orange cat sitting on a deck
+000000114884 Buses lined up letting people aboard them. 
+000000197796 A bathroom with a white sink and a towel rack.
+000000202339 A man carrying a sign walking past buses.
+000000367195 A dog laying down on a fluffy carpet
+000000341719 Skiers and snowboarders mill about on a mountain.
+000000214539 A soccer player prepares to kick a soccer ball.
+000000295809 A freshly paved road with a bicycle lane marker on the pavement.
+000000168458 there is a small glass vase that has purple flowers in it
+000000080949 A kitten laying over a keyboard on a laptop.
+000000232649 A woman is taking a photo of herself in the mirror of a bathroom.
+000000322352 A wooden table with four wooden chairs in front of two windows..
+000000176446 a yellow book with a black pen on it
+000000315187 an old passenger bus parked next to a walgreens 
+000000450488 A bedroom with a bed and a flat screen TV on top of a wooden dresser.
+000000327769 a cat laying in a bathroom sink while looking at the camera 
+000000010583 A large pile of food on a plate with knife.
+000000049759 A man serving a volley ball on a court.
+000000239717 Server smiling while carrying food inside a busy restaurant.
+000000124975 Two zebras facing opposite directions in a grassy field.
+000000371677 This computer set up has two monitors and an ergonomic keyboard.
+000000299720 A half of face of a giraffe and a tree.
+000000124798 Various vehicles driving on a road surrounded by trees.
+000000029596 A nicely decorated living room and dining area.
+000000454404 Young girl sitting on a bench checking her cell phone.
+000000266400 A row of parked motorcycle sitting on the side of a road.
+000000429718 An old- fashioned train rounds the corner on train tracks. 
+000000199771 People in a kitchen with cooking attire on
+000000567886 A white teddy bear laying on top of a blue rug.
+000000041488 An green and white overhead street sign on Interstate 278 for Queens and Bronx, showing a truck restriction.
+000000238410 A group of people sit outside with wine glasses in hand. 
+000000348216 There were six toilet all lined up along the wall.
+000000167540 The underside white and blue plane in sky.
+000000089271 A cat wearing a hat while resting it's paws on top of a chair.
+000000012748 A man and child next to a horse.
+000000011760 Three zebras stand side by side looking at the camera.
+000000080666 a white and brown cat sitting alone on a bench 
+000000116825 A cat reaching for a knife that has it's blade out.
+000000132375 a glass vase with three pink flowers and a drink
+000000570539 A man looking out the window of a bus
+000000057760 A large group of people on a sandy beach.
+000000293245 A Yamaha motorcycle parked outside of a building.
+000000460379 Giraffes stand in a field surrounded by woods in a zoo.
+000000550691 a passangers bus moving on the opposite side with the car
+000000581615 A bathroom with an auto flush urinal 
+000000368752 A young boy is smiling and he has food around him on a table.
+000000484404 The inside of a city bus in a bus depot.
+000000439994 people playing a baseball and many people watching
+000000223188 An Elvis impersonator riding on top of a statue.
+000000367195 A picture of a dog laying under a table on the rug.
+000000548555 A person holding a chili hot dog on a napkin.
+000000235241 A young lady that is making a diving catch to get the frisbee. 
+000000572678 A room with a couch, chairs, lamps, a table and curtains. 
+000000153797 A pitcher for the Camden baseball team throws a pitch.
+000000116068 This is a man fixing his kite with a woman on his left and a man to his back right.
+000000406417 three little kids that have different ties on
+000000450439 Two red, white and blue kites flying in the sky.
+000000492905 a picture of a pc on a makeshift computer table
+000000119365 Student working on the computer while another student shows her a mobile phone. 
+000000283268 A child plays with a toy picnic set including a plastic hotdog.
+000000228942 A bunch of cars are crossing an intersection.
+000000461275 Street signs on the corner of Furnace St and Crapo St.
+000000554291 a cat sitting inside of a plastic bucket
+000000245102 A snow covered ramp in a parking garage
+000000572517 A Polar Bear walks toward water, while a large bird lands on the opposite bank.
+000000520324 Someone is driving a cart near a single engine airplane.
+000000319607 A sign post with an electric traffic signal attached.
+000000100489 A light blue bird perched on top of a tree branch.
+000000419974 He is working on the meal while the dog is patiently waiting. 
+000000081061 a chair,couch and suitcase sitting by the road 
+000000397327 A white bathroom is shown with a blue towel.
+000000205333 Two people on skies on a snowy mountain slope.
+000000007386 A motorcycle frame that has been made in to a tricycle. 
+000000286422 A sailboat in a river with people and a dog on it. 
+000000253819 a male skateboarder in a black jacket doing a trick
+000000463174 A tennis player hits the tennis ball with her racket
+000000359833 a kid is reaching up to some apples
+000000183716 A couple of little girls sitting next to each other.
+000000248400 A man opens pizza box and stares at the camera.
+000000442822 Empty boats docked on the water in a harbor.
+000000403584 Two surfers stand with their surfboards on a beach.
+000000565853 The man is standing in front of a television playing a game 
+000000142971 A man and three children are on the surfboards
+000000343453 A skier stands on skis on a snowy path.
+000000497867 A white bus turns a corner on a city street.
+000000203864 Two men with tennis rackets playing doubles tennis.
+000000491071 A sleek modern bathroom design with square sinks.
+000000361268 A brown and white cow standing in front of an iron fence.
+000000022396 a jumbo jet flies through the sky with the moon peeking out
+000000390246 a person riding a surf board on a wave
+000000369812 signs hanging from a metal pole bordering a street.
+000000490413 The airplane has landed behind a fence with barbed wire
+000000165351 Sliced bananas sit on top of a pastry dish on a plate.
+000000418281 two cows outside one laying down and the other standing near a building 
+000000509699 A woman sits on a chair and watches a vintage TV.
+000000325306 A young zebra grooms its self in the sun.
+000000330818 A young man stands in a kitchen watching a griddle. 
+000000447611 Two people are sticking knives into a laptop.
+000000320696 A man riding a wave on top of a surfboard in the ocean.
+000000336587 A stop sign on a broken post across the street from houses.
+000000388927 a hand holding a gray and blue clock and some peoples legs
+000000491725 Picture of docked boats while sun is setting in distance
+000000394940 A young boy sits at a table with his head in his hand.
+000000556873 A cake that is sitting in the grass.
+000000459809 A group of people flying a kite on the water.
+000000574810 A long-haired cat sits on a window sill and stares outside.
+000000414170 3 people separated by a short distance skiing on a ski slope
+000000068078 A puppy sitting on the ground in a bathroom
+000000544565 a tray with pastries drizzled with chocolate and whip cream
+000000023751 A statue of a boy and a girl flying a kite in front of a building.
+000000278353 A sheep standing on the grass in front of a fence
+000000185472 A person getting off a bike and a train coming on the tracks.
+000000399560 a small kitten inside of a laptop on the floor
+000000365886 A person using a photo filter holding a samsung cell phone.
+000000532901 A Victorian house looked very much like this model one does.
+000000392818 A dog in a leash harness posing for a photo.
+000000407614 A smaller kitchen with a very decorated fridge. 
+000000079565 An adult and a younger giraffe are facing the same direction.
+000000397279 The tennis player does not know what to expect with the next serve.
+000000550426 A vase of flowers on a dining table.
+000000078170 Two women look at their phones on the train.
+000000368684 A room with a tv in the background and a laptop screen in the foreground 
+000000500716 A man is wearing surgical type scissors as bifocal glasses.
+000000343315 Patchy lawn, a road sign and a fire hydrant
+000000405279 Kites are flying in the sky over the water at a park.
+000000447789 A Do Not Enter Sign with a Two Way Street Sign above it.
+000000524850 A group of people standing next to a bus under an airplane.
+000000157390 A bowl of cut up raw carrots and broccoli.
+000000050828 Two twin beds with no comforters on them.
+000000103723 A woman is watching an elephant in an enclosure.
+000000173799 Two men herding a pack of elephants across a field.
+000000481404 A golden cross ornament sits in the middle of an empty church.
+000000167898 A kitchen with a toilet, sink and a bathtub.
+000000450399 Amish men and women making homemade doughnuts at a flea market.
+000000431727 A pair of bears climb up in a tree.
+000000095069 A large, old brick building with a clock tower at one corner.
+000000261097 a man plays tennis in front of a large audience
+000000352582 A young man participates in a Frisbee game.
+000000234779 Two sloppy joe sandwiches on two separate plates
+000000570736 A view of a bathroom that is clean.
+000000293804 A view of a fire place with a light on it.
+000000343453 a woman wearing skis on a snowy mountain posing for the camera. 
+000000546976 A man riding on the back of a motorcycle.
+000000565012 A time lapse image of a city street.
+000000211120 Several teddy bears appear to have a picnic on the grass.
+000000470121 Two plates that have different types of food on it near a bottle.
+000000194875 A line of neon colored motorcycles parked in front of a bar.
+000000581357 a person jumping a skate board in the air 
+000000459437 Several lambs leaning up and over a wooden post towards a red cup held by a man.
+000000306893 A red and white train near a terminal in a city setting.
+000000570782 A computer desk is filled with laptop computers.
+000000239347 A picture of a recently married couple displayed behind glass.
+000000233567 a black goat standing next to two white goats 
+000000089697 this woman is sitting on a wood bench
+000000536343 A space has been made to install a bath tub. 
+000000109916 an image of two cakes on the stove
+000000293804 A living room with a fire place and furniture.
+000000258883 Pizzas are cooking on the small grill outside.
+000000151962 A dog looking out the window as seen through a mirror.
+000000527528 A crowd of people flying kites in the grass.
+000000246436 A young man prepares food in a modern kitchen.
+000000414676 A very tall brick clock tower with a clock on each of it's sides.
+000000242724 Two women in bikinis on horses in the ocean.
+000000094326 A sculpture of a man reading a newspaper sitting at a bench.
+000000210855 A view of a bathroom that appears to be very elegant. 
+000000156278 The shelves in a kitchen are covered with pans.
+000000310862 Man in vest and tie standing in room.
+000000434204 A man riding on a skateboard on the sidewalk.
+000000381971 A horse drawn carriage is near a fire hydrant by a curb.
+000000267351 The white and yellow clock with a red roof reads 11:44.
+000000350388 A bowl of soup with green vegetables floating on top of it.
+000000294783 A bed set up on the wooden floor of a large wooden house
+000000398438 The hands of a couple that is cutting a wedding cake.
+000000206027 A slice of pizza with a swirl of sauce on top.
+000000404678 A family sits around a wooden table talking and eating in a restaurant.
+000000353970 Gray cat lying on seat cushion on outdoor wooden chair.
+000000242678 Horse above hurdle at Kentucky horse show event. 
+000000054628 Nicely decorated three tier wedding cake with topper.
+000000336356 A woman sitting at a table in front of her pizza. 
+000000143961 A yellow red umbrella in a crowd of people.
+000000082765 Black and white picture of an object laying on a bed with sun coming through the window.
+000000537053 A lone giraffe standing in a large grassy field
+000000369370 A ham sandwich on a white plate with a kale garnish.
+000000170670 this is a meal oina wood picnic table
+000000484415 A person's hand flushing a toilet with a button on top of the tank. 
+000000001490 A man with a wet suit on standing on a surfboard in the water.
+000000210394 A truck driving down the middle of a street next to a crowd of people.
+000000200667 A cow that is leaning under a metal wall.
+000000157390 Broccoli and carrots together in a bowl on a table.
+000000568147 A parking meter has some colored wrap around the pole
+000000187243 A person is laying on the bathroom floor in front of toilet.
+000000405432 A table with a bag of beans and donuts box on it. 
+000000210030 A white plate with a slice of cake on top of it.
+000000415238 A painting of flowers in vase setting on a table with two chairs.
+000000300039 A person looking at their cell phone at another person taking a picture.
+000000376478 A glass vase on a bench has a rose in it.
+000000402473 two cats in a room next to a white wall 
+000000186042 A person in the snow on some skis.
+000000124277 A train turns on the tracks in a different direction.
+000000405195 This a case full of doughnuts and cinnamon buns.
+000000378673 A person at a skateboard park, jumping his skateboard.
+000000290592 A herd of sheep all gathered near a fence area.
+000000272212 two black and white cows on a green hill
+000000419096 A train going down the train track. 
+000000562243 A guy wearing glasses, a red shirt and a tie.
+000000008690 This is an image of two girls with goats
+000000117645 A person riding a snowboard down a snowy slope.
+000000332901 Tagged, black faced sheep in a pen in a livestock yard
+000000448256 Several men standing next to a truck with it's hood open and a boat.
+000000524456 Two people are at a computer desk with a mouse.
+000000277197 A large formal living room with stained glass windows.
+000000331280 A couple of horses in what appears to be a circus.
+000000276285 A small pizza in the middle of a table.
+000000568981 A man hiding behind a concrete skate ramp taking pictures
+000000426376 A man riding a snowboard down the side of a snow covered slope.
+000000292082 A little girl sitting on a man's lap in an old photograph.
+000000217219 a white pillow white sheets and an orange and tan bedspread
+000000377723 A blue bus parked in front of a tall building.
+000000404249 Boy riding a skateboard down a road or paved sidewalk 
+000000205324 The men are playing a game of Frisbee on the field. 
+000000489339 A woman walking on the beach holding a surfboard. 
+000000429281 There are many bananas sitting on the shelf.
+000000249550 A bed made up with a flowered bedspread and pillows with a night stand.
+000000537506 The person holds a plate with two red apples near pedestrians holding umbrellas. 
+000000451043 a person snow skiing down a mountain  between trees
+000000038070 Looking down into a toilet in a very small stall.
+000000036844 a black couch a green chair a red chair a mirror and a ceiling fan
+000000169356 a young male in mid air flipping a skateboard off of a ramp
+000000459662 Two young girls peeking through the holes in their donuts.
+000000445439 A zebra standing up next to a chain linked fence and eating grass that is on the ground.
+000000312263 Black and white, looking up the side of a building, with a clock and with rows of windows.
+000000130566 a long cargo train going down a track by some trees 
+000000285894 Children looking at a zoo giraffe and its baby 
+000000382734 a white room with not that much in it
+000000404128 The older airplane is about ready to take off from the runway. 
+000000143998 A group of carrots with an organically grown strip tag around them.
+000000064898 two kids with red hair riding surfboards onto the beach
+000000226592 a bedroom with a blanket and some pillows
+000000445439 A zebra is bent down eating some grass.
+000000376278 Several zebras laying down in a rocky field.
+000000338325 An airplane is ascending into the white sky
+000000568981 A man taking pictures of kids at a skate park.
+000000526728 A couple standing outside of a restaurant on a sidewalk.
+000000393569 Three people live in one room who share a bathroom.
+000000189806 A cat observing a dog that's sniffing something on the floor.
+000000121031 The shirtless men are riding horses through the lake. 
+000000162130 A park bench placed near an open field
+000000081766 Three dogs sitting on the levels of an empty tiered garden.
+000000356968 A brown horse with a white mane is on the grass.
+000000360393 a pizza with meat and red peppers and Parmesan cheese
+000000046497 Two little girls sitting on the side of a white boat.
+000000181816 a bench with a statue stting on one side of it 
+000000507015 a group of young people playing baseball in a field
+000000119995 a male tennis player with a blue shirt is playing tennis
+000000392818 a black brown and white dog wearing a pink and white jacket
+000000132329 a refrigerator with four plastic bottles and one is dark blue
+000000214192 two people on dirt bikes making turns on a track
+000000082085 A brightly colored train is stopped on the tracks to let passengers on.
+000000168883 A woman dressed as a zombie in a pink dress with a little girl.
+000000427256 A proffessional baseball player up to bat with a crowd behind him.
+000000242060 There is a teapot and food on a plate.
+000000526706 a small calf nursing a cow in a pasture
+000000129054 A group of zebra stand in the plain.
+000000078032 a couple of park benches on some concrete
+000000131131 A cat looking at another cat on the television.
+000000174018 Two teddy bears are laying side by side on a quilt. 
+000000262895 A man in a black tie and white shirt has a stern look on his face.
+000000014439 A woman and a young girl holding a kite on a green field.
+000000122962 Large group of little kids and adults sitting and surrounding a long table with blue plates.
+000000351530 There are people sitting on chairs lined up
+000000096493 A boy sitting on top of a couch under a pink blanket.
+000000151000 A woman taking a picture on a sunny day. 
+000000007991 Carrots and small green beans on a white plate. 
+000000090003 Two dogs in a grassy field with one dog holding an orange disc.
+000000403565 A person wearing a red biker shirt stands next to his bike.
+000000095155 Four photographs of a person in red jacket snowboarding.
+000000410487 open oven door with metal pie plate of spilled food
+000000545594 A man holding on to a pole on a bus.
+000000190307 A woman is standing on a dolphins head.
+000000018737 A motor bike is parked next to a sandy area. 
+000000051938 a skier approaches a red bar stuck into the snow on a snowy hill
+000000481413 Baby boy with Frisbee on beach near surf watching the waves. 
+000000078959 A drooping pink and white flower with green leaves.
+000000531707 A group of people sitting on a wooden bench next to a  field.
+000000455448 A person holding a cup cake with chocolate icing and sprinkles.
+000000461036 A small black bear is walking through the woods
+000000466835 A bunch of bananas hanging from a tree.
+000000300233 A close up of a bowl of uncooked rice and a jar of oil.
+000000201426 A couple of kids at a table with some food.
+000000034760 Cleaning products are on the counter of the bathroom.
+000000119452 a big pile of fruit that is on top of each other
+000000420840 A young toddler with food on her face sticking out her tongue. 
+000000001425 A plate with a burger that is halfway eaten. 
+000000171788 A man is running to hit a tennis ball.
+000000102331 Kid jumps a hill while riding on dirt bike
+000000304396 A tall building with several windows and two clocks on it.
+000000404678 A family sitting at a large table in a restaurant.
+000000508586 A group of zebras that are in a field.
+000000255664 A yellow dog runs to grab a yellow frisbee in the grass 
+000000213605 A city bus stopped by the side of the road for passengers
+000000136334 A person riding skis down a snow covered slope past a flag.
+000000415748 a man in a white shirt is riding an elephant and some buildings
+000000267903 A very old and dirty rest room with sink and toilet.
+000000509131 A bunch of mini bananas are hanging above a bowl of apples and oranges.
+000000111951 A bathroom with sinks, a brass counter and decorative shower stall.
+000000465585 A large tray is filled with broccoli and chunks of cheese.
+000000335658 Thee computer keyboard is white and it has a matching white mouse. 
+000000074200 A group of surfboards sitting up against wooden poles.
+000000559543 A little girl in a pink shirt holding a game controller.
+000000085576 An empty toilet stall with two rolls of used toilet paper and one full roll.
+000000453341 A large wooden shelf with a bench of action figures on top of it.
+000000356505 A woman riding a surfboard on a wave.
+000000194471 A skate board rider skating on the edge of a ramp.
+000000255165 A computer desk with two monitors and a laptop.
+000000176901 Several people traveling up a hill wearing snow gear. 
+000000097278 A woman in blue jacket standing on a snowboard.
+000000276434 a person that is cutting a cake on a tasble
+000000395633 There is a large ship in the middle of the ocean. 
+000000521405 A toothbrush is being pushed against the stuffed animal
+000000398810 A cat staring out a window at another cat.
+000000041872 A bedroom with  a large bed facing a fireplace that has a television over the top of it, in a shelf space.
+000000268375 A group of elephants contains a baby one.
+000000250282 A picture of a lot of girls and boys in front of a building. 
+000000012639 A little boy holding a bat standing at home plate.
+000000506310 A room with an empty and open refrigerator and a shelf of bottles.
+000000190753 There is a group of four men sitting on park benches.
+000000484351 In a classroom school lunch for the school boys.
+000000116825 A black and white cat sitting on a bench reaching for a toothbrush on a table.
+000000348881 Airline employees by an aircraft parked at the gate
+000000475150 Head and neck of a giraffe in natural feeding habitat.
+000000276055 A woman holding up a large carrot in a backyard.
+000000445722 A covered area with a guitar and other luggage
+000000206135 A herd of cattle sitting in front of a church with a steeple.
+000000273712 a refrigerator in a room next to a wall
+000000176446 a rock like table with a notebook and pen on it
+000000401250 The skier is upside down in the air.
+000000078426 A gray cat on a table with a pen and phone handset
+000000315001 Seagulls in flight and standing on a beach at sunset
+000000405279 People flying kites over a lake in a park 
+000000345261 Two giraffe standing on to of a dry grass covered field.
+000000440617 an image of a bus that is going down the tracks
+000000462643 A person holding a broken cell phone while looking at the screen.
+000000060823 A large brown cow standing in the middle of a flock of birds.
+000000066635 a small wood table with a small computer on it 
+000000476415 A man wearing a white shirt and tie standing in  a room.
+000000497599 A man with a beard and mustache thoughtfully stares into a computer screen at some sort of a meeting. 
+000000294350 A man preparing food in a restaurant kitchen.
+000000350054 One cat sitting in front of the television screen.
+000000378515 A man kneeling down over piles of bananas.
+000000336658 A city road with buses and people on the sidewalk
+000000076416 A double decker passenger bus with advertising for a technology product.
+000000105455 A colorful clock tower along a street and near apartments.
+000000533493 There are a group of friends laughing together and licking out there tongue
+000000369751 A view of a parking lot with parking meters.
+000000464689 A picture of the entrance to an old house.
+000000308476 A polar bear walking on the rocks near a fence.
+000000037670 A man holding holding a giant remote control.
+000000474021 Two men are standing while playing a video game.
+000000399764 A man is walking with a cow who's sporting a red ribbon. 
+000000168330 a city street at night with a lit clock on the corner
+000000437351 A young woman sits near three suitcases of luggage.
+000000036678 A scenic river view with the a palace in the background.
+000000404191 a fridge with a bunch of words on it with magnets 
+000000213593 A stop sign on the side of a street.
+000000050145 This gentleman is shirtless, walking his bicycle down the street of an Asian city.
+000000194716 A woman in sunglasses is outside holding a baby.
+000000194724 A small pizza and a soda pop sit upon a table with magazines, a phone and other items.
+000000378453 A herd of zebras standing on a sandy platform near a small body of water.
+000000179392 A cat sitting on a desk in front of a computer.
+000000131444 A man wearing a hat and a gray jacket.
+000000174018 teddy bear dressed in pink with red ribbon around neck
+000000531036 there is a double decker bus that is next to a small bus
+000000209530 A few guys at a restaurant talking and eating.
+000000530854 Opened up umbrellas in different colors and patterns.
+000000108503 two surfers walking along the ocean beach in the sand
+000000256407 A clock that has replaced numbers with teacups
+000000454661 A STREET WITH A BUNCH OF CARS ON IT 
+000000066635 An apple computer monitor, keyboard and mouse sits on top of a clean wooden desk. 
+000000161609 A person in a subway station with a dog in a backpack.
+000000002587 a banana and a chocolate frosted donut sitting in a baggie
+000000450439 A red, white and blue kit flying among the sky.
+000000033221 A woman holding a baseball bat over her head on the sand.
+000000089761 A toilet is placed at the base of a tree.
+000000187990 A boy doing a trick on a skateboard on a ramp.
+000000563648 A group of skateboarders watching a skateboarder doing a trick in a skate park.
+000000344614 A clock tower with multiple columns in a city.
+000000416745 There is a small bathroom with a white vanity.
+000000316404 Two men posing for a picture in a tennis game in a funny way.
+000000225532 A highway at dusk with cars traveling between cities.
+000000251119 Sandwich with container of fruit and container of juice in small cardboard box.
+000000025228 A woman looks to her right while straddling a surfboard in the ocean.
+000000376310 A bathroom with mirror, hand dryers, and multiple sinks.
+000000260657 A worn and a new sign for the same Mexican food restaurant.
+000000494869 A woman standing at a kitchen counter with a child and a dog is behind her.
+000000442993 A large clock tower towering over a city glowing brightly at night.
+000000280710 People crossing the street near a parked sightseeing bus.
+000000304396 A very tall pink and white clock tower.
+000000327605 The woman wearing skis and a helmet is standing with skis and ski poles. 
+000000054593 a young kid with a baseball bat gets ready for the pitch
+000000323709 Large commercial cargo plane sits on tarmac next to radar equipment.
+000000011760 three zebra facing the camera in a field
+000000002592 a close up of a knife and a cup on a surface
+000000461405 A group of sheep in a grassy area with trees in the back ground.
+000000512403 A vase of feathers on a shelf by some art.
+000000145591 There is a large clock in the foreground and a large building in the background.
+000000479912 A large passenger jumbo jet takes passengers on the runway
+000000512657 two people are playing a video game and a male is watching
+000000103548 A herd of animals traveling down a country road surrounded by a lush green landscape.
+000000153217 A black cat with arched back walking past a motorcycle
+000000435208 Woman sitting on top of a couch in front of two laptops. 
+000000442463 A kid is riding down the street on a skateboard.
+000000432468 The brown cat is lying on a luggage bag.
+000000463690 A man is laying down in a pavement balancing a ball on his foot while others watch.
+000000346905 A family of three elephants standing next to a forest.
+000000059386 A group of giraffes that are standing in the grass.
+000000461751 The man is riding his motorcycle while smoking a cigarette. 
+000000338905 a couple of people are eating at a table
+000000515445 Surfers stand with surfboards on a beach for the early morning sunrise.
+000000283717 A kitchen counter with a microwave and a toaster oven on a shelf 
+000000147223 A large bus on a open city street.
+000000186980 A very pretty kitchen with a small wooden dining table.
+000000243989 A group of young men holding skateboards in front of a building.
+000000404922 A mother with her two children holding tennis rackets.
+000000375015 A brown bear walking in its zoo enclosure
+000000455448 a hand holding a cupcake with brown frosting and colorful balls on it
+000000502737 A person holding up a tasty looking treat.
+000000323751 A train is making its way down the tracks. 
+000000499313 A deep dish and thick crusted pizza with toppings.
+000000446117 an orange sitting on top of a wooden table
+000000299355 A plate topped with broccoli and vegetables on a table.
+000000482275 The dork with the earring stands next to the Asian beauty who is way out of his league.
+000000136715 A sheriff riding a motorcycle in a parade.
+000000054164 a person riding a surf board on a wave 
+000000339870 Street lights in night time in urban setting.
+000000511999 A train station with people standing on a platform beside a train. 
+000000176901 Three people are skiing on the snowy mountain.
+000000350054 there is a cat that is sitting in front of a tv
+000000511321 Two large ferries passing by each other in a city scape,
+000000494913 A picture of someones living room white a person is watching TV.
+000000140583 A herd of animals on the side of the road.
+000000167572 A sandwich with nachos and a salad on a plate.
+000000412362 Two women and one man pose for a picture.
+000000094614 A person in a green jacket skiing down a slope
+000000430377 A woman is standing in the snow wearing a pair of skis.
+000000114049 A group of people sitting and standing on top of a train.
+000000441468 a red and black fire hydrant sitting next to a crosswalk
+000000477805 Man using his laptop computer in a dark room 
+000000380203 A horse standing on top of a grass covered field.
+000000018837 some people sitting on top of a green truck as another man tries to get in it 
+000000125062 A line of small teddy bears are in front of several DVD cases.
+000000051738 A king bead and chair with lamps in a hotel room.
+000000060932 A group of people standing in a room.
+000000248334 A group of people riding on the back of a boat in a river.
+000000131556 A man riding skis on top of a ramp.
+000000270908 A male tennis player lays on the ground
+000000276024 Two men guiding cows while behind on horses. 
+000000258883 Pitas on a grill get topped with vegetables.
+000000112626 a metallic suit case in front of a couch
+000000368961 Two mini elephants walking  toward people on a plank 
+000000335427 a bowl of soup with broccoli next to eating utensils. 
+000000011197 A man prepares to cross the street at a crosswalk
+000000254016 A row of doughnuts being passed through a fryer.
+000000278006 There are a few street signs next to a house.
+000000383337 A person reaching into an open box of doughnuts
+000000426253 Food is cooking inside of a microwave oven.
+000000516173 A woman walks down the water with a surfboard.
+000000411530 some skiers are in a line for the chair lift
+000000148662 Man with a hot dog in a paper rapper in his hand.
+000000180101 A boy getting ready to blow out birthday candles with a girl watching.
+000000327601 A woman sitting on a unique chair beside a vase. 
+000000340697 a close up of many drinks in a fridge 
+000000276285 A table with a plate of pizza and a plate of shrimp on it
+000000319696 there is a red and black microwave on a counter
+000000022705 A woman in a short black dress is holding a champagne flute and standing in front of a refrigerator.
+000000068387 The baseball player is participating in an intense game,
+000000283412 a close up of a dog next to a news paper on a table
+000000069138 A street scene with a close of of a stop light.
+000000235857 some cows are eating grass in a field
+000000146831 A person is riding a skateboard on a curved ramp.
+000000524742 a picture of graffiti on a subway train.
+000000429281 a table full of bananas being sold outside
+000000034205 Extreme close up of a cooked chicken, mushroom, and broccoli dish.
+000000355240 Small dog sitting on display behind the window. 
+000000211042 A cat pears into the bowl of a toilet.
+000000157756 A clock tower that is sitting in the middle of the street.
+000000408830 A row of foreign motorcycles is lined up in front of a wooden fence.
+000000558114 A basketball couch walking with his team on a court.
+000000129416 Several cows grazing on grass near water with trees in the background.
+000000441442 The person is riding a horse in the pen.
+000000156292 a wall has many different clocks on display
+000000027972 a surfer in a red top is surfing on a white and green board
+000000437331 A person on a surf board riding a wave.
+000000343976 A city street lit up in a night scene with cars in the background. 
+000000490936 two covered motorcycles in a row of motorcycles parked on the street
+000000486112 A man is spraying an elephant with a water hose.
+000000152686 a man is swinging a tennis racket on a court
+000000389109 A personal pizza on table in restaurant 
+000000089296 The baseball player is practicing his swing for his favorite game.
+000000240940 A cat sitting of a TV on top of a TV stand.
+000000494634 a cat that is looking into a computer screen
+000000401862 people standing in line beside a food truck
+000000407650 A young person jumping their snow board in the air as they ski down a mountain.
+000000359833 A young boy reaching up to grab a red apple.
+000000544565 A plate filled with pastry on a table.
+000000137106 it looks like a hand of a small child with pink and white top
+000000443498 A bus parks behind a cone in a large building. 
+000000108495 A girl is sitting on a skateboard outside
+000000311190 A dog that is in a car with a hat.
+000000273420 A room with a man in it talking on his phone.
+000000039480 a woman walking on a tennis court holding a tennis racket.
+000000366178 A fire hydrant and sign buried in the snow.  
+000000176634 A couple of zebra standing on top of a rock.
+000000457078 The inside of a room with a large collection of CD's.
+000000183716 Two small girls wearing blue jackets pose for a photo.
+000000309938 The brown fuzzy dog has round blue eyes.
+000000581100 Two giraffes looking at the other animals in the enclosure. 
+000000453860 Two different brands but similar looking devices sitting by each other.
+000000339823 A woman smiling and holding an umbrella high
+000000192191 A frozen pizza in the oven already cooked
+000000252716 A frisbee that is laying in the grass near a tree.
+000000538067 a hilly grassy area filled with people sitting down and taking pictures
+000000489611 A boy in grey shirt holding a Nintendo Wii controller.
+000000329455 Some people sitting at a table with a large oval shaped pizza.
+000000146489 Pizza on a cutting board is next to goblets with a beer bottle.
+000000085157 A group of people enjoying a meat at a restaurant.
+000000560011 An office desk with a computer keyboard and mouse.
+000000581482 A clock in a brightly lit room with windows.
+000000007386 a home made three wheeled trike sitting in a driveway
+000000249180 Two men walking opposite directions on a staircase.
+000000016228 A horse pulling a wagon with a conductor down the road. 
+000000470952 A man with skis and poles is on snow.
+000000181796 Fancy food on a plate on a wooden table.
+000000447611 A couple of people holding knives to the bottom of a laptop.
+000000441442 a girl in full riding gear riding a brown horse
+000000172330 A grey and white cat standing in the grass in a parking lot. 
+000000378139 Covered canal tour boat in Europe on a sunny day
+000000251065 A toilet bowl with a black toilet seat cover
+000000008899 A red brick building with a graffiti covered front.
+000000106912 A young boy riding a skateboard across a parking lot.
+000000146489 Half eaten pizza on cutting board next to alcohol.
+000000553788 Someone sitting on their bed looking at their laptop
+000000494913 A person sitting a the couch in a living room. 
+000000018193 A woman in green sweater eating a hotdog by campfire.
+000000079031 A kid on a surboard riding a little wave
+000000404601 A Cave sign is next to a restaurant and lounge.
+000000289222 The light shows safe for animal crossing on the street.
+000000489046 a pretty bird sitting on a branch among some leaves 
+000000155341 A car and a large truck on a city street.
+000000142238 A group of people are holding other people on a soccer field.
+000000560911 a person sitting down with a cell phone
+000000144300 a motorcycle with a boot on the back wheel
+000000420281 A man wearing glasses eating a hot dog.
+000000129492 Siblings playing video games on the floor at Dad's feet
+000000085329 A young brunette woman with multiple face piercings.
+000000511453 A prepared plate of food has pizza and french fries.
+000000505169 A toilet next to a chain shower stall in a bathroom.
+000000486479 A white dog is peeking out from under a hanging cloth.
+000000446703 A teddy bear inside of a wooden frame. 
+000000511599 Several people are standing outside a large boat. 
+000000554579 A group of people are standing next to a dog.
+000000193494 A bed in a ten with a colorful ceiling.
+000000138639 People in cars, on bikes, and walking near city buildings.
+000000468505 A man is surrounded by fast food at a restaurant booth.
+000000329447 A sheepdog runs around a herd of cows in a fenced-in field.
+000000181542 A woman in white crosses the road against the flow of motorcycle traffic.
+000000351096 A giant pair of scissors sitting on top of camera boxes.
+000000141328 a large bowl full of pasta with many other foods
+000000156924 A man and woman playing games on the Nintendo Wii.
+000000044260 Crisp, delicious apples hang on the dying branches of the tree.
+000000319721 The woman stands next to the large brown horse.
+000000445834 Men climbing on board a bus that needs to be repaired.
+000000009769 THERE IS A RED TRUCK THAT IS ON THE STREET 
+000000241326 A cat sitting on a couch with a dog on the floor.
+000000409268 A stuffed teddy bear covered in dirt on the ground.
+000000370375 A small boy holding a bat with a hat pulled over his face.
+000000564336 there is a man sitting at a table and using a lap top
+000000384949 A bridge with a clock tower at one side and a ferris wheel at the other side.
+000000366884 an image of a dog sitting on a sitting 
+000000297830 A dirty dog sits on the front patio of a home. 
+000000132116 Two bowls of stir fried beef and broccoli with chopsticks next to the bowls.
+000000019402 Two people with heavy winter coats are eating
+000000363875 The guy is travelling on his motorcycle outside.
+000000226058 Two stuffed teddy bears are placed on a couch together.  
+000000132375 A drink in a mason jar sitting beside a vase of pink flowers.
+000000161642 A very ornately decorated and brightly colored clock.
+000000509451 A group of people seated on a bench posing for a photo
+000000058111 A lone cat lounging on top of a counter.
+000000007784 This is a large kite flying high in the sky.
+000000122969 Two Zebra's standing very close to each other.
+000000523175 A spoon resting on the edge of a bowl of soup.
+000000133087 Downtown seems is really quiet for the night night
+000000036844 A spacious living room with three windows and a wooden floor.
+000000066523 A bed with white and gray bedding and netting overhead.
+000000344611 A yellow hazard sign sitting on the side of a road.
+000000050006 Looking out over a bay with many tourist boats moored
+000000008021 People watching an on screen presentation of a gentleman in a suit.
+000000039405 Spectators watching as competitors play a double tennis game.
+000000040036 Rider on a horse jumping over a fence
+000000568195 A man standing next to a woman near a cake.
+000000542073 A street light in front of a store front window.
+000000057672  a river.A group of people with life jacket rowing a long boat in
+000000491008 A waitress and restaurant owner showing off a pizza in front of a mural of Venice.
+000000293390 A vanity stink area with a marble counter top and wooden cabinets.
+000000416745 An open bathroom door reveals a toilet, sink and wall cabinets.
+000000182441 A bunch of birds flying around a couple of waves near the ocean.
+000000546829 A white dog is sitting on a wooden bench
+000000458702 A parking meter near the side of the road.
+000000320232 Buildings with large signs "Park Here" with rainy cement
+000000442993 A row of attached buildings situated by a river.
+000000386457 Cat wedged between two large oak doors trying to get out.
+000000559547 An old time photo of three baseball players.
+000000033005 A tennis player getting ready to his a tennis ball.
+000000217614 A giraffe standing in the middle of  a lush green field.
+000000414133 A sign is shown on the side of a road.
+000000492758 A living room with windows, chairs, table, lamps, and a person sitting in a chair with a walker beside it.
+000000198805 An old yellow truck in a parking lot.
+000000146489 wooden cutting board with a slice of pizza
+000000359781 A giraffe in a field looking down at something.
+000000333402 A man standing next to a stage coach bus parked at a bus stop.
+000000315219 There is a white cord hanging down a wall.
+000000336658 A city skyline after the sun has set. 
+000000303566 An outdoor market has a wide variety of fruit to choose from.
+000000341681 A man with a tennis racket is in the air with legs spread apart next to a tennis ball in motion.
+000000465180 People riding and washing elephants in a stream.
+000000054967 a couple of cars are driving down a city street
+000000132375 A vase filled with pink flowers on top of a table.
+000000185473 Men on horses are driving many other horses in a group.
+000000484760 A statue that is in front of a building.
+000000278848 People are waiting for the bus with umbrellas. 
+000000046497 Children are sitting on the side of the boat in the water.
+000000539962 A pool with two people inside is shown.
+000000469174 A plane flying in the sky near an American flag.
+000000442306 A young child who is holding an umbrella.
+000000009448 a young girl stands with her arms wrapped around a large blue umbrella
+000000396518 A women who is rowing on a boat.
+000000186980 A table sitting in a kitchen area with a fridge
+000000005001 A man uses a pair of big scissors to cut a pink ribbon.
+000000375078 A brown and white vase with foliage on a small table.
+000000338718 The Norfolk and Cambridge road signs have fallen to the ground while a van drives by. 
+000000563349 a person using a parachute with a wake board
+000000552371 Two horses stand outside in their corral, with mountains in the background.
+000000217060 A propeller plane that is flying in the air.
+000000158945 Seats placed on top of an elephant that is standing outside. 
+000000499181 a parking lot with a few cars in it
+000000044260 An apple tree is shown in the middle of an orchard.
+000000342295 a bathtub with several large paintings in it
+000000547816 a tub and toilet in a small bathroom
+000000322959 An assortment of meats and cheeses with bread.
+000000022192 A dog is sitting on an unmade bed with pillows.
+000000415882 A woman laying in bed next to a window.
+000000503841 two traffic lights in front of a huge building 
+000000029187 A person in a buggy drawn by a horse.
+000000253433 The teddy bear is all chained up on the bed
+000000532481 A person kitesurfing over the waves of the ocean's shore.
+000000079229 A man rides a horse and his dog follows.
+000000562818 a couple of people skiing on a snowy slope of ground
+000000433774 A dual parking meter near the grassy area of a residential area.
+000000521717 A woman standing on a tennis court holding a racquet.
+000000261036 A boy standing in the grass with a frisbee.
+000000507667 A map of the Boston subway printed on a statue of a cow.
+000000192191 A frozen pizza is coming out of an oven.
+000000236308 A couple of men in skies on a snowy slope
+000000551815 Two cats cuddle next to each other on a flowery blanket.
+000000137294 A gentleman is removing food from a cooker in a yard.
+000000435205 A person holding a teddy bear as she writes on her belly.
+000000090208 A green and two red double decker tour buses.
+000000409630 A wireless keyboard and mouse are sitting next to a laptop.
+000000104803 a bathroom with black walls and a toilet with a silver toilet seat
+000000022479 A person doing a trick on a skateboard
+000000245448 Men in neon vests are in a parking lot with a motorcycle.
+000000161799 a man on a skateboard performing a trick at a skate park
+000000415741 A house where you can see the bedroom and relaxing room.
+000000569825 A player in a baseball uniform on the field.
+000000569565 Two large trucks are travelling along a tree-lined roadway.
+000000422706 People watching a small boat being raised up.
+000000180878 A table with a happy birthday message holds several items that were probably birthday gifts.
+000000261097 Male tennis player on the middle of the court.
+000000038825 A pair of zebra graze near a barbed wire fence.
+000000339442 some fruit is next to a sandwich on a table
+000000085665 A blue kite is flying in the sky
+000000427500 a short fire hydrand sitting by the curb of a weedy street
+000000427655 Two ladies are riding in a wagon being pulled by a horse. 
+000000464522 A small dog with a chain collar looking at another dog.
+000000510329 Two men on old architecture pull out umbrella and securing their belongings.
+000000500613 A fire engine parked in a parking lot.
+000000570834 Bikes are being transported in a subway car.
+000000411754 A man checks his cell phone in a waiting area. 
+000000080274 A mama elephant and her baby inside a fence.
+000000233238 A group of people gathered around a table filled with food.
+000000460147 Traffic is traveling both ways on the busy streeet.
+000000296284 An assortment of doughnuts are arranged in a display case.
+000000533855 A white donut with a jumping rope to work it off.
+000000474452 A building near the sidewalk with a yellow curb and a no parking sign next to it
+000000055022 a row of bicycles for sale in a store
+000000259571 There are man lining the sides of the dumb truck.
+000000562207 Three tourists pose for a picture next to an elephant.
+000000520832 AA man walks down a city street next to a shop window.
+000000489924 A girl is skate boarding on a short skateboard.
+000000376900 Spectators watching a man swinging at a tennis ball.
+000000272136 A military plane parked in a hanger with a person sitting nearby.
+000000319935 There is a living room with a wide open view
+000000356427 A man standing next to train tracks with bags of luggage.
+000000227985 The white plate contains a meat sandwich on toast, garnished with lettuce, onions, and tomato.
+000000273712 A black refrigerator in a newly decorated house.
+000000260266 A street that has traffic lights that are very high up.
+000000395388 People looking at sheep in a series of indoor pens.
+000000054123 The zebras are standing around and grazing in the pasture.
+000000070739 Two women who are holding wine glasses. 
+000000380711 A male walking out of the water holding a surf board.
+000000068933 a few zebra are out in the open eating some hay
+000000454661 Cars bumper to bumper stuck in city traffic.
+000000269682 an intersection with different poles filled with street lights and a camera
+000000215245 A large group of zebra standing in the grass.
+000000015517 A city road with buses and bus park
+000000105912 A red fire hydrant sitting on a sidewalk.
+000000059635 a black and white photo of people with surf boards
+000000364557 Two surfers walk down the beach holding their boards.
+000000533536 a cat intently watching something that caught its interest  on tv
+000000213086 There is a man opening the oven door.
+000000519688 A woman in blue jacket holding a black bear cub.
+000000007991 A plate on a table is filled with carrots and beans. 
+000000355677 A cordoned swimming area with warning sign and lifeguarding equipment.
+000000251140 THERE IS AN IMAGE OF A PHOTO OF A MAN AND A BIKE 
+000000577584 A tray extends from below a microwave where a dish sits.
+000000334006 Garbage truck parked in back alley of large office building.
+000000574810 An orange cat perched in a window watching outside.
+000000120420 A guy in a green shirt holding his cell phone to take a picture.
+000000179141 Four people are smiling together over a box of pizza.
+000000047112 A pizza sitting on a plate in front of a glass of wine.
+000000568690 A small cat sits on top of a white toilet missing its seat in a bathroom.
+000000466156 A gray cat is walking next to a truck.
+000000119911 an image of a person on skates doing tricks
+000000052507 A man holding a yellow surfboard walking into the ocean.
+000000331280 White show horses and handlers performing during public event.
+000000269121 A guy in the middle of serving on a tennis court.
+000000007281 Two people on horses on a beach readying for a picture
+000000551439 A woman that is holding a book sitting on a bed.
+000000166747 Group of people standing on a train behind a white horse.
+000000572388 A toy smiley face in the middle of a doughnut. 
+000000110211 Zebras share a grassland with wild pigs in Africa.
+000000152771 A bike in the middle of an empty parking lot next to trees
+000000509008 A bus is painted half pink and half yellow.
+000000360661 Jockeys races horses on the beach on a clear day.
+000000224200 A dog laying on the ground its leash tied to a fire hydrant.
+000000080666 A cat sitting on a bench in front of a house
+000000142472 there are many houses and stores on this street
+000000185890 Two barefoot women holding game controllers in each hand.
+000000446207 A Sony camera and a cell phone are on a table.
+000000151516 An eagle flies low over the open water.
+000000579321 a dog wearing a scarf and shirt on a leash
+000000157928 Two men with backpacks on, one of which is fastening skis to a sled.
+000000225757 A computer desk with a laptop, desktop monitors and a television.
+000000552371 Some horses standing near each other in a fenced in area
+000000442463 A young kid in elbow and knee pads riding a skateboard.
+000000215778 a laptop on the top of a wood desk with a bookcase
+000000177489 two people sitting at a table with plates of food
+000000031817 A man on a motor scooter waiting in a group of motor scooters.
+000000147498 A beach area with black birds flying over it.
+000000131938 A cat standing on a bed and wearing a tie.
+000000347370 A person that is holding a lime in their hand.
+000000419312 A table with a plate of dinner and something to drink. 
+000000356612 Several black bulls are walking down the street.
+000000122166 a person riding on a bicycle on the street in front of some cars
+000000412362 A man with a nametag in a suit and tie and two women holding glasses on each side of him.
+000000549055 A man riding a wave on top of a surfboard.
+000000006818 a couple of buckets in a white room
+000000258388 A man ignites spray from a can next to his friend on a skateboard
+000000050380 A man guides a pony with a young boy on it's back.
+000000522638 A stuffed teddy bear and monkey share the picture.
+000000079651 A variety of beer sitting behind some bananas.
+000000051309 Three large horses eating hay while a small horse stands behind.
+000000460333 A little boy sits on a bed with a chair next to it.
+000000080057 A teddy bear is sitting near to a toy with pegs.
+000000448076 A group of people wait in a red reception area.
+000000351823 a tennis player standing and looking across the court 
+000000458992 The woman has a knife and fork for eating the big pizza.
+000000165518 We are looking at a motorcycle riding through the fox.
+000000367680 Several vehicles and a horse drawn cart pull up outside of a building.
+000000473015 A duck is floating in the pond next to muddy mounds in the water.
+000000255749 A bus stopped at a curb to allow people to board the bus. 
+000000252776 a man glares as other people behind him look on
+000000480944 An intersection with traffic signals and several street signs.
+000000076625 A blue and yellow train passing by people and building.
+000000324158 A man on a skateboard with a dog outside. 
+000000068933 A trio of zebra's leaned over eating hay on the ground.
+000000562243 A man with glasses and wearing a tie poses for a profile picture.
+000000369310 A boy wearing a helmet is playing video games.
+000000559707 A plate of broccoli, ham, pasta and parmesan.
+000000416104 many people sitting at different tables under a tent
+000000502910 Portable sign stating Wegomlegging which is Detour in Dutch
+000000147518 A bathroom that has magazine rack and small cabinet.
+000000299553 A lot of people on a hillside watching hot air balloons.
+000000526728 Two people smile while posing behind luggage suitcases on the sidewalk. 
+000000540932 People enjoying the beach on a nice day.
+000000412887 A cow is laying down with mouth open as if to say something.
+000000223959 A man holding a tennis racquet on a tennis court.
+000000228942 A car on a city street with a moving train directly behind it.
+000000504415 A man playing tennis on a court in a neighborhood.
+000000225670 a dog is jumping in the air toward a disk
+000000002153 Batter preparing to swing at pitch during major game.
+000000242678 The horse and rider are jumping over the yellow structure. 
+000000363072 this trains station has a white building in the background
+000000278006 A couple of green street signs sitting in front of a gray building.
+000000378454 a man is playing with a green flying disk
+000000032861 A man holding an umbrella in a hallway.
+000000333956 The ocean sits near a sidewalk and viewing contraption. 
+000000377575 An adorable grey poodle standing next to a fire hydrant.
+000000228981 A smart phone, digital camera and mobile phone on display.
+000000032038 A large cheese pizza being served to customers.
+000000292415 The setting sun silhouettes a skateboarder turning at the top of the ramp.
+000000516173 A woman in a wet suit carrying a surfboard into the ocean.
+000000343706 A woman has crocheted case for her iPhone. 
+000000447789 A street sign near a large leafy tree.
+000000034417 View through window of donut being made in kitchen assembly line.
+000000091406 Four women stand behind boxes of large pizzas. 
+000000512929 Two young girls cooking some fresh asparagus on a stove.
+000000394328 A bathroom with a red wall and poster in front of a toilet.
+000000052462 Two zebras standing in a field of tall grass.
+000000351530 There is a long line of wooden chairs and two women sit. 
+000000509260 A old television set is displaying an old computer game in front of two bookshelves.
+000000511076 there is a man sitting around many birds
+000000542776 A little finch that landed on the edge of a book. 
+000000032735 A person on skis making a cross in the air
+000000331569 A fruit smoothie is on display on a plate with two strawberries for garnish.
+000000391290 One girl trying to block another girl from catching a Frisbee.
+000000299720 A camel or a giraffe is playing with the camera-man. 
+000000153343 big statues of two beers dressed in clothes, outside
+000000568710 A rustic fireplace along a wall in a living room. 
+000000181499 a desk with a laptop and holding a disc
+000000431727 Two bears are perched on trees out in the fog. 
+000000145781 A black dog laying on the floor sitting alcohol from a bottle.
+000000392933 A giraffe stands at the edge of some trees. 
+000000220858 A boat is on the beach while two men dig in the sand in the distance.
+000000498747 A man at a party talking on a cell phone.
+000000350122 A busy bus station with ramp going downstairs.
+000000509131 a group of bananas are resting on a stand 
+000000489014 A dog stands on the bow of a white sailboat in the water.
+000000375430 A plate of pancakes sitting on a plate on a table.
+000000015751 An insect sitting on top of a wooden deck.
+000000356261 white horse with saddle on overlooks a river
+000000248980 We are looking at a plate full of salad.
+000000538458 two people standing on a paved sidewalk with skateboards
+000000290619 A banana and a container with a lid on a counter.
+000000322864 A picture of an animal is on a pole and next to it is a yellow taxi.
+000000562207 Three people stand with an elephant in front of a stream.
+000000233370 A person lying under a large black suitcase near a block wall.
+000000299720 A giraffe has its head pressed against the wall. 
+000000002923 Several ships in the distance behind a grassy marsh with egrets
+000000439593 A passenger train pulls into a train station.
+000000409475 Two people on skis in front of ski lift.
+000000079651 A row of beer bottles and bananas on a wooden counter.
+000000081061 A couch, chair, and suitcase sitting on the curb.
+000000124659 A living room with a large green couch and office desk.
+000000328959 A bathroom with a toilet, shower and an hanging chain to flush the toilet. 
+000000316666 A vase with flowers next to a mirror.
+000000432898 The man is flying his kite in the sky. 
+000000529762 a man sittig by a table taking of photo of his food 
+000000402765 A white Wii component with two buttons and four lights.
+000000243495 There are two rolls of toilet paper and a hand dispenser on the bathroom wall.
+000000058384 A herd of elephants standing next to each other.
+000000496409 The skier is coming down the slope on a very snowy day.
+000000353970 A gray and white cat looking up from a chair.
+000000514797 A couple of men reaching up towards a blue kite.
+000000488166 The man stands on fence railing beside a table with food on it. 
+000000134034 a horse and a donkey eating grass in a fenced in pasture 
+000000277020 Asian businessman in a restaurant with one of them being served. 
+000000551439 A woman holding a red book in her hand while sitting on a bed. 
+000000132622 A large brown bear walking across a grassy field.
+000000201072 A man that is on a tennis court with a racquet.
+000000263474 A bathroom toilet seat lid is highly decorated.
+000000479248 An entertainment center with a tv, movies, and decor.
+000000290768 A kitchen with sunshine shining off of black and white tiles
+000000365098 An empty boat in the water near a tree
+000000569972 a wake boarder setting up his raft in the ocean
+000000459757 A giraffe and some deer like animals at a wildlife park.
+000000193245 A group of people flying kites by the river.
+000000161781 A woman is skiing down a hill with safety netting.
+000000034873 a black counter a sink and a table and two chairs
+000000451144 Two skiers looking out at the view from a high cliff 
+000000232684 A man uses a laptop in an otherwise dark room.
+000000442161 Three individuals are at a restaurant and drinking wine.
+000000349837 The refrigerators are lined up against the wall.
+000000244750 A table topped with three wine glasses and a laptop.
+000000293390 A VERY NICE BATHROOM IN THE MIDDLE OF A REMODEL
+000000131556 a person riding skis on a rail 
+000000534673 A blue and white bus with large windows on it.
+000000553669 A man is sitting on a park bench using a computer.
+000000117914 A toilet and towel rack are in a bathroom.
+000000258883 A person cooking some food on a grill.
+000000546659 A freight train passes by drab industrial buildings.
+000000562843 A cellphone and scissors are next to a coffee mug.
+000000118515 Small kitten sitting on top of a wooden crate. 
+000000243495 A toilet in a small room with a two-roll dispenser for toilet paper.
+000000392481 A military jet taking off into the sky 
+000000509824 a couch in a living room with three pillows
+000000087038 Athletes performing tricks on a BMX bicycle and a skateboard.
+000000540928 A cat is walking past a chair. 
+000000047010 Three giraffes with heads pointing ahead walk on a dirt path.
+000000293474 A yellow and black frisbee sitting on a  sidewalk.
+000000327701 A mother and her children play on a grass tennis court
+000000532855 A man on a skateboard grinding on a pole
+000000023126 A man is standing on top of two horses and holding the reigns.
+000000467776 A cow and calf are standing in a grassy pasture.
+000000205776 A large bear laying down on a leather hammock.
+000000174004 A yellow truck is sitting in high grass. 
+000000197022 There is an image of man doing things on the table. 
+000000391648 A white clock tower next to a brown building at night.
+000000176606 A goat stands on its high legs and leans against a grassy bank.
+000000056288 A large half a sandwich sitting on a plate next to a pepperocini
+000000437514 a close up of a bench near a potted flower
+000000177714  A plate containing potatoes, carrots and chicken. 
+000000560178 The end of a carrot is shown after someone has been eating it.
+000000215723 Workers in fluorescent yellow jackets standing under a fluorescent orange umbrella.
+000000235836 A boy playing baseball about to hit a ball.
+000000377723 group of people waiting to get onto a bus in the city
+000000369310 A boy in helmet playing a game with a Nintendo Wii controller.
+000000367818 A pretty young woman is posing with her horse.
+000000157124 there are several different sinks of all different colors
+000000087476 A man teaching a boy how to ride a skateboard.
+000000481159 A horse in a snowy paddock looking over the fence
+000000239843 Two green street signs crossing each other above a stop sign.
+000000031296 a large group of people are sharing a meal at a restaurant
+000000157138 A peice of cake sit on a red plate with a fork beside it.
+000000281447 A tan horse with a blond mane inside of a coral.
+000000078565 We see a very old picture of people enjoying the shore.
+000000158227 A child has a baseball helmet and a bat.
+000000476770 A baseball player reacts to the play during a baseball game.
+000000321790 a man that is holding a tennis racket
+000000330396 A man and a woman watching waves at beach.
+000000159684 The red and white train is waiting at the station.
+000000261116 A table filled with a cake and paper plates with ice cream and cake.
+000000289343 A person riding a very tall bike in the street.
+000000101420 Cat perched on the sofa looking out of the window
+000000270066 A white and black motorcycle sits next to a van in a parking lot.
+000000256775 Three children fly a kite along the beach.
+000000195165 A bathroom sink sitting above a trash can.
+000000428280 Chairs and a table with a laptop on it sitting outside. 
+000000344059 Two giraffe standing in a field next to trees.
+000000211120 three teddy bears sitting on a picnic blankets 
+000000322429 a cabinet with vases pots bowls cups and other china 
+000000376856 A clock tower made of brick with vines growing up one side.
+000000009914 A sandwich on a sesame seed bun next to a pile of french fries and a cup of ketchup.
+000000436617 A bunch of people sitting at a table having a discussion. 
+000000570782 A desktop containing various computers missing their screens.
+000000186632 A fancy bathroom has glass sinks, a chandelier, and a TV.
+000000113354 Three zebras standing in the wild surrounded by trees.
+000000360137 A person leaning up against a metal rail while holding a rainbow colored umbrella.
+000000530457 A clear vase holding drooping orange and red flowers.
+000000547886 A group of people are riding horses on a beach.
+000000194832 A television screen is displayed in an empty bus.
+000000171382 A young man doing a flip on a skateboard in the middle of a busy street.
+000000297147 A silver and black sports motorcycle parked near a building.
+000000039956 a messy bed room a bed a chair and boxes
+000000048555 Horses with jockeys racing across the coast of some beach.
+000000325527 A small brown teddy bear sitting on top of a box.
+000000149770 A single surfer striking a pose after catching a wave.
+000000460683 A man that has the ten commandments on his tie
+000000281687 A woman with a hand bag on a cell phone next to another woman.
+000000506310 A refrigerator in a kitchen with its door opened.
+000000117492 a group of moms and kids sitting out in a field
+000000462904 A black horse is standing in the sand, with people in the background. 
+000000483050 The bed has been neatly made and the pillows arranged.
+000000557916 A girl sitting on a bed in a room.
+000000289393 A display of vintage animal toys on the floor.
+000000528399 Trays of pastries and sandwiches beside a bowl of soup.
+000000041633 A classic red truck is parked in the lot.
+000000068286 Crisp white bedding creates a stately environment when combined with a cherry ensemble.
+000000031749 A demonic looking chucky like doll standing next to a white clock.
+000000503841 two street lights that are both on the green light
+000000438774 a coupe of people are standing around a small table
+000000014380 A long bridge with cars has a train going underneath.
+000000486573 A clear vase of simple flowers is sitting in a windowsill.
+000000530854 Several different opened umbrellas all located near each other.
+000000509656 a zebra standing in the foreground  and another zebra standing behind
+000000328601 Tennis player getting ready to back hand the ball over the net.
+000000375763 The herd of sheep is standing on the grass.
+000000053529 A dog with a green hat sitting in a truck.
+000000127987 A desk with two computer screens and a Cecil Cooper bobble head.
+000000335427 A bowl filled with broccoli soup next to a spoon and fork.
+000000449909 A boy takes a picture of a seated giraffe
+000000377588 Pitcher in a baseball game tries to out a player attempting to steal base.
+000000480944 A traffic light sitting on the side of a road.
+000000271116 Two men sitting in chairs at a table with cell phones in their hands. 
+000000227399 A white and yellow train traveling down a road next to a red motorcycle.
+000000287291 A city street with buildings and cars on a rainy day.
+000000014439 A group of people standing on a field flying a colorful kite.
+000000360951 A computer workstation with a laptop and a desktop computer.
+000000480985 The front end of a red motorcycle that is on display.
+000000094871 a number of people standing near a number of sheep 
+000000361730 A large group of people flying kites beside a body of water
+000000189078 Several assorted mixed fruits and vegetables on a fruit plate.
+000000153568 A stop sign with additional warnings taped to it.
+000000532493 Surfer at the height of a large wave looks down. 
+000000418696 A tall church tower sitting on the corner of a street.
+000000344888 A couple of horses by a fence on a field.
+000000377113 A boy wearing  skating boats in water
+000000380711 There is a blurry photo of a surfer walking out of the water
+000000045090 A couple of men in the water on surfboards.
+000000434996 a cute little kitty laying among stuffed animals
+000000147415 a woman writing something down on paper while the laptop sits on the table 
+000000556498 The old fashioned light also serves as a street sign.
+000000500464 A sink, mirror and toiletries in a bathroom. 
+000000553664 Red laptop on a table showing you can learn anywhere!
+000000558854 One person is eating chilli and a grilled cheese and the other is having a sandwich and fries.
+000000394199 A woman is posing next to a red motorcycle.
+000000061471 Puppy chewing on toilet paper in the bathroom. 
+000000151657 A man and woman in costume like clothing standing together.
+000000400794 A man sitting at a table eating some food
+000000081988 Four kids sitting on surfboards with man in background.
+000000422836 A man is carrying a suitcase through the street
+000000125806 A couple of black cows standing next to a tree.
+000000482978 This is a multi colored train at this train station
+000000134722 a long passenger train pulling up to a station
+000000181542 A woman walking across a street near motorcycles.
+000000394206 A large group of men traveling on the back of a truck.
+000000537355 A long fire plug stands some distance away from the building.
+000000244379 View down a two lane road at a red stop signal.
+000000188296 a couple of football players in the air
+000000101787 A big commercial plane flying in the sky over a wire.
+000000321118 A stuffed bear head and paw on a laptop computer.
+000000008021 An audience is looking at an film of a man taking that is projected onto a wall. 
+000000105335 A person holding a plate with food 
+000000142472 some people some buildings cars and a red double decker bus
+000000172330 A cat in between two cars in a parking lot.
+000000343315 A red fire hydrant on a street corner near a sign
+000000348243 A chubby black pony in a pasture looking ahead.
+000000078170 two women sitting in two separate train cars.
+000000256941 A bicycle with a blue umbrella and baskets attached to carry things.
+000000546826 A pink and green marker, next to another object. 
+000000311909 A large bus parked by the curb 
+000000336658 The buses are parked next to the pole.
+000000263644 An odd bed sits near two separate windows.
+000000012667 A phone with a banana where the receiver should be.
+000000398742 Two men hold up a man holding a soccer ball and all the men have on orange shirts.
+000000166478 A man sits on the couch and is on laptop and phone.
+000000210789 A child and person stand under an umbrella. 
+000000358525 A very young boy is using a keyboard on a laptop computer.
+000000571857 two people outside sitting on chairs with planes around them.
+000000264535 An old man reads a book on a bench. 
+000000407298 A kid with a glove and a ball.
+000000550471 A sliced orange on a plate by a fork.
+000000410428 Curious sheep on a large, well used pasture.
+000000142324 A man rides with a child on a motorcycle down a busy street with other pedestrians and bicyclists
+000000284282 A blender is sitting next to a printer.
+000000308531 Palm trees are in the foreground of a picture with a clock tower standing above them.
+000000128675 A man holding one water ski at the lake.
+000000029675 A plate topped with three hot dogs on top of a table.
+000000001532 Many cars traveling on a busy road with exit signs overhead.
+000000355240 a dog is siting behind a large window
+000000448448 Two giraffes grazing from a tree in a field.
+000000109827 A brown teddy bear sitting up against a tree.
+000000373705 a skinny red fire hydrogen  by some people
+000000149770 A man riding a wave on top of a surfboard.
+000000057597 A soccer game where a girl in white is dribbling ahead of the girl in blue.
+000000041633 An old model red Chevrolet truck is on the pavement.
+000000378605 A white plate with a donut and napkin sculpture.
+000000568981 a man that is taking a picture of a little boy
+000000571264 A black and white photo of a guy and his motorcycle.
+000000521259 The man is playing Frisbee with the young boy.
+000000513283 A small toy is sitting on a plate of pizza.
+000000486438 A person sitting on the floor near a plate of donuts.
+000000323496 A young man standing on a tennis court holding a racquet.
+000000503755 A woman holding a tennis racket standing on a tennis court.
+000000280325 A parking meter in front of a wall that has been written on. 
+000000214205 A large jet liner flying under the moon.
+000000256868 A man filming a person skateboarding on a wall.
+000000384666 A group of young children riding skis down a snow covered mountain.
+000000286908 Several different pictures of various types of food.
+000000001761 Two planes flying in the sky over a bridge.
+000000421923 A room witb a statue, bookshelves, books and a vase with flowers in it.
+000000421834 A bird sits on a seed filled bird feeder.
+000000073702 A man with an umbrella hat stands next to another man.
+000000532901 A couch sitting in a living room in front of a fire place.
+000000227491 A woman is sitting on a bench and is eating something.
+000000117914 A restroom with a toilet, sink and window
+000000004134 Two men shaking hands after a dinner speech.
+000000149622 Two owls sitting together on a tree branch.
+000000475064 A person is holding up a half eaten pastry.
+000000024919 Two elephants that are pressing their heads together.
+000000177383 a yellow and red trains engine on its track 
+000000504415 A person on a court near a fence.
+000000003934 A young girl playing a video game while others talk.
+000000509824 A living room with a cream colored couch
+000000080666 A cat sitting on a bench in front of a building.
+000000350054 A cat sitting in front of a flat screen TV.
+000000458410 a family room with a couch and a tv
+000000500477 A person sheering the wool from a sheep.
+000000188465 a baseball game with a batter swinging the bat and fielder awaiting the ball
+000000436315 An oddly shaped pizza sitting on a table
+000000199442 A man riding a wave on top of a surfboard.
+000000150417 An old lady sitting at a table while looking at her birthday cake. 
+000000439854 Skateboard rider doing a trick near a beach during sunset.
+000000502336 Pedestrians walking on cobbled sidewalk past multiple storefronts..
+000000153527 Tools are lying in a road on pieces of thick wood. 
+000000161642 A coo coo clock mounted to the side of a wall.
+000000257478 a close up of a baseball player holding a glove on the field
+000000299553 A crowd of people flying kites in a blue cloudy sky.
+000000014831 Close up of some fuzzy animal laying on the bed
+000000453708 Person carrying green bananas on their shoulder and a market. 
+000000311002 A batter at the plate during a baseball game.
+000000005992 A group of shaggy haired sheep standing beside a building.
+000000520301 A brown medium sized dog stares out of a window.
+000000116589 The group of zebra are eating hay together.
+000000488270 A white airplane is driving down the runway.
+000000506933 a man holding a cell phone that is powered on 
+000000276018 A group of children are holding various stuffed animals and dolls.
+000000489014 A dog on a boat with a person in the water.
+000000063965 A close up of a person holding a piece of cake.
+000000343076 a cat lays down on a dvd player 
+000000465675 A beach with two boats and a tractor.
+000000232649 a woman taking a picture in a bathroom mirror and a toilet and sink
+000000039956 A messy room with door and bed and chair.
+000000186929 A plate topped with sliced up oranges full of juice.
+000000382734 Unfinished bathroom with only a shower base installed
+000000479126 Two girls are smoking cigarettes while one is playing on the laptop.
+000000263679 A man standing on top of a sandy beach near a colorful kite.
+000000579091 Small plate of food which includes small pieces of broccoli
+000000474078 a baseball player bunting at a baseball on a field
+000000263796 A red toilet in a very small bathroom.
+000000512648 Cow standing under the only tree to catch some shade
+000000213593 Street sign displayed on pole at intersection in metropolitan region.
+000000110999 Some art work with a man with a hat on and some fruit in a bowl.
+000000185472 A group of passengers at a train station standing on the platform as a train arrives
+000000095843 A double decker bus parked near a curb by trees. 
+000000050638 A young boy walking on a path lined with flowers, with a teddy bear.
+000000126110 A dog driving a yellow big rig truck next to a man.
+000000028449 a family of elephants standing together in the jungle
+000000201072 A tennis player prepares to react during a match.
+000000558558 A traffic light on a street in a dark city.
+000000064574 A set of women's personal care items sitting on a bed.
+000000205514 A room with all white furniture in front of a flat screen T.V. 
+000000194216 A family of bears walking through a forest with lots of trees.
+000000410880 Large stuffed bear sitting outside of a building.
+000000273642 a dog is laying with a remote controller
+000000001268 A beautiful woman taking a picture with her smart phone.
+000000304560 A black ct with green eyes sitting in the sun.
+000000378605 A table with a napkin, doughnut on a plate, and a cup of coffee.
+000000154213 There is a pie crust inside of a pie ring
+000000551794 A motorcyclist is pulled to the side of the road.
+000000250619 A woman laying on the beach under an umbrella.
+000000184321 A passenger train pulls into a train station.
+000000427160 A boy in a blue baseball uniform running to tag out a boy in a red uniform.
+000000251119 The lunchbox has a cold sandwich, strawberry yogurt and orange juice. 
+000000279769 A yellow ripe banana sitting on top of  a wooden table.
+000000071756 Two bears walking together in the night. 
+000000232538 A train is pulling into the train station.
+000000549738 A huge dragon kite flown with a steeple in the background.
+000000083531 a number of hot dogs cooking on a grill
+000000386210 the fruits are all in a container in the window.
+000000458768 A kitchen filled with lots of white counter space.
+000000295138 A young man on a skate board riding a ramp.
+000000204329 A surf board sticking up in the sand.
+000000250205 Two seagulls stand on the rocks by the shore.
+000000384527 The room is crowded with many things including chairs, a bicycle, and a table with cups on it.
+000000371529 A woman standing in front of a white toilet.
+000000500049 The airplane is on the tarmac at the airport.
+000000048153 A woman showing off her pink pointy-toed heels.
+000000013923 a room that has some furniture and a table in it
+000000080340 A boy in a tie holding a small suit case.
+000000067315 A close-up picture of a man wearing a tie.
+000000165831 A foil pan is full of orange type food.
+000000246963 motor cycle riders stopping at a stop sign
+000000395575 a bench sits between a pond and the tree line
+000000511076 a person sitting on a bench near many birds 
+000000078404 A group of women sitting on a full bench.
+000000342295 Someone put their art collection in their bath tub.
+000000388903 A woman talking on the phone in front of a person selling apples.
+000000158660 bananas and apples sitting next to each other on a counter
+000000236845 there is a street sign that has been written on
+000000234526 Kids feeding a giraffe at a zoo with plants.
+000000313182 A purple and white bus driving down a street.
+000000195045 An empty field with older structure in the background.
+000000000139 A room with chairs, a table, and a woman in it.
+000000499031 The three women are posing for pictures. 
+000000054592 Couple of people walking up the snowy mountain with skis
+000000427338 Person on motorcycle in a very scenic rock area.
+000000267940 A ferry the is pulling into the dock for off loading. 
+000000485972 A blue and white plate with a chocolate dessert on the plate and powdered sugar on top.
+000000252701 a man on a blue surfboard on top of some rough water
+000000372819 A group of dogs running across a lush green field.
+000000096493 a kid is wearing a blanket on a coach
+000000011511 A metal statue of two women sits on a bench in a city street.
+000000499266 An electronic device is being held as it displays a message and an image.
+000000147729 some people sitting and one man looking at his cellphone
+000000428111 A skateboarder flies through the air in a skate park.
+000000424975 a building with so many arts on it
+000000149406 An old motorcycle rests near a rundown building. 
+000000079588 A very old building has a large clock tower.
+000000290163 a little girl in a blue apron and chefs hat and a girl in a red apron and hat
+000000033707 Two large giraffes inside a fenced area stand together near many rocks
+000000355240 A chihuahua sits in a window display with a price tag.
+000000236914 two teddy bears at a table with honey
+000000192607 A man stands in front of a pickup truck.
+000000535156 A large elephant rests on a black pole.
+000000001425 part of a sandwich sitting on a table
+000000055072 An adult giraffe grazing in a brown field.
+000000163314 People are walking in front of some shops.
+000000193926 Four different photos of bananas and banana holders.
+000000376206 A man on a surfboard  surfing a nice big wave
+000000223789 a rome with rose flowers and a sink 
+000000083531 Several rows of cooking hot dogs on a grill.
+000000312192 A plate filled with food sitting on a table next to a drink.
+000000343315 A red fire hydrant sitting on a patch of green grass.
+000000121417 a woman is sitting with an umbrella outside
+000000098520 A plane siting on a showy tarmac at an airport
+000000221872 A tray with carrots, snap beans, mash potatoes and an egg.
+000000255965 A black cat with white stomach takes a pose.
+000000568195 The couple are getting ready to cut the cake. 
+000000515982 A man sliding into a base next to another baseball player.
+000000370813 a man does a trick on his skate board
+000000283038 A red stop sign sitting on top of a green planter.
+000000063602 A black laptop at a desk with a printer.
+000000146825 a train covered in black dirt sitting in a fancy train station
+000000150638 Outdoor impromptu bar scene on wooden bench, ice coolers, blender, bottles and several other items, appearing very crowded. 
+000000549674 Two computer monitors are turned on on a computer desk.
+000000193674 A woman that is next to a surfboard with a dog.
+000000165713 a fire hydrant near two small poles on the curb of a street
+000000521819 A group of youngsters contend for a frisbee in a park-like setting.
+000000313588 The woman holding an umbrella smiles on the narrow street beside the sidewalk.
+000000130699 People playing a game on green grass in front of a building.
+000000395388 People at a stock yard looking at animals 
+000000491366 A couple of men in front of a television with a remote.
+000000329319 A cat sitting on a chair looking down.
+000000413404 a park with benches and lots of trees.
+000000054931 An attractive young woman leads a grey horse through a paddock.
+000000064462 Female skier smiling while standing in the snow. 
+000000465718 A desk has two computer monitors, a keyboard, and a laptop that is all connected.
+000000044590 A motorcycle club stopping along the highway to take a break and talk.
+000000241319 A bathroom sink with two toothbrushes facing each other.
+000000524108 A street with many motor bikes parked together.
+000000329542 A picture of a person touching a cupcake.
+000000378673 A skateboarder is separated from his board as he performs a jump in front of a crowd.
+000000101068 The man is hitting the water balloon with a bat.
+000000288685 Many people watch a man and dog herd sheep.
+000000096825 A man with skis and ski poles is standing next to a hill covered in snow
+000000142092 An overhead view of a pizza on a pizza stone.
+000000221693 A husky dog has an orange frisbee in it's mouth.
+000000109055 Cat sitting on the ground between bikes showing its teeth.
+000000545129 A group of zebra standing next to each other on a dirt field.
+000000148707 Two yellow fruits hanging on branches full of leaves. 
+000000397351 A man holding up a bunch of root vegetables for sale.
+000000104455 A couple of zebra eating a small pile of hay.
+000000264335 A bird is flying against a blue sky.
+000000532493 A surfer on a white board riding a small wave.
+000000293794 Man at street market with table full of bananas.
+000000224093 several cows grazing in a grassy  open field.
+000000241297 A train with bright yellow engine on tracks beside tall leafy trees.
+000000165257 a black counter on a wood cabinet in a kitchen
+000000423506 Two men in ties smile at the camera 
+000000400922 A bank sign that also has a clock attached to it.
+000000287874 A large blue and yellow train stops near a fairly-non urban train-station.
+000000568690 A cat sitting on a toilet in a bathroom 
+000000095069 a couple of towers on top of a building
+000000336209 A person that is doing a trick on a skateboard.
+000000182202 A cellphone next to a laptop computer. 
+000000405205 A passenger bus that is parked in a parking lot.
+000000369323 A girl in a tennis outfit getting ready to swing a tennis raquet on a tennis court.
+000000554735 A person is picking up a piece of pizza.
+000000421455 A person taking a picture with a cell phone of them and a dog in a mirror.
+000000291490 A cat laying down on a blue comforter with pink sheets.
+000000001490 Black and white of a person wearing a wetsuit standing on a surfboard and holding out a paddle, in a large body of water outside.
+000000462576 Breakfast items including juice are on the table.
+000000512776 A catcher with a catcher's mitt on squatting behind a baseball player.
+000000367386 A living room with sofas, tables, a lamp and windows.
+000000182805 A woman holding an umbrella sitting on a blanket next to two dogs.
+000000546475 A man holding a baby taking a picture.
+000000453166 two soldiers in uniforms holding plaques in front of them.
+000000493772 A person is standing in the middle of a road while its raining.
+000000163057 People fly kites in a large park in the middle of a city.
+000000535608 A sandy beach topped with lots of lawn chairs and people.
+000000218362 A large digital clock mounted to a wall.
+000000311392 a dairy cow standing in a stall full of hay
+000000117744 A young girl in a white top poses with a yellow tennis racket.
+000000190307 A woman in a wet suit standing on a dolphins head
+000000101022 A sign on the side of a snowy road stating avalanche zone.
+000000581781 A fruit stand that has bananas, papaya, and plantains.  
+000000575815 A pepperoni pizza sitting on top of a wooden cutting board.
+000000170613 A couple of women are at the beach with surfboards
+000000464872 a couple of zebras that are running through some grass
+000000161861 A woman standing on a tennis court holding a racquet.
+000000348012 A pair of black handled scissors lying on a roll of tape.
+000000232088 View of a living room with a sloped ceiling over the windows.
+000000000724 A stop sign is mounted upside-down on it's post. 
+000000292330 A dog is standing and sniffing an abandoned ball bat.
+000000282296 An old refrigerator stands against a wood-panelled wall
+000000147205 Some food that is on a glass plate.
+000000127530 A man hitting a tennis ball on a professional court.
+000000054123 A herd of zebras is grazing on the plains in the sunshine.
+000000197658 A man getting ready to catch a baseball.
+000000007108 An elephant is seen head on amongst other elephants.
+000000246968 A woman standing in a  kitchen preparing food.
+000000228771 Two street signs side by side on a street corner.
+000000358195 A little girl wearing pajamas is riding on a skateboard.
+000000415716 The cabinet has a various assortment of decorations on it.
+000000106281 Two trains passing each other on seperate tacks.
+000000019042 A bird that is standing on a rock in the water.
+000000490470 A sunset view of a marina with boats docked.
+000000289741 two cruise ships docked near a beach resort
+000000437898 A kitchen area with a stove, cabinets and a refrigerator.
+000000122166 a woman rides a bike in front of some cars 
+000000171740 a living room with with a couch and a table 
+000000250619 A woman lying on her back with knees bent on a beach towel under a multicolored, striped beach umbrella, surrounded by sand, and with clear blue sky above.
+000000349152 an open box of doughnuts on a wooden table
+000000469067 A woman laying in bed next to a cat.
+000000338905 a group of people sitting around a long table while eating from plates of food 
+000000291490 A cat peeking it's head out of the blankets.
+000000244099 A PERSON IS RIDING A HORSE THAT IS RUNNING VERY FAST
+000000270883 Young woman erotically bound on top of bed.
+000000529105 A horse is laying down in a field.
+000000148739 The person in the bodysuit is surfing a wave.
+000000474293 A full view of a beautiful store in a town. 
+000000312278 A crate on top of a car holding various luggage.
+000000081594 A woman with an umbrella standing on a deck. 
+000000535094 Cow being fed a popsicle by a person over a fence.
+000000520264 A man and a woman stand next to each other in a kitchen.
+000000317024 A zebra standing on the ground inside a fence.
+000000237928 A small kitchen with stainless steel appliances and white cupboards.
+000000021903 A man reaching out to an elephants trunk near a gate.
+000000492937 A bicycle is chained up to a pole at a train station.
+000000348881 A large white airplane and a person on a lot.
+000000263463 A dog is shown in a car rear view mirror.
+000000000724 A stop sign that is hanging upside down.
+000000057238 A white refrigerator freezer sitting inside of a kitchen.
+000000350148 A man taking a picture of plates of food.
+000000375763 A very cute flock of sheep in a grassy field.
+000000377723 A group of people that are standing outside of a bus.
+000000064523 A black and white photograph of a female tennis club.
+000000581206 Two hot dogs garnishing with sesame seeds and green things
+000000287347 A pretty young lady making a pizza next to a man in a kitchen.
+000000118515 A spotted kitten sitting on a wooden bench.
+000000496854 A woman is holding an umbrella over a bride and groom.
+000000337498 A large slice of pizza with cheese and marinara sauce on a plate. 
+000000348481 A laptop, phone, keys and other accessories sitting on a table.
+000000439593 A closeup of a train at the station for people to board. 
+000000443303 A cat is laying in an open suitcase on top of some clothes.
+000000352582 Two men one getting ready to throw a freebie. 
+000000070774 A motorcycle is sitting outside near some plants.
+000000579900 an image of a small pizza on a plate
+000000556193 Two men both working on the same computer screen.
+000000322574 Food items served in small tray displayed on table.
+000000005503 Point of view of a man peeing in an airplane restroom.
+000000036494 This crowded restaurant has an enormous clock on the wall
+000000476810 A black and white cat laying on white blanket next to a remote.
+000000282046 A bench in the sand facing an empty beach.
+000000376322 People sitting on the long table with plates of food. 
+000000558558 A city traffic signal and street at night.
+000000339870 A church steeple is shown amidst telephone wires and lights. 
+000000530052 A bunch of carrots sitting on a table.
+000000508917 a train on a track pulling into a station 
+000000309452 A colorful bird sits on a tree branch. 
+000000279145 A white cat sitting before a bench surrounded by plants in planters.
+000000229849 A person leaping into the air while holding a tennis racquet.
+000000131379 A brown crust of pie with strawberry filing. 
+000000280710 People walking on the sidewalk of a city with a tour bus in the background.
+000000476119 A person wearing a motorcycle helmet is riding a skateboard.
+000000153632 Some interesting food sitting on a white plate by a knife
+000000250766 A dinner plate with a colorful salad and grilled meat.
+000000409358 A white toilet and sink in a room.
+000000263644 The antique bed has elaborate wood decoration on the frame..
+000000377393 A large clock tower sitting in the middle of a paved walkway.
+000000259625 An elephant walks around a fenced in area. 
+000000380913 A group of friends riding a subway while using cell phones.
+000000315219 The cord and warning label on a hair dryer.
+000000019432 a man playing tennis going for a low ball
+000000407614 The kitchen is coral with wooden cabinets and colorful decorations.
+000000402783 Several orange wedges sit on a white plate.
+000000097585 a white wall displaying art that look like vases
+000000092053 Three different meals on plates with silverware and beverages.
+000000498857 A giraffe eats leaves through a wire fence.
+000000129812 A fuzzy picture of a man in the middle of a skiing jump
+000000217872 Woman snow boarding off of a cliff in the air.
+000000462756 Motorcycle sitting on a dirt road in front of a farm house advertising produce.
+000000022935 A soccer player in uniform runs toward the ball.
+000000412286 A man swinging a tennis racquet towards a tennis ball.
+000000091654 A bowl of food with mushrooms in it.
+000000363784 A bathroom scene with a dirty tub, toilet and sink.
+000000364166 An adult and a baby zebra are on the savannah.
+000000189436 A carved bear that has a ribbon around the neck.
+000000157124 The kitchen is in the process of being remodeled. 
+000000256407 A white tray topped with lots of different coffee cups.
+000000176634 A zebra walking around looking for something to eat
+000000402720 three men sitting at  a table with some wine glasses
+000000485424 Various items sit on top of tables and counters.
+000000460347 City bus next to traffic cones in the far right lane of a busy freeway.
+000000360325 a close up of a lot of animals with a fence
+000000514508 A man in  a jacket standing near a  giant tv
+000000405970 A large living room with a wall painted green
+000000283113 A hot dog in a bun is topped with mustard.
+000000184384 A bluebery cake is on a plate and is topped with butter.
+000000407002 A person standing on skies in the snow on top of a mountain.
+000000358525 The little boy is having fun playing at the computer keybord
+000000390555 A group of young children sitting next to each other.
+000000267300 A dog eats people food from a plate on the floor next to a plaid bed.
+000000442993 The back row of buildings of residences at night.
+000000377393 View of plaza with tower near beach on sunny day.
+000000382111 two young men on horseback at a gas station near some gas pumps and trucks
+000000460333 a young boy sitting on a bed with a lamp on beside it
+000000229747 A photo of an outside with various things in the scene.
+000000084241 Two women are laughing in the kitchen while they cook.
+000000151657 A man standing next to a woman in a living room.
+000000001296 A woman holding a Hello Kitty phone on her hands.
+000000292488 A woman in a jacket with a hood looking at a cell phone .
+000000052007 The bus stopped at the bus stop and people are getting in.
+000000474786 A large mirror reflecting a bathroom sing counter.
+000000255536 4 chairs rest on the concrete outside in front of a fence.
+000000188439 a truck on a street with hay in the back
+000000506178 Blurry image of men in living room playing a stand-up video game.
+000000276707 A red and white traffic sign sitting in front of buildings.
+000000246883 The two surfers art talking on the beach.
+000000267351 A yellow and white clock tower against a blue sky
+000000428867 Green, yellow, and pink teddy bears sitting on top of a table on display. 
+000000449603 A surfer is para sailing along a large wave.
+000000173057 A large clock tower in front of a building next to a fountain.
+000000426203 Parkour while skating on a rail in urban
+000000315187 Traffic in front of Walgreens on a town street.
+000000152465 A rusted red fire hydrant encircled by an old tire stands in the bare earth beside a chain link fence.
+000000296317 A large grassy field with giraffes and a few other animals.
+000000530457 PIcked peach flowers sit in a vase with water.
+000000098497 A yellow sign that is at the top of a pole.
+000000447611 A computer laying upside down with two hands holding knives prying open a section of the back.
+000000474078 A baseball player swinging a bat a a baseball.
+000000303818 A curve in a city road with a bus following a car and people walking on the side.
+000000329041 Several pieces of luggage are on display. 
+000000251140 A book about understanding and maintaining a ten-speed bicycle.
+000000419408 The skateboarder is jumping over the railing in a park.
+000000268729 A grouping of zebras in a pen next to a giraffe.
+000000529939 A man walks up to where surfboards are propped up against a wall and wetsuits are hanging on a rack.
+000000240049 A woman holding a little child as they look at giraffes.
+000000084477 A beach chair and umbrella in the sand on the beach.
+000000523782 A tiny red bird perched on a car's mirror
+000000465129 a person standing next to a metal pot holder in a kitchen 
+000000544565 A with plate with a chocolate pastry and whip cream. 
+000000210230 The woman is putting a hot dog into the bun.
+000000190007 a man on surf board riding a wave over his head
+000000230983 a man wearing a hat does a trick on  a skateboard
+000000146498 A street sign at the corner of S Lane St and 12th Ave.
+000000000885 This woman has just returned a volley in tennis
+000000332901 a few animals that are in side a fence
+000000088345 A woman standing in front of a fiber optic tree.
+000000112298 The man holds a ping pong paddle under the shelf on a wall.
+000000094852 A mother elephant with her young one walking around.
+000000019109 A row of parked motorcycles sitting in front of a restaurants.
+000000259690 A man standing next to another person in a wheel chair.
+000000092416 A young man holding a bottle of beer while wearing a suit.
+000000488710 A woman standing next to a princess parking only sign.
+000000327617 An older photo of a woman on a tennis court posing with her raquet.
+000000031735 A modern living room with very unique furniture.
+000000267169 a baby elephant drinking milk out of a bottle
+000000365745 A bomb squad truck turning at a crosswalk. 
+000000343937 A woman in white jacket snowboarding on snowy hill.
+000000474344 A man who is holding a baseball bat.
+000000282298 A group of people walking across an airport.
+000000529528 A bowl of mash potatoes with a mixer in it.
+000000060835 A dog sitting inside a cage, in the dirt.
+000000057149 A silver car is stopped in front of train tracks as a train crosses.
+000000447187 A group of boys playing baseball on a field.
+000000203294 People standing next to a bus with a cat face on the front.
+000000236412 A plate of food is sitting on a red clothe.
+000000167902 A large yellow and black toucan sitting on top of a tree branch.
+000000580294 A woman makes eggs on the stove top.
+000000440508 An orange train is on the track with many cars behind it.
+000000475191 A blurry woman is holding a tennis racquet in the air
+000000430871 A parket meter sitting next to a parking space.
+000000318908 Cute little dog poses for a picture with the top of it's head dyed
+000000094336 A cat is sitting inside of a bathroom sink.
+000000023751 A statue of a boy is holding a kite in his hand and standing next to a girl statue.
+000000459272 A woman holding a smart phone next to a group of people.
+000000465549 Two young people playing a game on a video game console.
+000000311789 a close up of a cat laying on a desk near a keyboard
+000000175535 a white plate with some food and two trays of sauce
+000000441442 A young girl rides her horse in the middle of a course.
+000000106266 A plate topped with potatoes pan cakes and other foods.
+000000280930 A little girl smiling as she is about to put something in the oven. 
+000000571718 A man standing in front of a fruit stand containing several bunches of bananas.
+000000546475 A man and his baby girl taking a selfie.
+000000053529 A dog with a green hat in a truck and a green shamrock posted to the truck's side mirror. 
+000000557258 A white toilet sitting in a bathroom next to a tiled wall.
+000000416343 A boy sitting next to a girl holding a big white surfboard.
+000000580418 a herd of cows walks down a city street 
+000000261161 A dog is sitting next to a bench. 
+000000106912 A skateboarder behind a house on a cement pad.
+000000231508 A uniformed baseball player being watched by people.
+000000034139 a huge suitcase with a bunch of stickers on it
+000000257566 People in a small boat traveling in water along a walkway
+000000159684 A large long train on a steel track.
+000000230993 Two people walk down the street with umbrellas.
+000000399205 A group of people on a sidewalk with their mobile phones.
+000000223959 A man in white shirt and black shorts playing tennis.
+000000314177 A youth on a toilet taking a "selfie" in a mirror.
+000000033005 A person on a tennis court standing on one leg
+000000557258 a toilet with a lid up leaning against a tiled wall 
+000000312720 A man flying through the air while skiing.
+000000581781 A group of bananas surround a small display of kiwi.
+000000216277 An partially eaten apply and a cell phone set on a table.
+000000306700 A woman getting ready to eat a pizza on the table.
+000000513484 A large brown bear standing next to a pile of rocks.
+000000159112 A pizza topped with cheese, vegetables and lemon wedges.
+000000177861 A group of people walking down a wet rain soaked sidewalk.
+000000226408 A blurry black an white image of a girl laying down.
+000000239318 a lap top that is sitting on a table
+000000186296 White and grey kitten lying on a messy green carpet.
+000000494863 A train on the train tracks in the daytime.
+000000163951 An overripe banana in a tiny chair with a flashlight and boombox.
+000000287649 A kitty cat looks up from a desk, near a keyboard.
+000000421455 The woman is holding her phone toward the dog hanging out the car window.
+000000530466 A train covered in blue paint and graffiti.
+000000166642 An elephant standing on top of a rocky hillside.
+000000369310 boy wearing a helmet while playing a video game
+000000135872 a desk with a laptop and a desktop computer 
+000000345469 a bunch of food is on display in a case
+000000222299 assorted items sitting neatly on a bed 
+000000080949 A kitty cat lies down on a computer keyboard.
+000000270386 The black street pole has three different signs on it. 
+000000174123 a person holding a knife and fork over a pizza
+000000460147 A busy highway is being viewed from a distance.
+000000155179 A large round cake is sitting on a plate.
+000000438017 a street with traffic and wet roads from the rain
+000000145597 A group of woman sitting around at a picnic.
+000000545219 A bus that is decked out with multi colored lights on it.
+000000078843 A man is kissing a woman on the top of the head.
+000000368456 A large long train on a steel track.
+000000108026 A man holds a controller and keyboard with his hands
+000000353051 a woman in green hairs on the field with a brown horse
+000000445846 A kitchen scene is shown with cabinets and appliances.
+000000301135 Several people who are sitting on some benches.
+000000222458 A woman in a coat walks along a snow covered sidewalk. 
+000000071226 Two dogs and a cat sleeping on their human's bed
+000000048564 A young man with ski goggles talking on a cell phone on a snowy day.
+000000116208 A very large pizza covered in cheese and toppings.
+000000359677 A kite surfer gets high in the air while his kite is out of the picture frame.
+000000463690 A man laying on his back while kicking a soccer ball.
+000000429690 A crowd watches a batter try to hit a baseball.
+000000173183 A street sign gives the cross streets on a sunny day.
+000000393469 A man sitting on his knees in the snow.
+000000163611 A pizza sits on a plate with one piece taken out of it.
+000000172595 A desk with multiple computers inside a office 
+000000581615 A picture of a white toilet in the rest room. 
+000000223090 A cutting board with carrots and some other leafy green vegetables along side a knife.
+000000404479 An airplane landing on a runway in an airport
+000000181796 A dinner plate with beans, bones, and veggies. 
+000000369757 Pizza are on display for the hungry diners
+000000302760 Soap and a water bottle are on top of a vanity in a teal bathroom.
+000000151480 A group of baseball players that are walking in the outfield.
+000000520531 an image of a cat sitting on top of the desk area
+000000106266 A plate of food has potatoes and fruit.
+000000312278 A collection of luggage that has been placed on a wooden carrier.
+000000308193 A building with a clock built into it. 
+000000334719 Children & one adult flying a kite in an open area
+000000575357 there is a dog in the air going to catch a frisbee
+000000521956 A man getting ready to serve the ball in a tennis game.
+000000453981 Two black bears playfully biting each other on the grass.
+000000418959 A bird has it's head below the surface of the water.
+000000504635 A group of zebra's running through a field.
+000000526103 An elephant standing by a plaque with a description of it.
+000000136334 A person skiing down a hill maneuvers around a flag. 
+000000183437 Three Asian people are riding on top of an elephant. 
+000000537812 A small bathroom with two containers catching plumbing drips.
+000000415990 a bunch of cows are in a grassy field
+000000121744 A tennis player with his leg up in the air and a racket in one hand.
+000000292225 Two people are playing tennis in an outside court.
+000000210230 a bunch of women serving hot dogs 
+000000507893 A bathroom with gray painted walls and white tile near the bathtub.
+000000283785 In a library children look on as costumed characters give a presentation
+000000565778 A blue commuter train traveling towards a train tunnel.
+000000226984 A kitchen with an oven with a green rag and an orange rag.
+000000262895 Elderly  gray haired man in a suit scowling into the camera.
+000000291664 a brown dog with a collar sniffing a red fire hydrant
+000000437392 A pink monster made out of a toilet seat with googley eyes.
+000000229216 Two women sitting next to each other using a cell phone.
+000000308631 People standing around a home built motorcycle used for racing.
+000000068765 A clean white keyboard laying on the table
+000000297830 A black and white dog sits next to a bottle on the ground.
+000000575187 A man feeding a giraffe through a metal fence
+000000497628 A bed that has an old fashioned wooden frame.
+000000052565 A cow is standing in sand by some water.
+000000504389 A couple of people playing a weird game of tennis frisbee.
+000000431896 A passanger train stopped on the railroad tracks at a station.
+000000139077 A small child stands in front of a television playing a video game.
+000000064898 Two children who are on wake boards in the ocean.
+000000166642 The grey elephant is walking between the rocks 
+000000247806 A brick clock tower ascending towards the heavens
+000000453708 A pile of green bananas on top of a woman's shoulder.
+000000486040 A monochromatic image of a desk with two laptops.
+000000426166 A bicycle near a pole in front of a building. 
+000000304365 A commuter train traveling down the tracks into the next station.
+000000048924 A dirt bike parked near a tent in the woods.
+000000366225 a monitor sits on a desk in front of a keyboard 
+000000201072 A man in a defensive stance with a tennis racquet.
+000000300842 A red skateboard with blue wheels on the floor with someones foot on it. 
+000000481413 A toddler is standing on the shore holding a Frisbee.
+000000155145 Three people one in the water with two boats one with an umbrella.
+000000243034 The clock tower features a very unusual walkway.
+000000351331 the pizza is on a plate with a fork next to it 
+000000522751 Some yellow street lights on some poles in the middle of a street.
+000000397279 A man leaning over holding a tennis racket.
+000000243867 A bus parked along the side of a busy street.
+000000476770 A man in a baseball game running to base and others trying to tag him out.
+000000553788 A girl on a bed studying something on her laptop.
+000000539445 A red train traveling down tracks next to a field.
+000000454978 A rural road with trees, road signs of a squiggly arrow and of 55 MPH, and motorcycle parked beside it.
+000000018575 Two plates, one with a sandwich and fries, the other with just a sandwich and a pickle.
+000000108495 Cute girl sitting on a skateboard in the driveway 
+000000031735 A living room with red walls and dark furniture, with light streaming in two windows.
+000000170474 A young man serving a tennis ball with a green wall behind him.
+000000179487 The woman is on the court playing a game of tennis. 
+000000346232 these people are walking together down a road
+000000176847 A black bird is eating a berry from a bush.
+000000312552 A man eating a piece of pizza on top of a plate.
+000000327605 This woman has stopped snowboarding to pose for a picture.
+000000294350 A cook in the kitchen is making something very good to eat.
+000000378454 A man in white shorts jumping in the air to catch a green Frisbee. 
+000000073326 This refrigerator is packed with many foods and drinks
+000000094614 A man riding skis down the side of a snow covered ski slope.
+000000332318 Cows lounge in a field with a mountain backdrop.
+000000164363 There is an old clock showing the current time
+000000214720 A man sitting at a table in front of a plate of food.
+000000152740 Numerous head of cattle are grazing in the grass.
+000000133233 A large pier holding several boats overlooked by mountains.
+000000144706 Windshield with rain drops coming up to a stoplight.
+000000551439 A woman smiles from behind the book she has been reading.
+000000138954 A plate of apples, cucumbers, tuna and peas sitting on a table.
+000000304291 a horse that is walking around a field
+000000048555 A black and white photo of jockeys riding their horses
+000000135670 A bike is leaning against the railing by the window.
+000000483667 A man wearing a top hate with a red tie and black shirt.
+000000329614 A bus and other cars are traveling on a street.
+000000081738 A person cutting a sheet cake with a large knife.
+000000336209 A man rides to the top of a skateboard ramp.
+000000117374 Two cats sitting in the dirt under a tree.
+000000031322 A flock of swans swims in a bay.
+000000328430 Woman in white outfit swinging up to hit tennis ball. 
+000000399655 A small dog is laying on a couch with a shoe.
+000000001818 Baby zebra sucking milk from its mothers teat. 
+000000521540 A plate with a banana that is cut up, cookies and a spoon on it. 
+000000125257 A man on a snowboard on a groomed slope.
+000000341921 A young boy sits and reads a book on a bench at a quiet park.
+000000203488 A sign pinned to a tree next to a parking meter.
+000000173371 The pizza has small plates with various foods around it.
+000000237864 Adult elephant running after a small brown dog. 
+000000318238 Several dogs sleeping together on a bed next to eachother.
+000000436738 Vintage street with cars and buses lining it
+000000025139 The two zebras are looking through the fence.
+000000528977 Street signs cast shadows on the graffiti decorated wall.
+000000325114 A table with a plate of food and a urinal basin
+000000500423 a train parked next to a railway platform 
+000000201025 two huge elephant with tusks hidden among the trees
+000000166768 An older woman standing next to an older man.
+000000177893 A parked bus next to a building and palm trees.
+000000433134 a black and white cat on some grass
+000000571264 A man stands by a motorcycle on the side of the road.
+000000159977 A giraffe and a zebra checking each other out.
+000000114770 Display featuring military vehicles and large gun with huge mural.
+000000384468 THERE ARE PEOPLE RIDING ON AN ELEPHANT BACK
+000000297343 A stop sign standing up against a wall.
+000000079031 A young surf boarder is riding in the water
+000000053624 Elephant enjoying a drink of water in zoo.
+000000548246 A group of people standing on top of a tennis court.
+000000423944 Headless statues show of clothing beneath a colored background.
+000000554328 A group of three men standing next to each other on a baseball field.
+000000444879 A commuter train traveling down snow covered train tracks.
+000000410934 A broccoli plant growing in a garden is dwarfed by it's leaves.
+000000215644 A picture of an unfinished back splash in the kitchen with a microwave and other small kitchen appliances.
+000000205542 A stuffed teddy bear sitting on a metal sculpture
+000000458109 A passenger train rides on the rails, with another set of rails nearby.
+000000221502 bouquets of flower sit on either side of a memorial bench.
+000000138639 Urban downtown city center with a bicyclist. and pedestrians 
+000000243344 a cat sits on top of a fridge in a kitchen
+000000152120 A man in a jean jacket riding a motorcycle on a road.
+000000344795 A dish covered in aluminum foil is baking in an oven. 
+000000119233 A black and white cat laying across the keyboard of a laptop.
+000000306139 A group of people sitting around a table with laptops and notebooks.
+000000449909 A young boy standing on a field next to a giraffe.
+000000464358 A bed and chair sitting next to a window in a bedroom.
+000000218439 Man brushing his teeth and looking at his phone.
+000000438774 A group of people that are standing in front of a island.
+000000396568 A train chugs along parallel to a lush forest.
+000000153217 A cat walks next to a motorcycle on a sidewalk. 
+000000327605 A woman poses on the slope with her skis. 
+000000163682 A man in a blue dress shirt and tie sitting in an office chair.
+000000500478 a baseball player gets ready to throw a pitch 
+000000405279 some people a bench some grass trees and kites
+000000263679 A man holds strings connected to a large parachute or kite on the beach.
+000000205834 a close up of a dog near a bowl
+000000438269 There is a bird standing in the sand.
+000000355817 A bus on the road outside a large building.
+000000163746 A pretty young girl standing next to another pretty young girl on a beach.
+000000166664 A big elephant makes its way across the road with onlookers in the car.
+000000424642 Young men playing disc golf in treed area of park.
+000000292456 A man and a woman on a blue motorbike are picking up a piece of luggage from a man.
+000000501523 A cat laying down in a bathroom sink. 
+000000288584 A person dressed as a giraffe carrying a bullhorn.
+000000276804 Baseball players engaged in a baseball game. 
+000000157807 a cat that is standing on a toilet and next to a sink
+000000366141 a living room with a couch and large television
+000000498747 A man talking a phone up to his face near a crows of people.
+000000376365 A glass of milk is sitting next to a vase with a reflection in it.
+000000454978 The motorcycle rider is taking a break before continuing his ride.
+000000347254 Bearded man in a suit about to enjoy an adult beverage
+000000419408 A man riding a skateboard over a rope.
+000000404805 A women is laying on a board surfing a small wave.
+000000533958 A plate with a green salad and an egg
+000000169996 Two people are riding bikes through the street traffic.
+000000419974 A man slices meat in his kitchen while his dog begs from the floor.
+000000522156 Close up of an hors d'oeuvre on a plate.
+000000070229 A teddy bear sitting on a bench in the middle of the park.
+000000075393 Some wooden objects and beads in a display case.
+000000368940 a bathroom with a mirror and a sink
+000000031817 A man riding a motorcycle while eating food.
+000000570456 This is what your ceiling fan thinks your house looks like.
+000000284106 A plate of chicken and vegetables sits next to a bowl of rice. 
+000000427077 A man stands poised with a baseball bat.
+000000218249 An order of food sits on a table.
+000000233567 Black and white sheep in hay next to wooden fence.
+000000234607 A man and a woman playing a game on the Nintendo Wii.
+000000190753 4 men play a game on a park bench.
+000000130579 A Boston Red Sox pitcher standing on the mound
+000000329827 A person skiing on a mountain covered with snow
+000000325838 A woman works at a computer workstation in an office.
+000000233825 some windows a fireplace books tables and a couch
+000000531134 A traffic light has sad faces on the lights.
+000000329456 Several people crossing a road with one person having food in their hand.
+000000335954 Pair of fruit cups on kitchen marble counter.
+000000399462 A girl with a kite running in the grass.
+000000406417 A jumping boy wearing three different neck ties
+000000421455 A person holds a phone in a car with the window rolled down
+000000092939 A happy couple is cutting a decorated cake.
+000000126110 Dog sitting inside the cab of a yellow truck. 
+000000320632 A group of zebras stand together in a field.
+000000261161 A dog with a leash on is sitting near a park bench
+000000102820 A young child that is sitting by a red cake.
+000000190648 A messy and unmade bed and a red chair
+000000219440 a cow staring into a camera standing on a field 
+000000104619 A small giraffe standing on top of a dirt field.
+000000571598 Those look like remote or game controls in their hands.
+000000035062 Child in bed covered with a flower patterned blanket. 
+000000108495 A girl is sitting on the sidewalk on her skate board.
+000000511647 Man flying a kite high above hotel buildings
+000000180011 A man is offering leaves to four giraffes.
+000000121673 a person riding a surf board on a wave 
+000000047740 A man in shorts is laying on the beach.
+000000259097 A man jumps to catch a Frisbee flying through the air.
+000000361730 A crowd of people standing on top of a field flying colorful kites.
+000000455301 A baby looking at a picture book while mother looks on.
+000000515579 Man holding a ball he is about to throw.
+000000179898 A child is holding two hot dogs up to the photographer.
+000000281409 A major league baseball player is swinging the bat to hit the ball.  
+000000148707 There is an orange growing in the tree.
+000000513580 A motorcycle driving by a red and white biplane.
+000000410735 Four pictures showing the way a stew being cooked looks at various stages.
+000000214224  A blender sitting next to bottles of booze under a sign that reads PUB.
+000000014226 A man in a green suiit leaning over a laptop
+000000475365 a close up of an the front of an amtrak train 
+000000372718 A vase with some water and dead flowers in it.
+000000227187 some green yellow black white and orange birds and some plants
+000000368940 Small wooden and white kitchen sink with a small mirror. 
+000000119445 a baseball player swinging a bat at a ball
+000000474028 A little boy grips a soccer ball in his arms surrounded by other youth soccer players.
+000000398742 The team mates are lifting up one of their members.
+000000073702 A couple of men standing next to each other on a  tennis court.
+000000091615 A kitchen has light wooden colored cabinetry, beige countertops, and stainless steel appliances in an open concept house.
+000000490171 A man and a dog riding a surf board in the water .
+000000218439 A kid on a bed with dozens of photos on the wall behind him.
+000000369771 Two plastic containers sitting on top of a wooden table.
+000000231508 The pitcher  on the mound has just let go of a baseball directed to the batter. 
+000000395575 View of a pond with reflection of the surrounding scenery, with a bench and statue surrounded with flowers and trees,
+000000408112 A small propeller plane sitting on top of an airport tarmac.
+000000148739 a person riding a surf board on a wave
+000000276707 Shopping district with a banned sign for anything with wheels.
+000000203629 A group of people gathered inside of a room. 
+000000311928 A young boy swings at a ball with a bat
+000000559707 A white plate with noodles, chicken and broccoli.
+000000297681 The two signs give directions to upcoming cars.
+000000260261 A man with fruit on the back of a pickup truck.
+000000564023 A public bathroom with a hole in the floor.
+000000243344 A cat sitting on top of a metallic refrigerator freezer.
+000000470173 a statue us standing in some green grass
+000000153669 A man swinging a bat on top of a baseball field.
+000000290833 Two zebras eat grass in a fenced field.
+000000531036 the driver of a double decked bus standing outside
+000000071451 Bed made with folded towels and walls with window
+000000275727 A train is moving along the tracks past a traffic signal.
+000000490125 Several white birds standing in a puddle in a parking lot.
+000000195754 The spaceous living room has a large television and a fireplace.
+000000544444 A man is enjoying his day of skiing, trying to stay up float. 
+000000047740 A man lies on the beach while someone else holds a kite.
+000000553990 there is a woman jockey riding a hose over the hurdle
+000000064523 Eleven girls in tennis outfits pose for a picture with their racquets on a tennis court.
+000000263068 a person riding a skate board at a skate park
+000000574823 The men are playing baseball in the field.
+000000463849 River down the middle of a city with many parked boats.
+000000217285 some ball players are looking home plate and talking
+000000205776 A bear laying in a canvas type net
+000000401862 People are lined up by a red food truck.
+000000378453 A group of zebras standing by a water hole.
+000000547144 There are two desktop computer and one laptop on the desk.
+000000149568 The man plays on the beach with his two dogs, one of which is a black standard poodle.
+000000253742 a woman holding an umbrella at the park
+000000276720 A house lined road with red trucks on the side of the street
+000000217614 A giraffe stands in the brush near trees.
+000000381971 A horse and carriage with people riding in it.
+000000084674 A woman is holding a baby with another young boy beside her.
+000000472046 Empty stylish kitchen and living room with big picture windows.
+000000453584 A sidewalk scene with focus on a red bench.
+000000248752 a man is holding a baseball bat at a game
+000000132544 A girl taking a swing at a baseball during a game.
+000000115118 A train driving down the tracks near several trees.
+000000144706 a window with rain drops on it looking at a street light 
+000000320664 a plate with a hot dog in a bun, macaroni and cheese and stuffing on it
+000000465806 A table with bananas close to the road.
+000000461751 A man on a motorcycle driving beside a van.
+000000357060 A small bird pecking at the sand and water.
+000000489046 A small bird sitting on a thin tree branch.
+000000207306 a teddy bear dressed in prison stripes sitting on top of a trash can
+000000089045 A baby grand piano sits in the corner of the living room.
+000000037689 There are two snowboarders in the air completing stunts.
+000000474786 a bathroom with a sink a monitor in the mirror and a tub
+000000136033 A bathroom sink with mirror are shown in this picture.
+000000368752 a little boy smiling at someone across the table 
+000000011760 Three zebra standing side by side on a dirt field next to a flowery forest.
+000000226058 Some very cute stuffed animals on a couch.
+000000365387 A room with a sink, bed and toilet in it
+000000572620 A man on a skateboard on a ramp in front of spectators.  
+000000196754 A view of a street intersection through a window.
+000000532058 A young boy lying in a bed with a sippy cup.
+000000191845 a crowded street of people all with umbrellas
+000000213830 Couple of men during couples match playing tennis about to contact ball
+000000184978 there is a snowboarder coming down a very steep hill
+000000306136 An "Apple" logo burns bright over a city bus.
+000000498919 A selection of pears, orange, and watermelons on display. 
+000000226130 The sausages are served with parsley on the side.
+000000186042 a person skiing through the snow near some trees toward a sunset
+000000094326 A white statue holding open a newspaper while sitting on a bench.
+000000571718 A man with graying hair looks down at a stand full of yellow bananas.
+000000085195 a slice of cake with some syrup next to it 
+000000469192 A car is parked on dry grass with kites overhead.
+000000450488 A bedroom with a bed, TV and ceiling fan.
+000000221708 Partial picture of a kitchen and a table and chairs under a lamp.
+000000535156 An elephant leans against a post in a fenced area.
+000000356968 A brown horse walking outside in the grass
+000000175443 A teddy bear handing on a lit christmas tree.
+000000183965 A close up of two pastries and a cup of coffee.
+000000226662 A customized motorcycle with more in the background.
+000000384661 An oven with a stove, pots and utensils, and a refrigerator next to it.
+000000089648 People using electronic devises all facing in the same direction.
+000000185473 A herd of cattle grazing on a lush green field.
+000000288685 a sheep in a field with a dog and people watching 
+000000530820 a man standing on the court holding onto a tennis racket 
+000000172571 A sliced pizza sitting on a table next to a glass of water and a glass of wine. 
+000000407960 A cat looks up at the stove and is reflected in the oven's glass. 
+000000171611 A man sailing a boat with his dog riding in front.
+000000317433 A man guiding a plow being pulled by two horses.
+000000374982 A young girl looking at a huge pizza in amazement.
+000000144784 A person presses the power button on a blender containing liquid
+000000061108 A dog waits in the basket of the bicycle.
+000000342186 A veggie pizza is sitting on a white plate.
+000000247838 A small herd of buffalo resting on the ither side of the stream.
+000000140658 The exterior of a building with a clock.
+000000284296 Three giraffe drinking from a pond with brush in back.
+000000397639 A lamb and a sheep are out in the field together. 
+000000537672 A large black refrigerator with various bits of paper stuck to it with magnets.
+000000370711 A man's reflection taking a picture through a shop window.
+000000139883 A view of a piece of art near a tree.
+000000332845 A man in white shirt sitting at a laptop in bedroom.
+000000380706 A young person on a scooter in an urban area.
+000000456394 A blue motorcycle parked on display  next to other vehicles.
+000000140640 Three women that are standing in front of a table.
+000000323355 A young kid is holding a box of pizza. 
+000000377575 A tog is leashed to a post on the sidewalk.
+000000466986 a man is standing inside of a food truck
+000000448410 A train is entering its stop with people waiting for it.
+000000289393 Children's toy animals are strewn across a floor.
+000000568290 Articulated bus, first part is red, back part is grey, on dull red street.
+000000463690 A man balancing a soccer ball on his foot.
+000000492758 a living room that has a couple of chairs in it
+000000378099 A computer mouse and keyboard on a table.
+000000457884 The young boys are playing a game of baseball in the park. 
+000000556873 A cake laying on the ground near flowers.
+000000073946 some men are looking at a design of a shark
+000000212895 A giraffe standing in a open grassy field.
+000000298697 Horses are eating the hay in a large field.
+000000322944 An injured woman holding a teddy bear close to her chest
+000000245026 A woman wearing glasses poses behind her birthday cake. 
+000000372317 A red bus driving down a street next to a tall building.
+000000195045 A park filled with green grass and a tall leaf filled tree.
+000000088462 A school bus in the street behind other cars.
+000000404678 A family sitting down at a table for a meal.
+000000270386 A one-way sign at Library Way and Madison Ave.
+000000206838 Two women in bikinis on horses stand in the ocean.
+000000416269 A train is in the train station with the lights on.
+000000100428 A man pulling himself out of a mirror by his own tie.
+000000144932 A small boat makes its way across the water on an overcast day.
+000000453981 A black bear holds a baby black bear as they lie on the ground.
+000000537827 Two baseball players stand near the baseball field.
+000000289586 A giraffe in the display at the zoo.
+000000042563 A train traveling along a forest through the snow.
+000000364126 A person on a surfboard in the water.
+000000263679 two people standing on the sand of a beach in front of a large colorful kite
+000000512330 An open refrigerator filled with lots of food.
+000000320642 A woman holding a game controller as her friend watches.
+000000228144 No one is in the room but there are chairs.
+000000156292 A person standing looking at a  large statue with clocks built into it.
+000000159311 Two zebras grazing from grass next to a tree.
+000000066231 A group of people work in a kitchen together.
+000000275198 A group of people holding umbrellas eating a meal outdoors.
+000000284282 A GREEN TABLE WITH A OLD RUSTY BLENDER AND A PRINTER 
+000000447917 Two girls and a boy have fun at soccer practice
+000000493566 A train caboose on tracks next to another train.
+000000060090 A yellow passenger train pulling up to a station.
+000000452784 A plate of food that includes lentils and leafy greens.
+000000396205 Two cows standing and grazing in an open pasture.
+000000265108 A red balloon tied to a purse with an umbrella and luggage.
+000000139684 A living room with walls painted beige and furnishings and accents in shades of blue and purple.
+000000285894 Some children look at two giraffes in an enclosure. 
+000000312586 This is a picture of three bears in a field.
+000000164602 A guy and a pretty girl taking a picture in the mirror.
+000000181859 A cute cat laying down in a sink.
+000000270066 A motorcycle is parked near a puddle and a van.
+000000411530 Many skiers are waiting to ride a chair lift.
+000000407825 An alarm clock that is shaped like an apple.
+000000087144 A baby sitting between a lady and a man on a bench under a tree.
+000000489046 There is a bird perched on the limb of a tree.
+000000065350 A young person is doing a trick on their skateboard.
+000000452793 A kitchen with a stainless steel fridge and granite counters. 
+000000180792 A toilet stall with exposed pipes and a window.
+000000096960 A polar bear is under water next to a rock wall.
+000000006954 A group of young children sitting in the grass.
+000000087470 A bunch of cows walking on the side of the road
+000000351609 Peeled tangerines sitting in a bowl in a kitchen.
+000000395180 A cyclist pauses near the water to observe a canoe.
+000000291861 Black and white photograph of a zebra between rocks.
+000000350488 A zebra grazes on tall grass at the edge of a clear blue river.
+000000548780 Two women sitting on a metal park bench
+000000143572 A man holding a tennis racquet on top of a tennis court.
+000000328601 a tennis player getting ready to swing a racket
+000000491497 The living room is empty with the television on. 
+000000488166 a slice of pizza sitting next to a coffee cup on an outdoor table 
+000000425361 A young man that is having some wine and something to eat. 
+000000498286 The small dog has a tiny black nose.
+000000243075 A wet polar bear looks up at the sky in his zoo enclosure
+000000352760 A skier in mid air after a jump 
+000000155154 Some sort of bathroom facility, possibly in an airplane.
+000000213035 Two sets of hands are each holding a cell phone while another hand in the background is holding a glass.
+000000275392 A smiling woman is on a brown horse.
+000000355169 A electronic sign at an airport listing the flights.
+000000137950 A plane flying over several houses and lots of trees. 
+000000537053 A beautiful zebra standing in a green field.
+000000231339 a little girl looking inside the refrigerator. 
+000000080659 a group of racing horses are racing around the track
+000000507042 Three giraffes looking in different directions from each other. 
+000000217957 A clock is shown with gold detailing around it.
+000000225670 A brown dog leaping to catch a flying disc.
+000000052007 People are boarding a bus parked at a stop.
+000000523100 A metal bowl filled with orange slices on a table.
+000000297578 A man holding matches and a cigarette in his mouth.
+000000027768 A bus which is part of the Metropolitan Transit System.
+000000189451 A wedge of chocolate cake leftover on a platter.
+000000122046 a young Asian woman model sitting on a folding chair in a sleeveless dress, high heels and holding an umbrella
+000000520264 A man and a woman are posing for a picture in a kitchen.
+000000578500 A living room with the walls painted orange-red color.  
+000000158548 A sheep is walking along a wide field.
+000000416256 A CAT LAYING ON THE FLOOR AMIDST A COMPUTER,SPEAKERS,CORDS
+000000475678 Many pieces of luggage gathers together, all with tags attached.
+000000382088 A big and white horse in a dirt cage. 
+000000311950 A hot dog with mustard and catchup on it
+000000437205 a small boy in a blue shirt holding a banana peel
+000000229358 A man is looking at himself in the mirror.
+000000491497 A pillow covered reading chair in the corner of the living room
+000000411938 A boy is riding a bike with a teddy bear as a person sitting is toasting a beer.
+000000418959 A bird floating on top of water in the rain.
+000000328430 A woman in a very short skirt swinging a tennis racquet at a ball.
+000000330818 A young man working at a range in a commercial kitchen.
+000000560266 Two adult bears play fight in the water.
+000000099054 An airplane being prepped for flight, as another plane passes by.
+000000161128 A no cursing sign posted as a street sign.
+000000061333 A orange and white cat sleeping on top of a blanket on a bed.
+000000522751 a traffic light attached to a pole with a sky in the background
+000000295316 some people are riding surfboards on a wave
+000000073946 TWO MEN IMITATING A SHARK PICTURE ON A SURFBOARD
+000000108253 A plate of cheese bread next to bread sticks and wine.
+000000000285 Closeup of a brown bear sitting in a grassy area.
+000000025424 A tennis player hitting a ball on the court 
+000000499313 A herb topped pizza lying on a plate.
+000000186637 A giraffe eating out of a food bucket in its enclosure.
+000000142971 A man riding on the back of a surfboard next to kids.
+000000401244 a person playing frisbee on a field in sport wear
+000000094871 A group of people standing around sheep on a field.
+000000357748 A pink city bus at a bus stop.
+000000216516 A man that is standing on ski's in the snow.
+000000206218 a laptop computer sitting on top of a homemade machine with wheels
+000000576031 a person riding skis on a snowy slope
+000000205401 A BIPLANE AND A SMALL BOAT OUT ON THE WATER
+000000565776 a kitchen view of a counter cabinets stove and pots
+000000007088 A little girl holding an umbrella while standing outside.
+000000139260 A portrait of a banana split in a plastic bowl.
+000000363784 A bathtub is filled with junk in a dirty bathroom.
+000000431140 A bathroom with a sink, mirror, and toilet with handicap rail.
+000000019786 Two people with Wii remote controllers near a sofa.
+000000119828 A cat sitting on top of a wooden desk.
+000000490125 White birds getting a drink from the water puddles.
+000000514508 A man looking at a picture of a TV screen.
+000000166478 I am unable to see an image above.
+000000117645 A person on a snowy slope on a snowboard.
+000000156071 A contestant has caught a sheep during a rodeo event.
+000000496571 A person is eating at a table with plates, a fork, knife, spoon, cup, and cell phone on it.
+000000147740 Woman carrying a bunch of bananas on top of her hat. 
+000000515577 A few people on skis skiing on a snowy mountain.
+000000128654 A man is doing tricks with a frisbee
+000000460682 black and white of a horse lying  down in a field
+000000091500 Two men are sitting in chairs, playing a video game.
+000000199236 A knight riding on the back of a brown horse through a lush green field.
+000000152740 A herd of cows grazing in a grassy field.
+000000288042 A man holding an umbrella next to a waterway.
+000000342295 A white toilet sitting next to a bath tub filled with paintings.
+000000308165 a man on a skateboard in mid air going over a cone in some dirt
+000000205647 A semi truck pulling a chain of three trailers.
+000000345252 A young man sitting at a computer desk in front of a computer.
+000000118209 Two horses stand and look out beyond a fence.
+000000451090 a white boat some green hills and water
+000000055022 A row of bikes parked next to each other.
+000000213422 Large made up bed in modern bedroom, with small desk.
+000000118209 there are two horses standing behind a fence 
+000000370677 Three female workers stand in front of rows of pastries. 
+000000329219 A man standing next to a dog on the ground.
+000000410510 A boy playing tennis on a blue and green tennis court. 
+000000186422 a brown bear is walking away from a river
+000000223955 A man with long hair is throwing a Frisbee.
+000000073118 Two teddy bears sitting next to each other on a couch. 
+000000391140 A man makes a face while holding a video game controller.
+000000170595 The plate has bread, meat, eggs and tomatoes.
+000000405249 a child blows out the candles on a birthday cake.
+000000162092 There are many people who are gathered at a get together or a party
+000000572303 A train has oval like mirrors on the sides.
+000000057725 A large building with a clock tower near a river.
+000000009448 A little girl holds up a big blue umbrella.
+000000093717 a collage of a skateboarders trick on his skateboard
+000000405249 Several children are around a table at a birthday party.
+000000116206  A wooden cutting board filled with chopped vegetables.
+000000504589 A woman in a yellow shirt is about to catch a white frisbee.
+000000382122 Large brick building with a clock tower at its height. 
+000000479126 Two girls sitting outisde smoking and one girl on a laptop.
+000000443844 A man riding a skateboard on top of a wooden bench.
+000000095862 some baseball players are playing baseball on a field 
+000000348243 A BLACK HORSE IN THE GROUND IN A HILLY AREA.
+000000555012 A white toilet sits next to a bath tub.
+000000040036 A person riding the horse jumping over a wooden obstacle.
+000000226130 two hot dogs in buns sitting on a white plate
+000000479155 Two kids play in the fresh produce section of a busy grocery store.
+000000111179 A church steeple that has a clock on it
+000000083113 Animals trying to either move or go around fallen tree branches
+000000154358 A person setting in a chair in a living room with a fireplace and windows.
+000000001993 A room with purple walls, carpet and bed comforter.
+000000422998 A plate that has food on it on a table.
+000000025394 Hey man that is opening up a bottle of wine.
+000000447088 Three baseball players from the Chicago cubs teams.
+000000500613 A fire truck parked in a parking lot with other trucks. 
+000000014007 a cat sitting on top of a blue refrigerator 
+000000191471 The stop sign is visible for everyone to see.
+000000526706 The young cow is standing in the field with the older cows. 
+000000355817 A bus parked in front of a brick building.
+000000461573 Metal park benches sitting in the grass and mud.
+000000303863 A train driving down the rail road tracks in a city.
+000000434459 Two people are cutting into a large cake.
+000000164115 A surfboard mounted on a bicycle on the side of the road.
+000000037689 People doing tricks in the air on their snow boards. 
+000000282037 some baseball players on a field playing baseball
+000000404805 A person on a boogie board in the water. 
+000000399655 A puppy cuddles with a shoe on a couch.
+000000453981 A mother black bear holding a baby black bear.
+000000237864 An elephant is chasing a dog around in the dirt.
+000000122166 A woman on a bike rides on the street in front of cars.
+000000218424 A group of zebra standing next to each other.
+000000217400 A silver and green train pulling into a train station.
+000000100510 a woman getting her horse to jump over a couple of poles
+000000527960 A green park bench underneath a tree in a park.
+000000290592 Close together group of white, beige, and brown sheep, from the rear, standing in grass at fence gate.
+000000359677 A person high up in the air over the water.
+000000429109 Electric trolleys, bus, limo bus, and bicycle on a street.
+000000502599 Multiple planes hanging on top of a ceiling. 
+000000328117 THERE ARE FRIED DOUGHNUTS IN A PAN TO BUY
+000000397327 a sink and a toilet that is in a small bathroom
+000000544565 a plate of scones covered with chocolate sauce
+000000316015 A cat is sitting on a desk next to a computer.
+000000201148 A train is approaching a train signal in an area with several tracks.
+000000433915 A video screen with faces of a couple of guys on it.
+000000293071 A red. white, and blue fire hydrant on the side of the road. 
+000000002431 A table with a plate, wine-filled glass, and bread, among other items
+000000309484 A dog is laying down and getting his teeth brushed.
+000000495054 A plane is parked in a airport field.
+000000564280 A dog sitting on the couch with it's paw over a tray that holds remotes for the tv and Wii.
+000000050380 A young boy and an elderly man taking a pony out for a walk.
+000000331817 A persons arm reaching over a table filled with food.
+000000273132 A woman is sitting on a bench near a fence. 
+000000293324 A group of large passenger jets parked in front of an airport.
+000000384616 A red fire hydrant in a forest setting.
+000000449603 Man in a drysuit kite surfing in a rough ocean
+000000157807 A tabby cat standing on a closed toilet lid.
+000000013291 A group of people playing with Frisbee's on the grass.
+000000571313 A make shift office space in a bedroom.
+000000001503 A computer desk with several pieces of computer equipment.
+000000107094 A man is skiing on a bridge with snow. 
+000000319607 A traffic light on the intersection of a city, with a large building and a bicyclist in the background.
+000000063047 A man carries a bag of food and a drink away from a roadside eatery.
+000000058655 There is a clock right outside of the tall building.
+000000144798 A cup sits on the edge of a white tub in a bathroom.
+000000411953 A young man singing and playing the guitar.
+000000517056 Two women are cutting and serving a large cake.
+000000310862 A man in formal business attire stands in an office and looks angry.
+000000139872 A happy dog sitting on the grass next to a frisbee.
+000000353027 A man pulling a slice of pizza away from a pizza.
+000000499109 A sandwich comprised of rustic bread, tomatoes and sprouts on a plate along side a sliced tomato, pickles and chips.
+000000245513 A giraffe and a goose stand in a fenced area. 
+000000130465 A Church entrance with Saint and Angel Statues and a cross on top.
+000000229601 A baseball player running to a base during a game at night. 
+000000132329 Four bottles of liquid set in a refrigerator.
+000000111951 Modern bathroom with two sinks a toilet and a shower 
+000000028449 A group of elephants are standing in the trees.
+000000449198 Motion blur photograph of an oriental street at night
+000000386210 A store window with a tray of grass and vegetables
+000000114907 A woman next to a cow is giving an explanation of milking to a crowd.
+000000177015 A cat sitting on the man's lap while the man types on the laptop. 
+000000249180 Men are walking up and down the stairs.
+000000127263 The men are playing a game of Frisbee on the grassy field,
+000000432085 A man eating a sandwich off of a white plate.
+000000501368 A woman wearing a towel holding a blow dryer.
+000000574810 A cat sitting on top of a window ledge in the daytime
+000000232684 Man looking at laptop playing video game in the dark
+000000340175  a living room with a big table next to a book shelf 
+000000569976 a small child is getting their hair dried at the salon.
+000000476215 A black and white photo of a farmer standing next to horses.
+000000313454 A man in a suit is talking to a group
+000000477441 An airplane pulled up to a gate on the tarmac.
+000000161032 A woman holding a red and white umbrella while standing in a cave.
+000000321214 Baby boy at the table eating cake frosting off his hand.
+000000167898 A white porcelain pedestal sink in a bathroom.
+000000125778 A living room with a white carpet and black furniture.
+000000007977 a boy on a skateboard at a skatepark
+000000065736 A woman tennis player in a black army shirt and tennis skirt, swinging a tennis racket.
+000000350405 A man on a snowboard in the snow.
+000000531707 a group of old people sitting on a bench talking 
+000000506656 A man standing in front of a few horses on the street.
+000000102356 Woman on a motorcycle with a sidecar in front of scenic backdrop in a showroom.
+000000022396 A commercial plane flying by the perspective of a half moon.
+000000506656 a man pets a horse attached to other horses 
+000000102820 A child sitting cross legged in front of a birthday cake.
+000000222458 A person walking down the street past snow covered benches
+000000543043 A red Mustang is parked in front of a schoolbus/boat/house.
+000000286994 A elephant drinks from a stream with several other elephants walking in the background.
+000000522713 A wooden bench overlooking a harbor beneath a cloudy sky.
+000000356261 A brown horse standing on top of a lush green hillside
+000000384808 A bathroom sink sitting under a bathroom mirror.
+000000578500 The room has red wall, white carpet and matching furniture.
+000000038070 A white toilet sitting next to a trash can in a restroom.
+000000480944 Traffic lights on a corner of a road with vehicles 
+000000563882 an image of a person on their cellular phone
+000000425702 A man holding a surfboard walking on a beach next to the ocean.
+000000395701 A chair and a television in a room.
+000000146825 Several people stand by a train in a train station.
+000000357238 Some one is flying a kite high on the beach.
+000000449603 A man riding a wave on a surfboard in the ocean.
+000000488673 A couple in the kitchen with a birthday cake.
+000000485972 Pastry dessert with powdered sugar and carrots on a blue and white plate.
+000000575357 A dog chasing a coaster frisbee in a field near a wire fence.
+000000071711 An air hanger with two world war two airplanes parked next to each other.
+000000279730 A person holding a sandwich next to a cup of coffee.
+000000214869 A close shot of someone pointing to their cased Alltel phone. 
+000000450559 A boy who is skateboarding does a trick. 
+000000261318 A man in grey shirt sitting in chair on cellphone.
+000000168337 There are two people standing on the side of a street.
+000000031093 a man doing a skateboard trick on top of pool 
+000000193743 A tennis player on an indoor court pauses to raise three fingers in the air.
+000000139077 A boy stands in front of a TV in a living room.
+000000284764 A man cutting up his meal before consumption.
+000000574297 The jockey is riding a horse on the beaten path.
+000000276720 An empty street with brick buildings and cars.
+000000323496 A young lad swings his Wilson tennis racquet.
+000000089271 A cat is wearing an orange and brown hat.
+000000574315 A cat laying on the track pad of a laptop.
+000000189451 A piece of cake and a knife on a plate.
+000000416991 Covered seating in front of a restaurant by a lake.
+000000561889 Plate of food with broccoli on glass plate with table cloth
+000000366199 Small cat under the blankets on a bed. 
+000000297084 Woman caught off guard while eating food from her plate
+000000491090 A MAN IS SMILING SITTING ON A MOTOR BIKE 
+000000474164 There is a dog on top of a truck outside.
+000000419882 A train yard with several parked trains. 
+000000086483 There is a sign in front of a brick house.
+000000022479 Skateboarder holding the bottom of his board in the air. 
+000000311295 A field with various wildlife and trees in the background.
+000000354753 Oncoming traffic shining their lights at a traffic stop. 
+000000335954 Two cups of fruits placed next to each other.
+000000134322 People congregated around a lake flying a large kite. 
+000000017627 Cars are parked on the the street in front of the building.
+000000222118 A woman on a cell phone with a bag.
+000000276804 A person stands at the base ready to hit a ball.
+000000491090 A woman riding on the back of a motorcycle.
+000000050165 Small train running down the tracks in the middle of a field.
+000000372203 a wood bench and post light on a sidewalk
+000000009772 A person takes a picture in a hotel bathroom.
+000000060932 A man in a suit that is standing in front of a group of people.
+000000435299 The cat is perched on the edge of the bed.
+000000262895 A man wearing a tie, jacket and white shirt.
+000000430056 This is a plate holding a double decker sandwich.
+000000240049 A woman holding a young child looking at some giraffes
+000000415882 there is a woman with a robe on laying in bed
+000000473237 Small girl in green shirt holding a slice of pizza to her face. 
+000000151662 A giraffe behind a fences reaches for the hay in the basket.
+000000066135 A train at the station with no passengers around.
+000000033854 Two very old time cars in front of a group of motorcycles.
+000000476787 a homemade pizza  is on a cutting board
+000000356498 there is some type of flat bread with topping on the top of it
+000000278006 A grey house sitting between a couple of other houses.
+000000153299 Two giraffes in a sanctuary standing close to the wall
+000000040757 A child is landing on a shallow pillow.
+000000065455 A group of giraffe Standing up against a dirt wall in front of a crowd of children.
+000000521601 A small doughnut inside a cup that's sitting on a table.
+000000224200 A dog sitting with his leash tied to a fire hydrant
+000000230008 Police officer laying a speed trap for the purpose of writing a ticket.
+000000180135 A guy holding a skate board is speaking into a microphone.
+000000343976 A picture of a street during the night.
+000000331075 A dog enjoying playing on the beach in the sun
+000000179285 A blue tennis racket has a yellow tennis ball on it.
+000000205282 A herd of giraffe standing on to of a grass covered field.
+000000419601 The shades are partly drawn on this black and white motif room.
+000000231088 Several umbrellas of different colors are hanging in the sun.
+000000323751 A passenger train that is traveling down railroad tracks.
+000000130826 Two men sitting next to each other on a wooden bench.
+000000289741 Beautiful view of two cruise ships at dock.
+000000557501 A restroom with two sinks and a red colored stall.
+000000297085 The cat sits directly in front of the television.
+000000569700 a green vase with some flowers and greenery
+000000380711 On a gray day a surfer carrying a white board walks on a beach.
+000000060770 A couple of giraffes are standing by trees and grass.
+000000051712 a man in a big jacket skiing down a snowy slope 
+000000125211 A zebra is standing in an open field.
+000000288882 The back side of two giraffes inside a gated area.
+000000202445 A cat is lying covered in bed with white sheets
+000000376093 Two men and woman opening boxes of pizza in a room.
+000000064574 A line of products on a white surface.
+000000432468 there is a cat laying on top of luggage
+000000352684 Persons of different origin standing in a hall.
+000000217872 a person on a snow board flying in the air down the snow
+000000402473 A couple of cats laying next to each other.
+000000080057 A teddy bear with a hammer next to a peg board toy.
+000000168593 Two refrigerators standing side by side in a room.
+000000383384 some baseball players are playing baseball and a camera man
+000000293625 Two friends are sitting on the couch, enjoying a game played on the wii.
+000000426329 Several desserts sitting behind a glass window with display cards in front of them.
+000000344795 there is a round pan covered with aluminum in the stove
+000000424721 A pan with carrots, apples, meat, and potatoes. 
+000000262487 A player is up to bat in a baseball game.
+000000435003 A desk with two computer monitors and a laptop.
+000000410735 Various different pictures of food in a bowl.
+000000157847 a large bird flying across a valley with a lot of trees in it 
+000000281687 Woman talking on cell phone on sidewalk with pedestrians nearby.
+000000308793 Two skiers heading up the side of a mountain
+000000409268 a poor little teddy bear smashed and covered with dirt
+000000253835 A woman standing on the side walk with her child.
+000000377635 Guy jumps high doing skateboard flip off a ramp
+000000292060 a kitchen full of cabinets, a stove, sink and refridgeratio
+000000015440 A stop sign sitting on top of a metal column.
+000000501005 A man with a baseball bat that is in the dirt.
+000000338304 sheep running through town all together in a crowd
+000000062554 A bowl of broccoli, noodles and other various foods.
+000000393115 The bottom of the skateboard deck shows lots of wear and tear.
+000000140439 a boquet of flowers sits on a table 
+000000167902 A black and yellow bird with a colorful beak
+000000128598 Pears and oranges have dirt and brown spots.
+000000000802 The refrigerator is brand new and was delivered today.
+000000166768 A woman holding a man by the arm, with a tie.
+000000307658 The bear looks like he is posing, beautifully coiffed.
+000000514376 A city bus driving down a street next to a hotel.
+000000540280 It is a lighting rig setup by a photographer 
+000000257624 A little girl standing in the grass with two soccer balls in front of her.
+000000455352 A clock tower with a compass sitting on top of it.
+000000543528 A plane soars hundreds of feet in the air.
+000000203580 An umbrella is placed outside of a shop.
+000000364166 2 Zebras standing next to each other in plaines
+000000449432 People in uniform standing in front of a bus.
+000000044260 A tree has dying leaves and apples in a forest.
+000000559160 A person with a skateboard on a street.
+000000508917 Many people are gathered on the platforms of a train station.
+000000056127 A picture of a restaurant interior is taken through a fish-eye lens.
+000000186042 Cross country skier skiing along a well-worn trail.
+000000449661 A woman lies in a hospitable bed in a panoramic scene of the room.
+000000406611 A bunch of people are standing in the snow
+000000308391 A fire department with a sign out front 
+000000295713 A woman sitting on a bench near a statue, checking her phone.
+000000329080 A couple of beds sitting next to each other in a bedroom.
+000000086582 A cloth bag is on the keyboard of a laptop.
+000000482585 Two trains parked on the tracks near a platform.
+000000257478 A baseball player signaling to another player or coach.
+000000547519 a bear on grass near small trees 
+000000002261 A boy riding a boogie board in the ocean waves.
+000000244181 Here are  two kinds of sandwiches and a pie with a bite out of it.
+000000227686 A horse that is in the dirt by trees.
+000000448263 a baseball player kneeling down talking to a young child
+000000521719 A structure with colorful kites flying from it's side.
+000000578500 a living room that has some couches and tables in it
+000000228981 A table with three different electronic devices. 
+000000057150 A girl in plain dress standing next to a white teddy bear.
+000000266206 Partially open door leading to a kitchen from a hallway.
+000000276024 a few people that are riding some horses 
+000000185890 Two woman wearing no shoes playing video games
+000000343496 A person walking near a stop sign in a residential neighborhood.
+000000512776 The home plate with the back catcher and someone ready to bat.
+000000082986 A trio of men throwing a Frisbee in a field.
+000000365766 The small kitchen with the spacious counters is clean.
+000000400573 A man taking a bite of a doughnut while wearing glasses.
+000000017379 a tv in a bathroom mirror next to sinks
+000000013201 The skate boarder is doing a jumping trick.
+000000307074 a large street with buildings and street lights
+000000321214 A toddler eats cake with his hands in his high chair.
+000000158744 There is a man standing next to the street.
+000000203931 The baseball player is ready to swing the bat.
+000000461036 Animal in shadows of woods surrounded by foliage. 
+000000157046 A hand is holding a tag with a cross on it.
+000000291634 A group of bikes and motorcycles parked on a city street.
+000000045472 A table with five red cups, a pineapple and some oranges on a plate
+000000055072 A single giraffe looks over the green brush.
+000000400082 Two boxes of takeout food are sitting on a park bench.
+000000255917 Some cars on a road with some buildings to the side
+000000466256 Various foods are sitting on the plate with a fork
+000000384850 A clean bathroom has a shower curtain with palm trees on it.
+000000039951 A boy is hitting a tennis ball on a tennis court. 
+000000113235 A hand is holding a carrot for a llama to chew. 
+000000535253 A variety of items is shown in a shopping cart.
+000000132375 The small vase on the table is holding three flowers.
+000000323828 Train with lights on, multiple track area, near over/under pass.
+000000210520 A table has colorful flowers and food on it.
+000000152214 A guy in a business suit and glasses smiling
+000000394559 A woman playing tennis and holding a racquet in her hand.
+000000291861 a zebra is standing by a large rock
+000000196185 A passenger jet airline taking off from the runway.
+000000180011 Four young giraffes in a zoo, with one of them being fed leaves by a person
+000000324258 two men standing in the living room and playing a video game with wii motes
+000000061658 a mixture of broccoli and cauliflower on a white platter
+000000011760 A group of zebras walking away from trees.
+000000058539 A man with a red hat is in the city.
+000000022623 A close up photo of a smart phone lit up.
+000000196754 A stop sign taken through the window of a train
+000000530052 A lot of carrots on a wood board for sale. 
+000000192191  white stove top oven with a pizza cooking inside of it.
+000000384666 There are some skiers heading down a slope
+000000132703 a man that is on a skateboard on a rail
+000000439773 A baby laying on a floor next to a purple cell phone.
+000000030213 Kitchen with many old appliances and many pots and pans.
+000000413689 Umbrellas are used as parasols to deter the sun on hot streets.
+000000366141 Cat sitting on recliner back in room facing the television
+000000365207 A dog sticks its head out of the car and is reflected in the side mirror. 
+000000050828 A bedroom with two beds placed next to one another with a desk to one side and a small window on the opposite side.
+000000292997 A snowy ladnscape with a car crashed into a pole
+000000076416 A double decker bus driving past very tall buildings.
+000000203389 A group of young people riding motorcycles near a river.
+000000400815 An old man sits in the back of a truck.
+000000338718 A street sign reading Cambridge st and Norfolk st's crossing lies broken on the ground. 
+000000325114 A urinal in a public restroom near a wooden table.
+000000006894 A man is playing with an elephant in a field.
+000000181303 A young child holding onto a kite while standing on a green grass covered field.
+000000071877 A man in a hat balancing his skateboard on a highway barrier.
+000000494634 A cat standing next to a laptop computer on top of a table.
+000000408830 a colorful motorcycle sitting next to a fence 
+000000153527 Several pieces of wood lined up near a lot with several axes around.
+000000333956 A parking meter beside a walkway near the water.
+000000010977 A bath and a toilet in a small room.
+000000210030 A piece of cake on a white plate with one bite taken.
+000000317433 Two irking horses pulling a piece of equipment guided by a man.
+000000488664 The blue train engine emits black smoke as it speeds down the track.
+000000106389 A tabby cat siting on an orange chair
+000000253452 A plate with creamy chicken and vegetables, a side of onion rings, a cup of coffee and a slice of cheesecake.
+000000307598 A yellow and green train is coming down the track.
+000000478286 a train on a train track on an elevated bridge
+000000215778 A laptop with mouse set up on a desk.
+000000099428 a laptop with some other electronics on top of it 
+000000248400 A man is pointing at a pizza box with a some words on the inner side.
+000000183049 There are books on all three shelves of this book shelf.
+000000099053 A bowl of food that looks like worms and some greens.
+000000295231 Several sheep standing around in the grass. 
+000000522007 A very small kitchen with small white appliances.
+000000466416 A building with two clock towers next to water at night.
+000000455716 Several motorcycles are parked on the green grass.
+000000248111 This basic kitchen has tools sitting on the floor
+000000375493 A woman wearing a yellow top and a jean skirt standing next to a yellow fire hydrant. 
+000000086483 A road sign outside of a building in a city.
+000000384670 A young man in a yellow shirt preparing to serve in an tennis match.
+000000326627 A city street with a rainbow in the background 
+000000102805 A man in holds a Frisbee in the parking lot by the river.
+000000478136 A plate full of tasty looking assorted baked items.
+000000125072 Five cows are grazing at a lake next to a road.
+000000211069 Two giraffe moving very quickly in the woods.
+000000221017 A brown chicken standing on top of a lush green field.
+000000375469 Two surfers riding a wave back to shore.
+000000322844 A teddy bear is next to a goldfish toy.
+000000172083 A very colorful vanity in a public restroom.
+000000378453 A large herd of zebra gathering around a watering hole
+000000491130 a person jumping a snow board in the air 
+000000311303 A pepsi soda on the table with a tray of bread and vegetables
+000000411530 A LOT OF PEOPLE ARE SKIING ON THE MOUNTAIN 
+000000138819 A white sink in a small, tiled bathroom.
+000000474170 two elephants walking in a field near a tree
+000000203580 A sidewalk with sandwich board, table and umbrella.
+000000466256 A plate of rice, steak, and a garden salad.
+000000146363 The sun is about set on the beach.
+000000460841 a cat is rinsed off with soap and aqua
+000000379533 A zebra and two elephants are on a grassy field.
+000000491757 A cat sitting on top of a bed on the comforter 
+000000153217 a black cat standing in front of a motorcycle
+000000138954 The plate has sliced zucchinis, sliced apples, and diced meats.
+000000474881 The elk have horns and are eating grass. 
+000000376365 A glass of milk next to a round purple vase.
+000000293200 A group of people flying kites in grassy field.
+000000439623 A white fire hydrant stand on a side walk.
+000000169356 A young man performs a trick on a skate board.
+000000192904 A tray on a bed with food and drink.
+000000297427 A man holding a cheeseburger made out of donuts.
+000000183648 A woman pats an elephant as a couple men watch.
+000000521141 A shot of a road in a city with a no peddling sign.
+000000121417 A person sitting on a bench under an umbrella.
+000000140286 A white horse drawn carriage in front of a yellow building.
+000000449996 Some parked airplanes and one airplane in the sky
+000000486112 A elephant standing next to an enclosure getting washed.
+000000007816 A person on a motor bike travels around a sharp corner.
+000000079969 A tennis player jumps and stretches to return the ball.
+000000121586 A tv on the wall with a cat on the screen.
+000000253835 woman standing with backpacks on platform with train 
+000000471567 a giraffe's head at a zoo with a mountain in the background
+000000480936 A woman sitting in a chair with a plate of food on her lap.
+000000390555 a bunch of people are gathered for a photograph
+000000513524 a man riding a blue surfboard on top of a wave.
+000000303499 Two people riding on the back of horses near a tall building.
+000000053626 A group of people standing on ski's in the snow.
+000000575243 a man is walking in the rain with an umbrella held close to his head
+000000205105 A man throws a tennis ball and poses his racquet. 
+000000553094 A jet with smoke pouring from its wings.
+000000179112 Woman taking her picture in a mirror with her cell phone.
+000000578545 A pregnant woman is in bed reading a large book.
+000000494759 Two people are on a beach flying a kite.
+000000532690 A woman playing a Wii game with other people.
+000000146358 A couple stands next to a wedding cake.
+000000557172 A bathroom sink and mirror reflecting the shower and bathtub.
+000000506656 a couple of horses are standing in the street
+000000097230 Two elephants are eating some leaves by the road.
+000000054164 a man in a wetsuit riding a wave on a surfboard
+000000012062 A sheep in a cage looking at sheep fur. 
+000000127987 A bobble head is shown next to computer screens.
+000000289417 A small boy skiing down a snowy hill
+000000502168 The men wear hardhats as they work on the aircraft carrier.
+000000144114 A small red and white plane flying in the sky.
+000000523807 A hotdog with mustard and relish is being prepared in a food cart.
+000000229659 A laptop in a bright yellow coffee house.
+000000560371 A street sign is pointing towards 8th avenue and the other is pointing towards 22 3/4 street in the middle of the forest.
+000000283318 There is a sign showing a street running one direction
+000000006012 There are bananas around another piece of fruit. 
+000000196009 A couple of very large carrots compared to the size of a spoon.
+000000262227 A computer is lit up on the checkout counter.
+000000526751 A boat out of the water on the shore with other boats in the background.
+000000310862 A man is frowning while standing in an empty room.
+000000189775 The tennis player is ready to return the ball.
+000000326462 A red icing doughnut with sprinkles on top.
+000000437514 a wood bench is next to a tree
+000000430973 A wooden pan holding a pizza with cheese and olives.
+000000055167 A yellow and blue bus traveling down a road next to a park.
+000000464689 a building that has some stairs out front
+000000006818 A standing toilet in a bathroom next to a window.
+000000228214 A man in uniform is looking at his phone.
+000000260925 The cat is laying on the hood of the car. 
+000000153343 two big stuffed bears sitting behind a chained fenced
+000000332318 Cattle on a level pasture in a mountainous area.
+000000229948 Clydesdale horses pulling a wagon in front of a large combine.
+000000400367 A picture taken at slow shutter speed on a city street at night.
+000000115946 A street intersection with old buildings in the background.
+000000252507 A cross country skier on the slopes on a sunny day
+000000564091 A woman sitting next to a display on the ground of jewelry is talking on the phone.
+000000292082 an old photo of a little girl sitting on her dads lap 
+000000133000 A grey and blue train passing over a city area.
+000000487583 A cup and a plate with food on it on a table next to a toilet.
+000000064718 A male Tennis player is about to swing with his racket. 
+000000017182 A wooden teacher's desk with books stacked on it and a blackboard in the background.
+000000002157 A table topped with a cake covered in berries next to a plate of sandwiches.
+000000532761 this living room is painted yellow with beige couches
+000000421834 A large green bird feeder with a bird on it.
+000000000802 Kitchen appliances and cabinets as seen through opening.
+000000445846 An office kitchen with open windows and no food.
+000000575081 A young boy in his pajamas is playing a video game.
+000000284282 An old blender is sitting next to a printer.
+000000014038 A long room with balloons and arrows on the floor.
+000000542089 A bathroom is outfitted with brown and white.
+000000431545 A baseball runner slows as he arrives at third base
+000000096001 Several antique trinkets are arranged on a table. 
+000000067310 A man is in the air as he performs a stunt on a skateboard.
+000000179765 A Honda motorcycle parked in a grass driveway
+000000281759 Women in warm weather clothes with multi-colored parasols.
+000000054628 A wedding cake with flowers in a banquet hall.
+000000438955 a group of men are competing in a skiing tournament
+000000281693 an airline with airplanes from one world parked on the air field
+000000188689 Black and white photograph of boats pulled up onto the beach
+000000156076 three guys sitting at a table with two pizzas
+000000306437 A guy on a skateboard grinding on a short concrete platform.
+000000491613 A zebra standing by its reflection in a body of water.
+000000225184 Three sheep running from a dog in the grass
+000000102331 A person on a motorcycle in the dirt doing tricks. 
+000000058384 Three elephants in a cage at a green field.
+000000528399  A array of food consisting of soup, sandwich and dinner rolls
+000000376442 A black plush toy sitting on the lid of a toilet.
+000000338532 A giraffe stands in the middle of a sandy area.
+000000206135 There are four cows standing on the side of a road
+000000216739 A giraffe sticking his head in the sky next to a tree.
+000000304404 A batter swings at the ball during a baseball game
+000000424776 A silver jet is flying high in the sky above the clouds.
+000000037740 a desk with a laptop a monitor and a chair
+000000434479 Two mini figurines sitting beside a plate with food.
+000000078170 Two woman are sitting on a passenger train.
+000000335427 A bowl full of broth and pieces of broccoli with a spoon and fork.
+000000088848 Two pictures of a fire hydrant with two pictures of people.
+000000565776 A kitchen with a center white island topped with a bowl of greens.
+000000151000 A woman taking pictures on a busy street.
+000000346707 Picture of multiple different colored plants and fruit.
+000000468632 A group of men that are on a field playing frisbee.
+000000140420 A motorcycle is parked on a tree lined dirt road.
+000000378454 A man gets air catching the Frisbee near the water.
+000000008532 A man with a plaid hat, tie, dress shirt and glasses on.
+000000185250 A woman standing next to a  brown and white dog.
+000000138819 A bathroom has a sink, toilet, and trash can in it.  
+000000431848 A man is doing a trick on a sidewalk curb.
+000000195754 A very big country style living room with a wagon wheel.
+000000198960 Two sheep and a ram stand next to a fence in the yard 
+000000234526 A boy holds out a leaf to a giraffe.
+000000372260 A car parked by a clock and some flowers.
+000000452122 A large aircraft is shown in the air.
+000000207728 A mother sheep and a baby sheep pictured here.
+000000434297 A plate with chicken, bread, broccoli and carrots.
+000000417779 There is a fireplug at the corner of this intersection in the residential section of town.
+000000265108 A balloon is tried to a bag right next to a luggage bag. 
+000000494869 An older image of a toddler, dog, and adult doing dishes. 
+000000186422 A brown bear walking along a river filled with rocks.
+000000145020 A giant Donald Duck standing in front of a Christmas Tree with lots of toys around it.
+000000160012 A pizza that is sitting on a wooden platter.
+000000165831 here is a large tray of vegetables sitting on a counter top.
+000000213033 An old man standing on the waters edge with an umbrella. 
+000000365095 Soigo'd is soccer player number 8 preparing to kick a soccer ball.
+000000512403 The case on the shelf is holding several feathers. 
+000000293625 Two women in glasses using Nintendo wii remote controllers.
+000000148508 A black and white picture of a busy street, only red is in color, a red double decker bus drives down the road. 
+000000079565 Two giraffes traveling across a rocky terrain with patchy grass.
+000000549055 A guy is splashing through the waves on his surfboard.
+000000223130 A giraffe reaching up to some tree branches
+000000407943 The shadow of a man's head is seen cast on an airbag.
+000000260657 a number of business signs in a tall pole 
+000000056545 A gray bird is standing on small brown branch.
+000000474028 A group of kids playing soccer wearing blue and orange uniforms. 
+000000214224 A pub sign for a small home bar has a crooked letter.
+000000334371 Double-decker buses sit at the curb in front of an old building.
+000000005586 Indoor tennis match, with a full crowd of spectators.
+000000190007 A man that is on a surfboard in the water.
+000000125211 A zebra showing the left side of itself on a field.
+000000415741 a hallway next to a bedroom doorway and a blue padded chair.
+000000135561 A kitchen illuminated by a track lighting unit. 
+000000067406 Two people standing in a field flying a kite
+000000480212 A smart phone sitting next to a canned soft drink and a pen on a wooden table.
+000000100428 there are two men that are pulling their ties
+000000281693 A large jet airplane being loaded with cargo at an airport.
+000000513484 A large brown bear laying on a rock.
+000000258911 Several cows that are lined up next to one another.
+000000015335 A man, woman, and boy are sitting at a table.
+000000221281 Giraffes in a dirt field with some patches of green.
+000000427338 Two men are riding motorcycles through some ruins.
+000000107554 A surfboard leans against a wheelbarrow on the lawn.
+000000570834 An officer watching over bicycles in a subway cart.
+000000259854 There is a billboard add on the side of the street.
+000000161820 A large bunch of ripening banana type fruit.
+000000338625 A city white bus stopped at a bus stop in front of tall buldings.
+000000145020 Daisy ducks stands at a Christmas decorated Disney World.
+000000076211 The side of the building has a large clock and several windows.
+000000371699 A dog lies near the table next to bookshelves.
+000000492937 A close up of a bicycle  parked on a train platform.
+000000001532 California traffic goes by road signs on an overpass indicating famous Hollywood streets.
+000000270883 A woman in lingerie is laying on a soft bed.
+000000313454 a man speaking at a podium during an event
+000000086755 THERE ARE PEOPLE THAT ARE ON THE SNOW SLOPE SKIING
+000000044699 Sheep grazing in grass in the mountains 
+000000252559 a number of people walking on a side walk near a building 
+000000101068 A baseball player holding a bat while standing on a field.
+000000425906 A person is taken in this very picture.
+000000286849 Four giraffes poking their heads out from behind a rock.
+000000186980 A wooden table with two chairs next to a sink.
+000000424545 A green-eyed tabby cat lies on a sunny window ledge
+000000425221 A plane is flying away from the airport
+000000196759 A Frito Lay delivery van parked outside in a parking lot.
+000000049810 A white and orange cat looking at itself in front of a mirror. 
+000000561958 many people holding skis and snow boards on a snowy surface
+000000084664 Dinner plate with potatoes, meat, an egg, and a chocolate-dipped strawberry.
+000000212573 Two people under an umbrella standing in the rain. 
+000000247917 A little league team wears orange shirts and black caps.
+000000551815 Two cats lay together on a blanket. 
+000000451043 A skier in the snow in a wooded area.
+000000074646 A beach area that has people on the sand and one person flying a kite overhead.
+000000489764 A woman holding a game controller with a man looking on.
+000000091654 An orange bowl filled with lots of noodles and beef.
+000000031050 The vase is holding an arrangement of dried vines. 
+000000006723 A building lined street with three lanes and light traffic.
+000000402346 A small bowl that has tomatoes in it.
+000000004795 A cat sitting in front of a computer monitor.
+000000502737 A person holding up a chocolate doughnut with a face drawn on it.
+000000216277 A table topped with a half eaten apple next to a smart phone.
+000000505573 A dog sitting on top of stairs.] with a red bow tie.
+000000561335 A couple of boats turned over in a forest.
+000000495732 Two men play Wii in a cramped living room.
+000000050896 a big bowl of oranges that is on the floor
+000000235399 Bull dog sitting on a furniture pad in a truck.
+000000303499 two men in red coats riding horses in front of a palace
+000000257566 A gondola like boat crossing over a bridge
+000000162543 A herd of elephants standing on top of a dirt road.
+000000265777 A woman sitting at a table with a pizza on a plate.
+000000493334 A large sheet cake with a frosted mobile phone.
+000000384527 there is a small table with tea cups and three chairs around it
+000000297343 A red stop sign sitting on the floor next to a wall.
+000000277584 A cat sitting outside a glass window. 
+000000368900 The front room of an apartment with the basics.
+000000079588 An old stone and brick building three stories high.
+000000398028 A baseball player holding a baseball bat during a game.
+000000359540 A spirited game of baseball takes place behind a chain link fence.
+000000514508 There are many people sitting inside a screen for love tv
+000000475387 An old train is pulled up to a passenger terminal.
+000000520659 People on a beach and a line of surfboards.
+000000133819 A large blue bus is driving down a crowded street.
+000000572678 A glass table sitting next to a couch and chair.
+000000382009 a tennis player on a court with a racket
+000000036861 A double parking meter on a pole with a bicycle rack.
+000000340930 Scissors with orange handles are in a container with pens.
+000000292415 A man riding a skateboard prepares to roll down a ramp.
+000000015956 A woman with a brown horse in a dirt area of building.
+000000353027 A person taking a slice of pizza to eat
+000000154339 A blurry image of a gauge on a pipe. 
+000000079144 Two young black bears playing at the zoo
+000000225670 A dog leaping into the air to catch a frisbee.
+000000364636 A blue eyed dog panting as he walks by.
+000000491008 The man and woman at the restaurant are holding a full pizza. 
+000000537241 A cinnamon roll with icing sitting on wax paper.
+000000084031 A group of baseball players are playing ball while a crowd looks on. 
+000000021879 there are many surfers that are in the water
+000000190236 an office with desk computer and chair and laptop.
+000000451150 Six donuts, all being a different kind of flavor. 
+000000298994 some different vegetables laid out on a wooden pallet
+000000458255 a woman laying in bed reading a book with a cat sitting on top
+000000233139 Two square clocks behind a window with Asian writing on it.
+000000496954 A serving slice of cake on a table.  
+000000269316 four men playing frisbee in a fenced park
+000000450686 The hand is holding an iPhone for the picture.
+000000503755 a person on a tennis court in tennis wear 
+000000448076 Girl trying out a game at a trade show.
+000000505942 Two people have a conversation at an intersection
+000000091406 Several women gathered together posing with 6 pizzas in take out boxes.
+000000532530 A map and street sign with building in background.
+000000538364 a police on a motorcycle on the side of the road
+000000537672 A black refrigerator freezer inside of a kitchen.
+000000334977 A new nokia phone has the features to muti-task
+000000003501 The meal in the bowl has rice and broccoli in it.
+000000461751 A man sitting on a motorcycle smoking a cigarette.
+000000022371 A man in shirt and tie at desk with laptop.
+000000388215 A young man is in blue and white playing tennis.
+000000367195 A Corgi dog resting on frizzy beige carpet 
+000000441286 A person riding waves in a body of water.
+000000538458 young men filming skating tricks on their skateboards
+000000500211 All of the umbrellas are folded up. 
+000000512248 A jewelry store sign that looks like a clock.
+000000402519 A bird perched on a branch of a tree. 
+000000192904 The tray on the bed has a pastry and two mugs on it.
+000000296224 theres a number 41 bus with a green stripe on it 
+000000087742 A vase filled with peacock feathers sits in front of the window.
+000000047769 A small living room with furniture and two windows.
+000000342186 A closeup of a pizza on a plate. 
+000000508482 The open doors of a public transport train.
+000000267670 A bathroom sink with a large walk in shower.
+000000250127 the girl is standing under a clear umbrella
+000000499031 A group of women pose in front of a sign hanging on a building.
+000000379842 A Nintendo Wii game controller sitting on top of instruction manuals.
+000000224337 A man with a piece of food in his mouth.
+000000229948 there are two horses that are pulling a carriage
+000000351589 A wooden bench sitting outside of a home.
+000000255965 A blue eyed tabby cat crouched on the ground. 
+000000273493 People playing tennis on a court surrounded by green hedges.
+000000455085 A large long bus going down a city street.
+000000295420 a blue rubbish removal truck and two vehicles behind it
+000000186422 The Bear is on the rocks has a fish in the mouth. 
+000000272416 A white toilet with a blue tank above it.
+000000413689 Two women holding umbrellas while standing on a city street.
+000000150265 People walking down a city sidewalk showing one with headphones.
+000000098392 A large clock tower in the middle of a street.
+000000176799 A town square that has a skateboarding ramp that a boy is using.
+000000190007 Man on surfboard riding large breaking wave at ocean.
+000000452891 A small white dog standing on a bench.
+000000581357 Skate boarder hang time with young man appearing to streak though the air.
+000000464824 A person is bending over to pick something off the ground.
+000000458755 A child places his hands on the head and neck of a sheep while another sheep looks at his face.
+000000424521 A boy jumping high in the sky on his skate board
+000000252219 City dwellers walk by as a homeless man begs for cash.
+000000227511 Several cars parked along a no parking area.
+000000516804 A street sign reads "Bus Stop Not In Use".
+000000458045 A young boy swings a bat during a children's baseball game.
+000000230983 A man in black shirt doing a trick on a skateboard.
+000000403353 Bedside with lamp, large bound book, and humidifier. 
+000000540932 Two people standing on a beach next to the ocean.
+000000486479 A large white dog laying underneath a bed.
+000000190753 A group of people sitting in a park doing various things such as playing chess, waiting for something with luggage, or just relaxing.
+000000522638 THIS IS A CLOSE UP PICTURE OF A STUFFED BEAR AND MONKEY
+000000263594 There is a car parked in the handicapped spot.
+000000210273 Buses and cars on interstate going both ways. 
+000000386134 Food is in a styrofoam take out container.
+000000500716 A man is looking through the handle of some scissors.
+000000415536 A bus is driving on a wet road with many green trees on the roadside.
+000000055002 A metallic toilet sitting in a small bathroom.
+000000206027 A slice of pizza that is on top of a napkin.
+000000579902 A motorcyclist is driving down the road next to the road barrier.
+000000129322 a bath room with a toilet and toilet paper 
+000000292908 A man is surfing on a wave in the ocean.
+000000016958 A living room with an old fireplace and desk.
+000000022705 The woman in the dress is holding her drink.
+000000568213 a man is throwing a frisbee in a field
+000000187585 A man who is riding on a skateboard.
+000000414673 Boys playing baseball, with one little boy at bat and parents watching.
+000000453634 A bathroom with a toilette with it's seat down.
+000000370208 The ten speed bike is attached to the pole.
+000000236721 Bananas, cereal, juice and other breakfast foods on a tray.
+000000302536 A man in a living room playing video games.
+000000459887 A steel strainer with four ripe banana's on a table.
+000000229997 Black bear outside with large birds and fallen trees.
+000000347335 A white plate topped with three different types of foods.
+000000106266 A plate that has been filled with meat and fruit 
+000000026941 A bunch of suitcases that are on a cart.
+000000140270 A person is riding a horse around on grasses
+000000093437 An old man standing in a kitchen posing for a picture.
+000000011149 Two parked bikes on a sidewalk with a person riding a motor bike.
+000000466986 A man cooks in a kitchen that is made of all stainless steel.
+000000460379 three adult giraffes in a field among trees
+000000138979 An SUV is driving past a city bus that is letting passengers on.
+000000156076 Guys sitting at a table eating pizza and talking 
+000000463037 A small passenger airplane sits in the grass at the airport.
+000000002592 a white mug  showing pirate skull and bones and a large knife on a counter top.
+000000384670 A tennis player grabs an extra ball from his pocket.
+000000225946 a green white engine is pulling some gray train cars
+000000439715 A man in uniform rides a horse down the street.
+000000547144 A laptop sits on a table next to desktop computers. 
+000000404479 A large airplane is ascending from the runway.
+000000562448 A bus driving down a street at night.
+000000447169 A crowded kitchen that is cluttered and messy.
+000000321790 A tennis player getting ready to play a game.
+000000337987 Colorful bird sitting on a branch of a tree.
+000000454750 two zebras standing in line with the head of one resting on the back of the other 
+000000441468 A red fire hydrant on the right side of a road. 
+000000298251 Several zebras stand in a field and graze.
+000000262938 Two stuffed dogs under a blanket looking at a picture book.
+000000223130 A giraffe standing under a leaf filled tree.
+000000177714 Plate full of cooked carrots, potatoes, and other vegetables.
+000000186449 there is an old woman sitting on a bench
+000000151857 An old train is going down the railroad underpass.
+000000304291 A horse walking in a field with trees in the background.
+000000084362 A young boy walking through a living room towards  a cat.
+000000096001 A table with an umbrella and other things
+000000148620 A home office with two computers and a picture.
+000000144114 A small airplane is flying through the empty sky.
+000000239537 A man in white shirt and shorts playing a game of tennis.
+000000171190 People sit around a table that is covered in liquor bottles.
+000000155051 a close of up a clock that makes the moon look small next to it
+000000019742 A painted glass in Front of some lights
+000000365766 A kitchen area with a stove, sink and microwave.
+000000567432 A red airplane jet flies over power lines in an empty sky.
+000000142324 Busy street scene in India including various modes of transportation.
+000000017714 Food on a train with a pie and some vegetable
+000000035682 A man sitting at a table that has plates with donuts on it.
+000000390902 A man intensely looking while holding a tennis racquet.
+000000493442 Skier in green coat plowing through powdered snow on slope.
+000000285894 A giraffe has come to a fence line to greet visitors.
+000000294695 People on a beach with a colored windsock and American flag colored windsocks.
+000000085911 The kite has sank halfway down into the water.
+000000110721 a plane with the attachment to let passengers on the plane 
+000000002149 A bowl full of fresh green apples are kept.
+000000130826 An older couple sitting on a public bench in a city.
+000000256518 A breakfast plate with toast, vegetables, eggs and meat.
+000000309452 A yellow and black bird is perched on a tree. 
+000000556000 A bunch of kids in a living room playing something.
+000000085911 A kite sitting in the middle of a large body of water.
+000000000285 A big burly grizzly bear is show with grass in the background.
+000000429761 Two tennis players are facing each other over a tennis net.
+000000011615 A bunch of street signs hanging on a pole
+000000231580 A man speaks to some children on a farm.  
+000000424162 A person who is on a bike with a dog on a leash. 
+000000279278 The man is riding his skateboard down the street. 
+000000449909 A boy looking at a giraffe laying down.
+000000410880 A large and over-sized stuffed teddy bear sitting in a chair.
+000000012576 Pizza and empty glasses on the dinner table.
+000000509258 An old brick building has Florida direction signs in front.
+000000348216 A dim room with toilet bowls lined along the wall
+000000055022 A pink bike in a bike shop with hardwood floors.
+000000034071 A close up of what appears to be two traffic machines. 
+000000530061 A child eating from a very large bowl 
+000000401991 two dogs laying in a bed with thick covers
+000000033114 The back view of an airplane on a runway.
+000000455157 A man sitting at a wooden bench holding a grey umbrella.
+000000351609 A bowl full of oranges that have been peeled.
+000000333772 Two cats laying on a keyboard in front of a monitor.
+000000554595 a close up of a person sitting on a beach next to a surf board
+000000069138 A stop sign gives traffic a frown face.
+000000319935 A living room with white furniture and a small wooden table.
+000000262682 A kitchen interior with olive green wooden cabinets
+000000008629 a close up of a slice of pizza on a plate
+000000522751 There are a group of traffic lights on a nice day.
+000000326627 a street view of cars and buildings with a rainbow in the sky
+000000262938 Two stuffed animal dogs reading a picture book about dogs.
+000000331817 A table with several dishes on it with a television.
+000000074209 A modern kitchen and with snacks set on the table.
+000000178744 A Chicago logo boat is traveling in the water. 
+000000293044 A plastic dish with the food sectioned off.
+000000034760 A bathroom with a shower, toilet, and sink.
+000000150930 a building with a clock coming out of it 
+000000581062 A man rides his skateboard near a drain pipe system. 
+000000128112 this is a man riding skis down a hill
+000000565012 A city street at night filled with blurry traffic.
+000000499266 A hand holding a cell phone using the internet
+000000104803 a toilet with glitter paint on it 
+000000060507 A baseball player hitting a baseball into the air.
+000000272136 The airplane is store in the hangar for people to take a look at it. 
+000000006771 A women in costume who is talking on a cell phone. 
+000000039477 Room with sofa, sewing machine, and potted plants.
+000000464144 A man on skies and a snow covered field with trees.
+000000338986 A photo of a train station with the train pulling in.
+000000558114 Basketball players walking on court with coach in wearing dark suit.
+000000182441 A man standing on a beach in front of the ocean.
+000000021465 A blue shelf holds different sized silver vases.
+000000045229 The top window in a kitchen is open.
+000000463730 A couple of people are crossing a busy street
+000000190756 MOTOR CROSS BIKE AND RIDER IN THE AIR DOING TRICKS
+000000414133 A small red sign that says whoa in a stop sign shape.
+000000043435 Two men ride surfboards through the crashing waves.
+000000420916 Three people are sitting on a bench looking out at mountains and a lake.
+000000002157 a close up of a table with many plates of food
+000000361142 A very tall building sitting along side of a street.
+000000355325 Several slices of meat are on top of a pizza.
+000000151516 A bird swooping down to the water to catch food.
+000000439773 A baby crawls on the floor beside a blue cell phone.
+000000066038 A man and a girl under an umbrella at the beach.
+000000382696 A kitchen in the process of being remodeled with wiring hanging from the ceiling
+000000252294 A guy peeks out from behind a curtain.
+000000173044 A very tall clock tower sitting next to a tall glass building.
+000000550797 A blue toilet that is underneath a towel.
+000000289741 A beach area with grass umbrellas and boats in the distance.
+000000376856 A clock is at the top of a tall brick building. 
+000000525155 A man in black carrying a sheep on his shoulders.
+000000286182 there is a man that is sitting at a table with lots of food
+000000064084 Sliced french bread is on a plat that is lying on a table.
+000000334719 A woman and kids on a beach with a large rocky area in background.
+000000357501 A pair of gold scissors laying on a bag of decorations.
+000000182202 A vodafone sitting on a table next to a Mac laptop.
+000000004495 A television, couch and chair in the corner of a room.
+000000095899 a construction site sits in front of a stop sign 
+000000339442 A sandwich and some grapes are sitting on top of a tablecloth.
+000000177357 this is a man riding a board in the water
+000000460967 A bus that is red white and blue with advertising on it driving down the street.
+000000163258 White bathroom showing toilet, shower, sink and mirror.
+000000248400 Man displaying open pizza box with message written on inside.
+000000446651 Young girl with helmet on hold on tight to her father driving the motorcycle.
+000000146457 Two little girls in the kitchen making  cheese pizzas.
+000000562448 A bus driving down the road really fast
+000000287667 A brown wooden table with a block of cheese with a carrot and two onion balls.
+000000463690 A man laying on the street playing with a soccer ball while people look on
+000000132408 Two large brown horses pulling a plow as a farmer guides them.
+000000527784 A counter with two plates with sandwiches and eggs on top of them.
+000000007278 there is a male surfer that is riding in the wave
+000000495448 A cake is shown with three tiers and flowers.
+000000404568 A seagull standing near the ocean on the sand.
+000000015338 a yellow bus and a white bus on a empty road
+000000171382 a man is doing skateboard tricks off the side of a busy street
+000000231822 Two sandwiches and macaroni and cheese and desert on a cafeteria tray.
+000000489091 A vanity and toilet are in a bathroom under a mirror.
+000000381360 A surfer steadies his move through the waves.
+000000367228 Young man flying a kite over a grassy area by water.
+000000095862 a number of baseball players on a field 
+000000579635 A man on a surfboard riding a wave 
+000000379453 The bottom view of an airplane flying in the air.
+000000407943 A man walking in front of an open umbrella.
+000000453302 The kitchen has a refrigerator, toaster, and microwaves.
+000000051309 A group of horses and a baby horse eat hay at a farm.
+000000523957 A guy is skateboarding through cones on a street.
+000000494863 A train next to another train next to a car.
+000000350833 A bathroom with a vanity with sink, mirror, toilet and tub.
+000000295478 A woman with headphones in walking her white dog.
+000000154431 There is a statue of a body in the room.
+000000092124 A bathroom with a white toilet and a sink net to a shower.
+000000270474 A child looks on in amusement while holding a baseball bat. 
+000000370478 Two pieces of luggage sitting near a dresser and laptop.
+000000190648 A room with holes in the wall and dirt on the bed looking utterly disgusting. 
+000000440475 A large living and dining room area with rich wood features.
+000000047740 A man laying on top of a sandy beach.
+000000330396 A man and woman stand on the beach and look at the ocean.
+000000558854 A table topped with a sandwich, chili beans and a container of fries.
+000000361142 A large building is shown next to a busy street with cars on it.
+000000521282 clear vase filled with white and yellow flowers with water
+000000062355 A red-shirted team playing a game of baseball.
+000000527528 Kites fly over the grass on a clear day.
+000000315450 Two large buses traveling down a busy city street.
+000000351589 A bench that is sitting in the dirt.
+000000214200 A stop sign vandalized with an "eating animals" sticker below the word "stop."
+000000571718 Bananas being sold and a man looking at them.
+000000308587 People are in the water with kites flying over them.
+000000289415 A pizza buffet with many pizza varieties of pizza on plates. 
+000000478136 A plate with frosted donuts and other pastries. 
+000000196442 A snow skier is being pulled by a rope overhead.
+000000199310 An old person is holding a tennis racket. 
+000000442456 a man in a red shirt sitting on the tire of a broken down truck
+000000022623 A close-up image of a multi media device.
+000000284991 A man is drinking water from a red bottle.
+000000481413 A small child is holding a frisbee while looking at the beach. 
+000000522638 A teddy bear and a stuffed monkey sit side by side.
+000000383337 There is a box filled with glazed doughnuts
+000000115898 A surfer is on the water and is waiting for a wave. 
+000000260470 A row of assorted baked goods is behind glass at a store.
+000000359677 A man holding on to a para sail handle being lifted into the air over the ocean.
+000000562243 A man in a purple shirt and matching tie looking pensive.
+000000093261 A man fixing a kite on the beach.
+000000545100 A man is taking a photo of a city with his phone.
+000000357060 Bird drinking water and walking on beach and ocean surface
+000000545730 a herd of zebras crossing a shallow part of a river
+000000226417 Several motorcycles are parked at a red light in the road in a city with tall buildings.
+000000456662 The young girl is standing in front of the refrigerator.
+000000393115 Nan or woman is enjoying a day of surfboarding on a brown/orange board.
+000000525286 A thermometer near a tiled wall beneath trees.
+000000045728 A plate of cooked noodles, a fork, and a knife is displayed. 
+000000011122 A gated grassy noel has a shorted stop sign stuck in part of the land.
+000000115946 The street light is green above a quiet city street.
+000000252559 There is a lot of foot traffic on this street in the city.
+000000497568 a very large airplane that is on a runway
+000000344268 the person addressing on a microphone and a flag behind 
+000000501368 A woman in a bathroom blow drying her hair.
+000000121153 a street sign in english and asian writing
+000000042178 A foggy view above trains on train tracks.
+000000520531 A large brown and grey cat sits on top of a desk.
+000000529122 The people are selecting their donuts at the counter.
+000000451090 A white sail boat moored in a lake near a mountain range.
+000000522007 The corner of a kitchen with a hanging plant.
+000000303818 A group of people walk alongside the road.
+000000447187 An umpire is in jump action while the pitcher launches the ball.
+000000569972 The man in swim gear is dealing with his object
+000000533816 some people talking to a referee on a tennis court 
+000000284764 Man sitting at a table holding eating utensils in his hands over his food. 
+000000196843 A school bus painted with museum information.  
+000000013923 A living room and kitchen areas features red, white, and wooden furniture.
+000000331604 A man kiteboarding over the ocean under a blue sky.
+000000182202 A flip phone sitting next to a laptop computer.
+000000154339 Gauges attached to pipes displayed in dimly lit area.
+000000089296 A man gets ready to swing at a pitch during batting practice
+000000413552 A toddler in a pink shirt sits on the lap of a woman.
+000000325991 A man gets ready to throw a frisbee in a park.
+000000411754 A man with glasses looking at his cell phone.
+000000561366 A cell phone on top of a calculator near a computer keyboard.
+000000292997 a car crashed against a light pole in the snow.
+000000154644 a woman smoking while talking on a cell phone
+000000205542 a teddy bear sitting on some kind of seat.
+000000466835 some banana trees bananas dirt and bushes 
+000000463802 A stuffed bear is watching a tennis player on television.
+000000201775 Men's restroom with toilet and urinals along tiled walls.
+000000517523 A 787 airplane is landing at the airport.
+000000092839 A large white polar bear sitting on top of a rock.
+000000407518 Bird sitting on top a chain link fence. 
+000000016228 A large white horse pulling a trolley car down the street. 
+000000097585 A mosaic of vases hung on a wall.
+000000136334 A person skiing next to blue check points.
+000000365886 A person posing with there cell phone with thumbs up.
+000000007888 A poll that has a clock on the top of it.
+000000039914 Just a little help from Mom will get his kite up in the air.
+000000567197 There are two old trucks parked next to the building.
+000000449432 Young servicemen and servicewomen stand next to a blue bus with luggage.  
+000000026941 a little green cart filled with assorted suitcases 
+000000140929 a black and white clock is hanging up outside
+000000007386 A tricycle sits outside of a garage. 
+000000019924 a woman with a tie smoking a cigarette with a big smile
+000000377497 Horse with odd striped head and white body in a sanctuary.
+000000481567 A man riding on the side of a red motorcycle.
+000000150930 A clock hanging from a store with letters instead of numbers on it. 
+000000468245 An unmade bed with pillows and a decorative head board.
+000000466156 The striped cat is sitting on top of the car.
+000000007816 A person on a motorcycles driving past a group of people behind a fence.
+000000257370 two people are playing a baseball video game
+000000029640 a pot of food on a stove with broccoli 
+000000546826 A pair of red scissors on top of a desk.
+000000437392 A pink think with a mop for hair and bulging eyeballs
+000000089078 Female snowboarder riding down a trail smiling at camera.
+000000492282 A couple of people who are on the back of a horse.
+000000505565 Is anyone familiar with this species of bird?
+000000106235 THERE IS AN IMAGE OF A LIVING WITH A RUG AND AN END TABLE 
+000000043581 A small ham and pineapple pizza on a plate next to a spicy pepper shaker.
+000000480985 A brand new motorcycle on display at a show.
+000000322844 A teddy bear sitting with a Finding Nemo stuffed animal on a bench.
+000000574810 A cat is seated in a window sell looking outside the window
+000000561465 Food on plates and a box of doughnuts. 
+000000182805 A woman and a dog sitting on the ground.
+000000161978 The collage shows many different sets of skateboards.
+000000404479 A large jetliner sitting on top of an airport runway.
+000000551804 A woman in a short skirt playing tennis.
+000000177489 A man and a woman sitting inside eating salad
+000000541664 A group of lego figures standing on a and near a laptop.
+000000099810 Small certain was child eating donuts off a blue plate. 
+000000414261 A baby bear that is laying on a rock.
+000000400803 a boat with some people on it drive through a water way 
+000000185472 The train is approaching and a man is getting off his bicycle.
+000000154000 A dump truck with a dumpster on the back
+000000404923 a profession baseball player holding a bat 
+000000364126 A surfer is in the middle of an ocean wave.
+000000551780 A analogue clock is places next to a sign. 
+000000441491 Three girls share a bite of pizza clowning for the camera.
+000000003553 A man riding a skateboard on piece of concrete in a park.
+000000462614 A bathroom has red walls with yellow accents.
+000000392933 A giraffe running across a grass covered field.
+000000475150 A giraffe walking near a tree with very few leaves.
+000000250901 A group of people getting their food from a bag.
+000000318238 two dachshunds and a cat sleep on a bed
+000000319100 Two people sitting together on a couch playing wii
+000000558073 A tiger cat sitting in a white bathtub.
+000000539445 A large red electrical passenger train riding down railroad tracks.
+000000442822 Several boats in the water docked at a marina.
+000000505638 An old fashioned train engine caked up to a caboose.
+000000088462 A yellow school bus traveling down a street.
+000000203580 A colorful umbrella sitting outside of a pink building.
+000000311950 A hot dog sitting on a table covered in confetti.
+000000458223 A man with a bike walks towards a red lighthouse.
+000000110449 Two people playing with an interactive gaming unit.
+000000534639 A public transit train at one of its stations.
+000000248631 A laptop with a stationary mouse attached to it.
+000000119911 The boy plays with his skate board in the street.
+000000377000 A cat sitting in a window watching the rain drip on the window glass.
+000000544519 An Asian child brushes their teeth with a blue toothbrush.
+000000076547 the man is working on his laptop while he waits for the train.
+000000275392 Woman sitting ontop of a large brown and white horse.
+000000489091 A white bathroom that has been newly cleaned
+000000183709 a group of men walking across the street
+000000482970 A laptop sitting on a desk with a chat window displayed.
+000000554266 A series of photos showing a person laying on a bed.
+000000212800 A crowd of people riding on the back of a red and white boat.
+000000009769 A truck with a snow plow attached to the front.
+000000244411 Two cows in a large green grassy field.
+000000460683 A man holds out his tie, which has writing on it.
+000000267169 An elephant drinking from a white square bottle.
+000000022935 A girl chasing after a soccer ball with an opposing team member close behind
+000000144003 A group of people watching a woman cutting a cake.
+000000093353 A close up of a sliced hot dogs placed in a sandwich.
+000000297681 A car is driving through a flooded street. 
+000000314294 An elephant walks alone past some big rocks boulders in an open field
+000000573943 A parking lot with various trucks in it next to a building.
+000000409867 Two cats are sitting in a window sill. 
+000000419974 A dog sits waiting while his owner cuts some meat.  
+000000460927 a big brown bear standing in an open field next to some rocks
+000000244411 A couple of black cows standing on the top of a grassy hill.
+000000426376 A person snowboards on top of a snowy path. 
+000000126107 A bird that is flying over the sand.
+000000110359 Commercial jet passing overhead on bright cloudless sky.
+000000114049 People with luggage on an airport transfer bus.
+000000047585 this is a photo of a man and wife on there wedding day
+000000125572 a woman riding a bike in the street and a building with a clock in the background
+000000049269 A horse in a pasture of tall grass observing an energetic dog.
+000000167067 A man is posing for a camera in a room.
+000000088951 Dog and his human walking partner relax in a park.
+000000018737 A motorcycle is sitting next to a beach.
+000000544811 Two large birds are flying over the calm water. 
+000000532481 A guy is waterboarding in the ocean on a windy day.
+000000229553 Helmeted young skateboarder displaying ability skills on cement surface.
+000000571313 A small room with different types of furniture and appliances that it makes it hard to tell what type of room it is.
+000000507037 Individauls gathered outside a building and an American flag at half mast.
+000000523175 A bowl that has soup and a spoon in it.
+000000177934 Two young women are washing two motorcycles with hoses..
+000000404601 A yellow and black sign for a restaurant and lounge with red arrow pointing downwards.
+000000176901 Skiers on a ski slope; one laying on the ground.
+000000302990 A person with a surfboard in a room.
+000000565962 A cat looking at a buddha statue through a window.
+000000272364 four brown and white giraffes eating greens together
+000000201072 there is a male tennis player playing on the court
+000000509403 A family outside in a yard playing frisbee with their dog. 
+000000348488 A decorative vase with several black horses and a rider. 
+000000301563 a man skiing down a hill by himself 
+000000189806 A cat looking at a dog as it sniffs the floor.
+000000416170 a black and white cat sitting by a window and looking outside
+000000468632 Some people playing with a disc in a field.
+000000052007 A bus stopped at a bus stop, letting people board.
+000000002431 A meal of burnt toast slices with condiments on the side.
+000000445248 A group of elephants walk around within an enclosure.
+000000085911 Kite with a person's face designed on it is floating in the water
+000000109992 A person riding a snow board down a snow covered slope.
+000000334483 A group of women standing around a cake cutting slices.
+000000300341 A couple of people standing in a room with some remotes.
+000000445999 A woman with foam coming out of her mouth, brushing her teeth.
+000000279730 sandwhich with everything on it and a coke
+000000553664 the red laptop is sitting on the table with headphones
+000000473118 A person kicking up on their skateboard at the top of a ramp.
+000000312549 There are two white street signs showing directions
+000000089761 A toilet seat sits in front of a tree.
+000000213935 there is apples oranges and pears on the table
+000000050380 A man holding a horse, so a little boy can take a ride. 
+000000455352 a bell tower with a clock built inside of it 
+000000524850 A crowd of people standing around a bus next to a parked jetliner.
+000000311394 A small baby is holding a white and blue toothbrush.
+000000066635 a desk top with a monitor and keyboard on it
+000000214200 A stop sign with a eating animals sticker on it.
+000000220764 A person holding a pole in each hand in the snow.
+000000324258 Adults playing in living room using remote controllers.
+000000231169 Locomotive engine and cargo cars traveling on train tracks.
+000000017899 An older woman preparing cookies and bread at a table.
+000000495732 Two men playing video games on the Nintendo Wii.
+000000028449 A herd of elephants is walking together down a dirt trail.
+000000441586 A man on a beech vehicle making his way thru the sand.
+000000180798 A smart device sitting on top of a blue blanket.
+000000014007 A dark colored cat that has a collar on it, sitting o the top of a blue refrigerator.
+000000293324 Multiple commercial passenger airplanes at their gate at an airport.
+000000365385 Fancy colored toothbrush sitting in decorative mug, in a bathroom.
+000000134882 a cat in a bedroom laying on a striped covered bed.
+000000233139 A display of clocks inside a shop window.
+000000130566 A train pulls past an intersection in the rail in a rural area.
+000000229216 two ladies sitting down looking at a cell phone
+000000321790 A young man holds up a tennis racket.
+000000119911 A boy in white jersey doing a trick on a skateboard.
+000000025603 A girl with some food and drink at a table.
+000000353180 A group of people who are pushing a school bus.
+000000021503 Crackers with spread on table with remote in background
+000000130613 A picture of cooked fish in a tray.
+000000107087 A cat inside of a car during the day.
+000000570736 The bathroom features a vanity sink and tub/shower combination.
+000000436738 A white and green bus driving down a street.
+000000177015 A man using his laptop computer while a cat sits on his lap.
+000000514540 Both streets of the intersection are marked as one way.
+000000169169 People are walking around a plaza that has a sign that reads "Spring in the City".
+000000560256 Many types of fruit on display on a street.
+000000084674 A young gir holding a child with a boy standing next to them
+000000155291 The cat on the windowsill is relaxing in the sun.
+000000154425 A boy holding a kite while standing on a sidewalk.
+000000386912 The woman is working on her computer at the desk.
+000000234366 a building with a short clocktower in the front of it 
+000000108864 a open field with large elephants standing in it.
+000000115898 A man in black wetsuit on white surfboard riding wave.
+000000186422 A bear is coming out of the water and onto the rocks. 
+000000105335 A white bowl filled with meat, vegetables and broth.
+000000231879 A baby who is being held by a man.
+000000358427 Two pieces of broccoli being held up on a fork
+000000466986 A smiling man is behind the counter at a restaurant. 
+000000459396 Cow eating grass while a bull watches nearby
+000000320632 A herd of zebras standing in close proximity to each other.
+000000230819 A wetsuit clad surfer catching a wave in front of a pier.
+000000012280 An airport with a woman carrying a piece of luggage.
+000000289992 A plain, white bathroom with a sink and a tub.
+000000345385 Two man spooning in a bed with a brown duvet.
+000000572408 A cow standing in a field with a smaller cow.
+000000179642 A man without a shirt sitting in front of a fan.
+000000093353 A piece of bread with a hot dog in the middle is shown.
+000000290179 An aerial view of the city of London on a cloudy day.
+000000242287 a bike leaning on a door of a building
+000000256916 Pizzas are sitting on trays on a table.
+000000392228 a clock on a building with a sky background
+000000125806 A cow and another playing next to a tree.
+000000103585 A clean bathroom with mirrors, sinks, and a toilet.
+000000290592 A cluster of black and white sheep hang out by a fence.
+000000492937 The bicycle is locked to the metal pole as the young man waits to board the mass  transit train.
+000000130699 Several men are playing sports on a field near some trees, wall, bus, and several buildings in the background.
+000000267537 A group of young women sitting next to each other under an umbrella.
+000000426372 A white truck with a red car on its bed.
+000000196141 A baseball player taking a swing at the ball
+000000346905 A little elephant stands between two large, tusked elephants
+000000525322 Two jets are flying against a bright blue sky.
+000000141328 A big bowl filled with some pasta and meat and veggies.
+000000537355 A white building on a green lawn with trees and a fire hydrant.
+000000355257 A young girl lying in a hospital room hooked up to machines
+000000166521 A tv on a tv stand in a room at Christmas time 
+000000038829 A man riding a bicycle with a boy on the back of it.
+000000548524 A surfer stands riding down the side of a small wave. 
+000000106563 A station hallway has a glass ceiling and clock.
+000000190140 A man driving a boat with his dog looking out over the bow.
+000000210388 Man and woman walking up a snowy mountainous path, with luggage.
+000000046497 LITTLE GIRLS ARE ON THE EDGE OF A BOAT 
+000000551820 A group of people on a court playing tennis.
+000000424162 a couple of women that have a dog on a leash
+000000296222 A man wearing sunglasses while holding a plate of pizza.
+000000002431 A tray of food on the top of a table.
+000000512330 n open refrigerator door exposing all the food inside
+000000478136 A plate of assorted donuts, muffins, and danishes.
+000000229849 A tennis player is jumping and reaching to hit the ball.
+000000212559 a close up of sheep on hay ground indoors
+000000012120 A couple of women playing a game of tennis on a tennis court.
+000000476787 There is a pizza on the cutting board
+000000502599 Old and new airplane models handing inside a building.
+000000405972 An adult and two baby elephants standing near a fence.
+000000232646 A snow covered road with a stop sign next to it.
+000000184611 People riding and pushing tricycle carriages down the street.
+000000008532 A man that has glasses and a hat.
+000000088485 A boy is throwing a frisbee outside a building.
+000000239773 Baseball player in blue and white uniform swinging at a ball. 
+000000199977 A biplane leaves a smoke trail while doing a trick.
+000000535578 Sheep graze in a lushly green mountain meadow
+000000560911 A man sitting on top of a couch holding a smart phone.
+000000025986 A bowl filled with salad and plates of food.
+000000068409 A  group of boys in suits holding a chalkboard.
+000000409198 Man on a marked field holding a frisbee in his hand. 
+000000022623 Closeup of a cellphone device with various buttons.
+000000511760 Adult showing little kid how to fly the kite
+000000310980 A stuffed animal is in front of a notebook computer.
+000000175251 A toddler in the bathroom brushing his teeth
+000000250137 A woman holding an umbrella while standing next to a crowd.
+000000370208 This is a bike locked to a parking meter.
+000000126137 A young man standing on top of a tennis court holding a racquet.
+000000446207 A camera is placed next to a cellular phone.
+000000496571 A woman taking a picture of a man eating.
+000000417285 Three images of a pie and a cup of tea or coffee. 
+000000484029 A street signs is on top of the stop sign
+000000574520 A guy is showing his prowess while surfing the waves.
+000000361551 Some are standing outside a building with suitcases.
+000000304545 A black suitcase with a handle and white background.
+000000278749 Many people walk on a sidewalk as a man holds a restaurant sign.
+000000371552 A boy standing by a soccer ball on a field.
+000000281754 A beautiful young lady in short shorts sitting next to a little boy.
+000000079408 A stop sgn with a branch in front of it
+000000320696 Surfer riding out end of wave with large breaking wave in background.
+000000370478 some luggage and a back back sitting on teh ground next to a lap top
+000000493799 A plate of food containing carrots, potatoes and meat.
+000000305309 A kid hold a stick in his hands in the air.
+000000024243 Two ladies seated with cups of drinks, one smoking while the other seated inside a fridge
+000000530457 Some pretty flowers sitting on a hearth next to a fireplace. 
+000000330554 Blender with portions of its contents spilled around it. 
+000000528705 A stuffed animal with a smaller one on its back.
+000000080666 A cat sitting on top of a bench in a field.
+000000278353 A small sheep is standing under a wooden fence post.
+000000320425 a couple of giraffes are standing in a field
+000000199771 A woman teaches a class of aspiring chefs. 
+000000129062 Parking meters are in front of an eating establishment. 
+000000563758 A stuffed bear with no mouth wearing a red handkerchief.
+000000370375 A little boy wearing a baseball hat holding a baseball bat.
+000000278848 A group of people standing outside of a bus.
+000000430875 A double traffic signal shows three green lights.
+000000371552 a boy with a green shirt and a black pair of shorts playing soccer
+000000500613 A fire truck sitting in the stations park lot.
+000000479953 A soccer playing getting ready to kick a ball during a game.
+000000384808 A man is taking a picture of himself in the bathroom using the mirror for a reflection.
+000000050679 An orange lying near the white stripe of a highway.
+000000499181 There are many cars that are parked in the parking lot
+000000098716 A man lounges on a bench watching the ocean.
+000000384949 BEAUTIFUL SCENE OF THE WATER, THE BRIDGE, AND A TALL CLOCK TOWER
+000000232692 The man is about to perform a stunt in the water.
+000000404534 a fence some buildings and some different signs
+000000151480 Members of a baseball team are walking across a field.
+000000145665 A  person is lying down in a parking space.
+000000373382 A tennis player has their racquet raised in the air.
+000000318455 Many cupcakes and half of a cake sit on a table.
+000000458992 a lady using a knife and fork to cut a pizza
+000000170545 A pay phone sitting on the side of a street.
+000000477118 A metal pole covered in different street signs.
+000000080932 I young boy eating a large piece of pizza.
+000000309678 There is a tray filled with sauce covered food
+000000308466 A picture of a white bathroom with a sink,toilet and bathtub.
+000000098287 A group of people are playing baseball on a field.
+000000521231 A polar bear standing by rocks in a grassy habitat.
+000000018193 a person eating an apple with a plate of food
+000000262440 A bathroom with lighting and a toilet and other things.
+000000370999 A dime sits in the middle of a group of stones.
+000000284743 a close up of a person holding a bat in the dark
+000000440507 a blue suitcase sits on the floor in front of 2 pair of shoes
+000000285788 Two giraffes and an ostrich at a zoo enclosure habitat.
+000000530162 A man holding an umbrella for other people. 
+000000007977 A teenager is doing a wheelie on a skateboard.
+000000362520 A man flying through the air on top of a skateboard.
+000000192964 a male skateboarder in a white shirt is doing a trick
+000000261982 A person is riding on a skateboard on the street.
+000000398652 A large glass table topped with different types of plants.
+000000008844 a black woman standing over a bushel of yellow bananas
+000000023666 A very old bathroom has dusty toilet and clawfoot tub.
+000000357742 A man is cross country skiing in the snow.
+000000109118 A trio of men in baseball uniforms posing for a picture.
+000000019042 a bird is sitting out on some rocks in the water
+000000025603 A girl in a Cleopatra outfit eating pizza at a table.
+000000550714 The little  girl is eating a banana.
+000000553094 a fighter het flying up in the air 
+000000140556 these man are standing near a fire hydrant
+000000293324 A group of planes near control tower in an airport.
+000000561009 A bird that is perched in a tree.
+000000368335 A horse walking between the parking spaces of two cars.
+000000186449 A black and white image of an old person sitting on a bench
+000000092053 a couple of plates with food inside sits next to each other 
+000000571943 A picture of a street light and sign showing the street names.
+000000272566 A man and young girl surrounded by various fruits.
+000000507223 A group of baseball players standing next to each other on top of a field.
+000000537802 A set of plush toy teddy bears sitting in a sled.
+000000133233 A bunch of boats docked by the mountains
+000000104619 a small giraffe walking on a dirt ground 
+000000535608 some people sitting and standing on the beach
+000000457884 Three children are playing baseball outside in a field.
+000000402720 A group of people that are sitting around a table.
+000000538067 People in a rural area overlooking a mountainous region
+000000230993 Two women with open umbrellas walking down a street.
+000000027768 A white and red bus driving down a street.
+000000339823 A woman holding a umbrella in the middle of the day.
+000000101022 Two signs that are on a pole in the snow.
+000000246963 A group of men riding on the back of motorcycles.
+000000133087 a street corner in a town all bright from lights
+000000401446 A woman wearing a dress and holding an umbrella.
+000000374551 A bird standing on a tree branch by itself.
+000000135890 The Big Ben clock tower towering over the city of London.
+000000532493 A woman surfer riding the waves in the ocean.
+000000180383 A cow sculpture with a TV sticking out of it's center.
+000000459757 A giraffe hangs its neck low in a large field.
+000000310072 a metal bench resting on a sidewalk with cars on a street in the background
+000000303653 two ladies riding horses there's a reflection of one of them in a mirror
+000000105249 A very small bathroom with only the essentials.
+000000025057 A man throwing a frisbee while holding a beer
+000000239318 a laptop on a desk with an extra keyboard
+000000080659 horses racing on a race track with jockeys.
+000000401250 A skier performing a backflip with the sky as background
+000000377239 A dark skinned woman posses beneath an umbrella. 
+000000160728 The man stands next to a child standing on a canoe with a paddle.
+000000030828 Person sleeping on bench in sleeping bag near parking meters.
+000000549390 An older woman rides a bicycle down the street.
+000000467776 A baby cow and a large cow standing next to each other. 
+000000406570 some apples oranges and another yellow and green fruit
+000000558421 A train is rounding the train tracks next to the trees.
+000000277689 Lavishly decorated cake and empty drinking glasses await their tropical guests
+000000201426 Two young boys sit at a table with pizza.
+000000094157 Large clock tower sitting in front of a park. 
+000000229221 a black and white cow is standing in a field
+000000261706 A big cat laying down on a remote.
+000000286523 A man talking on his cell phone with his other hand in his ear.
+000000139872 A dog laying in the grass with a frisbee near his arms. 
+000000312263 A very tall building with lots of windows and a clock mounted to it's side.
+000000565853 Individual having an astounding time and getting a charge out of some life. 
+000000222863 A large cow with horns by some bushes.
+000000532855 A man riding a skateboard on the side of a  metal rail.
+000000474452 a no parking sign at some sort of shopping mall.
+000000550471 A slice of orange laying on a mostly clean plate.
+000000006213 A bathroom has two sinks, a bathtub, and a shower.
+000000203317 A red bicycle leaned against a red stucco building with a graffiti covered door. 
+000000255165 a desk with keyboards and a laptop and a pair of monitors
+000000116825 A cat leans over a desk and paws at an army knife. 
+000000034071 Two parking meter with the sunshine in the background.
+000000498919 The fruit are piled up on display in the store.
+000000368982 Some scrap book scissors are on a brown table.
+000000352582 A man throwing a white frisbee in a tree covered park.
+000000301718 Two people sitting at a table eating.  
+000000455555 A young boy riding on top of a skateboard.
+000000205514 A flat screen mounted to a wall in a living room.
+000000251065 A white toilet with a black seat sitting in a small stall.
+000000032610 many laptops  on a wooden table on being in use
+000000264535 An old man reading a book on a park bench.
+000000404249 A kid in a camo shirt riding a skateboard down the street.
+000000519522 A clock tower on top of a tall historic building.
+000000457262 A wooden table with miniature bananas next to a Malaysian coin from 2009.
+000000032941 A yellow sign sitting in the middle of a road.
+000000529966 Two bowls of oranges are sitting on a metal surface.
+000000548267 There is a herd of sheep grazing on the grassy field. 
+000000310980 A person who is sitting and looking up at a computer monitor. 
+000000568213 a young man attempting to throw a Frisbee past a defender 
+000000116479 A poster bed with canopy in a room. 
+000000051738 A white bed in a bedroom next to a table on a lamp.
+000000153797 A baseball player throwing a pitch onto the field
+000000400815 A man is sitting in the flatbed of a truck.
+000000334309 A man taking a selfie while eating pizza.
+000000063154 a person lying on a surfboard in the ocean.
+000000107087 A black cat sits in a car and looks out.
+000000497568 An airplane sits on a runway in a airport
+000000293300 THERE ARE ELEPHANTS BEHIND A FENCE AT A ZOO
+000000157601 A woman eating a sandwich with a drink in front of her.
+000000319369 A picture for rentals with two women walking up to them.
+000000275198 A family having lunch in their backyard in the rain.
+000000028285 A large white and blue clock tower sitting over a roof.
+000000373353 People crossing in a crosswalk in front of traffic in the city.
+000000362716 A guy in a green shirt is preparing to hit a tennis ball.
+000000448810 Number of street signs in front of a big building.
+000000563267 Two young men standing around a table in front of a chalk board.
+000000237517 Food set out on an outdoor picnic table.
+000000222458 People walking through the snow past park benches.
+000000307074 A traffic light hanging over a road at a cross walk.
+000000032081 A woman swings her tennis racket while reaching upward.
+000000087875 A fire hydrant that is sitting in the grass.
+000000002685 A group of people standing around each other.
+000000476787 The pizza is ready to go into the oven.
+000000458054 A row of toilets with broken seat tops on the ground.
+000000465129 A lady in a bath robe touching something near the ceiling.
+000000386879 Woman in red outfit holding a tennis racket in her hand. 
+000000191761 A dish with some orange slices and something else.
+000000368038 A colorful train sits on the tracks in the sunshine.
+000000130465 An old church building with clocks built into the tower and an arched doorway. 
+000000447169 A kitchen with fish tanks and many many cabinets is shown.
+000000539883 A television that has been placed outside, a video game is being played on it.
+000000509131 Apples and oranges and a tree with bananas in front of potted plants.
+000000061418 a couple of kids playing a game of checkers.
+000000369675 A white and red train at a train depot with passengers sitting inside.
+000000327780 A fork cutting a piece of cake on a plate
+000000236914 Two teddy bears sitting at a table with honey
+000000398438 A couple at a wedding cutting their wedding cake together.
+000000100582 Lettuce and tomatoes on a slice of pita bread.
+000000493284 A group of elephants that are walking together. 
+000000554838 I do not know what this is supposed to be..
+000000425390 This cat has no intention of giving up the computer keyboard to a human.
+000000039480 Tennis player smiles on the court while holding her tennis racket. 
+000000250282 The large group of children are outside posing for a photo. 
+000000197528 A cat sitting on a window sill behind mini blinds.
+000000088250 Two elephants locking noses on a dirt surface.
+000000535253 A shopping cart full of food that includes bananas and milk.
+000000569273 A yellow fire hydrant that is in the wilderness. 
+000000269314 A cat sitting in a tree, staring at the camera.
+000000039785 A surfer is surfing in a wave simulator.
+000000140439 A clear vase sits on a table and holds long stemmed tulips.
+000000165257 A large black counter top sitting next to a sink.
+000000287649 A cat laying on top of a wooden computer desk.
+000000519039 A truck drives in the middle of a neighborhood road.
+000000347544 There is an old multi colored volvo parked next to a motel
+000000172547 A grassy hill side with some animals in the distance 
+000000513041 two people some drinks some plates and a small pizza
+000000420230 A large elephants walks along toward the camera.
+000000189310 a living room with a large door way and a chair
+000000328683 The parking meter is decorated in different colors.
+000000274219 Talk about a bad hair day, his is frightful.
+000000149406 A motorcycle just outside a building with green, peeling paint
+000000434230 People in a shop looking at a wide variety of pottery.
+000000463199 A group of men standing on a subway with other people.
+000000304984 a couple of sandwiches are on a white plate
+000000398810 A small cat sitting on top of a couch looking out a window at another cat.
+000000263969 a bride and groom couple cuts the wedding cake 
+000000138954 A white plate on top of a table topped with fruits and vegetables.
+000000542073 A light pole that has a street sign.
+000000032887 a person with a backpack and a walking stick 
+000000356498 a plate of polents a tomatoe and brocolli
+000000122672 A downward picture of a surfer catching a wave.
+000000067896 a big clock a rod iron roof and a round window
+000000378099 Looking down at a computer keyboard and mouse
+000000377575 A dark  colored poodle is tied to a post on a sidewalk.
+000000401862 PEOPLE WAITING IN LINE TO GET FOOD FROM A FOOD TRUCK
+000000572517 a polar bear following a big winged bird
+000000164883 an image of a guy making a video game character
+000000015660 A windsurfer watches people kite surfing at the beach
+000000298396 Kitchen area with counter top space and dining room table. 
+000000472030 A lush green forest filled with lots of trees.
+000000426836 THIS IS A BREATHTAKING PHOTO OF SOMEONE FLYING A KITE
+000000523782 A cardinal perches on the side window mirror of a parked sedan.
+000000127092 A row of buildings and a city street on a hill that is pretty steep.
+000000234779 Two sloppy joe sandwiches, each on it's own plate.
+000000416837 An animal grazing on a grassy hill near mountains.
+000000192904 A tray with coffee and a pastry on it.
+000000073153 A person talking on a phone with a tree in the foreground.
+000000513181 A ship sitting at port on the water front.
+000000148508 black and white picture with a single red bus and a crowd of people lining the streets 
+000000579655 A woman sitting in front of several glasses, talking on her phone. 
+000000384808 A man in blue shirt taking a picture in bathroom mirror.
+000000234807 a number of horses in a field near a building 
+000000341681 A tennis player lunges to return a wide serve
+000000423519 The red bus is driving down the street.
+000000367386 a living room with some tables and couches
+000000459272 A woman in yellow scarf holding a cellphone.
+000000429281 Display for bananas with a euro money sign
+000000187144 Passengers standing outside a green and white tour bus.
+000000166426 Beef and vegetables on a plate sitting on a table.
+000000032285 You can see into part of a bathroom with the linen closet on the outside.
+000000473821 A red couch next to a brown chair in a living room.
+000000555597 Historic buildings cover an area with architectural elements alongside the street.
+000000432085 Without using hands, a man tries to bite a pair of sandwiches.
+000000298396 A kitchen filled with appliances dishes and a bowl of fruit.
+000000100582 Small personal pizza with bacon and spinach. 
+000000426166 A light blue bicycle chained to a pole on the sidewalk in front of a red building.
+000000320425 Ten giraffes loitering around a building out in the day.
+000000057027 An adult elephant is pictured walking with a baby elephant.
+000000177935 A white stove in a very small kitchen.
+000000577959 A man flying a kite at the beach near the ocean
+000000376856 A clock is seen at the top of a very tall building.
+000000210789 A couple of people standing with a umbrella.
+000000243204 A man is cooking while wearing a jacket and scarf
+000000189310 A pastel green room with pictures mounted on the wall and a red chair near a large window.
+000000529762 A man is taking a picture of his food
+000000056288 A half sandwich sitting on a paper plate located on a table
+000000183965 A tray holding a sandwich and cappuccino, next to the pastry.
+000000537991 A person holding a phone in each hand and wearing a head set in front  of a Christmas tree.
+000000471869 A beige teddy bear is sitting on a chair.
+000000000872 Two guys playing baseball, with trees in the back.
+000000394940 a kid eating a cake on  table looking around
+000000258911 A woman in a straw hat takes photos of brown cows inside a barn.
+000000546829 Cute white fluffy dog standing on an old wooden bench
+000000227898 A man and lady taking a horse and carriage ride.
+000000282037 A full view of some baseball players in the field.
+000000287714 A white bathroom sink sitting under a bathroom mirror.
+000000248616 there are three pictures of a man hitting a tennis ball
+000000321557 A motorcycle parked next to a white building.
+000000200162 A train driving down the tracks near a platform.
+000000476119 A man with a helmet on riding a skateboard in the street. 
+000000402473 Two different color cats laying beside each other
+000000000139 A living area with a television and a table
+000000324927 A plate of two slices of pizza and a cup of juice.
+000000248111 A white refrigerator freezer sitting inside of a kitchen.
+000000565597 A pizza sitting on a table, with a couple slices on it.
+000000063965 A woman holding a plate of cake covered in frosting.
+000000336209 A man riding a skateboard up a wooden ramp.
+000000144984 A child in a room with a remote in hand.
+000000344611 a street sign mentioning water on the road
+000000482735 The waves crash behind a man surfing in the ocean. 
+000000002261 A man on a blue raft attempting to catch a ride on a large wave.
+000000016451 Several items including two surf boards on the beach
+000000580757 A fire hydrant painted exactly as the patriotic flag
+000000523194 A fire hydrant and a street sign are on the side of a street.
+000000512248 A large clock on a pole near a building.
+000000356347 A wooden handle in a pot with soup or stew next to an empty bowl.
+000000329080 The hotel guests have messed up their room.
+000000047828 A bridge in the distance with multiple different colored lights.
+000000425227 People on a beach fly a kite near the water
+000000555412 A close up photo of a pizza with a lettuce and prosciutto topping. 
+000000265816 A horse is standing with a black carriage
+000000084650 a cat laying on a luggage bag on the ground
+000000276707 A sign that is sitting on a wooden pole.
+000000288430 A man plays with his family on the beach 
+000000159791 A refrigerator that is covered in magnets from top to bottom.
+000000194832 The inside of a commuter city bus with palide seats
+000000125472 a man in the air standing above the skateboard
+000000131938 A cat is wearing a tie and sitting on some clothes.
+000000213171 A man in black shirt standing in field with baseball mitt.
+000000536073 Blender on a counter with alcohol and a lime.
+000000177489 a man and a woman sitting at table
+000000132931 Two baseball players standing next to a wall on a baseball field.
+000000465585 The broccoli and cheese dish was sitting in a tin pan.
+000000386879 Lady wearing a red tennis outfit holding a racket and laughing.
+000000363072 a train on a track near many trees 
+000000491683 There is only one horse in the grassy field. 
+000000534827 five men riding their motor cycles on the street
+000000471991 A room with black and white tiled flooring.
+000000209142 A bowl of green vegetables are mixed together.
+000000545007 A statue holding a lamp on top of a clock.
+000000123213 Two teams are playing baseball with the umpire touching the catcher.
+000000424545 A striped cat sleeping on a sunny window sill
+000000148508 People standing on the sidewalk waiting for a parade to start.
+000000011197 Couple of people standing at an intersection waiting to cross.
+000000338625 A city street scene with a bus and buildings.
+000000389316 Two elephants are standing inside of an enclosure
+000000535523 A bagel machine is making bagels while people people work in the background.
+000000382743 Many kites can be seen in the air through umbrellas.
+000000058393 Two people are sitting on a bench together in front of water.
+000000140583 A herd of sheep with two sherds moving down a road in the mountains.
+000000500716 A man looking through a thumb hole on a scissors.
+000000084031 People stand near a fence watching the baseball game.
+000000555012 a white toilet and tub in a bathroom
+000000312421 A lunch box of some sorts is placed on the table. 
+000000416343 a person sitting down holding a surf board 
+000000424349 A pizza chef putting a pizza in an oven in a professional kitchen.
+000000481390 Several basketball players in the middle of a game.
+000000555005 The large dog is asleep on a couch.
+000000530052 some colorful carrots sitting on a table next to some other veggies
+000000221017 A chicken walking around on the ocean shore.
+000000006213 The bathroom had a tub with two sinks and a separate shower.
+000000479248 The movie "Walk the Line" is playing on a flatscreen tv.
+000000039477 A beach towel covers a TV across from plants on the wall and in front of the window.
+000000058111 An orange cat rests on a white counter top
+000000229858 A deer standing next to a small deer in a forest.
+000000455267 A skillet filled with a bunch of vegetables being cooked.
+000000009483 A man in a suit stands near a table with computer equipment on it.
+000000385997 A dog laying down in his bed in the living room.
+000000050811 The man sits at a table with a bowl and three beer bottles.
+000000046031 A small tablet computer next to a larger monitor and keyboard.
+000000030504 A person cross-country skiing on a winters day.
+000000287291 Cars and people stopped at a city intersection.
+000000360097 A gate is near lots and lots of luggage that is all grouped together in different clusters.
+000000089078 A woman that is standing on a snowboard in the snow.
+000000413552 A baby sitting on a females lap staring into the camera.
+000000468577 A young girl lying in a convertible toddler crib.
+000000098392 A photo of a busy intersection with a bus, several cars and a clock tower.
+000000259830 Some bicycles and a scooter are parked nearby each other
+000000387098 Three monitors sit on a small and mess covered desk.
+000000060363 A large clock on a wall with an industrial building in the background.
+000000377814 A variety of donuts sit in a box.
+000000463802 A stuffed animal watching an excited tennis player on TV.
+000000183391 Man swinging tennis racket while group of people watches
+000000265518 A white and black plate topped with food next to a cup of coffee.
+000000569917 a a small bathroom with pink tile and trim
+000000450303 A number of people around a large table working on laptop computers.
+000000409424 The fruit are arranges on the surface in a brown wicker basket.
+000000086582 Something knitted is lying on top of a laptop.
+000000491008 A man and woman are holding up a pizza.
+000000360951 A laptop next to a monitor, keyboard and mouse on a desk.
+000000530975 a round outdoor courtyard table with a hotdog and a bag of chips on it.
+000000139883 A Frisbee golf net in a park with several Frisbee in it.
+000000468632 A group of young men standing on top of a soccer field.
+000000350607 a bench that has many colors on it
+000000253742 A woman standing next to a herd of animals.
+000000484760 A statue of a woman holding a ball stands in front of the old church.
+000000479448 A plate of waffles with bananas and strawberries on top.
+000000446522 A black and brown dog sits curled in a flowered wing chair.
+000000232692 Man in black shirt wake boarding over a rail.
+000000161879 a skateboarder in a gray sweatshirt is doing a trick
+000000039480 Lady on tennis court wearing red tennis outfit and holding racket.
+000000555597 We are looking at an old photo of a city street.
+000000299355 The meal of chicken and rice has green brocolli with it.
+000000496854 Bride and groom walking under an umbrella held by a woman.
+000000205834 A dog is laying with its water dish in front of it.
+000000089880 Two dogs playing with each other in the grass.
+000000074058 Black and white photo bicycle with sidewalk and umbrellas
+000000159458 A dog sits on top of a bed in a room.
+000000293044 A container that has a packed lunch in it.
+000000412240 A dog sits on the floor next to some shoes. 
+000000415727 A couple of men standing next to each other.
+000000297022 Three men rest against their truck for a break from work.
+000000004395 A young man wearing a white shirt and tie.
+000000316666 this is a restaurant with flowers on the bar
+000000368038 Colorful train cars are on the track next to some grass.
+000000270066 A white motorcycle sitting in front of a van.
+000000188439 a big truck with some branches in the back of it 
+000000502347 a little plane flying above a lake full of boats 
+000000535858 A woman rides on a surfboard in the ocean.
+000000245102 A light shines over a canopy on a snow covered sidewalk.
+000000105249 A small bathroom has an open toilet seat.
+000000241319 A sink surrounded by lots of tooth brushes and glasses.
+000000103585 A picture of double sinks in a bathroom.
+000000436315 A home made pizza with pepperoni and sausage. 
+000000052565 a cow standing in some sand next to some water
+000000376322 A long table full of people on both sides.
+000000474452 A street sign on the side of a lot next to a building.
+000000449603 A kite surfer killin it while shreddin the GNAR
+000000084664 A plate filled with several types of decadent foods. 
+000000415194 A very old fashioned kitchen with retro floor tiles.
+000000348045 Three white urinals mounted to a bathroom wall.
+000000325347 The tennis player is running to meat the ball.
+000000577976 A street with orange cones and a work truck on it.
+000000309495 An out of focus image of a toilet and toilet paper
+000000388258 A bicycle leaning against a bench as an airplane flies by.
+000000194832 A vacant bus has seats facing in multiple directions.
+000000044877 a person in a wheel chair swinging a tennis racket
+000000051610 A little girl sitting in front of a laptop computer.
+000000514979 A very large elephant standing next to a baby elephant.
+000000245651 A cake with fondant stars cut out on white fondant
+000000171611 A man and his dog in a boat, on the water.
+000000037689 Two snowboarders high in the air off of a jump
+000000279927 PERFORMERS DOING A SHOW IN FRONT OF SPECTATORS
+000000511384 an orange disk if flying through the air at a park
+000000486438 The chocolate and glazed donuts are stacked on a plate.
+000000214539 A crowd is watching the two teams play soccer. 
+000000174231 A cat is laying inside of a half open lap top.
+000000060886 A baseball team crowd together on the field
+000000109916 A couple of dishes are sitting on a stove.
+000000163057 People flying kites in a park on a windy day.
+000000147498 A flock of birds flying over water and sand with a volley ball net on the sand.
+000000284698 A man taking a picture of himself in the rear view mirror.
+000000351589 The bench is in a shady area surrounded by plants
+000000013201 A man is grinding a ramp on a skateboard
+000000492937 A close up of a bike brake on a bicycle.
+000000430973 An olive and cheese pizza that has been placed on a table.
+000000329827 Blurry photograph of a woman skiing down a slope
+000000097022 An open kitchen is shown with light brown cabinets.
+000000470952 A man is in his skiing gear, while posing for the camera. 
+000000386879 a person smiling holding a tennis racket near a bush
+000000283318 A fire hydrant and some cars on the side of the street.
+000000196843 A school bus is covered in some decoration
+000000537827 Two baseball players are having fun practicing in the outfield.
+000000273715 The skier is happy to have won the race.
+000000355677 A lifeguard post is empty next to water.
+000000092416 A young man in a suit and tie with a beer in his hand.
+000000000885 A man holding a tennis racket playing tennis.
+000000064574 A pen, hairbrush, and other objects sitting on a bed.
+000000334483 People outside at table eating cake on plates.
+000000485130 there is a birthday cake that looks like a boxing ring
+000000299553 A crowd of people gathered watching kites in the air
+000000205289 A PICTURE OF A  MAN IN A TRUCK DATED 1917
+000000289586 Giraffe trying to reach some leafs on a tree.
+000000283113 A couple of hot dogs on a paper plate.
+000000297343 A stop sign has been placed upside-down in the grass beside a building. 
+000000458790 A building seen through a rain and fog covered window.
+000000442009 A bathroom scene complete with a cast iron style tub.
+000000334399 A clock that is standing on a table.
+000000461036 A black bear standing on top of a tree.
+000000008899 A graffiti-ridden building in an urban setting, a fire hydrant at curbside.
+000000502336 The passersby ignore the sign for the Canon Chinese restaurant.
+000000226147 a woman in a white top at a table with food
+000000429718 An old style green train moving along train tracks.
+000000429281 A big bin filled with some ripe yellow bananas.
+000000508370 A couple of men as one points to the right next to a bicycle.
+000000400082 A hotdog, fries and a drink sit on a table beside a book.
+000000232489 A paper plate that has two pieces of pizza on it.
+000000535608 Several people who are laying out on a beach.
+000000029596 A comfortable hotel room featuring a big screen television.
+000000440475 A room with a Tv, potted plant, tables and chairs
+000000170670 The table is covered with plates of food and bottles of beer. 
+000000336587 A broken telephone pole in a neighborhood with a stop sign attached to it.
+000000535094 A brown cow sniffing on a person hand
+000000053624 A large gray elephant standing on a pool of water.
+000000113354 A group of zebras eating from a pile of hay
+000000491757 A cat with blue eyes sitting on a pink bed.
+000000320706 Blender mixing food together on kitchen counter of tile
+000000031749 A mirror image of a red haired doll clock.
+000000034452 Man standing in wooded area playing with flying discs.
+000000542423 A person that is sitting alone on a park bench.
+000000546659 A locomotive train carrying carts down a track.
+000000068286 Two blankets have been folded on a large bed.
+000000542127 Two skiers flipping in the air with people in the background.
+000000426297 A zebra has its neck turned and is looking to its side
+000000074860 A man sitting on the ground with a snowboard on his feet.
+000000254016 A machine that holds a lot of donuts and cooks them. 
+000000014226 A person sitting on train doing something on their lap top.
+000000057232 Two white cushions on a chair made of baseball bats.
+000000431140 A bathroom with sink, mirror, and toilet in it.
+000000491757 there is a cat that is laying on top of a pink bed sheet
+000000289702 A neglected black and white Pitbull Terrier puppy in a filthy environment with garbage is reflected in an old mirror.
+000000534827 a group of young men are riding their motocycles down the street
+000000047801 A man with glasses on biting into a piece of cake.
+000000568981 A man pokes his head through a wall to photograph boys skateboarding
+000000029640 A skillet contains diced meat and mixed vegetables.
+000000249180 One man traveling up a stairway and another waling down the stairway on a phone.
+000000309467 Two people carrying backpacks are cross country skiing.
+000000062808 Two people sitting at a table with a pizza pie.
+000000392481 A nice jet plane flying into some grey skies.
+000000386912 a person sitting at a desk with a keyboard and monitor 
+000000508917 The passenger train is pulling has arrived at the station.
+000000365208 A black and white photo shows someone standing on a beach behind a curtain
+000000398377 There are women that are talking on the cell phone 
+000000210099 A cat sits on a wicker chair in the sun.
+000000284764 A man eats from a green dish while holding a handheld radio.
+000000186345 Seahorses are photos hoped into a blue sky above sand dunes. 
+000000210708 A large elephant standing next to a small elephant playing in the water.
+000000486112 a man sprays an elephant with a water hose 
+000000323263 A woman is selling fruit in a poor neighborhood.
+000000518770 People playing the Wii together in  a room.
+000000556765 A brown elephant walking in the moving water. 
+000000503823 Old wooden boats in dry dock on a grey dreary day.
+000000507473 A woman taking a picture of herself while brushing her teeth
+000000579970 some stairs a couch a table and a television
+000000389109 Pizza pie sitting on top of a table in a restaurant. 
+000000159112 a close up of a pizza with several toppings 
+000000522940 A stop sign with words added to it in marker
+000000099114 People standing and walking at a city intersection
+000000095155 Different angles of a person who is using a snowboard. 
+000000194940 A bowl of carrots, broccoli, and peppers on a table.
+000000548246 some people standing on a court with a racket 
+000000538236 Troisgros are piled up for sale at a busy market.
+000000123321 This is a broccoli carrot soup with a lot of broth.
+000000365385 A toothbrush in a cup sitting next to a sink.
+000000135410 A bike leaning up to a parking meter.
+000000031322 White swans swimming in a harbor with docked boats.
+000000087742 A glass vase filled with feathers on a table
+000000060102 a close up of two people playing soccer
+000000423519 A yellow and red bus in the middle of the street. 
+000000201934 A line of vehicles parked on the side of the road.
+000000153527 Several tools sit on the wood boundaries of a field.
+000000361571 A dog looking up with mouth open and wearing a hat.
+000000333069 Two giraffes in the trees, one standing up.
+000000363072 A silver train sitting next to a forest next to a tall building.
+000000416758 Two black and white cows and two white cows grazing in the grass.
+000000168330 An old clock is illuminated at night on a street corner.
+000000562229 Young boy on skateboard riding near circular drive.
+000000241677 a bunch of people on camels walk up some stairs 
+000000512929 Two young girls are cooking green beans on the stove.
+000000153797 A baseball pitcher follows through after releasing the ball.
+000000347335 a plate of food on that is sitting on a table
+000000022589 Two sheep in a caged area with bushes 
+000000162092 Woman with plated food near crowd of people.
+000000158548 A sheep is on a field of grass with two men walking around it.
+000000350488 A zebra grazing on grass near a river.
+000000465675 Large running mower sitting next to two boats on a ocean side.
+000000300233 ingredients placed on a table for a dish.
+000000493286 A vintage red car in front of a vintage army prop plane.
+000000064523 A team of girl tennis players gather together at the net for a group picture.
+000000528862 A group of giraffes wonder around in a fenced off area with other animals. 
+000000312192 A plate with pizza, lettuce and ham sit on a plate white we see hands holding silverware and a bottle of wine with glasses.
+000000355817 A large bus parked on the side of the road.
+000000575205 A Plane on runway taking off above ground.
+000000148999 A biilboard with a railcar in front of it. 
+000000171298 A group of busses parked in a parking lot.
+000000396863 A small girl riding a horse along a river
+000000430961 A man in a Rays uniform is throwing a baseball
+000000213171 a man in a baseball uniform standing with his mit on
+000000130465 this is a church made of white stone
+000000023359 a man kind of high in the air while doing a snowboard trick 
+000000475064 A person  holding a piece of food in front of a cafe.
+000000058393 A couple is sitting on a bench in front of the water. 
+000000420069 an informal baseball game is held in an open field.
+000000230166 Many sheep grazing on grass in a field. 
+000000389109 Pizza on a table at an outdoor restaurant at night
+000000011051 Girl looks back as she fixes flower on boy's suit
+000000197870 A wooden park bench with a statue of a chicken under it.
+000000043737 A photograph of a sidewalk covered in snow.
+000000449190 A close-up of a burrito and potato wedges.
+000000241319 Two toothbrushes each in a glass on the sink.
+000000343937 a girl in a ski jacket and hat riding a snow board
+000000373382 A tennis player is in an awkward position with racket raised an finger pointed.
+000000574702 A narrow city street is filled with people and motorbikes.
+000000374369 A man dressed in black snow gear stands on skis in the snow.
+000000276285 pizza an other food on a table at a restaurant
+000000071938 A man laying on top of a pillow with a phone on top of his head.
+000000205324 A couple of men standing on top of a grass covered field.
+000000486479 White dog sticking his nose out from under red and white striped bed ruffle.
+000000001000 The people are posing for a group photo.
+000000431876 Four cops on the back of horses stand in the street.
+000000407614 an orange kitchen with a white refrigerator stove oven and dishwasher
+000000134886 A playing flying over a van with multiple people around it.
+000000429011 Model of two cars with a man near a blue phonebooth.
+000000057725 A bridge over water near several tall buildings.
+000000389109 A pizza with tomatoes on it sitting on a table in a restaurant.
+000000170191 Man dressed in leopard robe next to a bed.
+000000414340 A clock tower stands on the side of a building.
+000000139077 a kid with a remote playing video game
+000000340272 Three sheep are grazing in the field next to the water.
+000000460229 a blue pickup truck stopped at a stop sign
+000000463542 Many people are skiing down a hill that is full of snow.
+000000007574 A metallic refrigerator freezer sitting inside of a kitchen.
+000000341921 A young man sitting on a park bench next to a playground.
+000000397133 Man in apron standing on front of oven with pans and bakeware
+000000500826 New York street sign showing Broadway and West 122nd Street
+000000488251 A man in a black hat posing with his horse.
+000000334719 A family are playing frisbee on the beach.
+000000132587 A person sits on a bench with a water bottle next to him
+000000284282 A blender laying on a table in a room.
+000000549674 A computer on a desk with two monitors 
+000000298904 There is a vase filled with water that has rocks and a plant in it
+000000524742 A wall covered in graffiti next to a sidewalk.
+000000292456 A man and a woman riding on a motorcycle are getting ready to hit the road.
+000000357903 A fried egg with a runny yolk tops a mini whole wheat pizza made with melted cheese and spinach. 
+000000512929 Two girls at stove cooking food in a skillet.
+000000504000 a blue jet is sitting next to a sign
+000000560266 Two bears on the water fighting each other.
+000000060823 What has attracted this many birds to the plaza?
+000000419201 A herd of animals grazing on top of a grass field.
+000000138979 a bus stops to load and unload passengers
+000000455555 a black and white picture of a little boy and girl
+000000110359 A large aircraft in the blue sky by itself
+000000419096 A train traveling down tracks net to a  bridge.
+000000289415 a variety of partially eaten pizzas are sitting on several plates.
+000000030504 A tall man is on snow skis on a pathway.
+000000569030 A fire hydrant in a city has a hose attached. 
+000000035279 a man lies on the floor next to some gadgets
+000000119516 A woman wearing a dark colored dress stands curbside near a large building.
+000000068286 A large bed with white sheets and blankets.
+000000352760 A skier attempts a jump hon his skis.
+000000147415 A woman writes in her notebook at her desk.
+000000186929 The oranges are peeled open with the inside showing.
+000000414034 A kid sitting on a bed with a remote.
+000000053626 some people standing next to each other on top of a snowy hill 
+000000064495 A parking meter next to a parking lot
+000000372203 A wooden bench sitting out in the middle of a desert.
+000000399205 men staning in the street and holding cell phones
+000000017714 a plate of fruit sitting next to another plate of food on a table 
+000000236599 Kites tied to a fence flying on a windy day.
+000000314541 A rusted up park bench sitting in the middle of a park.
+000000022892 A dog and a cat are watching the plants on the table..
+000000245026 A woman holding up a cake for her birthday
+000000398905 Frizzy haired guy standing at a keyboard listening to the music.
+000000364587 Old rusted transit train cars sit on the tracks. 
+000000363207 A just married coule feed each other cake while a photographer takes photos.
+000000372260 A clock sits in a planter decorated with flowers.
+000000255718 A rusted fire hydrant planted firmly on the ground.
+000000227985 A plate with two ham sandwiches garnished with lettuce
+000000230008 a traffic motorcycle cop waits to give a ticket
+000000094871 A crowd of people by the steps feed sheep.
+000000013004 A white plate topped with a banana and pudding.
+000000439854 A man riding a skateboard along a sandy beach.
+000000082696 A bird is walking right through a dining room.
+000000166287 Three cows standing at a watering hole with birds flying overhead.
+000000051738 a hotel room with one bed and blue chair.
+000000521540 Banana with peanuts butter and crackers on a plate. 
+000000176799  A skateboarder mid-air above a building covered in graffiti.
+000000191580 A plate with broccoli and cheese bread on a bamboo mat.
+000000035682 Eating a donut makes for a quick and easy breakfast.
+000000424349 A man placing a pizza pie in the oven
+000000424642 two men in the middle of the woods playing frisbee golf.
+000000555005 Large dog laying down on a blanket next to a table. 
+000000119445 There is a baseball player who swung for the ball
+000000341469 The red suitcases are setting in a corner, waiting to be bought for a trip.
+000000064359 Zebras and giraffes together in a zoo cage.
+000000214869 Someone pointing at their cell phone in a case on their hip.
+000000136466 an oven with a pot and kettle on top 
+000000322574 French fries and a bag with bread sticking out 
+000000061471 A dog playing with a roll of toilet paper in a bathroom.
+000000105455 Colorful clock tower, mostly red with typical European buildings in background. 
+000000053626 A GROUP OF PEOPLE SKIING STOP TO POSE FOR PHOTO
+000000310622 A cluster of street signs hang on a street pole.
+000000315492 A little girl reaches for some wipes while her stuffed bear sits on the toilet.
+000000120572 There is a church near a cemetary. 
+000000035326 A stone backsplash covers the wall behind a cooktop stove. 
+000000294162 The white laptop is turned off but left open. 
+000000434297 A plate of prepared food has a sandwich and vegetables.
+000000520009 A bus traveling down the street in front of a large building with a clock tower.
+000000277689 clear glasses on a table with a white cake with red flowers on it
+000000127660 a female in a black shirt is in a suitcase
+000000336628 A Man dressed like clown getting off bus.
+000000493286 And antique red car is positioned in front of an airplane. 
+000000098853 There is a man hitting a tennis ball.
+000000117425 a young child looking at a birthday cupcake
+000000525286 A large clock  is on the wall where the paint is peeling.
+000000224664 people on a quiet beach playing with kites
+000000217957 An ornate clock on a building with tree branches.
+000000119828 A black cat sitting on a table in front of a laptop. 
+000000245448 A man standing next to a parked motorcycle.
+000000284623 A black cat looking intently at the camera
+000000175364 A kitchen with two stoves, an island, and appliances.
+000000378673 a skateboarder in brown pants is doing a trick
+000000431140 A bathroom with a white toilet and a white sink and a mirror
+000000110638 a small elephant walks underneath a big elephant 
+000000507797 A group of people getting off of a bus.
+000000177893 a city bus sits parked next to a side walk
+000000478286 The train is traveling across the high bridge.
+000000174482 A blue bicycle sits on a sidewalk near a street.
+000000463037 White and silver Scottish Airways plane on the ground. 
+000000475064 A person standing before a big sign holding up a pastry.
+000000400794 A man eating crust sitting by a table.
+000000507015 Baseball players are on the field during a game.
+000000341921 A boy sits on a bench in a park, working on homework.
+000000140987 The young girl smiles brightly with a plate of pizza.
+000000565153 A rural road with three octagonal signs, two are stop signs and one has the word "Go" painted on it.
+000000071877 A boy doing a skateboard trick on a concrete barrier.
+000000526103 an elephant behind a fence near some water 
+000000473406 A young child is sitting on top of the kitchen counter.
+000000370486 Woman standing in a crowd holding a colorful umbrella. 
+000000377486 a person riding a horse on a dirt ground 
+000000519338 A blue rusted train engine sitting on top of rail tracks.
+000000174123 A huge plate of yummy food with fork to eat. 
+000000023899 A group of men playing a game with remote controllers.
+000000550084 Roadwork equipment, at a gas station, next to a thruway.
+000000038070 a little bathroom with some feet standing next to it 
+000000066817 A white plate topped with meat veggies and rice with sauce.
+000000255718 A close up picture of a fire hydrant on a street.
+000000172617 A woman sitting on stone steps with a suitcase full of books.
+000000395633 A boat that is sitting in the water.
+000000543043 There is a red car parked next to an old bus.
+000000288430 A young woman standing on a sandy beach.
+000000319696 bright red microwave sitting on the kitchen counter
+000000042888 A cartoon monster underneath a no parking sign.
+000000398905 A man with a beard at sound board in front of microphone.
+000000215778 A laptop is sitting on a computer desk.
+000000471991 An elaborate church with tiled floors and stained glass.
+000000187249 A TV with a man and a woman on it's display.
+000000475150 A giraffe standing next to a leaf free tree.
+000000122046 A girl with an umbrella in a chair
+000000029397 a man standing on a bench with a sign next to them 
+000000370711 A display of shoes and umbrellas are in a window.
+000000190923 A traffic light with a one way sign near a trash can.
+000000081988 five surfers learning how to surf by the ocean
+000000235064 A bear walking across a stone wall near trees.
+000000292330 A dog that is sniffing a baseball bat.
+000000099054 An airplane is at an airport with yellow hoses by it.
+000000550426 A vase filled with red and white flowers on top of a table.
+000000066886 The young girl is brushing her hair. 
+000000370486 a woman holding a colorful striped umbrella 
+000000196754 There is a stop sign outside of a bus window.
+000000474021 A group of guys are playing a game with the controllers.
+000000036660 Someone in sandals is standing over a broken cell phone in pieces. 
+000000287959 The sandwich had been cut and was ready to eat.
+000000225670 a black dog in the air catching a frisbee
+000000265108 a suitcase, umbrella, flowers and balloon sitting next to a fence 
+000000270244 A zebra standing in a field of grass outside.
+000000534601 A white stuffed teddy bear sleeping on top of a woman's bosom.
+000000059598 a laptop on top of a table with a plate with food next to it.
+000000054164 a man that is surfing on some water
+000000215259 Two people playing wii in a living room.
+000000262587 A man riding a wave on top of a yellow surfboard.
+000000400922 A key bank sign with a clock on a building
+000000435003 A computer desk that has three monitors on top of it.
+000000032038 a close up of a pizza on a wooden table  
+000000323895 a tennis player running to swing ata ball
+000000259097 A person trying to reach a Frisbee in a field with high brown grass.
+000000301563 A person skiing down a mountain in the snow.
+000000435003 a computer desk with multiple computers and screens
+000000463842 A bed on an empty floor surrounded by plastic bottles. 
+000000383339 one person is kite surfing at the beach
+000000152771 A bicycle parked in an empty parking lot by itself
+000000150417 A woman sitting at a table with a cake in front of her.
+000000151629 A young male tennis player in action on a court.
+000000230362 Nice open stretches of land that is alive.
+000000481480 Horses and jockeys running on a racing track.
+000000441491 Three women are taking a bite out of the same piece of pizza.
+000000226662 Black and white photo of a parked motorcycle
+000000150224 The crowd is sitting under the umbrella at the beach.
+000000191614 A person riding a horse in a horse park while another horse wonders in nearby grass. 
+000000151000 A woman holding a camera is taking a picture.
+000000467511 two people stand in front of a motorcycle 
+000000248631 A Windows laptop computer with a large mouse.
+000000293300 Two large elephants standing next to a wire fence.
+000000425226 A person kneels down to peer into a refrigerator
+000000177213 Small pizza sits on a plate on a restaurant table. 
+000000567898 a couple of tins with some food in them
+000000287714 a bathroom with a tub and sink inside of it 
+000000424776 a grey colored jet plane flying over a snowy mountain range.
+000000138115 A man purses his lips while holding up an orange in front of his face.
+000000546219 A large group of people pose for a photo at dinner.
+000000530470 A large yellow kite is flying over a park in the daytime.
+000000461036 Some leafy trees are hiding a black bear.
+000000359937 A large double decked bus parked in a lot.
+000000493905 A tennis player is in mid air with his racquet.
+000000140286 A white horse pulling a white carriage.on the street
+000000302990 A man is using his board to surf a wave
+000000044068 A brown teddy bear sitting in a red chair.
+000000475904 A white swan swimming through a lake next to a boat.
+000000044590 A large group of motorcyclists have parked on the side of a road.
+000000300276 Two women are selling cakes and other items inside of an outdoor stand.
+000000291634 A row of motorcycles next to a bicycle.
+000000555012 A white toilet sitting next to a white bath tub.
+000000395633 A red and black boat floating on top of water.
+000000514797 The men are trying to fly their kite by the water. 
+000000051610 A laptop on the table and a baby looking at it
+000000326542 there is a male skier going down a hill
+000000085772 A man is about to hit a tennis ball with a racquette. 
+000000473219 Woman and man preparing to be married with a pastor.
+000000559842 Group of small children running down the field playing soccer.
+000000084492 A skateboarder grinds down a railing in front of buildings.
+000000526706 A baby cow gets milk from its mother while two other cows stand nearby.
+000000545407 An airplane is flying through clear blue sky.
+000000443969 A little girl inside of a shopping cart.
+000000453001 a girl watching another girl trip while pulling her suitcase
+000000168593 An olive green refrigerator next to a white refrigerator in an old kitchen. 
+000000474854 a man sitting in a chair and eating a donut outside near some shrubbery.
+000000280930 Little girl laughing and holding a pan of food 
+000000387916 A group of people with skis standing in the snow.
+000000345356 two males at a table with food and one has a cellphone
+000000485027 A guy throws a tennis ball in the air to hit it. 
+000000559547 A group of vintage men with baseball attire on them
+000000089697 a couple of people are sitting on a wood bench
+000000125936 People on both sides of a counter at a small food market.
+000000248334 The people are floating in the water in the small wooden boat.
+000000134886 The landing airplane comes in very close to the vehicle in the road.
+000000449579 A women who is about to swing at the ball while playing tennis.
+000000312421 Suitcase with stickers outside on table in city.
+000000554291 A cat's image is blurred as he tumbles in a bowl near a food scale and pen and paper on a table.
+000000079408 A red stop sign sitting next to a forest.
+000000276707 A street sign prohibiting vehicles in front of a store.
+000000267940 A ferry in a port with mountains in the background.
+000000213816 A photo of the open door of a train.
+000000466602 A person standing in the snow on skies. 
+000000485130 Wrestling ring sitting in the middle of an open room.
+000000378099 A computer keyboard sitting next to a mouse on a wooden table.
+000000159399 A woman with backpack flying a kite in grassy field.
+000000453634 A fancy bathroom with a clean toilet and piping.
+000000449406 A lone zebra stands outside on a rocky hillside.
+000000570688 Several colorful kites in the sky by several persons in the ground.
+000000507975 three men riding horses through a grassy field
+000000499031 A group of two women and an older woman between them.
+000000384651 A table lamp is on revealing part of a bed and an upside down glass.
+000000452122 A large white airplane flies in the gray sky.
+000000084492 A man in white grinding on a skateboard.
+000000099053 A yellow and black bowl holding noodles, beef and broccoli.
+000000398742 A couple of men are holding up a team member
+000000130599 A man is miming punching a giraffe statue.
+000000411774 A woman wearing glasses holding a tennis racket. 
+000000228981 three different kinds of electronic devices all in a row
+000000294783 A room with pillows, sheets, bookshelves, and hanging lights.
+000000177357 a man is riding on a surfboard at the beach
+000000022371 A man in thought sitting in front of a laptop with a pen in his hand. 
+000000564023 A public restroom with a urinal and grab bars.
+000000130586 A kid is running while pulling a kite behind him. 
+000000440184 People playing doubles tennis on a bright sunny day.
+000000341094 A statue depicting a bear breaking into a car.
+000000183391 A MAN IS PLAYING TENNIS ON THE COURT RACKET IN HAND
+000000532575 A GERMAN SHEPARD LOOKING OUT AT THE WATER ON A SHIP 
+000000212166 a small child with his fingers in his mouth and plate of food
+000000159977 A giraffe and a zebra are on a grassy field by the water.
+000000351609 A group of peeled and whole oranges in a bowl.
+000000296969 A group of giraffe standing next to each other.
+000000565624 two elephants grazing in the grass behind a fence
+000000192964 A man flying through the air on top of a wooden ramp.
+000000438955 competitive skiers ski forward on a flat surface
+000000143068 Kites being used by people on a beach.
+000000419201 a grassy field with some animals in it
+000000536343 Photo of a bathtub with two towels placed on a rack at the far end of the tub. 
+000000163118 Two young men playing a game of frisbee.
+000000370999 some little stones sitting around a dime 
+000000520531 A large calico cat sitting on a wooden table next to a remote control.
+000000574425 A small older bus parked alongside a roadway and another behind it.
+000000434548 A baseball player warming up on the field.
+000000460929 A hot dog and bubble up drink on a table.
+000000553731 A man dressed as a clown talking on his phone.
+000000370208 Someone has parked a classic orange ten speed bicycle.
+000000484760 That looks like a statue of the Virgin Mary beside the church.
+000000054593 The action in an organized youth baseball game.
+000000447789 A red do not enter sign under a green street sign.
+000000191672 A man makes a splash as he surfs on a wave.
+000000344816 A train is traveling underneath a cement bridge.
+000000049810 An orange and white cat standing in front of a mirror.
+000000189752 A table with plates of food that include pizza and salad.
+000000159399 A person in a field flying a kite in the sky.
+000000349837 Several refrigerators are lined up by a wall.
+000000257865 A man holding a tennis racket picks rubs his eye.
+000000345252 The boy is enjoying playing the game on the computer.
+000000345385 Young couple happily snuggling on their bed together
+000000397327 A white toilet sitting next to a bathroom sink.
+000000153229 Three men jumping for a frisbee while others play around them.
+000000064718 A man getting ready to swing at a tennis ball.
+000000376478 A large flower in a vase is sitting on some wood. 
+000000102356 a woman sits on a motor bike with a cart extended 
+000000196009 A spoon laying beside a few big carrots
+000000289516 A clock that is surrounded by tall grass in a park.
+000000537053 a giraffe in a field with trees in the background
+000000217060 The airplane is flying near a cloud in the sky. 
+000000079565 a pair of giraffe standing in a big open area
+000000259830 A moped is parked next to bicycles on a sidewalk.
+000000015517 a photo of a bus terminal taken from a distance
+000000022969 Pair of giraffes exploring a large, airy zoo enclosure.
+000000405306 a cat lays down on a bed 
+000000293804 a comfortable living room with brown leather couches and a fireplace
+000000373382 A man in white playing tennis, about to hit a serve.
+000000499031 Three women standing in front of a sign about bananas and a bunch of bananas.
+000000121586 A room that has a tv, window, and books, in it.
+000000238039 there is a goat that has a leash on 
+000000089880 Two dogs are playing together on the grass
+000000289586 A giraffe raises its head in front of rocks.
+000000261712 Two giraffes standing on a dirt path in a fenced in area with trees and green grass surrounding them.
+000000453584 What appears to be a beach-side eatery has an outdoor area with a metal railing and high stools, the whole of which rests on a tilted walkway with a sign and  a bench with a logo on it. 
+000000149568 A person stands between two dogs on a beach, one of them holding a frisbee.
+000000147729 People gathered taking pictures with a cell phone.
+000000414795 a herd of elephants standing in a water hole with a crowd of people watching 
+000000071711 An indoor museum exhibit features several old planes. 
+000000507223 The player approaches home plate to take his turn at bat.
+000000229358 A man standing in front of a mirror in a room.
+000000430048 Purple and blue building which has an atm inside.
+000000092053 A bunch of plates with food on them on a table. 
+000000492110 A man sits in a coffee shop working on his laptop computer.
+000000507015 five baseball players with a runner on the base
+000000578500 A well-lit and well-decorated living room shows a glimpse of a glass front door through the corridor. 
+000000099024 A person sitting on the floor constructing a white kite.
+000000262938 Stuffed animals looking a pictures of other animals in a book.
+000000094157 A large, ornate clock tower rises above another skyscraper.
+000000335658 A mouse and a keyboard on top of a desk
+000000016451 A sandy beach with surf boards sitting on top of it
+000000443498 A passenger bus drives through a street at night. 
+000000537153 Two people on a bench near the water.
+000000376625 The trolley car runs up and down the street on train rails.
+000000462904 a miniature horse walking near a restaurants outside deck 
+000000419096 A freight train running along tracks beneath a bridge.
+000000404568 A seagull standing on the sand near the water.
+000000052996 Two woman soldiers talking to other women in a kitchen
+000000460927 A majestic bear looks out across a grass plain.
+000000067310 a man on a skateboard in the air coming on to a ramp
+000000287527 a guy and a girl posing for a picture
+000000276284 A lone motorcycle is parked in the lighted parking lot.
+000000322610 A woman walking past a shop filled with merchandise.
+000000273760 a man on a field with a tennis racket in his hand
+000000243344 A long haired cat sitting on a refrigerator
+000000003845 A plate of vegetables, chicken, and white rice.
+000000521282 A close up of a clear vase with flowers.
+000000325306 A zebra standing on top of a dirt and gravel covered field.
+000000500257 A few men working on moving luggage at an airport
+000000027932 A baseball cap and catchers mitt sitting on the ground.
+000000295797 A weather vane atop a cathedral clock tower.
+000000569273 A fire hydrant is near people viewing a canyon. 
+000000226130 A sausage on italian roll sandwich, cut in half, is on a white plate, with a garnish.
+000000229311 An empty kitchen sink underneath a window, on the counter.
+000000545407 A side view of a plane flying in the clear blue sky
+000000535578 A flock of sheep are grazing on a grassy slope.
+000000416330 A black and white cat lies on a bed and relaxes
+000000451714 A man poses while skiing on a mountain at a snow resort. 
+000000143961 A group of people sitting on the grassy ground with coolers and umbrellas.
+000000553221 A group of people enjoying the water with sails.
+000000023899 Three men laughing at some pictures from a projector.
+000000406129 A male tennis player reaching up to hit the ball.
+000000026941 A green cart with old luggage on it standing by.
+000000455157 The man with glasses is holding an umbrella and sitting at a picnic table.
+000000161609 Man with a backpack carrying his dog in the pouch.
+000000459634 A back end of a vehicle with a metal rack and an antennae.
+000000397639 The two sheep are standing next to each other in the field. 
+000000074457 A person walks on the beach, carrying a surf board.
+000000414385 a road filled with cars in a desert
+000000173302 A busy kitchen next to a dining table and counters with chairs and tables
+000000012120 a couple of people that are on a tennis court
+000000078565 Sail boats sail over a body of water while people stand near the shoreline. 
+000000121586 A cat is on the television screen in an office.
+000000490515 a person riding a snow board falling to the ground
+000000134882 A cat curled up on a bed for a nap.
+000000269632 A big pretty yellow city bus in the dark.
+000000422836 A man walking down the street carrying a suitcase.
+000000514540 An upward view of a building and several signs, including a one way sign. 
+000000560178 A person's hand holding up a bitten radish.
+000000316666 A clear glass vase holding flowers in front of a mirror. 
+000000245513 A giraffe and bird in a grassy area with a fence.
+000000092939 A bride and groom are cutting their wedding cake. 
+000000363875 A man riding around on a scooter with luggage on his lap.
+000000308631 An old with motorcycle with people standing beside it.
+000000050828 A bedroom with two small beds right next to each other.
+000000036660 A woman with pretty painted toe nails standing in front of a broken cell phone.
+000000032334 A man and a woman toast their wine glasses.
+000000041990 Three people on skis are posing for a picture in the snow.
+000000544052 A number of action figures on the ground and two of them on skateboards. 
+000000208363 a group of people sitting together eating pizza
+000000049259 A man who is sitting down next to a body of water.
+000000302452 A giraffe drinking from a bottle in front of a group of people.
+000000155571 An elephant walking on a grassy area during the day. 
+000000060347 A man sitting on a bench in a wood.
+000000005060 A man sitting on the floor taking a picture of himself in a mirror.
+000000171611 A man on a boat with his dog.
+000000171757 a couple of people that are standing up in a room
+000000177383 A yellow and burgundy train that is on tracks. 
+000000185409 Several zebras run and play in a grassy, wooded area
+000000138492 A black and white dog catching a blue flying disc
+000000445602 The skier is at the bottom of the mountain in the snow.
+000000473121 A man in skis passing an orange flag on a snowy race course.
+000000407083 A dog looking out the back window of a car.
+000000562121 Two zebras fighting in a cloud of dust
+000000393838 There are diced vegetables inside a saucepan on top of the stove.
+000000506279 There is a full glass sitting on the table.
+000000305317 A bunch of people are holding kites and the wind is blowing them on a grassy field.
+000000002149 A bowl filled with many shiny  green apples.
+000000245576 An orange cat sitting on top of a computer desk.
+000000344621 A living room with a couch, chair and a coffee table.
+000000033638 A woman cooking on an old fashioned stove.
+000000486573 A vase filled with water and white flowers.
+000000229997 A black bear is surrounded by black birds on grass.
+000000221291 A young boy is posing in a large grassy area.
+000000262682 A kitchen with dark green cabinets and personal items on the windowsill. 
+000000406611 Many people are all scattered around the ski station. 
+000000535578 A flock of sheep walking along a grassy hillside grazing.
+000000276284 a motorcycle in a parking lot with a sky background
+000000087875 A fire hydrant in a grassy field next to a bush
+000000229948 a wagon that is being pulled by some horse
+000000123213 batter receiving a pitch at a baseball game
+000000548246 A group of men talk on a tennis court.
+000000430973 Large whole pizza pie with cheese and olive toppings.
+000000181796 Plate of food that includes chicken, beans and a pickle.
+000000457884 A group of kids playing a game of baseball.
+000000431848 a man skating on a drying ground with his hands up
+000000010707 Three people with cups on the couch and one with remote standing
+000000429530 There are four vases that have butterfly sculptures on them.
+000000338905 People eating a meal at a long table covered with umbrellas outside a restaurant 
+000000574810 A yellow fluffy cat sitting in a window sill next to a window.
+000000455872 Black and white photo of an old dock with low water.
+000000294831 Someone is holding a black plate with two slices of pizza. 
+000000505942 A black and white photo of a street sign with an arrow on it
+000000097679 A car with a snowboard on it's ski rack parked in a foggy parking lot next to two cars with empty ski racks. 
+000000211825 Man blowing out white candles on top of a brown cake. 
+000000383384 A baseball player holding a bat next to home plate.
+000000308394 An older woman riding a train while sitting under it's window.
+000000577584 A large white bowl sits on a slide out shelf under a microwave.
+000000051008 A cat sitting in front of a laptop computer.
+000000271997 The young man possesses a very unusual face.
+000000561256 An older woman in a flower dress takes a self photo in the mirror.
+000000343466 In the kitchen, there is a sink, stove, window and some cabinets.
+000000516871 A bento box filled with sushi and vegetables.
+000000229601 there is a baseball player that has just run to the base
+000000563882 A woman sitting in a seat holding a cell phone.
+000000140929 a large clock hangs from a building and reads 12:43
+000000576052 A person on a horse with large snow capped mountains in the background.
+000000074860 there is a snowboarder that has fallen n the snow
+000000237316 The lid is up on a toilet viewed through the open door of a bathroom.
+000000007784 A large white and red parasail in the blue sky. 
+000000319607 A bent traffic light next to the side of a street.
+000000267191 Two people are riding elephants beside some trees.
+000000126226 A dug out filled with baseball players next to baseball equipment.
+000000429761 two men on a tennis court shaking hands over the net
+000000221213 Many train compartments parked along the train tracks.
+000000560474 Many different items are lined out for display on a table
+000000581615 A view of a man's urinal inside a bathroom stall.
+000000112634 Two giraffe's in a zoo with a large rock.
+000000238039 A small animal is being lead on the leash
+000000139872 A happy black and white dog next to a frisbee
+000000043435 Two people are riding on surfboards in the ocean.
+000000354547 A man wearing a suit and standing on a field of grass.
+000000072813 Dog lies on bed looking towards the camera
+000000515579 A man walking while holding a ball in his hand. 
+000000484351 A full view of a classroom with students and politicians.
+000000242060 a plate filled with some little tiny cakes 
+000000407574 A stack of bananas are sitting in a tree by signs. 
+000000111207 Two snapshots of a skate boarder doing stunts in an open area
+000000280891 A woman standing next to a young man near a pile of fruit.
+000000266206 A bathroom with the lights on visible through a slightly ajar door.
+000000117197 This is a woman talking on her cell phone while sitting on the curb.
+000000216739 A tall giraffe standing next to a tree.
+000000565153 a sidewalk with grass and signs next to it
+000000153217 A BLACK CAT POSING IN FRONT OF MOTORCYLE
+000000184791 A painting of a table with fruit on top of it.
+000000162366 A black parking meter next to a brick building. 
+000000368456 Empty parked train painted with large, colorful graffiti.
+000000060449 a laptop on a table on balcony, blue sky.
+000000416885 multiple plates of food on a wooden table
+000000446117 A ripe tangerine atop a worn wooden picnic table.
+000000118594 two really long horned cows eating in a field
+000000580757 A american flag painted fire hydrant with chains hanging at it side.
+000000002006 City bus driving through pedestrian saturated area near crosswalk.
+000000001675 A black cat is sitting on a metal platform. 
+000000407298 A young boy wearing a catchers mitt holding a ball.
+000000483050 a bed room with a neatly made bed with night stands
+000000556000 four children in a living room with one of the children holding a game controller before a television.
+000000163118 Two guys leap for the Frisbee as they play their game in some sort of a gym.
+000000402720 A trio of men drinking wine at a restaurant table
+000000245513 A giraffe standing in a field next to a fence.
+000000542856 A city bus carrying ads featuring women coming to town.
+000000097585 Glass is hanging on the white wall all bunched together.
+000000239274 A pontoon boat ferry full of passengers pushes off from shore. 
+000000484029 there is a stop sign and a street sign at the corner 
+000000433204 a person riding a motorcycle on a city road with a sky background
+000000242724 a couple of women ride horses through some water 
+000000309495 A roll of toilet paper sits next to a toilet. 
+000000193348 The small child is brushing it's teeth with a green toothbrush. 
+000000459195 The group of three friends are sitting on a fallen tree in the woods.
+000000090208 Several green and red double decker buses lined up parked.
+000000575243 a man that is walking down the street with a umbrella
+000000002592 Decorated coffee cup and knife sitting on a patterned surface.
+000000119233 A cat sprawled out over the top of a laptop computer keyboard.
+000000286708 A cat with a pink and fuzzy hat on it's head.
+000000569917 A sink and some shelves are in this small bathroom.
+000000294162 An open laptop compute with a computer mouse.
+000000314914 Several signs attached to a post detail regulations related to parking and other issues.
+000000379332 A tennis player is looking up to the sky.
+000000148707 Two unripe oranges growing on a tree. 
+000000502336 A group of ladies walking down a street
+000000381639 A little girl walking along a row of orange cones.
+000000575205 a small plane taking off from a run way in mid air
+000000205776 a brown bear is lying asleep in a hammock
+000000336628 A clown dressed in a suit bedecked with dinosaurs steps off a cable car.
+000000522713  a bench sitting in the grass facing the water and boats 
+000000087470 A herd of cattle walking down a road being followed by a cowboy.
+000000308753 A cave on a side of a rocky hill.
+000000520077 A stack of text and comic books on a table.
+000000019402 Young woman wearing winter clothing alongside friend preparing to eat slice of pizza.
+000000015746 A fire hydrant in front of a house with overgrown grass
+000000091779 A display of various hot dogs in serving platters.
+000000349860 a young man doing a jump with his skateboard in a skate park
+000000345469 The shelves have on different types of donuts. 
+000000200152 A wall street street sign in front of a lit up street.
+000000367095 Adults working in open office setting at desks.
+000000093717 A boy on a bike watches a boy on a skateboard go over a ramp.
+000000135902 A red train pulling into the station under a steel framed roof.
+000000094185 The store front of a teddy bear shop 
+000000417465 An adult black bear let's his long tongue protrude.
+000000096825 a person on skis riding down a snowy slope
+000000537270 three people sitting at a table with their cellphones
+000000153568 a stop sign with alot of  writing on it
+000000168974 A baby girl is sitting on the floor looking at a cell phone.
+000000232538 A passenger train stopped near a train station.
+000000227482 a male in a gray shirt grass and a white frisbee
+000000013597 a piece of a chocolate cake on a plate 
+000000292456 A man and woman on a scooter receiving baggage from a bellhop.
+000000450303 A group of people sitting around a table in front of computers.
+000000235857 Three cows grazing in a field of green grass
+000000200252 Two green shoes lined up on a bed.
+000000086956 A man standing behind a woman holding a bat.
+000000523957 A skateboard going around cones in a skateboard competition. 
+000000549738 A picture of a very large kite flying in the air.
+000000270705 A falcon perched on a window looking off in the distance.
+000000229858 An antelope walking out of the bushes towards a baby antelope.
+000000110282 a stop light that is next to some trees and grass
+000000142472 A bunch of pedestrians as well as cars are on a busy city street. 
+000000039484 A city has colorful buildings and signs among cars.
+000000503823 some boats are docked on the sand at a beach
+000000416534 A living room with hard wood flooring and two book shelves.
+000000545219 A neon double decker bus is driving by a few pedestrians.
+000000426376 Person snowboarding down a steep snow covered slope.
+000000203931 A batter ready to swing at home plate.
+000000110638 A mother elephant with her baby elephant underneath her.
+000000036936 A man and woman playing a wii game.
+000000249219 A double decker bus driving down a city street.
+000000356498 Some kind of a pancake that has broccoli and sauce on it.
+000000239717 a smiling man holding a little plate of food 
+000000296224 A city bus is parked on the curb waiting for people
+000000525247 A cat is looking at a laptop computer screen.
+000000480275 A flat bed cart with three large bunches of bananas and a tied lidded basket.
+000000572555 A long train is going down the train tracks.
+000000409475 A man and a woman standing in the snow on skis.
+000000384661 The kitchen is clean and organized ready for us to use. 
+000000230983 A person is doing a trick over a trash can.
+000000289586 A giraffe stretching its next long to reach a branch.
+000000004395 Man in a dress shirt and tie looks in front of him. 
+000000030213 A kitchen has an island, icebox, and sink in it.
+000000244592 Three zebras interacting in a zoo zebra enclosure.
+000000258911 A row of cows standing behind a metal fence.
+000000049259 A man sitting on a wall next to water next to a barrier
+000000511384 A Frisbee is tossed across green field to another person. 
+000000161925 A woman holding a large knife next to another lady cutting pizza.
+000000125062 A group of teddy bears leaning against a row of movies
+000000319184  This is a picture of some friends playing frisbee in the park.
+000000057760 A group of people together on a beach.
+000000326174 A group of surfers are carrying their surf boards into the ocean.
+000000570169 a brown teddy bear and some wooden block toys
+000000442822 Boats floating in rows at the dock in the river
+000000236690 A white and gray bird soaring over the blue ocean.
+000000348881 A large jetliner sitting on top of an airport runway.
+000000089670 A waffle with cream and strawberries sits on a plate. 
+000000111951 A double sink vanity with mirror, shower stall and toilet in a bathroom.
+000000471991 A large chapel facing towards the alter with pews on either side and beautiful glass and marble adornments. 
+000000261712 Two giraffe standing on dirt near flowering trees.
+000000373705 A red and yellow fire hydrant near the water fountain. 
+000000518770 Some children who are playing video games together.
+000000019109 Numerous motor scooters parked by backing in facing the street
+000000109313 A man holding a Wii game controller while standing in a living room.
+000000299355 A plate with flower print  and broccoli on it
+000000129492 two young children sitting on the ground playing nintendo 
+000000079188 There are two giraffes sitting together in the wild
+000000187513 A bathroom with a sliding shower door next to a toilet.
+000000308430 Someone is cooking a vegtable stir fry for dinner
+000000060363 A large clock mounted to the side of a building.
+000000457078 A room with a t.v. and other bedroom items and accessories in it. 
+000000165351 The desert is made and ready to be eaten. 
+000000482970 A laptop computer sitting on a desk with a speaker.
+000000330369 Small red and white metal ship near a stone embankment.
+000000559707 A small portion of broccoli, ham, and pasta salad
+000000115118 a train of many colors is coming down the track
+000000239347 A photo of a man and his wife on a bed
+000000510095 A man in a field of wild flowers holding an orange bat
+000000346638 A bottle of beer sits next to a computer at a desk
+000000101022 Sign in the snow that warns mountain goers.
+000000007278 A surfer in the ocean on the crest of a wave.
+000000093437 An older man is wearing a funny hat in his dining room.
+000000029187 A man races a horse on a dirt track.
+000000526706 A small cow is getting milk from it's mother.
+000000393569 A young woman using a laptop computer while sitting on a bunk bed.
+000000496409 some guy that is skiing down a hill in some snow
+000000155291 A cat sitting on top of a wooden window sill.
+000000484415 A hand is reaching out to the top if a toilet. 
+000000200839 An educational bus is parked in a residential area.
+000000322163 A group of three chefs preparing food in a kitchen.
+000000407650 a person jumping a snow board in the air 
+000000324614 A girl brushing her teeth holding up her other hand trying to shield from the camera.
+000000159791 A large refrigerator covered in magnets, photos and postcards.
+000000310622 A West 32nd sign under a Korea Way road sign. 
+000000344621 Magenta and plaid furniture sit in a brightly lit living room.
+000000439715 Person in red peacoat and hat riding a horse.
+000000455301 A mother with her baby laying in bed.
+000000476415 A guy wearing a tie has his hands in his pockets. 
+000000175443 A teddy bear wearing a red bow that is sitting on a Christmas tree.
+000000014473 A high speed train in movement, with workers.
+000000373382 People sitting in the stands watching a man play tennis.
+000000237984 A green bench sitting on the side of a sidewalk.
+000000301421 a glass topped desk with a computer and speakers
+000000377368 A cake shaped like a teddy bear sits on a platter.
+000000327617 an old black and white photo of a tennis player
+000000143931 a large campaign trailer parked in a parking lot.
+000000312489 The expert surfer is trying to also parasail. 
+000000507235 A table filled with dishes of very tasty looking food.
+000000451879 Baseball players in action, one person dressed in green running while the others in white are looking and reaching toward the ball.
+000000027982 Modern bathroom with stone sink area and tiled walls.
+000000229311 A kitchen area with a large stainless steel sink.
+000000565778 A blue train on some train tracks about to go under a bridge
+000000418281 There white cows in grassy area with temples in background.
+000000035062 A bed with a person sleeping in it covered in a blanket.
+000000547854 The little girl is busy eating her pizza at the table.
+000000142324 A man riding on a motor scooter with a child in his lap.
+000000125245 an aquarium with some blue black and yellow fish
+000000434247 A picture taken from a distance of people riding horses at the beach. 
+000000362434 a box full of stuffed animals and other children items 
+000000577584 A dish in front of a toaster oven. 
+000000470121 A plate full of sushi,with more behind it.
+000000334530 There is a surfer on a surfboard in a pipeline
+000000036494 a bunch of people gathered inside of a building 
+000000137294 A man reaching into a freezer while standing next to a woman in a park.
+000000236599 People hold flying kites in the air by the water
+000000125850 A cat is curled up and sleeping in a round tin.
+000000463542 A group of people riding on skis on a snowy slope.
+000000036844 A living room with wooden floors and white walls.
+000000099242 An older man is skiing down a snowy mountain.
+000000368752 A small child is smiling and there is some dessert in front of him.
+000000570471 Man cutting a cake on the table in the kitchen.
+000000270474 a young girl wearing a dress plays with a baseball bat
+000000297427 A burger with a doughnut for the bun.
+000000133418 A male tennis player has just finished hitting the ball
+000000021879 Young girl on surfboard riding wave with young boys nearby looking on.
+000000319617 A hotdog with mustard and ketchup on a white oval plate.
+000000532575 a close up of a dog on a boat 
+000000022935 A soccer player in a blue uniform is running after a soccer ball.
+000000307074 a stop light right across the street from a fire station 
+000000239843 A stop sign underneath two street signs with a tree in the background.
+000000553731 Mandor stuff is a clown holding his cell phone up to his face. 
+000000041990 Three people with ski poles standing in the snow
+000000256518 A plate full of different types of foods. 
+000000384616 A fire hydrant and a stake in a dirt area with a few plants.
+000000289059 A man in blue shirt and shorts playing a game of tennis.
+000000101068 A little kid is swinging at a water balloon. 
+000000022479 A man on a skateboard performs a trick on a ramp
+000000140270 a woman dressed in English riding clothes rides a horse 
+000000429623 A large pizza and a soda sitting on a table.
+000000207306 A white teddy bear sitting on top of a trash can.
+000000325306 A zebra cleaning it's tail for hygenie purposes.
+000000440171 A man riding on the back of an elephant along a dirt road.
+000000558854 A table with food and a drink on it 
+000000580410 a living room with a tray holding a laptop next to a couch 
+000000556000 A boy playing wii in the living room while several other children watch.
+000000506707 A baseball player runs into home plate during a game.
+000000176857 a group of people that are petting a cat
+000000407298 a boy with a baseball glove and ball
+000000079969 A man swinging a tennis racquet on a court.
+000000239627 Two sinks in a kitchen with dirty dishes in them.
+000000120584 A clock high up on a wall in a room.
+000000287874 People are waiting beside the blue train pulling beside the platform. 
+000000287545 Two giraffes look over a railing blockade. 
+000000360943 An orange cat sits on a stoop in front of a door.
+000000224119 The inside of an airplane about half way back with television screens pulled down
+000000248112 A person standing on a court with a tennis racket.
+000000309391 Vehicle traffic on a hilly street in a small town.
+000000491071 A couple of bathroom sinks sitting underneath a mirror.
+000000356531 A woman sitting in a restaurant with Mexican food on her plate 
+000000294163 A large white clock tower filled with light.
+000000469174 The plane was flying high near the flag in the air
+000000267940 A ship anchored to the bay and close to some building
+000000002532 A skier stands on skis at the top of a snowy plateau.
+000000104455 A few zebras resting underneath a tree during the day.
+000000060052 A woman with a striped shirt is holding an umbrella.
+000000376093 A group of young people eat pizzas and drink from pitchers.
+000000320642 A group of people are gathered together for an event.
+000000512564 A two way street has minimal cars on it.
+000000072852 a white green and black sign and a bicycle without wheels
+000000449198 time shot of cars driving down Asian highway
+000000022371 A man contemplating something at a desk with a laptop.
+000000491071 Clean and near modern appearing restroom with sinks and counter.
+000000221281 Two giraffes walking around in the grass and dirt.
+000000213422 The bed and desk in a modern hotel room
+000000507042 Group of giraffes standing next to each other in a pen. 
+000000523194 A yellow fire hydrant by a wall and a sign
+000000018380 A lot of people sitting around a table full of food.
+000000200667 A cow is looking through a gap into some machinery.
+000000413395 A man sits on the couch with two cats on his lap.
+000000201676 Three zebras standing next to each other in a meadow with other zebras visible behind them
+000000116479 A canopy bed with white see through curtains. 
+000000159977 A giraffe and a zebra walking through an African plain.
+000000567825 A BIG HAMBURGER SITTING ON TOP OF A COUNTER AND A GLASS OF WINE 
+000000484404 The inside of an empty public bus with black seats.
+000000086755 A couple of people riding a pair of skis down a snow covered slope.
+000000277584 A cat looking in a door from the outside 
+000000565962 A cat peers through a window at a Chinese Buddha figure. 
+000000251572 A young dog relaxing with his or her human at the end of the day
+000000269866 Chrome colored toaster oven with time on 10.14
+000000034139 Suitcase sitting on the ground with stickers from various countries on it.
+000000096493 A little boy sitting down with his blanket and a toy.
+000000349594 A young boy is standing against a wall eating an apple.
+000000411665 A white and gray kitten pokes its paw at a mirror and looks at its reflection.
+000000250205 Two heron are standing on some rocks near some water.
+000000475387 Engine and train stopped at an indoor platform.
+000000282912 A close up of a giraffes neck and face.
+000000050943 A person riding a wave on top of a surfboard.
+000000235778 a teenager hanging paper tassels from a long string
+000000312237 People at the beach enjoying the day and surfing
+000000331352 A close-up picture of a toilet with a fountain.
+000000527220 A red Ferrari at a gas station showing the gas prices and three mopeds
+000000574297 a man sits on top of a horse 
+000000082807 A dog is sitting in a chair at a table.
+000000230008 A police officer on a motorcycle sitting at the corner. 
+000000089697 A man and woman sitting on top of benches.
+000000456865 A large white airplane parked upon a runway
+000000241326 a cat sitting on a couch with a dog in front of it
+000000095069 Several green spires on towers are part of the architecture.
+000000345385 A man snuggled up to another man under the covers in a bed.
+000000211674 a double decker bus with people waving from the top
+000000155443 A train traveling down the rail road tracks.
+000000289702 Large black dog sitting in front of a big mirror. 
+000000290843 A cat watches as its owner uses the laptop.
+000000513283 A small bunny character is on a plate of pizza.
+000000566436 A person sits on top of a bench.
+000000121153 Blue mile marker sign advertising two locations in two languages. 
+000000530836 An old advertisement shows a large kitchen with white cabinets and appliances and yellow walls.
+000000084650 A cat lying on a suit case and looking alertly.
+000000273420 a man on his phone in some kind of room
+000000148620 a table that has some computers on it
+000000485027 A tennis player is serving a tennis ball.
+000000396729 A cell phone and game case on a bed.
+000000291861 A lone Zebra near a rock with large rock face in background.
+000000573943 Parking lot full of different colored semi trucks.
+000000295478 A woman is walking with a dog trailing behind her on a leash. 
+000000475223 A bird standing next to a lot of tall grass.
+000000482978 a close up of a train with a view inside
+000000431896 A train parked at a waiting station where passengers board.
+000000213605 A bus parked next to a white building with red trimming. 
+000000393056 A young man ridding a small wave on a surfboard.
+000000431876 an image of horses that are standing on the sidewalk
+000000447200 A black and white small dog sitting next to a brown and white small dog.
+000000334767 Several people are skiing on a snow covered field.
+000000223955 A person stretches to catch a red Frisbee.
+000000424721 Meat with vegetable and fruit displayed in roasting pan.
+000000162581 A man on the court with his tennis racket raised.
+000000284743 a woman being photographed holding a baseball bat
+000000442746 A girl is playing in a grassy area.
+000000165500 three short furry bears sniffing in some weeds next to a grassy bank 
+000000076468 to black women tennis players shake hands on court
+000000104198 A sign is next to an ornate white gazebo.
+000000338718 A street scene with the street sign knocked over
+000000272212 One cow walking across a field and another cow laying down on the ground.
+000000463527 A blue and black ray holding sandwich, coffee and liquor bottles.
+000000269316 A group of men playing a game of frisbee.
+000000329455 A square-shaped pizza on a wooden board near white plates.
+000000244379 A red traffic light at an empty intersection
+000000477955 a para sailor launching his sail at the beach.
+000000053909 A person holding a cell phone with multiple images of their hand holding the phone.
+000000075456 A small homemade pizza in a pie plate.
+000000385029 A person cutting a pizza next to a salad and bottles of wine on wooden table.
+000000471087 A man with dark hair, a shirt and a tie posing for a picture.
+000000044877 A girl sitting in a wheelchair is playing tennis.
+000000347174 A tattooed woman wearing a white shirt is laying in a bed with a side table next to it.
+000000249025 An orange reddish rose in a vase filled with water on top of a table.
+000000049761 Wild zebra and wildebeest grazing near a watering hole. 
+000000192607 A muddied man standing next to a muddied truck.
+000000383386 These stuffed animals are decor on a bed
+000000485972 A chocolate dessert cake ready on a plate with a fork
+000000217060 a small air plane flying in the sky
+000000190676 A panoramic shot of several people standing near a plane.
+000000370042 Large assortment of small and large vases displayed outdoors.
+000000491090 A man sitting on one of a group of motorcycles.
+000000117719 A man changes a light bulb in a darkened bar.
+000000125572 People moving around on the streets surrounding an old castle.
+000000307658 A grizzly bear stares in the direction of the photographer.
+000000384661 a stainless steel stove and oven with a shelf beside it
+000000422998 A woman is preparing a sandwich on a paper plate.
+000000456662 A young girl stands in front of the refrigerator.
+000000365521 A young man riding a skateboard down a rail.
+000000549674 Two computer monitors behind a keyboard and mouse.
+000000518326 Creepy large doll heads laying in giant oversized fake beds.
+000000029596 An entertainment room set up with a table for two for coffee
+000000127955 there are two brown bears that are playing together in the water
+000000323571 A street sign on the street near a building.
+000000080671 The woman is learning how to ski on the snow.
+000000427500 A broken fire hydrant on a city sidewalk.
+000000127987 A bobble head baseball figurine on a desk. 
+000000157418 Two sandwiches and some spam on a table.
+000000434247 A group of people on horses on a beach.
+000000042276 A bathroom with ski equipment laid out on the floor.
+000000549136 A stuffed animal sits atop a barbed wire fence.
+000000297396 Chocolate dessert bars covered in frosting and sprinkles.
+000000192670 A player swings at a baseball while the catcher is ready to catch the ball.
+000000161861 A female tennis player is posed and ready for the ball to come.
+000000001993 A bedroom with a bed and small table near by.
+000000042296 a polar bear partially submerged in a body of water
+000000096001 A counter with porcelain figurines, an umbrella and a bowl. 
+000000255483 A person holding many papers and cutting one with scissors.
+000000449996 Planes flying and landing on a runway at an airport.
+000000063965 A woman holding a plate of cake with frosting and icing.
+000000329080 Messy bed next to bed with clothes on it in bright room.
+000000354829 A car parked in front of a building and fire hydrant in the rain.
+000000336053 two children are sitting at a table with food and drinks
+000000394275 a bus moving down the road with people inside
+000000433134 A gray and white cat standing on a lush green field.
+000000133087 Night time view of empty streets in commercial district.
+000000239843 A stop sign at the cross section of two streets.
+000000544444 A person skis down a snowy hill. 
+000000509735 A group of giraffes drink water in the wilderness.
+000000016451 two surf boars on a beach near the water 
+000000258793 a number of buildings in the distance 
+000000559707 A plate of prepared food has pasta, broccoli, and meat.
+000000002157 A plate of finger foods next to a blue and raspberry topped cake.
+000000309938 The puppy is sitting on the books in ront of the cubbyholes
+000000224675 A woman and man hold parts of a kite together.
+000000046804 A sheep standing in grass next to a rock wall.
+000000231580 a man talks in front of a group of kids 
+000000456394 A blue motorcycle is displayed in a showroom.
+000000187734 The people was looking at the balls and chair
+000000260106 A horse and rider making a jump over a fence on a field
+000000210708 A large and small elephant bathing in the river.
+000000314914 A group of signs on a pole on a city street.
+000000484351 A group if children having a snack at their desks while a group of adults in suits visits them. 
+000000559513 A blue plate filled with a large serving of foood
+000000262440 a bath room with a bath tub a toilet and a sink
+000000550349 An Orange bus is parked next to the people. 
+000000384651 The nightstand of a hotel room holds an unused class and an ice bucket. 
+000000008629 Pictures of several different types of quiche prepared differently.
+000000051610 A little child sitting in front of a laptop watching whats is on the screen.
+000000249025 Two orange roses in a small vase with water. 
+000000183127 A huge, crashing wave with lots of spray is carrying a kneeling wet-suited surfer to shore.
+000000520832 A person walking down a street in front of a building.
+000000400803 A man standing on top of an orange boat on a river.
+000000482275 A bride and groom are cutting a cake as they smile.
+000000459153 a person operating a laptop on a bench
+000000368900 A TV sits in a room with many other things. 
+000000161642 an image of clock that is on the wall
+000000190307 A woman rides the nose of a dolphin in a show
+000000518326 Three beds in a street with cartoon figures in the bed.
+000000346905 A LOT OF ELEPHANTS ARE GROUP TOGETHER 
+000000369751 A car is driving down the road near some road signs. 
+000000366711 A man surrounded by lemons and oranges making fresh juice
+000000279769 a banana some papers and a pencil on a wooden table
+000000239627 A white kitchen sink filled with dishes and eating utensils.
+000000250901 A sandwich that has a lot of lettuce on it.
+000000115118 Cargo train is traveling on a track next to a forest. 
+000000456496 A man sitting on cement by some birds
+000000356498 A pancake with vegetables piled up on it
+000000419653 Many containers of food are covered up on the table. 
+000000074256 Two women standing next to each other in a train station.
+000000545407 A gray and blue plane flies in the blue sky.
+000000250901 Three people opening a wrapped sandwich out on the grass.
+000000282298 A group of people milling about inside a large space in the winter.
+000000386277 An orange hanging alone at the end of a branch.
+000000161044 Two planes that are flying in the sky.
+000000330369 A boat going through the water with a wall in back.
+000000531036 A double decker bus next to a normal sized bus. 
+000000224724 There is a woman standing on a city street.
+000000093437 An old man is wearing an odd hat
+000000356347 A wooden spoon sticking out of a pot of food.
+000000539143 A train sits on train tracks under a bridge.
+000000273132 Older woman in dress sitting on a park bench.
+000000430056 A sandwich on a plate with three slices of bread, tomatoes and chips on the side. 
+000000331075 a dog that's been playing in the water
+000000498807 A young girl tries to keep her balance on a surfboard.
+000000508602 The white bird is standing on the roof of a car.
+000000045070 A small girl sits atop a saddled animal.
+000000246968 A woman standing in a kitchen making food.
+000000263644 A bedroom with a framed bed and a vanity.
+000000204329 A blue, yellow and green surfboard sticking out of a sandy beach.
+000000097230 Two elephants are facing each other on the side of a dirt road
+000000061658 Broccoli and cauliflower florets cut into small pieces and spread around.
+000000542089 A tile bathroom contains a tub and large sink. 
+000000245026 an image of a woman holding a cake 
+000000114049 The travelers are all traveling with their best luggage.
+000000122962 A group of young children sitting around a long table.
+000000033104 A group of people standing around each other in skis.
+000000344909 A multicolored fire hydrant that is on the sidewalk.
+000000422670 The little baby is reaching up with his hand.
+000000055072 a giraffe standing in a straw field next to shrubbery.
+000000166277 A cat is drinking something from a glass.
+000000369675 A red and white train traveling down train tracks.
+000000148730 an old photo of a boat parked next to the dock 
+000000109441 A picture of a reflection in a building. The view is from the street.
+000000229221 The calf is in the pasture staring at the camera 
+000000429623 A pizza sitting in front of a can of pepsi
+000000377635 A person doing skateboard ticks at a skate park
+000000179214 A white plate topped with a pizza and cheese.
+000000292024 a woman looking a pizza with assorted toppings 
+000000185292 a person is on a surfboard in the water
+000000111951 A bathroom with a glass shower, glass and wood vanity and a toilet.
+000000126137 A tennis player poses, racket in his right hand, left arm behind him.
+000000324818 Two parrots sitting on a branch side by side 
+000000078959 a bunch of bananas are hanging from the banana tree
+000000139099 A person on a motorcycle talking to a person with a watermelon.
+000000175251 a young boy brushing his teeth in the bathroom.
+000000008021 a large screen showing a person wearing a suit
+000000289594 A man is standing on a sidewalk leaning on a pole.
+000000314251 A group of scooters rides down a street
+000000522889 A female tennis player swinging her racket and running
+000000500211 several beach umbrellas folded in front of a large house with lots of windows
+000000349302 two giraffes standing under a tree to get some shade
+000000451435 A man holding a tennis racket playing on a tennis court.
+000000128112 a man that is on a snowy trail
+000000031217 a boy playing tennis runs to hit the ball 
+000000224051 Bicycle parked by the side of busy highway on a concrete slab. 
+000000248616 Three images show the man swinging a racket at the tennis ball. 
+000000387387 A school bus sitting in some overgrown grass
+000000154705  A couple of computers that are sitting on a desk.
+000000147338 A yellow fire hydrant with a cartoon face drawn on it.
+000000298738 A panda sitting in a field and eating.
+000000448365 a little boy that is jumping a skateboard
+000000015746 A picture of a fire hydrant in the grass.
+000000341196 A man riding a wave on top of a surfboard.
+000000493442 A woman is skiing down a snowy hill.
+000000505169 Stylish circular shower stall, next to a toilet in a compact bathroom.
+000000042628 Some people are walking down a snowy streets with white snow. 
+000000483050 A bed that is very nice and clean.
+000000552612 an image of a herd of sheep grazing in the pasture
+000000527528 people flying kites on a bright sunny day
+000000016010 A bunch of zebras playing on a grassy field.
+000000479448 Some bananas and strawberries on a large waffle
+000000074092 A woman jumping with a tennis racket in her hand.
+000000080413 The small stuffed bear is propped into the car dashboard.
+000000469192 A truck driving through a field with people flying kites.
+000000129945 a man is swinging a baseball bat at a game
+000000545958 A cow standing next to a white wall and bright blue door.
+000000465718 A computer desk with two monitors and a laptop.
+000000187055 A man with a ball facing a girl with a racquet.
+000000138550 there is a red fire hydrant on a street between two identical white poles
+000000360097 Luggage is arranged in groups  on a concrete  platform. 
+000000313034 A man in a green shirt stands by a girl holding a piece of cake on a plate.
+000000367386 A living room with some empty couches in it 
+000000207538 a microwave sits built in some cupboards 
+000000202228 A man takes his picture using the bathroom mirror.
+000000521509 A girl sits on the bed and photographs herself in the mirror with her tripod.
+000000320632 A herd of zebra standing next to each other on a green grass field.
+000000536947 A refrigerator on a patio flooded with water, beverage cans and a beer bottle. 
+000000307074 two stop and go lights at an intersection
+000000463174 A woman hits the tennis ball during a match.
+000000568439 A man walking down a street past a double decker bus.
+000000154947 one sheep is standing in some tall grass
+000000415194 An old fashioned kitchen with the light on. 
+000000384527 A group of chairs sitting around a table.
+000000096493 A little boy with a pink blanket and a remote control sitting on a couch.
+000000358525 A child in a hat playing on a laptop computer
+000000031269 A group of zebra standing on top of a grass covered hillside.
+000000352582 This is an image of a man trowing a frisbee.
+000000189226 A fire hydrant outside a shop with graffiti.
+000000194471 A boy on a skateboard at the skate park.
+000000251572 A woman laying on a couch with a white and brown puppy.
+000000117719 A man sitting at a bar that contains a lot of bottles.
+000000334309 A picture of a person eating some food.
+000000539445 a red train is riding down the tracks 
+000000026690 A skateboarder doing a trick at a competition.
+000000451084 A little kid that is doing a skateboard trick in the air.
+000000052413 Someone is holding a pick cellular flip phone. 
+000000463730 A street with two busses and people walking.
+000000488736 Large four sided clock hangs on the corner of the building.
+000000385190 a street corner is boarded by painted walls
+000000450758 THIS IS A FAMILY OF GIRAFFES RUNNING THROUGH THE BUSH
+000000388056 A baseball player is waiting for the pitch while standing in the batters box.
+000000263299 Large black train system surrounded by trees and nature.
+000000572620 a skateboarder in a blue shirt is doing a trick
+000000042528 someone that is holding a cell phone and has blue socks
+000000278749 a black and white photo of a person holding a sign
+000000570539 An older man is sitting on the bus above a movie advertisement.
+000000550084 The large machinery sits on the opposite side of the fence from the highway sign.
+000000390246 Person is riding a surfboard in the ocean.
+000000253695 A professional baseball player with his glove out.
+000000331817 A close up of a table with food, bowls and cups.
+000000061471 A brown dog standing next to a toilet in a bathroom.
+000000466256 A plate of food with a fork on a table 
+000000387148 Many livestock grazing in a field under a cloudy sky. 
+000000217285 The man is getting ready to hit the ball in the baseball game. 
+000000007088 A little girl holding an umbrella in the rain. 
+000000292005 a bathroom with some orange walls on it 
+000000146155 two people at a bar holding drinks 
+000000080274 A mother elephant and her baby are standing alongside a dry water pool.
+000000199551 A picture of a young boy laying on top of a bed.
+000000515266 A wooden bench with an old tire sitting on it.
+000000546011 A trio of zebras stand together in the grass.
+000000000632 Bedroom scene with a bookcase, blue comforter and window.
+000000340894 Two computers on a large wooden computer desk.
+000000169996 A street filled with traffic and men on bikes.
+000000231169 Small train coming down the rail road tracks.
+000000239041 Bathroom scene showing a television and a hair dryer.
+000000119828 a cat sits down in front of a laptop 
+000000581317 A woman holding a small item in a field.
+000000462728 A man is in the ocean waves near his surfboard with a dog on it.
+000000068093 A  pair of people sitting on a motorcycle, in the grass.
+000000009483 A man standing near a table with video equipment. 
+000000222559 A large body of water with boats in the water and various people in the water at the shoreline.
+000000282037 A man who is throwing a baseball to another man.
+000000034417 A picture of a factory making doughnuts full of joy. 
+000000012576 A woman sitting at a table topped with pizza.
+000000113720 A man and woman are sitting at a table eating pizza.
+000000297562  A baby getting a hair cut in the salon sitting in the chair
+000000104666 A reading room,in the corner of which paint board with a painting is kept.
+000000146155 Two men, one holding a baby, enjoy some wine at a wine bar.
+000000437110 A large building with a tower that has a clock.
+000000547519 A bear with an aggressive face lumbers through a forest. 
+000000004134 Two men standing next to each other holding hands.
+000000098839 A cat that is looking at a television screen.
+000000509260 some books shelves a television and a table
+000000398203 A bunch of little kids playing soccer with each other. 
+000000320706 The blender is full all the way to the top.
+000000130386 A sign on the side of a busy street outside a shopping center
+000000200667 A very cute cow by a big metal piece of equipment.
+000000553669 People sitting on benches in the park, and traffic in the street
+000000179487 a woman playing tennis and others watching her
+000000464476 A laptop computer sitting on top of a table.
+000000153782 A large clock standing among a building with a lamp in the from of the picture
+000000572303 A train speeding along the railroad tracks during the day.
+000000466125 A child with a backpack underneath an umbrella.
+000000447917 Children in green jerseys play soccer at the park.
+000000188592 A child in green pants is eating an apple. 
+000000564280 A dog is sitting on a couch by a remote.
+000000384527 The furniture is posed in the room with a sign that says do not touch. 
+000000146363 A beach chair and umbrella sit empty near the water.
+000000037670 Electronic wand displayed for modern type of toilet.
+000000352618 A papia, orange and bruised banana sitting in a bowl.
+000000154431 A living room with TV and entertainment center beside a sculpture of a torso.
+000000145665 A smiling young man laying in a parking spot.
+000000550691 A red double decker bus on a street.
+000000481404 The inside view of a large decorated church.
+000000116068 Man walking down a beach with a multicolored kite.
+000000362716 A tennis player is swinging at a tennis ball on a sunny day. 
+000000577862 Two giraffes standing next to each other in a holding pen. 
+000000128675 Several people engaged in para-sailing on a lake.
+000000025139 A zebra putting its face up close to a fence rail.
+000000259640 very well decorated and fruits in a very big box
+000000276055 a women that has a carrot in her hand
+000000516708 Sparta street sign hanging on a lamp in a street.
+000000378116 A surfer wearing a wetsuit is riding a wave. 
+000000090003 Two dogs who are playing with a Frisbee. 
+000000535306 A person with a skate board going getting ready to go down a ramp.
+000000459809 A group of people fly kites in the air.
+000000476215 A man holding horse reins connected to 2 horses on a dirt field.
+000000541291 There are several sinks and soap dispensers in the bathroom.
+000000063047 A man walking with a drink and a bag while listening to earbud headphones.
+000000398652 Round glass table full of potted plants and description cards
+000000551804 The woman is playing tennis on the court. 
+000000216636 a candle next to a bunt cake on a green plate
+000000537672 A kitchen area with a chair, tiled floor and a refrigerator.
+000000355905 Medium sized white dog enjoying a trot on the beach
+000000057232 Old baseball bats are turned into a seat.
+000000570169 Teddy bear holding a hammer in front of block set
+000000411817 some people are looking at different television sets
+000000210915 A person on a surfboard riding a wave.
+000000023359 A young man is snowboarding on top of some snow. 
+000000155145 Two boats beside each other with people holding on to the boats. 
+000000483999 Group of adults enjoying food and conversation together.
+000000163057 People are flying kites in the park and walking.
+000000134886 A low flying airplane over a parked silver van. 
+000000472298 a man and a woman sitting on a small boat
+000000387387 A school bus, parked in a rural area.
+000000312549 This is a close up photo of a street sign of a cross street.
+000000010764 a catcher kneeling at the mound waiting for a baseball 
+000000031217 The woman is looking at the tennis ball.
+000000239857 A woman is playing frisbee at a camp site.
+000000490413 A plane that is sitting on the asphalt behind a fence.
+000000423506 Two men standing next to one another, and both are smiling and wearing neckties.
+000000226154 A large green bus and people on a motor bike.
+000000453722 Living room area of home with wood flooring and fireplace.
+000000482800 an image of three people playing Frisbee outside
+000000007991 A plate that has a pile of carrots on a counter.
+000000111086 Two rows of cars parked next to a boat.
+000000427077 a photo of a baseball player, taken by Walzl's imperial portrait studios.
+000000078915 A lady is taking a shot at doing her best Tennis serve. 
+000000363784 An old and run down bathroom with dark red accent colors.
+000000551822 sandwich sitting on table next to two glasses 
+000000140929 a big clock on a pole saying it is 12:43
+000000052007 Two people about to get on-board a bus. 
+000000495146 People are in a parking lot  beside the water, while a train is in the background.
+000000052017 An old yellow plane is flying in the sky.
+000000444275 The microwave is out on the side of the street. 
+000000332570 A person holding a smart device in their left  hand.
+000000225757 A desk has two monitors and other electronics equipment.
+000000400922 A photograph of Key Bank with a clock under the sign.
+000000493905 A man jumping up in the air swinging a tennis racket at a ball.
+000000212453 A bath scene with a toilet and a wallpaper with a woman on it.
+000000562207 A group of people are standing next to an elephant emerging from the water.
+000000163562 Person standing near the water with a red disc in hand.
+000000178469 An open laptop on a desk surrounded by clutter.
+000000281447 A horse standing in the dirt near a fence.
+000000442661 The stuffed giraffe is looking out the high rise window.
+000000066523 A full size bed with mosquito netting above, all in neutral colors.
+000000087144 A man and woman sit on benches with a small child.
+000000051961 A graffiti covered wall and parking meter are seen in this image.
+000000178469 a page left open on  a laptop on a cluttered desk.
+000000476514 A man with red hair standing next to his bride.
+000000488251 A man standing in front of a brown horse near a small building
+000000463542 A group of skiers going downhill together. 
+000000330369 A red and white boat floating along a river.
+000000481413 A very small boy on the beach with a disc.
+000000216277 A partially eaten apple by a Verizon device.
+000000025986 A bunch of items that are on a counter.
+000000037988 A female tennis player on the court serving a ball.
+000000342186 A pizza with toppings served on a plate
+000000244019 A rusty fire hydrant is red white and blue.
+000000025139 A zebra grazing on food behind a metal fence.
+000000097988 A runner poses beneath a "Run for Rights" sign in a green city park. 
+000000147725 a public transit bus on a city street 
+000000214753 Two equestrians riding their horses on the beach together.
+000000132116 Two white bowls with vegetables, meats and herbs and chopsticks nearby.
+000000079034 a blue fire hydrant standing close to a tree 
+000000172547 some animals are grazing in a large field
+000000370478 A bedroom with a laptop, suitcase and backpack on the floor.
+000000166664 A big elephant walks on the road during a safari
+000000038048 A red fire hydrant is on the sidewalk next to a sign.
+000000327780 A fork cutting into a piece of cake
+000000239347 a man and a woman are laying in a bed
+000000508370 There are two men standing next to a bike and a doll.
+000000209747 A gray and white kitten in a white bathroom sink. 
+000000177935 A kitchen has a vintage gas range and yellow walls.
+000000347265 A young woman skiing through a snow covered forest.
+000000130699 Men playing soccer on a field at dusk.
+000000501023 Image of a street approaching California 138 with stop sign ahead.
+000000017115 The back end view of two zebras standing at a fence.
+000000540928 A cat walking on the wood floor beside a chair
+000000546826 A very close up view of some scissors and markers.
+000000460683 A man is wearing a tie with many words on it
+000000458045 An umpire officiates a game of little league baseball.
+000000110042 There is an old toilet sitting under a sign for this picture.
+000000082715 A man on a surfboard riding an ocean wave.
+000000512657 A couple of people in a room with remotes.
+000000286523 A man on a cellphone sticking his finger in his ear.
+000000369323 A woman holding a tennis racquet on top of a tennis court.
+000000547502 A group of dogs that are standing in the grass.
+000000252559 A group of people walking across a street below a tall building.
+000000473974 A herd of zebra in dry vegetation near dry trees.
+000000442836 older man tries to block younger man's frisbee throw
+000000198805 The yellow pick up truck is parked on the side of the road. 
+000000476514 TWO WOMEN BRIDES GETTING MARRIED STANDING UNDER AND UMBRELLA
+000000035197 Black and white photograph of person on a skateboard.
+000000413404 a couple of park benches next to some trees
+000000433192 A computer laptop is placed on a bed.
+000000092091 A little girl is using the toy brush for a chew toy rather than brushing the teddy.
+000000072795 A crowd watching a professional baseball game on a sunny day
+000000296231 A silver TV sitting on top of a TV stand.
+000000339442 Meat sandwich and dark berries on plaid cloth in front of tomato.
+000000323496 A LITTLE BOY PLAYING A GAME OF TENNIS 
+000000240049 A woman holding a young child in her arms.
+000000343934 A blue motorcycle is parked in a corner covered in graffiti.
+000000018150 A young man and a small boy are talking about a piece of pizza. 
+000000297830 happy dog poses outside a door with root beer bottle
+000000528399 A wooden table topped with plates and bows filled with food
+000000125850 A cat laying in a bowl asleep on the ground.
+000000210099 An orange cat sunning itself on a lawn chair.
+000000527695 A white and blue plate full of broccoli.
+000000246436 A man making food on a range in a kitchen.
+000000056288 A very tall sandwich is cut in half.
+000000309495 A blurry photo of a toilet seat in a bathroom.
+000000243199 A laptop sitting on a bed near a window
+000000383621 A passenger airplane is taking off into a cloudy sky.
+000000003845 A dinner plate that has white steamed rice with stir fry vegetables and chicken.
+000000558421 Several train tracks with a train on the far tracks.
+000000555972 There are purple flowers in a wooden vase. 
+000000288430 People on a sandy beach with a kite.
+000000416885 Several desserts on white plates near a glass of wine.
+000000160728 A man standing next to a child in a green boat.
+000000183709 People looking at a stop light on the other side of the road. 
+000000207538 A white microwave hangs above the oven under cabinets
+000000568584 A man sitting on a chair with his feet proper up .
+000000451043 A person skiing down a hill with goggles on.
+000000219271 A crowd of people walking along a snow covered mountain side.
+000000410712 a busy road full of street signs and cars 
+000000068628 Guy jumps off the steps doing a flip trick with his skateboard
+000000161128 A curse word written on a traffic sign.
+000000365745 A bomb squad vehicle in the middle of an intersection.
+000000275198 A woman standing next to a table covered with plates of mostly eaten food.
+000000084270 A busy airport with many people walking around.
+000000477805 A man with glasses sitting in front of a laptop computer.
+000000267537 People sitting under umbrellas on bleachers with other spectators.
+000000469192 A pickup truck is parked underneath multiple flying kites.
+000000168883 two women and two girls in makeup and one is talking on a cellphone
+000000058384 Sepia photograph of small elephants in a wildlife park
+000000500423 A two story train stopped at an indoor train stop.
+000000154004 People are sitting on the sand at the beach 
+000000032887 A man holding a stick standing next to a  green hillside.
+000000447522 Blue plate with green, white, and red vegetables on it. 
+000000130826 two elder people sit on a bench on a bus street
+000000402765 The controller of a Wii system is pictured.
+000000166165 A young woman plays tennis and wipes sweat from her face in this double shot.
+000000004765 a man in his wetsuit riding a big wave on his surfboard
+000000201418 A person waits for their luggage at an airport
+000000524742 The parked bus is covered in colorful graffiti.
+000000247838 That looks like a herd of moose in the snow.
+000000457848 a bunch of parked motorcycles sit in it line 
+000000088040 A display of fine food and spices on a wooden table.
+000000462629 Kitchen appliances outside a building with yellow doors. 
+000000128372 People loading onto a orange and silver metro bus.
+000000425925 large colorful clock with a brown top on a tall building near parked cars
+000000333237 A black and white photo of a bedroom.
+000000177893 a red and white passenger bus with a 'final destination 5' billboard on the side of it
+000000006818 This picture looks like a janitors closet with buckets on the floor.
+000000414170 People cross country skiing on a path between trees.
+000000348708 a bowl of food next to a very close banana
+000000108503 Two surfers in wetsuits carrying surfboards along the beach.
+000000482100 Two toilets sit outside a fence on the sidewalk
+000000284991 A man has a red water bottle up to his mouth.
+000000483531 Two single beds are pushed up against each other.
+000000460682 A horse laying in the grass by the water in a black and white photo 
+000000235241 some people and a woman in a pink shirt and a yellow frisbee
+000000293200 People standing in a field and flying kites.
+000000060932 A man holding a brides garter as a woman shows her muscles.
+000000311295 A group of zebras in a field by some trees.
+000000338532 A small giraffe standing on sand in its zoo enclosure
+000000126107 A big, white seagull flies over a beach.
+000000417465 A bear extends its long tongue while yawning
+000000022396 A passenger jet is flying across the sky.
+000000023034 A man riding on the back of a brown horse.
+000000246454 A girl kneels down to pet two dogs.
+000000366711 A man is squeezing orange juice from a pile of oranges.
+000000172330 A grey and white cat watches from between parked cars.
+000000064868 a man holding a cat in front of a stove 
+000000011699 Two women are imposing together as one holds her luggage.
+000000290179 Big Ben towering over the city of London England
+000000419312 Plates loaded with thanksgiving dinner and dessert with two glasses. 
+000000200667 A cow peaking its head into a room that contains mechanical equipment.
+000000054592 Two people on a snow covered hill with skis.
+000000136600 A laptop computer sitting on top of a wooden desk.
+000000222455 A filtered image of a microwave available to use in a store. 
+000000027620 A desk that has a laptop computer on it.
+000000235399 A dog laying in the back of a moving truck.
+000000530624 a close up of a dog laying on a bed 
+000000004795 A cat observing a computer screen next to a laptop and a cordless phone.
+000000349184 A woman sits on a bench watching the park.
+000000192964 A skateboarder performing a trick on an indoor ramp.
+000000206487 A large bus is making a turn in traffic. 
+000000270474 A little girl holding a baseball bat on grass.
+000000456496 A woman sitting in front of the Eiffel tower near pigeons.
+000000132931 A baseball game is in progress during the day time.
+000000177539 A man standing next to a surfboard on the sand.
+000000479732 A very interesting sandwich containing lots of vegetables.
+000000311950 A hot dog on a bun covered in ketchup and mustard.
+000000224807 A group of people that are sitting at a table talking.
+000000232692 A man jumping with a ramp while wake boarding
+000000377588 A group of men during a baseball game with wall in the background.
+000000240023 A tennis player dribbling the ball with  his tennis racket while wiping his face with his shirt.
+000000502336 A group of people walking down a street next to a tall building.
+000000411938 A person holding a huge teddy bear while riding a bike in the street.
+000000305609 Various plates of food that include hashbrowns, waffles and grits.
+000000344621 A living room filled with furniture and  set of glass doors.
+000000383606 A person taking a picture inside a bathroom with a blue sink and a brown door
+000000562581 A person is ready to hit a tennis ball on a tennis court.
+000000085576 A toilet in a restroom next to a busted toilet paper dispenser.
+000000044068 A teddy bear is sitting on a chair.
+000000181542 A woman that is standing in a crosswalk near mopeds.
+000000001675 A black and white cat relaxing inside a laptop.
+000000571313 A computer is on a desk across from a bed.
+000000329542 A hand adding a cherry to some small tarts.
+000000163290 Zebra and giraffe at a zoo in a cage. 
+000000397351 A vegetable vendor organizing his food for sale
+000000466339 A cat is sitting on the carpet in front of a door.
+000000328959 a vintage photo of a toilet sitting next to a shower 
+000000578093 a white yellow and blue train at an empty train station
+000000217425 A clock with a light in a clock tower at dusk.
+000000080932 A young boy sinking his teeth into a piece of pizza
+000000447088 A group of baseball players holding a black bat.
+000000429690 A man is hitting a baseball with a baseball bat.
+000000064523 A group of pretty young ladies standing on a tennis court.
+000000541634 a table that has a bowl of food and a cup on it
+000000212166 A small child is eating out of a tray.
+000000321887 A shorn sheep looks to the left of the picture. 
+000000270122 A surfer in the middle of two waves that have crested
+000000181542 A woman walking on a cross walk with cars driving on it.
+000000540932 A person rowing a kayak onto the shore of a beach
+000000494869 An adult is cooking at the kitchen counter near a baby and a dog.
+000000395903 A person in a wet suit walking into the ocean
+000000250205 Two seagulls standing on a rock looking out over the beach.
+000000477227 The boats are lined up at the dock on the water. 
+000000226984 The counters in this kitchen are covered with stuff.
+000000294163 A tall clock tower and a tree against a blue sky.
+000000374551 a brown gray and black bird sitting on a branch
+000000322211 A box of pastries is sitting on a table.
+000000410221 An airliner as seen from the ground, through some power lines.
+000000490413 A plane is surrounded by a high barbed-wire fence.
+000000394940 A young boy wearing a striped shirt eating some food. 
+000000420840 An adorable baby girl sitting at a table with food on her face.
+000000216636 A viw of a cake with pieces cut off of it.
+000000110884 A small cramped bathroom slightly cluttered with dingy walls.
+000000119452 A fruit basket showing a banana and oranges.  
+000000066706 a table that has a plate of food on it
+000000021503 Plate of crackers with something spread on them next to a keyboard
+000000121153 A street sign has a freeway sign in a different language.
+000000465675 Black and white photograph of a tractor on beach next to boats.
+000000459467 A small wheeled airplane on an open runway.
+000000235784 A women who is walking in the snow on skis.
+000000372819 a number of dogs running in a field with people on a bench
+000000386457 a cat stares through an opening between two doors 
+000000474039 A baby sitting in between to large stuffed animals.
+000000032817 Two men stand over the green water spilled from a toilet.
+000000148508 A red double-decker bus travels down the road
+000000531134 A traffic light with an orange and a red having faces drawn on them.
+000000042528 There is a person holding a cell phone in their hand.
+000000070739 Two women sitting at a table with wine glasses in their hands. 
+000000121153 Some large city signs in front of a big mountain.
+000000548267 A large grassy pasture with grazing sheep in it
+000000504074 a woman is sitting outside with a laptop
+000000567740 There are people standing in the snow on skis.
+000000288391 The cell phone is gold and hooked up to a wall. 
+000000216497 A kitchen featuring an open island for seating with high back stools.
+000000325991 A man in a red shirt tossing a white frisbee.
+000000101787 A large jetliner flying through a cloudy blue sky.
+000000218439 A man using a toothbrush in his room on bed
+000000199055 A man seems to be disguised as Marty McFly here, from Back to the Future. 
+000000064495 an image of a parking lot meter on the street
+000000058705 There are two people posing and one man is holding a banana 
+000000163057 A group of people flying kites on a grassy field next to buildings.
+000000577584 A microwave has a little board coming out from underneath.
+000000088269 A bowl of tomato soup and two pieces of grilled cheese.
+000000474854 A boy is sitting and eating a banana while smiling.
+000000286458 A briefcase sitting on top of a bed with lots of items in them.
+000000500663 some black cattle are grazing in a green field
+000000104803 Toilet with white splatter paint on a black seat in a restroom stall.
+000000094185 A teddy bear store is decorated with signs and bears.
+000000456394 BLUE MOTORCYCLE ON DISPLAY, SURROUNDED BY OTHER VEHICLES
+000000072281 A guy on a skate board high in the air.
+000000106235 A living room featuring a comfortable seating arrangement of chairs and tables.
+000000348481 A desk with various items that include cellphone, mouse and pocket change.
+000000512836 A woman with a red umbrella is walking two dogs in the snow.
+000000312237 People are having fun on a crowded beach.
+000000143068 A group of people flying kites on top of a sandy beach.
+000000124659 Fully decorated basement has no people in it.
+000000149622 Two small owls on a branch in a tree.
+000000471756 A smiling young boy lying on a surfboard rides in on the surf.
+000000058350 A young woman is holding a slice of pizza with her mouth.
+000000412894 People cross the street at the corner of luxury buildings.
+000000327592 a person holding a plate with food on it in one hand and  a spoon in the other.
+000000523100 Person canning oranges showing a large bowl of fruit wedges.
+000000085089 Three wine glasses being clinked together in a toast.
+000000205282 Giraffes in a stand of trees stare in the same direction.
+000000561223 A surfer wearing red and black shorts rides a white surf board.
+000000136033 A bathroom with orange walls and black tile features.
+000000279145 A wooden bench surrounded by potted plants in front of a house.
+000000392933 A giraffe is standing on a pile of logs in a grassy field next to trees.
+000000468632 A couple of men are running around in a field
+000000043816 Batter takes a swing at the incoming ball during the baseball game
+000000417608 A sandwich has meat, cheese, lettuce and tomato.
+000000483667 A man with black hair and a top hat
+000000079144 Two bear cubs are playing on a log.
+000000452891 A white dog standing on top of a wooden bench.
+000000338905 A group of people sitting at a table eating food.
+000000482319 a man and woman play with a small child 
+000000015440 A stop sign on post with a building in the background.
+000000356387 A stone street with bicycles and motor bikes parked on the side and people standing on the sidewalks in front of buildings.
+000000527695 Cooked broccoli in serving dish sitting on cloth hot pad.
+000000403122 Female tennis player preparing to serve the ball
+000000285047 A person sitting at table with a cup looking at a cell phone.
+000000050006 Boats are docked in a lake by a road.
+000000253695 A baseball player holds his glove in the air, waiting to catch a ball.
+000000346232 A group of people walking down a road.
+000000042178 A long train yard full of different equipment
+000000138115 Bald man making a face and pursing lips looking at orange.
+000000235778 A boy hangs up paper kites on a string. 
+000000052462 Two zebras are walking through tall brown grass.
+000000353096 A computer with an image of lighting on the screen. 
+000000421060 A person is skiing on a slope covered in snow.
+000000305695 A bunch of zebras grazing near a road where vehicles are driving by.
+000000009769 The pick up truck is plowing the snow. 
+000000261535 A woman who is holding a tennis racket and scratching her butt.
+000000411665 A kitten is looking into a mirror and touching the mirror with one paw.
+000000355905 A white dog is on a sandy beach while the sea foam washes ashore behind it.
+000000222455 a microwave oven with some food inside of it 
+000000301981 A lone giraffe standing next to a wall.
+000000207538 White microwave oven installed beneath cabinets in kitchens.
+000000022705 A woman is standing and smiling as she holds a wine glass in front of a refrigerator.
+000000085682 There is a picture of an outside territory.
+000000505451 A surfer is on top of an ocean wave.
+000000287545 A couple of giraffe standing in a building.
+000000555705 Orange and brown cat sitting on top of white shoes. 
+000000172595 A computer chair, with a computer desk, with both a laptop and desktop on top.
+000000005001 A group of people holding a large pair of scissors to a ribbon.
+000000100238 Some people posting in front of a camera for a picture. 
+000000180296 Doll wearing hat laid up next to open luggage 
+000000220732 a public bus driving on a city street
+000000319100 A man and woman sitting on a couch playing a game.
+000000092124 A white bathroom with an orange shower curtain.
+000000209829 A man that is standing on a surfboard in the water.
+000000312213 A grey and white cat sits near a laptop.
+000000308476 Polar bear licking rock near ball and fence.
+000000297427 A hamburger stuffed with fillings but a glazed doughnut in place of the top bun.
+000000153669 A player in action up to bat in a baseball game.
+000000518213 A yak walking through a field going towards a tree. 
+000000161609 A dog peeks out from the backpack of a man at the airport
+000000478721 Two people carrying a kite to be flown later.
+000000283785 A man reads a story to a group of children.  
+000000356094 A man prepares to swing his bat while another man looks at him on the baseball field. 
+000000311518 A green boat floating on top of a body of water.
+000000279278 Three people stand on the street talking next to a dog.
+000000081394 A white and blue truck parked on a street. 
+000000160772 A boat on the water with a lighthouse in the background. 
+000000570834 A train conductor standing next to a group of bikes.
+000000382743 Two kites flying in the sky over an open and closed umbrella.
+000000045070 A girl holding her leg out while riding on the back of a horse.
+000000384808 A man taking a picture of his reflection in a mirror.
+000000061333 an image of a cat sitting on the bed with a blanket
+000000222235 A cat plays with grass outside in the garden.
+000000225405 Two young boys kick a soccer ball on a soccer field
+000000085157 Group of men and women sitting at a table eating and talking
+000000344100 A half eaten sandwich sits on a table with a coffee near by.
+000000512648 a brown cow is standing under a green tree
+000000219440 A small group of cows standing in front of the camera.
+000000127494 Assortment of cooked vegetable on white plate on table.
+000000160556 A peson talks on a phone while playing on a computer
+000000357978 Woman with checkered shirt trying Wii Fit at Expo.
+000000479099 A red-brick wall with a gray bench in a lush green space.
+000000511453 French fries on a pizza with a bottle of beer.
+000000029984 Several people are swimming in the ocean with two empty chairs and an umbrella on the beach.
+000000316015 A cat on a desk next to a keyboard and computer
+000000472030 A park bench with a backpack on it in a forest of trees.
+000000430286 A sleep number bed with the comfort remotes.
+000000166563 A giant elephant is next to the Aurora rents sign. 
+000000572620 A crowd looks on as a young man does a skateboard trick.
+000000157756 A view of a clock tower, in the middle of the night.
+000000194506 Two bear cubs are playing together in water
+000000013546 A man on a skateboard doing tricks at the park
+000000004395 A man in a shirt and tie motioning with his hand.
+000000082765 Black and white photograph of a bed with a laptop on it.
+000000540280 A digital camera beside an umbrella type photography lighting equipment
+000000166277 a cat stands on a table drinking water out of a glass 
+000000445846 Spacious white kitchen with brown cabinetry, sink and appliances. 
+000000519569 A very nice large modern style kitchen with a bar.
+000000309391 Several vehicles are moving down a city street.
+000000019742 A glass with a lady bunched up holding her legs and head tilted with a with candle on the side and lights in the background.
+000000579307 A young boy who is holding a kite in his hands.
+000000293300 two elephants in a encloseur at a zoo
+000000107554 a surf board leaning on a wheel barrel 
+000000547519 A bear walking in the woods near several trees.
+000000416269 Passenger rail train traveling through tunnel with lights in.
+000000447169 A cramped kitchen in pictured in this image.
+000000452321 some red and black buses are parked and a building
+000000051008 A cat that is sitting on a laptop.
+000000291551 Littleboy performing a trick on his skateboard outside
+000000052996 A group of people standing in a kitchen.
+000000199236 A jouster rides on a horse through a field while an audience watches.
+000000043435 Surfers ride towering waves on the open ocean.
+000000267434 A herd of cows are by a brick house.
+000000411817 People shopping for interactive video games at Best Buy.
+000000347544 A Volvo bus parked on a road near a hotel.
+000000205514 An all white living room with stone flooring
+000000170474 a tennis player with a ball and a racket 
+000000556193 Two men are playing a Wii video game.
+000000244833 A yellow and a green motorcycle in the back of an auditorium.
+000000054628 A tall multi layer cake sitting on top of a blue table cloth.
+000000351810 A woman sitting at a table in front of a pile of luggage
+000000064718 A man hitting a tennis ball during a tennis match.
+000000263966 A horse standing near a wood fence being looked at through a camera lens
+000000286849 A group of three giraffe standing next to each other behind a rock.
+000000514914 A bathroom not being used contains a toilet, sink, shower and bathtub.
+000000260266 The traffic signal is at the intersection near a large building.
+000000146457 Two little girls standing next to each other making pizza.
+000000032334 Some friends pose for a picture while holding wine glasses.
+000000532129 The pizza has been cut into tiny square pieces.
+000000411774 A woman is sitting holding a bug swatter shaped like a tennis racket.
+000000318138 A young person is playing with a video game controller.
+000000428454 A man on a snowboard using a parachute.
+000000423944 When looking at mannequins without heads the clothing is superficial. 
+000000397354 a woman is giving a baby a bottle while others are congregated in the kitchen
+000000109798 Donuts with frosting and glazed toppings sit on table next to coffee maker
+000000229311 A stainless steel kitchen sink on a black granite countertop.
+000000509735 A group of zebras and giraffes by a water hole.
+000000520910 A person and a toilet standing in a room.
+000000057027 A baby elephant walks between the legs of its mother.
+000000575187 A man in black sweater feeding a giraffe through a fence.
+000000032887 Two men with back packs and walking sticks 
+000000506454 A park bench that is sitting in grass under a tree.
+000000061471 A puppy has pulled toilet paper across the bathroom floor.
+000000149375 A skateboarder is performing a tick in a skate park.
+000000115870 A group of people sitting around two couches.
+000000535306 A boy stands at the top of a jump with his skateboard.
+000000010583 A white plate sitting on a table filled with food. 
+000000338191 a collage with nine fire hydrants of different color combinations
+000000352584 A sculpture of a toilet made from woven wood. 
+000000059635 Woman in wet suit considers waves near forested coastline.
+000000162366 A box where you pay for your parking.
+000000435206 People and cattle come to the edge of the water on a lake.
+000000376442 There is a stuffed animal sitting on the toliet.
+000000559099 A herd of cattle stand in the field with a mountain in the background.
+000000500478 A baseball player is winding up for the pitch.
+000000568710 A charming living room utilizes a cast iron stove as a focal point.
+000000175387 Bathroom stall with green door in commercial business.
+000000057232 A chair is fashioned from baseball bats and two cushions.
+000000574520 A man surfing on his surfboard through the waves of the ocean.
+000000158548 Outdoor show with sheep prominent on grass field.
+000000110359 A jet plane is flying across the sky.
+000000353970 This cat is looking up from a chair.
+000000223182 a tennis player drying off his face with a towel 
+000000081738 Someone about to cut and birthday cake with blue and yellow frosting. 
+000000537241 a doughnut like pastry sitting on parchment paper
+000000440475 a living room with a lot of chairs and  a big entertainment center
+000000130566 A train with a red and yellow engine on a railroad track.
+000000394199 A woman with glasses and a red bandana around her neck, smiling next to a red motorcycle. 
+000000217219 A made up bed with a pile of newspapers on it.
+000000458045 A baseball player holding a bat next to home plate.
+000000299887 a man and a woman are standing beside a red motorcycle
+000000017905 A man standing next to a light and a sign.
+000000008762 A stoplight that is flashing green lights at cars.
+000000470173 a satute in a field of grass near a plant 
+000000409358 Brightly lit tiled wall in modern bathroom setting.
+000000086956 A group of people with a man in the background with a stick.
+000000451571 A close up of a corn on the cob that has been eating.
+000000015335 A boy sitting between a man and a woman.
+000000324715 A picture of a computer and some items on the table.
+000000477623 a close up of a train on a track with trees in the background
+000000543047 A living room and dining room have two tables, couches, and multiple chairs.
+000000383676 two giraffes taking turns eating leaves from a tree
+000000071877 Man in grey and black clothes grinding with his skateboard.
+000000555412 Leafy greens and sun dried tomatoes on something
+000000439290 A tall banana plant with bunches of green bananas.
+000000392722 A group of people standing outside of a double decker bus.
+000000129416 Several animals standing in the grass near a lake.
+000000261097 The tennis player is walking on the court.
+000000544605 A street pole with one ways, except deliveries, street and no walking signs.
+000000105923 A street with a shop that has a fake elephant in the window.
+000000359937 A double decker bus is parked in the parking lot.
+000000267537 A black and white scene with two couples under umbrellas and rain jackets sitting at a stadium.
+000000016598 a woman with blue hair holding her phone up
+000000032901 four men are posing for a picture at an event
+000000071938 A man sleeping with a phone on his cheek
+000000262227 A picture of a screen in the living room. 
+000000568147 A knit cover and note hang below a parking meter. 
+000000109827 A big teddy bear is leaning against a tree.
+000000161861 A woman tennis player getting ready to be served the ball.
+000000033759 A boy swinging a baseball bat at a game.
+000000548246 A group of people on a tennis court.
+000000058111 a cat laying on a table next to a stack of papers
+000000245320 A man riding a skateboard down a cement ramp.
+000000295797 The steeple had a clock and wind direction piece on top.
+000000308631  A motorcycle parked in front of a crowd of people.
+000000476415 A person stands and shows his tie off.
+000000221155 Two cow figures are standing in a yard.
+000000089271 Cat wearing a baseball cap with ears sticking out. 
+000000477955 A female kite surfer hoping a kite and a surfboard.
+000000242411 A grey and white clock tower next to street full of cars.
+000000329614 a double deckered bus on a city street
+000000503755 a woman in tennis gear holding a tennis racket on a tennis court
+000000279774 A baseball player is getting ready to hit a ball.
+000000149222 a keyboard and a monitor a mouse and a picture
+000000359781 The giraffe is walking toward a rock on a trail. 
+000000436738 An intersection with antique cars and a bus at it.
+000000077460 A group of people standing on a field under a cloudy blue sky.
+000000311303 There is a cup of pepsi, And a plate of food.
+000000135673 Two large airplanes flying through a blue sky.
+000000499266 A IMAGE THAT SAYS MAKE MEANING MATTER 
+000000001584 A double decker bus rolls along the streets.
+000000404534 A diner sitting on a street with the side of the building fenced in.
+000000423971 A modern looking commode in a very tiny bathroom. 
+000000410221 A jet liner flying above in the clear blue sky.
+000000459634 People are riding on motorcycles on a scenic road.
+000000451714 A skier is standing and posing for a photo
+000000297698 The skier is leaning forward while jumping through the air. 
+000000554579 Three women in long skirts and a dog standing near a Christmas tree.
+000000120853 A yummy looking sandwich with a salad and a drink.
+000000089296 A man holding a baseball bat preparing to swing it
+000000126216 The elephants are walking through the trees in the shade.
+000000173004 Plates of Pizza with silverware next to ketchup and other condiments. 
+000000294831 A plate with two slices of pizza on it.
+000000267434 Cows and steers grazing in an old-English pasture.
+000000168337 a city scene of a sidewalk with a red fire hydrant on a sidewalk next to an ATM
+000000143556 Some people riding some very nice looking motorcycles.
+000000066886 A baby girl is holding a pink brush as she scratches her head. 
+000000571008 The stop sign has stenciled spray paint on it. 
+000000458702 The parking meter stands unused next to the street.
+000000537802 Three teddy bears sit on a fake sled in fake snow.
+000000078565 A beach scene with sail boats, a dog, and people wading.
+000000449579 A woman holding a tennis racquet on a tennis court.
+000000409630 a keyboard and a mouse sit in front of a laptop 
+000000084270 An airport terminal with many people carrying their luggage.
+000000307658 a picture of a grizzly bear in the wild
+000000289741 A giant cruise ship next to another giant cruise ship next to a long pier.
+000000536073 A blender next to some tequila, a lime and a mixed drink.
+000000235057 A giraffe lapping up water with tongue near a pond.
+000000259690 A man in a wheel chair who is holding a baseball bat.
+000000553339 A road that has signs, trees, and telephone poles near it.
+000000003845 The food is prepared on the dish on the table. 
+000000383621 A large commercial airplane taking off into the sky
+000000571943 a pole with a street light and signs next to a bare tree
+000000361551 Some people who are placing luggage on a runway.
+000000210032 This is two birds pecking at the remnants of a burger at an outdoor restaurant.
+000000349837 A row of refrigerator freezers sitting next to each other.
+000000534827 Motorcyclists are riding down the street with each other.
+000000306139 Seven people seated at table talking and working on computer devices.
+000000136355 A kitchen with wooden furniture and a vase filled with red flowers.
+000000330554 A blender on the counter with liquid in it and spilled on the counter. 
+000000343315 A sidewalk grass median with a street sign and a fire hydrant.
+000000079969 The man is playing a game of tennis on the court.
+000000220584 A surfer takes a spill on a small wave.
+000000137294 a man outside cooking at an event and serving customers
+000000361268 Why would the cow be grazing in front of those homes?
+000000333697 Two signs that are on the side of a wall.
+000000535523 A machine in a factory with a blue chair sitting next to it.
+000000125778 A living room with dark colored furniture and a fire place.
+000000128658 A plate has a waffle, some fruit and ice cream on it.
+000000104603 A bear walks through tall grass near several trees.
+000000023023 Three pieces of luggage sitting on a floor in front of a curtain.
+000000234807 Horses are standing in a field outside of a crumbling castle.
+000000266082 An orange fire hydrant outside a modern office building
+000000091615 A very clean kitchen has stainless steel appliances.
+000000351810 A pile of luggage in a room with two women nearby.
+000000187271 A  refrigerator standing in front of some cabinets.
+000000471087 A man wearing a tie poses for a photo
+000000312213 Gray and white cat sitting next to open laptop.
+000000234413 A white toilet sits in a bathroom, with the lid open.
+000000309678 A person is holding a tomato above a tray.
+000000523194 A fire hydrant on the side of a road.
+000000143961 People are gathered together sitting for an event.
+000000437110 A building with many windows and a clock next to grassy field.
+000000499768 A fire hydrant with grass grown around it on a curb side.
+000000387098 A desk area with several computer screens and coffee mugs.
+000000194940 The small black bowls have peppers and carrots in them.
+000000397303 Two men at a table with ties sitting near a laptop.
+000000417085 An adult and a child cow standing on the side of the road.
+000000433103 Several children sit together while playing with plastic laptop computers.
+000000347174 A woman laying on quilted white bed in a room.
+000000442306 A girl in a jacket and boots with a black umbrella.
+000000163682 A businessman is sitting in a chair and smiling for a photo.
+000000017379 a flat screen TV embedded in a restroom mirror.
+000000172946 A group of people standing with remotes in hand.
+000000428867 Stuffed animals are displayed down a table with notecards.
+000000555412 A pizza covered in lots of greens on top of a table.
+000000184321 A blue train is next to a sidewalk on the rails.
+000000171050 The subject of this photo is a fire plug on a city street.
+000000226171 A desk with a computer, computer keyboard, mug, empty plastic bottle, red pen, and a laptop.
+000000220858 A small white boat floating on the shore of a lake.
+000000579893 A close up of the stop sign and to street signs.
+000000176847 The bird is eating berries from the tree.
+000000312549 Street signs for the intersection of Anza and 21st Avenue in front of a large building
+000000561679 there is a orange street sign with a white sign underneath
+000000033854 A bus, old cabin cars, and motor cycles with people 
+000000359855 Closed suit case laying on the ground next to a plant.
+000000044260 A collection of apples growing on a tree. 
+000000147518 an image of a clean full bathroom 
+000000414676 Tall stone tower building and beautiful blue sky.
+000000177934 Girls wash a motorcycle while men look on.
+000000078748 A gathering of people and several motor bikes.
+000000512330 There is a refrigerator that is filled with food.
+000000150265 A group of people walking down a sidewalk.
+000000473237 A young girl with a green jacket eating a piece of pepperoni pizza.
+000000449996 Two jumbo jets are driving on the runway of an airport.
+000000030675 The train engine number 6309 is operated by BNSF. 
+000000039951 A young man hitting a ball with a racquet.
+000000245311 A man working the counter at a pastry shop.
+000000417876 Two giraffes are standing together outside near a wall.
+000000104198 A stop sign with the work stop written in Arabic.
+000000037751 A person on a motorcycle riding on a mountain.
+000000507473 A woman brushing her teeth while taking a picture of herself
+000000431848 A person jumping up in the air on a skateboard.
+000000371552 a young boy on a soccer field standing in front of a ball.
+000000093437 A man in a jacket and hat looks at the camera.
+000000180792 A bathroom with a toilet, toilet roll and safety rail.
+000000479248 The TV is placed in a well decorated room.
+000000122166 A woman is riding her bike down the street in front of traffic.
+000000151657 There are two people dressed up in formal attire
+000000492992 a bird is standing on the back of a cow
+000000425906 a snow boarder doing a flip for the audience 
+000000194216 An outdoor area with a large bear and two smaller bears facing the wooded tree area.
+000000013774 A man standing on top of a river next to the sun and a flying kite.
+000000284279 A small bird resting and eating from a bird feeder.
+000000402519 A yellow and red bird is sitting in a tree.
+000000280918 Two beautiful young women baking a turkey in a pan..
+000000349860 A BOY WITH A BLUE SHIRT AND JEAN PANTS DOING A TRICK WITH HIS SKATEBOARD
+000000140987 A little girl grins at her plate of pizza.
+000000231831 A cat that is standing on a desk.
+000000468124 A double-decker bus is parked in a large field.
+000000285788 a couple of giraffes are standing by the rocks
+000000082846 A red trolley car that is sitting on pavement.
+000000071756 Two bears are interrupted on their walk by a surprise picture.
+000000218091 A small hotel room is furnished for business
+000000082715 A person that is surfing in the water.
+000000097337 A TV is on in this unique and eclectic living room.
+000000260105 A white plate with breakfast foods on it
+000000570539 THERE IS A MAN THAT IS LOOKING OUT OF THE WINDOW 
+000000090891 Five people sit in a room filled with bananas.
+000000037777 A small kitchen has various appliances and a table.
+000000390301 Broken fire hydrant has ice hanging out of it.
+000000303908 An empty wooden bench near the waters edge.
+000000287667 a wooden cutting board a knife some carrots and onions
+000000436617 a group of people sitting at a long dining table in a restaurant
+000000493772 A person holds an umbrella in the middle of a rode. 
+000000119445 A baseball player swings his bat while a ball floats above his head towards the catcher and umpire.
+000000186422 A grizzly bear exiting the water on a rocky shore.
+000000511999 Rail workers stand on a platform where a train is waiting.
+000000252716 A blue frisbee sitting on top of a lush green field.
+000000083172 A woman with a tennis racquet that is on a tennis court.
+000000033368 A person standing on a blue floor holding a tennis racket
+000000133778 A mother cow with a young calf feeding in a grassy area.
+000000424162 A few people riding bikes next to a dog on a leash.
+000000327306 A guy on a skate board does tricks on it 
+000000453860 Two pieces of hard luggage are seen here.
+000000365208 A person stands under tent on the beach. 
+000000189436 The large bear is made up of clay.
+000000500270 a person skateboarding near a skateboarding park at sunset.
+000000031620 a newly married couple cutting their wedding cake
+000000012667 A banana replacing the phone on an answering machine
+000000336309 A airplane is flying through a cloudy blue sky.
+000000221017 Rooster walking outside on the sand dunes at the beach.
+000000332570 the portable device is displaying a very important message.
+000000269121 A white-clad tennis player hurls a ball upward to deliver a serve.
+000000103585 A toilet with two sinks and mirrors with lights on
+000000172977 A zebra standing on top of a dirt and grass field.
+000000471756 A boy smiles as he rides his boogie board to shore.
+000000411953 a man is playing a guitar on a stage 
+000000371677 A couple of computer monitors sitting on top of a wooden desk.
+000000367818 A woman in tight blue jeans standing next to a brown horse.
+000000104424 A person on a court with a tennis racket.
+000000478286 A large long train on a steel track.
+000000514979 A baby elephant stands next to an adult elephant
+000000036936 A couple sits on a couch while playing wii.
+000000533536 A cat looking at a tv monitor on a desk
+000000235784 A woman is standing in the middle of the snow with ski sticks as the sun rises in the background.
+000000122969 The two zebras are standing beside each other. 
+000000193926 A series of photos showing a banana holder.
+000000127263 A couple of men playing a game of frisbee.
+000000214703 Pair of surfers paddling out to open ocean.
+000000486112 A man sprays an elephant with a water hose.
+000000484029 A stop sign with a street sign above it.
+000000426329 some frosted cakes with little decorations on top in a display case for sale
+000000507081 A view of a narrow kitchen with the only light coming from a glass door.
+000000445602 A man is at the bottom a ski slope.
+000000281929 A man standing outside a building holding a bicycle. 
+000000340015 A pizza with ingredients that include tomatoes and meatballs.
+000000369812 Various street signs including one that reads "Newlon Hale Village."
+000000357081 A photograph of two cows grazing on a green pasture.
+000000058705 A man and a woman posing together for a picture
+000000314182 An assortment of crackers, dip and veggies sitting on a white counter.
+000000557672 It's important to protect a baby from the sun.
+000000172649 A skater rides along a steep red wall.
+000000009483 A man in a suit standing next to a control board and computer.
+000000088432 The street corner of a busy intersection in a town.
+000000520910 Lady standing in a retro pink and turquoise bathroom.
+000000458702 A view of a street with some trees around it.
+000000124277 An elevated subway train in a major city on a sunny day.
+000000112634 Trees mark the far side of a fence that encloses a large environment space with man made rocks and two giraffes, one close up and very large, the other small, and seemingly far away.  
+000000078032 A picture of some type of park with benches and no people around.
+000000463037  a twin propeller air plane parked in the grass
+000000254814 Lot of two wheelers crossing across the road at a signal.
+000000080932 A child taking a bite of a pizza slice on a plate.
+000000527960 A green bench with graffiti in an area with trees.  
+000000097988 A guy smiling while standing under a run for rights banner.
+000000163155 A gray and white tiger striped cat sitting in front of a brickwall
+000000416170 a cat is sitting in a window that has a dirty exterior
+000000236426 A tennis player is jumping in the air at a tennis match.
+000000375493 a person standing next to a yellow fire hydrant
+000000105923 A stuffed elephant standing in a museum window.
+000000187249 A tv on with a man holding a knife at a guy's head and a bottle of mineral water in the foreground 
+000000425702 A lone surfer carries his board next to the shore.
+000000479596 Fruits sit in containers, ready for sale in a store.
+000000275058 A windsurfer glides across choppy seas in the glistening sun. 
+000000148999 A train driving past mural of working men while billowing smoke.
+000000526256 The cathedral has two clocks on each of it's walls. 
+000000367082 large dog in a brown chair by a window
+000000344795 A casserole covered in foil is inside an oven.
+000000229601 A baseball player who is running on a baseball field.
+000000323751 A white train with a yellow front running on rails. 
+000000501368 A person with a towel over them blow drying their hair
+000000390902 A man on a tennis court holding his racket and waiting
+000000435206 There are people standing on the shore next to a body of water.
+000000330396 A man and a woman standing on a stony beach with driftwood
+000000081738 a person holding a knife cuts into a cake 
+000000037670 A man stands next to a copier machine
+000000180135 A young man holding a microphone and a skate board.
+000000239537 A man holding a tennis racket getting ready to hit a tennis ball.
+000000554002 The black terrier stands about knee high to the group of people near him.
+000000177357 a man riding on top of a wave with a surfboard
+000000572900 A man holding a white frisbee in his right hand.
+000000512194 A vase with a handle has wild flowers in it.
+000000069224 two people at a table eating doughnuts and drinking milk
+000000155341 The street is blocked by a truck and a crane.
+000000184321 A blue and silver train next to train station and trees.
+000000206838 Two beautiful women riding horses in the ocean in bikinis.
+000000383842 Woman playing Nintendo Wii in font of business people in office
+000000183500 An old biplane that is flying low for the crowd.
+000000209142 a bowl of peas, broccoli and carrots with utensils
+000000509719 a close of up a small bird on a dirt ground
+000000308753 A cave is situated in a rocky wall.
+000000154087 People standing near a fence in the snow with Canadian flags flying.
+000000427655 a horse is pulling a carriage with people down a street
+000000279774 Two young boys playing t-ball at ball park
+000000453001 A girl is having a hard time pulling a suit case.
+000000282298 a large crowd of people walking through town near a large clock display with a christmas tree in the background
+000000529568 A nice kitchen has a large crystal chandelier.
+000000170474 A man who is about to server a tennis ball.
+000000082986 A group of men playing a game of frisbee.
+000000458702 These are two currently unused city parking meters.
+000000509656 A zebra in the shade on a sunny day 
+000000173057 People sit around a pool and a clock tower.
+000000154644 A LADY TALKING ON THE PHONE AND SMOKING A CIGARETTE. 
+000000199681 The door to a microwave is open and is exposing a stuffed animal is propped inside.
+000000061418 some kids playing with some large checkerboards painted on the ground 
+000000493019 A herd of giraffe standing next to each other eating.
+000000246883 two surfers walking on the beach carrying their surfboards
+000000326174 A mixed age group is going toward the ocean with surfboards.
+000000509656 There is a zebra in his habitat at the zoo.
+000000320632 There are several zebras standing close to each other.
+000000140556 some guys in green jackets standing around a fire hydrant 
+000000521141 The people are walking down the street in the city. 
+000000013597 A piece of chocolate pie sitting on top of a plate.
+000000303566 People are standing outside near bins of fruit.
+000000365207 A mirror shot of a dog sitting in a car.
+000000335450 Two horses stand by a fence in a grassy field.
+000000365655 One woman standing behind large specialized bus in mountains by glacier.
+000000559160 several boys hanging out at a skate park 
+000000360325 Crowded sheep of different colors stand in an enclosure.
+000000507037 A group of people standing outside of a building.
+000000095786 there are three vases made of clay on a table
+000000018833 A cat sleeping on the ground using a shoe as a pillow.
+000000378515 Vendors selling produce at an outside market. 
+000000185157 A blue sign points the way to the road.
+000000554266 Adults seen lying in bed in mosaic photograph montage.
+000000181816 A statue is sitting on a street bench.
+000000097022 An angled shot of a fully stocked kitchen with wooden cabinets.
+000000205647 A Fedex truck drives in front of hills.
+000000142790 Two snow boarders during a snowy day at the hill.
+000000362716 A male tennis player hitting a tennis ball. 
+000000181303 some douchey kid in vest who is about to fail at kite flying.
+000000284623 A black cat sitting inside a bathroom sink
+000000148707 A tree with lots of leaves and two lemons in it.
+000000035770 a couple of pans you would use to pee in
+000000146831 Someone riding a skateboard is going up the ramp.
+000000199055 A man wearing sunglasses getting ready to ride a skate board.
+000000154718 A person is standing over a dirty toilet in a bathroom stall. 
+000000394559 A young woman holding a tennis racquet on a tennis court.
+000000115885 A cat staring at the screensaver on a laptop.
+000000579070 A group of people sitting around a table. 
+000000389451 A herd of cattle walking across a beach next to the ocean.
+000000417779 A fire hydrant in the middle of the street curbside
+000000433134 A close up of a black and white cat in the grass.
+000000088848 A collage of photos including two which portray colorful fire hydrants.
+000000105923 Elephant displayed in window of city building near roadway.
+000000127530 a tennis player swinging a racket at a ball
+000000065798 A baseball player preparing to bat during a game.
+000000343453 A woman is on skis on the ski slope.
+000000489339 A woman in a wet suite carrying a surf board.
+000000114871 A white plate with two slices of cheese and a whole banana unpealed.
+000000151662 A giraffe standing on a fence to eat some grass.
+000000505565 The tall, skinny birds are standing in the grass. 
+000000343496 A person walking by stop sign at street intersection.
+000000336232 Vehicles are traveling at both ends of the intersection with no traffic control.
+000000559547 a black and white photo of three people standing 
+000000036539 A man and a child who are in the snow.
+000000031248 Living room with white chairs and couches, fireplace and books on bookshelves.
+000000066841 A toaster has been decorated with other kitchen items and googly eyes to look like a person. 
+000000383443 Large white bathroom showing sink, tub, TV, and countertops.
+000000453981 two baby animals are play fighting with eat other
+000000139871 a small plane is sitting on a runway
+000000504000 An airplane is vintage and is on display.
+000000139099 People at a veggie and fruit market looking at the merchandise.
+000000193429 A snow boarder sits in the snow looking at the camera. 
+000000335328 A man and a young child in wet suits surfing in the ocean.
+000000206025 A man wearing glasses and a lavender shirt. 
+000000564280 A dog on a red couch with the tv and Wii remotes.
+000000147729 A man holding up his smart phone to take a picture.
+000000474095 A man on the toilet drinking a beer and smoking a cigarette. 
+000000184791 A painting of a candlestick holder with a candle, several pieces of fruit and a vase, with a gold frame around the painting.
+000000024919 a number of horses standing near one another 
+000000547816 A bathroom has a shower, sink, and toilet in it.
+000000020059 Two zebras at the zoo eating some grass.
+000000132622 The large bear is looking in the direction of the camera.
+000000211674 A bus full of happy tourists drives down the street.
+000000430286 Three remotes sitting on top of a mattress.
+000000468332 The man is holding the child and lighting the candle.
+000000009590 A group of men sipping drinks and talking at a table.
+000000195045 A picture of a little house and some grass.
+000000396903 A red fighter plane on runway next to building.
+000000082812 People at a train station board and unboard a train
+000000286849 Four giraffes sticking their heads through a hole in some large rocks.
+000000104572 A bathroom with orange decorated wallpaper and a mirror. 
+000000159977 A giraffe looks down at a zebra on the field
+000000155443 A train traveling down tracks towards a train station.
+000000396205 Two cows in a meadow with a historic structure in the background.
+000000515025 A man sitting at a table with a cat eating food.
+000000445834 a couple of people are getting on a white bus
+000000206027 A piece of pizza with thousand island dressing and a Coke Zero on the side.
+000000029984 Four people are swimming in the sea with chairs on the beach.
+000000239537 a man swinging a tennis racket and hitting a ball on a tennis court.
+000000572555 A red, yellow and silver train traveling down train tracks.
+000000051326 Three Jack-O-Laturns of various shapes, one of which has flowers in it.
+000000169076 A dog laying on a bed with a television in the background.
+000000320696 A guy in a wet suite riding a surfboard on a wave.
+000000243989 A boy is looking to skate on his skateboard
+000000471567 the neck and head of a giraffe at the zoo
+000000244496 a man with a mustache is wearing a tie
+000000109313 A guy in a green shirt casually plays the Nintendo Wii.
+000000331317 A clock reads almost 2:30 with a building behind it
+000000575372 A street light with a mountain view in the background
+000000112634 Two giraffes standing near a large rock inside a gated corral.
+000000506279 a close up of a glass of wine on a table
+000000366199 A cat hides underneath the cover of blankets.
+000000071451 A clean bed in a hotel room with towels and soap laid out
+000000437239 A group of people stand next to a bench outside.
+000000402096 A sign with a red circle is sitting next to a curb.
+000000060899 A young boy stands in the living room playing with a brush
+000000348045 A view of a bunch of urinals with stickers above them.
+000000312552 Woman standing while eating pizza from a plate.
+000000173004 A dining table with some very tasty looking pizza.
+000000433243 Giraffe and a zebra standing side by side 
+000000283038 A large stop sign is set against city buildings.
+000000112378 a table covered with banansas and a coconut
+000000370813 Skateboard in middle of maneuver in the air with rider 
+000000368961 Two large elephants waiting to enter their shelter.
+000000089078 A woman in a pink jacket and printed pants snowboarding.
+000000010092 Bed in room with candles covered by a mosquito net
+000000579900 A small loaded pizza on a yellow plate
+000000481480 Horses racing on a dirt track at a race track.
+000000170595 A plate with a peanut butter sandwich, ham, and tomatoes. 
+000000180751 Man pouring different types of win into drinking glasses.
+000000379533 A zebra and elephants in a grassy area with sky in the background.
+000000005992 sheep standing around next to a brick wall 
+000000426795 A person is eating a gigantic pastry next to a chair.
+000000174018 Two teddy bears with one wearing a lrge pink shirt.
+000000468501 Three good friends are having a good time playing a game.
+000000028285 A small white tower off of a tiled roof.
+000000580197 Older man in tuxedo sitting next to another younger man in tuxedo.
+000000451155 A woman with her laptop on a bed in a dark room.
+000000404128 A plan is shown sitting on the pavement.
+000000202445 A cat laying in bed on top of a blanket.
+000000031217 A player on a tennis court swings a racket.
+000000370711 A display on a store front with lots of items on display.
+000000007888 A pedestal clock strikes 6:55 in a rainy sky.
+000000318114 The dessert is next to a glass of orange juice.
+000000073533 Two young boys sitting on a bench texting on their cell phones.
+000000513567 Two Asian women eating chili dogs while standing on a street.
+000000101884 The large brick building covered in vines has a clock on it. 
+000000358427 a close up of a  fork holding some cooked broccoli
+000000488736 a building with a large clock on the outside.
+000000108440 A train that is on a rail over a bridge.
+000000463522 A horse pulling  carriage travels through a busy area in a city
+000000370042 There are many vases on display outside the building.
+000000411938 People riding bicycles with a teddy bear on a city street
+000000294783 A large room with wooden floors and a bookshelf.
+000000541664 A laptop sitting on plastic blocks with some toy people.
+000000335529 A wooden bench next to the water where a lot a boats are.
+000000498032 A woman plays a video game with a controller.
+000000565607 Fire hydrant location in the middle of a flower field
+000000213224 A wooden table holding two vases of flowers.
+000000308165 A kid on skateboard jumps over the cone
+000000376307 A market has many items for sale including fruit.
+000000382125 The man smiles with a slice of pizza while next to a friend.
+000000289059 a man leaning over waiting for a tennis ball to hit 
+000000480275 There are many bananas being laid on the wagon.
+000000426241 A home office with a chrome and glass desk and computer equipment.
+000000231125 a person putting some pastries into a bag 
+000000157847 A large bird in the air over a heavily forested area.
+000000281032 A man stirring something in a foil roasting pan.
+000000041488 A large and green interstate exit sign above a road.
+000000248616 A man swings at a tennis ball on a tennis court.
+000000421455 Guy takes a picture  of his dog in the backseat through his side mirror. 
+000000159282 a glass vase with some flower coming out of it 
+000000055022 Bikes are lined up in the store on the floor
+000000579893 The stop sign is at a four way intersection. 
+000000345356 A couple of guys that were eating and are now on their cell phones.
+000000140203 woman is walking with a dog in her hands
+000000529528 A bowl full of batter below a blender.
+000000397681 Many pictures of different turnips and turnip like foods
+000000261535 A barefoot woman walking with a tennis racket and her hand on her butt. 
+000000048564 A man standing in the snow while talking on a cell phone.
+000000118594 A Texas longhorn bull eating grass by a barbed wire fence
+000000236592 A dog watching food cook in an oven.
+000000221754 There is very little traffic at this city intersection.
+000000470121 An assortment of pastries on a red table
+000000228436 there is a bike that is parked on a small bridge
+000000178744 A yellow boat with people is in a canal.
+000000329456 A woman holding something out in her hand
+000000118921 A woman on skis sliding down a snowy hill.
+000000036539 A man on snowboard standing by a mountain.
+000000579158 A Southwest Airlines jet pulled up to the terminal for loading.
+000000052017 A small airplane is flying in the sky.
+000000264335 a big bird flies through the air 
+000000174371 A woman in uniform shows something on her cell phone to a group of girls.
+000000115870 A large brown sectional sofa has people on it.
+000000543047 A colorful living area with living room, dining room and a kitchen.
+000000189698 a clean street sign that reads bodacious dr
+000000157928 A MAN IS PACKING UP SKIES ON THE SNOW LAND
+000000133819 A passenger bus that is driving down a street.
+000000194506 Two baby bears are playing with each other in the lake. 
+000000297353 A MAN STANDING ON A BED PLAYING A GUITAR
+000000316404 Two men wearing polos, headbands and shorts, holding tennis balls and rackets.
+000000532071 A field with a lot of dead trees.
+000000102805 A man wearing a hat and holding an object in his hand.
+000000545407  A jet airliner flying in the air carrying some passengers. 
+000000011122 The stop sign is behind the fence instead of on the street.
+000000481573 Boy standing on a skate board as seen from under the board.
+000000251065 A small tiled bethroom stall with a black toilet seat.
+000000156071 A group of people standing around at a rodeo.
+000000549390 A woman in business attire riding a bicycle. 
+000000032081 A woman holding a tennis racquet on a tennis court.
+000000425925 A large clock tower in the center of a city.
+000000022589 A couple of sheep standing next to each other on a field.
+000000315187 A bus parked in front of a building.
+000000089045 a large piano in a home living room
+000000021879 Several people are surfing or swimming at the ocean.
+000000052507 Surfer in board shorts walking in to the ocean
+000000418281 The white scared cow in a Tibetan city.
+000000506454 A park bench sitting on a bed of mulch next to a tree.
+000000127530 A man in blue shirt and white shorts playing tennis.
+000000360393 A thing is in the outline and it shows up like something 
+000000198960 Three big horn sheep are in an enclosed pasture. 
+000000060932 A black and white image of a formal occasion. 
+000000469246 A commuter train stops at a train station for passengers.
+000000356968 A brown horse with blonde hair standing in an open field.
+000000335529 An empty bench looking out over a bay with numerous boats on it.
+000000553339 A green road sign sitting on the side of a road.
+000000312213 A cat trying next to his owner who is using the computer. 
+000000181969 A cold dog curled up and going to sleep.
+000000133233 A marina filled with boats floating in crystal blue water.
+000000402992 Brown cows laying down in a field of grass. 
+000000464786 Hot dog with mustard on it sitting on a white napkin. 
+000000089271 A cat peeks over a chair while wearing a hat
+000000269932 A small horse on a large wooden chair.
+000000528578 The Big Ben clock tower towering over the city of London.
+000000308631 People stand around an antique motorcycle in a grassy area.
+000000403122 A female tennis player on grass about to serve
+000000433243 A giraffe and a zebra stand side by side as they eat.
+000000122672 An over head shot of a young person surfing on a surfboard. 
+000000226662 A black and white photograph of a motorcycle.
+000000247806 A clock tower is part of an old building.
+000000206411 A hand holding a spraying hose to a toilet bowl in a small toilet stall.
+000000466986 A man in a kitchen smiles towards a camera
+000000486479 A dog is on the floor hiding under a curtain.
+000000397351 Man selling yellow and red carrots at a vegetable stand 
+000000013177 A man in a green shirt working on a rim.
+000000075456 A pizza in a pan topped with cheese and meat. 
+000000288685 A man standing next to a herd of white sheep.
+000000206135 Four cows stand in the parking lot of a church. 
+000000403353 A lamp sitting next to a small bed with a book on top of it.
+000000463199 A group of commuters on a busy subway car or a bus.
+000000046252 A baseball player in a red shirt is ready to hit the ball.
+000000357430 A large combination pizza with two pieces gone.
+000000137106 a person holds their hand on a bed
+000000076417 A dog sticking it's head out of the drivers side window.
+000000164602 A couple taking a photo of themselves cleaning their teeth
+000000405691 Glasses of wine, salad and french bread on a wooden table. 
+000000520009 TWO PEOPLE ARE WALKING NEAR A MINI TRUCK IN A STREET.
+000000298251 A field full of zebras on a savannah grazing.
+000000365766 A kitchen is shown with cabinets and appliances.
+000000104119 A black and white chicken is walking through tall plants.
+000000474164 A dog sits on top of a truck.
+000000213547 A woman holds a cake with candles while riding a mechanical horse in a store.
+000000210502 A train on the tracks has a black and white striped door.
+000000546011 A few zebras are standing near tall grass.
+000000389933 A dog sitting on a couch with its head cocked to the side.
+000000492992 There is a bird standing on the back of the cow.
+000000050380 A  man guiding a pony with a boy riding on it.
+000000069138 A red traffic light with a sad face drawn over it.
+000000227686 A bridled mule stands in front of trees and leaves. 
+000000309391 A view of street with a bus parked on the side.
+000000142620 A man that is  sitting down on a bench in front of a table.
+000000312192 A white plate of food on a table.
+000000483999 A group of people at a dining table talking to each other.
+000000091500 two people sitting in chairs playing nintendo wii 
+000000368294 A woman working with a pair of scissors as she makes cigars.
+000000375015 A baby brown bear standing on top of a rock.
+000000532855 A man grinding his skateboard on a rail.
+000000460967 Photograph of an outdoor place with a bus.
+000000033114 Searchlights in front view and a plane on a wide runway in the background.
+000000458054 LOTS OF BROKEN TOILETS SITTING OUT ON A LAWN
+000000070254 A bright green train is pulled up for awaiting customers.
+000000460683 A man is showing off his religious tie
+000000449190 A table topped with a cut in half burrito next to cup of veggies.
+000000076261 Two birds flying in a gray sky next to a mountain.
+000000096427 A young boy holding a tennis racquet on a court.
+000000422886 a man eating a hotdog that has a lot of cheese
+000000290768 Three mugs are on the kitchen counter of a stove.
+000000381360 A man is surfing a big wave towards shore.
+000000424349 people in kitchen taking a pizza out of an oven.
+000000006614 Apples, grapes, an oranges and peanuts on the white surface in a picture.
+000000505638 A steam engine train traveling down train tracks.
+000000005586 A tennis player in a crowded arena waits for the next volley.
+000000311295 Zebras and wort hogs living together on the plains.
+000000003156 A man fixing a toilet in a black and white photo.
+000000276804 A small lit baseball field at night with people playing baseball.
+000000095843 a double decker bus sits parked on a curb 
+000000579655 A woman talking on a phone sitting next to glasses.
+000000426836 A boy is playing with a kite in a wet land.
+000000357060 A bird is trying to find food in the water. 
+000000410428 The small herd of sheep are grazing on the grassy field. 
+000000477288 Young people dressed in costume carry black umbrellas.
+000000000632 a bed room with a neatly made bed a window and a book shelf
+000000189436 a statue of  a bear wearing a red ribbon
+000000054628 Ornate wedding cake ready at the hotel reception
+000000126107 An up close shot of a seagull flying in front of the beach
+000000029397 A person wearing orange pants standing on a bench
+000000206271 Two urinals in a restroom with multicolored tile.
+000000276434 A couple cuts into a wedding cake with grapes and oranges. 
+000000133343 A man in a suite wearing a pair of sunglasses.
+000000216296 A tennis player takes a wide stance to return a ball.
+000000166509 a road with some trees on both sides of it 
+000000011122 a stop sign is in grass behind a fence
+000000516143 the signs states all directions and points up
+000000259625 An elephant that is using it's own trunk to touch a tree trunk.
+000000404249 A small boy rides a skateboard down the walkway.
+000000004765 A man in a yellow shirt and wetsuit surfing.
+000000308545 A young boy in a black jacket riding a white horse.
+000000101022 a street sign stuck inside of some snow 
+000000423104 A man getting ready to throw a Frisbee on a field.
+000000092053 Three white dishes with three different sorts of food on them.
+000000529966 Two bowls of lemons setting on a shelf.
+000000371749 Two woman playing a video game in their living room.
+000000437392 A pink, handmade puppet sits on the floor.
+000000269113 A pack of dogs are playing outside in the grass.
+000000293390 a white sink a towel and some brown cabinets
+000000044652 An old airplane is flying through the sky.
+000000209753 A young woman looking at her cell phone.
+000000290293 A train is in an overhead structure with lights.
+000000204186 A young man performing a trick on an ATV on a field.
+000000190007 this is man riding a wave on a board
+000000109313 A man is playing Nintendo Wii in a living room.
+000000394199 a woman on her bike in her yard
+000000221155 Black and white photo of cows near a windmill, house and shed.
+000000190853 A woman's hand holding a half peeled banana with a large bite out of it
+000000520910 a lady dressed in khakis standing in a bathroom next to the sink. 
+000000226147 A women who is taking a picture of her food.
+000000402346 an image of a bowl of tomatoes and a flower
+000000346707 A woman standing behind a very small banana tree with green bananas on it.
+000000577182 A bird sitting on top of a park bench.
+000000232088 a room with a couch and a tv in it
+000000011051 A woman pinning a flower onto a man's blazer.
+000000505451 A man surfs on a wave in the ocean.
+000000196185 A jet airplane taking off from a runway.
+000000555005 A large dogs comfortably sleeping on someones bed
+000000384651 A bedstand with an ice bucket, upside down glass, lamp, and phone sits next to a bed with pillows and sheets
+000000234660 Two white trains on tracks next to a station.
+000000466567 A tree with a donut as an ornament
+000000140076 A view of a table with a vase on the middle of it.
+000000460229 A red traffic stop sign with a blank white sign below it.
+000000127517 A row of surfboards leaned up against a wood rail in the sand.
+000000389451 A group of cows walking on a beach and people and a boat in the water.
+000000091615 a kitchen that has a stove and a sink in it
+000000522393 Sail boat of older design, in middle of expansive bay
+000000315001 A flock of seagulls flying over the water together.
+000000213171 A man standing in a dry baseball field.
+000000299720 A giraffe standing next to a building near a tree.
+000000042563 A metro style train speeding down the tracks between evergreen trees.
+000000505451 A surfer riding the break on a small wave.
+000000186632 A hotel room with a chandelier and glass fixtures.
+000000228144 A chair sitting in the middle of the room, in a black and white photo.
+000000495448 A wedding cake has pink and white flowers on it.
+000000154425 A young boy gets ready to fly a kite.
+000000210915 A person on a surfboard in the water.
+000000548506 A variety of fruit is shown at close up range.
+000000416269 A train that is at a train station at night time. 
+000000501243 three zebras standing together in a line by some trees 
+000000276921 Two teddy bears hang from the inside of a car.
+000000313783 Two burgers sit on a plate next to carrots and potatoes.
+000000135890 Big Ben and part of parliament from across the street.
+000000006763 A man and woman hugging in a restaurant
+000000236784 two dogs laying down on a brown couch
+000000309173 A red and white traffic sign attached to a pole in a city.
+000000084031 a baseball game featuring a batter a catcher and an umpire 
+000000210520 A bunch of food and flowers sits on a table.
+000000278973 A man starting to stand up on a surfboard in the ocean.
+000000023230 Four geese floating in the middle of a river.
+000000163155 A cat crouched on the edge of a porch 
+000000308466 A bathroom with a toilet, tub, mirror, window and a shower pole. 
+000000033707 Two giraffes stand together near rocks and a building.
+000000389381 One tray of rice and a tray of fruits and veggies.
+000000382009 A person on a court with a tennis racket.
+000000289417 A little kid that is snowboarding in the snow.
+000000378244 A skier skiing down a snow filled slope.
+000000509699 A PERSON IS STIING ON A COUCH WATCHING TV
+000000100624 A woman walking down a street talking on a cell phone.
+000000354547 A man with a beard wearing a white suit standing near a garden,
+000000315257 A bird with outstretched blue wings is sitting on some bird feeder.
+000000378244 Skier in a black going down the side of a mountain. 
+000000429718 Smoke is emanating from a train that is on a track that is curving.
+000000323355 A boy with a fedora on holding a pizza box open.
+000000570736 A bathroom with a vanity, toilet and shower.
+000000506004 the river has many buildings surrounding it 
+000000261061 A WOMAN IS STANDING LOOKING AT SOMETHING 
+000000212453 A white toilet sitting in a corner of a room.
+000000392228 A building with a very pointy roof and a clock.
+000000513524 a person riding a surf board on a wave
+000000103585 a bathroom with two sinks with a light on above them.
+000000534664 Several suitcases sitting on a shelf with tags on them.
+000000141821 Gourmet food on a picnic table with drinks
+000000575187 A man feeding a giraffe through a fence.
+000000286182 a man sitting in a room with a food on a table
+000000424135 An elaborate decoration hangs on the side of a tall building.
+000000509451 some adults and a child are in a photograph
+000000187513 A bathroom has green walls and a glass shower door.
+000000467776 A cow and a calf standing in a field
+000000075612 a bunch of kids are standing in a room
+000000033759 A young boy swinging a baseball bat at a baseball.
+000000173044 A clock tower stands in front of a large mirrored sky scraper. 
\ No newline at end of file
diff --git a/calculate_FID.py b/calculate_FID.py
new file mode 100644
index 00000000..9571c9fe
--- /dev/null
+++ b/calculate_FID.py
@@ -0,0 +1,90 @@
+import os
+import requests
+import zipfile
+from PIL import Image
+from pytorch_fid import fid_score
+import torch
+import argparse
+
+
+# Paths
+COCO_DOWNLOAD_URL = "http://images.cocodataset.org/zips/val2017.zip"
+COCO_DIR = "coco_dataset"
+REAL_IMAGES_DIR = os.path.join(COCO_DIR, "real_images")
+
+
+def download_coco(output_dir):
+    os.makedirs(output_dir, exist_ok=True)
+    coco_zip_path = os.path.join(output_dir, "val2017.zip")
+    
+    # Download MS-COCO validation set
+    print("Downloading MS-COCO validation images...")
+    with requests.get(COCO_DOWNLOAD_URL, stream=True) as r:
+        r.raise_for_status()
+        with open(coco_zip_path, "wb") as f:
+            for chunk in r.iter_content(chunk_size=8192):
+                f.write(chunk)
+    
+    print("Extracting images...")
+    with zipfile.ZipFile(coco_zip_path, "r") as zip_ref:
+        zip_ref.extractall(output_dir)
+    
+    os.rename(os.path.join(output_dir, "val2017"), REAL_IMAGES_DIR)
+    os.remove(coco_zip_path)
+    print(f"MS-COCO images downloaded and extracted to {REAL_IMAGES_DIR}")
+
+# Function to preprocess images (resize to 299x299)
+def preprocess_images(input_dir, output_dir, target_size=(299, 299)):
+    os.makedirs(output_dir, exist_ok=True)
+    for img_name in os.listdir(input_dir):
+        img_path = os.path.join(input_dir, img_name)
+        output_path = os.path.join(output_dir, img_name)
+        with Image.open(img_path) as img:
+            img = img.convert("RGB").resize(target_size)
+            img.save(output_path)
+    print(f"Preprocessed images saved to {output_dir}")
+
+# Function to calculate FID
+def calculate_fid(real_dir, generated_dir):
+    print("Calculating FID score...")
+    fid_value = fid_score.calculate_fid_given_paths(
+        [real_dir, generated_dir],
+        batch_size=50,  # Adjust based on your hardware
+        device="cuda" if torch.cuda.is_available() else "cpu",
+        dims=2048,  # Default feature dimensions for Inception-v3 pool3
+    )
+    print(f"FID Score: {fid_value}")
+    return fid_value
+
+
+
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Calculate FID for generated images.")
+    parser.add_argument(
+        "generated_images_dir",
+        type=str,
+        help="Path to the directory containing generated images."
+    )
+    args = parser.parse_args()
+
+    GENERATED_IMAGES_DIR = args.generated_images_dir
+    if not os.path.exists(REAL_IMAGES_DIR):
+        download_coco(COCO_DIR)
+    preprocessed_coco_dir = os.path.join(COCO_DIR, "preprocessed_real_images")
+    preprocess_images(REAL_IMAGES_DIR, preprocessed_coco_dir)
+
+    if not os.path.exists(GENERATED_IMAGES_DIR):
+        raise ValueError("Generated images directory does not exist. Add your images to 'generated_images/'.")
+    
+    print(f"Calculating FID for {GENERATED_IMAGES_DIR}...")
+    fid_value = calculate_fid(preprocessed_coco_dir, GENERATED_IMAGES_DIR)
+
+    fid_score_file = os.path.join(GENERATED_IMAGES_DIR, "fid_score.txt")
+    with open(fid_score_file, "w") as f:
+        f.write(f"FID Score: {fid_value}\\n")
+    print(f"FID score saved to {fid_score_file}")
+    
+    
+    
+# python calculate_FID.py --generated_images_dir outputs/flux-pab
\ No newline at end of file
diff --git a/examples/flux/sample.py b/examples/flux/sample.py
new file mode 100644
index 00000000..5d8d6a2a
--- /dev/null
+++ b/examples/flux/sample.py
@@ -0,0 +1,111 @@
+from videosys.pipelines.flux.pipeline_flux_pab import FluxConfig, FluxPipeline, FluxPABConfig
+import torch
+import time 
+
+
+def run_base():
+    # change num_gpus for multi-gpu inference
+    # sampling parameters are defined in the config
+    
+    # config = OpenSoraConfig(num_sampling_steps=30, cfg_scale=7.0, num_gpus=1)
+    # engine = VideoSysEngine(config)
+    config = FluxConfig()
+    
+    pipe = FluxPipeline(
+        config = config
+    )
+    
+    prompts = [
+        "Sunset over the sea.",
+        "A group of people wading around a large body of water.",
+        "A cabinet towels a toilet and a sink.",
+        "A few boys read comic books together outside.",
+        "There is a man and a young girl snowboarding.",
+        "A male snowboarder sits with his board on a snowy hill.",
+        "A woman wearing a skirt shows off her tattoos.",
+        "A large dog is tied up to a fire hydrant.",
+        "A man making a pizza in a kitchen."
+    ]
+
+    # Generate and save images
+    for prompt in prompts:
+        start_time = time.time()
+        image = pipe(
+            prompt,
+            height=1024,
+            width=1024,
+            guidance_scale=3.5,
+            num_inference_steps=50,
+            max_sequence_length=512,
+            generator=torch.Generator("cuda:0").manual_seed(0)
+        ).images[0]
+        end_time = time.time()
+        elapsed_time = end_time - start_time
+        print(f"'{prompt}' | {elapsed_time:.2f} s.")
+        pipe.save_image(image, f"./outputs/flux/{prompt}.png")
+
+def run_pab():
+    pab_config = FluxPABConfig(
+        spatial_broadcast=True,
+        spatial_threshold=[100, 930],
+        spatial_range=5,
+        temporal_broadcast=False,
+        cross_broadcast=True,
+        cross_threshold=[100, 930],
+        cross_range=5,
+        mlp_broadcast=True
+    )
+    config = FluxConfig(
+        enable_pab=True,
+        pab_config=pab_config)
+    pipe = FluxPipeline(
+        config = config
+    )
+
+    prompts = [
+        "Sunset over the sea.",
+        "A group of people wading around a large body of water.",
+        "A cabinet towels a toilet and a sink.",
+        "A few boys read comic books together outside.",
+        "There is a man and a young girl snowboarding.",
+        "A male snowboarder sits with his board on a snowy hill.",
+        "A woman wearing a skirt shows off her tattoos.",
+        "A large dog is tied up to a fire hydrant.",
+        "A man making a pizza in a kitchen."
+    ]
+
+    for prompt in prompts:
+        start_time = time.time()
+        image = pipe(
+            prompt,
+            height=1024,
+            width=1024,
+            guidance_scale=3.5,
+            num_inference_steps=50,
+            max_sequence_length=512,
+            generator=torch.Generator("cuda:0").manual_seed(0)
+        ).images[0]
+        end_time = time.time()
+        elapsed_time = end_time - start_time
+        print(f"'{prompt}' | {elapsed_time:.2f} s.")
+        pipe.save_image(image, f"./outputs/flux-pab/{prompt.replace(' ', '_')}.png")
+
+    # results = pipe(
+    #     prompts,
+    #     height=1024,
+    #     width=1024,
+    #     guidance_scale=3.5,
+    #     num_inference_steps=50,
+    #     max_sequence_length=512,
+    #     generator=torch.Generator("cuda:0").manual_seed(0)
+    # )
+
+    # for idx, image in enumerate(results.images):
+    #     safe_filename = f"./outputs/flux-pab-batch/image_{idx}.png"
+    #     pipe.save_image(image, safe_filename)
+        
+        
+if __name__ == "__main__":
+    run_base()
+    # run_low_mem()
+    run_pab()
diff --git a/examples/flux/sample_pab.py b/examples/flux/sample_pab.py
new file mode 100644
index 00000000..d4f7d485
--- /dev/null
+++ b/examples/flux/sample_pab.py
@@ -0,0 +1,139 @@
+# process_captions.py
+import os
+import time
+import argparse
+import csv
+from datetime import datetime
+from typing import List, Tuple
+from videosys.pipelines.flux.pipeline_flux_pab import FluxConfig, FluxPipeline, FluxPABConfig
+
+def read_caption_file(caption_file: str, start_idx: int, end_idx: int) -> List[Tuple[str, str]]:
+    """Read specific range of captions from file"""
+    caption_data = []
+    with open(caption_file, 'r') as f:
+        for i, line in enumerate(f):
+            if i >= start_idx and i < end_idx:
+                image_id, caption = line.strip().split(' ', 1)
+                caption_data.append((image_id, caption))
+            if i >= end_idx:
+                break
+    return caption_data
+
+def log_to_csv(log_file: str, entries: List[dict]):
+    """Write generation log to CSV file"""
+    fieldnames = ['batch_id', 'image_ids', 'prompts', 'start_time', 'end_time', 'processing_time']
+    
+    file_exists = os.path.isfile(log_file)
+    with open(log_file, 'a', newline='', encoding='utf-8') as f:
+        writer = csv.DictWriter(f, fieldnames=fieldnames)
+        if not file_exists:
+            writer.writeheader()
+        writer.writerows(entries)
+
+def process_captions(start_idx: int, end_idx: int, 
+                    caption_file: str = 'FID_caption.txt',
+                    output_dir: str = './outputs/flux-pab',
+                    log_file: str = './logs/generation_log.csv',
+                    batch_size: int = 4,
+                    height: int = 1024,
+                    width: int = 1024,
+                    guidance_scale: float = 3.5,
+                    num_steps: int = 50,
+                    max_seq_len: int = 512):
+    
+    # Load captions
+    caption_data = read_caption_file(caption_file, start_idx, end_idx)
+    print(f"Processing captions from index {start_idx} to {end_idx}")
+    
+    # Configure pipeline
+    pab_config = FluxPABConfig(
+        spatial_broadcast=True,
+        spatial_threshold=[100, 930],
+        spatial_range=5,
+        temporal_broadcast=False,
+        cross_broadcast=True,
+        cross_threshold=[100, 930],
+        cross_range=5,
+        mlp_broadcast=True
+    )
+    config = FluxConfig(
+        enable_pab=True,
+        pab_config=pab_config
+    )
+    
+    # Initialize pipeline
+    pipe = FluxPipeline(
+        config=config,
+        device="cuda"
+    )
+    
+    # Create output directories
+    os.makedirs(output_dir, exist_ok=True)
+    os.makedirs(os.path.dirname(log_file), exist_ok=True)
+    
+    # Process captions
+    log_entries = []
+    total_batches = (len(caption_data) + batch_size - 1) // batch_size
+    
+    for batch_idx in range(total_batches):
+        start_pos = batch_idx * batch_size
+        end_pos = min(start_pos + batch_size, len(caption_data))
+        batch_data = caption_data[start_pos:end_pos]
+        batch_ids, batch_prompts = zip(*batch_data)
+        
+        start_time = datetime.now()
+        
+        # Generate images
+        images = pipe(
+            list(batch_prompts),
+            height=height,
+            width=width,
+            guidance_scale=guidance_scale,
+            num_inference_steps=num_steps,
+            max_sequence_length=max_seq_len
+        ).images
+        
+        end_time = datetime.now()
+        processing_time = (end_time - start_time).total_seconds()
+        
+        # Save images
+        for image_id, image in zip(batch_ids, images):
+            save_path = os.path.join(output_dir, f"{image_id}.png")
+            image.save(save_path)
+        
+        # Log batch information
+        log_entry = {
+            'batch_id': batch_idx,
+            'image_ids': ','.join(batch_ids),
+            'prompts': '|'.join(batch_prompts),
+            'start_time': start_time.strftime('%Y-%m-%d %H:%M:%S.%f'),
+            'end_time': end_time.strftime('%Y-%m-%d %H:%M:%S.%f'),
+            'processing_time': processing_time
+        }
+        log_entries.append(log_entry)
+        
+        print(f"Batch {batch_idx + 1}/{total_batches} processed in {processing_time:.2f}s")
+        
+        if len(log_entries) >= 5:
+            log_to_csv(log_file, log_entries)
+            log_entries = []
+    
+    if log_entries:
+        log_to_csv(log_file, log_entries)
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description='Process captions with Flux pipeline')
+    parser.add_argument('--start-idx', type=int, required=True)
+    parser.add_argument('--end-idx', type=int, required=True)
+    parser.add_argument('--caption-file', type=str, default='FID_caption.txt')
+    parser.add_argument('--output-dir', type=str, default='./outputs/flux-pab')
+    parser.add_argument('--log-file', type=str, default='./logs/generation_log.csv')
+    parser.add_argument('--batch-size', type=int, default=4)
+    parser.add_argument('--height', type=int, default=1024)
+    parser.add_argument('--width', type=int, default=1024)
+    parser.add_argument('--guidance-scale', type=float, default=3.5)
+    parser.add_argument('--num-steps', type=int, default=50)
+    parser.add_argument('--max-seq-len', type=int, default=512)
+    
+    args = parser.parse_args()
+    process_captions(**vars(args))
\ No newline at end of file
diff --git a/examples/open_sora_plan/sample.py b/examples/open_sora_plan/sample.py
index f98a8016..d81a7eca 100644
--- a/examples/open_sora_plan/sample.py
+++ b/examples/open_sora_plan/sample.py
@@ -15,7 +15,7 @@ def run_base():
         guidance_scale=7.5,
         num_inference_steps=100,
         seed=-1,
-    ).video[0]
+    ).images[0]
     engine.save_video(video, f"./outputs/{prompt}.mp4")
 
 
diff --git a/flux-test.py b/flux-test.py
new file mode 100644
index 00000000..205d6b8b
--- /dev/null
+++ b/flux-test.py
@@ -0,0 +1,43 @@
+import os
+
+import torch
+
+os.environ["CUDA_VISIBLE_DEVICES"] = "5,6"
+
+from videosys.pipelines.flux.pipeline_flux_pab import FluxConfig, FluxPipeline
+
+pipe = FluxPipeline(
+    # "black-forest-labs/FLUX.1-dev",
+    # torch_dtype=torch.bfloat16,
+    # device_map="balanced",
+    config=FluxConfig()
+)
+
+# transformer = FluxTransformer2DModel.from_pretrained(
+#     "black-forest-labs/FLUX.1-dev",
+#     subfolder="transformer",
+#     torch_dtype=torch.bfloat16,
+#     low_cpu_mem_usage=True,
+#     offload_state_dict=False,
+# ).to("cuda:1")
+
+from diffusers import FluxPipeline
+
+pipe = FluxPipeline.from_pretrained(
+    "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16, device_map="balanced"  # 自动分配到多个 GPU
+)
+pipe.enable_model_cpu_offload()
+
+prompt = "A cat holding a sign that says hello world"
+
+image = pipe(
+    prompt,
+    height=1024,
+    width=1024,
+    guidance_scale=3.5,
+    num_inference_steps=50,
+    max_sequence_length=512,
+    generator=torch.Generator("cuda:0").manual_seed(0),
+).images[0]
+
+image.save("flux-dev.png")
diff --git a/flux_FID.sh b/flux_FID.sh
new file mode 100644
index 00000000..baba908c
--- /dev/null
+++ b/flux_FID.sh
@@ -0,0 +1,5 @@
+#!/bin/bash
+
+python calculate_FID.py --generated_images_dir outputs/flux-pab
+
+echo "calculate FID completed!"
\ No newline at end of file
diff --git a/flux_exp.sh b/flux_exp.sh
new file mode 100644
index 00000000..73f8958d
--- /dev/null
+++ b/flux_exp.sh
@@ -0,0 +1,57 @@
+#!/bin/bash
+
+# 设置总数和每个GPU处理的数量
+TOTAL_CAPTIONS=5000
+CAPTIONS_PER_GPU=$((TOTAL_CAPTIONS / 5))
+
+# 创建日志目录
+mkdir -p ./logs
+
+# 通用参数
+COMMON_ARGS="--caption-file FID_caption.txt \
+             --batch-size 4 \
+             --height 1024 \
+             --width 1024 \
+             --guidance-scale 3.5 \
+             --num-steps 50 \
+             --max-seq-len 512 \
+             --output-dir ./outputs/flux-pab"
+
+# 为每个GPU分配任务
+CUDA_VISIBLE_DEVICES=0 python examples/flux/sample_pab.py \
+    --start-idx 0 \
+    --end-idx $CAPTIONS_PER_GPU \
+    --log-file ./logs/gpu0_generation.csv \
+    $COMMON_ARGS \
+    > ./logs/gpu0.log 2>&1 &
+
+CUDA_VISIBLE_DEVICES=1 python examples/flux/sample_pab.py \
+    --start-idx $CAPTIONS_PER_GPU \
+    --end-idx $((CAPTIONS_PER_GPU * 2)) \
+    --log-file ./logs/gpu1_generation.csv \
+    $COMMON_ARGS \
+    > ./logs/gpu1.log 2>&1 &
+
+CUDA_VISIBLE_DEVICES=2 python examples/flux/sample_pab.py \
+    --start-idx $((CAPTIONS_PER_GPU * 2)) \
+    --end-idx $((CAPTIONS_PER_GPU * 3)) \
+    --log-file ./logs/gpu2_generation.csv \
+    $COMMON_ARGS \
+    > ./logs/gpu2.log 2>&1 &
+
+CUDA_VISIBLE_DEVICES=3 python examples/flux/sample_pab.py \
+    --start-idx $((CAPTIONS_PER_GPU * 3)) \
+    --end-idx $((CAPTIONS_PER_GPU * 4)) \
+    --log-file ./logs/gpu3_generation.csv \
+    $COMMON_ARGS \
+    > ./logs/gpu3.log 2>&1 &
+
+CUDA_VISIBLE_DEVICES=4 python examples/flux/sample_pab.py \
+    --start-idx $((CAPTIONS_PER_GPU * 4)) \
+    --end-idx $TOTAL_CAPTIONS \
+    --log-file ./logs/gpu4_generation.csv \
+    $COMMON_ARGS \
+    > ./logs/gpu4.log 2>&1 &
+
+# 等待所有进程完成
+wait
\ No newline at end of file
diff --git a/img1.png b/img1.png
new file mode 100644
index 00000000..084f2c95
Binary files /dev/null and b/img1.png differ
diff --git a/img2.png b/img2.png
new file mode 100644
index 00000000..545725e5
Binary files /dev/null and b/img2.png differ
diff --git a/latte_spatial_average_mse_plot.pdf b/latte_spatial_average_mse_plot.pdf
new file mode 100644
index 00000000..42ba644a
Binary files /dev/null and b/latte_spatial_average_mse_plot.pdf differ
diff --git a/latte_spatial_block_mse_plot.pdf b/latte_spatial_block_mse_plot.pdf
new file mode 100644
index 00000000..0df17951
Binary files /dev/null and b/latte_spatial_block_mse_plot.pdf differ
diff --git a/latte_temporal_average_mse_plot.pdf b/latte_temporal_average_mse_plot.pdf
new file mode 100644
index 00000000..62aea467
Binary files /dev/null and b/latte_temporal_average_mse_plot.pdf differ
diff --git a/latte_temporal_block_mse_plot.pdf b/latte_temporal_block_mse_plot.pdf
new file mode 100644
index 00000000..78a445c1
Binary files /dev/null and b/latte_temporal_block_mse_plot.pdf differ
diff --git a/open_sora_plan_spatial_average_mse_plot.pdf b/open_sora_plan_spatial_average_mse_plot.pdf
new file mode 100644
index 00000000..9b89289e
Binary files /dev/null and b/open_sora_plan_spatial_average_mse_plot.pdf differ
diff --git a/open_sora_plan_spatial_block_mse_plot.pdf b/open_sora_plan_spatial_block_mse_plot.pdf
new file mode 100644
index 00000000..cce6ec16
Binary files /dev/null and b/open_sora_plan_spatial_block_mse_plot.pdf differ
diff --git a/open_sora_plan_temporal_average_mse_plot.pdf b/open_sora_plan_temporal_average_mse_plot.pdf
new file mode 100644
index 00000000..12201870
Binary files /dev/null and b/open_sora_plan_temporal_average_mse_plot.pdf differ
diff --git a/open_sora_plan_temporal_block_mse_plot.pdf b/open_sora_plan_temporal_block_mse_plot.pdf
new file mode 100644
index 00000000..bf22017d
Binary files /dev/null and b/open_sora_plan_temporal_block_mse_plot.pdf differ
diff --git a/open_sora_spatial_average_mse_plot.pdf b/open_sora_spatial_average_mse_plot.pdf
new file mode 100644
index 00000000..ca97f3f7
Binary files /dev/null and b/open_sora_spatial_average_mse_plot.pdf differ
diff --git a/open_sora_spatial_block_mse_plot.pdf b/open_sora_spatial_block_mse_plot.pdf
new file mode 100644
index 00000000..d9074146
Binary files /dev/null and b/open_sora_spatial_block_mse_plot.pdf differ
diff --git a/open_sora_temporal_average_mse_plot.pdf b/open_sora_temporal_average_mse_plot.pdf
new file mode 100644
index 00000000..a2b720d3
Binary files /dev/null and b/open_sora_temporal_average_mse_plot.pdf differ
diff --git a/open_sora_temporal_block_mse_plot.pdf b/open_sora_temporal_block_mse_plot.pdf
new file mode 100644
index 00000000..429aa7d7
Binary files /dev/null and b/open_sora_temporal_block_mse_plot.pdf differ
diff --git a/sample_latte_mse.txt b/sample_latte_mse.txt
new file mode 100644
index 00000000..27d8b783
--- /dev/null
+++ b/sample_latte_mse.txt
@@ -0,0 +1,562 @@
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/diffusers/models/transformers/transformer_2d.py:34: FutureWarning: `Transformer2DModelOutput` is deprecated and will be removed in version 1.0.0. Importing `Transformer2DModelOutput` from `diffusers.models.transformer_2d` is deprecated and this will be removed in a future version. Please use `from diffusers.models.modeling_outputs import Transformer2DModelOutput`, instead.
+  deprecate("Transformer2DModelOutput", "1.0.0", deprecation_message)
+[10/01/24 15:05:16] INFO     colossalai - colossalai - INFO:
+                             /home/jin509/anaconda3/envs/opendit/lib/python3.10/
+                             site-packages/colossalai/initialize.py:60 launch
+                    INFO     colossalai - colossalai - INFO: Distributed
+                             environment is initialized, world size: 1
+
+Init SkipManager:
+                steps=50
+                cross_broadcast=True, cross_threshold=[100, 800], cross_gap=6
+                spatial_broadcast=True, spatial_threshold=[100, 800], spatial_gap=2, spatial_layer_range=[0, 28]
+                temporal_broadcast=True, temporal_threshold=[100, 800], temporal_gap=3
+                diffusion_skip=False, diffusion_timestep_respacing=None
+
+Some weights of the model checkpoint were not used when initializing LatteT2V_mse:
+ ['caption_projection.y_embedding']
+Special tokens have been added in the vocabulary, make sure the associated word embeddings are fine-tuned or trained.
+
+Downloading shards:   0%|          | 0/4 [00:00<?, ?it/s]
+Downloading shards: 100%|██████████| 4/4 [00:00<00:00, 699.37it/s]
+
+Loading checkpoint shards:   0%|          | 0/4 [00:00<?, ?it/s]
+Loading checkpoint shards:  25%|██▌       | 1/4 [00:03<00:11,  3.86s/it]
+Loading checkpoint shards:  50%|█████     | 2/4 [00:05<00:05,  2.60s/it]
+Loading checkpoint shards:  75%|███████▌  | 3/4 [00:07<00:02,  2.33s/it]
+Loading checkpoint shards: 100%|██████████| 4/4 [00:08<00:00,  1.79s/it]
+Loading checkpoint shards: 100%|██████████| 4/4 [00:08<00:00,  2.14s/it]
+Processing the (Sunset over the sea.) prompt
+
+  0%|          | 0/50 [00:00<?, ?it/s]
+  2%|▏         | 1/50 [00:05<04:53,  5.99s/it]Time step 960, Spatial MSE: [(0, 1.0004599971580319e-05), (1, 1.6409954923801706e-06), (2, 2.2087740944698453e-05), (3, 2.5976463803090155e-05), (4, 6.559863686561584e-05), (5, 4.8225701902993023e-05), (6, 8.425926353083923e-05), (7, 8.252511906903237e-05), (8, 0.00014971272321417928), (9, 0.00030662270728498697), (10, 0.0005700086476281285), (11, 0.000661017547827214), (12, 0.0006852964870631695), (13, 0.000755543471314013), (14, 0.003775363089516759), (15, 0.004852415528148413), (16, 0.03589743375778198), (17, 0.003639827249571681), (18, 0.004708926193416119), (19, 0.003987183328717947), (20, 0.0072733815759420395), (21, 0.0042415098287165165), (22, 0.0011923187412321568), (23, 0.0019837617874145508), (24, 0.005299860145896673), (25, 0.005599646829068661), (26, 0.004709301516413689), (27, 0.0020401340443640947)], Temporal MSE: [(0, 1.4483610357274301e-05), (1, 4.8103811423061416e-06), (2, 4.44661372966948e-06), (3, 8.535286724509206e-06), (4, 7.957279194670264e-06), (5, 1.306928515987238e-05), (6, 4.011851342511363e-05), (7, 6.913705146871507e-05), (8, 4.1768125811358914e-05), (9, 0.0002559793647378683), (10, 0.00038942249375395477), (11, 0.0003439167339820415), (12, 0.0004543345421552658), (13, 0.0005753498990088701), (14, 0.0008833783213049173), (15, 0.0004146500723436475), (16, 0.0008664758643135428), (17, 0.0004927171394228935), (18, 0.0003545025538187474), (19, 0.00017459708033129573), (20, 0.0005503034917637706), (21, 0.0006551106343977153), (22, 0.0004814696731045842), (23, 0.00041146459989249706), (24, 0.0004769708029925823), (25, 0.0005793106975033879), (26, 0.00029073975747451186), (27, 1.0152251888939645e-05)]
+
+  4%|▍         | 2/50 [00:16<06:45,  8.44s/it]Time step 940, Spatial MSE: [(0, 1.130597411247436e-05), (1, 1.8848315903596813e-06), (2, 2.180473347834777e-05), (3, 2.670003959792666e-05), (4, 5.263106504571624e-05), (5, 5.976920510875061e-05), (6, 0.00015111399989109486), (7, 0.00014421477681025863), (8, 0.00025616446509957314), (9, 0.0005170294898562133), (10, 0.0008074894431047142), (11, 0.000969650864135474), (12, 0.0011835292680189013), (13, 0.0011619632132351398), (14, 0.003910616040229797), (15, 0.005173811689019203), (16, 0.010542245581746101), (17, 0.0047149513848125935), (18, 0.005427862983196974), (19, 0.004216693341732025), (20, 0.008960708044469357), (21, 0.0040329452604055405), (22, 0.001628792379051447), (23, 0.002299085259437561), (24, 0.006162436679005623), (25, 0.007711181882768869), (26, 0.006203847471624613), (27, 0.0022086927201598883)], Temporal MSE: [(0, 1.562560828460846e-05), (1, 5.107947799842805e-06), (2, 5.210534254729282e-06), (3, 9.954833330994006e-06), (4, 8.566194992454257e-06), (5, 1.6934509403654374e-05), (6, 6.367373862303793e-05), (7, 8.414554758928716e-05), (8, 5.984687959426083e-05), (9, 0.00032865567482076585), (10, 0.00045923335710540414), (11, 0.0005340451607480645), (12, 0.0007651126361452043), (13, 0.0008183387690223753), (14, 0.0006142946658656001), (15, 0.0006101510371081531), (16, 0.0008704702486284077), (17, 0.000579786254093051), (18, 0.00037380075082182884), (19, 0.00021661515347659588), (20, 0.0005715799634344876), (21, 0.00045402528485283256), (22, 0.00047705534962005913), (23, 0.00047996899229474366), (24, 0.0005957384128123522), (25, 0.0007546633132733405), (26, 0.0003560635959729552), (27, 1.9425353457336314e-05)]
+
+  6%|▌         | 3/50 [00:27<07:48,  9.96s/it]Time step 920, Spatial MSE: [(0, 1.1775808161473833e-05), (1, 2.0007591956527904e-06), (2, 2.7083113309345208e-05), (3, 2.798098466882948e-05), (4, 4.647642344934866e-05), (5, 6.860559369670227e-05), (6, 0.00023742724442854524), (7, 0.00020579191914293915), (8, 0.0003424649185035378), (9, 0.0006722743273712695), (10, 0.0010069411946460605), (11, 0.0010634331265464425), (12, 0.001456133439205587), (13, 0.0013208183227106929), (14, 0.0034803906455636024), (15, 0.004724318161606789), (16, 0.01655917800962925), (17, 0.004779322072863579), (18, 0.0047007277607917786), (19, 0.004485907964408398), (20, 0.009380357339978218), (21, 0.003837226890027523), (22, 0.0016531123546883464), (23, 0.0023896796628832817), (24, 0.005931406281888485), (25, 0.009527095593512058), (26, 0.00691976910457015), (27, 0.002535630948841572)], Temporal MSE: [(0, 1.8000479030888528e-05), (1, 5.879944183107e-06), (2, 6.163899342936929e-06), (3, 1.204570071422495e-05), (4, 9.908755600918084e-06), (5, 2.0873916582786478e-05), (6, 7.772589742671698e-05), (7, 7.782198372296989e-05), (8, 7.42898482712917e-05), (9, 0.00037824956234544516), (10, 0.0004759605217259377), (11, 0.0006154854199849069), (12, 0.0008725501247681677), (13, 0.0009284872212447226), (14, 0.0008507939637638628), (15, 0.0006441243458539248), (16, 0.0008092052885331213), (17, 0.0006026425398886204), (18, 0.00040591589640825987), (19, 0.00022701825946569443), (20, 0.0005608931533060968), (21, 0.0005576232215389609), (22, 0.0005346618709154427), (23, 0.0005030888132750988), (24, 0.0006762503180652857), (25, 0.0009379482362419367), (26, 0.00039434258360415697), (27, 1.8174094293499365e-05)]
+
+  8%|▊         | 4/50 [00:36<07:15,  9.48s/it]Time step 900, Spatial MSE: [(0, 1.54780409502564e-05), (1, 2.3423265247402014e-06), (2, 3.2665804610587656e-05), (3, 3.4093409340130165e-05), (4, 4.487500336836092e-05), (5, 8.670583338243887e-05), (6, 0.00038590849726460874), (7, 0.0002969808701891452), (8, 0.00043960261973552406), (9, 0.0008082733838818967), (10, 0.0011954648653045297), (11, 0.0011440441012382507), (12, 0.0016279718838632107), (13, 0.0014078172389417887), (14, 0.003492486895993352), (15, 0.004234957043081522), (16, 0.019146909937262535), (17, 0.0051206559874117374), (18, 0.004443884827196598), (19, 0.0046932450495660305), (20, 0.009881289675831795), (21, 0.0037499601021409035), (22, 0.0017187506891787052), (23, 0.0023501866962760687), (24, 0.005415400955826044), (25, 0.010961074382066727), (26, 0.008936137892305851), (27, 0.0032041044905781746)], Temporal MSE: [(0, 2.220024907728657e-05), (1, 6.8676690716529265e-06), (2, 8.192887435143348e-06), (3, 1.6257117749773897e-05), (4, 1.2901757145300508e-05), (5, 3.2097235816763714e-05), (6, 9.729102021083236e-05), (7, 8.411816088482738e-05), (8, 0.00010222456330666319), (9, 0.0005058373208157718), (10, 0.0005570962093770504), (11, 0.0007084361859597266), (12, 0.0009199366322718561), (13, 0.0009881219593808055), (14, 0.0009188256808556616), (15, 0.0006238509085960686), (16, 0.0007723880116827786), (17, 0.0006243109819479287), (18, 0.0004087756387889385), (19, 0.0002614544937387109), (20, 0.0006064270855858922), (21, 0.0007213943754322827), (22, 0.0006741551915183663), (23, 0.0005967100150883198), (24, 0.000819002918433398), (25, 0.0012508021900430322), (26, 0.0004480942734517157), (27, 2.2868200176162645e-05)]
+
+ 10%|█         | 5/50 [00:48<07:37, 10.16s/it]Time step 880, Spatial MSE: [(0, 1.2200918718008325e-05), (1, 2.221475142505369e-06), (2, 3.9146299968706444e-05), (3, 2.74186568276491e-05), (4, 4.627216549124569e-05), (5, 8.758977492107078e-05), (6, 0.00033395038917660713), (7, 0.00035914056934416294), (8, 0.0004311249067541212), (9, 0.0007912490982562304), (10, 0.0010979186045005918), (11, 0.0011269174283370376), (12, 0.0016015070723369718), (13, 0.0012936347629874945), (14, 0.003305373014882207), (15, 0.004117573145776987), (16, 0.01672208495438099), (17, 0.0045096613466739655), (18, 0.0030168055091053247), (19, 0.0033804643899202347), (20, 0.0075416406616568565), (21, 0.0038904030807316303), (22, 0.0016472337301820517), (23, 0.002066843444481492), (24, 0.0049316054210066795), (25, 0.012624504044651985), (26, 0.006217173766344786), (27, 0.002993677044287324)], Temporal MSE: [(0, 1.882776996353641e-05), (1, 6.3547763602400664e-06), (2, 8.231228093791287e-06), (3, 1.3117050912114792e-05), (4, 1.438257095287554e-05), (5, 3.087832374149002e-05), (6, 0.00012245953257661313), (7, 8.749938569962978e-05), (8, 0.00011224779882468283), (9, 0.0004972488386556506), (10, 0.0005654257256537676), (11, 0.0007576883072033525), (12, 0.0008077445672824979), (13, 0.0010096157202497125), (14, 0.0009514306439086795), (15, 0.0006267433636821806), (16, 0.0006803994765505195), (17, 0.0006437339470721781), (18, 0.0004285034956410527), (19, 0.000299464154522866), (20, 0.0005857906653545797), (21, 0.0008353802841156721), (22, 0.000629801070317626), (23, 0.0005610439111478627), (24, 0.0007680190028622746), (25, 0.0011430152226239443), (26, 0.00037491682451218367), (27, 2.2261898266151547e-05)]
+
+ 12%|█▏        | 6/50 [00:57<07:23, 10.09s/it]Time step 860, Spatial MSE: [(0, 4.4900018110638484e-05), (1, 7.467044270015322e-06), (2, 4.354764678282663e-05), (3, 8.003390394151211e-05), (4, 0.0001395691797370091), (5, 0.00016734100063331425), (6, 0.0005923149874433875), (7, 0.0004827038792427629), (8, 0.0006415403913706541), (9, 0.0010295549873262644), (10, 0.0015170702245086432), (11, 0.0016341368900611997), (12, 0.0017016100464388728), (13, 0.0018567357910797), (14, 0.003918240778148174), (15, 0.006397046148777008), (16, 0.02412761189043522), (17, 0.00758601538836956), (18, 0.008656608872115612), (19, 0.007181904278695583), (20, 0.01755145564675331), (21, 0.011288395151495934), (22, 0.003037876682356), (23, 0.0032068314030766487), (24, 0.006694425363093615), (25, 0.0217900350689888), (26, 0.02296268194913864), (27, 0.011426301673054695)], Temporal MSE: [(0, 6.925113120814785e-05), (1, 1.9529881683411077e-05), (2, 2.3277843865798786e-05), (3, 3.0412678825086914e-05), (4, 3.000621472892817e-05), (5, 0.0001048940175678581), (6, 0.00012338435044512153), (7, 0.00013889283582102507), (8, 0.00018067556084133685), (9, 0.000761501956731081), (10, 0.0009851469658315182), (11, 0.001156673301011324), (12, 0.0010922917863354087), (13, 0.0016809299122542143), (14, 0.001285797799937427), (15, 0.0008081790874712169), (16, 0.0010394646087661386), (17, 0.0012418051483109593), (18, 0.0007350422674790025), (19, 0.00048293021973222494), (20, 0.0009841376449912786), (21, 0.0016095624305307865), (22, 0.0016829483211040497), (23, 0.0011798657942563295), (24, 0.0015944299520924687), (25, 0.0030295366887003183), (26, 0.0009952269028872252), (27, 0.00011528792674653232)]
+
+ 14%|█▍        | 7/50 [01:09<07:30, 10.48s/it]Time step 840, Spatial MSE: [(0, 5.012794645153917e-06), (1, 2.801358050419367e-06), (2, 5.8947473007719964e-05), (3, 1.786705615813844e-05), (4, 3.451944576227106e-05), (5, 5.356393739930354e-05), (6, 0.00022892198467161506), (7, 0.00025793214445002377), (8, 0.0003242660022806376), (9, 0.0005755130550824106), (10, 0.0007874852162785828), (11, 0.0007950991857796907), (12, 0.000790190533734858), (13, 0.0010111145675182343), (14, 0.001345641678199172), (15, 0.0016183798434212804), (16, 0.0012352861231192946), (17, 0.000987225561402738), (18, 0.0008750851848162711), (19, 0.001159542822279036), (20, 0.0017417581984773278), (21, 0.0008638601866550744), (22, 0.0008248954545706511), (23, 0.001095634768716991), (24, 0.0023189301136881113), (25, 0.010238254442811012), (26, 0.0032390542328357697), (27, 0.004584218841046095)], Temporal MSE: [(0, 8.438489203399513e-06), (1, 3.10590144181333e-06), (2, 7.783930414007045e-06), (3, 1.5975823771441355e-05), (4, 1.2311092177696992e-05), (5, 2.0943136405549012e-05), (6, 9.90235639619641e-05), (7, 6.534433487104252e-05), (8, 8.305694063892588e-05), (9, 0.00019594974583014846), (10, 0.0003822371072601527), (11, 0.0004294735554140061), (12, 0.00031260584364645183), (13, 0.0005677127046510577), (14, 0.0006805642042309046), (15, 0.00039443743298761547), (16, 0.00041145269642584026), (17, 0.0005161703447811306), (18, 0.00045665621291846037), (19, 0.0005381451337598264), (20, 0.0008637594874016941), (21, 0.0008422645623795688), (22, 0.0005402770475484431), (23, 0.0004694348608609289), (24, 0.0009608497493900359), (25, 0.0013567350106313825), (26, 0.00036757250200025737), (27, 3.53473296854645e-05)]
+
+ 16%|█▌        | 8/50 [01:21<07:38, 10.92s/it]Time step 820, Spatial MSE: [(0, 1.2880926988145802e-05), (1, 2.6339503165218048e-06), (2, 6.244335236260667e-05), (3, 3.242698949179612e-05), (4, 8.387464185943827e-05), (5, 8.002847607713193e-05), (6, 0.0003476240672171116), (7, 0.00034619454527273774), (8, 0.0003833285882137716), (9, 0.0006416966789402068), (10, 0.0009451353107579052), (11, 0.0008412243332713842), (12, 0.0008308895048685372), (13, 0.0010052337311208248), (14, 0.001706784125417471), (15, 0.0018961740424856544), (16, 0.0022794133983552456), (17, 0.0017033941112458706), (18, 0.0015151248080655932), (19, 0.0022644847631454468), (20, 0.004985900595784187), (21, 0.0026208613999187946), (22, 0.0012965825153514743), (23, 0.0015767525183036923), (24, 0.003313920460641384), (25, 0.017890863120555878), (26, 0.010409306734800339), (27, 0.0038925898261368275)], Temporal MSE: [(0, 1.8748207367025316e-05), (1, 6.274560291785747e-06), (2, 1.5170901860983577e-05), (3, 2.0067356672370806e-05), (4, 1.8345637727179565e-05), (5, 4.561388414003886e-05), (6, 0.00010577763896435499), (7, 8.947491005528718e-05), (8, 0.00012024649913655594), (9, 0.0003694626211654395), (10, 0.000605148496106267), (11, 0.0007116458145901561), (12, 0.0004966477281413972), (13, 0.0007650509942322969), (14, 0.0006821519345976412), (15, 0.00048540180432610214), (16, 0.0004966090782545507), (17, 0.0007569415029138327), (18, 0.0005047463346272707), (19, 0.0006152294808998704), (20, 0.0008567156037315726), (21, 0.001248213928192854), (22, 0.0009336344082839787), (23, 0.0007906037499196827), (24, 0.0012558629969134927), (25, 0.002389185596257448), (26, 0.0006521213799715042), (27, 4.305051697883755e-05)]
+
+ 18%|█▊        | 9/50 [01:29<06:49,  9.99s/it]Time step 800, Spatial MSE: [(0, 1.2726863133138977e-05), (1, 3.12896531795559e-06), (2, 7.811508112354204e-05), (3, 3.139224645565264e-05), (4, 9.708053403301165e-05), (5, 7.868415559642017e-05), (6, 0.00025839259615167975), (7, 0.0003213979071006179), (8, 0.00036914378870278597), (9, 0.000591723364777863), (10, 0.0010299826972186565), (11, 0.0007945071556605399), (12, 0.000788831734098494), (13, 0.0009609038825146854), (14, 0.0017463825643062592), (15, 0.002029283670708537), (16, 0.002216068096458912), (17, 0.0017545404843986034), (18, 0.0018425113521516323), (19, 0.0024309689179062843), (20, 0.005345581099390984), (21, 0.004193762317299843), (22, 0.0013159530935809016), (23, 0.0016614892520010471), (24, 0.003545469371601939), (25, 0.019399555400013924), (26, 0.00984680000692606), (27, 0.004402470774948597)], Temporal MSE: [(0, 1.8798577002598904e-05), (1, 6.033080353518017e-06), (2, 1.642720053496305e-05), (3, 1.954699109774083e-05), (4, 2.06251788767986e-05), (5, 4.297222039895132e-05), (6, 9.82664932962507e-05), (7, 9.077105642063543e-05), (8, 0.00012615029118023813), (9, 0.00036205287324264646), (10, 0.0005295243463478982), (11, 0.0006106580840423703), (12, 0.0004339663137216121), (13, 0.0006643964443355799), (14, 0.0006255283369682729), (15, 0.0005206167115829885), (16, 0.0004988067666999996), (17, 0.0007913003209978342), (18, 0.0005527404719032347), (19, 0.0006730066961608827), (20, 0.0008948344038799405), (21, 0.0014579107519239187), (22, 0.0009628403931856155), (23, 0.0008911953191272914), (24, 0.0015193331055343151), (25, 0.0029453346505761147), (26, 0.0007398661691695452), (27, 5.5188564147101715e-05)]
+
+ 20%|██        | 10/50 [01:38<06:36,  9.92s/it]Time step 780, Spatial MSE: [(0, 9.888132808555383e-06), (1, 2.7246560421190225e-06), (2, 6.893301178934053e-05), (3, 2.00634549401002e-05), (4, 2.6660598450689577e-05), (5, 3.8118545489851385e-05), (6, 5.736707680625841e-05), (7, 0.00014112566714175045), (8, 0.00015879236161708832), (9, 0.0001970492012333125), (10, 0.0007737191626802087), (11, 0.0007603054400533438), (12, 0.0006614312878809869), (13, 0.0010273582302033901), (14, 0.0016887316014617682), (15, 0.0014684689231216908), (16, 0.0013318778946995735), (17, 0.001620435155928135), (18, 0.0009107545483857393), (19, 0.0009006019681692123), (20, 0.0013489164412021637), (21, 0.0011147395707666874), (22, 0.0006143148639239371), (23, 0.0011993490625172853), (24, 0.0027458530385047197), (25, 0.007259247358888388), (26, 0.0029661592561751604), (27, 0.00514334999024868)], Temporal MSE: [(0, 1.3063781807431951e-05), (1, 5.170998065295862e-06), (2, 1.0949936950055417e-05), (3, 1.2218935808050446e-05), (4, 1.1425684533605818e-05), (5, 2.3577265892527066e-05), (6, 7.388753147097304e-05), (7, 0.00012981724285054952), (8, 7.237630052259192e-05), (9, 0.0002983918529935181), (10, 0.0006027296767570078), (11, 0.0004585782589856535), (12, 0.0004823015769943595), (13, 0.0006292805774137378), (14, 0.0007755337283015251), (15, 0.000783377792686224), (16, 0.0010677666869014502), (17, 0.0008090613409876823), (18, 0.0009203556110151112), (19, 0.000363610393833369), (20, 0.0009737354121170938), (21, 0.00125784520059824), (22, 0.0002626229834277183), (23, 0.0005574710667133331), (24, 0.0009677994530647993), (25, 0.0017883897526189685), (26, 0.0004580723471008241), (27, 4.2442108679097146e-05)]
+
+ 22%|██▏       | 11/50 [01:49<06:40, 10.28s/it]Time step 760, Spatial MSE: [(0, 9.641084034228697e-06), (1, 3.1807273899175925e-06), (2, 5.734912338084541e-05), (3, 1.1447725228208583e-05), (4, 1.0762858437374234e-05), (5, 6.706311523885233e-06), (6, 1.2328840057307389e-05), (7, 1.449384853913216e-05), (8, 2.1681329599232413e-05), (9, 2.79244650300825e-05), (10, 4.786934005096555e-05), (11, 5.825315383845009e-05), (12, 8.06116295279935e-05), (13, 0.00016612948093097657), (14, 0.00016839522868394852), (15, 0.00019823454204015434), (16, 0.001087187323719263), (17, 0.00020957482047379017), (18, 0.0002775621833279729), (19, 0.0002386443520663306), (20, 0.00030152357066981494), (21, 0.00048396302736364305), (22, 0.0001672100042924285), (23, 0.00037383523886092007), (24, 0.0009452239028178155), (25, 0.0016270781634375453), (26, 0.0012528563383966684), (27, 0.004320116713643074)], Temporal MSE: [(0, 1.720512955216691e-05), (1, 5.948454600002151e-06), (2, 5.344112651073374e-06), (3, 4.7323096623586025e-06), (4, 2.4719765860936604e-06), (5, 4.3813433876493946e-06), (6, 5.29448379893438e-06), (7, 8.114674528769683e-06), (8, 6.778643637517234e-06), (9, 2.6072399123222567e-05), (10, 2.5462992198299617e-05), (11, 5.7164463214576244e-05), (12, 5.834399053128436e-05), (13, 0.00016223143029492348), (14, 0.000155533169163391), (15, 0.0001280812284676358), (16, 0.00022690540936309844), (17, 0.00012000964488834143), (18, 0.0001922359806485474), (19, 0.00010454479343025014), (20, 0.00024918909184634686), (21, 0.00039824130362831056), (22, 0.00011045959399780259), (23, 0.00018460818682797253), (24, 0.00017576558457221836), (25, 0.0007034367881715298), (26, 0.0001728633651509881), (27, 6.745883001713082e-05)]
+
+ 24%|██▍       | 12/50 [01:57<06:04,  9.60s/it]Time step 740, Spatial MSE: [(0, 2.280522858200129e-05), (1, 1.228048131451942e-05), (2, 0.00018002602155320346), (3, 0.00011613816604949534), (4, 0.00045990297803655267), (5, 0.0003574328438844532), (6, 0.0008965000160969794), (7, 0.0015296771889552474), (8, 0.0021526615601032972), (9, 0.003542742459103465), (10, 0.005451900884509087), (11, 0.004533661529421806), (12, 0.0047112638130784035), (13, 0.0059006717056035995), (14, 0.007850159890949726), (15, 0.009793873876333237), (16, 0.014857631176710129), (17, 0.008385736495256424), (18, 0.010804368183016777), (19, 0.010583806782960892), (20, 0.02138475328683853), (21, 0.02805303782224655), (22, 0.006105704233050346), (23, 0.007397081237286329), (24, 0.018686063587665558), (25, 0.11120780557394028), (26, 0.04440383240580559), (27, 0.006749643012881279)], Temporal MSE: [(0, 4.2378262151032686e-05), (1, 7.697925866523292e-06), (2, 5.956448148936033e-05), (3, 6.930661766091362e-05), (4, 0.00011423707474023104), (5, 0.0002225412317784503), (6, 0.00048670146497897804), (7, 0.0005259013269096613), (8, 0.000701001612469554), (9, 0.0014233350520953536), (10, 0.002158950548619032), (11, 0.0023271036334335804), (12, 0.002234028885141015), (13, 0.003166751004755497), (14, 0.004177231807261705), (15, 0.003891461528837681), (16, 0.005092706996947527), (17, 0.0035113017074763775), (18, 0.0036903992295265198), (19, 0.003493740689009428), (20, 0.0068615204654634), (21, 0.006213831249624491), (22, 0.004633523989468813), (23, 0.004489695653319359), (24, 0.011193950660526752), (25, 0.018020348623394966), (26, 0.005282899830490351), (27, 0.000313278054818511)]
+
+ 26%|██▌       | 13/50 [02:09<06:14, 10.12s/it]Time step 720, Spatial MSE: [(0, 8.032806363189593e-06), (1, 4.382621682452736e-06), (2, 9.025642066262662e-05), (3, 1.5058742064866237e-05), (4, 1.2999321370443795e-05), (5, 8.150681424012873e-06), (6, 1.8842760255211033e-05), (7, 2.1206958990660496e-05), (8, 2.0398214473971166e-05), (9, 3.7280406104400754e-05), (10, 4.48629871243611e-05), (11, 7.34452114556916e-05), (12, 9.068005601875484e-05), (13, 0.0001479402999393642), (14, 0.00016645627329126), (15, 0.0002198283764300868), (16, 0.0019294230733066797), (17, 0.0001743612374411896), (18, 0.00039229579851962626), (19, 0.00020167982438579202), (20, 0.00029555370565503836), (21, 0.0005599236465059221), (22, 0.0002581912267487496), (23, 0.00048342818627133965), (24, 0.0010950290597975254), (25, 0.0020174942910671234), (26, 0.001618137233890593), (27, 0.005207069218158722)], Temporal MSE: [(0, 1.3964645404485054e-05), (1, 4.910533334623324e-06), (2, 6.426936579373432e-06), (3, 4.784839802596252e-06), (4, 3.0195960789569654e-06), (5, 5.899667030462297e-06), (6, 9.345502803626005e-06), (7, 1.2559415154100861e-05), (8, 8.60027466842439e-06), (9, 2.647366090968717e-05), (10, 3.2488955184817314e-05), (11, 6.134850991657004e-05), (12, 6.967617082409561e-05), (13, 0.0001323283213423565), (14, 0.00015524115588050336), (15, 0.0001483994274167344), (16, 0.00014849672152195126), (17, 9.600289922673255e-05), (18, 0.00023023871472105384), (19, 0.00011865127453347668), (20, 0.00019137896015308797), (21, 0.00042230120743624866), (22, 0.00011977428948739544), (23, 0.0002764632226899266), (24, 0.000268946256255731), (25, 0.0009085072670131922), (26, 0.0002473957429174334), (27, 7.106627163011581e-05)]
+
+ 28%|██▊       | 14/50 [02:18<05:55,  9.89s/it]Time step 700, Spatial MSE: [(0, 2.0955776562914252e-05), (1, 9.013845556182787e-06), (2, 0.00015075434930622578), (3, 4.50986553914845e-05), (4, 7.919571362435818e-05), (5, 0.00010224033030681312), (6, 0.00037534639704972506), (7, 0.0003939108573831618), (8, 0.0008658711449243128), (9, 0.0008810576400719583), (10, 0.0026340449694544077), (11, 0.00312562077306211), (12, 0.0031061822082847357), (13, 0.00425737863406539), (14, 0.0069409748539328575), (15, 0.004313988611102104), (16, 0.02194252237677574), (17, 0.0039340113289654255), (18, 0.0017769108526408672), (19, 0.0020524875726550817), (20, 0.0020470486488193274), (21, 0.0018703200621530414), (22, 0.0016917575849220157), (23, 0.002976382616907358), (24, 0.007896444760262966), (25, 0.022762373089790344), (26, 0.007252157665789127), (27, 0.009434948675334454)], Temporal MSE: [(0, 1.2825710655306466e-05), (1, 4.816411092178896e-06), (2, 2.893048440455459e-05), (3, 3.2798678148537874e-05), (4, 3.2766416552476585e-05), (5, 0.00011168993660248816), (6, 0.0002596955746412277), (7, 0.0003978134482167661), (8, 0.00024865931482054293), (9, 0.0006449559004977345), (10, 0.0012987952213734388), (11, 0.0012936113635078073), (12, 0.0013552452437579632), (13, 0.0019115702016279101), (14, 0.002564388094469905), (15, 0.0019605776760727167), (16, 0.0023417090997099876), (17, 0.0018540811724960804), (18, 0.0018194769509136677), (19, 0.0009219597559422255), (20, 0.0019593362230807543), (21, 0.0030738881323486567), (22, 0.0010717776603996754), (23, 0.001831602887250483), (24, 0.004448581952601671), (25, 0.008609067648649216), (26, 0.0018004963640123606), (27, 0.00014252927212510258)]
+
+ 30%|███       | 15/50 [02:28<05:44,  9.83s/it]Time step 680, Spatial MSE: [(0, 6.935333658475429e-06), (1, 1.7445185221731663e-05), (2, 0.00015616582822985947), (3, 4.864269430981949e-05), (4, 0.00010934664896922186), (5, 0.00011624914623098448), (6, 0.00045936714741401374), (7, 0.0007241284474730492), (8, 0.00027060339925810695), (9, 0.00046770006883889437), (10, 0.0005414292099885643), (11, 0.0007395928259938955), (12, 0.0008241735049523413), (13, 0.0011793335434049368), (14, 0.0019398927688598633), (15, 0.004500681068748236), (16, 0.017049040645360947), (17, 0.00133981229737401), (18, 0.0029207998886704445), (19, 0.0017931250622496009), (20, 0.004892423748970032), (21, 0.006654342170804739), (22, 0.003178471466526389), (23, 0.0020138216204941273), (24, 0.005034032743424177), (25, 0.02063165046274662), (26, 0.006030057556927204), (27, 0.0063581764698028564)], Temporal MSE: [(0, 1.9874123609042726e-05), (1, 7.209056093415711e-06), (2, 3.301575270597823e-05), (3, 8.151615475071594e-05), (4, 6.389443296939135e-05), (5, 0.00024057961127255112), (6, 0.00028127303812652826), (7, 0.0001799156452761963), (8, 0.00013444443175103515), (9, 0.0003163317160215229), (10, 0.0007646434823982418), (11, 0.0008239687886089087), (12, 0.0009172550635412335), (13, 0.0018879702547565103), (14, 0.0030926852487027645), (15, 0.001870172331109643), (16, 0.002018911996856332), (17, 0.0010644469875842333), (18, 0.000710873631760478), (19, 0.0008037213701754808), (20, 0.0035208314657211304), (21, 0.002736023161560297), (22, 0.0017326927045360208), (23, 0.0012819617986679077), (24, 0.004362013190984726), (25, 0.007418086286634207), (26, 0.0034556901082396507), (27, 0.00010964623652398586)]
+
+ 32%|███▏      | 16/50 [02:40<05:54, 10.42s/it]Time step 660, Spatial MSE: [(0, 1.6590012819506228e-05), (1, 1.094646086130524e-05), (2, 0.00015236390754580498), (3, 7.10491367499344e-05), (4, 0.00015954005357343704), (5, 0.0001959671062650159), (6, 0.0006925397319719195), (7, 0.0006039988365955651), (8, 0.0008902328554540873), (9, 0.0011501400731503963), (10, 0.001664391835220158), (11, 0.0018622324569150805), (12, 0.002200210699811578), (13, 0.0036598825827240944), (14, 0.005151714198291302), (15, 0.004116929601877928), (16, 0.018103187903761864), (17, 0.002482120878994465), (18, 0.0033490797504782677), (19, 0.0026237210258841515), (20, 0.00310714915394783), (21, 0.00522420322522521), (22, 0.004174042027443647), (23, 0.0048798006027936935), (24, 0.020000286400318146), (25, 0.033906128257513046), (26, 0.008722386322915554), (27, 0.006993875838816166)], Temporal MSE: [(0, 9.525982022751123e-06), (1, 3.771278898057062e-06), (2, 2.013404809986241e-05), (3, 4.512061786954291e-05), (4, 8.6859698058106e-05), (5, 0.00013135591871105134), (6, 0.00017222535097971559), (7, 0.00045286331442184746), (8, 0.00035644779563881457), (9, 0.0006179856136441231), (10, 0.0009706583223305643), (11, 0.00114853261038661), (12, 0.0010786432540044188), (13, 0.002306886250153184), (14, 0.0028119164053350687), (15, 0.0015138299204409122), (16, 0.0014152051880955696), (17, 0.0011311223497614264), (18, 0.001498572644777596), (19, 0.0015677809715270996), (20, 0.003277601907029748), (21, 0.00517401285469532), (22, 0.0033743204548954964), (23, 0.0037985246162861586), (24, 0.01368364505469799), (25, 0.04639224708080292), (26, 0.00924462266266346), (27, 0.0002516136737540364)]
+
+ 34%|███▍      | 17/50 [02:50<05:40, 10.31s/it]Time step 640, Spatial MSE: [(0, 7.272123184520751e-06), (1, 9.877726370177697e-06), (2, 0.00010524912795517594), (3, 2.405972009000834e-05), (4, 1.3470051271724515e-05), (5, 9.700917871668935e-06), (6, 3.8956037315074354e-05), (7, 4.0922866901382804e-05), (8, 1.6686592061887495e-05), (9, 3.4415967093082145e-05), (10, 4.410173278301954e-05), (11, 7.375234417850152e-05), (12, 7.855032163206488e-05), (13, 9.131645492743701e-05), (14, 0.000141384283779189), (15, 0.00014917227963451296), (16, 0.00010291133366990834), (17, 0.00013681131531484425), (18, 0.00026958223315887153), (19, 0.00011052016634494066), (20, 0.000182277915882878), (21, 0.0004130617016926408), (22, 0.0001586491271154955), (23, 0.0004492370062507689), (24, 0.0009192789439111948), (25, 0.002063254127278924), (26, 0.0029195856768637896), (27, 0.006498869974166155)], Temporal MSE: [(0, 8.757031537243165e-06), (1, 3.054624357901048e-06), (2, 6.1995979194762185e-06), (3, 4.359857939562062e-06), (4, 3.954237854486564e-06), (5, 7.3347732723050285e-06), (6, 1.9575003534555435e-05), (7, 2.1417485186248086e-05), (8, 9.76959927356802e-06), (9, 2.3324304493144155e-05), (10, 3.84153172490187e-05), (11, 5.6304543250007555e-05), (12, 4.9130187107948586e-05), (13, 8.145537140080705e-05), (14, 8.901929686544463e-05), (15, 0.00011661223106784746), (16, 9.12456089281477e-05), (17, 6.706411659251899e-05), (18, 0.00017498066881671548), (19, 8.25708411866799e-05), (20, 8.606424671597779e-05), (21, 0.00027552625397220254), (22, 5.831990347360261e-05), (23, 0.00020477127691265196), (24, 0.0002819735382217914), (25, 0.0007650434854440391), (26, 0.0003526823129504919), (27, 5.4499116231454536e-05)]
+
+ 36%|███▌      | 18/50 [03:02<05:48, 10.89s/it]Time step 620, Spatial MSE: [(0, 1.815926043491345e-05), (1, 2.484946344338823e-05), (2, 0.00020333884458523244), (3, 6.495504203485325e-05), (4, 0.0001666116004344076), (5, 0.0004009771510027349), (6, 0.0015607327222824097), (7, 0.0032661270815879107), (8, 0.004102834966033697), (9, 0.006462352350354195), (10, 0.006907599046826363), (11, 0.008688889443874359), (12, 0.009524352848529816), (13, 0.011598285287618637), (14, 0.01740567944943905), (15, 0.017417103052139282), (16, 0.01549376081675291), (17, 0.016552558168768883), (18, 0.02010352909564972), (19, 0.012627820484340191), (20, 0.01382431574165821), (21, 0.016807569190859795), (22, 0.014216176234185696), (23, 0.018320538103580475), (24, 0.04732324182987213), (25, 0.08081299066543579), (26, 0.03868916630744934), (27, 0.007925152778625488)], Temporal MSE: [(0, 1.855163100117352e-05), (1, 7.565540727227926e-06), (2, 2.570163269410841e-05), (3, 9.382612915942445e-05), (4, 0.00022122908558230847), (5, 0.00047467820695601404), (6, 0.0013645498547703028), (7, 0.002082150662317872), (8, 0.0015813022619113326), (9, 0.004103857558220625), (10, 0.006541325710713863), (11, 0.005765223875641823), (12, 0.005797129590064287), (13, 0.009691505692899227), (14, 0.010368277318775654), (15, 0.008990809321403503), (16, 0.009142235852777958), (17, 0.008382330648601055), (18, 0.009585943073034286), (19, 0.00839824415743351), (20, 0.020081065595149994), (21, 0.015136104077100754), (22, 0.009287855587899685), (23, 0.01236761175096035), (24, 0.033440664410591125), (25, 0.07670481503009796), (26, 0.017259757965803146), (27, 0.0007570449961349368)]
+
+ 38%|███▊      | 19/50 [03:14<05:49, 11.28s/it]Time step 600, Spatial MSE: [(0, 7.1020122049958445e-06), (1, 1.3056694115221035e-05), (2, 0.00010747402848210186), (3, 3.0160528694977984e-05), (4, 1.2781512850779109e-05), (5, 1.0834563909156714e-05), (6, 3.573373396648094e-05), (7, 4.8988222260959446e-05), (8, 1.6446210793219507e-05), (9, 3.396086685825139e-05), (10, 4.497385452850722e-05), (11, 7.541311788372695e-05), (12, 8.267511293524876e-05), (13, 8.64013418322429e-05), (14, 0.00012535024143289775), (15, 0.00014529493637382984), (16, 9.030967339640483e-05), (17, 0.00012366975715849549), (18, 0.00023138300457503647), (19, 0.0001025229794322513), (20, 0.0001577544171595946), (21, 0.0003313563938718289), (22, 0.00016142259119078517), (23, 0.0004353026452008635), (24, 0.0008776106406003237), (25, 0.002023011213168502), (26, 0.0034408913925290108), (27, 0.007112021092325449)], Temporal MSE: [(0, 7.374119832093129e-06), (1, 2.700104005270987e-06), (2, 6.219761871761875e-06), (3, 4.398651071824133e-06), (4, 4.670873295253841e-06), (5, 8.187864295905456e-06), (6, 2.2271207853918895e-05), (7, 2.0644793039537035e-05), (8, 1.0360244232288096e-05), (9, 2.2784095563110895e-05), (10, 3.797344834310934e-05), (11, 5.438689549919218e-05), (12, 4.999291923013516e-05), (13, 7.869440014474094e-05), (14, 8.322371286340058e-05), (15, 0.00010599641245789826), (16, 8.101372077362612e-05), (17, 6.699408550048247e-05), (18, 0.00012808319297619164), (19, 6.897118146298453e-05), (20, 6.935557030374184e-05), (21, 0.000198305060621351), (22, 5.837048593093641e-05), (23, 0.00017308752285316586), (24, 0.0002718657488003373), (25, 0.0007312259403988719), (26, 0.00046127967652864754), (27, 5.500691258930601e-05)]
+
+ 40%|████      | 20/50 [03:23<05:20, 10.67s/it]Time step 580, Spatial MSE: [(0, 1.718951534712687e-05), (1, 1.816645817598328e-05), (2, 0.00018212992290500551), (3, 3.7928046367596835e-05), (4, 6.719890370732173e-05), (5, 5.967594188405201e-05), (6, 0.0005494925426319242), (7, 0.00021857691172044724), (8, 0.0006087321089580655), (9, 0.0007850422989577055), (10, 0.0014168923953548074), (11, 0.002332929754629731), (12, 0.0026044659316539764), (13, 0.0029997541569173336), (14, 0.004790658131241798), (15, 0.002450964879244566), (16, 0.0013677839888259768), (17, 0.0013764481991529465), (18, 0.0009879641002044082), (19, 0.0007812949479557574), (20, 0.0009962720796465874), (21, 0.0013539446517825127), (22, 0.0012283417163416743), (23, 0.002067845081910491), (24, 0.005952473729848862), (25, 0.011831204406917095), (26, 0.0077916658483445644), (27, 0.010028337128460407)], Temporal MSE: [(0, 7.136448402889073e-06), (1, 3.3842125048977323e-06), (2, 2.1147430743440054e-05), (3, 2.4974104235298e-05), (4, 2.555589162511751e-05), (5, 8.38681953609921e-05), (6, 9.83867357717827e-05), (7, 0.00014792109141126275), (8, 0.00013655511429533362), (9, 0.00022847135551273823), (10, 0.00040906097274273634), (11, 0.0006285067647695541), (12, 0.0007664235308766365), (13, 0.0010259656701236963), (14, 0.0012443456798791885), (15, 0.0007492604199796915), (16, 0.0006920381565578282), (17, 0.0005408124998211861), (18, 0.0005338414339348674), (19, 0.0003833265509456396), (20, 0.0006795786321163177), (21, 0.001223225612193346), (22, 0.0007458606269210577), (23, 0.0013091543223708868), (24, 0.004137905780225992), (25, 0.010698674246668816), (26, 0.002398265525698662), (27, 0.00017388726701028645)]
+
+ 42%|████▏     | 21/50 [03:34<05:05, 10.53s/it]Time step 560, Spatial MSE: [(0, 6.477393526438391e-06), (1, 2.377511191298254e-05), (2, 0.00010383822518633679), (3, 4.659167097997852e-05), (4, 5.342452277545817e-05), (5, 0.00010045733506558463), (6, 0.00017734456923790276), (7, 0.0004992263857275248), (8, 0.000262518908130005), (9, 0.0003118723107036203), (10, 0.000416145077906549), (11, 0.00056044029770419), (12, 0.0007052724831737578), (13, 0.0012214253656566143), (14, 0.003094553481787443), (15, 0.0034288978204131126), (16, 0.10385891795158386), (17, 0.0009078209404833615), (18, 0.001957106636837125), (19, 0.0027237827889621258), (20, 0.0027474751695990562), (21, 0.003740607062354684), (22, 0.0016682767309248447), (23, 0.0015277923084795475), (24, 0.004038115963339806), (25, 0.011112815700471401), (26, 0.004244053736329079), (27, 0.009441779926419258)], Temporal MSE: [(0, 8.266495569841936e-06), (1, 4.661696948460303e-06), (2, 2.7234484150540084e-05), (3, 5.7882578403223306e-05), (4, 7.128273136913776e-05), (5, 0.0001019086193991825), (6, 0.00014783034566789865), (7, 0.00014972538338042796), (8, 0.00012846682511735708), (9, 0.0003701558453030884), (10, 0.0007687856559641659), (11, 0.0009773105848580599), (12, 0.0011670129606500268), (13, 0.0026472662575542927), (14, 0.0036574089899659157), (15, 0.0010499487398192286), (16, 0.0011649272637441754), (17, 0.0007293369853869081), (18, 0.0009851075010374188), (19, 0.0008348824921995401), (20, 0.002260783454403281), (21, 0.0029639038257300854), (22, 0.002031331416219473), (23, 0.001797510078176856), (24, 0.004214724525809288), (25, 0.009116044268012047), (26, 0.003935770131647587), (27, 0.00027672582655213773)]
+
+ 44%|████▍     | 22/50 [03:43<04:44, 10.15s/it]Time step 540, Spatial MSE: [(0, 1.5292134776245803e-05), (1, 1.982292451430112e-05), (2, 0.00018512058886699378), (3, 6.332668999675661e-05), (4, 0.00010322419984731823), (5, 0.00013322968152351677), (6, 0.0006422105361707509), (7, 0.0005115751409903169), (8, 0.0005490542389452457), (9, 0.0007746442570351064), (10, 0.001064554089680314), (11, 0.0015304696280509233), (12, 0.001981208799406886), (13, 0.0033726701512932777), (14, 0.006142498925328255), (15, 0.004770275671035051), (16, 0.09938445687294006), (17, 0.002719299402087927), (18, 0.0037255340721458197), (19, 0.0036409790627658367), (20, 0.003051162464544177), (21, 0.0041092270985245705), (22, 0.0032964246347546577), (23, 0.004432269372045994), (24, 0.013059459626674652), (25, 0.028826763853430748), (26, 0.014925145544111729), (27, 0.009811099618673325)], Temporal MSE: [(0, 6.151075467641931e-06), (1, 3.1588965612172615e-06), (2, 1.402200541633647e-05), (3, 3.7024969060439616e-05), (4, 8.603306923760101e-05), (5, 0.0001365737698506564), (6, 0.00019403039186727256), (7, 0.00021929557260591537), (8, 0.00023337420134339482), (9, 0.000404109014198184), (10, 0.0005815213080495596), (11, 0.0009636218892410398), (12, 0.001135253463871777), (13, 0.0025824157055467367), (14, 0.004043417051434517), (15, 0.0016740624560043216), (16, 0.0016770982183516026), (17, 0.0013276903191581368), (18, 0.0018257759511470795), (19, 0.0014243561308830976), (20, 0.003365411190316081), (21, 0.003058536909520626), (22, 0.0028425974305719137), (23, 0.004164928570389748), (24, 0.008615934289991856), (25, 0.031117191538214684), (26, 0.01166341919451952), (27, 0.0005051984917372465)]
+
+ 46%|████▌     | 23/50 [03:53<04:31, 10.06s/it]Time step 520, Spatial MSE: [(0, 6.276243311731378e-06), (1, 1.613980930414982e-05), (2, 8.229319064412266e-05), (3, 2.9402011932688765e-05), (4, 1.1732256098184735e-05), (5, 1.060964132193476e-05), (6, 2.345978282392025e-05), (7, 4.904040542896837e-05), (8, 1.6446965673821978e-05), (9, 2.699122160265688e-05), (10, 3.5045268305111676e-05), (11, 6.416070391424e-05), (12, 6.070976087357849e-05), (13, 8.490776235703379e-05), (14, 8.523956057615578e-05), (15, 8.675051503814757e-05), (16, 9.96841408777982e-05), (17, 7.79963520471938e-05), (18, 0.0001414156286045909), (19, 7.91805432527326e-05), (20, 0.00011229274241486564), (21, 0.0001570811728015542), (22, 0.00012379762483760715), (23, 0.0002945087617263198), (24, 0.0005542169092223048), (25, 0.0014866746496409178), (26, 0.0033596146386116743), (27, 0.0072136870585381985)], Temporal MSE: [(0, 5.4655633903166745e-06), (1, 2.1906998881604522e-06), (2, 4.652241386793321e-06), (3, 4.455730049812701e-06), (4, 4.68007056042552e-06), (5, 7.341580385400448e-06), (6, 1.9385690393391997e-05), (7, 1.745834742905572e-05), (8, 9.474921171204187e-06), (9, 1.8591908883536234e-05), (10, 3.353691135998815e-05), (11, 4.41815827798564e-05), (12, 3.946820288547315e-05), (13, 7.329467189265415e-05), (14, 5.3857769671594724e-05), (15, 5.705949661205523e-05), (16, 5.363275340641849e-05), (17, 4.075832839589566e-05), (18, 5.048888851888478e-05), (19, 4.37297931057401e-05), (20, 4.3617506889859214e-05), (21, 9.040764416567981e-05), (22, 5.003272963222116e-05), (23, 0.00011120775161543861), (24, 0.00021225740783847868), (25, 0.0004916799953207374), (26, 0.0004647783935070038), (27, 5.278600656311028e-05)]
+
+ 48%|████▊     | 24/50 [04:03<04:25, 10.20s/it]Time step 500, Spatial MSE: [(0, 1.4335665582621004e-05), (1, 2.8457181542762555e-05), (2, 0.00015736528439447284), (3, 5.6186981964856386e-05), (4, 0.00021141566685400903), (5, 0.0004376681172288954), (6, 0.0015210863202810287), (7, 0.0029612050857394934), (8, 0.003939289133995771), (9, 0.005595264956355095), (10, 0.009293894283473492), (11, 0.010177984833717346), (12, 0.012053255923092365), (13, 0.014150787144899368), (14, 0.025321096181869507), (15, 0.02878998965024948), (16, 0.0615207701921463), (17, 0.020554520189762115), (18, 0.02545139193534851), (19, 0.018541675060987473), (20, 0.0228596031665802), (21, 0.03035784512758255), (22, 0.020291725173592567), (23, 0.027395591139793396), (24, 0.05792511999607086), (25, 0.08921875059604645), (26, 0.06821078807115555), (27, 0.008649762719869614)], Temporal MSE: [(0, 6.775399015168659e-06), (1, 4.676632215705467e-06), (2, 2.2424192138714716e-05), (3, 0.0001505510153947398), (4, 0.00031036988366395235), (5, 0.00049446034245193), (6, 0.0012046205811202526), (7, 0.0014650600496679544), (8, 0.0016618778463453054), (9, 0.00357322976924479), (10, 0.005893101450055838), (11, 0.007348239421844482), (12, 0.007639972493052483), (13, 0.011568751186132431), (14, 0.013162363320589066), (15, 0.014599267393350601), (16, 0.014801900833845139), (17, 0.012297559529542923), (18, 0.01126022543758154), (19, 0.01146619115024805), (20, 0.02303893119096756), (21, 0.017641399055719376), (22, 0.011999977752566338), (23, 0.016545981168746948), (24, 0.04328133165836334), (25, 0.08385241031646729), (26, 0.024352462962269783), (27, 0.001011638785712421)]
+
+ 50%|█████     | 25/50 [04:14<04:20, 10.41s/it]Time step 480, Spatial MSE: [(0, 6.09002381679602e-06), (1, 1.7551035853102803e-05), (2, 7.043106597848237e-05), (3, 3.217950870748609e-05), (4, 1.1682173862936907e-05), (5, 1.1843997526739258e-05), (6, 1.893957596621476e-05), (7, 4.912752046948299e-05), (8, 1.6308313206536695e-05), (9, 2.418673102511093e-05), (10, 3.31746632582508e-05), (11, 5.637570939143188e-05), (12, 5.2492323447950184e-05), (13, 6.004620809108019e-05), (14, 7.57887028157711e-05), (15, 7.55040964577347e-05), (16, 6.093152114772238e-05), (17, 5.137051266501658e-05), (18, 0.00015936825366225094), (19, 7.559773803222924e-05), (20, 9.096917347051203e-05), (21, 0.00017747403762768954), (22, 0.00012996734585613012), (23, 0.0002435691567370668), (24, 0.0005503028514795005), (25, 0.001216889126226306), (26, 0.003429003991186619), (27, 0.007722950540482998)], Temporal MSE: [(0, 4.732172783405986e-06), (1, 2.0963836959708715e-06), (2, 4.4079547478759196e-06), (3, 5.213179520069389e-06), (4, 4.838290806219447e-06), (5, 7.853457645978779e-06), (6, 1.935849286383018e-05), (7, 1.654544030316174e-05), (8, 8.730260560696479e-06), (9, 1.8564131096354686e-05), (10, 3.005622056662105e-05), (11, 3.7734858779003844e-05), (12, 3.249184010201134e-05), (13, 5.42191082786303e-05), (14, 4.474259549169801e-05), (15, 5.0765567721100524e-05), (16, 4.119596997043118e-05), (17, 2.734799636527896e-05), (18, 4.1872961446642876e-05), (19, 3.398339686100371e-05), (20, 4.2579860746627674e-05), (21, 9.287585999118164e-05), (22, 5.804635293316096e-05), (23, 9.936402057064697e-05), (24, 0.00022291200002655387), (25, 0.00045380182564258575), (26, 0.0005474580102600157), (27, 7.242985884658992e-05)]
+
+ 52%|█████▏    | 26/50 [04:24<04:08, 10.35s/it]Time step 460, Spatial MSE: [(0, 1.523615082987817e-05), (1, 2.1109617591719143e-05), (2, 0.00011665817874018103), (3, 3.542288322933018e-05), (4, 6.308816955424845e-05), (5, 6.311942706815898e-05), (6, 0.00040511685074307024), (7, 0.00015636997704859823), (8, 0.00043712256592698395), (9, 0.0005708280368708074), (10, 0.0009314659982919693), (11, 0.0015085074119269848), (12, 0.0018407396273687482), (13, 0.0019966731779277325), (14, 0.003506562439724803), (15, 0.0019674370996654034), (16, 0.0010106118861585855), (17, 0.0008572996011935174), (18, 0.0008660363964736462), (19, 0.000614538905210793), (20, 0.000920531339943409), (21, 0.00180121383164078), (22, 0.0011145486496388912), (23, 0.0017220074078068137), (24, 0.004238440655171871), (25, 0.008576812222599983), (26, 0.013619598001241684), (27, 0.007732405327260494)], Temporal MSE: [(0, 4.851702669839142e-06), (1, 2.680700163182337e-06), (2, 2.1268082491587847e-05), (3, 4.0514285501558334e-05), (4, 3.695955092553049e-05), (5, 6.55959956930019e-05), (6, 6.109619425842538e-05), (7, 9.446017065783963e-05), (8, 0.00010243311407975852), (9, 0.00017048604786396027), (10, 0.0002979356504511088), (11, 0.00047328841174021363), (12, 0.0005714370054192841), (13, 0.0009333157213404775), (14, 0.0010964515386149287), (15, 0.0006595247541554272), (16, 0.00046714069321751595), (17, 0.0004022884531877935), (18, 0.00038490226143039763), (19, 0.0003282979305367917), (20, 0.0006244358373805881), (21, 0.001014690031297505), (22, 0.0005941580748185515), (23, 0.0009565090294927359), (24, 0.003275363938882947), (25, 0.008485131897032261), (26, 0.0026167321484535933), (27, 0.0002600431616883725)]
+
+ 54%|█████▍    | 27/50 [04:33<03:46,  9.86s/it]Time step 440, Spatial MSE: [(0, 5.434469130705111e-06), (1, 2.327426955162082e-05), (2, 6.078166188672185e-05), (3, 3.760879917535931e-05), (4, 7.057143375277519e-05), (5, 0.0001274559326702729), (6, 0.00011486937728477642), (7, 0.0004048057016916573), (8, 0.00027152206166647375), (9, 0.0003077939327340573), (10, 0.00037852145032957196), (11, 0.0004786847566720098), (12, 0.0006002418813295662), (13, 0.001293738023377955), (14, 0.0037815661635249853), (15, 0.0030754217877984047), (16, 0.13429337739944458), (17, 0.0008229187224060297), (18, 0.0014093448407948017), (19, 0.003234976204112172), (20, 0.0025605331175029278), (21, 0.0032957359217107296), (22, 0.0014652053359895945), (23, 0.0012308618752285838), (24, 0.0031031540129333735), (25, 0.006820396054536104), (26, 0.0035803483333438635), (27, 0.012975623831152916)], Temporal MSE: [(0, 4.656167675420875e-06), (1, 3.3031451494025532e-06), (2, 2.6132100174436346e-05), (3, 0.00012167526438133791), (4, 0.0001305787154706195), (5, 0.00010148667206522077), (6, 0.0001587670558365062), (7, 0.00015831715427339077), (8, 0.00013314193347468972), (9, 0.0003740499378181994), (10, 0.000597609905526042), (11, 0.0008803019300103188), (12, 0.0011126790195703506), (13, 0.002938272897154093), (14, 0.00430280901491642), (15, 0.0008991460199467838), (16, 0.001313240616582334), (17, 0.0006744152633473277), (18, 0.0009811444906517863), (19, 0.0008478006930090487), (20, 0.0020645265467464924), (21, 0.002742673270404339), (22, 0.0019196460489183664), (23, 0.0018990091048181057), (24, 0.0036736023612320423), (25, 0.006340600084513426), (26, 0.003938642796128988), (27, 0.0005605968763120472)]
+
+ 56%|█████▌    | 28/50 [04:44<03:47, 10.34s/it]Time step 420, Spatial MSE: [(0, 1.3782283531327266e-05), (1, 2.245702125946991e-05), (2, 0.000104773513157852), (3, 5.494936340255663e-05), (4, 9.644647070672363e-05), (5, 0.0001648535835556686), (6, 0.0003869382489938289), (7, 0.0005073443171568215), (8, 0.000572784396354109), (9, 0.000743074924685061), (10, 0.0009909538784995675), (11, 0.001469155540689826), (12, 0.0019099062774330378), (13, 0.003332392079755664), (14, 0.007760124281048775), (15, 0.00623591523617506), (16, 0.12587149441242218), (17, 0.0022391844540834427), (18, 0.002875751815736294), (19, 0.003979111090302467), (20, 0.0034700653050094843), (21, 0.003943619783967733), (22, 0.0029845540411770344), (23, 0.003776293247938156), (24, 0.008635031059384346), (25, 0.026507757604122162), (26, 0.028661217540502548), (27, 0.012756714597344398)], Temporal MSE: [(0, 4.253510724083753e-06), (1, 2.5770009415282402e-06), (2, 1.5955893104546703e-05), (3, 3.628529157140292e-05), (4, 0.00011929296306334436), (5, 0.00019378091383259743), (6, 0.00022031938715372235), (7, 0.0002645108615979552), (8, 0.0002664334315340966), (9, 0.0004352661198936403), (10, 0.0006222308147698641), (11, 0.0010656480444595218), (12, 0.0012085526250302792), (13, 0.0031424686312675476), (14, 0.005034790374338627), (15, 0.0021302071399986744), (16, 0.0022562164813280106), (17, 0.0013384915655478835), (18, 0.0016654367791488767), (19, 0.001480733510106802), (20, 0.004245614167302847), (21, 0.00342462002299726), (22, 0.0026435614563524723), (23, 0.003901821793988347), (24, 0.006968623958528042), (25, 0.026297109201550484), (26, 0.009341956116259098), (27, 0.0007207614253275096)]
+
+ 58%|█████▊    | 29/50 [04:57<03:48, 10.88s/it]Time step 400, Spatial MSE: [(0, 5.746761416958179e-06), (1, 1.863495708676055e-05), (2, 4.3120009650010616e-05), (3, 3.047852987947408e-05), (4, 1.0878036846406758e-05), (5, 1.222543232870521e-05), (6, 1.4622773051087279e-05), (7, 4.597442602971569e-05), (8, 1.864071236923337e-05), (9, 2.132328700099606e-05), (10, 3.2064672268461436e-05), (11, 4.998662552679889e-05), (12, 5.056414374848828e-05), (13, 5.764343586633913e-05), (14, 7.465195085387677e-05), (15, 6.11783834756352e-05), (16, 6.124412175267935e-05), (17, 5.170750227989629e-05), (18, 9.258852514903992e-05), (19, 7.314268441405147e-05), (20, 6.803602445870638e-05), (21, 0.00011786788672907278), (22, 0.00011271966650383547), (23, 0.00022474316938314587), (24, 0.0004248961922712624), (25, 0.0010740099241957068), (26, 0.003195066936314106), (27, 0.008681987412273884)], Temporal MSE: [(0, 4.025625912618125e-06), (1, 2.0201678125886247e-06), (2, 3.977727374149254e-06), (3, 5.427226369647542e-06), (4, 4.713684575108346e-06), (5, 8.185706064978149e-06), (6, 1.707743649603799e-05), (7, 1.5071146663103718e-05), (8, 9.184878763335291e-06), (9, 1.931517181219533e-05), (10, 3.049017504963558e-05), (11, 3.508777081151493e-05), (12, 3.4117514587705955e-05), (13, 5.7940771512221545e-05), (14, 4.54860673926305e-05), (15, 4.262344737071544e-05), (16, 3.980789188062772e-05), (17, 2.5884519345709123e-05), (18, 2.4198214305215515e-05), (19, 2.9470933441189118e-05), (20, 4.10297216149047e-05), (21, 5.939749826211482e-05), (22, 6.102771294536069e-05), (23, 9.569495887262747e-05), (24, 0.0002024168352363631), (25, 0.0004068845009896904), (26, 0.000559138017706573), (27, 0.00011335213639540598)]
+
+ 60%|██████    | 30/50 [05:05<03:23, 10.18s/it]Time step 380, Spatial MSE: [(0, 1.2515474736574106e-05), (1, 2.5619128791731782e-05), (2, 7.187295705080032e-05), (3, 5.285710722091608e-05), (4, 0.00016500402125529945), (5, 0.0004590083844959736), (6, 0.0011066613951697946), (7, 0.0020434590987861156), (8, 0.0032824857626110315), (9, 0.004364704247564077), (10, 0.005880777724087238), (11, 0.005862034857273102), (12, 0.00709720142185688), (13, 0.008293534629046917), (14, 0.01365961879491806), (15, 0.01578628271818161), (16, 0.05137712135910988), (17, 0.011527879163622856), (18, 0.014038334600627422), (19, 0.011693264357745647), (20, 0.015295419842004776), (21, 0.014748002402484417), (22, 0.012103108689188957), (23, 0.016933567821979523), (24, 0.032491862773895264), (25, 0.05655600503087044), (26, 0.060666996985673904), (27, 0.008076419122517109)], Temporal MSE: [(0, 4.285049726604484e-06), (1, 3.102643404417904e-06), (2, 2.047413363470696e-05), (3, 0.00011550298950169235), (4, 0.0002528514014557004), (5, 0.0004370823735371232), (6, 0.0010639741085469723), (7, 0.0011921740369871259), (8, 0.0013510467251762748), (9, 0.0027349505107849836), (10, 0.00421344256028533), (11, 0.005225698463618755), (12, 0.005134926177561283), (13, 0.008394211530685425), (14, 0.01053399033844471), (15, 0.008111363276839256), (16, 0.007978064939379692), (17, 0.007518642581999302), (18, 0.0067323604598641396), (19, 0.006868571043014526), (20, 0.013269742019474506), (21, 0.011662516742944717), (22, 0.00839900877326727), (23, 0.011395137757062912), (24, 0.02591300755739212), (25, 0.05215948820114136), (26, 0.018380654975771904), (27, 0.0009311728645116091)]
+
+ 62%|██████▏   | 31/50 [05:16<03:19, 10.50s/it]Time step 360, Spatial MSE: [(0, 5.724512448068708e-06), (1, 1.9459917893982492e-05), (2, 3.340543480589986e-05), (3, 3.390281199244782e-05), (4, 1.1153800187457819e-05), (5, 1.4824207028141245e-05), (6, 1.5023387277324218e-05), (7, 4.361947503639385e-05), (8, 1.9558665371732786e-05), (9, 2.1806330551044084e-05), (10, 3.4587290429044515e-05), (11, 5.177528510103002e-05), (12, 5.224744381848723e-05), (13, 5.729976328439079e-05), (14, 7.152034959290177e-05), (15, 5.7022662076633424e-05), (16, 6.249466969165951e-05), (17, 4.7276727855205536e-05), (18, 0.00011820331565104425), (19, 7.991720485733822e-05), (20, 7.340261072386056e-05), (21, 0.00012692000018432736), (22, 0.0001303090393776074), (23, 0.00024019104603212327), (24, 0.00046713443589396775), (25, 0.0010693199001252651), (26, 0.0029445895925164223), (27, 0.009580194018781185)], Temporal MSE: [(0, 3.785263515965198e-06), (1, 2.0759594008268323e-06), (2, 4.5234282879391685e-06), (3, 6.9679063017247245e-06), (4, 5.204665740166092e-06), (5, 9.086597856367007e-06), (6, 1.6899968613870442e-05), (7, 1.4677551916975062e-05), (8, 1.0088128874485847e-05), (9, 2.149756437574979e-05), (10, 3.187334368703887e-05), (11, 3.652308078017086e-05), (12, 3.444136382313445e-05), (13, 5.762631553807296e-05), (14, 4.4427550164982677e-05), (15, 3.96931754949037e-05), (16, 3.8422378565883264e-05), (17, 2.4961898816400208e-05), (18, 2.5601482775527984e-05), (19, 2.8596459742402658e-05), (20, 4.743475074064918e-05), (21, 7.308003114303574e-05), (22, 7.665505836484954e-05), (23, 0.00011473084305180237), (24, 0.0002344590611755848), (25, 0.00045302597573027015), (26, 0.0006427748594433069), (27, 0.00014967744937166572)]
+
+ 64%|██████▍   | 32/50 [05:27<03:10, 10.60s/it]Time step 340, Spatial MSE: [(0, 1.3651082554133609e-05), (1, 2.0960173060302623e-05), (2, 4.0985636587720364e-05), (3, 3.376900713192299e-05), (4, 4.876610910287127e-05), (5, 7.364290650002658e-05), (6, 0.0002456990769132972), (7, 0.00015880675346124917), (8, 0.00034522011992521584), (9, 0.0004870474513154477), (10, 0.0008579426212236285), (11, 0.0012399611296132207), (12, 0.0016102667432278395), (13, 0.0018651570426300168), (14, 0.004130242392420769), (15, 0.0023523971904069185), (16, 0.001137338113039732), (17, 0.0009004446910694242), (18, 0.0009577448945492506), (19, 0.0007285850588232279), (20, 0.0011936016380786896), (21, 0.0021182061173021793), (22, 0.0011196484556421638), (23, 0.0016580554656684399), (24, 0.004610983654856682), (25, 0.010580720379948616), (26, 0.02120421640574932), (27, 0.007183540146797895)], Temporal MSE: [(0, 4.122012342122616e-06), (1, 2.4201026462833397e-06), (2, 2.0217652490828186e-05), (3, 4.976778291165829e-05), (4, 5.081679410068318e-05), (5, 7.12929613655433e-05), (6, 6.951810064492747e-05), (7, 9.335426875622943e-05), (8, 0.00010232422209810466), (9, 0.0001859131152741611), (10, 0.0003013903624378145), (11, 0.00048095654346980155), (12, 0.0005873867776244879), (13, 0.0011510065523907542), (14, 0.0017315674340352416), (15, 0.0008708573877811432), (16, 0.000522602116689086), (17, 0.00046091346303001046), (18, 0.0004813808191101998), (19, 0.00041607741150073707), (20, 0.0008358459454029799), (21, 0.0009377010865136981), (22, 0.0005830114823766053), (23, 0.0009744387934915721), (24, 0.0029383774381130934), (25, 0.007856355048716068), (26, 0.0030600344762206078), (27, 0.00048300609341822565)]
+
+ 66%|██████▌   | 33/50 [05:37<02:56, 10.38s/it]Time step 320, Spatial MSE: [(0, 5.602880264632404e-06), (1, 2.3134056391427293e-05), (2, 2.962175858556293e-05), (3, 3.1851508538238704e-05), (4, 8.505526784574613e-05), (5, 0.00015838687249924988), (6, 0.00014622628805227578), (7, 0.00046441692393273115), (8, 0.00030134955886751413), (9, 0.00032367618405260146), (10, 0.0003937305591534823), (11, 0.0005008286680094898), (12, 0.0006251577869988978), (13, 0.0012957779690623283), (14, 0.004183897282928228), (15, 0.003443609457463026), (16, 0.1707615703344345), (17, 0.0009623910882510245), (18, 0.001604588469490409), (19, 0.003740118583664298), (20, 0.0031395626720041037), (21, 0.0033118417486548424), (22, 0.0013664236757904291), (23, 0.001303221331909299), (24, 0.003163920482620597), (25, 0.006805717013776302), (26, 0.0038712364621460438), (27, 0.016184154897928238)], Temporal MSE: [(0, 4.162201094004558e-06), (1, 2.8357897008390864e-06), (2, 2.4015385861275718e-05), (3, 0.00015332236944232136), (4, 0.00016808371583465487), (5, 0.00014186564658302814), (6, 0.0001768888032529503), (7, 0.00019180646631866693), (8, 0.00017288667731918395), (9, 0.0004041864303871989), (10, 0.000623679778072983), (11, 0.0008401703089475632), (12, 0.0010785850463435054), (13, 0.002895898185670376), (14, 0.004980287980288267), (15, 0.0011459288652986288), (16, 0.0017520913388580084), (17, 0.0008106346940621734), (18, 0.0011623390018939972), (19, 0.0010459092445671558), (20, 0.0027685449458658695), (21, 0.0031395426485687494), (22, 0.0021247602999210358), (23, 0.0022054763976484537), (24, 0.003579291980713606), (25, 0.005404217168688774), (26, 0.0045738425105810165), (27, 0.0009886709740385413)]
+
+ 68%|██████▊   | 34/50 [05:49<02:51, 10.73s/it]Time step 300, Spatial MSE: [(0, 1.2061620509484783e-05), (1, 2.0678544387919828e-05), (2, 3.0012080969754606e-05), (3, 4.529409488895908e-05), (4, 0.00010022091737482697), (5, 0.00019778759451583028), (6, 0.0002841518435161561), (7, 0.0005982702132314444), (8, 0.0006494157714769244), (9, 0.0007686912431381643), (10, 0.0009588145767338574), (11, 0.0013832643162459135), (12, 0.0017934340285137296), (13, 0.003447131486609578), (14, 0.008081554435193539), (15, 0.007163569331169128), (16, 0.15357525646686554), (17, 0.0025715557858347893), (18, 0.0035085314884781837), (19, 0.004819510504603386), (20, 0.0046875616535544395), (21, 0.004598407540470362), (22, 0.003309507854282856), (23, 0.004385160282254219), (24, 0.011169458739459515), (25, 0.03524710237979889), (26, 0.04025785252451897), (27, 0.015950269997119904)], Temporal MSE: [(0, 4.463166078494396e-06), (1, 2.4667569960001856e-06), (2, 1.4094854122959077e-05), (3, 3.39742582582403e-05), (4, 0.0001381940091960132), (5, 0.000253150676144287), (6, 0.0002544538292568177), (7, 0.0003355127410031855), (8, 0.0003429493517614901), (9, 0.000508204335346818), (10, 0.0007743840105831623), (11, 0.0013981831725686789), (12, 0.001465547364205122), (13, 0.003780948230996728), (14, 0.006041062995791435), (15, 0.0026627141050994396), (16, 0.0027496989350765944), (17, 0.0014931012410670519), (18, 0.002152944914996624), (19, 0.001882284297607839), (20, 0.005819567479193211), (21, 0.0038505191914737225), (22, 0.0031452528201043606), (23, 0.0050729019567370415), (24, 0.008430649526417255), (25, 0.022919295355677605), (26, 0.007947983220219612), (27, 0.0008915879298001528)]
+
+ 70%|███████   | 35/50 [05:59<02:40, 10.67s/it]Time step 280, Spatial MSE: [(0, 6.0889669839525595e-06), (1, 2.0186766050755978e-05), (2, 2.7798228984465823e-05), (3, 3.3208962122444063e-05), (4, 1.2030381185468286e-05), (5, 1.96080727619119e-05), (6, 1.6442738342448138e-05), (7, 4.1057024645851925e-05), (8, 2.381767990300432e-05), (9, 2.4089593352982774e-05), (10, 3.688998913276009e-05), (11, 5.337134643923491e-05), (12, 5.468268136610277e-05), (13, 8.842859097057953e-05), (14, 8.526589954271913e-05), (15, 7.368471415247768e-05), (16, 0.00010790669330162928), (17, 5.015039641875774e-05), (18, 0.00013780564768239856), (19, 9.980861796066165e-05), (20, 8.153813541866839e-05), (21, 0.0001764716289471835), (22, 0.00016291213978547603), (23, 0.0002920354891102761), (24, 0.0005626659258268774), (25, 0.0011163961607962847), (26, 0.0026374226436018944), (27, 0.011230435222387314)], Temporal MSE: [(0, 4.285387149138842e-06), (1, 2.4769951778580435e-06), (2, 5.290205081109889e-06), (3, 6.9729567258036695e-06), (4, 5.724962647946086e-06), (5, 1.123060610552784e-05), (6, 1.7914355339598842e-05), (7, 1.5181164599198382e-05), (8, 1.3592147297458723e-05), (9, 2.4683580704731867e-05), (10, 3.613391163526103e-05), (11, 4.04501952289138e-05), (12, 4.081896258867346e-05), (13, 8.556248940294608e-05), (14, 5.6078391935443506e-05), (15, 5.649744707625359e-05), (16, 3.531579568516463e-05), (17, 2.727479295572266e-05), (18, 3.034641849808395e-05), (19, 3.5050885344389826e-05), (20, 5.8496410929365084e-05), (21, 8.78510472830385e-05), (22, 9.722769755171612e-05), (23, 0.00015486986376345158), (24, 0.0002616576384752989), (25, 0.0005044369027018547), (26, 0.0006705249543301761), (27, 0.0002735219313763082)]
+
+ 72%|███████▏  | 36/50 [06:10<02:29, 10.67s/it]Time step 260, Spatial MSE: [(0, 1.08921658465988e-05), (1, 2.437929651932791e-05), (2, 4.3512744014151394e-05), (3, 5.4037009249441326e-05), (4, 0.00015531227109022439), (5, 0.0003861508157569915), (6, 0.0007371322717517614), (7, 0.0016708162147551775), (8, 0.002629500813782215), (9, 0.00373389245942235), (10, 0.003588692285120487), (11, 0.004080485552549362), (12, 0.005060570314526558), (13, 0.005939458962529898), (14, 0.009276248514652252), (15, 0.013984541408717632), (16, 0.011253638193011284), (17, 0.009365768171846867), (18, 0.010958199389278889), (19, 0.008601645939052105), (20, 0.010277044028043747), (21, 0.010730660520493984), (22, 0.01075688749551773), (23, 0.014748526737093925), (24, 0.02886362187564373), (25, 0.049290962517261505), (26, 0.05515918508172035), (27, 0.008173673413693905)], Temporal MSE: [(0, 5.632872216665419e-06), (1, 3.1757483611727366e-06), (2, 2.2599861040362157e-05), (3, 0.00012462092854548246), (4, 0.00022258645913098007), (5, 0.00037051396793685853), (6, 0.00080271897604689), (7, 0.0010247969767078757), (8, 0.0011942869750782847), (9, 0.0024146426003426313), (10, 0.0033926917240023613), (11, 0.00431458093225956), (12, 0.004364750348031521), (13, 0.0076272436417639256), (14, 0.0091100400313735), (15, 0.0077489223331213), (16, 0.00657955277711153), (17, 0.006044358015060425), (18, 0.0054113492369651794), (19, 0.006102160084992647), (20, 0.011719737201929092), (21, 0.010784674435853958), (22, 0.007175387814640999), (23, 0.010013594292104244), (24, 0.023484893143177032), (25, 0.045580796897411346), (26, 0.0199308879673481), (27, 0.001402357709594071)]
+
+ 74%|███████▍  | 37/50 [06:20<02:15, 10.41s/it]Time step 240, Spatial MSE: [(0, 6.798631147830747e-06), (1, 2.2651611288893037e-05), (2, 3.7473168049473315e-05), (3, 3.738301165867597e-05), (4, 1.4048006960365456e-05), (5, 2.397125535935629e-05), (6, 1.9172814063495025e-05), (7, 3.6036282835993916e-05), (8, 2.6547411835053936e-05), (9, 2.793361636577174e-05), (10, 4.2070143535966054e-05), (11, 5.658566442434676e-05), (12, 6.074593693483621e-05), (13, 6.716648931615055e-05), (14, 9.303080150857568e-05), (15, 7.670045306440443e-05), (16, 7.064014789648354e-05), (17, 6.40872967778705e-05), (18, 0.00011756383901229128), (19, 0.00011705132783390582), (20, 0.00010085190297104418), (21, 0.00018802002887241542), (22, 0.0001910907740239054), (23, 0.00035698863212019205), (24, 0.0006562306080013514), (25, 0.0012540839379653335), (26, 0.0026356561575084925), (27, 0.012365725822746754)], Temporal MSE: [(0, 5.136627805768512e-06), (1, 3.0077292194619076e-06), (2, 6.507840225822292e-06), (3, 9.352838787890505e-06), (4, 7.386240213236306e-06), (5, 1.3954903806734364e-05), (6, 1.9487006284180097e-05), (7, 1.6718120605219156e-05), (8, 1.7272324839723296e-05), (9, 2.8860951715614647e-05), (10, 3.7486959627131e-05), (11, 4.5044795115245506e-05), (12, 4.644507134798914e-05), (13, 8.081860869424418e-05), (14, 6.0917038354091346e-05), (15, 6.388092879205942e-05), (16, 4.5317505282582715e-05), (17, 3.58187171514146e-05), (18, 3.621937503339723e-05), (19, 4.383729174151085e-05), (20, 7.817251025699079e-05), (21, 0.00010080173524329439), (22, 0.00010761104931589216), (23, 0.0001804569037631154), (24, 0.00031902542104944587), (25, 0.0006130095571279526), (26, 0.0008186555933207273), (27, 0.0003569318796508014)]
+
+ 76%|███████▌  | 38/50 [06:29<02:01, 10.08s/it]Time step 220, Spatial MSE: [(0, 1.131135013565654e-05), (1, 2.2674550564261153e-05), (2, 7.300012657651678e-05), (3, 4.123684993828647e-05), (4, 4.709222412202507e-05), (5, 0.00011562116560526192), (6, 0.00023528897145297378), (7, 0.00021339766681194305), (8, 0.00036050734343007207), (9, 0.0005586330662481487), (10, 0.0009390978957526386), (11, 0.0013323058374226093), (12, 0.0017736392328515649), (13, 0.002119749551638961), (14, 0.004191621206700802), (15, 0.002820660825818777), (16, 0.001352144405245781), (17, 0.0010626479052007198), (18, 0.001487682107836008), (19, 0.0011542555876076221), (20, 0.001804401515983045), (21, 0.0021160789765417576), (22, 0.001252062269486487), (23, 0.0023064499255269766), (24, 0.007449689786881208), (25, 0.019542597234249115), (26, 0.030673256143927574), (27, 0.008014419116079807)], Temporal MSE: [(0, 6.386856966855703e-06), (1, 3.2286407076753676e-06), (2, 1.793166848074179e-05), (3, 5.467348455567844e-05), (4, 6.422409933293238e-05), (5, 0.0001003003926598467), (6, 0.00012726482236757874), (7, 0.00012442034494597465), (8, 0.00014726458175573498), (9, 0.00024065302568487823), (10, 0.00037321788840927184), (11, 0.0006253960309550166), (12, 0.0007653300417587161), (13, 0.0015637739561498165), (14, 0.0022997725754976273), (15, 0.00121408817358315), (16, 0.0006870093056932092), (17, 0.0005695299478247762), (18, 0.0008162104641087353), (19, 0.0006472631357610226), (20, 0.0012861838331446052), (21, 0.0010228422470390797), (22, 0.000733102613594383), (23, 0.0013308337656781077), (24, 0.0036115923430770636), (25, 0.00995869841426611), (26, 0.00440615089610219), (27, 0.0009722309187054634)]
+
+ 78%|███████▊  | 39/50 [06:39<01:49,  9.99s/it]Time step 200, Spatial MSE: [(0, 7.373775588348508e-06), (1, 2.722610588534735e-05), (2, 7.564145926153287e-05), (3, 3.322513657622039e-05), (4, 7.818610174581409e-05), (5, 0.00016150008013937622), (6, 0.00018487901252228767), (7, 0.0005618444993160665), (8, 0.0003497103461995721), (9, 0.0003728628798853606), (10, 0.00047275557881221175), (11, 0.0005851704627275467), (12, 0.000753937172703445), (13, 0.0012486096238717437), (14, 0.0028358648996800184), (15, 0.0035442663356661797), (16, 0.06668669730424881), (17, 0.0012647011317312717), (18, 0.0017217559507116675), (19, 0.00193664594553411), (20, 0.0024600571487098932), (21, 0.003144661197438836), (22, 0.001506245811469853), (23, 0.0015546954236924648), (24, 0.0036082651931792498), (25, 0.0067557720467448235), (26, 0.004425799008458853), (27, 0.016429081559181213)], Temporal MSE: [(0, 7.095834007486701e-06), (1, 3.817307060671737e-06), (2, 2.2031994376447983e-05), (3, 0.00013499852502718568), (4, 0.0001635601365705952), (5, 0.0001536169002065435), (6, 0.0002536505344323814), (7, 0.0002607476490084082), (8, 0.0002377296332269907), (9, 0.0004744284087792039), (10, 0.0006896331906318665), (11, 0.0009472076199017465), (12, 0.0012615405721589923), (13, 0.002913611475378275), (14, 0.003536188742145896), (15, 0.0016065018717199564), (16, 0.0015264515532180667), (17, 0.0010009193792939186), (18, 0.0008857318316586316), (19, 0.0010523126693442464), (20, 0.0020711086690425873), (21, 0.0020093561615794897), (22, 0.0013955985195934772), (23, 0.0018307925201952457), (24, 0.0037448634393513203), (25, 0.00548098050057888), (26, 0.005581976845860481), (27, 0.0016782842576503754)]
+
+ 80%|████████  | 40/50 [06:49<01:40, 10.09s/it]Time step 180, Spatial MSE: [(0, 1.1037363947252743e-05), (1, 2.7169120585313067e-05), (2, 0.00015957720461301506), (3, 5.1658498705364764e-05), (4, 0.00010249813203699887), (5, 0.00022447148512583226), (6, 0.00035790487891063094), (7, 0.0007028828840702772), (8, 0.0007990371668711305), (9, 0.0008764690137468278), (10, 0.001035855384543538), (11, 0.0014374115271493793), (12, 0.0018841394921764731), (13, 0.00319307716563344), (14, 0.006937734317034483), (15, 0.007594684604555368), (16, 0.057576995342969894), (17, 0.002996186027303338), (18, 0.0038836433086544275), (19, 0.004330698866397142), (20, 0.005634892266243696), (21, 0.00534317409619689), (22, 0.004759017378091812), (23, 0.006799411028623581), (24, 0.017779594287276268), (25, 0.05802522599697113), (26, 0.06035633385181427), (27, 0.016010278835892677)], Temporal MSE: [(0, 8.443061233265325e-06), (1, 4.103540049982257e-06), (2, 1.9792063540080562e-05), (3, 3.892970926244743e-05), (4, 0.00012994972348678857), (5, 0.0002784595126286149), (6, 0.00033916052780114114), (7, 0.0004055759636685252), (8, 0.00048456128570251167), (9, 0.000642933591734618), (10, 0.0010133603354915977), (11, 0.0019121579825878143), (12, 0.0019103166414424777), (13, 0.0041410475969314575), (14, 0.005523188970983028), (15, 0.0034037609584629536), (16, 0.0025246269069612026), (17, 0.0020217904821038246), (18, 0.002068290254101157), (19, 0.0020761084742844105), (20, 0.005516538396477699), (21, 0.00359907653182745), (22, 0.0031707710586488247), (23, 0.004901545122265816), (24, 0.010967940092086792), (25, 0.02746574394404888), (26, 0.00951562449336052), (27, 0.0010213881032541394)]
+
+ 82%|████████▏ | 41/50 [06:59<01:30, 10.01s/it]Time step 160, Spatial MSE: [(0, 8.901847650122363e-06), (1, 3.3339852961944416e-05), (2, 0.00010366836795583367), (3, 3.930761886294931e-05), (4, 2.1788075173390098e-05), (5, 3.741048203664832e-05), (6, 2.8892003683722578e-05), (7, 5.854344635736197e-05), (8, 4.2701871279859915e-05), (9, 4.26829865318723e-05), (10, 5.5865901231300086e-05), (11, 7.046667451504618e-05), (12, 8.037428051466122e-05), (13, 0.000100770965218544), (14, 0.0001299776486121118), (15, 0.00011585339962039143), (16, 0.00012670195428654552), (17, 0.0001041836294461973), (18, 0.00014130136696621776), (19, 0.00021185634250286967), (20, 0.00017430513980798423), (21, 0.00021751328313257545), (22, 0.00032625303720124066), (23, 0.0006247421260923147), (24, 0.0010662222048267722), (25, 0.0019223830895498395), (26, 0.0028684099670499563), (27, 0.015136201865971088)], Temporal MSE: [(0, 8.634295227238908e-06), (1, 4.676677690440556e-06), (2, 9.263048923457973e-06), (3, 1.0176391697314102e-05), (4, 1.093060382117983e-05), (5, 2.141380355169531e-05), (6, 2.8578702767845243e-05), (7, 2.7739804863813333e-05), (8, 3.176785321556963e-05), (9, 4.352708128863014e-05), (10, 5.75373851461336e-05), (11, 6.740519893355668e-05), (12, 7.118699431885034e-05), (13, 0.0001303285243920982), (14, 9.19951344258152e-05), (15, 9.711311577120796e-05), (16, 7.933322922326624e-05), (17, 6.338467937894166e-05), (18, 8.965760935097933e-05), (19, 7.319505675695837e-05), (20, 0.00011801649816334248), (21, 0.00014167421613819897), (22, 0.0001653444196563214), (23, 0.0002925494918599725), (24, 0.000445984915131703), (25, 0.0008320041815750301), (26, 0.0010578841902315617), (27, 0.0006284150294959545)]
+
+ 84%|████████▍ | 42/50 [07:08<01:17,  9.67s/it]Time step 140, Spatial MSE: [(0, 1.2556575711641926e-05), (1, 3.422801091801375e-05), (2, 0.00037863096804358065), (3, 7.145612471504137e-05), (4, 0.00020097252854611725), (5, 0.00033601862378418446), (6, 0.000653645780403167), (7, 0.0016225248109549284), (8, 0.001972883939743042), (9, 0.003034898778423667), (10, 0.0029330363031476736), (11, 0.0036634269636124372), (12, 0.004175031092017889), (13, 0.0047719827853143215), (14, 0.008279728703200817), (15, 0.014099452644586563), (16, 0.010439569130539894), (17, 0.009106813929975033), (18, 0.010002958588302135), (19, 0.007818853482604027), (20, 0.010507014580070972), (21, 0.01032082550227642), (22, 0.009995860978960991), (23, 0.01645859144628048), (24, 0.03125026449561119), (25, 0.056366171687841415), (26, 0.06841887533664703), (27, 0.009382511489093304)], Temporal MSE: [(0, 1.2118389349780045e-05), (1, 6.329089046630543e-06), (2, 3.225500768166967e-05), (3, 0.0001055835818988271), (4, 0.00019891693955287337), (5, 0.00033000868279486895), (6, 0.0008272646809928119), (7, 0.0010117770871147513), (8, 0.001106668496504426), (9, 0.0019019506871700287), (10, 0.0029672535602003336), (11, 0.004061990417540073), (12, 0.004438137635588646), (13, 0.008591590449213982), (14, 0.011288687586784363), (15, 0.008495948277413845), (16, 0.006155809387564659), (17, 0.005462991539388895), (18, 0.004528236109763384), (19, 0.005553664173930883), (20, 0.010464374907314777), (21, 0.008666999638080597), (22, 0.00663079833611846), (23, 0.009941383264958858), (24, 0.02339170314371586), (25, 0.04474740847945213), (26, 0.021445104852318764), (27, 0.0022783048916608095)]
+
+ 86%|████████▌ | 43/50 [07:19<01:11, 10.23s/it]Time step 120, Spatial MSE: [(0, 1.150273146777181e-05), (1, 4.800968599738553e-05), (2, 0.00019212001643609256), (3, 5.0384329369990155e-05), (4, 3.105034556938335e-05), (5, 5.871472239959985e-05), (6, 4.472661385079846e-05), (7, 6.429603672586381e-05), (8, 6.219414353836328e-05), (9, 6.058496001060121e-05), (10, 7.713243394391611e-05), (11, 8.951337076723576e-05), (12, 0.00010277031105943024), (13, 0.0001332763204118237), (14, 0.0001779075973900035), (15, 0.00016601216339040548), (16, 0.00014065088180359453), (17, 0.00015224266098812222), (18, 0.0002149468200514093), (19, 0.0002837343199644238), (20, 0.00025795481633394957), (21, 0.00044742345926351845), (22, 0.0004998425720259547), (23, 0.0009360786061733961), (24, 0.0015813373029232025), (25, 0.0027685139793902636), (26, 0.003474956611171365), (27, 0.017565814778208733)], Temporal MSE: [(0, 1.2106014764867723e-05), (1, 6.439374374167528e-06), (2, 1.3256175407150295e-05), (3, 1.4990227100497577e-05), (4, 1.6872756532393396e-05), (5, 3.275448761996813e-05), (6, 4.212748899590224e-05), (7, 4.020171400043182e-05), (8, 4.8282592615578324e-05), (9, 6.27657791483216e-05), (10, 7.402069604722783e-05), (11, 9.555382712278515e-05), (12, 0.00010220819240203127), (13, 0.00016828509978950024), (14, 0.0001223170111188665), (15, 0.00015671418805141002), (16, 0.00010607206058921292), (17, 9.346046135760844e-05), (18, 0.0001050387290888466), (19, 0.00011947212624363601), (20, 0.000193800384295173), (21, 0.0002702396013773978), (22, 0.00022366823395714164), (23, 0.00039151444798335433), (24, 0.0006908773211762309), (25, 0.0011851801536977291), (26, 0.0015583653002977371), (27, 0.0009154431172646582)]
+
+ 88%|████████▊ | 44/50 [07:28<00:58,  9.73s/it]Time step 100, Spatial MSE: [(0, 1.4405472938960884e-05), (1, 5.111682912684046e-05), (2, 0.0006906518246978521), (3, 7.736470433883369e-05), (4, 0.00014398856728803366), (5, 0.0002100516576319933), (6, 0.0004254803934600204), (7, 0.000604062806814909), (8, 0.0006902686436660588), (9, 0.0011128995101898909), (10, 0.0011350489221513271), (11, 0.001369172241538763), (12, 0.001580126117914915), (13, 0.0019170386949554086), (14, 0.0031497885938733816), (15, 0.004998032934963703), (16, 0.0031187364365905523), (17, 0.002907011890783906), (18, 0.003343817312270403), (19, 0.003073466010391712), (20, 0.0034670988097786903), (21, 0.005095274653285742), (22, 0.0053720660507678986), (23, 0.009602529928088188), (24, 0.01790737360715866), (25, 0.051703162491321564), (26, 0.07033784687519073), (27, 0.011372135952115059)], Temporal MSE: [(0, 1.7052943803719245e-05), (1, 8.754668670007959e-06), (2, 2.531413883843925e-05), (3, 4.362304389360361e-05), (4, 6.807261524954811e-05), (5, 0.0001201532140839845), (6, 0.0002492490748409182), (7, 0.00028675762587226927), (8, 0.00038501882227137685), (9, 0.0006839582929387689), (10, 0.0007930834544822574), (11, 0.0010828272206708789), (12, 0.001326742465607822), (13, 0.0023134236689656973), (14, 0.002661430276930332), (15, 0.002290064003318548), (16, 0.0019051459385082126), (17, 0.001789122587069869), (18, 0.001806455198675394), (19, 0.0016789842629805207), (20, 0.002824866445735097), (21, 0.0022754324600100517), (22, 0.0021396821830421686), (23, 0.003990307450294495), (24, 0.007055153604596853), (25, 0.012571815401315689), (26, 0.00979334581643343), (27, 0.0018988405354321003)]
+
+ 90%|█████████ | 45/50 [07:37<00:47,  9.46s/it]Time step 80, Spatial MSE: [(0, 1.5027874724182766e-05), (1, 7.053583976812661e-05), (2, 0.0006261523813009262), (3, 5.135327228344977e-05), (4, 8.137373515637591e-05), (5, 0.00014142831787467003), (6, 0.00019328994676470757), (7, 0.00026764068752527237), (8, 0.0003092814004048705), (9, 0.000412302790209651), (10, 0.0004006950184702873), (11, 0.0005159830907359719), (12, 0.0005719875334762037), (13, 0.0006836416432633996), (14, 0.0012644811067730188), (15, 0.002085470361635089), (16, 0.0012246303958818316), (17, 0.0012367701856419444), (18, 0.0014288617530837655), (19, 0.0015024030581116676), (20, 0.0016434855060651898), (21, 0.0022904514335095882), (22, 0.0025839097797870636), (23, 0.005100317299365997), (24, 0.00905788317322731), (25, 0.020114846527576447), (26, 0.02837781421840191), (27, 0.015564288944005966)], Temporal MSE: [(0, 1.971908932318911e-05), (1, 9.981647963286377e-06), (2, 1.8192597053712234e-05), (3, 2.238639171991963e-05), (4, 3.112628837698139e-05), (5, 5.550417336053215e-05), (6, 0.00011188969074282795), (7, 0.00011661987809929997), (8, 0.00017627427587285638), (9, 0.0002841031819116324), (10, 0.000289334129774943), (11, 0.0004196169029455632), (12, 0.0005447918665595353), (13, 0.0008212528191506863), (14, 0.0009587088134139776), (15, 0.0009288087021559477), (16, 0.0008061628323048353), (17, 0.0007392401457764208), (18, 0.0007848721579648554), (19, 0.0007719604764133692), (20, 0.001386775984428823), (21, 0.0010934798046946526), (22, 0.0010037090396508574), (23, 0.001982197631150484), (24, 0.0033223028294742107), (25, 0.00604067649692297), (26, 0.004795941058546305), (27, 0.001842991914600134)]
+
+ 92%|█████████▏| 46/50 [07:45<00:36,  9.16s/it]Time step 60, Spatial MSE: [(0, 1.997205981751904e-05), (1, 9.932173998095095e-05), (2, 0.0008797853370197117), (3, 6.864764145575464e-05), (4, 0.00011539394472492859), (5, 0.00019651808543130755), (6, 0.00026554844225756824), (7, 0.0003535657888278365), (8, 0.0004116635536774993), (9, 0.0005560507997870445), (10, 0.0004837636079173535), (11, 0.000629503047093749), (12, 0.000739692070055753), (13, 0.0008602996822446585), (14, 0.0014963987050577998), (15, 0.002506269607692957), (16, 0.0015665731625631452), (17, 0.0015511878300458193), (18, 0.0020247953943908215), (19, 0.002153337001800537), (20, 0.0023613134399056435), (21, 0.003579474985599518), (22, 0.003517736680805683), (23, 0.0071559143252670765), (24, 0.01269470527768135), (25, 0.02608802542090416), (26, 0.03482541814446449), (27, 0.0182206891477108)], Temporal MSE: [(0, 2.657723962329328e-05), (1, 1.348048863292206e-05), (2, 2.609600778669119e-05), (3, 2.7797630536952056e-05), (4, 4.0465973143000156e-05), (5, 7.779250154271722e-05), (6, 0.00014834129251539707), (7, 0.00015541435277555138), (8, 0.0002488440368324518), (9, 0.0003978520689997822), (10, 0.00037460163002833724), (11, 0.0005361035000532866), (12, 0.0007030111155472696), (13, 0.0010294828098267317), (14, 0.0012069103540852666), (15, 0.0012289719888940454), (16, 0.001102203968912363), (17, 0.0009822744177654386), (18, 0.0010344809852540493), (19, 0.0011149733327329159), (20, 0.00209834729321301), (21, 0.001600509276613593), (22, 0.0013228771276772022), (23, 0.0026395481545478106), (24, 0.004660526290535927), (25, 0.008271378464996815), (26, 0.006970223505049944), (27, 0.005698848981410265)]
+
+ 94%|█████████▍| 47/50 [07:55<00:27,  9.32s/it]Time step 40, Spatial MSE: [(0, 2.9766117222607136e-05), (1, 0.00015409632760565728), (2, 0.0012805711012333632), (3, 9.678301285021007e-05), (4, 0.0001677791733527556), (5, 0.0002832798636518419), (6, 0.0003794560907408595), (7, 0.0004997492651455104), (8, 0.0006050656083971262), (9, 0.0007729720673523843), (10, 0.0006468771025538445), (11, 0.0008260598406195641), (12, 0.0009337245719507337), (13, 0.0012032121885567904), (14, 0.002419819822534919), (15, 0.0033835957292467356), (16, 0.013301844708621502), (17, 0.002221927512437105), (18, 0.003045968944206834), (19, 0.0033621753100305796), (20, 0.003698271233588457), (21, 0.0054123420268297195), (22, 0.005006543360650539), (23, 0.009976282715797424), (24, 0.018477793782949448), (25, 0.03289889916777611), (26, 0.0452471598982811), (27, 0.027889959514141083)], Temporal MSE: [(0, 3.905894482159056e-05), (1, 1.9537550542736426e-05), (2, 4.495030952966772e-05), (3, 3.727068906300701e-05), (4, 5.4889722377993166e-05), (5, 0.00011612888920353726), (6, 0.00021830118203070015), (7, 0.00022650290338788182), (8, 0.00037242757389321923), (9, 0.0005584637401625514), (10, 0.0005150320939719677), (11, 0.0007150879828259349), (12, 0.000938546727411449), (13, 0.0014197858981788158), (14, 0.0017471557948738337), (15, 0.0017311061965301633), (16, 0.001599105540663004), (17, 0.0013744148891419172), (18, 0.0013892194256186485), (19, 0.0016835194546729326), (20, 0.0031892447732388973), (21, 0.002493972424417734), (22, 0.0018844841979444027), (23, 0.0037204171530902386), (24, 0.007162809371948242), (25, 0.012201258912682533), (26, 0.011262700892984867), (27, 0.02594068832695484)]
+
+ 96%|█████████▌| 48/50 [08:04<00:18,  9.15s/it]Time step 20, Spatial MSE: [(0, 5.661993782268837e-05), (1, 0.0003001861914526671), (2, 0.0021332385949790478), (3, 0.00016662241250742227), (4, 0.0002954692463390529), (5, 0.000487403420265764), (6, 0.0006233266904018819), (7, 0.0008890463504940271), (8, 0.001160412561148405), (9, 0.001397656393237412), (10, 0.0011307557579129934), (11, 0.0013187380973249674), (12, 0.0014476844808086753), (13, 0.002004213398322463), (14, 0.0042916517704725266), (15, 0.0056774369440972805), (16, 0.063730888068676), (17, 0.003898901166394353), (18, 0.005319132469594479), (19, 0.006546275690197945), (20, 0.00694199837744236), (21, 0.00898880884051323), (22, 0.008525820448994637), (23, 0.015585565008223057), (24, 0.03202076256275177), (25, 0.05075177550315857), (26, 0.07007624953985214), (27, 0.0774790346622467)], Temporal MSE: [(0, 7.015348091954365e-05), (1, 3.427678893785924e-05), (2, 9.608727850718424e-05), (3, 6.309655145741999e-05), (4, 9.311808389611542e-05), (5, 0.00020998796389903873), (6, 0.0003993168647866696), (7, 0.00040904307388700545), (8, 0.0006365837180055678), (9, 0.0008904337882995605), (10, 0.0008446623687632382), (11, 0.0011178075801581144), (12, 0.0014592843363061547), (13, 0.0023997933603823185), (14, 0.003603173652663827), (15, 0.00285691162571311), (16, 0.0027132073882967234), (17, 0.002267109462991357), (18, 0.002358010271564126), (19, 0.002849898301064968), (20, 0.0052313487976789474), (21, 0.004406035412102938), (22, 0.0033391688484698534), (23, 0.00640142522752285), (24, 0.014093490317463875), (25, 0.023412508890032768), (26, 0.025795811787247658), (27, 0.21991875767707825)]
+
+ 98%|█████████▊| 49/50 [08:15<00:09,  9.74s/it]Time step 0, Spatial MSE: [(0, 0.0003917538851965219), (1, 0.0017043319530785084), (2, 0.009577520191669464), (3, 0.0009448090568184853), (4, 0.0019754129461944103), (5, 0.002345579443499446), (6, 0.0026826097164303064), (7, 0.004048488102853298), (8, 0.007653249427676201), (9, 0.009374118410050869), (10, 0.008335217833518982), (11, 0.006053987890481949), (12, 0.006221517454832792), (13, 0.007053909357637167), (14, 0.009742319583892822), (15, 0.02096656709909439), (16, 0.06332537531852722), (17, 0.013664890080690384), (18, 0.021336253732442856), (19, 0.027073374018073082), (20, 0.02623002976179123), (21, 0.02797764539718628), (22, 0.05201015993952751), (23, 0.1417423039674759), (24, 0.27012908458709717), (25, 0.8077110648155212), (26, 1.3570364713668823), (27, 1.6697068214416504)], Temporal MSE: [(0, 0.0005473540513776243), (1, 0.0002125277096638456), (2, 0.0007511857547797263), (3, 0.0003847114276140928), (4, 0.0005995228420943022), (5, 0.0010105023393407464), (6, 0.0022009864915162325), (7, 0.001968302996829152), (8, 0.0023901062086224556), (9, 0.0028190354350954294), (10, 0.0036779840011149645), (11, 0.004363720770925283), (12, 0.005824522580951452), (13, 0.00785150658339262), (14, 0.010560904629528522), (15, 0.009752865880727768), (16, 0.007691245060414076), (17, 0.007242592517286539), (18, 0.009567037224769592), (19, 0.009191319346427917), (20, 0.013213162310421467), (21, 0.012870102189481258), (22, 0.022275004535913467), (23, 0.03954176977276802), (24, 0.10355062782764435), (25, 0.17843668162822723), (26, 0.4115753471851349), (27, 3.80991530418396)]
+
+100%|██████████| 50/50 [08:26<00:00, 10.21s/it]
+100%|██████████| 50/50 [08:26<00:00, 10.13s/it]
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv3d(
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:456: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv2d(input, weight, bias, self.stride,
+Saved video to ./samples/latte/Sunset_over_the_sea._0000.mp4
+Saved spatial MLP MSE outputs to ./samples/latte/Sunset_over_the_sea._0000_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/latte/Sunset_over_the_sea._0000_temporal_mlp_mse.pt
+Processing the (A dog in astronaut suit and sunglasses floating in space.) prompt
+
+  0%|          | 0/50 [00:00<?, ?it/s]
+  2%|▏         | 1/50 [00:04<03:41,  4.52s/it]Time step 960, Spatial MSE: [(0, 1.0308115633961279e-05), (1, 1.5921305021038279e-06), (2, 2.2948817786527798e-05), (3, 2.7016914827981964e-05), (4, 7.8623641456943e-05), (5, 5.0501126679591835e-05), (6, 6.497796130133793e-05), (7, 7.238682155730203e-05), (8, 9.64604623732157e-05), (9, 0.00022023623751010746), (10, 0.0005157039267942309), (11, 0.0011324048973619938), (12, 0.0009513854165561497), (13, 0.0017683545593172312), (14, 0.0037185284309089184), (15, 0.006156992632895708), (16, 0.016454624012112617), (17, 0.003380389651283622), (18, 0.004302598536014557), (19, 0.0034681421238929033), (20, 0.007634567096829414), (21, 0.004038236103951931), (22, 0.0011577834375202656), (23, 0.00253680651076138), (24, 0.007087287958711386), (25, 0.005501533392816782), (26, 0.004889015574008226), (27, 0.002073562704026699)], Temporal MSE: [(0, 1.4408116840058938e-05), (1, 4.767839527630713e-06), (2, 4.119595359952655e-06), (3, 7.037304385448806e-06), (4, 7.645812729606405e-06), (5, 1.6961081200861372e-05), (6, 4.6930155804147944e-05), (7, 6.319463136605918e-05), (8, 6.064030094421469e-05), (9, 0.0003194160817656666), (10, 0.000440495612565428), (11, 0.000499530287925154), (12, 0.0007653117645531893), (13, 0.0007812491967342794), (14, 0.0005217030411586165), (15, 0.00046117196325212717), (16, 0.0008716167067177594), (17, 0.00042108504567295313), (18, 0.00022620888194069266), (19, 0.00013646986917592585), (20, 0.000228345554205589), (21, 0.0003817106189671904), (22, 0.0003216468612663448), (23, 0.00038559423410333693), (24, 0.0006992589333094656), (25, 0.0005396882770583034), (26, 0.00034158502239733934), (27, 9.939512892742641e-06)]
+
+  4%|▍         | 2/50 [00:15<06:48,  8.51s/it]Time step 940, Spatial MSE: [(0, 1.1758880646084435e-05), (1, 1.8322955384064699e-06), (2, 2.2524922314914875e-05), (3, 2.884050627471879e-05), (4, 6.387326720869169e-05), (5, 6.179004412842914e-05), (6, 9.488972864346579e-05), (7, 9.608067193767056e-05), (8, 0.00016318439156748354), (9, 0.0003724828129634261), (10, 0.0007820830796845257), (11, 0.0015776043292135), (12, 0.0019141238881275058), (13, 0.0030644533690065145), (14, 0.005523473024368286), (15, 0.008436973206698895), (16, 0.008974006399512291), (17, 0.004555968567728996), (18, 0.005638436414301395), (19, 0.003723087487742305), (20, 0.008932815864682198), (21, 0.00451379269361496), (22, 0.001433035940863192), (23, 0.002780939918011427), (24, 0.007238825783133507), (25, 0.006734249647706747), (26, 0.00583653524518013), (27, 0.0021931182127445936)], Temporal MSE: [(0, 1.5511448509641923e-05), (1, 5.064039669377962e-06), (2, 4.8087954382936005e-06), (3, 7.798191290930845e-06), (4, 8.564030395064037e-06), (5, 2.1987441868986934e-05), (6, 7.099996582837775e-05), (7, 9.241918451152742e-05), (8, 9.68961394391954e-05), (9, 0.0005223573534749448), (10, 0.0005610815715044737), (11, 0.0008727657841518521), (12, 0.0014968272298574448), (13, 0.0014189290814101696), (14, 0.0007253324729390442), (15, 0.0009791747434064746), (16, 0.001369734643958509), (17, 0.0006177113973535597), (18, 0.000274827383691445), (19, 0.00017991033382713795), (20, 0.0002903986896853894), (21, 0.0005134613020345569), (22, 0.00041501550003886223), (23, 0.0004860082408413291), (24, 0.0009650385472923517), (25, 0.0006893121171742678), (26, 0.0003993844147771597), (27, 2.0120825865888037e-05)]
+
+  6%|▌         | 3/50 [00:26<07:21,  9.40s/it]Time step 920, Spatial MSE: [(0, 1.2048408279952127e-05), (1, 1.9057315512327477e-06), (2, 2.5620402084314264e-05), (3, 2.766206489468459e-05), (4, 4.756005000672303e-05), (5, 5.8300443924963474e-05), (6, 0.00011866368731716648), (7, 0.00010552470484981313), (8, 0.00019999509095214307), (9, 0.0004464272642508149), (10, 0.001186709851026535), (11, 0.0017324733780696988), (12, 0.0020672862883657217), (13, 0.003006517654284835), (14, 0.009856989607214928), (15, 0.011409787461161613), (16, 0.01182110607624054), (17, 0.005711589939892292), (18, 0.005759710446000099), (19, 0.0029768263921141624), (20, 0.007405360694974661), (21, 0.005482485052198172), (22, 0.001182219828478992), (23, 0.0018976061837747693), (24, 0.005501012317836285), (25, 0.006264592986553907), (26, 0.005511360242962837), (27, 0.002550181932747364)], Temporal MSE: [(0, 1.7837019186117686e-05), (1, 5.839902769366745e-06), (2, 5.40923474545707e-06), (3, 7.695076419622637e-06), (4, 9.24722735362593e-06), (5, 2.188085090892855e-05), (6, 8.095670636976138e-05), (7, 0.00012591689301189035), (8, 0.00013381705502979457), (9, 0.000746116740629077), (10, 0.0007476513856090605), (11, 0.0011703980853781104), (12, 0.0019387754146009684), (13, 0.0018924684263765812), (14, 0.0009484928450547159), (15, 0.0013504757080227137), (16, 0.0018806704320013523), (17, 0.0009191799908876419), (18, 0.00045569095527753234), (19, 0.00019084473024122417), (20, 0.000291848904453218), (21, 0.000637306715361774), (22, 0.00046415627002716064), (23, 0.00045641453471034765), (24, 0.0008297119638882577), (25, 0.0006564543000422418), (26, 0.0003889967920258641), (27, 1.6140656953211874e-05)]
+
+  8%|▊         | 4/50 [00:37<07:42, 10.05s/it]Time step 900, Spatial MSE: [(0, 1.571132270328235e-05), (1, 2.2145202365209116e-06), (2, 2.9520826501538977e-05), (3, 3.279231532360427e-05), (4, 3.6166966310702264e-05), (5, 6.859967834316194e-05), (6, 0.00021811824990436435), (7, 0.0001696907711448148), (8, 0.0003530555695760995), (9, 0.0012321951799094677), (10, 0.003744852263480425), (11, 0.004102974198758602), (12, 0.004220260307192802), (13, 0.004660570994019508), (14, 0.020276755094528198), (15, 0.018712656572461128), (16, 0.019801022484898567), (17, 0.008726514875888824), (18, 0.008312505669891834), (19, 0.0034631346352398396), (20, 0.0075671011582016945), (21, 0.006346370559185743), (22, 0.0013450036058202386), (23, 0.001965845236554742), (24, 0.005486036650836468), (25, 0.007216806057840586), (26, 0.007813132368028164), (27, 0.0032452873419970274)], Temporal MSE: [(0, 2.1978501536068507e-05), (1, 6.832843610027339e-06), (2, 7.08632251189556e-06), (3, 9.414440683030989e-06), (4, 1.1793104022217449e-05), (5, 3.1385421607410535e-05), (6, 0.00013809661322738975), (7, 0.0002463945420458913), (8, 0.0003062324831262231), (9, 0.0016190944006666541), (10, 0.0016868425300344825), (11, 0.002425091341137886), (12, 0.0037167349364608526), (13, 0.0031701303087174892), (14, 0.0020061105024069548), (15, 0.002415640978142619), (16, 0.0028213050682097673), (17, 0.00185391993727535), (18, 0.00104123807977885), (19, 0.0004698633565567434), (20, 0.00061815045773983), (21, 0.0012086383067071438), (22, 0.0008194553083740175), (23, 0.0007274670642800629), (24, 0.0011759850895032287), (25, 0.0010760602308437228), (26, 0.0006457131821662188), (27, 2.3461740056518465e-05)]
+
+ 10%|█         | 5/50 [00:48<07:45, 10.35s/it]Time step 880, Spatial MSE: [(0, 1.222960781888105e-05), (1, 2.1669511625077575e-06), (2, 3.523439954733476e-05), (3, 2.4892018700484186e-05), (4, 3.2841970096342266e-05), (5, 7.159837696235627e-05), (6, 0.00027349128504283726), (7, 0.00027226528618484735), (8, 0.0005106000462546945), (9, 0.0027817650698125362), (10, 0.007845592685043812), (11, 0.005284290295094252), (12, 0.005343626718968153), (13, 0.0042309449054300785), (14, 0.016255971044301987), (15, 0.015766654163599014), (16, 0.015498657710850239), (17, 0.0077858781442046165), (18, 0.007973859086632729), (19, 0.004609246272593737), (20, 0.008185197599232197), (21, 0.004746817983686924), (22, 0.0019922107458114624), (23, 0.002685511251911521), (24, 0.005930202081799507), (25, 0.012624312192201614), (26, 0.007110575679689646), (27, 0.0030569781083613634)], Temporal MSE: [(0, 1.86609395314008e-05), (1, 6.354630386340432e-06), (2, 6.800213668611832e-06), (3, 7.019503755145706e-06), (4, 1.4142089639790356e-05), (5, 4.2593481339281425e-05), (6, 0.0002860739186871797), (7, 0.00038935901829972863), (8, 0.0004590385942719877), (9, 0.001955808140337467), (10, 0.0019217352382838726), (11, 0.0023654864635318518), (12, 0.0025833973195403814), (13, 0.0021454626694321632), (14, 0.0019338851561769843), (15, 0.0017903060652315617), (16, 0.0016015064902603626), (17, 0.0018071250524371862), (18, 0.001159469480626285), (19, 0.0006766414735466242), (20, 0.0008079573744907975), (21, 0.0011495425133034587), (22, 0.0010258456459268928), (23, 0.0008400471415370703), (24, 0.0013601137325167656), (25, 0.0014822547091171145), (26, 0.0006857336265966296), (27, 2.0508938177954406e-05)]
+
+ 12%|█▏        | 6/50 [00:58<07:39, 10.45s/it]Time step 860, Spatial MSE: [(0, 4.503183299675584e-05), (1, 7.3457922553643584e-06), (2, 3.9211532566696405e-05), (3, 7.909393752925098e-05), (4, 0.00013441785995382816), (5, 0.00016121591033879668), (6, 0.000605572247877717), (7, 0.0007326523773372173), (8, 0.0009457826963625848), (9, 0.0029951026663184166), (10, 0.008428145200014114), (11, 0.0043296292424201965), (12, 0.004353228025138378), (13, 0.0036507307086139917), (14, 0.012357987463474274), (15, 0.014379961416125298), (16, 0.013675983063876629), (17, 0.008619129657745361), (18, 0.010133817791938782), (19, 0.007559970021247864), (20, 0.01414657011628151), (21, 0.007383796852082014), (22, 0.003342271549627185), (23, 0.00396680599078536), (24, 0.007368453778326511), (25, 0.018981512635946274), (26, 0.022532084956765175), (27, 0.011138183064758778)], Temporal MSE: [(0, 6.859096902189776e-05), (1, 1.951863305293955e-05), (2, 2.041518200712744e-05), (3, 2.2832638933323324e-05), (4, 3.715788261615671e-05), (5, 9.423252049600706e-05), (6, 0.0005134706734679639), (7, 0.0006011062650941312), (8, 0.0004600680840667337), (9, 0.0017342151841148734), (10, 0.0017074005445465446), (11, 0.0017870863666757941), (12, 0.00174809992313385), (13, 0.0020852868910878897), (14, 0.0024134914856404066), (15, 0.0017028535949066281), (16, 0.0016501449281349778), (17, 0.001928283367305994), (18, 0.0011360952630639076), (19, 0.0007130287121981382), (20, 0.0012137419544160366), (21, 0.0014583184383809566), (22, 0.0017344803782179952), (23, 0.001335185137577355), (24, 0.0023153929505497217), (25, 0.0031620264053344727), (26, 0.0014506797306239605), (27, 9.176744788419455e-05)]
+
+ 14%|█▍        | 7/50 [01:08<07:14, 10.09s/it]Time step 840, Spatial MSE: [(0, 5.180720108910464e-06), (1, 2.714005631787586e-06), (2, 5.460221291286871e-05), (3, 1.5901292499620467e-05), (4, 2.3478320144931786e-05), (5, 6.930715608177707e-05), (6, 0.0003583781945053488), (7, 0.0003462102322373539), (8, 0.0005307505489327013), (9, 0.0012281801318749785), (10, 0.0018787477165460587), (11, 0.0011121820425614715), (12, 0.0011972576612606645), (13, 0.0010282050352543592), (14, 0.0038618138059973717), (15, 0.0035367468371987343), (16, 0.0022102477960288525), (17, 0.0019311372889205813), (18, 0.001635017804801464), (19, 0.0011209237854927778), (20, 0.0015336950309574604), (21, 0.0017281605396419764), (22, 0.0008898737723939121), (23, 0.001267745508812368), (24, 0.0029436240438371897), (25, 0.006208838429301977), (26, 0.0025735958479344845), (27, 0.004705002065747976)], Temporal MSE: [(0, 8.41179644339718e-06), (1, 3.0939308999222703e-06), (2, 6.136984666227363e-06), (3, 6.645248049608199e-06), (4, 1.3602063518192153e-05), (5, 7.601099059684202e-05), (6, 0.00038608082104474306), (7, 0.00023076889920048416), (8, 0.00017160862626042217), (9, 0.0004183416021987796), (10, 0.0005142703885212541), (11, 0.0005918159149587154), (12, 0.0006490280502475798), (13, 0.0008681642357259989), (14, 0.001167566399089992), (15, 0.0008766807150095701), (16, 0.0009197519393637776), (17, 0.0008418436045758426), (18, 0.000995768466964364), (19, 0.0006057649734430015), (20, 0.0008737525786273181), (21, 0.0009881312726065516), (22, 0.0004401570768095553), (23, 0.00044992094626650214), (24, 0.0011253247503191233), (25, 0.0011331957066431642), (26, 0.000450843945145607), (27, 2.7706750188372098e-05)]
+
+ 16%|█▌        | 8/50 [01:17<06:48,  9.72s/it]Time step 820, Spatial MSE: [(0, 1.3417189620668069e-05), (1, 2.567112915130565e-06), (2, 5.6011893320828676e-05), (3, 3.0994215194368735e-05), (4, 8.759504999034107e-05), (5, 0.00010870926053030416), (6, 0.00039419226231984794), (7, 0.0009214135934598744), (8, 0.0007332891691476107), (9, 0.001269067288376391), (10, 0.0018828277243301272), (11, 0.0014441298553720117), (12, 0.0017483942210674286), (13, 0.0014247223734855652), (14, 0.006202802062034607), (15, 0.005208242684602737), (16, 0.004166083876043558), (17, 0.0035189075861126184), (18, 0.0028137906920164824), (19, 0.0018814901122823358), (20, 0.0028013649862259626), (21, 0.003052076557651162), (22, 0.0013327001361176372), (23, 0.0017450544983148575), (24, 0.004083854146301746), (25, 0.009720023721456528), (26, 0.00856509618461132), (27, 0.004057351034134626)], Temporal MSE: [(0, 1.8521797755965963e-05), (1, 6.282056347117759e-06), (2, 1.237910873896908e-05), (3, 1.861893542809412e-05), (4, 3.214017488062382e-05), (5, 7.952401938382536e-05), (6, 0.00048578897258266807), (7, 0.0003217242774553597), (8, 0.00021787226432934403), (9, 0.0004902931395918131), (10, 0.0006063851178623736), (11, 0.0007108569843694568), (12, 0.0008400434162467718), (13, 0.001158329308964312), (14, 0.0015639924677088857), (15, 0.0011324095539748669), (16, 0.0012058336287736893), (17, 0.00137134559918195), (18, 0.001350001897662878), (19, 0.0008567585027776659), (20, 0.0011338163167238235), (21, 0.0015986766666173935), (22, 0.0009013405069708824), (23, 0.0008589173667132854), (24, 0.001759978593327105), (25, 0.0019764688331633806), (26, 0.0008593245293013752), (27, 2.9201935831224546e-05)]
+
+ 18%|█▊        | 9/50 [01:26<06:38,  9.71s/it]Time step 800, Spatial MSE: [(0, 1.3456166925607249e-05), (1, 3.0018939014553325e-06), (2, 7.032697612885386e-05), (3, 3.0841078114463016e-05), (4, 0.00011996951070614159), (5, 0.00012419407721608877), (6, 0.00026652016094885767), (7, 0.0010107061825692654), (8, 0.0006769625470042229), (9, 0.0011826992267742753), (10, 0.0017745279474183917), (11, 0.0015751038445159793), (12, 0.001813583541661501), (13, 0.0015006362227723002), (14, 0.005476032383739948), (15, 0.0052108182571828365), (16, 0.0034061362966895103), (17, 0.0031584107782691717), (18, 0.002829996170476079), (19, 0.0018699587089940906), (20, 0.0025902597699314356), (21, 0.002284976653754711), (22, 0.001462138956412673), (23, 0.001868658815510571), (24, 0.004040038213133812), (25, 0.009682917967438698), (26, 0.008450695313513279), (27, 0.004612313583493233)], Temporal MSE: [(0, 1.8541375538916327e-05), (1, 6.0526831475726794e-06), (2, 1.332517058472149e-05), (3, 2.842465983121656e-05), (4, 4.105574043933302e-05), (5, 6.86304338159971e-05), (6, 0.0004156094801146537), (7, 0.0002916065277531743), (8, 0.0002243867056677118), (9, 0.0005125974421389401), (10, 0.0006361062405630946), (11, 0.0007436869782395661), (12, 0.0009037663694471121), (13, 0.0012538571609184146), (14, 0.0018606672529131174), (15, 0.0012609758414328098), (16, 0.0012015881948173046), (17, 0.001138443243689835), (18, 0.0007965924451127648), (19, 0.0006454696413129568), (20, 0.0008075986988842487), (21, 0.0010491956491023302), (22, 0.0007742654997855425), (23, 0.0008187509374693036), (24, 0.0018045814940705895), (25, 0.0020951880142092705), (26, 0.000977966352365911), (27, 3.532040864229202e-05)]
+
+ 20%|██        | 10/50 [01:37<06:38,  9.97s/it]Time step 780, Spatial MSE: [(0, 1.1074912436015438e-05), (1, 2.732448592723813e-06), (2, 6.887028575874865e-05), (3, 2.1766867575934157e-05), (4, 4.2152612877544016e-05), (5, 4.087307752342895e-05), (6, 0.00010455094161443412), (7, 0.00010798395669553429), (8, 0.0001471834839321673), (9, 0.00022173699107952416), (10, 0.0006149737746454775), (11, 0.0006513422704301775), (12, 0.0006162719801068306), (13, 0.0006179980118758976), (14, 0.0009912247769534588), (15, 0.0007263162406161427), (16, 0.0005235720309428871), (17, 0.0007772335084155202), (18, 0.0004286413313820958), (19, 0.00045073486398905516), (20, 0.0006573938298970461), (21, 0.0006434359238483012), (22, 0.0006098854937590659), (23, 0.0008639724110253155), (24, 0.001957655418664217), (25, 0.005633952096104622), (26, 0.003178413026034832), (27, 0.004984011873602867)], Temporal MSE: [(0, 1.3095579561195336e-05), (1, 5.1886395340261515e-06), (2, 1.0887539247050881e-05), (3, 1.2298522051423788e-05), (4, 1.665527088334784e-05), (5, 2.687927735678386e-05), (6, 6.97164359735325e-05), (7, 0.000130847271066159), (8, 7.659263792447746e-05), (9, 0.00022323486336972564), (10, 0.00039737080805934966), (11, 0.00029941051616333425), (12, 0.0004516690387390554), (13, 0.0004965834668837488), (14, 0.0004994476912543178), (15, 0.0004569098528008908), (16, 0.0003802453866228461), (17, 0.00025242401170544326), (18, 0.00040751631604507565), (19, 0.00015908951172605157), (20, 0.0003849792992696166), (21, 0.0007255434757098556), (22, 0.00022479954350274056), (23, 0.0003982259950134903), (24, 0.0008361780783161521), (25, 0.0013354925904422998), (26, 0.00049753044731915), (27, 3.471028321655467e-05)]
+
+ 22%|██▏       | 11/50 [01:47<06:31, 10.05s/it]Time step 760, Spatial MSE: [(0, 9.577050150255673e-06), (1, 3.109773388132453e-06), (2, 5.626600977848284e-05), (3, 9.67141295404872e-06), (4, 9.180388587992638e-06), (5, 7.439755790983327e-06), (6, 1.2293627150938846e-05), (7, 1.3335211406229064e-05), (8, 1.7921534890774637e-05), (9, 2.2119389541330747e-05), (10, 3.0403820346691646e-05), (11, 4.278549749869853e-05), (12, 6.213026063051075e-05), (13, 8.934425568440929e-05), (14, 8.899608656065539e-05), (15, 0.0001088019780581817), (16, 0.00023970362963154912), (17, 0.00014033890329301357), (18, 0.00016661634435877204), (19, 0.00016365807096008211), (20, 0.0002253073180327192), (21, 0.00037051853723824024), (22, 0.00016195101488847286), (23, 0.00034963907091878355), (24, 0.0009473207173869014), (25, 0.0014526861486956477), (26, 0.0011756120948120952), (27, 0.004280671942979097)], Temporal MSE: [(0, 1.717644408927299e-05), (1, 5.908997536607785e-06), (2, 5.013285772292875e-06), (3, 3.399154820726835e-06), (4, 2.3914183202577988e-06), (5, 4.024978807137813e-06), (6, 6.0911979744560085e-06), (7, 8.36068738863105e-06), (8, 7.490586995118065e-06), (9, 2.1261257643345743e-05), (10, 1.6008045349735767e-05), (11, 4.524448013398796e-05), (12, 5.081448034616187e-05), (13, 7.834233838366345e-05), (14, 6.659759674221277e-05), (15, 8.122855797410011e-05), (16, 0.00012497225543484092), (17, 6.520220631500706e-05), (18, 0.00015422528667841107), (19, 6.468877109000459e-05), (20, 0.00012931274250149727), (21, 0.0002943097788374871), (22, 8.798392809694633e-05), (23, 0.00013685958401765674), (24, 0.00016444189532194287), (25, 0.0004919435014016926), (26, 0.00016431204858236015), (27, 6.182537617860362e-05)]
+
+ 24%|██▍       | 12/50 [01:57<06:21, 10.04s/it]Time step 740, Spatial MSE: [(0, 2.6605117454892024e-05), (1, 1.1907523003173992e-05), (2, 0.00014411931624636054), (3, 0.00014564003504347056), (4, 0.0007642897544428706), (5, 0.00077989132842049), (6, 0.0014136859681457281), (7, 0.004766654688864946), (8, 0.0033541498705744743), (9, 0.005431528203189373), (10, 0.0072425478138029575), (11, 0.007124435622245073), (12, 0.007345939055085182), (13, 0.007996046915650368), (14, 0.01670573092997074), (15, 0.02150850184261799), (16, 0.015155952423810959), (17, 0.015483745373785496), (18, 0.015033883973956108), (19, 0.010906527750194073), (20, 0.01749001070857048), (21, 0.014888539910316467), (22, 0.008345948532223701), (23, 0.009464076720178127), (24, 0.01934426836669445), (25, 0.06958319991827011), (26, 0.04169817268848419), (27, 0.008100387640297413)], Temporal MSE: [(0, 3.972569538746029e-05), (1, 7.41308213036973e-06), (2, 4.6686604036949575e-05), (3, 0.00033458377583883703), (4, 0.00027239933842793107), (5, 0.0003946881624870002), (6, 0.0015281031373888254), (7, 0.0011906916042789817), (8, 0.0012436846736818552), (9, 0.002940817503258586), (10, 0.0037737239617854357), (11, 0.004047006368637085), (12, 0.00471894396468997), (13, 0.007161983288824558), (14, 0.011429239995777607), (15, 0.00650373799726367), (16, 0.006213712505996227), (17, 0.0058490508235991), (18, 0.004105715546756983), (19, 0.0031677945517003536), (20, 0.004725615028291941), (21, 0.004192498978227377), (22, 0.004177693743258715), (23, 0.004147987347096205), (24, 0.010093644261360168), (25, 0.012959463521838188), (26, 0.0058325049467384815), (27, 0.00024427060270681977)]
+
+ 26%|██▌       | 13/50 [02:08<06:19, 10.25s/it]Time step 720, Spatial MSE: [(0, 7.938165254017804e-06), (1, 4.255807652953081e-06), (2, 8.919177344068885e-05), (3, 1.2739588783006184e-05), (4, 1.0250023478874937e-05), (5, 9.550461072649341e-06), (6, 1.3526532711694017e-05), (7, 1.580070602358319e-05), (8, 1.794902163965162e-05), (9, 2.5625451598898508e-05), (10, 2.7335578124620952e-05), (11, 4.36480731877964e-05), (12, 5.899101233808324e-05), (13, 6.691923772450536e-05), (14, 6.85404083924368e-05), (15, 0.0001248646149178967), (16, 0.000155879752128385), (17, 0.00010943003871943802), (18, 0.0002912911004386842), (19, 0.00014685775386169553), (20, 0.0002564970054663718), (21, 0.0005017322837375104), (22, 0.00019824787159450352), (23, 0.0004206555604469031), (24, 0.0010910318233072758), (25, 0.0017940824618563056), (26, 0.0015384014695882797), (27, 0.005088495090603828)], Temporal MSE: [(0, 1.3903203580412082e-05), (1, 4.84597148897592e-06), (2, 5.890453394385986e-06), (3, 3.039700004592305e-06), (4, 3.1305999073083512e-06), (5, 5.342655185813783e-06), (6, 6.896841568959644e-06), (7, 9.537157893646508e-06), (8, 8.309136319439858e-06), (9, 1.6531899746041745e-05), (10, 1.7578067854628898e-05), (11, 3.5600431147031486e-05), (12, 5.0873342843260616e-05), (13, 6.25750981271267e-05), (14, 5.049394894740544e-05), (15, 8.353131852345541e-05), (16, 8.897791121853516e-05), (17, 5.388988574850373e-05), (18, 0.0001882874930743128), (19, 8.725647785468027e-05), (20, 0.00013918441254645586), (21, 0.0003238478093408048), (22, 8.638908184366301e-05), (23, 0.0001753268443280831), (24, 0.0002179637085646391), (25, 0.0006219733622856438), (26, 0.00022701580019202083), (27, 6.0488244344014674e-05)]
+
+ 28%|██▊       | 14/50 [02:17<05:54,  9.86s/it]Time step 700, Spatial MSE: [(0, 3.1276016670744866e-05), (1, 9.655129360908177e-06), (2, 0.00014626771735493094), (3, 5.051328844274394e-05), (4, 0.00016779531142674387), (5, 0.00013535612379200757), (6, 0.0002839681983459741), (7, 0.00027907744515687227), (8, 0.0005055420915596187), (9, 0.0006008266354911029), (10, 0.001317642512731254), (11, 0.0012923995964229107), (12, 0.001340095535852015), (13, 0.0014135611709207296), (14, 0.0018670572899281979), (15, 0.0017118378309533), (16, 0.0012520935852080584), (17, 0.0014828960411250591), (18, 0.0014897746732458472), (19, 0.0010297890985384583), (20, 0.0010927641997113824), (21, 0.0014590523205697536), (22, 0.0014504098799079657), (23, 0.001929488847963512), (24, 0.00599943520501256), (25, 0.013789991848170757), (26, 0.007830129936337471), (27, 0.008972172625362873)], Temporal MSE: [(0, 1.2552436601254158e-05), (1, 5.192221124161733e-06), (2, 2.9865381293348037e-05), (3, 3.906733400071971e-05), (4, 5.4490137699758634e-05), (5, 8.504712604917586e-05), (6, 0.00011946147424168885), (7, 0.00022874516434967518), (8, 0.0001654859952395782), (9, 0.000338713638484478), (10, 0.0005502429557964206), (11, 0.0005324641242623329), (12, 0.0007526753470301628), (13, 0.0011710897088050842), (14, 0.0008910354226827621), (15, 0.000897908816114068), (16, 0.0006041575106792152), (17, 0.0005391155136749148), (18, 0.0006378864636644721), (19, 0.00045103312004357576), (20, 0.0010862445924431086), (21, 0.001670277793891728), (22, 0.0007570950547233224), (23, 0.0012117390288040042), (24, 0.0032675350084900856), (25, 0.006239118054509163), (26, 0.0018609759863466024), (27, 0.00014112386270426214)]
+
+ 30%|███       | 15/50 [02:27<05:50, 10.02s/it]Time step 680, Spatial MSE: [(0, 6.924559784238227e-06), (1, 1.6336496628355235e-05), (2, 0.00013606254651676863), (3, 5.9331749071134254e-05), (4, 6.861178553663194e-05), (5, 0.00010441667109262198), (6, 0.0002615251869428903), (7, 0.00032887861016206443), (8, 0.0002628565125633031), (9, 0.0003065770724788308), (10, 0.0005119250854477286), (11, 0.0005878583760932088), (12, 0.0008500291733071208), (13, 0.0011527559254318476), (14, 0.001633030129596591), (15, 0.006419960409402847), (16, 0.005140959285199642), (17, 0.001791823422536254), (18, 0.002418607473373413), (19, 0.0011327192187309265), (20, 0.0034569399431347847), (21, 0.005073175765573978), (22, 0.002518140710890293), (23, 0.001644284464418888), (24, 0.0037384917959570885), (25, 0.012331699952483177), (26, 0.0027886023744940758), (27, 0.006761936470866203)], Temporal MSE: [(0, 1.935408181452658e-05), (1, 7.1906461016624235e-06), (2, 3.2802690839162096e-05), (3, 4.648278627428226e-05), (4, 8.362869266420603e-05), (5, 9.70471155596897e-05), (6, 0.00029765028739348054), (7, 0.00019881781190633774), (8, 0.00016377805150114), (9, 0.00042053184006363153), (10, 0.0009247175185009837), (11, 0.0011496953666210175), (12, 0.002015196718275547), (13, 0.0027886664029210806), (14, 0.003844713093712926), (15, 0.0020175345707684755), (16, 0.0013974845642223954), (17, 0.0007138490909710526), (18, 0.000541212095413357), (19, 0.0004240736016072333), (20, 0.00156837934628129), (21, 0.001003763871267438), (22, 0.0012485510669648647), (23, 0.0008633562247268856), (24, 0.0029451348818838596), (25, 0.004760585725307465), (26, 0.0031140281353145838), (27, 9.735448111314327e-05)]
+
+ 32%|███▏      | 16/50 [02:40<06:10, 10.89s/it]Time step 660, Spatial MSE: [(0, 2.890619362005964e-05), (1, 1.1999564776488114e-05), (2, 0.00014122405264060944), (3, 0.00010103622480528429), (4, 0.00022372456442099065), (5, 0.0002151016960851848), (6, 0.0004708274791482836), (7, 0.0007941892836242914), (8, 0.0009368619648739696), (9, 0.0010910936398431659), (10, 0.0013498476473614573), (11, 0.0016009798273444176), (12, 0.002217740286141634), (13, 0.0026561764534562826), (14, 0.007500449661165476), (15, 0.0061625223606824875), (16, 0.007291834335774183), (17, 0.0036905466113239527), (18, 0.004955417476594448), (19, 0.0032854126766324043), (20, 0.0038637882098555565), (21, 0.005217572208493948), (22, 0.0033949208445847034), (23, 0.003931555896997452), (24, 0.011235123500227928), (25, 0.022816251963377), (26, 0.013963243924081326), (27, 0.007475560065358877)], Temporal MSE: [(0, 9.389221304445527e-06), (1, 4.230964350426802e-06), (2, 2.221841896243859e-05), (3, 8.118498953990638e-05), (4, 0.00011699054448399693), (5, 0.0001400227047270164), (6, 0.00037169051938690245), (7, 0.0002627662615850568), (8, 0.00027416751254349947), (9, 0.000600190949626267), (10, 0.0007879030890762806), (11, 0.001005635131150484), (12, 0.0012347310548648238), (13, 0.002615341916680336), (14, 0.003282152581959963), (15, 0.002650616457685828), (16, 0.0018639516783878207), (17, 0.0016061357455328107), (18, 0.001732410630211234), (19, 0.0015941723249852657), (20, 0.0029384400695562363), (21, 0.0024227583780884743), (22, 0.0021340365055948496), (23, 0.0026582141872495413), (24, 0.007589144632220268), (25, 0.014984984882175922), (26, 0.014192919246852398), (27, 0.0002525787567719817)]
+
+ 34%|███▍      | 17/50 [02:50<05:52, 10.68s/it]Time step 640, Spatial MSE: [(0, 7.409497357002692e-06), (1, 9.358780516777188e-06), (2, 0.00010547874990152195), (3, 2.1199453840381466e-05), (4, 1.1765486306103412e-05), (5, 1.0458086762810126e-05), (6, 1.6650565157760866e-05), (7, 2.5998624550993554e-05), (8, 1.6662490452290513e-05), (9, 2.2551786969415843e-05), (10, 2.679473618627526e-05), (11, 3.344539072713815e-05), (12, 4.367305155028589e-05), (13, 3.66244676115457e-05), (14, 5.0584923883434385e-05), (15, 8.703031198820099e-05), (16, 7.871634443290532e-05), (17, 8.82290696608834e-05), (18, 0.0002325374080101028), (19, 0.0001041289433487691), (20, 0.00015549242380075157), (21, 0.000455158413387835), (22, 0.0001660828129388392), (23, 0.0004032413999084383), (24, 0.0009155262960121036), (25, 0.0020226698834449053), (26, 0.0029928337316960096), (27, 0.006427528336644173)], Temporal MSE: [(0, 8.678425729158334e-06), (1, 2.9918273867224343e-06), (2, 5.651783794746734e-06), (3, 2.628466063470114e-06), (4, 3.968490091210697e-06), (5, 6.684751951979706e-06), (6, 7.320129043364432e-06), (7, 1.3594923075288534e-05), (8, 8.149555469572078e-06), (9, 1.4357555301103275e-05), (10, 1.74146807694342e-05), (11, 2.5683568310341798e-05), (12, 2.7670961571857333e-05), (13, 3.705078052007593e-05), (14, 3.5640012356452644e-05), (15, 5.639509981847368e-05), (16, 4.675594391301274e-05), (17, 3.549184475559741e-05), (18, 0.00014022710092831403), (19, 6.432211375795305e-05), (20, 6.49334688205272e-05), (21, 0.000237139334785752), (22, 5.794175376649946e-05), (23, 0.00016409972158726305), (24, 0.0002535149978939444), (25, 0.0006181591306813061), (26, 0.00033171521499753), (27, 4.593700941768475e-05)]
+
+ 36%|███▌      | 18/50 [03:01<05:39, 10.62s/it]Time step 620, Spatial MSE: [(0, 2.9913868274888955e-05), (1, 2.4788620066829026e-05), (2, 0.00017126626335084438), (3, 0.00011674274719553068), (4, 0.00039089962956495583), (5, 0.0007887854008004069), (6, 0.0018634767038747668), (7, 0.004181021824479103), (8, 0.006343925837427378), (9, 0.009446106851100922), (10, 0.011257734149694443), (11, 0.012256931513547897), (12, 0.014418487437069416), (13, 0.021297290921211243), (14, 0.026646215468645096), (15, 0.04207811877131462), (16, 0.024652255699038506), (17, 0.025915602222085), (18, 0.029055170714855194), (19, 0.02173362486064434), (20, 0.025663722306489944), (21, 0.028990043327212334), (22, 0.022422824054956436), (23, 0.028377125039696693), (24, 0.06061400845646858), (25, 0.09508858621120453), (26, 0.04099433124065399), (27, 0.008581845089793205)], Temporal MSE: [(0, 1.778544537955895e-05), (1, 8.11198515293654e-06), (2, 3.994517828687094e-05), (3, 0.00025428729713894427), (4, 0.000388323882361874), (5, 0.0007290199282579124), (6, 0.0020416947081685066), (7, 0.0027993214316666126), (8, 0.0023224486503750086), (9, 0.005595272406935692), (10, 0.006960362195968628), (11, 0.007996123284101486), (12, 0.009528132155537605), (13, 0.014607559889554977), (14, 0.022306114435195923), (15, 0.01447338331490755), (16, 0.012449284084141254), (17, 0.013294761069118977), (18, 0.011988207697868347), (19, 0.01285878662019968), (20, 0.029709752649068832), (21, 0.020197974517941475), (22, 0.012194654904305935), (23, 0.014915121719241142), (24, 0.04037659615278244), (25, 0.04641294479370117), (26, 0.02158558927476406), (27, 0.0007369440281763673)]
+
+ 38%|███▊      | 19/50 [03:14<05:54, 11.42s/it]Time step 600, Spatial MSE: [(0, 7.441800335072912e-06), (1, 1.2336823601799551e-05), (2, 0.00010893661965383217), (3, 2.6054483896587044e-05), (4, 1.1716031622199807e-05), (5, 1.0774273505376186e-05), (6, 1.6175741620827466e-05), (7, 3.5772620321949944e-05), (8, 1.7453487089369446e-05), (9, 2.4966724595287815e-05), (10, 2.8709593607345596e-05), (11, 3.7588051782222465e-05), (12, 4.8948095354717225e-05), (13, 4.077118865097873e-05), (14, 5.322799916029908e-05), (15, 9.510570816928521e-05), (16, 7.821308099664748e-05), (17, 9.299108933191746e-05), (18, 0.00022407580399885774), (19, 9.806470916373655e-05), (20, 0.00014578427362721413), (21, 0.0004212625790387392), (22, 0.00016691589553374797), (23, 0.0004278048872947693), (24, 0.0008907188894227147), (25, 0.0020365482196211815), (26, 0.003623158670961857), (27, 0.007127504330128431)], Temporal MSE: [(0, 7.2926372922665905e-06), (1, 2.628409674798604e-06), (2, 5.6586959544802085e-06), (3, 2.6281888949597487e-06), (4, 4.263258233550005e-06), (5, 7.022407316981116e-06), (6, 9.105500794248655e-06), (7, 1.4488454326055944e-05), (8, 9.08688161871396e-06), (9, 1.398193580826046e-05), (10, 1.90354221558664e-05), (11, 2.5658036975073628e-05), (12, 2.857622894225642e-05), (13, 4.359525701147504e-05), (14, 3.9075486711226404e-05), (15, 5.980822243145667e-05), (16, 4.8105077439686283e-05), (17, 4.275858009350486e-05), (18, 0.00011441163223935291), (19, 6.368014874169603e-05), (20, 6.416962423827499e-05), (21, 0.00023458496434614062), (22, 5.68701907468494e-05), (23, 0.0001599338575033471), (24, 0.00024560748715884984), (25, 0.0005970082711428404), (26, 0.00039660706534050405), (27, 4.651727431337349e-05)]
+
+ 40%|████      | 20/50 [03:26<05:47, 11.57s/it]Time step 580, Spatial MSE: [(0, 3.6347169952932745e-05), (1, 1.9730081476154737e-05), (2, 0.00018425016605760902), (3, 4.6441491576842964e-05), (4, 5.9714147937484086e-05), (5, 9.372478962177411e-05), (6, 0.0002007208822760731), (7, 0.0003200093633495271), (8, 0.000466052966658026), (9, 0.0005716147716157138), (10, 0.0009779081447049975), (11, 0.0011578022968024015), (12, 0.001194765791296959), (13, 0.001257074996829033), (14, 0.0015431386418640614), (15, 0.00133243587333709), (16, 0.000829999684356153), (17, 0.0007965436670929193), (18, 0.001350667793303728), (19, 0.0008637409191578627), (20, 0.0013951805885881186), (21, 0.0011466232826933265), (22, 0.0011795238824561238), (23, 0.0015421275747939944), (24, 0.005460383836179972), (25, 0.008385982364416122), (26, 0.006564249750226736), (27, 0.009820261038839817)], Temporal MSE: [(0, 7.199434094218304e-06), (1, 4.215327862766571e-06), (2, 2.1821142581757158e-05), (3, 4.073102536494844e-05), (4, 4.52704043709673e-05), (5, 7.178534724516794e-05), (6, 0.00013150212180335075), (7, 0.00010443543578730896), (8, 0.0001187529051094316), (9, 0.0002257113519590348), (10, 0.00032439580536447465), (11, 0.00040851192898117006), (12, 0.0004713019006885588), (13, 0.0007480150670744479), (14, 0.0006225731922313571), (15, 0.0005233696429058909), (16, 0.00036303361412137747), (17, 0.00036978302523493767), (18, 0.0004175215435680002), (19, 0.000325247849104926), (20, 0.0007280579302459955), (21, 0.0008102440042421222), (22, 0.0006644357927143574), (23, 0.0010241607669740915), (24, 0.0030145663768053055), (25, 0.008209803141653538), (26, 0.0023347942624241114), (27, 0.0002123216981999576)]
+
+ 42%|████▏     | 21/50 [03:38<05:42, 11.83s/it]Time step 560, Spatial MSE: [(0, 6.90676597514539e-06), (1, 2.186395795433782e-05), (2, 9.068221697816625e-05), (3, 5.352700100047514e-05), (4, 4.601657565217465e-05), (5, 7.687124161748216e-05), (6, 0.0002039248211076483), (7, 0.0003090144891757518), (8, 0.00023587590840179473), (9, 0.00030210663680918515), (10, 0.0004750642692670226), (11, 0.0005392049206420779), (12, 0.0007359477458521724), (13, 0.0010684479493647814), (14, 0.0017948901513591409), (15, 0.00718662329018116), (16, 0.01658417098224163), (17, 0.0013287606416270137), (18, 0.0014161554863676429), (19, 0.0010857152519747615), (20, 0.0018000300042331219), (21, 0.0030875701922923326), (22, 0.0013322093291208148), (23, 0.0012880860595032573), (24, 0.002792269689962268), (25, 0.008446209132671356), (26, 0.003091593272984028), (27, 0.007431974168866873)], Temporal MSE: [(0, 7.971483682922553e-06), (1, 4.767038717545802e-06), (2, 2.583898458397016e-05), (3, 4.766723941429518e-05), (4, 8.454622002318501e-05), (5, 8.120371057884768e-05), (6, 0.0002302406937815249), (7, 0.00017947029846254736), (8, 0.00015748974692542106), (9, 0.0004687166365329176), (10, 0.0007813922129571438), (11, 0.0010241377167403698), (12, 0.001615811255760491), (13, 0.00290888175368309), (14, 0.005543071310967207), (15, 0.0015897840494289994), (16, 0.0007560034282505512), (17, 0.0006503094919025898), (18, 0.000663811108097434), (19, 0.0005140868597663939), (20, 0.000989444088190794), (21, 0.0009179351036436856), (22, 0.0010170632740482688), (23, 0.0010214465437456965), (24, 0.0029951760079711676), (25, 0.004969118162989616), (26, 0.0037994305603206158), (27, 0.00019897488527931273)]
+
+ 44%|████▍     | 22/50 [03:49<05:16, 11.31s/it]Time step 540, Spatial MSE: [(0, 3.40219667123165e-05), (1, 2.1128824300831184e-05), (2, 0.00018402605201117694), (3, 8.80367515492253e-05), (4, 0.00011579489364521578), (5, 0.00016673741629347205), (6, 0.00037534325383603573), (7, 0.0008550143684260547), (8, 0.0007139937952160835), (9, 0.0010204442078247666), (10, 0.0012045415351167321), (11, 0.0014271634863689542), (12, 0.0020824058447033167), (13, 0.0026124112773686647), (14, 0.00544478977099061), (15, 0.00495958561077714), (16, 0.018756337463855743), (17, 0.002924425760284066), (18, 0.003689892590045929), (19, 0.0029682687018066645), (20, 0.003411318175494671), (21, 0.004592412151396275), (22, 0.003164646215736866), (23, 0.0035707401111721992), (24, 0.009326227940618992), (25, 0.015733782202005386), (26, 0.011200256645679474), (27, 0.009031602181494236)], Temporal MSE: [(0, 6.231784027477261e-06), (1, 3.874265985359671e-06), (2, 1.5596153389196843e-05), (3, 5.4797234042780474e-05), (4, 0.00010570001904852688), (5, 0.00014276184083428234), (6, 0.00031630753073841333), (7, 0.0002833427570294589), (8, 0.0002678087039384991), (9, 0.0005403965478762984), (10, 0.0006932676187716424), (11, 0.0010348308132961392), (12, 0.0011265729553997517), (13, 0.002541555790230632), (14, 0.002778017660602927), (15, 0.0019856083672493696), (16, 0.0016740465071052313), (17, 0.001356478314846754), (18, 0.0014436757192015648), (19, 0.0012212509755045176), (20, 0.0023301206529140472), (21, 0.0022783083841204643), (22, 0.0016755437245592475), (23, 0.0022858718875795603), (24, 0.005874296184629202), (25, 0.011181863024830818), (26, 0.009076938033103943), (27, 0.0003559687174856663)]
+
+ 46%|████▌     | 23/50 [04:02<05:22, 11.96s/it]Time step 520, Spatial MSE: [(0, 6.975682026677532e-06), (1, 1.5100163182069082e-05), (2, 8.483960846206173e-05), (3, 2.728309846133925e-05), (4, 1.2065796909155324e-05), (5, 1.073448765964713e-05), (6, 1.3739319911110215e-05), (7, 4.4577045628102496e-05), (8, 1.5971791071933694e-05), (9, 2.006005342991557e-05), (10, 2.27549080591416e-05), (11, 2.572826633695513e-05), (12, 3.142519562970847e-05), (13, 3.252580791013315e-05), (14, 4.2240651964675635e-05), (15, 8.031564357224852e-05), (16, 0.0013813191326335073), (17, 6.045137706678361e-05), (18, 0.00015853256627451628), (19, 0.00011883630941156298), (20, 0.00010754918912425637), (21, 0.0003215619653929025), (22, 0.00013759701687376946), (23, 0.00029900766094215214), (24, 0.0005805297987535596), (25, 0.0015806431183591485), (26, 0.00368762225843966), (27, 0.00722423754632473)], Temporal MSE: [(0, 5.390105798142031e-06), (1, 2.141018740076106e-06), (2, 4.407394953886978e-06), (3, 2.878424083974096e-06), (4, 3.6593960430764128e-06), (5, 6.0857546486658975e-06), (6, 8.129347406793386e-06), (7, 1.4745882253919262e-05), (8, 7.604669917782303e-06), (9, 1.2646988579945173e-05), (10, 1.586059806868434e-05), (11, 1.90141327038873e-05), (12, 2.0314699213486165e-05), (13, 3.147836832795292e-05), (14, 6.889711221447214e-05), (15, 3.445371476118453e-05), (16, 9.632081491872668e-05), (17, 4.4954842451261356e-05), (18, 6.475186091847718e-05), (19, 5.406318814493716e-05), (20, 0.0001118185100494884), (21, 0.0001280844007851556), (22, 6.0586800827877596e-05), (23, 0.00010686145105864853), (24, 0.00019978368072770536), (25, 0.00043155995081178844), (26, 0.0004358977312222123), (27, 5.15963947691489e-05)]
+
+ 48%|████▊     | 24/50 [04:19<05:46, 13.34s/it]Time step 500, Spatial MSE: [(0, 3.2814990845508873e-05), (1, 2.7690199203789234e-05), (2, 0.00013634056085720658), (3, 0.00010065941023640335), (4, 0.00026361463824287057), (5, 0.000659669516608119), (6, 0.0019966636318713427), (7, 0.003975021652877331), (8, 0.006409439258277416), (9, 0.00857620406895876), (10, 0.010102879256010056), (11, 0.011340923607349396), (12, 0.013292254880070686), (13, 0.01651463471353054), (14, 0.025123469531536102), (15, 0.037878308445215225), (16, 0.024138376116752625), (17, 0.02395225316286087), (18, 0.0276893749833107), (19, 0.021198969334363937), (20, 0.025857092812657356), (21, 0.027991512790322304), (22, 0.02087344601750374), (23, 0.028135815635323524), (24, 0.06335414946079254), (25, 0.08246345818042755), (26, 0.04882189258933067), (27, 0.0066739944741129875)], Temporal MSE: [(0, 6.9971929406165145e-06), (1, 5.154175141797168e-06), (2, 3.126496812910773e-05), (3, 0.00017407859559170902), (4, 0.000307991576846689), (5, 0.000566945702303201), (6, 0.0020316019654273987), (7, 0.0025912891142070293), (8, 0.002546048955991864), (9, 0.006114421412348747), (10, 0.006846604403108358), (11, 0.0073326206766068935), (12, 0.008743165992200375), (13, 0.014750470407307148), (14, 0.020995208993554115), (15, 0.013629002496600151), (16, 0.011543038301169872), (17, 0.01223703846335411), (18, 0.011311452835798264), (19, 0.011282087303698063), (20, 0.022849373519420624), (21, 0.016637304797768593), (22, 0.01237183716148138), (23, 0.015621930360794067), (24, 0.03743220865726471), (25, 0.04688016697764397), (26, 0.020580628886818886), (27, 0.0008929733885452151)]
+
+ 50%|█████     | 25/50 [04:31<05:27, 13.12s/it]Time step 480, Spatial MSE: [(0, 6.789564849896124e-06), (1, 1.6305733879562467e-05), (2, 7.400012691505253e-05), (3, 2.9664697649423033e-05), (4, 1.2452532246243209e-05), (5, 1.1445514246588573e-05), (6, 1.2716762284981087e-05), (7, 4.678385448642075e-05), (8, 1.5166766388574615e-05), (9, 1.9232478734920733e-05), (10, 2.1562080291914754e-05), (11, 2.5236931833205745e-05), (12, 2.961940481327474e-05), (13, 3.437432678765617e-05), (14, 5.052377673564479e-05), (15, 6.315820792224258e-05), (16, 0.0020184917375445366), (17, 4.7594156058039516e-05), (18, 0.00018058664863929152), (19, 0.0001124721675296314), (20, 8.225783676607534e-05), (21, 0.000309375231154263), (22, 0.00016439124010503292), (23, 0.0002510235644876957), (24, 0.0006092449184507132), (25, 0.001278672250919044), (26, 0.00380181847140193), (27, 0.007510199211537838)], Temporal MSE: [(0, 4.635956884158077e-06), (1, 2.043088443315355e-06), (2, 4.0685645217308775e-06), (3, 3.261006895627361e-06), (4, 3.643881200332544e-06), (5, 6.3496891016256995e-06), (6, 9.778230378287844e-06), (7, 1.4692176591779571e-05), (8, 7.508422640967183e-06), (9, 1.2408817383402493e-05), (10, 1.6383644833695143e-05), (11, 1.891968167910818e-05), (12, 1.904781129269395e-05), (13, 3.282842590124346e-05), (14, 7.696246029809117e-05), (15, 3.368910984136164e-05), (16, 2.9348722819122486e-05), (17, 2.1360589016694576e-05), (18, 4.7090768930502236e-05), (19, 3.761710104299709e-05), (20, 5.668031735694967e-05), (21, 0.00011948862083954737), (22, 7.133011968107894e-05), (23, 0.00013755641703028232), (24, 0.0002342849038541317), (25, 0.0004079919308423996), (26, 0.000497195404022932), (27, 7.226439629448578e-05)]
+
+ 52%|█████▏    | 26/50 [04:41<04:49, 12.04s/it]Time step 460, Spatial MSE: [(0, 3.9079961425159127e-05), (1, 2.169933395634871e-05), (2, 0.00011936161172343418), (3, 4.9681824748404324e-05), (4, 4.895000529359095e-05), (5, 7.51743937144056e-05), (6, 0.00020845870312768966), (7, 0.00023024519032333046), (8, 0.0004975335905328393), (9, 0.0005828023422509432), (10, 0.0009043279569596052), (11, 0.001043002586811781), (12, 0.0011095775989815593), (13, 0.0012798430398106575), (14, 0.0015791497426107526), (15, 0.0012149048270657659), (16, 0.0028347740881145), (17, 0.0007353675318881869), (18, 0.00118713837582618), (19, 0.0008062840206548572), (20, 0.0011515290243551135), (21, 0.001079054782167077), (22, 0.001067499746568501), (23, 0.0014753054128959775), (24, 0.005078825168311596), (25, 0.01157995592802763), (26, 0.014037628658115864), (27, 0.007842793129384518)], Temporal MSE: [(0, 4.949758022121387e-06), (1, 3.3108858588093426e-06), (2, 1.9716684619197622e-05), (3, 4.5960368879605085e-05), (4, 5.353157757781446e-05), (5, 7.289089990081266e-05), (6, 0.00017812108853831887), (7, 9.700057125883177e-05), (8, 0.00012184428487671539), (9, 0.00019132497254759073), (10, 0.00031445585773326457), (11, 0.00040081399492919445), (12, 0.00043735545477829874), (13, 0.0007046862738206983), (14, 0.0006655246834270656), (15, 0.0004754421825055033), (16, 0.0003712533216457814), (17, 0.00035606857272796333), (18, 0.0004180195974186063), (19, 0.0003222395898774266), (20, 0.0007198132225312293), (21, 0.0007153492188081145), (22, 0.0006922371685504913), (23, 0.0009736085776239634), (24, 0.002615283941850066), (25, 0.010317429900169373), (26, 0.0030442324932664633), (27, 0.00035987928276881576)]
+
+ 54%|█████▍    | 27/50 [04:51<04:28, 11.66s/it]Time step 440, Spatial MSE: [(0, 6.236276021809317e-06), (1, 2.0887908249278553e-05), (2, 5.2766808948945254e-05), (3, 3.8672631490044296e-05), (4, 4.170299507677555e-05), (5, 7.911397551652044e-05), (6, 0.0001912492443807423), (7, 0.00037878425791859627), (8, 0.00022391100355889648), (9, 0.000300737505313009), (10, 0.00042874313658103347), (11, 0.0005061919218860567), (12, 0.0006463183090090752), (13, 0.0009064889745786786), (14, 0.0014151337090879679), (15, 0.007522170897573233), (16, 0.007131307385861874), (17, 0.0011371202999725938), (18, 0.0011620023287832737), (19, 0.0007733829552307725), (20, 0.0014295156579464674), (21, 0.0021439203992486), (22, 0.0009571645059622824), (23, 0.0009263672400265932), (24, 0.002216658554971218), (25, 0.00439669843763113), (26, 0.0033782157115638256), (27, 0.006252101622521877)], Temporal MSE: [(0, 4.440289103513351e-06), (1, 3.1830584248382365e-06), (2, 1.985827475436963e-05), (3, 5.54219659534283e-05), (4, 9.908328502206132e-05), (5, 8.328083640662953e-05), (6, 0.00024137066793628037), (7, 0.00017102966376114637), (8, 0.00014695173013024032), (9, 0.00044968060683459044), (10, 0.0006721848621964455), (11, 0.000899814476724714), (12, 0.0012528011575341225), (13, 0.002417505718767643), (14, 0.00553834717720747), (15, 0.0014322246424853802), (16, 0.0006758590461686254), (17, 0.0005957874818705022), (18, 0.0005631287349388003), (19, 0.0004920450155623257), (20, 0.0009802111890166998), (21, 0.0007670500199310482), (22, 0.0007834690622985363), (23, 0.0008067979943007231), (24, 0.0024404118303209543), (25, 0.004459267482161522), (26, 0.0043082404881715775), (27, 0.0003878094721585512)]
+
+ 56%|█████▌    | 28/50 [05:05<04:31, 12.36s/it]Time step 420, Spatial MSE: [(0, 3.2538679079152644e-05), (1, 2.245785617560614e-05), (2, 0.00010495140304556116), (3, 6.747295265085995e-05), (4, 9.029454668052495e-05), (5, 0.00016077718464657664), (6, 0.00034091429552063346), (7, 0.0007690798956900835), (8, 0.0006192710716277361), (9, 0.0009422539151273668), (10, 0.0010663459543138742), (11, 0.0013639404205605388), (12, 0.0019563797395676374), (13, 0.0025287268217653036), (14, 0.004816343076527119), (15, 0.005036979913711548), (16, 0.009012822061777115), (17, 0.002727391431108117), (18, 0.0032598190009593964), (19, 0.0026283515617251396), (20, 0.0032339836470782757), (21, 0.00365423783659935), (22, 0.0029032626189291477), (23, 0.003930421080440283), (24, 0.010473825968801975), (25, 0.018725117668509483), (26, 0.02357027865946293), (27, 0.011008750647306442)], Temporal MSE: [(0, 4.349652044766117e-06), (1, 2.982257910844055e-06), (2, 1.621439696464222e-05), (3, 4.925812390865758e-05), (4, 0.00010064304660772905), (5, 0.00017003367247525603), (6, 0.0003181312931701541), (7, 0.00029382214415818453), (8, 0.0002662966726347804), (9, 0.0004810437385458499), (10, 0.0007076200563460588), (11, 0.0011085063451901078), (12, 0.001029903069138527), (13, 0.002003867644816637), (14, 0.0021577824372798204), (15, 0.0018297601491212845), (16, 0.001459217513911426), (17, 0.0012992351548746228), (18, 0.0012981402687728405), (19, 0.0011283555068075657), (20, 0.002376839518547058), (21, 0.0030036817770451307), (22, 0.0017469655722379684), (23, 0.0021961515303701162), (24, 0.00596274621784687), (25, 0.014375803992152214), (26, 0.007599852047860622), (27, 0.0005758705083280802)]
+
+ 58%|█████▊    | 29/50 [05:16<04:08, 11.82s/it]Time step 400, Spatial MSE: [(0, 6.75614182910067e-06), (1, 1.6977965060505085e-05), (2, 4.659842306864448e-05), (3, 3.074439518968575e-05), (4, 1.2296155546209775e-05), (5, 1.2543056072900072e-05), (6, 1.2728302863251884e-05), (7, 5.313930523698218e-05), (8, 1.6021602277760394e-05), (9, 1.731759766698815e-05), (10, 2.116038740496151e-05), (11, 2.1118987206136808e-05), (12, 2.7089514333056286e-05), (13, 3.684423063532449e-05), (14, 3.1616400519851595e-05), (15, 4.2159270378760993e-05), (16, 5.6214164942502975e-05), (17, 4.0418675780529156e-05), (18, 9.170056728180498e-05), (19, 7.486152753699571e-05), (20, 6.014279279042967e-05), (21, 0.00012989630340598524), (22, 0.00011356195318512619), (23, 0.00022305325546767563), (24, 0.00045996278640814126), (25, 0.0011133203515782952), (26, 0.003653932362794876), (27, 0.008264432661235332)], Temporal MSE: [(0, 3.889576419169316e-06), (1, 1.9903916381736053e-06), (2, 3.6958310829504626e-06), (3, 3.871289663948119e-06), (4, 3.5596412999439053e-06), (5, 6.134721388662001e-06), (6, 9.671788575360551e-06), (7, 1.5212289326882455e-05), (8, 7.548467237938894e-06), (9, 1.4658215150120668e-05), (10, 1.5865329260122962e-05), (11, 1.7897216821438633e-05), (12, 1.9703133148141205e-05), (13, 3.482778265606612e-05), (14, 2.2968766643316485e-05), (15, 2.5654153432697058e-05), (16, 1.980322485906072e-05), (17, 1.782119579729624e-05), (18, 1.9305320165585726e-05), (19, 2.7809737730422057e-05), (20, 3.444270259933546e-05), (21, 6.779401155654341e-05), (22, 6.694631883874536e-05), (23, 9.56550138653256e-05), (24, 0.00018265782273374498), (25, 0.00035394166479818523), (26, 0.0005191012751311064), (27, 0.0001122474714065902)]
+
+ 60%|██████    | 30/50 [05:25<03:41, 11.09s/it]Time step 380, Spatial MSE: [(0, 3.0338291253428906e-05), (1, 2.4198208848247305e-05), (2, 6.916937854839489e-05), (3, 7.55297442083247e-05), (4, 0.00020613466040231287), (5, 0.0004842461785301566), (6, 0.001695588929578662), (7, 0.004432994406670332), (8, 0.006347752641886473), (9, 0.008466505445539951), (10, 0.008909343741834164), (11, 0.009873239323496819), (12, 0.011619864962995052), (13, 0.015319460071623325), (14, 0.025461416691541672), (15, 0.042012035846710205), (16, 0.03058457374572754), (17, 0.02918297052383423), (18, 0.03795890510082245), (19, 0.029196785762906075), (20, 0.03675578534603119), (21, 0.03547696769237518), (22, 0.03015451692044735), (23, 0.04316294193267822), (24, 0.0935567319393158), (25, 0.14746874570846558), (26, 0.08287427574396133), (27, 0.006403953768312931)], Temporal MSE: [(0, 4.222153620503377e-06), (1, 3.2270872907247394e-06), (2, 2.4545282940380275e-05), (3, 0.00018101153546012938), (4, 0.0002488518657628447), (5, 0.0004610063915606588), (6, 0.0015017897821962833), (7, 0.0022390803787857294), (8, 0.002510205376893282), (9, 0.005929604172706604), (10, 0.007217691279947758), (11, 0.007072835229337215), (12, 0.009298931807279587), (13, 0.015106171369552612), (14, 0.02405666932463646), (15, 0.016214964911341667), (16, 0.015109931118786335), (17, 0.016774877905845642), (18, 0.017705174162983894), (19, 0.017203517258167267), (20, 0.033959947526454926), (21, 0.0235900841653347), (22, 0.019191036000847816), (23, 0.02521059662103653), (24, 0.054643288254737854), (25, 0.08523431420326233), (26, 0.03452650457620621), (27, 0.0019189276499673724)]
+
+ 62%|██████▏   | 31/50 [05:37<03:36, 11.40s/it]Time step 360, Spatial MSE: [(0, 6.795432454964612e-06), (1, 1.7753738575265743e-05), (2, 3.812443173956126e-05), (3, 3.3522061130497605e-05), (4, 1.2489798791648354e-05), (5, 1.4199563338479493e-05), (6, 1.2904081813758239e-05), (7, 4.991694731870666e-05), (8, 1.612732194189448e-05), (9, 1.72883828781778e-05), (10, 2.1117491996847093e-05), (11, 2.0982464775443077e-05), (12, 2.6312489353585988e-05), (13, 4.5252872951095924e-05), (14, 3.0742132366867736e-05), (15, 3.678300708997995e-05), (16, 7.718158303759992e-05), (17, 3.466442649369128e-05), (18, 0.00010413624113425612), (19, 8.016813080757856e-05), (20, 6.27596746198833e-05), (21, 0.00012538480223156512), (22, 0.00012781837722286582), (23, 0.00023264119226951152), (24, 0.0004991082241758704), (25, 0.0010878914035856724), (26, 0.0033396685030311346), (27, 0.009043979458510876)], Temporal MSE: [(0, 3.62671198672615e-06), (1, 2.0367147044453304e-06), (2, 3.894756446243264e-06), (3, 4.586816885421285e-06), (4, 4.115115189051721e-06), (5, 7.079147962940624e-06), (6, 1.1230333257117309e-05), (7, 1.4229490261641331e-05), (8, 8.285274816444144e-06), (9, 1.5924242688925005e-05), (10, 1.6272084394586273e-05), (11, 1.7883650798466988e-05), (12, 2.018074519583024e-05), (13, 4.019026891910471e-05), (14, 2.2192247342900373e-05), (15, 2.1930278307991102e-05), (16, 2.0062996554770507e-05), (17, 1.72993095475249e-05), (18, 2.0245883206371218e-05), (19, 2.7524190954864025e-05), (20, 4.180934774922207e-05), (21, 7.356912101386115e-05), (22, 8.760325727052987e-05), (23, 0.00011876476492034271), (24, 0.00021320967061910778), (25, 0.000410685344832018), (26, 0.0005927989259362221), (27, 0.0001408544776495546)]
+
+ 64%|██████▍   | 32/50 [05:50<03:33, 11.85s/it]Time step 340, Spatial MSE: [(0, 3.188548726029694e-05), (1, 2.053058778983541e-05), (2, 3.957488661399111e-05), (3, 5.207854701438919e-05), (4, 5.092264473205432e-05), (5, 7.20770622137934e-05), (6, 0.000263589812675491), (7, 0.00021482471493072808), (8, 0.0005448553711175919), (9, 0.000636218290310353), (10, 0.0009110954706557095), (11, 0.0009269635775126517), (12, 0.0010859905742108822), (13, 0.0014131837524473667), (14, 0.0025910299737006426), (15, 0.0015419787960126996), (16, 0.020597824826836586), (17, 0.0008406455744989216), (18, 0.0011378847993910313), (19, 0.001149623654782772), (20, 0.0015066539635881782), (21, 0.001846596598625183), (22, 0.0011566993780434132), (23, 0.0017885935958474874), (24, 0.005761713720858097), (25, 0.018667148426175117), (26, 0.025954827666282654), (27, 0.007562965154647827)], Temporal MSE: [(0, 4.197826910967706e-06), (1, 2.7532669264473952e-06), (2, 1.9182694813935086e-05), (3, 5.027280349167995e-05), (4, 6.19435086264275e-05), (5, 8.162692392943427e-05), (6, 0.00021958371507935226), (7, 0.00011260819155722857), (8, 0.00014829059364274144), (9, 0.00020462959946598858), (10, 0.0003445780312176794), (11, 0.0004317811399232596), (12, 0.000462288357084617), (13, 0.0008117721299640834), (14, 0.0009924534242600203), (15, 0.000590387440752238), (16, 0.0005809917929582298), (17, 0.0004837046726606786), (18, 0.0005414469633251429), (19, 0.0004883040091954172), (20, 0.0010468343971297145), (21, 0.0009855660609900951), (22, 0.0008879097877070308), (23, 0.0011019391240552068), (24, 0.002842325484380126), (25, 0.011918023228645325), (26, 0.003767923451960087), (27, 0.0005576129187829792)]
+
+ 66%|██████▌   | 33/50 [06:01<03:13, 11.37s/it]Time step 320, Spatial MSE: [(0, 6.809489605075214e-06), (1, 2.0600538846338168e-05), (2, 2.5931592972483486e-05), (3, 3.2989253668347374e-05), (4, 4.3901221943087876e-05), (5, 9.956509165931493e-05), (6, 0.0002038506354438141), (7, 0.0004418307507876307), (8, 0.00023607496405020356), (9, 0.00029698555590584874), (10, 0.00042742874938994646), (11, 0.0004953521420247853), (12, 0.0005903499550186098), (13, 0.0011079235700890422), (14, 0.003003300866112113), (15, 0.007195190526545048), (16, 0.058562133461236954), (17, 0.0011479469249024987), (18, 0.00125544099137187), (19, 0.0019648363813757896), (20, 0.002106209984049201), (21, 0.002548040123656392), (22, 0.0011553692165762186), (23, 0.0012139955069869757), (24, 0.0028402165044099092), (25, 0.004752462729811668), (26, 0.00398592371493578), (27, 0.011567842215299606)], Temporal MSE: [(0, 3.839951205009129e-06), (1, 2.6110271846846445e-06), (2, 1.7126696548075415e-05), (3, 6.607613613596186e-05), (4, 0.00011942756827920675), (5, 9.57536103669554e-05), (6, 0.0002914189826697111), (7, 0.00018831872148439288), (8, 0.00016274448717013001), (9, 0.0004454708250705153), (10, 0.0007189647876657546), (11, 0.0008472166955471039), (12, 0.0011721415212377906), (13, 0.0033559866715222597), (14, 0.006682167295366526), (15, 0.0013882382772862911), (16, 0.0009668956627137959), (17, 0.0006882521556690335), (18, 0.0006762364646419883), (19, 0.0006184933590702713), (20, 0.0016501593636348844), (21, 0.0012910421937704086), (22, 0.001075803767889738), (23, 0.0014404141111299396), (24, 0.0032413294538855553), (25, 0.004684319254010916), (26, 0.004922906868159771), (27, 0.0010776014532893896)]
+
+ 68%|██████▊   | 34/50 [06:11<02:59, 11.19s/it]Time step 300, Spatial MSE: [(0, 2.5490104235359468e-05), (1, 2.002106157306116e-05), (2, 2.7919853891944513e-05), (3, 5.562263322644867e-05), (4, 8.258262096205726e-05), (5, 0.00017863005632534623), (6, 0.0003121057525277138), (7, 0.0007990588201209903), (8, 0.000626817112788558), (9, 0.0009478374850004911), (10, 0.001091612153686583), (11, 0.0012970541138201952), (12, 0.0019290511263534427), (13, 0.002406628103926778), (14, 0.006515805143862963), (15, 0.005693326238542795), (16, 0.06525487452745438), (17, 0.0029312209226191044), (18, 0.0034503284841775894), (19, 0.0039766463451087475), (20, 0.004065371584147215), (21, 0.00459674559533596), (22, 0.003305265912786126), (23, 0.004548557102680206), (24, 0.014475997537374496), (25, 0.024360476061701775), (26, 0.03778829798102379), (27, 0.018060151487588882)], Temporal MSE: [(0, 4.537185304798186e-06), (1, 2.6896295821643434e-06), (2, 1.5604397049173713e-05), (3, 4.569235534290783e-05), (4, 0.0001001639393507503), (5, 0.00019583827815949917), (6, 0.0003158011822961271), (7, 0.0003104821953456849), (8, 0.00029400771018117666), (9, 0.0005367490812204778), (10, 0.0008370664436370134), (11, 0.0012553759152069688), (12, 0.0010704249143600464), (13, 0.0024991100654006004), (14, 0.0036577379796653986), (15, 0.002082796534523368), (16, 0.002176494337618351), (17, 0.0015806560404598713), (18, 0.0016273455694317818), (19, 0.0015103603946045041), (20, 0.003978334832936525), (21, 0.0032543104607611895), (22, 0.0021420831326395273), (23, 0.0030292822048068047), (24, 0.008219059556722641), (25, 0.01803823746740818), (26, 0.007023102603852749), (27, 0.001023558434098959)]
+
+ 70%|███████   | 35/50 [06:22<02:45, 11.05s/it]Time step 280, Spatial MSE: [(0, 7.403253675875021e-06), (1, 1.846525992732495e-05), (2, 3.12617230520118e-05), (3, 3.451464726822451e-05), (4, 1.340296512353234e-05), (5, 1.945492476806976e-05), (6, 1.582310869707726e-05), (7, 5.623103788821027e-05), (8, 1.9608067304943688e-05), (9, 2.105710154864937e-05), (10, 2.4461876819259487e-05), (11, 2.2834166884422302e-05), (12, 2.9397468097158708e-05), (13, 8.925604197429493e-05), (14, 0.0003268137515988201), (15, 0.00012899590365123004), (16, 0.017732048407197), (17, 5.7527548051439226e-05), (18, 0.0001907467667479068), (19, 0.00028888805536553264), (20, 0.0001255113020306453), (21, 0.00026569495094008744), (22, 0.00025552563602104783), (23, 0.0003381043206900358), (24, 0.0007358830771408975), (25, 0.0014412159798666835), (26, 0.0030139663722366095), (27, 0.010989535599946976)], Temporal MSE: [(0, 4.091225946467603e-06), (1, 2.4301248231495265e-06), (2, 4.487341811909573e-06), (3, 5.266517291602213e-06), (4, 4.901648026134353e-06), (5, 8.224497832998168e-06), (6, 1.3325685358722694e-05), (7, 1.5814282960491255e-05), (8, 1.0938306331809144e-05), (9, 2.0677449356298894e-05), (10, 1.927682569657918e-05), (11, 2.1515614207601175e-05), (12, 2.712237255764194e-05), (13, 9.165883966488764e-05), (14, 0.0002704568614717573), (15, 6.090215902077034e-05), (16, 0.00016310143109876662), (17, 4.891376738669351e-05), (18, 0.00011373790039215237), (19, 8.261469338322058e-05), (20, 0.0003136985469609499), (21, 0.00017116039816755801), (22, 0.00024813530035316944), (23, 0.00042496551759541035), (24, 0.00040637675556354225), (25, 0.0005538872210308909), (26, 0.0006654594908468425), (27, 0.0002838039945345372)]
+
+ 72%|███████▏  | 36/50 [06:35<02:41, 11.51s/it]Time step 260, Spatial MSE: [(0, 2.1309730072971433e-05), (1, 2.2931206331122667e-05), (2, 4.859716136706993e-05), (3, 6.47444830974564e-05), (4, 0.0001666031457716599), (5, 0.0003964809002354741), (6, 0.0013776315608993173), (7, 0.003378218272700906), (8, 0.004088532645255327), (9, 0.005178837105631828), (10, 0.005834029987454414), (11, 0.006824650801718235), (12, 0.008253582753241062), (13, 0.010773037560284138), (14, 0.01941295713186264), (15, 0.03552873060107231), (16, 0.0990331843495369), (17, 0.02397369220852852), (18, 0.03604615852236748), (19, 0.027793284505605698), (20, 0.03714635968208313), (21, 0.034762896597385406), (22, 0.026044728234410286), (23, 0.03995519131422043), (24, 0.08697447180747986), (25, 0.12770479917526245), (26, 0.09559851139783859), (27, 0.010125864297151566)], Temporal MSE: [(0, 5.162988600204699e-06), (1, 3.2124144127010368e-06), (2, 2.135734393959865e-05), (3, 0.0001645078300498426), (4, 0.00023095325741451234), (5, 0.00033428645110689104), (6, 0.0009837502148002386), (7, 0.0015074997209012508), (8, 0.0017734047723934054), (9, 0.003829255700111389), (10, 0.005719863809645176), (11, 0.006490465719252825), (12, 0.007483005523681641), (13, 0.013377289287745953), (14, 0.021574534475803375), (15, 0.015728438273072243), (16, 0.01437746174633503), (17, 0.017402729019522667), (18, 0.019081877544522285), (19, 0.018608639016747475), (20, 0.02932554855942726), (21, 0.025161659345030785), (22, 0.018164092674851418), (23, 0.023713039234280586), (24, 0.049909695982933044), (25, 0.08237466961145401), (26, 0.03595571219921112), (27, 0.002635137876495719)]
+
+ 74%|███████▍  | 37/50 [06:47<02:34, 11.88s/it]Time step 240, Spatial MSE: [(0, 7.832039955246728e-06), (1, 2.0967734599253163e-05), (2, 4.122857717447914e-05), (3, 3.721098255482502e-05), (4, 1.4918534361640923e-05), (5, 2.272763777000364e-05), (6, 1.7398786440026015e-05), (7, 4.377956793177873e-05), (8, 2.2018852178007364e-05), (9, 2.447299448249396e-05), (10, 2.787345329124946e-05), (11, 2.606546513561625e-05), (12, 3.29080612573307e-05), (13, 7.2328737587668e-05), (14, 7.814716082066298e-05), (15, 6.137110176496208e-05), (16, 0.0037215969059616327), (17, 4.9512374971527606e-05), (18, 0.00012543066986836493), (19, 0.00018826175073627383), (20, 9.274336480302736e-05), (21, 0.0002691557165235281), (22, 0.00023085052089300007), (23, 0.0003746200236491859), (24, 0.0007017496973276138), (25, 0.00122031569480896), (26, 0.0028049142565578222), (27, 0.011422820389270782)], Temporal MSE: [(0, 4.913433258479927e-06), (1, 2.913927573899855e-06), (2, 5.183434495847905e-06), (3, 6.407183263945626e-06), (4, 6.639977527811425e-06), (5, 1.059205078490777e-05), (6, 1.4523763638862874e-05), (7, 1.5394301954074763e-05), (8, 1.4367087715072557e-05), (9, 2.5825866032391787e-05), (10, 2.170091465814039e-05), (11, 2.543587288528215e-05), (12, 3.244115214329213e-05), (13, 6.383864092640579e-05), (14, 0.00012809694453608245), (15, 3.890419247909449e-05), (16, 3.978882887167856e-05), (17, 2.9900831577833742e-05), (18, 5.170560325495899e-05), (19, 5.100518319522962e-05), (20, 9.921888704411685e-05), (21, 0.00011291243572486565), (22, 0.00015713753236923367), (23, 0.000266384711721912), (24, 0.00035214697709307075), (25, 0.000573618512134999), (26, 0.0007393299019895494), (27, 0.0003207567788194865)]
+
+ 76%|███████▌  | 38/50 [06:57<02:15, 11.30s/it]Time step 220, Spatial MSE: [(0, 2.0867768398602493e-05), (1, 2.112342008331325e-05), (2, 8.167908526957035e-05), (3, 5.68510549783241e-05), (4, 8.110116323223338e-05), (5, 9.70358814811334e-05), (6, 0.0003987920063082129), (7, 0.00025078229373320937), (8, 0.0005938075482845306), (9, 0.0007085783872753382), (10, 0.0008975894306786358), (11, 0.0009634018642827868), (12, 0.0012272860622033477), (13, 0.0015067990170791745), (14, 0.0021447278559207916), (15, 0.0018202149076387286), (16, 0.004741212353110313), (17, 0.000978143885731697), (18, 0.0015118722803890705), (19, 0.0012604240328073502), (20, 0.0017158902483060956), (21, 0.0018642826471477747), (22, 0.0014071404002606869), (23, 0.002602101769298315), (24, 0.008010443300008774), (25, 0.030454589053988457), (26, 0.03821006044745445), (27, 0.009014817886054516)], Temporal MSE: [(0, 6.415477400878444e-06), (1, 3.4699255593295675e-06), (2, 1.960601548489649e-05), (3, 5.190816955291666e-05), (4, 7.074055611155927e-05), (5, 9.970784594770521e-05), (6, 0.0002834060578607023), (7, 0.00014415250916499645), (8, 0.00019042915664613247), (9, 0.00023930224415380508), (10, 0.00041209126356989145), (11, 0.0005333749577403069), (12, 0.0005385823897086084), (13, 0.0008909187163226306), (14, 0.0010126474080607295), (15, 0.0007114777690730989), (16, 0.0005821228260174394), (17, 0.000563939509447664), (18, 0.0007909229025244713), (19, 0.0006814192747697234), (20, 0.0013703160220757127), (21, 0.0012420282000675797), (22, 0.0009936336427927017), (23, 0.0014453153125941753), (24, 0.003670714795589447), (25, 0.013171453028917313), (26, 0.004744199104607105), (27, 0.0008482381235808134)]
+
+ 78%|███████▊  | 39/50 [07:06<01:55, 10.47s/it]Time step 200, Spatial MSE: [(0, 8.435475137957837e-06), (1, 2.54789374594111e-05), (2, 7.065552199492231e-05), (3, 3.618749542511068e-05), (4, 4.977561911800876e-05), (5, 0.00013306929031386971), (6, 0.00020779165788553655), (7, 0.0005266577936708927), (8, 0.0002698666066862643), (9, 0.00032336206641048193), (10, 0.00045243604108691216), (11, 0.0005486468435265124), (12, 0.000641376303974539), (13, 0.0021744996774941683), (14, 0.004626238718628883), (15, 0.007727174088358879), (16, 0.09039297699928284), (17, 0.0013059629127383232), (18, 0.0016232655616477132), (19, 0.003112367121502757), (20, 0.002412350382655859), (21, 0.003828285960480571), (22, 0.0017267564544454217), (23, 0.0019277377286925912), (24, 0.004215848632156849), (25, 0.006788140162825584), (26, 0.004838532768189907), (27, 0.020554129034280777)], Temporal MSE: [(0, 6.683128503937041e-06), (1, 3.566261284504435e-06), (2, 1.7366674001095816e-05), (3, 7.155778439482674e-05), (4, 0.00014217622810974717), (5, 0.00011836228804895654), (6, 0.0004287301271688193), (7, 0.0002422382967779413), (8, 0.00019929058908019215), (9, 0.00048134929966181517), (10, 0.0007658415124751627), (11, 0.000809701276011765), (12, 0.001181499334052205), (13, 0.004774403292685747), (14, 0.00975696463137865), (15, 0.0015415401430800557), (16, 0.001002165605314076), (17, 0.0007645197329111397), (18, 0.0008073522476479411), (19, 0.000892949930857867), (20, 0.0024718395434319973), (21, 0.0015459596179425716), (22, 0.0013585604028776288), (23, 0.002460531424731016), (24, 0.0048909056931734085), (25, 0.005543692037463188), (26, 0.005899785086512566), (27, 0.002525373362004757)]
+
+ 80%|████████  | 40/50 [07:15<01:39,  9.91s/it]Time step 180, Spatial MSE: [(0, 1.6908830730244517e-05), (1, 2.5099039703491144e-05), (2, 0.0001702916924841702), (3, 5.5331853218376637e-05), (4, 8.797869668342173e-05), (5, 0.0002075411903206259), (6, 0.00029714996344409883), (7, 0.0008018864900805056), (8, 0.0006606526440009475), (9, 0.0009786711307242513), (10, 0.0011603470193222165), (11, 0.0013377773575484753), (12, 0.00197539571672678), (13, 0.002880996558815241), (14, 0.0071321832947432995), (15, 0.006282764952629805), (16, 0.07318287342786789), (17, 0.0030122590251266956), (18, 0.0040108212269842625), (19, 0.0051819635555148125), (20, 0.0051087611354887486), (21, 0.007607453502714634), (22, 0.004957671742886305), (23, 0.007065153680741787), (24, 0.024618996307253838), (25, 0.04698040336370468), (26, 0.05768923461437225), (27, 0.027163812890648842)], Temporal MSE: [(0, 8.383693966607098e-06), (1, 4.3601876313914545e-06), (2, 1.930155121954158e-05), (3, 4.763867036672309e-05), (4, 0.00010739393474068493), (5, 0.0002305999951204285), (6, 0.0003304346464574337), (7, 0.0003425244358368218), (8, 0.00034288890310563147), (9, 0.0005599876749329269), (10, 0.0010356606217101216), (11, 0.0016165978740900755), (12, 0.0012404514709487557), (13, 0.003168463706970215), (14, 0.006342253182083368), (15, 0.002928359666839242), (16, 0.003348599188029766), (17, 0.0018945661140605807), (18, 0.002069490496069193), (19, 0.0019411096582189202), (20, 0.005522498860955238), (21, 0.0047407024540007114), (22, 0.0027386306319385767), (23, 0.004185022786259651), (24, 0.011472414247691631), (25, 0.02324312925338745), (26, 0.008637083694338799), (27, 0.0018915062537416816)]
+
+ 82%|████████▏ | 41/50 [07:26<01:33, 10.37s/it]Time step 160, Spatial MSE: [(0, 9.888779459288344e-06), (1, 3.200276114512235e-05), (2, 0.0001069141726475209), (3, 3.961460242862813e-05), (4, 2.1199084585532546e-05), (5, 3.746517904801294e-05), (6, 3.0112665626802482e-05), (7, 7.82459246693179e-05), (8, 3.5403129004407674e-05), (9, 3.798044417635538e-05), (10, 4.54155306215398e-05), (11, 4.081364386365749e-05), (12, 5.2735871577169746e-05), (13, 0.0001057681511156261), (14, 9.528103691991419e-05), (15, 0.00012638008047360927), (16, 0.001541799632832408), (17, 8.025218266993761e-05), (18, 0.0001302794407820329), (19, 0.0002206419303547591), (20, 0.00014823890523985028), (21, 0.0002554390230216086), (22, 0.00031309391488321126), (23, 0.0006160047487355769), (24, 0.0010154928313568234), (25, 0.0017352142604067922), (26, 0.0028882822953164577), (27, 0.01393281389027834)], Temporal MSE: [(0, 8.557456567359623e-06), (1, 4.54338942290633e-06), (2, 7.689940503041726e-06), (3, 7.5681391535908915e-06), (4, 1.0366271453676745e-05), (5, 1.6180205420823768e-05), (6, 2.1471163563546725e-05), (7, 2.529958146624267e-05), (8, 2.531429345253855e-05), (9, 3.982962516602129e-05), (10, 3.8502610550494865e-05), (11, 4.7000790800666437e-05), (12, 5.3882162319496274e-05), (13, 0.00010070534335682169), (14, 0.0001301709853578359), (15, 7.519967039115727e-05), (16, 0.00021494683460332453), (17, 7.751771045150235e-05), (18, 9.504913759883493e-05), (19, 8.61924490891397e-05), (20, 0.00011255282879574224), (21, 0.00014638216816820204), (22, 0.00020070774189662188), (23, 0.00030735277687199414), (24, 0.00042419147212058306), (25, 0.0007549477159045637), (26, 0.0009196199825964868), (27, 0.000561163411475718)]
+
+ 84%|████████▍ | 42/50 [07:35<01:20, 10.05s/it]Time step 140, Spatial MSE: [(0, 1.5741992683615535e-05), (1, 3.243543687858619e-05), (2, 0.00040114656439982355), (3, 7.416520384140313e-05), (4, 0.00016009742103051394), (5, 0.0004212308849673718), (6, 0.0013213870115578175), (7, 0.002596145262941718), (8, 0.002880841726437211), (9, 0.0036193642299622297), (10, 0.0038473503664135933), (11, 0.004564263857901096), (12, 0.005270117893815041), (13, 0.0068364012986421585), (14, 0.013515911996364594), (15, 0.027293751016259193), (16, 0.11813477426767349), (17, 0.013861200772225857), (18, 0.017108893021941185), (19, 0.016703618690371513), (20, 0.021609365940093994), (21, 0.0201273150742054), (22, 0.016948668286204338), (23, 0.027667824178934097), (24, 0.0684259831905365), (25, 0.09340827912092209), (26, 0.08920955657958984), (27, 0.01392790861427784)], Temporal MSE: [(0, 1.141793927672552e-05), (1, 6.390899216057733e-06), (2, 2.9524871933972463e-05), (3, 0.0001253115915460512), (4, 0.0002334912569494918), (5, 0.00033442131825722754), (6, 0.0009580415789969265), (7, 0.001466805930249393), (8, 0.0014269952662289143), (9, 0.0027994865085929632), (10, 0.004158661235123873), (11, 0.005505159497261047), (12, 0.005205834750086069), (13, 0.009807825088500977), (14, 0.018002796918153763), (15, 0.011347528547048569), (16, 0.009587045758962631), (17, 0.009982610121369362), (18, 0.009574949741363525), (19, 0.009703993797302246), (20, 0.017688363790512085), (21, 0.01604127138853073), (22, 0.011729039251804352), (23, 0.015370877459645271), (24, 0.03831152617931366), (25, 0.06777416169643402), (26, 0.03101290948688984), (27, 0.0033087776973843575)]
+
+ 86%|████████▌ | 43/50 [07:44<01:08,  9.78s/it]Time step 120, Spatial MSE: [(0, 1.2367183444439434e-05), (1, 4.782999894814566e-05), (2, 0.0001952734892256558), (3, 4.995770359528251e-05), (4, 2.837955253198743e-05), (5, 5.9154506743652746e-05), (6, 4.653865471482277e-05), (7, 7.185603317338973e-05), (8, 5.107624747324735e-05), (9, 6.088496229494922e-05), (10, 6.448425847338513e-05), (11, 5.9314046666258946e-05), (12, 7.748642383376136e-05), (13, 0.00015741561946924776), (14, 0.00017505206051282585), (15, 0.0001541951933177188), (16, 0.004045224282890558), (17, 0.00011981398711213842), (18, 0.00020927393052261323), (19, 0.00033589141094125807), (20, 0.00021551159443333745), (21, 0.0006112859700806439), (22, 0.0005060124676674604), (23, 0.0009468077914789319), (24, 0.0014840058283880353), (25, 0.0025101935025304556), (26, 0.0033108973875641823), (27, 0.01644381321966648)], Temporal MSE: [(0, 1.2031703590764664e-05), (1, 6.264945113798603e-06), (2, 1.0803937584569212e-05), (3, 1.0733215276559349e-05), (4, 1.6494353985763155e-05), (5, 2.5269753678003326e-05), (6, 3.07530390273314e-05), (7, 3.574022412067279e-05), (8, 4.1600691474741325e-05), (9, 5.895494905416854e-05), (10, 5.8669997088145465e-05), (11, 7.853787974454463e-05), (12, 8.724935469217598e-05), (13, 0.00016635229985695332), (14, 0.0002342636144021526), (15, 0.00012100880849175155), (16, 9.053592657437548e-05), (17, 7.31706022634171e-05), (18, 0.00012120095925638452), (19, 0.00014551768254023045), (20, 0.00022300089767668396), (21, 0.00030572214745916426), (22, 0.000300967221846804), (23, 0.00048454722855240107), (24, 0.0007202955894172192), (25, 0.0011075622169300914), (26, 0.001344794873148203), (27, 0.0008790789870545268)]
+
+ 88%|████████▊ | 44/50 [07:53<00:55,  9.32s/it]Time step 100, Spatial MSE: [(0, 1.8080407244269736e-05), (1, 5.0723851018119603e-05), (2, 0.0007458837353624403), (3, 7.647177699254826e-05), (4, 0.00015623179206158966), (5, 0.0002491287887096405), (6, 0.0004229460610076785), (7, 0.0007196376682259142), (8, 0.0010105033870786428), (9, 0.0012995372526347637), (10, 0.0012738034129142761), (11, 0.0014435120392590761), (12, 0.0017835801700130105), (13, 0.0025089597329497337), (14, 0.0046312580816447735), (15, 0.010264837183058262), (16, 0.06302960216999054), (17, 0.004475723020732403), (18, 0.005156530067324638), (19, 0.005562900565564632), (20, 0.006539106834679842), (21, 0.0064770812168717384), (22, 0.00647602416574955), (23, 0.011782861314713955), (24, 0.020997216925024986), (25, 0.05179375037550926), (26, 0.07559046149253845), (27, 0.01490346435457468)], Temporal MSE: [(0, 1.6727306501707062e-05), (1, 8.780550160736311e-06), (2, 2.386211599514354e-05), (3, 3.249903966207057e-05), (4, 6.901368033140898e-05), (5, 9.751962352311239e-05), (6, 0.0002540277491789311), (7, 0.00033390766475349665), (8, 0.00037847086787223816), (9, 0.000687494408339262), (10, 0.0009213941521011293), (11, 0.0016000407049432397), (12, 0.0014331055572256446), (13, 0.002844328060746193), (14, 0.005513449665158987), (15, 0.003309512510895729), (16, 0.0025345520116388798), (17, 0.0026460113003849983), (18, 0.0029650286305695772), (19, 0.0027856798842549324), (20, 0.005157065112143755), (21, 0.004150228574872017), (22, 0.003062947653234005), (23, 0.005149076692759991), (24, 0.009485906921327114), (25, 0.013470613397657871), (26, 0.010591740719974041), (27, 0.0014639286091551185)]
+
+ 90%|█████████ | 45/50 [08:02<00:46,  9.34s/it]Time step 80, Spatial MSE: [(0, 1.676178362686187e-05), (1, 7.383115007542074e-05), (2, 0.0006576792802661657), (3, 4.8826506827026606e-05), (4, 8.463929407298565e-05), (5, 0.00015823374269530177), (6, 0.0001968305150512606), (7, 0.0003176814934704453), (8, 0.00041128150769509375), (9, 0.0004366192442830652), (10, 0.00045924040023237467), (11, 0.0005107878823764622), (12, 0.0006206015241332352), (13, 0.00082316406769678), (14, 0.0015662549994885921), (15, 0.004588724114000797), (16, 0.0020172300282865763), (17, 0.0018693743040785193), (18, 0.0021545225754380226), (19, 0.0021220268681645393), (20, 0.0025618653744459152), (21, 0.0026880258228629827), (22, 0.0029896837659180164), (23, 0.006018454674631357), (24, 0.009841631166636944), (25, 0.02091106027364731), (26, 0.02978106029331684), (27, 0.01695367321372032)], Temporal MSE: [(0, 1.9200571841793135e-05), (1, 9.940890777215827e-06), (2, 1.5069175788084976e-05), (3, 1.6844531273818575e-05), (4, 3.108999226242304e-05), (5, 4.484478267841041e-05), (6, 0.00011289185931673273), (7, 0.00012791022891178727), (8, 0.00017166232282761484), (9, 0.0002974936505779624), (10, 0.00035757262958213687), (11, 0.0006477725692093372), (12, 0.0005952976061962545), (13, 0.0009930470259860158), (14, 0.0019199198577553034), (15, 0.0013874702854081988), (16, 0.0010103716049343348), (17, 0.0010245773009955883), (18, 0.0012467772467061877), (19, 0.0012663903180509806), (20, 0.002168324775993824), (21, 0.0018032786902040243), (22, 0.0013643442653119564), (23, 0.0024460344575345516), (24, 0.004183297976851463), (25, 0.006082772742956877), (26, 0.004751165397465229), (27, 0.0011992361396551132)]
+
+ 92%|█████████▏| 46/50 [08:13<00:38,  9.74s/it]Time step 60, Spatial MSE: [(0, 2.1838586690137163e-05), (1, 0.0001077332635759376), (2, 0.0009328704327344894), (3, 6.612730794586241e-05), (4, 0.00012588976824190468), (5, 0.0002192825631937012), (6, 0.00028254848439246416), (7, 0.00043726578587666154), (8, 0.0005649140221066773), (9, 0.000571973214391619), (10, 0.0005715347942896187), (11, 0.0006302617257460952), (12, 0.0007837916491553187), (13, 0.0010945870308205485), (14, 0.004616407677531242), (15, 0.005583296995609999), (16, 0.0983625277876854), (17, 0.002267691073939204), (18, 0.002860981971025467), (19, 0.00410841591656208), (20, 0.0048171053640544415), (21, 0.004131380468606949), (22, 0.004081288352608681), (23, 0.008330509066581726), (24, 0.013627130538225174), (25, 0.028070317581295967), (26, 0.03627466410398483), (27, 0.018037505447864532)], Temporal MSE: [(0, 2.59115931839915e-05), (1, 1.3495548046194017e-05), (2, 2.0789868358406238e-05), (3, 2.221864997409284e-05), (4, 4.160143726039678e-05), (5, 6.707756983814761e-05), (6, 0.00015028382767923176), (7, 0.00017028786533046514), (8, 0.00024116525310091674), (9, 0.0003858448180835694), (10, 0.0004416221927385777), (11, 0.000765924050938338), (12, 0.0007056874455884099), (13, 0.0016924072988331318), (14, 0.0039759911596775055), (15, 0.0017648269422352314), (16, 0.0013240370899438858), (17, 0.001288259751163423), (18, 0.0015542837791144848), (19, 0.0017431338783353567), (20, 0.0030629760585725307), (21, 0.0033490536734461784), (22, 0.0018005134770646691), (23, 0.0032655682880431414), (24, 0.005628605838865042), (25, 0.00814090482890606), (26, 0.006627313327044249), (27, 0.0028150370344519615)]
+
+ 94%|█████████▍| 47/50 [08:23<00:29,  9.90s/it]Time step 40, Spatial MSE: [(0, 3.182711952831596e-05), (1, 0.00017288161325268447), (2, 0.0013534523313865066), (3, 9.517229773337021e-05), (4, 0.00018319633090868592), (5, 0.00032195181120187044), (6, 0.00044638331746682525), (7, 0.0006695322808809578), (8, 0.0008727441309019923), (9, 0.000806617783382535), (10, 0.0007856772281229496), (11, 0.0008412942988798022), (12, 0.0010043426882475615), (13, 0.0013569315196946263), (14, 0.0035075636114925146), (15, 0.006607043091207743), (16, 0.051978424191474915), (17, 0.0029817328322678804), (18, 0.004158273804932833), (19, 0.004471415653824806), (20, 0.0053057135082781315), (21, 0.005625113379210234), (22, 0.005631754174828529), (23, 0.011664407327771187), (24, 0.019867127761244774), (25, 0.036550432443618774), (26, 0.046779900789260864), (27, 0.022192474454641342)], Temporal MSE: [(0, 3.806409949902445e-05), (1, 1.9764493117691018e-05), (2, 3.3980191801674664e-05), (3, 3.16954537993297e-05), (4, 6.0759011830668896e-05), (5, 0.00010829538950929418), (6, 0.00022543770319316536), (7, 0.00025306842871941626), (8, 0.0003582217905204743), (9, 0.0005036290967836976), (10, 0.0005754164885729551), (11, 0.000934830866754055), (12, 0.0008769838605076075), (13, 0.0019023277563974261), (14, 0.003958654124289751), (15, 0.0022862243931740522), (16, 0.0017721632029861212), (17, 0.001697922358289361), (18, 0.0019749091006815434), (19, 0.0024294264148920774), (20, 0.004192118998616934), (21, 0.00377508532255888), (22, 0.0023343455977737904), (23, 0.004471366759389639), (24, 0.008065227419137955), (25, 0.01170204859226942), (26, 0.010188344866037369), (27, 0.011768508702516556)]
+
+ 96%|█████████▌| 48/50 [08:31<00:18,  9.36s/it]Time step 20, Spatial MSE: [(0, 5.981179856462404e-05), (1, 0.00035015400499105453), (2, 0.0022756666876375675), (3, 0.00016667462477926165), (4, 0.0002845611597876996), (5, 0.0005600659060291946), (6, 0.000853078905493021), (7, 0.00135241134557873), (8, 0.0018012329237535596), (9, 0.0015564364148303866), (10, 0.0014410853618755937), (11, 0.0013910636771470308), (12, 0.001634085550904274), (13, 0.0020142430439591408), (14, 0.0037162427324801683), (15, 0.009413974359631538), (16, 0.055662233382463455), (17, 0.004804885480552912), (18, 0.0067094662226736546), (19, 0.0067383102141320705), (20, 0.008078494109213352), (21, 0.009315828792750835), (22, 0.008665001951158047), (23, 0.017678912729024887), (24, 0.033737365156412125), (25, 0.05504868924617767), (26, 0.06888750195503235), (27, 0.04802195727825165)], Temporal MSE: [(0, 6.868189666420221e-05), (1, 3.5227287298766896e-05), (2, 7.109829311957583e-05), (3, 6.0341826610965654e-05), (4, 0.00010690426279325038), (5, 0.00020784998196177185), (6, 0.00042662920895963907), (7, 0.0004658998514059931), (8, 0.0006280072848312557), (9, 0.0007524613756686449), (10, 0.0008686974178999662), (11, 0.001312733511440456), (12, 0.0012443845625966787), (13, 0.0023045395500957966), (14, 0.0053957742638885975), (15, 0.0035093394108116627), (16, 0.0027812751941382885), (17, 0.002571375109255314), (18, 0.0029685499612241983), (19, 0.0035938892979174852), (20, 0.0060988725163042545), (21, 0.005525785032659769), (22, 0.003577059367671609), (23, 0.00693601556122303), (24, 0.014569537714123726), (25, 0.021208329126238823), (26, 0.021642670035362244), (27, 0.11189433932304382)]
+
+ 98%|█████████▊| 49/50 [08:41<00:09,  9.66s/it]Time step 0, Spatial MSE: [(0, 0.00046250803279690444), (1, 0.0020539932884275913), (2, 0.0112786078825593), (3, 0.0010554726468399167), (4, 0.0012395845260471106), (5, 0.0025297831743955612), (6, 0.003400900401175022), (7, 0.007076652720570564), (8, 0.010978658683598042), (9, 0.012233458459377289), (10, 0.010382810607552528), (11, 0.007342963479459286), (12, 0.008412884548306465), (13, 0.008741948753595352), (14, 0.01264217495918274), (15, 0.026912152767181396), (16, 0.11923302710056305), (17, 0.018499955534934998), (18, 0.025174394249916077), (19, 0.02924438565969467), (20, 0.03239548206329346), (21, 0.031231803819537163), (22, 0.04958043992519379), (23, 0.13243427872657776), (24, 0.2398829162120819), (25, 1.083013653755188), (26, 1.1450773477554321), (27, 2.571340560913086)], Temporal MSE: [(0, 0.0005453386693261564), (1, 0.0002299617335665971), (2, 0.0005818921490572393), (3, 0.0004559565568342805), (4, 0.0006170792039483786), (5, 0.0009998786263167858), (6, 0.002543471520766616), (7, 0.0020200340077281), (8, 0.0023236710112541914), (9, 0.002441429067403078), (10, 0.0035107252188026905), (11, 0.004451495595276356), (12, 0.0045089321210980415), (13, 0.00777781568467617), (14, 0.013398749753832817), (15, 0.00987719651311636), (16, 0.008118648082017899), (17, 0.006967712659388781), (18, 0.009457931853830814), (19, 0.009897412732243538), (20, 0.015332006849348545), (21, 0.014414194971323013), (22, 0.017907582223415375), (23, 0.03235480189323425), (24, 0.08183661103248596), (25, 0.15623997151851654), (26, 0.34925809502601624), (27, 3.4235849380493164)]
+
+100%|██████████| 50/50 [08:50<00:00,  9.40s/it]
+100%|██████████| 50/50 [08:50<00:00, 10.62s/it]
+Saved video to ./samples/latte/A_dog_in_astronaut_suit_and_sunglasses_floating_in_space._0000.mp4
+Saved spatial MLP MSE outputs to ./samples/latte/A_dog_in_astronaut_suit_and_sunglasses_floating_in_space._0000_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/latte/A_dog_in_astronaut_suit_and_sunglasses_floating_in_space._0000_temporal_mlp_mse.pt
+Processing the (An epic tornado attacking above aglowing city at night.) prompt
+
+  0%|          | 0/50 [00:00<?, ?it/s]
+  2%|▏         | 1/50 [00:05<04:23,  5.37s/it]Time step 960, Spatial MSE: [(0, 1.006701586447889e-05), (1, 1.69715224274114e-06), (2, 2.188262624258641e-05), (3, 2.9453649403876625e-05), (4, 6.288801523623988e-05), (5, 5.8764941059052944e-05), (6, 5.4757678299210966e-05), (7, 7.189926691353321e-05), (8, 9.216431499226019e-05), (9, 0.0001729585201246664), (10, 0.0006597497267648578), (11, 0.0006803557043895125), (12, 0.0006262772949412465), (13, 0.0008826266275718808), (14, 0.0028066704981029034), (15, 0.0050955540500581264), (16, 0.006106933578848839), (17, 0.003603377379477024), (18, 0.004602324683219194), (19, 0.0030800949316471815), (20, 0.007323800586163998), (21, 0.004140881355851889), (22, 0.0011250064708292484), (23, 0.0019477440509945154), (24, 0.0059351809322834015), (25, 0.005190015304833651), (26, 0.00455920584499836), (27, 0.002064343076199293)], Temporal MSE: [(0, 1.4393577657756396e-05), (1, 4.85795499116648e-06), (2, 4.2250162550772075e-06), (3, 8.88284012035001e-06), (4, 7.047358394629555e-06), (5, 1.4543609722750261e-05), (6, 2.4141399990185164e-05), (7, 4.56836714874953e-05), (8, 3.7805195461260155e-05), (9, 0.00022144685499370098), (10, 0.00032182762515731156), (11, 0.0002352173178223893), (12, 0.0003284211561549455), (13, 0.0003715466591529548), (14, 0.00021089341316837817), (15, 0.0003602545184548944), (16, 0.0007951974403113127), (17, 0.00043518521124497056), (18, 0.0003097744192928076), (19, 0.00010910296259680763), (20, 0.0002673331182450056), (21, 0.0002492090279702097), (22, 0.000336428580339998), (23, 0.00033038275432772934), (24, 0.0004788286460097879), (25, 0.000552066951058805), (26, 0.00029183446895331144), (27, 9.58514920057496e-06)]
+
+  4%|▍         | 2/50 [00:18<07:51,  9.83s/it]Time step 940, Spatial MSE: [(0, 1.1452278158685658e-05), (1, 1.9117733245366253e-06), (2, 2.1434965674416162e-05), (3, 2.9673603421542794e-05), (4, 4.976201671524905e-05), (5, 6.716734787914902e-05), (6, 7.841509068384767e-05), (7, 7.997050852281973e-05), (8, 0.00012913894897792488), (9, 0.0002818403299897909), (10, 0.0008150535286404192), (11, 0.0010754172690212727), (12, 0.001323074335232377), (13, 0.0015591110568493605), (14, 0.0039611379615962505), (15, 0.006130341440439224), (16, 0.008435804396867752), (17, 0.005562876351177692), (18, 0.007153163198381662), (19, 0.0043209814466536045), (20, 0.011962145566940308), (21, 0.006033024750649929), (22, 0.0017795924795791507), (23, 0.0025744950398802757), (24, 0.007853144779801369), (25, 0.007805453613400459), (26, 0.006323674693703651), (27, 0.002185047371312976)], Temporal MSE: [(0, 1.556169445393607e-05), (1, 5.139760105521418e-06), (2, 5.004529157304205e-06), (3, 9.71410918282345e-06), (4, 7.488554729206953e-06), (5, 1.7877178834169172e-05), (6, 3.5823832149617374e-05), (7, 5.6841243349481374e-05), (8, 4.7225090384017676e-05), (9, 0.0003018074494320899), (10, 0.00041763822082430124), (11, 0.00034259085077792406), (12, 0.0006673557218164206), (13, 0.0005947899189777672), (14, 0.00038797841989435256), (15, 0.0005789087153971195), (16, 0.0010368548100814223), (17, 0.0006362537969835103), (18, 0.00036506488686427474), (19, 0.00016329534992109984), (20, 0.00039393201586790383), (21, 0.00042036882950924337), (22, 0.0005131944781169295), (23, 0.00048721639905124903), (24, 0.0006876678671687841), (25, 0.0007555742049589753), (26, 0.0003864416794385761), (27, 1.8210341295343824e-05)]
+
+  6%|▌         | 3/50 [00:29<08:11, 10.46s/it]Time step 920, Spatial MSE: [(0, 1.1918589734705165e-05), (1, 1.9946321572206216e-06), (2, 2.5951521820388734e-05), (3, 2.945742198789958e-05), (4, 4.262637594365515e-05), (5, 6.622775981668383e-05), (6, 0.0001258206320926547), (7, 0.00011117498797830194), (8, 0.00020068968296982348), (9, 0.00045566519838757813), (10, 0.0008237510919570923), (11, 0.001049970043823123), (12, 0.0014256042195484042), (13, 0.0011626138584688306), (14, 0.0027829776518046856), (15, 0.005177331157028675), (16, 0.0070580425672233105), (17, 0.005054643843322992), (18, 0.005878201685845852), (19, 0.0043993424624204636), (20, 0.011624312028288841), (21, 0.005046061705797911), (22, 0.0019514395389705896), (23, 0.002667992142960429), (24, 0.006351304706186056), (25, 0.009636352770030499), (26, 0.007335724774748087), (27, 0.002530024852603674)], Temporal MSE: [(0, 1.7913009287440218e-05), (1, 5.918641363678034e-06), (2, 5.708466233045328e-06), (3, 1.1751479178201407e-05), (4, 8.373066521016881e-06), (5, 2.1398620447143912e-05), (6, 4.256323154550046e-05), (7, 5.6536999181844294e-05), (8, 5.0255148380529135e-05), (9, 0.00033422571141272783), (10, 0.0004694294766522944), (11, 0.0004365902568679303), (12, 0.0007703670999035239), (13, 0.0008555746171623468), (14, 0.0005955240922048688), (15, 0.0006556783337146044), (16, 0.0009705827687866986), (17, 0.0007193008786998689), (18, 0.0004244706651661545), (19, 0.00019587136921472847), (20, 0.00041992461774498224), (21, 0.0004950788570567966), (22, 0.0005899557145312428), (23, 0.0005359143833629787), (24, 0.0007639161194674671), (25, 0.0009153978899121284), (26, 0.0004595098434947431), (27, 1.9100512872682884e-05)]
+
+  8%|▊         | 4/50 [00:38<07:38,  9.97s/it]Time step 900, Spatial MSE: [(0, 1.5573956261505373e-05), (1, 2.344414269828121e-06), (2, 3.0263236112659797e-05), (3, 3.371744242031127e-05), (4, 3.5533172194845974e-05), (5, 7.355379784712568e-05), (6, 0.0002827458083629608), (7, 0.00018343697593081743), (8, 0.0003654105821624398), (9, 0.0007648371392861009), (10, 0.0012261683586984873), (11, 0.001452539931051433), (12, 0.0014002860989421606), (13, 0.0012298180954530835), (14, 0.0021049308124929667), (15, 0.003822609782218933), (16, 0.005164199974387884), (17, 0.003505548695102334), (18, 0.0040646884590387344), (19, 0.0038988653104752302), (20, 0.009306716732680798), (21, 0.003123523434624076), (22, 0.0019674678333103657), (23, 0.002455388894304633), (24, 0.004438367672264576), (25, 0.01067247986793518), (26, 0.009876466356217861), (27, 0.0031716881785541773)], Temporal MSE: [(0, 2.2073783839005046e-05), (1, 6.8842232394672465e-06), (2, 7.217682195914676e-06), (3, 1.6293743101414293e-05), (4, 1.0348515388614032e-05), (5, 3.5245800972916186e-05), (6, 5.9303401940269396e-05), (7, 6.856167601654306e-05), (8, 7.681592978769913e-05), (9, 0.00041477434569969773), (10, 0.0005607557250186801), (11, 0.0006046856287866831), (12, 0.000840265944134444), (13, 0.0012989509850740433), (14, 0.0008112148498184979), (15, 0.0006204948876984417), (16, 0.0008562932489439845), (17, 0.0007048316765576601), (18, 0.0003956042928621173), (19, 0.0002680509933270514), (20, 0.0005781409563496709), (21, 0.0006531099206767976), (22, 0.0007755332626402378), (23, 0.0006277561187744141), (24, 0.0009090221719816327), (25, 0.0012892041122540832), (26, 0.0005959952832199633), (27, 2.5763229132280685e-05)]
+
+ 10%|█         | 5/50 [00:48<07:22,  9.83s/it]Time step 880, Spatial MSE: [(0, 1.2077479368599597e-05), (1, 2.1810910766362213e-06), (2, 3.5963166737928987e-05), (3, 2.5428384105907753e-05), (4, 3.231521986890584e-05), (5, 6.713220500387251e-05), (6, 0.0003044040349777788), (7, 0.0001759993174346164), (8, 0.00027935096295550466), (9, 0.0005112190847285092), (10, 0.0007591054309159517), (11, 0.0007251169881783426), (12, 0.0009513963013887405), (13, 0.0007428980898112059), (14, 0.002489235484972596), (15, 0.004304282367229462), (16, 0.004731085151433945), (17, 0.003443232038989663), (18, 0.003305720863863826), (19, 0.0031206710264086723), (20, 0.007327018305659294), (21, 0.0035324357450008392), (22, 0.0020351780112832785), (23, 0.002350956667214632), (24, 0.004809003788977861), (25, 0.012979896739125252), (26, 0.007812506519258022), (27, 0.0029852446168661118)], Temporal MSE: [(0, 1.877213617262896e-05), (1, 6.408810804714449e-06), (2, 6.97213636158267e-06), (3, 1.31804099510191e-05), (4, 1.0222505807178095e-05), (5, 3.3350901503581554e-05), (6, 6.752710760338232e-05), (7, 7.252515206346288e-05), (8, 6.741755350958556e-05), (9, 0.0003293451154604554), (10, 0.0003862191515509039), (11, 0.00046665346599183977), (12, 0.0006132344133220613), (13, 0.0006737179355695844), (14, 0.0003933339030481875), (15, 0.00044727729982696474), (16, 0.0006509397644549608), (17, 0.0005871125613339245), (18, 0.00029803498182445765), (19, 0.00021199212642386556), (20, 0.0003349524922668934), (21, 0.0004143562400713563), (22, 0.0005931658670306206), (23, 0.0005078231915831566), (24, 0.0006575310253538191), (25, 0.0009788313182070851), (26, 0.0005632814136333764), (27, 2.7672578653437085e-05)]
+
+ 12%|█▏        | 6/50 [00:58<07:14,  9.87s/it]Time step 860, Spatial MSE: [(0, 4.483049997361377e-05), (1, 7.4573399615474045e-06), (2, 4.0019403968472034e-05), (3, 7.512739102821797e-05), (4, 9.550341928843409e-05), (5, 0.00014231065870262682), (6, 0.0010822839103639126), (7, 0.0003882746968884021), (8, 0.0005691974074579775), (9, 0.0008749169064685702), (10, 0.0011424231342971325), (11, 0.0011854334734380245), (12, 0.0012437683762982488), (13, 0.0012018567649647593), (14, 0.0030877499375492334), (15, 0.005947201047092676), (16, 0.006657245568931103), (17, 0.005004323553293943), (18, 0.006746913306415081), (19, 0.006010936573147774), (20, 0.012813013046979904), (21, 0.0061389910988509655), (22, 0.0038349356036633253), (23, 0.0040117125026881695), (24, 0.006215565837919712), (25, 0.02113179676234722), (26, 0.027260610833764076), (27, 0.011275646276772022)], Temporal MSE: [(0, 6.892418605275452e-05), (1, 1.947801501955837e-05), (2, 2.1005482267355546e-05), (3, 3.574421862140298e-05), (4, 2.681812839000486e-05), (5, 0.00016436036094091833), (6, 0.00012160080950707197), (7, 0.0001261242141481489), (8, 0.00014796064351685345), (9, 0.0005056319641880691), (10, 0.0005665116477757692), (11, 0.0006660545477643609), (12, 0.0007969466969370842), (13, 0.0010209237225353718), (14, 0.0006598429172299802), (15, 0.0006038045976310968), (16, 0.0008140408899635077), (17, 0.0007850524270907044), (18, 0.0004934676690027118), (19, 0.0003711788740474731), (20, 0.001081073540262878), (21, 0.000728551996871829), (22, 0.0014339664485305548), (23, 0.001119589083828032), (24, 0.0016726873582229018), (25, 0.0028112258296459913), (26, 0.0016114356694743037), (27, 0.00012370399781502783)]
+
+ 14%|█▍        | 7/50 [01:09<07:27, 10.41s/it]Time step 840, Spatial MSE: [(0, 5.100412181491265e-06), (1, 2.757006996034761e-06), (2, 5.59651707590092e-05), (3, 1.648558463784866e-05), (4, 2.3486887585022487e-05), (5, 5.0075919716618955e-05), (6, 0.0005207685753703117), (7, 0.00019834534032270312), (8, 0.0002859444357454777), (9, 0.0005605719052255154), (10, 0.000750271137803793), (11, 0.0007024403312243521), (12, 0.0007002662750892341), (13, 0.000746897654607892), (14, 0.0014690437819808722), (15, 0.0017178341513499618), (16, 0.0012804860016331077), (17, 0.0009627261315472424), (18, 0.0007859165198169649), (19, 0.0012068203650414944), (20, 0.0015943336766213179), (21, 0.0005558215780183673), (22, 0.0010099775390699506), (23, 0.0016424410277977586), (24, 0.0023360818158835173), (25, 0.011297263205051422), (26, 0.005609795916825533), (27, 0.004630604293197393)], Temporal MSE: [(0, 8.464083293802105e-06), (1, 3.1785773444426013e-06), (2, 5.952904757577926e-06), (3, 1.790379246813245e-05), (4, 1.1537872524058912e-05), (5, 2.9304126655915752e-05), (6, 5.9068221162306145e-05), (7, 5.741789937019348e-05), (8, 6.490806845249608e-05), (9, 0.00017818226478993893), (10, 0.00024096277775242925), (11, 0.0002386294218013063), (12, 0.00028462614864110947), (13, 0.0003995005099568516), (14, 0.00042953959200531244), (15, 0.00037596988840959966), (16, 0.0004111821763217449), (17, 0.00035505148116499186), (18, 0.0003507652145344764), (19, 0.0003576503659132868), (20, 0.0006206332473084331), (21, 0.0004914234159514308), (22, 0.0004973206669092178), (23, 0.000462387572042644), (24, 0.0009329469176009297), (25, 0.0013946698745712638), (26, 0.0008513474022038281), (27, 4.726212500827387e-05)]
+
+ 16%|█▌        | 8/50 [01:18<07:00, 10.02s/it]Time step 820, Spatial MSE: [(0, 1.2867663826909848e-05), (1, 2.597367483758717e-06), (2, 5.7718149037100375e-05), (3, 2.86295180558227e-05), (4, 6.057533391867764e-05), (5, 7.799762533977628e-05), (6, 0.0012696633348241448), (7, 0.00030205579241737723), (8, 0.00041104352567344904), (9, 0.0007476755417883396), (10, 0.0010250683408230543), (11, 0.0009648618870414793), (12, 0.0010493245208635926), (13, 0.0011274676071479917), (14, 0.00227361056022346), (15, 0.0025382244493812323), (16, 0.0022472005803138018), (17, 0.001538748387247324), (18, 0.0015765378484502435), (19, 0.0021919002756476402), (20, 0.0031609726138412952), (21, 0.0009910819353535771), (22, 0.0019146515987813473), (23, 0.002664208645001054), (24, 0.0032471464946866035), (25, 0.017878737300634384), (26, 0.01608322188258171), (27, 0.0038256836123764515)], Temporal MSE: [(0, 1.8679798813536763e-05), (1, 6.29637406746042e-06), (2, 1.2558637536130846e-05), (3, 2.949856934719719e-05), (4, 2.285491063958034e-05), (5, 0.00011049275781260803), (6, 0.00011466394062153995), (7, 9.848804620560259e-05), (8, 0.00010709581692935899), (9, 0.00029728899244219065), (10, 0.0003653608146123588), (11, 0.0003917055728379637), (12, 0.00047589963651262224), (13, 0.0007057845941744745), (14, 0.0006272199098020792), (15, 0.0005499494727700949), (16, 0.0005533064249902964), (17, 0.0005222933832556009), (18, 0.00045646741637028754), (19, 0.00048237646115012467), (20, 0.0009015527321025729), (21, 0.000788930687122047), (22, 0.0010257038520649076), (23, 0.0009340670658275485), (24, 0.001792561262845993), (25, 0.0029336120933294296), (26, 0.0018832103814929724), (27, 7.35836147214286e-05)]
+
+ 18%|█▊        | 9/50 [01:28<06:37,  9.70s/it]Time step 800, Spatial MSE: [(0, 1.2547864571388345e-05), (1, 3.0454527859546943e-06), (2, 7.233439828269184e-05), (3, 2.8070944608771242e-05), (4, 8.684638305567205e-05), (5, 8.640103624202311e-05), (6, 0.0010697492398321629), (7, 0.00028739983099512756), (8, 0.0003791879571508616), (9, 0.0006683373940177262), (10, 0.0009353057248517871), (11, 0.00087833619909361), (12, 0.0009152234415523708), (13, 0.0010104178218171), (14, 0.0017600262071937323), (15, 0.0017777425236999989), (16, 0.0014065232826396823), (17, 0.001031361985951662), (18, 0.0011307235108688474), (19, 0.0015139838214963675), (20, 0.0019268958130851388), (21, 0.0007609737222082913), (22, 0.001559644821099937), (23, 0.002135911723598838), (24, 0.0026713714469224215), (25, 0.014227587729692459), (26, 0.013409712351858616), (27, 0.004357123747467995)], Temporal MSE: [(0, 1.8723769244388677e-05), (1, 6.046172529750038e-06), (2, 1.3368340660235845e-05), (3, 3.439930878812447e-05), (4, 3.0038945624255575e-05), (5, 0.0001145016576629132), (6, 0.00015533003897871822), (7, 0.00010397013829788193), (8, 0.00010511917935218662), (9, 0.0002673007547855377), (10, 0.0003399043343961239), (11, 0.00036706068203784525), (12, 0.0003920220478903502), (13, 0.0006377517711371183), (14, 0.0005119742709212005), (15, 0.000450725550763309), (16, 0.0004102320526726544), (17, 0.0003877159906551242), (18, 0.00034498830791562796), (19, 0.00035671296063810587), (20, 0.0006427456391975284), (21, 0.0006437371484935284), (22, 0.0008366125402972102), (23, 0.0007793826516717672), (24, 0.001630082493647933), (25, 0.002775152213871479), (26, 0.0018365298165008426), (27, 8.502598211634904e-05)]
+
+ 20%|██        | 10/50 [01:38<06:33,  9.84s/it]Time step 780, Spatial MSE: [(0, 9.628917723603081e-06), (1, 2.6991906452167314e-06), (2, 6.661024963250384e-05), (3, 2.046557165158447e-05), (4, 2.445763129799161e-05), (5, 3.2050764275481924e-05), (6, 6.0853508330183104e-05), (7, 9.175442392006516e-05), (8, 0.00016452996351290494), (9, 0.00021937272686045617), (10, 0.0007209719624370337), (11, 0.0007773570250719786), (12, 0.000719154195394367), (13, 0.0008336084429174662), (14, 0.0016669082688167691), (15, 0.0015311532188206911), (16, 0.0011473889462649822), (17, 0.0014802203513681889), (18, 0.0008680865867063403), (19, 0.0008966547320596874), (20, 0.00127122038975358), (21, 0.000709326530341059), (22, 0.0006164599908515811), (23, 0.0011612738016992807), (24, 0.0025779870338737965), (25, 0.006144560873508453), (26, 0.00305519113317132), (27, 0.00511096091940999)], Temporal MSE: [(0, 1.3059855518804397e-05), (1, 5.195930498302914e-06), (2, 9.401233000971843e-06), (3, 1.5723046089988202e-05), (4, 1.3801801287627313e-05), (5, 2.291471537319012e-05), (6, 4.256548345438205e-05), (7, 0.00011482513946248218), (8, 7.153964543249458e-05), (9, 0.00022911495761945844), (10, 0.000553442514501512), (11, 0.00046726129949092865), (12, 0.0004396403965074569), (13, 0.0006771167973056436), (14, 0.0007433216087520123), (15, 0.0007239065016619861), (16, 0.0010174508206546307), (17, 0.0007333004614338279), (18, 0.0007835750002413988), (19, 0.00031683463021181524), (20, 0.0009033770766109228), (21, 0.000783583614975214), (22, 0.00030129923834465444), (23, 0.0005686041549779475), (24, 0.0009692960302345455), (25, 0.0013842538464814425), (26, 0.0005034853238612413), (27, 5.0635760999284685e-05)]
+
+ 22%|██▏       | 11/50 [01:45<05:56,  9.13s/it]Time step 760, Spatial MSE: [(0, 9.626786777516827e-06), (1, 3.1749234494782286e-06), (2, 5.475833313539624e-05), (3, 1.0869012839975767e-05), (4, 1.0752788512036204e-05), (5, 6.852126716694329e-06), (6, 1.1506549526529852e-05), (7, 1.2571760635182727e-05), (8, 1.9967726984759793e-05), (9, 2.262627640448045e-05), (10, 4.087998240720481e-05), (11, 4.763455217471346e-05), (12, 7.121694943634793e-05), (13, 0.00012529273226391524), (14, 0.00014632170496042818), (15, 0.00015232345322147012), (16, 0.00031545839738100767), (17, 0.0001861085183918476), (18, 0.00019994049216620624), (19, 0.0001765274500939995), (20, 0.0002624711487442255), (21, 0.0003771959454752505), (22, 0.00015190106933005154), (23, 0.0003533123235683888), (24, 0.0008689392707310617), (25, 0.0015002350555732846), (26, 0.0012815697118639946), (27, 0.004236249718815088)], Temporal MSE: [(0, 1.719043757475447e-05), (1, 5.919440354773542e-06), (2, 5.0826806727855e-06), (3, 4.872468707617372e-06), (4, 2.6129991965717636e-06), (5, 5.151139703230001e-06), (6, 4.87472470922512e-06), (7, 7.196584192570299e-06), (8, 5.563885224546539e-06), (9, 1.8216698663309216e-05), (10, 2.1228164769127034e-05), (11, 4.4398857426131144e-05), (12, 4.857456588069908e-05), (13, 0.00011504178837640211), (14, 0.00012176283053122461), (15, 0.00010059549094876274), (16, 0.0001780657039489597), (17, 0.00011218641884624958), (18, 0.00017842442321125418), (19, 8.835011249175295e-05), (20, 0.00020963026327081025), (21, 0.0003082974872086197), (22, 0.00010872999700950459), (23, 0.00016832070832606405), (24, 0.00017918460071086884), (25, 0.0006227719131857157), (26, 0.00018093954713549465), (27, 7.520651706727222e-05)]
+
+ 24%|██▍       | 12/50 [01:53<05:35,  8.82s/it]Time step 740, Spatial MSE: [(0, 2.1342611944419332e-05), (1, 1.2002808944089338e-05), (2, 0.0001604343269718811), (3, 0.0001177680678665638), (4, 0.0006372977513819933), (5, 0.00039734566234983504), (6, 0.0031941889319568872), (7, 0.0017526402371004224), (8, 0.0020956287626177073), (9, 0.0030737624038010836), (10, 0.003710792399942875), (11, 0.004508380778133869), (12, 0.004890932701528072), (13, 0.005894209258258343), (14, 0.008732332848012447), (15, 0.008883584290742874), (16, 0.00813340675085783), (17, 0.004971691407263279), (18, 0.004225282464176416), (19, 0.003385654417797923), (20, 0.0031890287064015865), (21, 0.0036715310998260975), (22, 0.004986590705811977), (23, 0.006315162405371666), (24, 0.00987186934798956), (25, 0.047646865248680115), (26, 0.04801372438669205), (27, 0.007046819664537907)], Temporal MSE: [(0, 4.166529106441885e-05), (1, 7.509172064601444e-06), (2, 4.8725549277151003e-05), (3, 0.00022524285304825753), (4, 0.00020431936718523502), (5, 0.0005081656272523105), (6, 0.0011886439751833677), (7, 0.0008658396545797586), (8, 0.0006573937134817243), (9, 0.0012541301548480988), (10, 0.0016889249673113227), (11, 0.0017374677117913961), (12, 0.0020270771346986294), (13, 0.002947745146229863), (14, 0.0032013431191444397), (15, 0.003028981853276491), (16, 0.004388412460684776), (17, 0.0033093425445258617), (18, 0.0031749894842505455), (19, 0.001947775948792696), (20, 0.006707634776830673), (21, 0.002540531102567911), (22, 0.003021557815372944), (23, 0.002999035408720374), (24, 0.008491120301187038), (25, 0.014766093343496323), (26, 0.010315692983567715), (27, 0.0004920989158563316)]
+
+ 26%|██▌       | 13/50 [02:07<06:20, 10.28s/it]Time step 720, Spatial MSE: [(0, 8.0160862125922e-06), (1, 4.323461780586513e-06), (2, 8.67009730427526e-05), (3, 1.4108574760030024e-05), (4, 1.459324084862601e-05), (5, 8.177516065188684e-06), (6, 1.6819687516544946e-05), (7, 1.968116885109339e-05), (8, 2.0113417122047395e-05), (9, 3.0525541660608724e-05), (10, 3.668107819976285e-05), (11, 5.215751662035473e-05), (12, 7.371855463134125e-05), (13, 0.00010307316551916301), (14, 0.00013933464651927352), (15, 0.00018425584130454808), (16, 0.00021011247008573264), (17, 0.00016507164400536567), (18, 0.0003375908127054572), (19, 0.0001599243114469573), (20, 0.0003001575532834977), (21, 0.0004703026788774878), (22, 0.00019017459999304265), (23, 0.00042078926344402134), (24, 0.0009840525453910232), (25, 0.0017633421812206507), (26, 0.001580169191583991), (27, 0.0050905924290418625)], Temporal MSE: [(0, 1.3976225091028027e-05), (1, 4.8725601118349005e-06), (2, 6.0685360949719325e-06), (3, 5.548515218833927e-06), (4, 3.229856247344287e-06), (5, 7.41154281058698e-06), (6, 7.18478077033069e-06), (7, 1.2346401490503922e-05), (8, 6.865697287139483e-06), (9, 1.6309604689013213e-05), (10, 2.365020009165164e-05), (11, 4.1772480471991e-05), (12, 5.187893475522287e-05), (13, 9.669604332884774e-05), (14, 9.762233821675181e-05), (15, 0.00011611621448537335), (16, 0.0001479337370255962), (17, 0.00010169077722821385), (18, 0.00022780406288802624), (19, 0.00011250231909798458), (20, 0.00023082172265276313), (21, 0.0003075022832490504), (22, 0.00010092424781760201), (23, 0.00020020295050926507), (24, 0.00023507658625021577), (25, 0.0007581091485917568), (26, 0.00025765455211512744), (27, 8.034481288632378e-05)]
+
+ 28%|██▊       | 14/50 [02:19<06:30, 10.86s/it]Time step 700, Spatial MSE: [(0, 1.9615072233136743e-05), (1, 9.070457053894643e-06), (2, 0.00014294737775344402), (3, 4.4626671297010034e-05), (4, 7.823073246981949e-05), (5, 0.00010595962521620095), (6, 0.0002483816060703248), (7, 0.00035690946970134974), (8, 0.0007721022120676935), (9, 0.0010332271922379732), (10, 0.0027776046190410852), (11, 0.0028147613629698753), (12, 0.0028254787903279066), (13, 0.0028021049220114946), (14, 0.00514642521739006), (15, 0.003639382543042302), (16, 0.002815886400640011), (17, 0.003219635458663106), (18, 0.0013846912188455462), (19, 0.0019016420701518655), (20, 0.0018877097172662616), (21, 0.000959560158662498), (22, 0.001524984952993691), (23, 0.002611558185890317), (24, 0.005921463947743177), (25, 0.017780503258109093), (26, 0.007031669374555349), (27, 0.009609546512365341)], Temporal MSE: [(0, 1.3184228919271845e-05), (1, 4.847605850955006e-06), (2, 2.353171475988347e-05), (3, 4.9585880333324894e-05), (4, 4.383407940622419e-05), (5, 0.00010641360131558031), (6, 0.00016493836301378906), (7, 0.0003805965243373066), (8, 0.0002312186552444473), (9, 0.0005849063745699823), (10, 0.0012902497546747327), (11, 0.0012481125304475427), (12, 0.0010955259203910828), (13, 0.0019307674374431372), (14, 0.0019247432937845588), (15, 0.0015968993538990617), (16, 0.0019515990279614925), (17, 0.0015916806878522038), (18, 0.001386148389428854), (19, 0.000766328361351043), (20, 0.0017228410579264164), (21, 0.0018164892680943012), (22, 0.0009033988462761045), (23, 0.0016584083205088973), (24, 0.004037871956825256), (25, 0.0058632902801036835), (26, 0.0020564754959195852), (27, 0.00018040029681287706)]
+
+ 30%|███       | 15/50 [02:27<05:52, 10.08s/it]Time step 680, Spatial MSE: [(0, 6.997563104960136e-06), (1, 1.7059339370462112e-05), (2, 0.0001487036788603291), (3, 4.6499808377120644e-05), (4, 0.0001495805918239057), (5, 0.00012715606135316193), (6, 0.00045220734318718314), (7, 0.0006808508187532425), (8, 0.00025357233243994415), (9, 0.00036397294024936855), (10, 0.0005423391121439636), (11, 0.0006111627444624901), (12, 0.0005813064053654671), (13, 0.0009039660217240453), (14, 0.0021392451599240303), (15, 0.0034458162263035774), (16, 0.032602034509181976), (17, 0.0015074297552928329), (18, 0.002745983889326453), (19, 0.001735940808430314), (20, 0.004039878491312265), (21, 0.006018427666276693), (22, 0.0027752574533224106), (23, 0.0019803906325250864), (24, 0.004343200009316206), (25, 0.02061544544994831), (26, 0.005095404572784901), (27, 0.007457940839231014)], Temporal MSE: [(0, 1.9205106582376175e-05), (1, 7.598276170028839e-06), (2, 2.770264291029889e-05), (3, 0.00013526796828955412), (4, 7.67227029427886e-05), (5, 0.0001955135667230934), (6, 0.0002768904669210315), (7, 0.00012481703015509993), (8, 0.00010174075578106567), (9, 0.00027046745526604354), (10, 0.0005971693899482489), (11, 0.0006661703810095787), (12, 0.000711333064828068), (13, 0.001497426419518888), (14, 0.003080476075410843), (15, 0.001602228614501655), (16, 0.0016630367608740926), (17, 0.0007086497498676181), (18, 0.0005319386254996061), (19, 0.0005555573152378201), (20, 0.0030525107868015766), (21, 0.0009954762645065784), (22, 0.0013461881317198277), (23, 0.0014606956392526627), (24, 0.004293068777769804), (25, 0.005412341095507145), (26, 0.003059971611946821), (27, 0.00020976518862880766)]
+
+ 32%|███▏      | 16/50 [02:36<05:32,  9.77s/it]Time step 660, Spatial MSE: [(0, 1.581848118803464e-05), (1, 1.1373123925295658e-05), (2, 0.00013793111429549754), (3, 8.027879084693268e-05), (4, 0.0001953653118107468), (5, 0.00018104571790900081), (6, 0.0004777394642587751), (7, 0.0005387914134189487), (8, 0.0006337129743769765), (9, 0.0010033600265160203), (10, 0.0012881152797490358), (11, 0.0014275277499109507), (12, 0.0016731698997318745), (13, 0.0028660958632826805), (14, 0.005564694292843342), (15, 0.0035385307855904102), (16, 0.03428461030125618), (17, 0.0017758235335350037), (18, 0.0018536957213655114), (19, 0.0027985163033008575), (20, 0.0020601346623152494), (21, 0.002041875384747982), (22, 0.0018360523972660303), (23, 0.0023293267004191875), (24, 0.007602176629006863), (25, 0.016718389466404915), (26, 0.0069092982448637486), (27, 0.006385800428688526)], Temporal MSE: [(0, 9.854967174760532e-06), (1, 3.724600219356944e-06), (2, 1.8417447790852748e-05), (3, 7.074845052557066e-05), (4, 8.598206477472559e-05), (5, 0.00014696476864628494), (6, 0.0001771825918694958), (7, 0.0002804456162266433), (8, 0.00020297862647566944), (9, 0.0005158209824003279), (10, 0.000580699706915766), (11, 0.0008419929072260857), (12, 0.0008587514166720212), (13, 0.002233257982879877), (14, 0.0031211813911795616), (15, 0.0014382431982085109), (16, 0.001765381544828415), (17, 0.0009963862830772996), (18, 0.00123414711561054), (19, 0.0011524499859660864), (20, 0.002523954026401043), (21, 0.0016676891827955842), (22, 0.0013119253562763333), (23, 0.001897865324281156), (24, 0.006082437001168728), (25, 0.011828439310193062), (26, 0.006489462219178677), (27, 0.00028839847072958946)]
+
+ 34%|███▍      | 17/50 [02:49<05:53, 10.72s/it]Time step 640, Spatial MSE: [(0, 7.4673371273092926e-06), (1, 9.559185855323449e-06), (2, 0.00010120365186594427), (3, 2.4170532924472354e-05), (4, 1.578245974087622e-05), (5, 9.136971129919402e-06), (6, 4.0137223550118506e-05), (7, 5.1306255045346916e-05), (8, 1.676633837632835e-05), (9, 2.8655507776420563e-05), (10, 3.794567965087481e-05), (11, 4.8578865971649066e-05), (12, 5.880791286472231e-05), (13, 5.608564242720604e-05), (14, 0.00011498216190375388), (15, 0.0001240379933733493), (16, 0.0001009824700304307), (17, 0.00013729269267059863), (18, 0.0002667409717105329), (19, 0.0001112834652303718), (20, 0.0001947617274709046), (21, 0.0005097548710182309), (22, 0.00014961708802729845), (23, 0.0003982317866757512), (24, 0.0008500580443069339), (25, 0.0018977754516527057), (26, 0.002647201530635357), (27, 0.006254764273762703)], Temporal MSE: [(0, 8.819381946523208e-06), (1, 3.006045290021575e-06), (2, 5.881979632249568e-06), (3, 5.322409833752317e-06), (4, 3.7581121432594955e-06), (5, 8.677207006257959e-06), (6, 1.767499816196505e-05), (7, 2.363785279158037e-05), (8, 7.812393050699029e-06), (9, 1.3730640603171196e-05), (10, 2.5855404601315968e-05), (11, 3.726873910636641e-05), (12, 3.144927541143261e-05), (13, 5.6705619499552995e-05), (14, 6.307735020527616e-05), (15, 8.742633508518338e-05), (16, 9.351359767606482e-05), (17, 7.627499144291505e-05), (18, 0.0001800569734768942), (19, 9.125742508331314e-05), (20, 0.0001327277277596295), (21, 0.00026249076472595334), (22, 6.133557326393202e-05), (23, 0.00017930720059666783), (24, 0.0002674917341209948), (25, 0.0007368054357357323), (26, 0.0003519877791404724), (27, 6.393260264303535e-05)]
+
+ 36%|███▌      | 18/50 [03:00<05:42, 10.71s/it]Time step 620, Spatial MSE: [(0, 1.5336503565777093e-05), (1, 2.4952923922683112e-05), (2, 0.00017483766714576632), (3, 9.065177437150851e-05), (4, 0.00029601954156532884), (5, 0.00037277789670042694), (6, 0.004523271694779396), (7, 0.0023747822269797325), (8, 0.0033213032875210047), (9, 0.005578373093158007), (10, 0.006900522857904434), (11, 0.010619810782372952), (12, 0.010340158827602863), (13, 0.013775322586297989), (14, 0.029345830902457237), (15, 0.02497721090912819), (16, 0.024961790069937706), (17, 0.017170701175928116), (18, 0.013999147340655327), (19, 0.011402792297303677), (20, 0.015271848998963833), (21, 0.015992427244782448), (22, 0.007738521322607994), (23, 0.012015115469694138), (24, 0.023390604183077812), (25, 0.03507554531097412), (26, 0.0247978363186121), (27, 0.00816844031214714)], Temporal MSE: [(0, 1.821313526306767e-05), (1, 7.1679078246233985e-06), (2, 2.6840383725357242e-05), (3, 0.0003104617935605347), (4, 0.00023199159477371722), (5, 0.0006286107236519456), (6, 0.002201203489676118), (7, 0.002251050202175975), (8, 0.001837796182371676), (9, 0.003327791579067707), (10, 0.005376370623707771), (11, 0.005450073629617691), (12, 0.005630167201161385), (13, 0.00856323353946209), (14, 0.009184573777019978), (15, 0.009163393639028072), (16, 0.010389546863734722), (17, 0.008717488497495651), (18, 0.007680067792534828), (19, 0.0061270445585250854), (20, 0.012415864504873753), (21, 0.008888975717127323), (22, 0.006076500751078129), (23, 0.008482173085212708), (24, 0.019055010750889778), (25, 0.03337586671113968), (26, 0.017988838255405426), (27, 0.0013550291769206524)]
+
+ 38%|███▊      | 19/50 [03:10<05:26, 10.54s/it]Time step 600, Spatial MSE: [(0, 7.4112667789449915e-06), (1, 1.2719088772428222e-05), (2, 0.00010314906830899417), (3, 2.9464652470778674e-05), (4, 1.594887362443842e-05), (5, 1.0047077012131922e-05), (6, 3.811120041063987e-05), (7, 5.510988921741955e-05), (8, 1.7665675841271877e-05), (9, 2.891635631385725e-05), (10, 3.710294913616963e-05), (11, 4.266513860784471e-05), (12, 5.915871588513255e-05), (13, 5.394867912400514e-05), (14, 8.632821845822036e-05), (15, 0.00012220960343256593), (16, 0.00010056406608782709), (17, 0.0001292102679144591), (18, 0.000229495984967798), (19, 0.0001042308213072829), (20, 0.00020109876641072333), (21, 0.00044675430399365723), (22, 0.00015529015217907727), (23, 0.0004239529080223292), (24, 0.0008894286584109068), (25, 0.0019281203858554363), (26, 0.003094620304182172), (27, 0.006861676927655935)], Temporal MSE: [(0, 7.470515811291989e-06), (1, 2.6361417440057267e-06), (2, 6.0031652537873015e-06), (3, 6.187755388964433e-06), (4, 4.3342811295588035e-06), (5, 9.486177077633329e-06), (6, 1.9181843526894227e-05), (7, 2.144680365745444e-05), (8, 8.305511983053293e-06), (9, 1.1893765076820273e-05), (10, 2.2427027943194844e-05), (11, 3.083854608121328e-05), (12, 2.7181748009752482e-05), (13, 5.681885886588134e-05), (14, 5.012453766539693e-05), (15, 7.463997462764382e-05), (16, 8.489244646625593e-05), (17, 6.824143201811239e-05), (18, 0.00014195707626640797), (19, 8.243645424954593e-05), (20, 0.0001135592901846394), (21, 0.00023916017380543053), (22, 6.484687037300318e-05), (23, 0.0001623421994736418), (24, 0.0002632094838190824), (25, 0.0007689381600357592), (26, 0.00044393917778506875), (27, 7.148274744395167e-05)]
+
+ 40%|████      | 20/50 [03:22<05:28, 10.96s/it]Time step 580, Spatial MSE: [(0, 1.5654051821911708e-05), (1, 1.9185639757779427e-05), (2, 0.00017130962805822492), (3, 3.855927207041532e-05), (4, 6.236929766600952e-05), (5, 6.07379297434818e-05), (6, 0.00038251301157288253), (7, 0.0002798908681143075), (8, 0.0007612997433170676), (9, 0.0008801090298220515), (10, 0.0016830679960548878), (11, 0.0023363707587122917), (12, 0.0028805481269955635), (13, 0.0027635726146399975), (14, 0.004142475314438343), (15, 0.002586065325886011), (16, 0.001265405211597681), (17, 0.0010888203978538513), (18, 0.0007744764443486929), (19, 0.0006797105306759477), (20, 0.000794912688434124), (21, 0.0007947772392071784), (22, 0.0010944250971078873), (23, 0.001611195970326662), (24, 0.00429858174175024), (25, 0.008192579261958599), (26, 0.006843249779194593), (27, 0.009894599206745625)], Temporal MSE: [(0, 7.513238870160421e-06), (1, 3.3015048757079057e-06), (2, 1.6829644664539956e-05), (3, 3.888286664732732e-05), (4, 2.9760807592538185e-05), (5, 7.69464168115519e-05), (6, 8.420810627285391e-05), (7, 0.00020541150297503918), (8, 0.00016497632896061987), (9, 0.0002766048419289291), (10, 0.0004476815811358392), (11, 0.0006095904391258955), (12, 0.0007604477577842772), (13, 0.0012417661491781473), (14, 0.0011409642174839973), (15, 0.0007408214150927961), (16, 0.0005618114955723286), (17, 0.00046177394688129425), (18, 0.0004274091625120491), (19, 0.00033785169944167137), (20, 0.0005329460254870355), (21, 0.0008737101452425122), (22, 0.000676765339449048), (23, 0.001049834885634482), (24, 0.0038779331371188164), (25, 0.006946460343897343), (26, 0.0026272342074662447), (27, 0.00028957761242054403)]
+
+ 42%|████▏     | 21/50 [03:31<04:59, 10.32s/it]Time step 560, Spatial MSE: [(0, 6.802685675211251e-06), (1, 2.3372103896690533e-05), (2, 9.863298328127712e-05), (3, 4.630608964362182e-05), (4, 5.658770169247873e-05), (5, 0.00010646446025930345), (6, 0.00021128993830643594), (7, 0.0005179820000194013), (8, 0.00022372308012563735), (9, 0.0002810544101521373), (10, 0.0005014788475818932), (11, 0.000459953531390056), (12, 0.0005178064457140863), (13, 0.0015756137436255813), (14, 0.0035450858995318413), (15, 0.002686974825337529), (16, 0.07043847441673279), (17, 0.000964838603977114), (18, 0.0014513410860672593), (19, 0.002773919142782688), (20, 0.0027664010412991047), (21, 0.004845428746193647), (22, 0.0017740707844495773), (23, 0.001620207680389285), (24, 0.0039422339759767056), (25, 0.012897727079689503), (26, 0.0032204713206738234), (27, 0.011072084307670593)], Temporal MSE: [(0, 8.140606041706633e-06), (1, 4.6916757128201425e-06), (2, 2.136120201612357e-05), (3, 6.748994201188907e-05), (4, 6.547046359628439e-05), (5, 7.475123857147992e-05), (6, 0.00012847338803112507), (7, 0.00010533953900448978), (8, 8.054177305893973e-05), (9, 0.00024406000738963485), (10, 0.0005040995310992002), (11, 0.0006331413169391453), (12, 0.0007521111983805895), (13, 0.003267450723797083), (14, 0.0046277474611997604), (15, 0.000873578421305865), (16, 0.0006381726707331836), (17, 0.0004151028406340629), (18, 0.00042986555490642786), (19, 0.00042513266089372337), (20, 0.0008844263502396643), (21, 0.0008962370920926332), (22, 0.0013907671673223376), (23, 0.002150318818166852), (24, 0.0047195544466376305), (25, 0.006548421457409859), (26, 0.004331028554588556), (27, 0.0005810752045363188)]
+
+ 44%|████▍     | 22/50 [03:39<04:32,  9.74s/it]Time step 540, Spatial MSE: [(0, 1.4520106560667045e-05), (1, 2.14756655623205e-05), (2, 0.00016813531692605466), (3, 6.858957931399345e-05), (4, 9.761995897861198e-05), (5, 0.00011433741019573063), (6, 0.0006040563457645476), (7, 0.0004704115563072264), (8, 0.0005319707561284304), (9, 0.0006619927589781582), (10, 0.0008172197267413139), (11, 0.0014213029062375426), (12, 0.00167818833142519), (13, 0.0034278002567589283), (14, 0.00743508106097579), (15, 0.004210516344755888), (16, 0.07434156537055969), (17, 0.001922709634527564), (18, 0.002113530645146966), (19, 0.0035283202305436134), (20, 0.002391513204202056), (21, 0.0032944551203399897), (22, 0.0024772833567112684), (23, 0.00277545559220016), (24, 0.0074319117702543736), (25, 0.01593095064163208), (26, 0.01511200238019228), (27, 0.010795125737786293)], Temporal MSE: [(0, 6.544506049976917e-06), (1, 3.1428307920577936e-06), (2, 1.3451451195578557e-05), (3, 5.7028890296351165e-05), (4, 6.829967605881393e-05), (5, 0.00011976808309555054), (6, 0.00012836679525207728), (7, 0.00019197262008674443), (8, 0.0002004159614443779), (9, 0.00038376517477445304), (10, 0.00039985659532248974), (11, 0.0007127859280444682), (12, 0.0008366730762645602), (13, 0.0025774671230465174), (14, 0.005022961180657148), (15, 0.001848265528678894), (16, 0.0026196951512247324), (17, 0.001251738634891808), (18, 0.0010450625559315085), (19, 0.001063925912603736), (20, 0.0033448149915784597), (21, 0.002080608857795596), (22, 0.0014837676426395774), (23, 0.00229904823936522), (24, 0.00630313390865922), (25, 0.01080299075692892), (26, 0.00788803119212389), (27, 0.0005228124209679663)]
+
+ 46%|████▌     | 23/50 [03:48<04:11,  9.32s/it]Time step 520, Spatial MSE: [(0, 6.672186373180011e-06), (1, 1.5844861991354264e-05), (2, 7.846942025935277e-05), (3, 3.0118893846520223e-05), (4, 1.4002333045937121e-05), (5, 1.0312660378986038e-05), (6, 2.947073880932294e-05), (7, 5.8295612689107656e-05), (8, 1.5178601643128786e-05), (9, 2.3021369997877628e-05), (10, 2.8391892556101084e-05), (11, 3.326399746583775e-05), (12, 3.884143370669335e-05), (13, 4.358775186119601e-05), (14, 5.100432827020995e-05), (15, 7.278654811671004e-05), (16, 7.476961036445573e-05), (17, 7.666532474104315e-05), (18, 0.00013248679169919342), (19, 7.470161654055119e-05), (20, 0.00014780329365748912), (21, 0.0002292837598361075), (22, 0.000116247363621369), (23, 0.0002952535869553685), (24, 0.000565662863664329), (25, 0.001436204300262034), (26, 0.002984263701364398), (27, 0.006880163215100765)], Temporal MSE: [(0, 5.569017957895994e-06), (1, 2.1295222722983453e-06), (2, 4.638078735297313e-06), (3, 6.222097454156028e-06), (4, 4.425183306011604e-06), (5, 8.851500751916319e-06), (6, 2.0823681552428752e-05), (7, 1.8241218640469015e-05), (8, 7.310061846510507e-06), (9, 1.0267232937621884e-05), (10, 1.779156445991248e-05), (11, 2.3750933905830607e-05), (12, 2.0564202714012936e-05), (13, 4.200174589641392e-05), (14, 3.070991442655213e-05), (15, 4.093121242476627e-05), (16, 5.2575531299225986e-05), (17, 3.815590389422141e-05), (18, 5.3425668738782406e-05), (19, 5.4250001994660124e-05), (20, 6.519989256048575e-05), (21, 0.00013456847227644175), (22, 4.932946467306465e-05), (23, 0.00010355579433962703), (24, 0.00021067618217784911), (25, 0.0005478563834913075), (26, 0.0004802905023097992), (27, 7.390342216240242e-05)]
+
+ 48%|████▊     | 24/50 [03:57<03:58,  9.18s/it]Time step 500, Spatial MSE: [(0, 1.3614649105875287e-05), (1, 2.9070701202726923e-05), (2, 0.00013549794675782323), (3, 6.437156116589904e-05), (4, 0.0001768145739333704), (5, 0.0003613198350649327), (6, 0.001512414775788784), (7, 0.00347900390625), (8, 0.003155067563056946), (9, 0.005367831327021122), (10, 0.006026150193065405), (11, 0.006566124968230724), (12, 0.006927494890987873), (13, 0.0077603706158697605), (14, 0.012470321729779243), (15, 0.01699535921216011), (16, 0.026124222204089165), (17, 0.017303936183452606), (18, 0.013783282600343227), (19, 0.008462483994662762), (20, 0.012889101170003414), (21, 0.023296060040593147), (22, 0.007849370129406452), (23, 0.010864321142435074), (24, 0.021527625620365143), (25, 0.03491419926285744), (26, 0.026630723848938942), (27, 0.007323155179619789)], Temporal MSE: [(0, 6.91078776071663e-06), (1, 4.312798864702927e-06), (2, 1.9645076463348232e-05), (3, 0.00020584890444297343), (4, 0.0002278134343214333), (5, 0.0005557470140047371), (6, 0.0015194020234048367), (7, 0.0011645194608718157), (8, 0.0011873225448653102), (9, 0.002218837384134531), (10, 0.002676413394510746), (11, 0.003914450760930777), (12, 0.003953572362661362), (13, 0.007395761087536812), (14, 0.00864518154412508), (15, 0.00887794978916645), (16, 0.009047066792845726), (17, 0.008379090577363968), (18, 0.0075804805383086205), (19, 0.007586550433188677), (20, 0.018322980031371117), (21, 0.010892603546380997), (22, 0.005837845150381327), (23, 0.0076090809889137745), (24, 0.019437722861766815), (25, 0.031025568023324013), (26, 0.015948152169585228), (27, 0.0012891078367829323)]
+
+ 50%|█████     | 25/50 [04:06<03:50,  9.20s/it]Time step 480, Spatial MSE: [(0, 6.462333658419084e-06), (1, 1.7891594325192273e-05), (2, 6.721753743477166e-05), (3, 3.235208714613691e-05), (4, 1.3413386113825254e-05), (5, 1.2186442290840205e-05), (6, 2.396301533735823e-05), (7, 5.734596197726205e-05), (8, 1.4725303117302246e-05), (9, 2.080116973957047e-05), (10, 2.4578148440923542e-05), (11, 2.938913894467987e-05), (12, 3.3676922612357885e-05), (13, 4.513563180807978e-05), (14, 4.603933120961301e-05), (15, 6.769433821318671e-05), (16, 6.959558959351853e-05), (17, 5.016498107579537e-05), (18, 0.0001615601358935237), (19, 6.922871398273855e-05), (20, 0.00010306108015356585), (21, 0.0002483032876625657), (22, 0.00012215129390824586), (23, 0.0002410211891401559), (24, 0.0005702989874407649), (25, 0.0011462015099823475), (26, 0.003205239772796631), (27, 0.007404349744319916)], Temporal MSE: [(0, 4.841465852223337e-06), (1, 2.0249806311767315e-06), (2, 4.346286914369557e-06), (3, 7.314808954106411e-06), (4, 4.808533049072139e-06), (5, 9.311239409726113e-06), (6, 2.0485436834860593e-05), (7, 1.6926243915804662e-05), (8, 7.117121185729047e-06), (9, 1.0531090993026737e-05), (10, 1.638643334445078e-05), (11, 2.2112515580374748e-05), (12, 1.9321199943078682e-05), (13, 4.1899682400980964e-05), (14, 3.030378502444364e-05), (15, 4.0526567318011075e-05), (16, 3.663994721136987e-05), (17, 2.485823461029213e-05), (18, 4.3152173020644113e-05), (19, 4.184115459793247e-05), (20, 4.703653394244611e-05), (21, 0.00013759919966105372), (22, 5.4935619118623435e-05), (23, 9.580405458109453e-05), (24, 0.0002122274017892778), (25, 0.0004939016653224826), (26, 0.0005403601680882275), (27, 9.452100493945181e-05)]
+
+ 52%|█████▏    | 26/50 [04:18<04:03, 10.15s/it]Time step 460, Spatial MSE: [(0, 1.4991237549111247e-05), (1, 2.312373362656217e-05), (2, 0.00010768409265438095), (3, 4.292478479328565e-05), (4, 7.214825018309057e-05), (5, 6.681673403363675e-05), (6, 0.0003696605272125453), (7, 0.00021167441445868462), (8, 0.0006472459062933922), (9, 0.0006377523532137275), (10, 0.0013064879458397627), (11, 0.0019164939876645803), (12, 0.0023753568530082703), (13, 0.0022316910326480865), (14, 0.004061878193169832), (15, 0.0017902503022924066), (16, 0.0010329880751669407), (17, 0.0007841855986043811), (18, 0.0008965164888650179), (19, 0.0005600869189947844), (20, 0.0007006922969594598), (21, 0.0008211263339035213), (22, 0.0008026737486943603), (23, 0.0012590223923325539), (24, 0.003504745662212372), (25, 0.005361427087336779), (26, 0.0114819360896945), (27, 0.007060712203383446)], Temporal MSE: [(0, 5.2947252697777e-06), (1, 2.719343228818616e-06), (2, 1.8580809410195798e-05), (3, 6.38080236967653e-05), (4, 4.054285091115162e-05), (5, 7.154580089263618e-05), (6, 0.00010498272604309022), (7, 0.00013686316378880292), (8, 0.00011930167238460854), (9, 0.00019291421631351113), (10, 0.000328548951074481), (11, 0.0005505569279193878), (12, 0.0006456744740717113), (13, 0.0010903123766183853), (14, 0.001265538390725851), (15, 0.0006844386225566268), (16, 0.00046427821507677436), (17, 0.00035285146441310644), (18, 0.000326905312249437), (19, 0.0002747691178228706), (20, 0.0005284247454255819), (21, 0.0006538934540003538), (22, 0.0005244287895038724), (23, 0.0007464410155080259), (24, 0.0029275931883603334), (25, 0.006708130240440369), (26, 0.0029047883581370115), (27, 0.0004775839624926448)]
+
+ 54%|█████▍    | 27/50 [04:29<03:57, 10.34s/it]Time step 440, Spatial MSE: [(0, 5.760610747529427e-06), (1, 2.3600099666509777e-05), (2, 5.580252764048055e-05), (3, 3.701807145262137e-05), (4, 7.749011274427176e-05), (5, 0.0001815218711271882), (6, 0.00021372712217271328), (7, 0.0005488084279932082), (8, 0.00024723942624405026), (9, 0.00031484858482144773), (10, 0.0004523702373262495), (11, 0.0004416274605318904), (12, 0.0005314724403433502), (13, 0.0036825293209403753), (14, 0.00828063115477562), (15, 0.0029788995161652565), (16, 0.16606974601745605), (17, 0.001095751067623496), (18, 0.001562762539833784), (19, 0.0059126378037035465), (20, 0.0036015009973198175), (21, 0.006136572919785976), (22, 0.002191429026424885), (23, 0.0017366413958370686), (24, 0.003971685189753771), (25, 0.008870230056345463), (26, 0.003062648233026266), (27, 0.020862795412540436)], Temporal MSE: [(0, 4.710833763965638e-06), (1, 3.19956302519131e-06), (2, 2.036921796388924e-05), (3, 0.00019200322276446968), (4, 0.00015347596490755677), (5, 0.00011329808330629021), (6, 0.00013666224549524486), (7, 0.00013776691048406065), (8, 9.230253635905683e-05), (9, 0.0002566987241152674), (10, 0.0004883988294750452), (11, 0.0006110830581746995), (12, 0.0008351036813110113), (13, 0.006479262840002775), (14, 0.010448126122355461), (15, 0.0009755084756761789), (16, 0.0008864220581017435), (17, 0.000565450987778604), (18, 0.000766103039495647), (19, 0.0007139600929804146), (20, 0.0016310277860611677), (21, 0.001395479659549892), (22, 0.001699523301795125), (23, 0.003305031219497323), (24, 0.00603240542113781), (25, 0.006742527242749929), (26, 0.005063652992248535), (27, 0.0014169169589877129)]
+
+ 56%|█████▌    | 28/50 [04:37<03:33,  9.70s/it]Time step 420, Spatial MSE: [(0, 1.42156750371214e-05), (1, 2.4836419470375404e-05), (2, 9.253143070964143e-05), (3, 5.2123868954367936e-05), (4, 0.00011434933549026027), (5, 0.00016931032587308437), (6, 0.00047143222764134407), (7, 0.0006169264670461416), (8, 0.0006003029411658645), (9, 0.0007348659564740956), (10, 0.000790711201261729), (11, 0.0013745584292337298), (12, 0.0016839310992509127), (13, 0.005491914227604866), (14, 0.01225210353732109), (15, 0.005231271032243967), (16, 0.1728731393814087), (17, 0.002552146790549159), (18, 0.002574340673163533), (19, 0.007036833092570305), (20, 0.0035790312103927135), (21, 0.005031404551118612), (22, 0.0038543844129890203), (23, 0.004019422456622124), (24, 0.01010742224752903), (25, 0.01762290857732296), (26, 0.02399221807718277), (27, 0.021407606080174446)], Temporal MSE: [(0, 4.688225544668967e-06), (1, 2.640003231135779e-06), (2, 1.7163118172902614e-05), (3, 4.996974894311279e-05), (4, 9.678592323325574e-05), (5, 0.00016508231055922806), (6, 0.000211295613553375), (7, 0.0002396603231318295), (8, 0.00025716988602653146), (9, 0.00037827531923539937), (10, 0.0004870127886533737), (11, 0.0008918310049921274), (12, 0.0009556778240948915), (13, 0.004686316009610891), (14, 0.010047818534076214), (15, 0.0031441475730389357), (16, 0.004719050135463476), (17, 0.0021558066364377737), (18, 0.001679960172623396), (19, 0.0016994683537632227), (20, 0.007714515086263418), (21, 0.004108164459466934), (22, 0.0022382186725735664), (23, 0.003791602561250329), (24, 0.008154860697686672), (25, 0.013407384045422077), (26, 0.00801563635468483), (27, 0.0012098596198484302)]
+
+ 58%|█████▊    | 29/50 [04:48<03:31, 10.09s/it]Time step 400, Spatial MSE: [(0, 6.161537385196425e-06), (1, 1.938916466315277e-05), (2, 4.150525637669489e-05), (3, 3.144555012113415e-05), (4, 1.1497020750539377e-05), (5, 1.3721581126446836e-05), (6, 2.0667219359893352e-05), (7, 5.5995318689383566e-05), (8, 1.5218912267300766e-05), (9, 1.8190467017120682e-05), (10, 2.376590964559e-05), (11, 2.679726094356738e-05), (12, 3.2261865271721035e-05), (13, 4.683889710577205e-05), (14, 4.2572391976136714e-05), (15, 5.307332321535796e-05), (16, 7.632652705069631e-05), (17, 4.99160960316658e-05), (18, 9.371925989398733e-05), (19, 6.606429815292358e-05), (20, 7.883764192229137e-05), (21, 0.00016352854436263442), (22, 0.0001038572154357098), (23, 0.00022012105910107493), (24, 0.0004133685724809766), (25, 0.0009919269941747189), (26, 0.0030693309381604195), (27, 0.008301777765154839)], Temporal MSE: [(0, 4.1073440115724225e-06), (1, 1.950700834640884e-06), (2, 3.875535639963346e-06), (3, 7.887677384132985e-06), (4, 5.1060737860098016e-06), (5, 1.0516002475924324e-05), (6, 2.0670113372034393e-05), (7, 1.5867837646510452e-05), (8, 7.204781923064729e-06), (9, 1.1812396223831456e-05), (10, 1.5909801732050255e-05), (11, 2.1540829038713127e-05), (12, 2.0952546037733555e-05), (13, 4.3259773519821465e-05), (14, 3.0807343136984855e-05), (15, 3.627781188697554e-05), (16, 3.469089278951287e-05), (17, 2.3924283595988527e-05), (18, 2.396100535406731e-05), (19, 3.610766725614667e-05), (20, 3.7338082620408386e-05), (21, 9.488617070019245e-05), (22, 5.66963899473194e-05), (23, 8.679476741235703e-05), (24, 0.00019541884830687195), (25, 0.00042122669401578605), (26, 0.0005865555722266436), (27, 0.00013339583529159427)]
+
+ 60%|██████    | 30/50 [05:01<03:36, 10.82s/it]Time step 380, Spatial MSE: [(0, 1.4342797840072308e-05), (1, 2.6774905563797802e-05), (2, 6.309279706329107e-05), (3, 5.4348711273632944e-05), (4, 0.00014429127622861415), (5, 0.00040195509791374207), (6, 0.001010275911539793), (7, 0.002212279476225376), (8, 0.0020717813167721033), (9, 0.0031432739924639463), (10, 0.0037374289240688086), (11, 0.004003686830401421), (12, 0.004468489903956652), (13, 0.005377817898988724), (14, 0.007779752369970083), (15, 0.010987941175699234), (16, 0.009814594872295856), (17, 0.009079860523343086), (18, 0.010471213608980179), (19, 0.00862888153642416), (20, 0.010260587558150291), (21, 0.013205626979470253), (22, 0.009623025543987751), (23, 0.012915821745991707), (24, 0.027643198147416115), (25, 0.038306452333927155), (26, 0.02985365502536297), (27, 0.006015332415699959)], Temporal MSE: [(0, 4.466083737497684e-06), (1, 2.8304280021984596e-06), (2, 2.2809381334809586e-05), (3, 0.00016105727991089225), (4, 0.00022733822697773576), (5, 0.0004429554392118007), (6, 0.0009276808123104274), (7, 0.0007937353220768273), (8, 0.0007093052845448256), (9, 0.001365711446851492), (10, 0.0016983188688755035), (11, 0.0028501173947006464), (12, 0.0033642956987023354), (13, 0.006737822201102972), (14, 0.00829388853162527), (15, 0.007419561501592398), (16, 0.007408577483147383), (17, 0.00642986036837101), (18, 0.00552113214507699), (19, 0.0063503761775791645), (20, 0.014125592075288296), (21, 0.010499452240765095), (22, 0.005935864523053169), (23, 0.008275583386421204), (24, 0.020031115040183067), (25, 0.03387029469013214), (26, 0.014802019111812115), (27, 0.0012599267065525055)]
+
+ 62%|██████▏   | 31/50 [05:10<03:15, 10.28s/it]Time step 360, Spatial MSE: [(0, 6.080734237912111e-06), (1, 2.113410118909087e-05), (2, 3.3848118619062006e-05), (3, 3.392207145225257e-05), (4, 1.1553300282685086e-05), (5, 1.670881647442002e-05), (6, 1.8814422219293192e-05), (7, 5.6006480008363724e-05), (8, 1.6424597561126575e-05), (9, 1.8097160136676393e-05), (10, 2.402188147243578e-05), (11, 2.7344171030563302e-05), (12, 3.0991952371550724e-05), (13, 5.6496370234526694e-05), (14, 4.119189907214604e-05), (15, 4.9211070290766656e-05), (16, 8.410558075411245e-05), (17, 4.479238123167306e-05), (18, 0.0001197794044855982), (19, 7.345330959651619e-05), (20, 8.007097494555637e-05), (21, 0.00016215528012253344), (22, 0.00011703606287483126), (23, 0.0002200245507992804), (24, 0.0004386541841086), (25, 0.0009685838595032692), (26, 0.002974913688376546), (27, 0.009244885295629501)], Temporal MSE: [(0, 3.834856670437148e-06), (1, 2.012633785852813e-06), (2, 4.234812422510004e-06), (3, 9.684667020337656e-06), (4, 5.823070750921033e-06), (5, 1.129699739976786e-05), (6, 1.9456783775240183e-05), (7, 1.5845402231207117e-05), (8, 7.851707778172567e-06), (9, 1.3574506738223135e-05), (10, 1.7842114175437018e-05), (11, 2.3342750864685513e-05), (12, 2.389822475379333e-05), (13, 4.8096007958520204e-05), (14, 3.205333268851973e-05), (15, 3.208013004041277e-05), (16, 3.131667835987173e-05), (17, 2.150169893866405e-05), (18, 2.3890668671811e-05), (19, 3.246004052925855e-05), (20, 4.005678783869371e-05), (21, 0.00010163180559175089), (22, 6.925774505361915e-05), (23, 9.805014269659296e-05), (24, 0.00021614917204715312), (25, 0.00044816697482019663), (26, 0.0006600762135349214), (27, 0.00016793949180282652)]
+
+ 64%|██████▍   | 32/50 [05:18<02:54,  9.72s/it]Time step 340, Spatial MSE: [(0, 1.5301780877052806e-05), (1, 2.3715514544164762e-05), (2, 3.761659900192171e-05), (3, 4.49699473392684e-05), (4, 5.8221325161866844e-05), (5, 8.681399776833132e-05), (6, 0.00032466629636473954), (7, 0.0002489292237441987), (8, 0.0005802283994853497), (9, 0.0005759051418863237), (10, 0.0010312087833881378), (11, 0.0017937690718099475), (12, 0.002017866587266326), (13, 0.0020989966578781605), (14, 0.004515651147812605), (15, 0.002052766503766179), (16, 0.0012664784444496036), (17, 0.0009538183803670108), (18, 0.0010188017040491104), (19, 0.0008142923470586538), (20, 0.0012342662084847689), (21, 0.001276161172427237), (22, 0.0007836524746380746), (23, 0.0011925935978069901), (24, 0.0038919923827052116), (25, 0.006865968927741051), (26, 0.01704397052526474), (27, 0.006425101775676012)], Temporal MSE: [(0, 4.5595843403134495e-06), (1, 2.57442570728017e-06), (2, 2.1169566025491804e-05), (3, 7.241885759867728e-05), (4, 4.963681203662418e-05), (5, 9.28077643038705e-05), (6, 0.0001668177283136174), (7, 0.00016485816740896553), (8, 0.00012356969818938524), (9, 0.00020241791207809), (10, 0.00035989179741591215), (11, 0.0006366532179526985), (12, 0.0007244960288517177), (13, 0.0012098606675863266), (14, 0.0015815540682524443), (15, 0.0008830713923089206), (16, 0.000554461614228785), (17, 0.00045973528176546097), (18, 0.00041819087346084416), (19, 0.0005003584083169699), (20, 0.0011717203306034207), (21, 0.0009156166343018413), (22, 0.0005285270162858069), (23, 0.0007267321343533695), (24, 0.002429170534014702), (25, 0.0069760968908667564), (26, 0.0030952407978475094), (27, 0.0007452130084857345)]
+
+ 66%|██████▌   | 33/50 [05:30<02:54, 10.24s/it]Time step 320, Spatial MSE: [(0, 5.887966835871339e-06), (1, 2.4790862880763598e-05), (2, 3.0159417292452417e-05), (3, 3.432048833929002e-05), (4, 0.00010124441905645654), (5, 0.00025304200244136155), (6, 0.0003852658555842936), (7, 0.0007395032444037497), (8, 0.00036738597555086017), (9, 0.00036882233689539135), (10, 0.0005436844658106565), (11, 0.0005015981150791049), (12, 0.0006047491915524006), (13, 0.002272726036608219), (14, 0.003066651988774538), (15, 0.003169373841956258), (16, 0.04726067930459976), (17, 0.001417858642525971), (18, 0.001529743429273367), (19, 0.002363750711083412), (20, 0.003190484829246998), (21, 0.004412045702338219), (22, 0.0013397015864029527), (23, 0.0012428174959495664), (24, 0.0029464485123753548), (25, 0.006429255474358797), (26, 0.003129621036350727), (27, 0.013264973647892475)], Temporal MSE: [(0, 4.156231625529472e-06), (1, 2.832488462445326e-06), (2, 2.283428875671234e-05), (3, 0.0002525248273741454), (4, 0.0002017555380007252), (5, 0.00016410411626566201), (6, 0.0002032416668953374), (7, 0.00016998624778352678), (8, 0.00011577521945582703), (9, 0.00032015502802096307), (10, 0.0005066977464593947), (11, 0.0006671443115919828), (12, 0.0010097394697368145), (13, 0.0027613425627350807), (14, 0.003622946562245488), (15, 0.0010493802838027477), (16, 0.000792861741501838), (17, 0.0005918622482568026), (18, 0.0006562357302755117), (19, 0.0007535354234278202), (20, 0.001709828618913889), (21, 0.001144460286013782), (22, 0.0010252678766846657), (23, 0.0015223039081320167), (24, 0.0034974967129528522), (25, 0.005217918194830418), (26, 0.0045398869551718235), (27, 0.0013857628218829632)]
+
+ 68%|██████▊   | 34/50 [05:42<02:54, 10.92s/it]Time step 300, Spatial MSE: [(0, 1.4160934370011091e-05), (1, 2.3766579033690505e-05), (2, 3.1712745112599805e-05), (3, 4.657805766328238e-05), (4, 0.00013648578897118568), (5, 0.0001885762030724436), (6, 0.00034366780892014503), (7, 0.0005632698303088546), (8, 0.0005629993393085897), (9, 0.0007228727918118238), (10, 0.000763413670938462), (11, 0.0011394319590181112), (12, 0.0013276268728077412), (13, 0.003335395362228155), (14, 0.006040256004780531), (15, 0.004210560582578182), (16, 0.049476467072963715), (17, 0.0025829714722931385), (18, 0.00284209824167192), (19, 0.004266810603439808), (20, 0.003707967232912779), (21, 0.00411990424618125), (22, 0.002637405414134264), (23, 0.003670772770419717), (24, 0.009147875010967255), (25, 0.018166009336709976), (26, 0.02893194928765297), (27, 0.017368711531162262)], Temporal MSE: [(0, 4.977347998647019e-06), (1, 2.6580078156257514e-06), (2, 1.992387478821911e-05), (3, 4.5332290028454736e-05), (4, 0.00011146088218083605), (5, 0.00018178632308263332), (6, 0.00021003882284276187), (7, 0.00023364332446362823), (8, 0.00024636538000777364), (9, 0.00039151887176558375), (10, 0.0005588229396380484), (11, 0.0009321221732534468), (12, 0.0010765183251351118), (13, 0.0022727963514626026), (14, 0.004065323155373335), (15, 0.002402956597507), (16, 0.0028639386873692274), (17, 0.0016923571238294244), (18, 0.0013137701898813248), (19, 0.0013602420222014189), (20, 0.005332437343895435), (21, 0.0036888262256979942), (22, 0.0017498640809208155), (23, 0.002420844277366996), (24, 0.007204040884971619), (25, 0.016537735238671303), (26, 0.007086085621267557), (27, 0.0008426959975622594)]
+
+ 70%|███████   | 35/50 [05:51<02:33, 10.23s/it]Time step 280, Spatial MSE: [(0, 6.234715328901075e-06), (1, 2.265978037030436e-05), (2, 3.1235787901096046e-05), (3, 3.274560731370002e-05), (4, 1.1810898286057636e-05), (5, 2.2522650397149846e-05), (6, 2.0525454601738602e-05), (7, 6.005535396980122e-05), (8, 2.064480577246286e-05), (9, 2.03140498342691e-05), (10, 2.7787451472249813e-05), (11, 3.163744622725062e-05), (12, 3.505289714667015e-05), (13, 7.492584700230509e-05), (14, 6.38364435872063e-05), (15, 7.454407750628889e-05), (16, 0.0019749479833990335), (17, 5.159290958545171e-05), (18, 0.00015143357450142503), (19, 0.0001526740670669824), (20, 8.745806553633884e-05), (21, 0.00025323688169009984), (22, 0.00016454362776130438), (23, 0.00024340178060811013), (24, 0.0005125040770508349), (25, 0.00093222432769835), (26, 0.0026766385417431593), (27, 0.010967114008963108)], Temporal MSE: [(0, 4.22638231611927e-06), (1, 2.429809683235362e-06), (2, 4.850654477195349e-06), (3, 1.0218308489129413e-05), (4, 6.826013304817025e-06), (5, 1.417110161128221e-05), (6, 2.3772678105160594e-05), (7, 1.7718251910991967e-05), (8, 1.0290729733242188e-05), (9, 1.7589649360161275e-05), (10, 2.28688440984115e-05), (11, 2.8072739951312542e-05), (12, 3.175057645421475e-05), (13, 7.493408338632435e-05), (14, 8.258604793809354e-05), (15, 7.613781781401485e-05), (16, 9.72141424426809e-05), (17, 5.321813659975305e-05), (18, 4.683159932028502e-05), (19, 5.589910506387241e-05), (20, 0.00016561921802349389), (21, 0.00015584893117193133), (22, 0.00010762042802525684), (23, 0.00015053601237013936), (24, 0.0002788329729810357), (25, 0.00047540091327391565), (26, 0.000698910269420594), (27, 0.000275645696092397)]
+
+ 72%|███████▏  | 36/50 [06:04<02:35, 11.11s/it]Time step 260, Spatial MSE: [(0, 1.3741660040977877e-05), (1, 2.6564490326563828e-05), (2, 6.428956112358719e-05), (3, 5.692221748176962e-05), (4, 0.00013451573613565415), (5, 0.00044531348976306617), (6, 0.0009285074775107205), (7, 0.0019834658596664667), (8, 0.001922896597534418), (9, 0.0030755968764424324), (10, 0.0033697113394737244), (11, 0.004370835144072771), (12, 0.0041682785376906395), (13, 0.004967823158949614), (14, 0.007329876068979502), (15, 0.011852608062326908), (16, 0.012425786815583706), (17, 0.010064133442938328), (18, 0.01121438480913639), (19, 0.009802854619920254), (20, 0.01167331263422966), (21, 0.013767759315669537), (22, 0.009613617323338985), (23, 0.0134981544688344), (24, 0.029584161937236786), (25, 0.04446645453572273), (26, 0.03313087299466133), (27, 0.006496441550552845)], Temporal MSE: [(0, 5.616372163785854e-06), (1, 3.13510327032418e-06), (2, 3.525233478285372e-05), (3, 0.00017710725660435855), (4, 0.00019438008894212544), (5, 0.0003688906435854733), (6, 0.00053267867770046), (7, 0.0006784374709241092), (8, 0.0007013553404249251), (9, 0.0012864824384450912), (10, 0.001730264280922711), (11, 0.0029727588407695293), (12, 0.0035380953922867775), (13, 0.006535546854138374), (14, 0.00881451740860939), (15, 0.007597794756293297), (16, 0.006455182563513517), (17, 0.00640392629429698), (18, 0.005510127637535334), (19, 0.0069132354110479355), (20, 0.014560606330633163), (21, 0.012577205896377563), (22, 0.006319466978311539), (23, 0.009421140886843204), (24, 0.023883001878857613), (25, 0.041491225361824036), (26, 0.015688994899392128), (27, 0.0016822160687297583)]
+
+ 74%|███████▍  | 37/50 [06:15<02:25, 11.22s/it]Time step 240, Spatial MSE: [(0, 6.717181804560823e-06), (1, 2.6213776436634362e-05), (2, 4.368957888800651e-05), (3, 3.583285433705896e-05), (4, 1.425821210432332e-05), (5, 2.642177787492983e-05), (6, 2.1834062863490544e-05), (7, 5.4963089496595785e-05), (8, 2.6261768653057516e-05), (9, 2.3170403437688947e-05), (10, 3.042678326892201e-05), (11, 3.788462345255539e-05), (12, 4.0349965274799615e-05), (13, 7.847457163734362e-05), (14, 5.3703595767728984e-05), (15, 6.983696221141145e-05), (16, 0.00034211340243928134), (17, 6.421929720090702e-05), (18, 0.0001392550184391439), (19, 0.0001163242050097324), (20, 9.984110511140898e-05), (21, 0.00021814045612700284), (22, 0.00016031268751248717), (23, 0.0002704637008719146), (24, 0.0005304606165736914), (25, 0.0009992090053856373), (26, 0.002715140115469694), (27, 0.012466040439903736)], Temporal MSE: [(0, 5.016928298573475e-06), (1, 2.9615180210385006e-06), (2, 5.9631402109516785e-06), (3, 1.3630851753987372e-05), (4, 8.89873263076879e-06), (5, 1.688672819000203e-05), (6, 2.4319315343745984e-05), (7, 1.9028682800126262e-05), (8, 1.3116437912685797e-05), (9, 2.2102249204181135e-05), (10, 2.8610453227884136e-05), (11, 3.40806545864325e-05), (12, 4.0619757783133537e-05), (13, 6.167070387164131e-05), (14, 4.4283850002102554e-05), (15, 5.418964792625047e-05), (16, 4.639819235308096e-05), (17, 3.7679867091355845e-05), (18, 4.135711787967011e-05), (19, 5.487696034833789e-05), (20, 8.156519470503554e-05), (21, 0.0001236385287484154), (22, 0.00010224981815554202), (23, 0.00014369492419064045), (24, 0.0002957988472189754), (25, 0.0005565264727920294), (26, 0.0008167100022546947), (27, 0.00035608047619462013)]
+
+ 76%|███████▌  | 38/50 [06:28<02:19, 11.60s/it]Time step 220, Spatial MSE: [(0, 1.343660187558271e-05), (1, 2.6426665499457158e-05), (2, 9.251922892872244e-05), (3, 5.140693610883318e-05), (4, 4.644772343453951e-05), (5, 0.00013112190936226398), (6, 0.00032579241087660193), (7, 0.00040701639954932034), (8, 0.0006991410627961159), (9, 0.0007304573082365096), (10, 0.0011366462567821145), (11, 0.002003462053835392), (12, 0.002299062442034483), (13, 0.0025132622104138136), (14, 0.005952836945652962), (15, 0.0027260270435363054), (16, 0.0019843035843223333), (17, 0.0013019070029258728), (18, 0.0013823519693687558), (19, 0.0012742073740810156), (20, 0.0020938885863870382), (21, 0.0018989646341651678), (22, 0.0009719358058646321), (23, 0.001661621150560677), (24, 0.005775297526270151), (25, 0.013946743682026863), (26, 0.025407124310731888), (27, 0.007475482765585184)], Temporal MSE: [(0, 6.800148184993304e-06), (1, 3.529004288793658e-06), (2, 2.3018224965198897e-05), (3, 6.697644130326807e-05), (4, 5.4720265325158834e-05), (5, 0.00013918693002779037), (6, 0.00021576731523964554), (7, 0.0002506823802832514), (8, 0.0001954283652594313), (9, 0.0002922146231867373), (10, 0.0005167410126887262), (11, 0.0008411705493927002), (12, 0.0010161739774048328), (13, 0.0015870920615270734), (14, 0.0020464928820729256), (15, 0.0011693467386066914), (16, 0.0007418824243359268), (17, 0.0006917673745192587), (18, 0.0006817001849412918), (19, 0.000803454197011888), (20, 0.0020411228761076927), (21, 0.0014335375744849443), (22, 0.0007410533144138753), (23, 0.0010043695801869035), (24, 0.002925262786448002), (25, 0.009095498360693455), (26, 0.003826590022072196), (27, 0.0011672829277813435)]
+
+ 78%|███████▊  | 39/50 [06:41<02:12, 12.06s/it]Time step 200, Spatial MSE: [(0, 7.287880634976318e-06), (1, 3.109341923845932e-05), (2, 8.580230496590957e-05), (3, 3.7930709368083626e-05), (4, 8.82309177541174e-05), (5, 0.00024373775522690266), (6, 0.00058260268997401), (7, 0.0009418490226380527), (8, 0.0005524139269255102), (9, 0.0004838938475586474), (10, 0.000700255623087287), (11, 0.0006110096583142877), (12, 0.0007216560770757496), (13, 0.0010926331160590053), (14, 0.0018550867680460215), (15, 0.004024344962090254), (16, 0.019607815891504288), (17, 0.0020206496119499207), (18, 0.002034480683505535), (19, 0.0015671709552407265), (20, 0.003451974131166935), (21, 0.00480281887575984), (22, 0.001539222546853125), (23, 0.0014461693353950977), (24, 0.0034242563415318727), (25, 0.00707831047475338), (26, 0.004097079858183861), (27, 0.011791082099080086)], Temporal MSE: [(0, 6.863314411020838e-06), (1, 4.129702119826106e-06), (2, 2.3437945856130682e-05), (3, 0.00020438966748770326), (4, 0.0001633390347706154), (5, 0.00017147959442809224), (6, 0.000304145592963323), (7, 0.00018847039609681815), (8, 0.00015399792755488306), (9, 0.0004820608301088214), (10, 0.000603588530793786), (11, 0.0008298574830405414), (12, 0.0012508452637121081), (13, 0.002231212565675378), (14, 0.0028243910055607557), (15, 0.0013587749563157558), (16, 0.0008870859164744616), (17, 0.0007570816669613123), (18, 0.0006977043813094497), (19, 0.0009955839486792684), (20, 0.002778426744043827), (21, 0.0015009662602096796), (22, 0.0009764658752828836), (23, 0.0013522575609385967), (24, 0.0037844362668693066), (25, 0.005238659679889679), (26, 0.004911669529974461), (27, 0.0021695271134376526)]
+
+ 80%|████████  | 40/50 [06:54<02:03, 12.31s/it]Time step 180, Spatial MSE: [(0, 1.3002495506952982e-05), (1, 3.186706453561783e-05), (2, 0.0002024247223744169), (3, 5.4030566388973966e-05), (4, 0.00012751799658872187), (5, 0.00018890437786467373), (6, 0.0003297640068922192), (7, 0.0005071017658337951), (8, 0.0006029143114574254), (9, 0.0006861722795292735), (10, 0.0007966785924509168), (11, 0.0010173830669373274), (12, 0.001207616413012147), (13, 0.0019375561969354749), (14, 0.0040929620154201984), (15, 0.004634925629943609), (16, 0.020430870354175568), (17, 0.0032894136384129524), (18, 0.0041228970512747765), (19, 0.004502324853092432), (20, 0.00594114325940609), (21, 0.00585198262706399), (22, 0.003549390472471714), (23, 0.005194838158786297), (24, 0.014323340728878975), (25, 0.032171521335840225), (26, 0.043408654630184174), (27, 0.017225997522473335)], Temporal MSE: [(0, 8.743027137825266e-06), (1, 4.444255864655133e-06), (2, 3.147474126308225e-05), (3, 5.42532216059044e-05), (4, 0.0001108154610847123), (5, 0.00020048135775141418), (6, 0.00022711341443937272), (7, 0.0002670598914846778), (8, 0.00025005859788507223), (9, 0.00042778506758622825), (10, 0.0006848911289125681), (11, 0.0011036243522539735), (12, 0.0012718891957774758), (13, 0.0022673960775136948), (14, 0.0034878014121204615), (15, 0.0026434154715389013), (16, 0.0027955775149166584), (17, 0.0020250515080988407), (18, 0.001691783545538783), (19, 0.0018315714551135898), (20, 0.007266199216246605), (21, 0.00615005474537611), (22, 0.0022348915226757526), (23, 0.0028917856980115175), (24, 0.009099256247282028), (25, 0.02083868719637394), (26, 0.008090381510555744), (27, 0.000826076720841229)]
+
+ 82%|████████▏ | 41/50 [07:06<01:49, 12.16s/it]Time step 160, Spatial MSE: [(0, 8.6169520727708e-06), (1, 3.8786751247243956e-05), (2, 0.00011613685637712479), (3, 3.8349437090801075e-05), (4, 2.1526811906369403e-05), (5, 3.6654815630754456e-05), (6, 2.9823710065102205e-05), (7, 0.00010114121687365696), (8, 4.7135101340245456e-05), (9, 3.6399509554030374e-05), (10, 4.742449164041318e-05), (11, 6.525382923427969e-05), (12, 6.597107858397067e-05), (13, 0.0001280746073462069), (14, 8.287563105113804e-05), (15, 0.00010587641736492515), (16, 0.00016961497021839023), (17, 0.00011058148083975539), (18, 0.00014689029194414616), (19, 0.00017967760504689068), (20, 0.00017739272152539343), (21, 0.00020051421597599983), (22, 0.0002711020933929831), (23, 0.0004537642525974661), (24, 0.0007965004188008606), (25, 0.0013672271743416786), (26, 0.0027472444344311953), (27, 0.015990793704986572)], Temporal MSE: [(0, 8.419920959568117e-06), (1, 4.6894647312001325e-06), (2, 8.811875886749476e-06), (3, 1.5667588741052896e-05), (4, 1.3072039109829348e-05), (5, 2.5245266442652792e-05), (6, 3.532263872330077e-05), (7, 3.543479760992341e-05), (8, 2.545330789871514e-05), (9, 4.006997187389061e-05), (10, 5.806388799101114e-05), (11, 6.318992382148281e-05), (12, 7.369219383690506e-05), (13, 0.00010241434938507155), (14, 7.344047480728477e-05), (15, 9.0766217908822e-05), (16, 8.027124567888677e-05), (17, 6.444384780479595e-05), (18, 6.522452895296738e-05), (19, 8.646171045256779e-05), (20, 0.00013851279800292104), (21, 0.0001507051201770082), (22, 0.000161853720783256), (23, 0.0002325437671970576), (24, 0.00043112694402225316), (25, 0.0007682446739636362), (26, 0.001042418647557497), (27, 0.0006590812699869275)]
+
+ 84%|████████▍ | 42/50 [07:15<01:31, 11.39s/it]Time step 140, Spatial MSE: [(0, 1.301257452723803e-05), (1, 3.888695209752768e-05), (2, 0.0004761779273394495), (3, 7.651271153008565e-05), (4, 0.00019341617007739842), (5, 0.0004863742215093225), (6, 0.0007994979969225824), (7, 0.0023596605751663446), (8, 0.002664398169144988), (9, 0.003409077413380146), (10, 0.003551599569618702), (11, 0.004701836034655571), (12, 0.004823267925530672), (13, 0.005536995828151703), (14, 0.008114639669656754), (15, 0.015884334221482277), (16, 0.013263052329421043), (17, 0.014001571573317051), (18, 0.017683789134025574), (19, 0.014252157881855965), (20, 0.017904331907629967), (21, 0.018749810755252838), (22, 0.013483730144798756), (23, 0.01908944733440876), (24, 0.04392930120229721), (25, 0.06804373860359192), (26, 0.05114675685763359), (27, 0.008830126374959946)], Temporal MSE: [(0, 1.1841983905469533e-05), (1, 6.504208613478113e-06), (2, 5.233641422819346e-05), (3, 0.00020226459309924394), (4, 0.00022438264568336308), (5, 0.00034259181120432913), (6, 0.0006301209796220064), (7, 0.00093028973788023), (8, 0.000879587372764945), (9, 0.0018640743801370263), (10, 0.002356192795559764), (11, 0.004246399272233248), (12, 0.004399130120873451), (13, 0.007654801942408085), (14, 0.011744705028831959), (15, 0.010431255213916302), (16, 0.007910766638815403), (17, 0.008042428642511368), (18, 0.0070947930216789246), (19, 0.008703100495040417), (20, 0.019135745242238045), (21, 0.014379744417965412), (22, 0.008361130021512508), (23, 0.01139433030039072), (24, 0.029940208420157433), (25, 0.04522557929158211), (26, 0.020282499492168427), (27, 0.0029306309297680855)]
+
+ 86%|████████▌ | 43/50 [07:25<01:16, 10.99s/it]Time step 120, Spatial MSE: [(0, 1.1260674909863155e-05), (1, 5.61300985282287e-05), (2, 0.00021230339189060032), (3, 5.035486537963152e-05), (4, 3.226926855859347e-05), (5, 5.6479319027857855e-05), (6, 5.2213570597814396e-05), (7, 0.0001074639440048486), (8, 7.8702432801947e-05), (9, 5.4759631893830374e-05), (10, 7.52326741348952e-05), (11, 0.00010264443699270487), (12, 9.866329492069781e-05), (13, 0.00018829556938726455), (14, 0.00011770107812481001), (15, 0.0001779940794222057), (16, 0.0002241530455648899), (17, 0.00017211149679496884), (18, 0.00023432174930348992), (19, 0.00027540663722902536), (20, 0.0002537937252782285), (21, 0.0004203159478493035), (22, 0.00040901685133576393), (23, 0.0007189624011516571), (24, 0.001232149195857346), (25, 0.0019959802739322186), (26, 0.0031387873459607363), (27, 0.019420815631747246)], Temporal MSE: [(0, 1.1829129107354674e-05), (1, 6.434448550862726e-06), (2, 1.330161921941908e-05), (3, 2.4188297174987383e-05), (4, 2.104764098476153e-05), (5, 4.0350827475776896e-05), (6, 5.359403803595342e-05), (7, 4.944389365846291e-05), (8, 4.358428850537166e-05), (9, 6.506847421405837e-05), (10, 0.00010263251169817522), (11, 0.00010442127677379176), (12, 0.00011886123684234917), (13, 0.00014485167048405856), (14, 0.00011264056956861168), (15, 0.00016366616182494909), (16, 0.00011711494880728424), (17, 0.00010200272663496435), (18, 0.00010832634870894253), (19, 0.00015387829625979066), (20, 0.000242657057242468), (21, 0.00029790468397550285), (22, 0.00023682315077167004), (23, 0.00034364976454526186), (24, 0.000685826176777482), (25, 0.0011521223932504654), (26, 0.001517393160611391), (27, 0.0010390955721959472)]
+
+ 88%|████████▊ | 44/50 [07:35<01:03, 10.66s/it]Time step 100, Spatial MSE: [(0, 1.5129183339013252e-05), (1, 5.879783566342667e-05), (2, 0.0008342535002157092), (3, 8.285828516818583e-05), (4, 0.00012379573308862746), (5, 0.00023054091434460133), (6, 0.0007287953048944473), (7, 0.001444635447114706), (8, 0.0018326507415622473), (9, 0.002014057943597436), (10, 0.0022275191731750965), (11, 0.00276862527243793), (12, 0.003213093848899007), (13, 0.0036845318973064423), (14, 0.006026492454111576), (15, 0.009213055483996868), (16, 0.007007586304098368), (17, 0.006839653477072716), (18, 0.007954672910273075), (19, 0.006640097126364708), (20, 0.010398750193417072), (21, 0.009637800045311451), (22, 0.006657106336206198), (23, 0.010664510540664196), (24, 0.023124726489186287), (25, 0.04756809026002884), (26, 0.05963095650076866), (27, 0.01365873496979475)], Temporal MSE: [(0, 1.6801661331555806e-05), (1, 8.96644723979989e-06), (2, 3.535323776304722e-05), (3, 6.613331788685173e-05), (4, 7.522350642830133e-05), (5, 0.00012646624236367643), (6, 0.00021093219402246177), (7, 0.0002750323910731822), (8, 0.000308984104776755), (9, 0.0007191531476564705), (10, 0.0009093137923628092), (11, 0.0017698766896501184), (12, 0.0019032808486372232), (13, 0.003198727034032345), (14, 0.003817823715507984), (15, 0.003916177432984114), (16, 0.002801088383421302), (17, 0.0030239855404943228), (18, 0.0033720338251441717), (19, 0.0032012725714594126), (20, 0.005668526515364647), (21, 0.0047872187569737434), (22, 0.0039538186974823475), (23, 0.005950360558927059), (24, 0.012135248631238937), (25, 0.015772219747304916), (26, 0.010879349894821644), (27, 0.001816901727579534)]
+
+ 90%|█████████ | 45/50 [07:47<00:54, 10.89s/it]Time step 80, Spatial MSE: [(0, 1.5899378922767937e-05), (1, 8.177475683623925e-05), (2, 0.0007254783413372934), (3, 5.858984513906762e-05), (4, 8.610470104031265e-05), (5, 0.00014733243733644485), (6, 0.0004099154903087765), (7, 0.0007184066926129162), (8, 0.0008959832484833896), (9, 0.0008823586977086961), (10, 0.001042691757902503), (11, 0.001240835408680141), (12, 0.0013236278900876641), (13, 0.0015538886655122042), (14, 0.002520043635740876), (15, 0.004163699224591255), (16, 0.003037200076505542), (17, 0.0033219922333955765), (18, 0.003822210244834423), (19, 0.0031981728971004486), (20, 0.004803345538675785), (21, 0.004967494402080774), (22, 0.0034952135756611824), (23, 0.006234081462025642), (24, 0.011870164424180984), (25, 0.0229784045368433), (26, 0.027334021404385567), (27, 0.018711071461439133)], Temporal MSE: [(0, 1.9371187590877526e-05), (1, 1.0108666174346581e-05), (2, 2.05401247512782e-05), (3, 3.9620976167498156e-05), (4, 4.0736907976679504e-05), (5, 6.353948265314102e-05), (6, 0.0001221799902850762), (7, 0.00012458737182896584), (8, 0.00015988612722139806), (9, 0.00032462587114423513), (10, 0.0003994510043412447), (11, 0.000727372826077044), (12, 0.0008473218185827136), (13, 0.0012313365004956722), (14, 0.0016320685390383005), (15, 0.0016919720219448209), (16, 0.0012707416899502277), (17, 0.001337173511274159), (18, 0.0014811636647209525), (19, 0.001580799464136362), (20, 0.002886801026761532), (21, 0.002347806002944708), (22, 0.0019493498839437962), (23, 0.00307521503418684), (24, 0.005861551500856876), (25, 0.007689621765166521), (26, 0.005706355907022953), (27, 0.0013299128040671349)]
+
+ 92%|█████████▏| 46/50 [07:57<00:42, 10.67s/it]Time step 60, Spatial MSE: [(0, 2.1079618818475865e-05), (1, 0.00011458758672233671), (2, 0.0010237792739644647), (3, 8.101553248707205e-05), (4, 0.0001375135179841891), (5, 0.0001992769684875384), (6, 0.0006562696071341634), (7, 0.00104077800642699), (8, 0.0012800395488739014), (9, 0.0013725635362789035), (10, 0.0014895638450980186), (11, 0.0016387253999710083), (12, 0.001743018627166748), (13, 0.002013410674408078), (14, 0.0033040291164070368), (15, 0.0052644130773842335), (16, 0.004116730764508247), (17, 0.004347872454673052), (18, 0.004948850721120834), (19, 0.004009020980447531), (20, 0.005977586377412081), (21, 0.005830318666994572), (22, 0.004638793878257275), (23, 0.008906620554625988), (24, 0.016323592513799667), (25, 0.031432244926691055), (26, 0.03506368026137352), (27, 0.02073380909860134)], Temporal MSE: [(0, 2.6061727112391964e-05), (1, 1.358381996396929e-05), (2, 2.9733126211795025e-05), (3, 5.088598118163645e-05), (4, 5.8771893236553296e-05), (5, 8.890601748134941e-05), (6, 0.00018246640684083104), (7, 0.00018566931248642504), (8, 0.00025773493689484894), (9, 0.00047279096907004714), (10, 0.0005239846650511026), (11, 0.000873049721121788), (12, 0.0010403210762888193), (13, 0.0014916211366653442), (14, 0.002097379881888628), (15, 0.0021313936449587345), (16, 0.0015844593290239573), (17, 0.0016154716722667217), (18, 0.0018059915164485574), (19, 0.0020562126301229), (20, 0.004113038070499897), (21, 0.0031147960107773542), (22, 0.002432778710499406), (23, 0.003949045669287443), (24, 0.007658080197870731), (25, 0.010145125910639763), (26, 0.008494112640619278), (27, 0.0024984069168567657)]
+
+ 94%|█████████▍| 47/50 [08:08<00:32, 10.97s/it]Time step 40, Spatial MSE: [(0, 3.1406223570229486e-05), (1, 0.0001755691773723811), (2, 0.0014867871068418026), (3, 0.00011711778643075377), (4, 0.0002465669240336865), (5, 0.0003022462478838861), (6, 0.0012130594113841653), (7, 0.001663689035922289), (8, 0.00204683979973197), (9, 0.0023368129041045904), (10, 0.002347287954762578), (11, 0.0022566881962120533), (12, 0.0022898998577147722), (13, 0.0026921515818685293), (14, 0.004553768318146467), (15, 0.007184176705777645), (16, 0.00564558245241642), (17, 0.0059098657220602036), (18, 0.006488041952252388), (19, 0.005202104337513447), (20, 0.007858957163989544), (21, 0.007995035499334335), (22, 0.006405171472579241), (23, 0.01304591353982687), (24, 0.02404315397143364), (25, 0.0446954220533371), (26, 0.048841703683137894), (27, 0.02521517500281334)], Temporal MSE: [(0, 3.821618520305492e-05), (1, 1.9605735360528342e-05), (2, 5.0923656090162694e-05), (3, 7.041383651085198e-05), (4, 9.824371954891831e-05), (5, 0.0001423587673343718), (6, 0.0003199068014509976), (7, 0.0003174892335664481), (8, 0.0004787564976140857), (9, 0.0007612109766341746), (10, 0.0007332057575695217), (11, 0.00108932179864496), (12, 0.0013202030677348375), (13, 0.0018852847861126065), (14, 0.002877155551686883), (15, 0.002716843970119953), (16, 0.0019448755774646997), (17, 0.0019565101247280836), (18, 0.0021884189918637276), (19, 0.0027901087887585163), (20, 0.005926819518208504), (21, 0.004471435211598873), (22, 0.003206954337656498), (23, 0.005426324438303709), (24, 0.010903719812631607), (25, 0.014900367707014084), (26, 0.014202752150595188), (27, 0.007077006623148918)]
+
+ 96%|█████████▌| 48/50 [08:17<00:20, 10.23s/it]Time step 20, Spatial MSE: [(0, 5.958143810858019e-05), (1, 0.00033414573408663273), (2, 0.002457085531204939), (3, 0.0002036023506661877), (4, 0.0005116269458085299), (5, 0.0005823729443363845), (6, 0.002882093656808138), (7, 0.0035527460277080536), (8, 0.00419303122907877), (9, 0.005259451922029257), (10, 0.004555115010589361), (11, 0.0035978895612061024), (12, 0.003392901737242937), (13, 0.0041620912961661816), (14, 0.007433250546455383), (15, 0.011148560792207718), (16, 0.008939623832702637), (17, 0.009688816033303738), (18, 0.009822516702115536), (19, 0.008105392567813396), (20, 0.012811566703021526), (21, 0.01353510096669197), (22, 0.010354251600801945), (23, 0.02156662382185459), (24, 0.042242612689733505), (25, 0.07151219248771667), (26, 0.0775667205452919), (27, 0.03206884488463402)], Temporal MSE: [(0, 6.868795026093721e-05), (1, 3.434641985222697e-05), (2, 0.00010830927203642204), (3, 0.00012323202099651098), (4, 0.00020716883591376245), (5, 0.0002973879163619131), (6, 0.0008032924379222095), (7, 0.0007482033688575029), (8, 0.0012497089337557554), (9, 0.0015657182084396482), (10, 0.0012620787601917982), (11, 0.0015877904370427132), (12, 0.0019504042575135827), (13, 0.0028610092122107744), (14, 0.004584381822496653), (15, 0.004082309547811747), (16, 0.002781579503789544), (17, 0.0027649179100990295), (18, 0.003305333899334073), (19, 0.004053644835948944), (20, 0.008300881832838058), (21, 0.006818373687565327), (22, 0.005282436963170767), (23, 0.009333889000117779), (24, 0.020252959802746773), (25, 0.028511634096503258), (26, 0.031508974730968475), (27, 0.04421958699822426)]
+
+ 98%|█████████▊| 49/50 [08:25<00:09,  9.72s/it]Time step 0, Spatial MSE: [(0, 0.0004437692114152014), (1, 0.001823050668463111), (2, 0.009987913072109222), (3, 0.001073867199011147), (4, 0.0026034414768218994), (5, 0.0033151612151414156), (6, 0.013162780553102493), (7, 0.020680030807852745), (8, 0.021948950365185738), (9, 0.03260108828544617), (10, 0.021477119997143745), (11, 0.012016904540359974), (12, 0.010250348597764969), (13, 0.01205455418676138), (14, 0.02359689399600029), (15, 0.03611163794994354), (16, 0.034075818955898285), (17, 0.03814941644668579), (18, 0.028573207557201385), (19, 0.031507719308137894), (20, 0.0497618168592453), (21, 0.04387305676937103), (22, 0.04339436814188957), (23, 0.07097747176885605), (24, 0.1758067011833191), (25, 0.5204743146896362), (26, 0.7333755493164062), (27, 1.4717851877212524)], Temporal MSE: [(0, 0.0005450269673019648), (1, 0.00021251877478789538), (2, 0.0007613196503371), (3, 0.000704410660546273), (4, 0.0013754263054579496), (5, 0.001876775175333023), (6, 0.008136063814163208), (7, 0.0064727505668997765), (8, 0.012423049658536911), (9, 0.009961452335119247), (10, 0.008531253784894943), (11, 0.005842977203428745), (12, 0.006600809749215841), (13, 0.008433195762336254), (14, 0.012236122041940689), (15, 0.011882961727678776), (16, 0.009128953330218792), (17, 0.008003069087862968), (18, 0.012378969229757786), (19, 0.014968623407185078), (20, 0.031553834676742554), (21, 0.01905292645096779), (22, 0.022697601467370987), (23, 0.0368245467543602), (24, 0.10991738736629486), (25, 0.20397897064685822), (26, 0.4293760657310486), (27, 4.145165920257568)]
+
+100%|██████████| 50/50 [08:34<00:00,  9.45s/it]
+100%|██████████| 50/50 [08:34<00:00, 10.30s/it]
+Saved video to ./samples/latte/An_epic_tornado_attacking_above_aglowing_city_at_night._0000.mp4
+Saved spatial MLP MSE outputs to ./samples/latte/An_epic_tornado_attacking_above_aglowing_city_at_night._0000_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/latte/An_epic_tornado_attacking_above_aglowing_city_at_night._0000_temporal_mlp_mse.pt
+Processing the (Snow falling over multiple houses and trees on winter landscape against night sky. christmas festivity and celebration concept.) prompt
+
+  0%|          | 0/50 [00:00<?, ?it/s]
+  2%|▏         | 1/50 [00:06<04:59,  6.10s/it]Time step 960, Spatial MSE: [(0, 1.0131829185411334e-05), (1, 1.6488434084749315e-06), (2, 2.157954077119939e-05), (3, 2.9487651772797108e-05), (4, 6.118885357864201e-05), (5, 5.3879382903687656e-05), (6, 5.5662807426415384e-05), (7, 6.691240560030565e-05), (8, 8.810023427940905e-05), (9, 0.00015335693024098873), (10, 0.0004045348323415965), (11, 0.0005393939791247249), (12, 0.0005466266302391887), (13, 0.0008503932040184736), (14, 0.0027494740206748247), (15, 0.004686701111495495), (16, 0.020556168630719185), (17, 0.002836558036506176), (18, 0.0034814332611858845), (19, 0.0027313323225826025), (20, 0.0054846457205712795), (21, 0.0033039762638509274), (22, 0.001042638556100428), (23, 0.0017866649432107806), (24, 0.005258239805698395), (25, 0.004536108113825321), (26, 0.004288462921977043), (27, 0.002057831734418869)], Temporal MSE: [(0, 1.4415993973670993e-05), (1, 4.8308843361155596e-06), (2, 4.418289790919516e-06), (3, 8.359560524695553e-06), (4, 6.571049198100809e-06), (5, 1.3426132682070602e-05), (6, 2.1935398763162084e-05), (7, 4.802115654456429e-05), (8, 4.0268594602821395e-05), (9, 0.0002126484178006649), (10, 0.0002656583965290338), (11, 0.00022719221306033432), (12, 0.00038563800626434386), (13, 0.00040653892210684717), (14, 0.00048067240277305245), (15, 0.00030373173649422824), (16, 0.0006622124346904457), (17, 0.00034924750798381865), (18, 0.0002637373108882457), (19, 9.645459067542106e-05), (20, 0.0001985541603062302), (21, 0.00030039669945836067), (22, 0.00034399438300170004), (23, 0.00027680324274115264), (24, 0.0004084416141267866), (25, 0.0004879288899246603), (26, 0.0002623782493174076), (27, 9.064174264494795e-06)]
+
+  4%|▍         | 2/50 [00:15<06:33,  8.21s/it]Time step 940, Spatial MSE: [(0, 1.1573056326596998e-05), (1, 1.8534615264798049e-06), (2, 2.0834570022998378e-05), (3, 3.0390241590794176e-05), (4, 4.856899977312423e-05), (5, 6.095839853514917e-05), (6, 8.192536915885285e-05), (7, 7.817782898200676e-05), (8, 0.00011699659808073193), (9, 0.00023458983923774213), (10, 0.0005001985118724406), (11, 0.0007081308285705745), (12, 0.0008651938405819237), (13, 0.0012094990815967321), (14, 0.003525580046698451), (15, 0.005690834950655699), (16, 0.020266322419047356), (17, 0.0037646819837391376), (18, 0.004749148152768612), (19, 0.003485239576548338), (20, 0.008283214643597603), (21, 0.00412739859893918), (22, 0.0015203901566565037), (23, 0.0021263803355395794), (24, 0.006265835836529732), (25, 0.0066702887415885925), (26, 0.0057304915972054005), (27, 0.0021816317457705736)], Temporal MSE: [(0, 1.5570976756862365e-05), (1, 5.125789812154835e-06), (2, 5.29870067111915e-06), (3, 9.005041647469625e-06), (4, 6.875204235257115e-06), (5, 1.6236708688666113e-05), (6, 3.225448017474264e-05), (7, 5.85425877943635e-05), (8, 5.086789315100759e-05), (9, 0.00029826146783307195), (10, 0.0003552714188117534), (11, 0.0003255119954701513), (12, 0.0005802466184832156), (13, 0.0006159993354231119), (14, 0.0005085021839477122), (15, 0.0004069332790095359), (16, 0.000820911256596446), (17, 0.00047931360313668847), (18, 0.0003168565162923187), (19, 0.0001455281162634492), (20, 0.00032141516567207873), (21, 0.000376590876840055), (22, 0.0004657375975511968), (23, 0.0003627045080065727), (24, 0.0004871774872299284), (25, 0.0006235999753698707), (26, 0.00032751786056905985), (27, 1.616538065718487e-05)]
+
+  6%|▌         | 3/50 [00:26<07:21,  9.40s/it]Time step 920, Spatial MSE: [(0, 1.2076739949407056e-05), (1, 1.9129440715914825e-06), (2, 2.6170409910264425e-05), (3, 3.080865280935541e-05), (4, 4.3372765503590927e-05), (5, 6.206910620676354e-05), (6, 0.00010697132529458031), (7, 9.248285641660914e-05), (8, 0.00013167312135919929), (9, 0.000274789024842903), (10, 0.0005003873957321048), (11, 0.000558959785848856), (12, 0.0007884440128691494), (13, 0.0007945788674987853), (14, 0.0025446442887187004), (15, 0.005037383176386356), (16, 0.006342528387904167), (17, 0.003874203423038125), (18, 0.004453425761312246), (19, 0.003504733555018902), (20, 0.00896748062223196), (21, 0.0037087814416736364), (22, 0.0015792208723723888), (23, 0.0021824988070875406), (24, 0.005730891600251198), (25, 0.008243340998888016), (26, 0.0063317399471998215), (27, 0.002537178574129939)], Temporal MSE: [(0, 1.793568480934482e-05), (1, 5.887851330044214e-06), (2, 6.181773187563522e-06), (3, 1.0475841008883435e-05), (4, 7.606286999362055e-06), (5, 1.8942995666293427e-05), (6, 3.4975768357980996e-05), (7, 5.4521235142601654e-05), (8, 4.696079122368246e-05), (9, 0.0003282061079517007), (10, 0.0003693739708978683), (11, 0.0003526144428178668), (12, 0.0005388619028963149), (13, 0.0006056868587620556), (14, 0.0003083912015426904), (15, 0.0004329209914430976), (16, 0.0007214132347144186), (17, 0.0005553446826525033), (18, 0.00033176213037222624), (19, 0.00014683253539260477), (20, 0.00028262491105124354), (21, 0.0002913491625804454), (22, 0.0004430107946973294), (23, 0.00037983187939971685), (24, 0.0004884609952569008), (25, 0.0007240480626933277), (26, 0.00035280577139928937), (27, 1.4975144949858077e-05)]
+
+  8%|▊         | 4/50 [00:35<06:55,  9.02s/it]Time step 900, Spatial MSE: [(0, 1.5800851542735472e-05), (1, 2.2219724087335635e-06), (2, 3.1716874218545854e-05), (3, 3.479344741208479e-05), (4, 3.722913606907241e-05), (5, 6.986378139117733e-05), (6, 0.00017891086463350803), (7, 0.00011756038293242455), (8, 0.00017527956515550613), (9, 0.00038087356369942427), (10, 0.0006739061209373176), (11, 0.0006320758257061243), (12, 0.0008339157793670893), (13, 0.0006870846846140921), (14, 0.002112401183694601), (15, 0.0040451944805681705), (16, 0.010485261678695679), (17, 0.0036988721694797277), (18, 0.004619701765477657), (19, 0.003732023062184453), (20, 0.00955064408481121), (21, 0.003446180373430252), (22, 0.0017283172346651554), (23, 0.002280124230310321), (24, 0.004994563292711973), (25, 0.009320094250142574), (26, 0.00848237331956625), (27, 0.003165337722748518)], Temporal MSE: [(0, 2.211583159805741e-05), (1, 6.858126198494574e-06), (2, 7.907221515779383e-06), (3, 1.3358062460611109e-05), (4, 9.301092177338433e-06), (5, 2.8624086553463712e-05), (6, 4.3134758016094565e-05), (7, 5.7408102293265983e-05), (8, 5.339544441085309e-05), (9, 0.0004014209844172001), (10, 0.0004762274620588869), (11, 0.0004301041772123426), (12, 0.0006123544299043715), (13, 0.000712850596755743), (14, 0.00047246585018001497), (15, 0.00046377850230783224), (16, 0.0006470066728070378), (17, 0.0006049348157830536), (18, 0.0002875341451726854), (19, 0.0001798729063011706), (20, 0.0003490528033580631), (21, 0.0004947876441292465), (22, 0.0005662322510033846), (23, 0.00047551654279232025), (24, 0.0006036522681824863), (25, 0.0009488636278547347), (26, 0.00044120807433500886), (27, 1.971478559426032e-05)]
+
+ 10%|█         | 5/50 [00:45<07:12,  9.62s/it]Time step 880, Spatial MSE: [(0, 1.2391773452691268e-05), (1, 2.1413575268525165e-06), (2, 3.814381852862425e-05), (3, 2.5918161554727703e-05), (4, 3.451130396570079e-05), (5, 6.709474109811708e-05), (6, 0.00017849636788014323), (7, 0.00012286951823625714), (8, 0.00019345928740222007), (9, 0.0004107943968847394), (10, 0.0007625268190167844), (11, 0.0006833994993939996), (12, 0.0008704154170118272), (13, 0.0007321182638406754), (14, 0.0021980591118335724), (15, 0.004013517405837774), (16, 0.009959219954907894), (17, 0.003764224471524358), (18, 0.0037148843985050917), (19, 0.0031198272481560707), (20, 0.007873876951634884), (21, 0.0033601222094148397), (22, 0.0019201386021450162), (23, 0.0024097126442939043), (24, 0.005052154418081045), (25, 0.012033232487738132), (26, 0.006840386427938938), (27, 0.002999134361743927)], Temporal MSE: [(0, 1.8780629034154117e-05), (1, 6.362458861985942e-06), (2, 7.656171874259599e-06), (3, 1.0117266356246546e-05), (4, 9.41093276196625e-06), (5, 2.6292043912690133e-05), (6, 5.374972533900291e-05), (7, 6.400082202162594e-05), (8, 5.475363286677748e-05), (9, 0.00041858942131511867), (10, 0.0004965882399119437), (11, 0.00047349013038910925), (12, 0.0006316328071989119), (13, 0.0006836187676526606), (14, 0.0005284188664518297), (15, 0.0004980866797268391), (16, 0.00061043567257002), (17, 0.0005863523692823946), (18, 0.0002817357308231294), (19, 0.0001995567581616342), (20, 0.00035163608845323324), (21, 0.0005152150988578796), (22, 0.000482794624986127), (23, 0.0004504001699388027), (24, 0.0005987103213556111), (25, 0.0008311395649798214), (26, 0.00045484703150577843), (27, 2.3187032638816163e-05)]
+
+ 12%|█▏        | 6/50 [00:56<07:23, 10.07s/it]Time step 860, Spatial MSE: [(0, 4.5344135287450626e-05), (1, 7.232251846289728e-06), (2, 4.3206695409026e-05), (3, 7.547744462499395e-05), (4, 8.348826668225229e-05), (5, 0.00013500764907803386), (6, 0.0004824857460334897), (7, 0.00024772275355644524), (8, 0.00036027454189024866), (9, 0.0006437904667109251), (10, 0.0011769646080210805), (11, 0.0011441960232332349), (12, 0.001099228742532432), (13, 0.0011935271322727203), (14, 0.003207116387784481), (15, 0.006650086492300034), (16, 0.008695360273122787), (17, 0.007153032347559929), (18, 0.008997786790132523), (19, 0.006615012418478727), (20, 0.0170282032340765), (21, 0.008186379447579384), (22, 0.004243403673171997), (23, 0.004126071464270353), (24, 0.007098508533090353), (25, 0.021020790562033653), (26, 0.02498282492160797), (27, 0.011042104102671146)], Temporal MSE: [(0, 6.911954551469535e-05), (1, 1.9408242224017158e-05), (2, 2.245251198473852e-05), (3, 2.4188560928450897e-05), (4, 2.2777574486099184e-05), (5, 0.00010643043788149953), (6, 9.324005804955959e-05), (7, 0.00010437132732477039), (8, 0.00010439515608595684), (9, 0.0006281591486185789), (10, 0.0006657500634901226), (11, 0.0006203207885846496), (12, 0.000848117982968688), (13, 0.0009782554116100073), (14, 0.0007201261469163001), (15, 0.0006649088463746011), (16, 0.0008621865999884903), (17, 0.0008620956214144826), (18, 0.0004392131231725216), (19, 0.00033919906127266586), (20, 0.0008285838994197547), (21, 0.0006139756296761334), (22, 0.001151377335190773), (23, 0.0009801116539165378), (24, 0.0013657797826454043), (25, 0.0022375499829649925), (26, 0.0012562996707856655), (27, 0.0001059608330251649)]
+
+ 14%|█▍        | 7/50 [01:08<07:38, 10.65s/it]Time step 840, Spatial MSE: [(0, 5.502924523170805e-06), (1, 2.6691382117860485e-06), (2, 5.866306310053915e-05), (3, 1.5629502740921453e-05), (4, 2.4128579752868973e-05), (5, 3.876172922900878e-05), (6, 0.0001427138049621135), (7, 9.829205373534933e-05), (8, 0.0001384900533594191), (9, 0.0003219397913198918), (10, 0.0006142580532468855), (11, 0.0006454155081883073), (12, 0.0005681122420355678), (13, 0.00062762969173491), (14, 0.0012587279779836535), (15, 0.0015290488954633474), (16, 0.003703495254740119), (17, 0.0008753579459153116), (18, 0.0005913763889111578), (19, 0.0008669269154779613), (20, 0.0017375645693391562), (21, 0.00047907058615237474), (22, 0.00091267324751243), (23, 0.0012777232332155108), (24, 0.0024913803208619356), (25, 0.009535041637718678), (26, 0.0037368470802903175), (27, 0.004867757670581341)], Temporal MSE: [(0, 8.40532811707817e-06), (1, 3.164034978908603e-06), (2, 6.6083134697692e-06), (3, 1.2120417522964999e-05), (4, 8.216195965360384e-06), (5, 1.7733840650180355e-05), (6, 2.9481669116648845e-05), (7, 3.444178219069727e-05), (8, 3.519784877425991e-05), (9, 0.00015175541921053082), (10, 0.0002049586910288781), (11, 0.0002560182474553585), (12, 0.0003393710940144956), (13, 0.00034578487975522876), (14, 0.0005510880146175623), (15, 0.00034837095881812274), (16, 0.0003316208312753588), (17, 0.0002675085561349988), (18, 0.00022206007270142436), (19, 0.00019603775581344962), (20, 0.00031776155810803175), (21, 0.0002183222968596965), (22, 0.0002588398347143084), (23, 0.00027977462741546333), (24, 0.000542256748303771), (25, 0.0007115227053873241), (26, 0.0004059381317347288), (27, 4.170745887677185e-05)]
+
+ 16%|█▌        | 8/50 [01:17<06:59,  9.99s/it]Time step 820, Spatial MSE: [(0, 1.3549083632824477e-05), (1, 2.5056112917809514e-06), (2, 6.35642500128597e-05), (3, 2.6545340006123297e-05), (4, 3.779468534048647e-05), (5, 5.643571785185486e-05), (6, 0.0003839494602289051), (7, 0.00017040122475009412), (8, 0.0002497357490938157), (9, 0.0005568353226408362), (10, 0.0009666311088949442), (11, 0.0010068370029330254), (12, 0.000889039714820683), (13, 0.001049863756634295), (14, 0.002080975566059351), (15, 0.0026399195194244385), (16, 0.004282588139176369), (17, 0.0022291706409305334), (18, 0.0015563248889520764), (19, 0.0019037449965253472), (20, 0.004520552232861519), (21, 0.0022353229578584433), (22, 0.002061393577605486), (23, 0.002458673669025302), (24, 0.004045708104968071), (25, 0.018312308937311172), (26, 0.013142243027687073), (27, 0.003927321173250675)], Temporal MSE: [(0, 1.8687627743929625e-05), (1, 6.27104282102664e-06), (2, 1.3703135664400179e-05), (3, 1.4522927813231945e-05), (4, 1.3227655472292099e-05), (5, 5.75581616431009e-05), (6, 6.495413254015148e-05), (7, 6.439728895202279e-05), (8, 6.946720532141626e-05), (9, 0.0003096151340287179), (10, 0.00036731749423779547), (11, 0.0003860440629068762), (12, 0.0006363799329847097), (13, 0.0006317443330772221), (14, 0.0007522639352828264), (15, 0.000594401964917779), (16, 0.0006923506152816117), (17, 0.00044401243212632835), (18, 0.000332729920046404), (19, 0.0002694913710001856), (20, 0.0008042328408919275), (21, 0.0005745110684074461), (22, 0.000576177379116416), (23, 0.0006088802474550903), (24, 0.0011604538885876536), (25, 0.0016532043227925897), (26, 0.0009960206225514412), (27, 5.2741303079528734e-05)]
+
+ 18%|█▊        | 9/50 [01:25<06:33,  9.60s/it]Time step 800, Spatial MSE: [(0, 1.3273687727632932e-05), (1, 2.95904510494438e-06), (2, 7.985729462234303e-05), (3, 2.5728701075422578e-05), (4, 4.8816349590197206e-05), (5, 6.383583240676671e-05), (6, 0.000436872971476987), (7, 0.00023456958297174424), (8, 0.00036251734127290547), (9, 0.0007905072416178882), (10, 0.001135366503149271), (11, 0.0012294090120121837), (12, 0.0011134424712508917), (13, 0.0014697886072099209), (14, 0.00222072284668684), (15, 0.0029229936189949512), (16, 0.007314709480851889), (17, 0.0023646072950214148), (18, 0.002402485581114888), (19, 0.002129798987880349), (20, 0.004036125726997852), (21, 0.0020461801905184984), (22, 0.0025719082914292812), (23, 0.0029877712950110435), (24, 0.004721314646303654), (25, 0.021046491339802742), (26, 0.015010936185717583), (27, 0.004465178120881319)], Temporal MSE: [(0, 1.8723649191088043e-05), (1, 6.0256829783611465e-06), (2, 1.4557996109942906e-05), (3, 1.5236229046422523e-05), (4, 1.662185059103649e-05), (5, 7.322196324821562e-05), (6, 0.00010349393414799124), (7, 9.024331666296348e-05), (8, 9.766224684426561e-05), (9, 0.0003145633963868022), (10, 0.00039907818427309394), (11, 0.0003831649955827743), (12, 0.0005654507549479604), (13, 0.0006681756349280477), (14, 0.0008763615041971207), (15, 0.0006693985778838396), (16, 0.0006563764181919396), (17, 0.0005477365339174867), (18, 0.00047159328823909163), (19, 0.00039987172931432724), (20, 0.001588735613040626), (21, 0.0005505467997863889), (22, 0.0006295557832345366), (23, 0.0007824784843251109), (24, 0.0018302303506061435), (25, 0.002339249709621072), (26, 0.0015561373438686132), (27, 8.125362364808097e-05)]
+
+ 20%|██        | 10/50 [01:35<06:18,  9.47s/it]Time step 780, Spatial MSE: [(0, 1.0293177183484659e-05), (1, 2.646603888933896e-06), (2, 6.822752766311169e-05), (3, 1.98768793779891e-05), (4, 2.3205724573927e-05), (5, 3.6762292438652366e-05), (6, 6.0938920796616e-05), (7, 0.00011402078962419182), (8, 0.00014051847392693162), (9, 0.00025077309692278504), (10, 0.0006499045412056148), (11, 0.0008836022461764514), (12, 0.0007243494037538767), (13, 0.0009891989175230265), (14, 0.0018659488996490836), (15, 0.0017279924359172583), (16, 0.0025497255846858025), (17, 0.0015506442869082093), (18, 0.000771253602579236), (19, 0.000958296877797693), (20, 0.0017539525870233774), (21, 0.0010390770621597767), (22, 0.0006157825118862092), (23, 0.0011791441356763244), (24, 0.0027098364662379026), (25, 0.007963264361023903), (26, 0.003308295737951994), (27, 0.004930047784000635)], Temporal MSE: [(0, 1.3165587915864307e-05), (1, 5.1860160965588875e-06), (2, 1.0986007509927731e-05), (3, 1.4909952369634993e-05), (4, 1.3272258911456447e-05), (5, 2.346256769669708e-05), (6, 4.264614108251408e-05), (7, 0.00014234607806429267), (8, 7.203888526419178e-05), (9, 0.0002913252974394709), (10, 0.0005319389747455716), (11, 0.0004151997563894838), (12, 0.0005069634644314647), (13, 0.0005953137297183275), (14, 0.0009079012670554221), (15, 0.0007133274339139462), (16, 0.0011552225332707167), (17, 0.0007149631273932755), (18, 0.0007486873655579984), (19, 0.0003438101848587394), (20, 0.0010325455805286765), (21, 0.000594901037402451), (22, 0.0002571438089944422), (23, 0.000502095848787576), (24, 0.0008491702028550208), (25, 0.0013206751318648458), (26, 0.0004552794271148741), (27, 4.762981188832782e-05)]
+
+ 22%|██▏       | 11/50 [01:48<06:51, 10.56s/it]Time step 760, Spatial MSE: [(0, 9.601464626030065e-06), (1, 3.2378770811192226e-06), (2, 5.593396781478077e-05), (3, 1.0861579539778177e-05), (4, 9.376160960528068e-06), (5, 7.133616236387752e-06), (6, 1.1768125659727957e-05), (7, 1.1767736395995598e-05), (8, 2.1509396901819855e-05), (9, 2.4919996576500125e-05), (10, 4.208676909911446e-05), (11, 4.871936107520014e-05), (12, 6.576067244168371e-05), (13, 0.00013408316590357572), (14, 0.00019717162649612874), (15, 0.00021089370420668274), (16, 0.0017491187900304794), (17, 0.00020856592163909227), (18, 0.00020215852418914437), (19, 0.00021751975873485208), (20, 0.0003280581149738282), (21, 0.0005722378264181316), (22, 0.00018737692153081298), (23, 0.0003770355542656034), (24, 0.0009786231676116586), (25, 0.0016667954623699188), (26, 0.0013204137794673443), (27, 0.00413831090554595)], Temporal MSE: [(0, 1.7303687855019234e-05), (1, 5.913715995120583e-06), (2, 5.2359064284246415e-06), (3, 4.314060788601637e-06), (4, 2.3785369194229133e-06), (5, 4.573754267767072e-06), (6, 4.636263838619925e-06), (7, 8.055578291532584e-06), (8, 5.4285223995975684e-06), (9, 2.0538846001727507e-05), (10, 2.0629622667911462e-05), (11, 4.8316251195501536e-05), (12, 5.32898266101256e-05), (13, 0.00012053554382873699), (14, 0.00019605750276241452), (15, 0.00010990232112817466), (16, 0.00021559868764597923), (17, 0.00011969020124524832), (18, 0.00018593788263387978), (19, 0.00010476436000317335), (20, 0.00023398554185405374), (21, 0.00032424088567495346), (22, 0.00012863960000686347), (23, 0.0002078849502140656), (24, 0.00021768234728369862), (25, 0.0006147621897980571), (26, 0.00017824741371441633), (27, 7.896647730376571e-05)]
+
+ 24%|██▍       | 12/50 [01:58<06:40, 10.55s/it]Time step 740, Spatial MSE: [(0, 2.3153826987254433e-05), (1, 1.199221696879249e-05), (2, 0.00019251472258474678), (3, 9.53751805354841e-05), (4, 0.00033850010368041694), (5, 0.00041037038317881525), (6, 0.0031293935608118773), (7, 0.002024873159825802), (8, 0.002765331882983446), (9, 0.004812420345842838), (10, 0.006648678332567215), (11, 0.008917412720620632), (12, 0.008845079690217972), (13, 0.01316632330417633), (14, 0.018219852820038795), (15, 0.02034372091293335), (16, 0.030185451731085777), (17, 0.01527358777821064), (18, 0.011576831340789795), (19, 0.007916678674519062), (20, 0.009865282103419304), (21, 0.0092178825289011), (22, 0.010845890268683434), (23, 0.011098659597337246), (24, 0.019016297534108162), (25, 0.0826331377029419), (26, 0.06805374473333359), (27, 0.007562618236988783)], Temporal MSE: [(0, 4.117209755349904e-05), (1, 7.594392172904918e-06), (2, 4.7462388465646654e-05), (3, 6.892644159961492e-05), (4, 0.00012190782581456006), (5, 0.0005987315089441836), (6, 0.0010785945923998952), (7, 0.0011145181488245726), (8, 0.000762693292926997), (9, 0.002204177901148796), (10, 0.0027033169753849506), (11, 0.0025810925289988518), (12, 0.0030916156247258186), (13, 0.0038680294528603554), (14, 0.00634003197774291), (15, 0.005617932882159948), (16, 0.006469087675213814), (17, 0.005190505646169186), (18, 0.005441318266093731), (19, 0.003081240924075246), (20, 0.01508202962577343), (21, 0.004821269307285547), (22, 0.0034280901309102774), (23, 0.004464359953999519), (24, 0.013665346428751945), (25, 0.017219185829162598), (26, 0.011355860158801079), (27, 0.0005058550741523504)]
+
+ 26%|██▌       | 13/50 [02:12<07:03, 11.46s/it]Time step 720, Spatial MSE: [(0, 7.955509317980614e-06), (1, 4.378861376608256e-06), (2, 8.800013893051073e-05), (3, 1.3750510333920829e-05), (4, 1.121911009249743e-05), (5, 8.406029337493237e-06), (6, 1.7133883375208825e-05), (7, 1.8805149011313915e-05), (8, 2.3388705812976696e-05), (9, 3.420215580263175e-05), (10, 3.483961700112559e-05), (11, 5.036452421336435e-05), (12, 6.4590283727739e-05), (13, 0.00010081700020236894), (14, 0.00015868070477154106), (15, 0.00021101179299876094), (16, 0.0002087923785438761), (17, 0.00018160560284741223), (18, 0.0003313480119686574), (19, 0.00017511645273771137), (20, 0.00030114673427306116), (21, 0.00048785682884044945), (22, 0.00016482413047924638), (23, 0.0004160813114140183), (24, 0.0009759432869032025), (25, 0.0018929557409137487), (26, 0.0016337825218215585), (27, 0.004888499155640602)], Temporal MSE: [(0, 1.4066827134229243e-05), (1, 4.8735860218585e-06), (2, 6.26631299383007e-06), (3, 4.339215138315922e-06), (4, 2.7894702725461684e-06), (5, 6.569684501300799e-06), (6, 6.974626103328774e-06), (7, 1.2898831300844904e-05), (8, 6.526023753394838e-06), (9, 1.7924037820193917e-05), (10, 2.202324139943812e-05), (11, 4.24885110987816e-05), (12, 5.435368075268343e-05), (13, 8.200012234738097e-05), (14, 0.00011976816313108429), (15, 0.00012007662007817999), (16, 0.00016461685299873352), (17, 0.00010921094508375973), (18, 0.00022644373530056328), (19, 0.00013089441927149892), (20, 0.0002496887755114585), (21, 0.0002704759535845369), (22, 8.924734720494598e-05), (23, 0.00018722143431659788), (24, 0.00021904979075770825), (25, 0.0007221433334052563), (26, 0.00024029410269577056), (27, 8.579338464187458e-05)]
+
+ 28%|██▊       | 14/50 [02:24<07:06, 11.84s/it]Time step 700, Spatial MSE: [(0, 2.0130521079408936e-05), (1, 8.491191692883149e-06), (2, 0.00014727108646184206), (3, 4.305585389374755e-05), (4, 7.741083391010761e-05), (5, 0.00012674019671976566), (6, 0.0002327286929357797), (7, 0.0004272516816854477), (8, 0.000629295245744288), (9, 0.0009212565491907299), (10, 0.0019324533641338348), (11, 0.0025427723303437233), (12, 0.002092483453452587), (13, 0.002778459805995226), (14, 0.00472697988152504), (15, 0.003367699682712555), (16, 0.0028404435142874718), (17, 0.0026873035822063684), (18, 0.001226180698722601), (19, 0.0019681823905557394), (20, 0.0025815297849476337), (21, 0.0013511873548850417), (22, 0.001451246440410614), (23, 0.002328739035874605), (24, 0.0052420212887227535), (25, 0.02348899096250534), (26, 0.008463140577077866), (27, 0.008933372795581818)], Temporal MSE: [(0, 1.3024803592998069e-05), (1, 4.84976771986112e-06), (2, 2.7357895305613056e-05), (3, 4.172522312728688e-05), (4, 3.9367507270071656e-05), (5, 0.00010438132449053228), (6, 0.00014199342695064843), (7, 0.0004462135548237711), (8, 0.00022918371541891247), (9, 0.0005459273234009743), (10, 0.0010539382928982377), (11, 0.0009202339570038021), (12, 0.000988271553069353), (13, 0.001425603753887117), (14, 0.0020444439724087715), (15, 0.0012960840249434114), (16, 0.001901410985738039), (17, 0.0015322748804464936), (18, 0.0012996005825698376), (19, 0.000978861702606082), (20, 0.0022070477716624737), (21, 0.001475274795666337), (22, 0.000763974676374346), (23, 0.0013012359850108624), (24, 0.003051517764106393), (25, 0.006285397801548243), (26, 0.0017922278493642807), (27, 0.00016577090718783438)]
+
+ 30%|███       | 15/50 [02:36<06:50, 11.72s/it]Time step 680, Spatial MSE: [(0, 6.900721928104758e-06), (1, 1.7170670616906136e-05), (2, 0.00016459972539450973), (3, 3.851876681437716e-05), (4, 0.00010205701255472377), (5, 0.00010595699859550223), (6, 0.0005739199696108699), (7, 0.00068828952498734), (8, 0.00027237876201979816), (9, 0.00048330577556043863), (10, 0.0007623776909895241), (11, 0.0005811697919853032), (12, 0.0005634089466184378), (13, 0.0008305524243041873), (14, 0.001921923249028623), (15, 0.004058247432112694), (16, 0.02326882630586624), (17, 0.0015969736268743873), (18, 0.0032312560360878706), (19, 0.001515047624707222), (20, 0.00447791488841176), (21, 0.006066137459129095), (22, 0.0021775749046355486), (23, 0.0017621753504499793), (24, 0.004084127955138683), (25, 0.017378654330968857), (26, 0.004261567257344723), (27, 0.00773833179846406)], Temporal MSE: [(0, 1.9699838958331384e-05), (1, 7.489106792490929e-06), (2, 2.353595118620433e-05), (3, 9.129972022492439e-05), (4, 6.736034265486524e-05), (5, 0.0002219717571279034), (6, 0.0002749588165897876), (7, 0.0001506423286627978), (8, 0.00012157305172877386), (9, 0.00023542607959825546), (10, 0.0005265597137622535), (11, 0.0006129505927674472), (12, 0.0006657277699559927), (13, 0.0013758228160440922), (14, 0.002294258214533329), (15, 0.001136889448389411), (16, 0.0014267159858718514), (17, 0.0007699428242631257), (18, 0.00045191694516688585), (19, 0.0005871476605534554), (20, 0.004229685757309198), (21, 0.0011601903242990375), (22, 0.0011091401102021337), (23, 0.0009818014223128557), (24, 0.0032068712171167135), (25, 0.004906719550490379), (26, 0.0028053170535713434), (27, 0.00017651327652856708)]
+
+ 32%|███▏      | 16/50 [02:47<06:37, 11.69s/it]Time step 660, Spatial MSE: [(0, 1.5678919226047583e-05), (1, 1.0422522791486699e-05), (2, 0.00014930371253285557), (3, 6.175493763294071e-05), (4, 0.00014812072913628072), (5, 0.0002048164897132665), (6, 0.00039071892388164997), (7, 0.0005324252997525036), (8, 0.000754988519474864), (9, 0.0012857471592724323), (10, 0.0018556948052719235), (11, 0.002030528150498867), (12, 0.002049036091193557), (13, 0.0039078239351511), (14, 0.005154819693416357), (15, 0.004633254371583462), (16, 0.033607810735702515), (17, 0.0025459553580731153), (18, 0.002980488119646907), (19, 0.0024915907997637987), (20, 0.0025681029073894024), (21, 0.002525546122342348), (22, 0.0021555377170443535), (23, 0.0024906331673264503), (24, 0.008139804005622864), (25, 0.019336262717843056), (26, 0.0104465801268816), (27, 0.008160240948200226)], Temporal MSE: [(0, 9.712114660942461e-06), (1, 3.749401912500616e-06), (2, 1.993523255805485e-05), (3, 3.724401176441461e-05), (4, 6.772720371372998e-05), (5, 0.0001686001050984487), (6, 0.00022682610142510384), (7, 0.0004502303490880877), (8, 0.0002608249196782708), (9, 0.0006551822298206389), (10, 0.00100044347345829), (11, 0.0011785150272771716), (12, 0.0009981426410377026), (13, 0.0021975392010062933), (14, 0.0027535855770111084), (15, 0.0014350417768582702), (16, 0.0015927943168208003), (17, 0.0014935246435925364), (18, 0.0010478831827640533), (19, 0.0011178553104400635), (20, 0.001977162202820182), (21, 0.0015733041800558567), (22, 0.0011885762214660645), (23, 0.001720269676297903), (24, 0.0051753465086221695), (25, 0.01214924268424511), (26, 0.00850355252623558), (27, 0.000252128200372681)]
+
+ 34%|███▍      | 17/50 [03:01<06:42, 12.18s/it]Time step 640, Spatial MSE: [(0, 7.345393896684982e-06), (1, 9.534656783216633e-06), (2, 0.00010229983308818191), (3, 2.3049829906085506e-05), (4, 1.2304015399422497e-05), (5, 9.0804896899499e-06), (6, 4.4994223571848124e-05), (7, 4.692430229624733e-05), (8, 1.692420119070448e-05), (9, 3.340992407174781e-05), (10, 3.5311812098370865e-05), (11, 4.853036443819292e-05), (12, 5.470647738547996e-05), (13, 6.817436951678246e-05), (14, 0.00013889098772779107), (15, 0.00014833707245998085), (16, 0.004873403813689947), (17, 0.00018143093620892614), (18, 0.00035421238862909377), (19, 0.00015274673933163285), (20, 0.00018718309001997113), (21, 0.000558667816221714), (22, 0.0001615914807189256), (23, 0.00040542459464631975), (24, 0.0009272219613194466), (25, 0.00220101117156446), (26, 0.0028133413288742304), (27, 0.006005847826600075)], Temporal MSE: [(0, 8.86720408743713e-06), (1, 3.0228291052480927e-06), (2, 6.0759939515264705e-06), (3, 3.912083684554091e-06), (4, 3.4228032745886594e-06), (5, 8.278691893792711e-06), (6, 1.641323251533322e-05), (7, 2.5251907572965138e-05), (8, 7.643136086699087e-06), (9, 1.648258330533281e-05), (10, 2.43515387410298e-05), (11, 3.880499571096152e-05), (12, 3.510466558509506e-05), (13, 6.264435069169849e-05), (14, 0.00011875468771904707), (15, 9.391421917825937e-05), (16, 0.00012670864816755056), (17, 8.623582834843546e-05), (18, 0.00022496974270325154), (19, 0.0001033385779010132), (20, 0.00016824608610477298), (21, 0.00034386146580800414), (22, 0.0001073336970875971), (23, 0.00019727501785382628), (24, 0.0002544995804782957), (25, 0.0007193490164354444), (26, 0.0003304672136437148), (27, 6.764990394003689e-05)]
+
+ 36%|███▌      | 18/50 [03:11<06:12, 11.63s/it]Time step 620, Spatial MSE: [(0, 1.4605330761696678e-05), (1, 2.4048124032560736e-05), (2, 0.00021911093790549785), (3, 7.32555563445203e-05), (4, 0.00029622146394103765), (5, 0.0006555074360221624), (6, 0.0044127777218818665), (7, 0.005133883096277714), (8, 0.0067504351027309895), (9, 0.01237928681075573), (10, 0.018326733261346817), (11, 0.02595045045018196), (12, 0.025358257815241814), (13, 0.03373005613684654), (14, 0.07746871560811996), (15, 0.06901097297668457), (16, 0.4342612624168396), (17, 0.06358969956636429), (18, 0.06910186260938644), (19, 0.04108389839529991), (20, 0.05688934028148651), (21, 0.14982253313064575), (22, 0.02226724661886692), (23, 0.02877095714211464), (24, 0.06263511627912521), (25, 0.12750206887722015), (26, 0.053799405694007874), (27, 0.010516166687011719)], Temporal MSE: [(0, 1.7779220797820017e-05), (1, 6.944268534425646e-06), (2, 2.1050969735370018e-05), (3, 9.321804100181907e-05), (4, 0.000286636408418417), (5, 0.0006700871745124459), (6, 0.0013063877122476697), (7, 0.002349702874198556), (8, 0.0021520406007766724), (9, 0.004932574462145567), (10, 0.007141590118408203), (11, 0.00868880096822977), (12, 0.010876725427806377), (13, 0.01457973476499319), (14, 0.02235310897231102), (15, 0.02181973122060299), (16, 0.02545442245900631), (17, 0.04042212292551994), (18, 0.03431384637951851), (19, 0.032583557069301605), (20, 0.03869042918086052), (21, 0.032795265316963196), (22, 0.016729865223169327), (23, 0.02064390480518341), (24, 0.04477440193295479), (25, 0.07027333229780197), (26, 0.031108425930142403), (27, 0.0012240330688655376)]
+
+ 38%|███▊      | 19/50 [03:21<05:45, 11.16s/it]Time step 600, Spatial MSE: [(0, 7.293309863598552e-06), (1, 1.2597305612871423e-05), (2, 0.0001034048618748784), (3, 2.7675569072016515e-05), (4, 1.2328602679190226e-05), (5, 1.0179994205827825e-05), (6, 3.82806574634742e-05), (7, 5.3221534471958876e-05), (8, 1.7042100807884708e-05), (9, 3.15081124426797e-05), (10, 3.438639032538049e-05), (11, 4.860253829974681e-05), (12, 5.8887326304102316e-05), (13, 5.7886918511940166e-05), (14, 9.68206804827787e-05), (15, 0.00015509611694142222), (16, 0.00014574904344044626), (17, 0.0001126477072830312), (18, 0.00019471868290565908), (19, 0.00010718153498601168), (20, 0.00014680749154649675), (21, 0.0004172738117631525), (22, 0.00015790114412084222), (23, 0.00040032275137491524), (24, 0.0009214504389092326), (25, 0.00211115600541234), (26, 0.0032955952920019627), (27, 0.006703090853989124)], Temporal MSE: [(0, 7.517240646848222e-06), (1, 2.6627944862411823e-06), (2, 6.073761596780969e-06), (3, 3.801605771514005e-06), (4, 3.988895059592323e-06), (5, 9.003862032841425e-06), (6, 1.822969534259755e-05), (7, 2.313627555849962e-05), (8, 7.78369212639518e-06), (9, 1.4768890650884714e-05), (10, 2.284358924953267e-05), (11, 3.641175862867385e-05), (12, 3.181710417266004e-05), (13, 4.801149771083146e-05), (14, 5.956217137281783e-05), (15, 7.703385199420154e-05), (16, 7.902758807176724e-05), (17, 6.667234993074089e-05), (18, 0.00012173960567452013), (19, 7.464792724931613e-05), (20, 8.697615703567863e-05), (21, 0.00017583294538781047), (22, 5.193830656935461e-05), (23, 0.0001458243787055835), (24, 0.00024593426496721804), (25, 0.0007390125538222492), (26, 0.00041251073707826436), (27, 6.141257472336292e-05)]
+
+ 40%|████      | 20/50 [03:30<05:18, 10.61s/it]Time step 580, Spatial MSE: [(0, 1.4318957255454734e-05), (1, 1.6939251509029418e-05), (2, 0.00017553626094013453), (3, 3.46916749549564e-05), (4, 4.917340993415564e-05), (5, 8.350154530489817e-05), (6, 0.0005184047622606158), (7, 0.0002641236933413893), (8, 0.0006394566153176129), (9, 0.0007973669562488794), (10, 0.001376430387608707), (11, 0.002026235917583108), (12, 0.002170627936720848), (13, 0.002241530455648899), (14, 0.00436092633754015), (15, 0.002120298333466053), (16, 0.008039753884077072), (17, 0.0012425245949998498), (18, 0.001170152798295021), (19, 0.00092527229571715), (20, 0.0011814715107902884), (21, 0.0014009769074618816), (22, 0.0012126262299716473), (23, 0.0016123815439641476), (24, 0.004684107378125191), (25, 0.009175486862659454), (26, 0.006345112342387438), (27, 0.009954463690519333)], Temporal MSE: [(0, 7.236226792883826e-06), (1, 3.392797225387767e-06), (2, 1.901134237414226e-05), (3, 3.087848745053634e-05), (4, 2.9326956791919656e-05), (5, 0.0001109823351725936), (6, 0.00010164143168367445), (7, 0.0002082372666336596), (8, 0.0001511344889877364), (9, 0.00023588904878124595), (10, 0.0004158244701102376), (11, 0.0005536733660846949), (12, 0.0007099842769093812), (13, 0.0010631848126649857), (14, 0.0016172581817954779), (15, 0.0007022079662419856), (16, 0.0007378132431767881), (17, 0.0006325389840640128), (18, 0.0006163347861729562), (19, 0.000556055223569274), (20, 0.000942332437261939), (21, 0.0011212901445105672), (22, 0.0007268996560014784), (23, 0.0010100597282871604), (24, 0.003299008822068572), (25, 0.007101326249539852), (26, 0.0024914992973208427), (27, 0.00028175607440061867)]
+
+ 42%|████▏     | 21/50 [03:42<05:15, 10.87s/it]Time step 560, Spatial MSE: [(0, 6.702957762172446e-06), (1, 2.3378295736620203e-05), (2, 0.00011111582716694102), (3, 4.334185359766707e-05), (4, 4.55412628070917e-05), (5, 9.230361320078373e-05), (6, 0.0003994634607806802), (7, 0.000659754266962409), (8, 0.00028241437394171953), (9, 0.00031821653828956187), (10, 0.00044408332905732095), (11, 0.00042062526335939765), (12, 0.0005417556967586279), (13, 0.0015541061293333769), (14, 0.005616215523332357), (15, 0.004778002388775349), (16, 0.19688943028450012), (17, 0.001330962753854692), (18, 0.002798637142404914), (19, 0.0042597586289048195), (20, 0.004067666362971067), (21, 0.005258429329842329), (22, 0.0016720463754609227), (23, 0.0016932213911786675), (24, 0.004248658195137978), (25, 0.012439638376235962), (26, 0.003538154996931553), (27, 0.012417674995958805)], Temporal MSE: [(0, 8.39722815726418e-06), (1, 4.605849426297937e-06), (2, 2.099031007674057e-05), (3, 4.87313445773907e-05), (4, 6.230748840607703e-05), (5, 0.00011396582704037428), (6, 0.00014784492668695748), (7, 0.00014340899360831827), (8, 9.976735600503162e-05), (9, 0.0002991206420119852), (10, 0.0007029027328826487), (11, 0.0006209073471836746), (12, 0.0010986513225361705), (13, 0.003205970861017704), (14, 0.005198946222662926), (15, 0.0009438620181754231), (16, 0.0022268241737037897), (17, 0.001194150885567069), (18, 0.0015362262492999434), (19, 0.0010262622963637114), (20, 0.002926567569375038), (21, 0.003063732758164406), (22, 0.0027025837916880846), (23, 0.00240109721198678), (24, 0.004084745887666941), (25, 0.005788950249552727), (26, 0.003978443332016468), (27, 0.0005249760579317808)]
+
+ 44%|████▍     | 22/50 [03:55<05:23, 11.54s/it]Time step 540, Spatial MSE: [(0, 1.2882677765446715e-05), (1, 1.871357744676061e-05), (2, 0.00018258379714097828), (3, 6.309515447355807e-05), (4, 0.00010244428995065391), (5, 0.00012923759641125798), (6, 0.0005762702785432339), (7, 0.0005787354311905801), (8, 0.0007876387680880725), (9, 0.0009901199955493212), (10, 0.0012129137758165598), (11, 0.001626168261282146), (12, 0.001830879133194685), (13, 0.0037543566431850195), (14, 0.010552821680903435), (15, 0.0066190725192427635), (16, 0.17768645286560059), (17, 0.0027878200635313988), (18, 0.004271172918379307), (19, 0.0049866437911987305), (20, 0.004214019514620304), (21, 0.004475991707295179), (22, 0.0035631076898425817), (23, 0.004220917820930481), (24, 0.013237999752163887), (25, 0.026383785530924797), (26, 0.018706228584051132), (27, 0.014361132867634296)], Temporal MSE: [(0, 6.283496986725368e-06), (1, 3.1357767511508428e-06), (2, 1.4237943105399609e-05), (3, 4.335486300988123e-05), (4, 6.192405999172479e-05), (5, 0.00014213928079698235), (6, 0.00016040372429415584), (7, 0.00030599668389186263), (8, 0.00023505174613092095), (9, 0.00047418102622032166), (10, 0.0005841521779075265), (11, 0.0007892328430898488), (12, 0.0009942908072844148), (13, 0.0023633914534002542), (14, 0.004959577694535255), (15, 0.001748507609590888), (16, 0.002804144285619259), (17, 0.00185980589594692), (18, 0.0022714349906891584), (19, 0.0016822048928588629), (20, 0.00586356408894062), (21, 0.0036658330354839563), (22, 0.0028389471117407084), (23, 0.0043875593692064285), (24, 0.008151497691869736), (25, 0.02060815319418907), (26, 0.010929014533758163), (27, 0.0006001077126711607)]
+
+ 46%|████▌     | 23/50 [04:07<05:12, 11.57s/it]Time step 520, Spatial MSE: [(0, 6.562737780768657e-06), (1, 1.5289248040062375e-05), (2, 7.79791662353091e-05), (3, 2.7668393158819526e-05), (4, 1.2015048923785798e-05), (5, 1.0687680514820386e-05), (6, 2.813793980749324e-05), (7, 5.227881410974078e-05), (8, 1.411992525390815e-05), (9, 2.340376886422746e-05), (10, 2.5569366698618978e-05), (11, 3.857795672956854e-05), (12, 3.850749635603279e-05), (13, 5.746866372646764e-05), (14, 6.025667244102806e-05), (15, 7.857615855755284e-05), (16, 0.00012179787154309452), (17, 6.562106864294037e-05), (18, 0.00011978421389358118), (19, 8.091296331258491e-05), (20, 0.00010106462286785245), (21, 0.00016819097800180316), (22, 0.00012059803702868521), (23, 0.00027398235397413373), (24, 0.0006009675562381744), (25, 0.0015837006503716111), (26, 0.003235545940697193), (27, 0.006809958256781101)], Temporal MSE: [(0, 5.644707925966941e-06), (1, 2.164785655622836e-06), (2, 4.60313231087639e-06), (3, 3.942041530535789e-06), (4, 3.884393663611263e-06), (5, 8.102205356408376e-06), (6, 1.7828719137469307e-05), (7, 1.8147140508517623e-05), (8, 6.505133114842465e-06), (9, 1.1700718459906057e-05), (10, 1.8030421415460296e-05), (11, 2.55313134402968e-05), (12, 2.2567150153918192e-05), (13, 5.1638606237247586e-05), (14, 4.489567072596401e-05), (15, 4.108470966457389e-05), (16, 4.730365253635682e-05), (17, 3.843239028356038e-05), (18, 4.804024865734391e-05), (19, 4.2761788790812716e-05), (20, 5.0339258450549096e-05), (21, 6.906157068442553e-05), (22, 4.456973692867905e-05), (23, 8.995511598186567e-05), (24, 0.00019593798788264394), (25, 0.0005172719829715788), (26, 0.0004404610372148454), (27, 5.632068496197462e-05)]
+
+ 48%|████▊     | 24/50 [04:19<05:05, 11.74s/it]Time step 500, Spatial MSE: [(0, 1.177665853901999e-05), (1, 2.738164948823396e-05), (2, 0.00017659965669736266), (3, 6.432432564906776e-05), (4, 0.00028646341525018215), (5, 0.0005079873371869326), (6, 0.005108175799250603), (7, 0.0063247173093259335), (8, 0.01008422952145338), (9, 0.013902310281991959), (10, 0.031038228422403336), (11, 0.029097246006131172), (12, 0.024377193301916122), (13, 0.02686651609838009), (14, 0.05659034475684166), (15, 0.052766602486371994), (16, 0.38865458965301514), (17, 0.05695990100502968), (18, 0.0803111121058464), (19, 0.10403083264827728), (20, 0.16553935408592224), (21, 0.12322355806827545), (22, 0.050487957894802094), (23, 0.06596777588129044), (24, 0.12883655726909637), (25, 0.24489133059978485), (26, 0.10450755059719086), (27, 0.01229347288608551)], Temporal MSE: [(0, 6.874508471810259e-06), (1, 3.954458861699095e-06), (2, 1.7815114915720187e-05), (3, 0.00025092833675444126), (4, 0.000284085312159732), (5, 0.0010710828937590122), (6, 0.0018735083285719156), (7, 0.002343852072954178), (8, 0.002879953710362315), (9, 0.006404170300811529), (10, 0.010372474789619446), (11, 0.009987126104533672), (12, 0.012247399426996708), (13, 0.017682768404483795), (14, 0.03367079794406891), (15, 0.02982637658715248), (16, 0.0372786708176136), (17, 0.03677879646420479), (18, 0.03801845386624336), (19, 0.04214909300208092), (20, 0.07020334899425507), (21, 0.06339067965745926), (22, 0.04003462195396423), (23, 0.046290770173072815), (24, 0.10214225202798843), (25, 0.17023426294326782), (26, 0.05369015783071518), (27, 0.002658543875440955)]
+
+ 50%|█████     | 25/50 [04:31<04:56, 11.84s/it]Time step 480, Spatial MSE: [(0, 6.322631179500604e-06), (1, 1.78343580046203e-05), (2, 6.724195554852486e-05), (3, 2.8712462153634988e-05), (4, 1.220781632582657e-05), (5, 1.2104004781576805e-05), (6, 1.94059557543369e-05), (7, 4.813685518456623e-05), (8, 1.4144390661385842e-05), (9, 2.0213010429870337e-05), (10, 2.3314969439525157e-05), (11, 3.31430601363536e-05), (12, 3.181684951414354e-05), (13, 4.326360431150533e-05), (14, 4.5839682570658624e-05), (15, 5.828201028634794e-05), (16, 6.646929978160188e-05), (17, 4.3884010665351525e-05), (18, 0.00013126974226906896), (19, 7.101565279299393e-05), (20, 8.489098399877548e-05), (21, 0.0001558003859827295), (22, 0.0001324971381109208), (23, 0.00022922812786418945), (24, 0.0005602797027677298), (25, 0.0013302307343110442), (26, 0.003254535375162959), (27, 0.007172508165240288)], Temporal MSE: [(0, 4.915078534395434e-06), (1, 2.073994437523652e-06), (2, 4.1839452933345456e-06), (3, 5.022649929742329e-06), (4, 4.103483661310747e-06), (5, 7.80776645115111e-06), (6, 1.609648643352557e-05), (7, 1.529281507828273e-05), (8, 6.725378625560552e-06), (9, 1.1672010259644594e-05), (10, 1.7051446775440127e-05), (11, 2.2263810024014674e-05), (12, 1.8906168406829238e-05), (13, 3.544090213836171e-05), (14, 2.657997902133502e-05), (15, 3.4661632525967434e-05), (16, 2.9977793019497767e-05), (17, 2.699751712498255e-05), (18, 4.361927858553827e-05), (19, 4.128113141632639e-05), (20, 5.944645090494305e-05), (21, 7.901242497609928e-05), (22, 6.452044908655807e-05), (23, 0.00010334633407182992), (24, 0.00023665561457164586), (25, 0.0005790536524727941), (26, 0.000529490236658603), (27, 6.64267354295589e-05)]
+
+ 52%|█████▏    | 26/50 [04:40<04:27, 11.14s/it]Time step 460, Spatial MSE: [(0, 1.225339565280592e-05), (1, 2.100275014527142e-05), (2, 0.00011474335042294115), (3, 3.629776256275363e-05), (4, 5.801291626994498e-05), (5, 6.912262324476615e-05), (6, 0.00038409189437516034), (7, 0.00020295847207307816), (8, 0.00043936798465438187), (9, 0.0005740026826970279), (10, 0.001131430733948946), (11, 0.001586486236192286), (12, 0.0017039473168551922), (13, 0.002001106971874833), (14, 0.004041182342916727), (15, 0.001722234534099698), (16, 0.0012208126718178391), (17, 0.0010890434496104717), (18, 0.0012958995066583157), (19, 0.0007215461228042841), (20, 0.0009267159621231258), (21, 0.0012955600395798683), (22, 0.0013409454841166735), (23, 0.0020244703628122807), (24, 0.007213590666651726), (25, 0.010127204470336437), (26, 0.016533417627215385), (27, 0.0070457616820931435)], Temporal MSE: [(0, 4.975870524503989e-06), (1, 2.5922574877768056e-06), (2, 1.7202823073603213e-05), (3, 4.710370922111906e-05), (4, 3.8360984035534784e-05), (5, 7.144117262214422e-05), (6, 0.00011976472160313278), (7, 0.00013276551908347756), (8, 0.0001109669974539429), (9, 0.00020003100507892668), (10, 0.00034326029708608985), (11, 0.0005053339991718531), (12, 0.0005908858729526401), (13, 0.001046614721417427), (14, 0.0013657379895448685), (15, 0.0007221804698929191), (16, 0.000555730948690325), (17, 0.0005214213160797954), (18, 0.0005600531003437936), (19, 0.0005543844308704138), (20, 0.000987929874099791), (21, 0.0011264663189649582), (22, 0.0010878817411139607), (23, 0.0012441860744729638), (24, 0.004660816863179207), (25, 0.013000787235796452), (26, 0.0039017503149807453), (27, 0.0005409861332736909)]
+
+ 54%|█████▍    | 27/50 [04:53<04:26, 11.57s/it]Time step 440, Spatial MSE: [(0, 5.596455139311729e-06), (1, 2.3939779566717334e-05), (2, 6.432671216316521e-05), (3, 3.425997419981286e-05), (4, 5.748491457779892e-05), (5, 0.00013522767403628677), (6, 0.00020990821940358728), (7, 0.00046823080629110336), (8, 0.00025315064704045653), (9, 0.00029011169681325555), (10, 0.0003750560572370887), (11, 0.0003765080473385751), (12, 0.0005137918051332235), (13, 0.0020791811402887106), (14, 0.003603354562073946), (15, 0.0032237640116363764), (16, 0.09084342420101166), (17, 0.0011238373117521405), (18, 0.001730659743770957), (19, 0.003269529901444912), (20, 0.003100044559687376), (21, 0.005312745459377766), (22, 0.002615583362057805), (23, 0.002020422834903002), (24, 0.004571438767015934), (25, 0.00924723781645298), (26, 0.004935849457979202), (27, 0.010370482690632343)], Temporal MSE: [(0, 4.880917458649492e-06), (1, 3.20641402140609e-06), (2, 2.0536341253318824e-05), (3, 0.0001216173404827714), (4, 0.00013658433454111218), (5, 0.00010017966269515455), (6, 0.00011360293137840927), (7, 0.00016003992641344666), (8, 0.00010405367356725037), (9, 0.0003461577871348709), (10, 0.000674832088407129), (11, 0.0006349710747599602), (12, 0.0012196345487609506), (13, 0.0031094427686184645), (14, 0.005526393186300993), (15, 0.0009933687979355454), (16, 0.000997935771010816), (17, 0.0008819771464914083), (18, 0.0007980524096637964), (19, 0.0008490770705975592), (20, 0.0019348723581060767), (21, 0.0023081572726368904), (22, 0.001699663931503892), (23, 0.0028586161788553), (24, 0.006086884997785091), (25, 0.008336037397384644), (26, 0.005324682220816612), (27, 0.0007879441254772246)]
+
+ 56%|█████▌    | 28/50 [05:06<04:23, 11.96s/it]Time step 420, Spatial MSE: [(0, 1.1419248949096072e-05), (1, 2.2528698536916636e-05), (2, 0.00010911390563705936), (3, 4.8386529670096934e-05), (4, 0.00010257362009724602), (5, 0.00014715519500896335), (6, 0.0003205527027603239), (7, 0.000604266591835767), (8, 0.0005650350358337164), (9, 0.0008431081660091877), (10, 0.0009679885115474463), (11, 0.0014183510793372989), (12, 0.0017465394921600819), (13, 0.004025116562843323), (14, 0.009332079440355301), (15, 0.00480644078925252), (16, 0.10235290974378586), (17, 0.0028635323978960514), (18, 0.005703361239284277), (19, 0.006571262609213591), (20, 0.006144767627120018), (21, 0.0070532807148993015), (22, 0.00729354377835989), (23, 0.010217742994427681), (24, 0.04476573318243027), (25, 0.06013922020792961), (26, 0.04020847752690315), (27, 0.012639153748750687)], Temporal MSE: [(0, 4.377785444376059e-06), (1, 2.454308969390695e-06), (2, 1.6041916751419194e-05), (3, 4.4706139306072146e-05), (4, 8.072670607361943e-05), (5, 0.0001586824655532837), (6, 0.00019302155124023557), (7, 0.00030018447432667017), (8, 0.0002765751560218632), (9, 0.000448377279099077), (10, 0.0005573972593992949), (11, 0.000935874180868268), (12, 0.001019681105390191), (13, 0.0024086746852844954), (14, 0.00622531259432435), (15, 0.002168368548154831), (16, 0.003681204281747341), (17, 0.003063320182263851), (18, 0.0033872174099087715), (19, 0.002812447724863887), (20, 0.009593639522790909), (21, 0.007857318967580795), (22, 0.00511896749958396), (23, 0.0073212566785514355), (24, 0.026009589433670044), (25, 0.0479004867374897), (26, 0.014602137729525566), (27, 0.0009815656812861562)]
+
+ 58%|█████▊    | 29/50 [05:18<04:13, 12.09s/it]Time step 400, Spatial MSE: [(0, 5.861182671651477e-06), (1, 1.933160820044577e-05), (2, 4.165139762335457e-05), (3, 2.7686552130035125e-05), (4, 1.0966919944621623e-05), (5, 1.2963401786691975e-05), (6, 1.6676358427503146e-05), (7, 4.5259726903168485e-05), (8, 1.4445792658079881e-05), (9, 1.806861473596655e-05), (10, 2.217040309915319e-05), (11, 2.9253282264107838e-05), (12, 2.9764358259853907e-05), (13, 4.287871706765145e-05), (14, 4.1786548536038026e-05), (15, 4.685939711635001e-05), (16, 6.0931139159947634e-05), (17, 4.1327995859319344e-05), (18, 7.82246861490421e-05), (19, 6.837819091742858e-05), (20, 8.123611769406125e-05), (21, 0.00010652071796357632), (22, 0.00011920180986635387), (23, 0.00023870645964052528), (24, 0.0004708293126896024), (25, 0.0012355592334643006), (26, 0.0031026131473481655), (27, 0.008006853982806206)], Temporal MSE: [(0, 4.210211500321748e-06), (1, 2.029416691584629e-06), (2, 3.5891773677576566e-06), (3, 5.548919943976216e-06), (4, 4.196337613393553e-06), (5, 8.29315740702441e-06), (6, 1.591540058143437e-05), (7, 1.3580123777501285e-05), (8, 7.655011359020136e-06), (9, 1.265789342141943e-05), (10, 1.6594334738329053e-05), (11, 1.9595465346355923e-05), (12, 1.8843942598323338e-05), (13, 3.241956073907204e-05), (14, 2.560775465099141e-05), (15, 2.9341417757677846e-05), (16, 2.8122383810114115e-05), (17, 2.453538400004618e-05), (18, 3.147407551296055e-05), (19, 4.071140574524179e-05), (20, 6.118381134001538e-05), (21, 6.834533996880054e-05), (22, 6.960488826734945e-05), (23, 0.00011430973245296627), (24, 0.0002317232429049909), (25, 0.0005567899788729846), (26, 0.0005993657396174967), (27, 0.00010277072578901425)]
+
+ 60%|██████    | 30/50 [05:31<04:05, 12.29s/it]Time step 380, Spatial MSE: [(0, 1.1231553799007088e-05), (1, 2.6227982743876055e-05), (2, 8.858902583597228e-05), (3, 5.425577546702698e-05), (4, 0.0002499890106264502), (5, 0.0006249045254662633), (6, 0.002453671069815755), (7, 0.004497851245105267), (8, 0.009384069591760635), (9, 0.01373698003590107), (10, 0.016415048390626907), (11, 0.01808670349419117), (12, 0.016113393008708954), (13, 0.018398039042949677), (14, 0.03504215180873871), (15, 0.04064386337995529), (16, 0.08421748131513596), (17, 0.057326529175043106), (18, 0.051598384976387024), (19, 0.045649170875549316), (20, 0.06029156967997551), (21, 0.07710495591163635), (22, 0.06840217113494873), (23, 0.1144215390086174), (24, 0.2720717489719391), (25, 0.3726326823234558), (26, 0.21021561324596405), (27, 0.008855474181473255)], Temporal MSE: [(0, 4.3269797060929704e-06), (1, 2.6291163521818817e-06), (2, 2.0966986994608305e-05), (3, 0.00033075851388275623), (4, 0.00036077992990612984), (5, 0.0010456339223310351), (6, 0.00191546103451401), (7, 0.0032245004549622536), (8, 0.0035547001753002405), (9, 0.0072584799490869045), (10, 0.011223506182432175), (11, 0.012787306681275368), (12, 0.012939495965838432), (13, 0.017273487523198128), (14, 0.02398153394460678), (15, 0.033369407057762146), (16, 0.047102540731430054), (17, 0.05570117011666298), (18, 0.058688268065452576), (19, 0.06711195409297943), (20, 0.08854618668556213), (21, 0.09236856549978256), (22, 0.06519561260938644), (23, 0.08160466700792313), (24, 0.18401604890823364), (25, 0.35594642162323), (26, 0.10422313958406448), (27, 0.00533738499507308)]
+
+ 62%|██████▏   | 31/50 [05:43<03:54, 12.34s/it]Time step 360, Spatial MSE: [(0, 5.623512606689474e-06), (1, 1.9108418200630695e-05), (2, 3.1637428037356585e-05), (3, 2.8507933166110888e-05), (4, 1.0657947314030025e-05), (5, 1.5100599739525933e-05), (6, 1.5573634300380945e-05), (7, 4.270942008588463e-05), (8, 1.515745771030197e-05), (9, 1.655315099924337e-05), (10, 1.994505146285519e-05), (11, 2.649684938660357e-05), (12, 2.7296053303871304e-05), (13, 5.19759050803259e-05), (14, 0.00011463248665677384), (15, 8.544580487068743e-05), (16, 0.006988204084336758), (17, 4.6828594349790365e-05), (18, 0.00013138627400621772), (19, 0.00021892081713303924), (20, 9.327151929028332e-05), (21, 0.00033550543594174087), (22, 0.00019097905897069722), (23, 0.00030370583408512175), (24, 0.000577524013351649), (25, 0.0012981077888980508), (26, 0.0024796996731311083), (27, 0.009275006130337715)], Temporal MSE: [(0, 3.930502316507045e-06), (1, 2.0561797100526746e-06), (2, 3.68708060705103e-06), (3, 6.862780992378248e-06), (4, 4.522693870967487e-06), (5, 8.583964699937496e-06), (6, 1.4497101801680401e-05), (7, 1.2976125617569778e-05), (8, 9.012649570649955e-06), (9, 1.2472386515582912e-05), (10, 1.6735521057853475e-05), (11, 1.9395723938941956e-05), (12, 2.0151752323727123e-05), (13, 4.788674777955748e-05), (14, 0.0002170922962250188), (15, 3.4897751902462915e-05), (16, 5.753599543822929e-05), (17, 3.478237704257481e-05), (18, 5.969848280074075e-05), (19, 5.120592686580494e-05), (20, 0.00010151138849323615), (21, 0.0001066699915099889), (22, 0.00013473443686962128), (23, 0.0002750126295723021), (24, 0.0004158894589636475), (25, 0.0007430280093103647), (26, 0.0006970142712816596), (27, 0.00016604620032012463)]
+
+ 64%|██████▍   | 32/50 [05:55<03:39, 12.19s/it]Time step 340, Spatial MSE: [(0, 1.1466621799627319e-05), (1, 2.0444567780941725e-05), (2, 3.9293503505177796e-05), (3, 3.979100802098401e-05), (4, 5.457617589854635e-05), (5, 8.124903979478404e-05), (6, 0.000311957934172824), (7, 0.0002955570234917104), (8, 0.00044721446465700865), (9, 0.0005269719404168427), (10, 0.0009531088871881366), (11, 0.0014395847683772445), (12, 0.0016017507296055555), (13, 0.0020001220982521772), (14, 0.004815103020519018), (15, 0.001985018840059638), (16, 0.008298050612211227), (17, 0.0010716241085901856), (18, 0.0010391883552074432), (19, 0.000994477653875947), (20, 0.0011504144640639424), (21, 0.001915577333420515), (22, 0.0011431635357439518), (23, 0.0023737738374620676), (24, 0.007080754265189171), (25, 0.01285591907799244), (26, 0.02380112186074257), (27, 0.006873203907161951)], Temporal MSE: [(0, 4.1977159526140895e-06), (1, 2.362696022828459e-06), (2, 1.824602804845199e-05), (3, 6.0298105381662026e-05), (4, 5.31929217686411e-05), (5, 8.077320671873167e-05), (6, 0.0001688744523562491), (7, 0.00017900879902299494), (8, 0.00015020825958345085), (9, 0.0002543506561778486), (10, 0.0003823246224783361), (11, 0.0005752438446506858), (12, 0.0006055140402168036), (13, 0.0010083842789754272), (14, 0.0017255209386348724), (15, 0.0008810904691927135), (16, 0.0006337821250781417), (17, 0.0005413193721324205), (18, 0.0006250048172660172), (19, 0.0005691719707101583), (20, 0.0011742572532966733), (21, 0.001199045917019248), (22, 0.001004297868348658), (23, 0.001714104786515236), (24, 0.004375712480396032), (25, 0.012560809031128883), (26, 0.004300123080611229), (27, 0.0009212206350639462)]
+
+ 66%|██████▌   | 33/50 [06:09<03:32, 12.53s/it]Time step 320, Spatial MSE: [(0, 5.376584340410773e-06), (1, 2.318350198038388e-05), (2, 2.8565063985297456e-05), (3, 2.8971806386834942e-05), (4, 8.509936014888808e-05), (5, 0.00018037798872683197), (6, 0.00032305147033184767), (7, 0.0005672818515449762), (8, 0.0002969763008877635), (9, 0.0003079431189689785), (10, 0.0004708020424004644), (11, 0.0004560764355119318), (12, 0.0006126848165877163), (13, 0.005856422241777182), (14, 0.009843933396041393), (15, 0.004416716285049915), (16, 0.206403449177742), (17, 0.001333222957327962), (18, 0.0018244170350953937), (19, 0.007093868218362331), (20, 0.004715419374406338), (21, 0.007865495048463345), (22, 0.0028906846418976784), (23, 0.002689349465072155), (24, 0.005580301396548748), (25, 0.011124912649393082), (26, 0.00550942262634635), (27, 0.02870466560125351)], Temporal MSE: [(0, 4.261395133653423e-06), (1, 2.866850081773009e-06), (2, 2.316312020411715e-05), (3, 0.00019770878134295344), (4, 0.00019031300325877964), (5, 0.0001525828120065853), (6, 0.00014913665654603392), (7, 0.00017268929514102638), (8, 0.00013216558727435768), (9, 0.00033931760117411613), (10, 0.0006106150103732944), (11, 0.0005888354498893023), (12, 0.001076278742402792), (13, 0.00677794823423028), (14, 0.013056812807917595), (15, 0.0011730751721188426), (16, 0.0010533853201195598), (17, 0.0008694246062077582), (18, 0.000893188756890595), (19, 0.0008991795475594699), (20, 0.0019781282171607018), (21, 0.0024703729432076216), (22, 0.0023189738858491182), (23, 0.004769559483975172), (24, 0.009339703246951103), (25, 0.009006526321172714), (26, 0.00680990144610405), (27, 0.002545541152358055)]
+
+ 68%|██████▊   | 34/50 [06:21<03:19, 12.50s/it]Time step 300, Spatial MSE: [(0, 1.0485283382877242e-05), (1, 2.0859464711975306e-05), (2, 2.9805689337081276e-05), (3, 3.8903624954400584e-05), (4, 0.00011370048014214262), (5, 0.00022357043053489178), (6, 0.0003419036802370101), (7, 0.0006464426987804472), (8, 0.0006098989979363978), (9, 0.0007628918974660337), (10, 0.0008457960793748498), (11, 0.001275639864616096), (12, 0.0015453131636604667), (13, 0.007054564077407122), (14, 0.014318574219942093), (15, 0.0050433906726539135), (16, 0.22988292574882507), (17, 0.0028557060286402702), (18, 0.0034986319951713085), (19, 0.009184484370052814), (20, 0.00692828930914402), (21, 0.009457491338253021), (22, 0.0066508036106824875), (23, 0.008769331499934196), (24, 0.03352220356464386), (25, 0.055277273058891296), (26, 0.05014643445611), (27, 0.027450835332274437)], Temporal MSE: [(0, 4.5755332394037396e-06), (1, 2.4233750082203187e-06), (2, 1.5371791960205883e-05), (3, 3.604881203500554e-05), (4, 0.00011358707706676796), (5, 0.00019013852579519153), (6, 0.00027415467775426805), (7, 0.00030136137502267957), (8, 0.0003162025532219559), (9, 0.00046792931971140206), (10, 0.0005820492515340447), (11, 0.0010917804902419448), (12, 0.0010277535766363144), (13, 0.004495083354413509), (14, 0.012094559147953987), (15, 0.003431011224165559), (16, 0.005851562134921551), (17, 0.003452661447227001), (18, 0.0028192433528602123), (19, 0.0026406932156533003), (20, 0.012114573270082474), (21, 0.007718461100012064), (22, 0.004591699689626694), (23, 0.007990296930074692), (24, 0.023942003026604652), (25, 0.04101189598441124), (26, 0.013093961402773857), (27, 0.0021371403709053993)]
+
+ 70%|███████   | 35/50 [06:32<03:01, 12.08s/it]Time step 280, Spatial MSE: [(0, 5.499610779224895e-06), (1, 2.004474663408473e-05), (2, 2.6875095500145108e-05), (3, 2.7434325602371246e-05), (4, 1.1178830391145311e-05), (5, 2.1009376723668538e-05), (6, 1.8213815565104596e-05), (7, 4.254123996361159e-05), (8, 1.9553592210286297e-05), (9, 1.907105433929246e-05), (10, 2.297916762472596e-05), (11, 3.097374064964242e-05), (12, 3.173072764184326e-05), (13, 5.587935447692871e-05), (14, 4.780390008818358e-05), (15, 6.127665983512998e-05), (16, 5.767100446973927e-05), (17, 4.231447746860795e-05), (18, 0.00010697431571315974), (19, 8.053556666709483e-05), (20, 8.486895967507735e-05), (21, 0.00014225409540813416), (22, 0.00015398154209833592), (23, 0.00035977852530777454), (24, 0.0005957020912319422), (25, 0.0011378355557098985), (26, 0.0022205437999218702), (27, 0.010437951423227787)], Temporal MSE: [(0, 4.322313998272875e-06), (1, 2.4324749574589077e-06), (2, 4.237938355799997e-06), (3, 7.034201644273708e-06), (4, 5.545914518734207e-06), (5, 1.0812188520503696e-05), (6, 1.7780503185349517e-05), (7, 1.390414126944961e-05), (8, 1.2996731129533146e-05), (9, 1.7118569303420372e-05), (10, 2.082515311485622e-05), (11, 2.3230182705447078e-05), (12, 2.6827965484699234e-05), (13, 4.695509051089175e-05), (14, 3.3550881198607385e-05), (15, 4.000395711045712e-05), (16, 3.1175419280771166e-05), (17, 2.8137716071796604e-05), (18, 4.364077904028818e-05), (19, 4.901642387267202e-05), (20, 7.390233804471791e-05), (21, 8.437835640506819e-05), (22, 0.00010541915253270417), (23, 0.00023779795446898788), (24, 0.00032640821882523596), (25, 0.0007313287351280451), (26, 0.0007440592162311077), (27, 0.00024974567350000143)]
+
+ 72%|███████▏  | 36/50 [06:46<02:55, 12.53s/it]Time step 260, Spatial MSE: [(0, 1.03626571217319e-05), (1, 2.4422412025160156e-05), (2, 3.247265703976154e-05), (3, 5.245990178082138e-05), (4, 0.0001513339957455173), (5, 0.000701965531334281), (6, 0.0015032070223242044), (7, 0.004428843501955271), (8, 0.009599145501852036), (9, 0.011325420811772346), (10, 0.008850767277181149), (11, 0.014097796753048897), (12, 0.014525409787893295), (13, 0.018610414117574692), (14, 0.03074011579155922), (15, 0.03692428767681122), (16, 0.03554506599903107), (17, 0.026206327602267265), (18, 0.025646142661571503), (19, 0.021541139110922813), (20, 0.024209624156355858), (21, 0.03168734908103943), (22, 0.029678411781787872), (23, 0.05483480915427208), (24, 0.10843091458082199), (25, 0.17080342769622803), (26, 0.13088981807231903), (27, 0.012119662016630173)], Temporal MSE: [(0, 5.255181349639315e-06), (1, 2.8603492410184117e-06), (2, 2.438069714116864e-05), (3, 0.00022181759413797408), (4, 0.00026127666933462024), (5, 0.0005733969155699015), (6, 0.0013069207780063152), (7, 0.0029646982438862324), (8, 0.0031195683404803276), (9, 0.006189312320202589), (10, 0.009633013978600502), (11, 0.01142830029129982), (12, 0.011214256286621094), (13, 0.014087299816310406), (14, 0.01787915825843811), (15, 0.021561844274401665), (16, 0.025190839543938637), (17, 0.02161891758441925), (18, 0.020725339651107788), (19, 0.021451139822602272), (20, 0.042425885796546936), (21, 0.038248077034950256), (22, 0.02899816632270813), (23, 0.040786828845739365), (24, 0.09701000154018402), (25, 0.13891859352588654), (26, 0.05384247377514839), (27, 0.004038508050143719)]
+
+ 74%|███████▍  | 37/50 [06:59<02:46, 12.83s/it]Time step 240, Spatial MSE: [(0, 5.803749445476569e-06), (1, 2.3442747988156043e-05), (2, 3.680096779135056e-05), (3, 2.9581809940282255e-05), (4, 1.3118824426783249e-05), (5, 2.680572470126208e-05), (6, 2.082931314362213e-05), (7, 3.673581522889435e-05), (8, 2.5149092834908515e-05), (9, 2.1255717001622543e-05), (10, 2.50787543336628e-05), (11, 3.3879168768180534e-05), (12, 3.647791163530201e-05), (13, 6.350670446408913e-05), (14, 5.943743963143788e-05), (15, 6.659387145191431e-05), (16, 0.00043159566121175885), (17, 5.495698860613629e-05), (18, 9.681218216428533e-05), (19, 9.724483243189752e-05), (20, 9.684310498414561e-05), (21, 0.0004954200703650713), (22, 0.0001877954346127808), (23, 0.0004392310802359134), (24, 0.0006774551002308726), (25, 0.0012757563963532448), (26, 0.002150088781490922), (27, 0.011690892279148102)], Temporal MSE: [(0, 5.105535365146352e-06), (1, 2.9097880087647354e-06), (2, 5.119793058838695e-06), (3, 9.195278835250065e-06), (4, 7.4011654760397505e-06), (5, 1.3835952813678887e-05), (6, 1.9826902644126676e-05), (7, 1.667886317591183e-05), (8, 1.893105400085915e-05), (9, 2.3244569092639722e-05), (10, 2.3907337890705094e-05), (11, 2.7884321752935648e-05), (12, 3.2303232728736475e-05), (13, 5.233272895566188e-05), (14, 4.3753978388849646e-05), (15, 4.4680622522719204e-05), (16, 3.7826986954314634e-05), (17, 3.764156281249598e-05), (18, 5.07024560647551e-05), (19, 5.8778776292456314e-05), (20, 8.392591553274542e-05), (21, 0.00011677950533339754), (22, 0.00012623349903151393), (23, 0.0002810831065289676), (24, 0.00038633521762676537), (25, 0.0008270667167380452), (26, 0.0008534283842891455), (27, 0.0003460639563854784)]
+
+ 76%|███████▌  | 38/50 [07:13<02:35, 13.00s/it]Time step 220, Spatial MSE: [(0, 1.0400010978628416e-05), (1, 2.4286437110276893e-05), (2, 6.441710866056383e-05), (3, 4.6467397623928264e-05), (4, 5.9619604144245386e-05), (5, 0.00012331217294558883), (6, 0.0003143158683087677), (7, 0.0004840052279178053), (8, 0.0004843204515054822), (9, 0.0005595250404439867), (10, 0.0009417195688001812), (11, 0.0018133491976186633), (12, 0.0020585914608091116), (13, 0.0023826579563319683), (14, 0.006229477934539318), (15, 0.0024210456758737564), (16, 0.010206339880824089), (17, 0.001279190881177783), (18, 0.0014452401082962751), (19, 0.0013014284195378423), (20, 0.002276267856359482), (21, 0.0019332205411046743), (22, 0.001399251981638372), (23, 0.003927611280232668), (24, 0.012595580890774727), (25, 0.023608040064573288), (26, 0.03196560963988304), (27, 0.008131683804094791)], Temporal MSE: [(0, 6.418529665097594e-06), (1, 3.2972207009152044e-06), (2, 1.7298623788519762e-05), (3, 5.279532342683524e-05), (4, 5.817969577037729e-05), (5, 0.00013284954184200615), (6, 0.00018243865633849055), (7, 0.00024606555234640837), (8, 0.00023244238400366157), (9, 0.00026723535847850144), (10, 0.0004282209265511483), (11, 0.0007009909022599459), (12, 0.0007487587863579392), (13, 0.0011514811776578426), (14, 0.0017855665646493435), (15, 0.001053656218573451), (16, 0.0007365333149209619), (17, 0.0006400413112714887), (18, 0.0007653979118913412), (19, 0.0006331898039206862), (20, 0.0014054849743843079), (21, 0.0011649304069578648), (22, 0.0009388975449837744), (23, 0.0021510049700737), (24, 0.004938963800668716), (25, 0.012201864272356033), (26, 0.005992262624204159), (27, 0.0014034545747563243)]
+
+ 78%|███████▊  | 39/50 [07:26<02:23, 13.08s/it]Time step 200, Spatial MSE: [(0, 6.303952886810293e-06), (1, 2.920571932918392e-05), (2, 6.850584031781182e-05), (3, 3.0686507670907304e-05), (4, 7.433046266669407e-05), (5, 0.00018128208466805518), (6, 0.0004684666928369552), (7, 0.0007596022915095091), (8, 0.0004180318210273981), (9, 0.00040992279537022114), (10, 0.0006828922196291387), (11, 0.0006090567912906408), (12, 0.0007948710699565709), (13, 0.006469734013080597), (14, 0.009022212587296963), (15, 0.005479124840348959), (16, 0.1612498164176941), (17, 0.0017129735788330436), (18, 0.002233457751572132), (19, 0.006377006880939007), (20, 0.005163286346942186), (21, 0.007089517079293728), (22, 0.00303479190915823), (23, 0.0032382227946072817), (24, 0.0059522236697375774), (25, 0.013339768163859844), (26, 0.00633544847369194), (27, 0.03873926028609276)], Temporal MSE: [(0, 6.921623480593553e-06), (1, 3.7011554923083168e-06), (2, 2.5167104467982426e-05), (3, 0.0001582456607138738), (4, 0.00014468819426838309), (5, 0.00014785825624130666), (6, 0.00027157331351190805), (7, 0.0002222116308985278), (8, 0.00020634621614590287), (9, 0.0003836220712400973), (10, 0.0006598088657483459), (11, 0.000808992306701839), (12, 0.0012658172054216266), (13, 0.007960663177073002), (14, 0.01052676048129797), (15, 0.0014282616320997477), (16, 0.0012529827654361725), (17, 0.0010803824989125133), (18, 0.0009788572788238525), (19, 0.001012787688523531), (20, 0.002438220428302884), (21, 0.002417665906250477), (22, 0.002328517846763134), (23, 0.00476468401029706), (24, 0.009349321946501732), (25, 0.010160229168832302), (26, 0.008449605666100979), (27, 0.004579557105898857)]
+
+ 80%|████████  | 40/50 [07:39<02:10, 13.03s/it]Time step 180, Spatial MSE: [(0, 1.03676329672453e-05), (1, 3.1247709557646886e-05), (2, 0.00012582849012687802), (3, 4.404550782055594e-05), (4, 0.00011050526518374681), (5, 0.00023053542827256024), (6, 0.00036339322105050087), (7, 0.0006733355112373829), (8, 0.0006451585795730352), (9, 0.0007443113136105239), (10, 0.0008662188774906099), (11, 0.0012983777560293674), (12, 0.0016309604980051517), (13, 0.007668417412787676), (14, 0.013408149592578411), (15, 0.004743422847241163), (16, 0.17610673606395721), (17, 0.0030123544856905937), (18, 0.004193232394754887), (19, 0.008624042384326458), (20, 0.006745191290974617), (21, 0.007402921561151743), (22, 0.006826425436884165), (23, 0.011059107258915901), (24, 0.03088497743010521), (25, 0.07014938443899155), (26, 0.06794627010822296), (27, 0.0367015041410923)], Temporal MSE: [(0, 8.434402843704447e-06), (1, 4.212184649077244e-06), (2, 1.642577990423888e-05), (3, 3.1938521715346724e-05), (4, 0.00010490126442164183), (5, 0.0002062020357698202), (6, 0.0002664162020664662), (7, 0.0003371460479684174), (8, 0.0003795504744630307), (9, 0.0004583284608088434), (10, 0.0006619801861234009), (11, 0.0012034982210025191), (12, 0.0011747170938178897), (13, 0.004856136627495289), (14, 0.008790817111730576), (15, 0.003652117680758238), (16, 0.0051934425719082355), (17, 0.0028066562954336405), (18, 0.0022889471147209406), (19, 0.0024157525040209293), (20, 0.00950400810688734), (21, 0.005225701257586479), (22, 0.003466172842308879), (23, 0.007194255944341421), (24, 0.016502372920513153), (25, 0.027772903442382812), (26, 0.012582727707922459), (27, 0.0031039707828313112)]
+
+ 82%|████████▏ | 41/50 [07:50<01:51, 12.43s/it]Time step 160, Spatial MSE: [(0, 7.538842965004733e-06), (1, 3.7364901800174266e-05), (2, 9.974413842428476e-05), (3, 3.2719359296606854e-05), (4, 1.9980823708465323e-05), (5, 4.504277603700757e-05), (6, 3.273781476309523e-05), (7, 6.675851182080805e-05), (8, 4.810609607375227e-05), (9, 3.389567427802831e-05), (10, 4.2495197703829035e-05), (11, 5.8757108490681276e-05), (12, 6.169554399093613e-05), (13, 0.00010324731556465849), (14, 9.39755336730741e-05), (15, 0.00010277690307702869), (16, 0.0001369711389997974), (17, 0.00010293662489857525), (18, 0.00012482613965403289), (19, 0.00016846002836246043), (20, 0.00015841700951568782), (21, 0.00018061994342133403), (22, 0.0002984559105243534), (23, 0.0007125994306989014), (24, 0.0011159204877912998), (25, 0.0019327385816723108), (26, 0.0025792638771235943), (27, 0.014371590688824654)], Temporal MSE: [(0, 8.445060302619822e-06), (1, 4.449128937267233e-06), (2, 7.843130333640147e-06), (3, 1.0679831575544085e-05), (4, 1.2307010365475435e-05), (5, 2.280498119944241e-05), (6, 2.8827789719798602e-05), (7, 2.9726299544563517e-05), (8, 3.719675078173168e-05), (9, 4.412418638821691e-05), (10, 4.436815288499929e-05), (11, 5.176108970772475e-05), (12, 5.469210009323433e-05), (13, 9.231968579115346e-05), (14, 6.052993921912275e-05), (15, 8.335991151398048e-05), (16, 8.059263200266287e-05), (17, 7.778631697874516e-05), (18, 0.00010429096437292174), (19, 9.98741015791893e-05), (20, 0.00014807592378929257), (21, 0.00013899322948418558), (22, 0.00017550771008245647), (23, 0.00038644325104542077), (24, 0.0005445482674986124), (25, 0.0010341220768168569), (26, 0.0012035827385261655), (27, 0.0006613535806536674)]
+
+ 84%|████████▍ | 42/50 [07:59<01:31, 11.47s/it]Time step 140, Spatial MSE: [(0, 1.132524266722612e-05), (1, 4.2038180254166946e-05), (2, 0.00028468467644415796), (3, 6.769635365344584e-05), (4, 0.00018747086869552732), (5, 0.0006054603145457804), (6, 0.0011737460736185312), (7, 0.002427330706268549), (8, 0.0036946022883057594), (9, 0.0057085175067186356), (10, 0.006107652094215155), (11, 0.009865224361419678), (12, 0.009582873433828354), (13, 0.011333536356687546), (14, 0.025649750605225563), (15, 0.029646137729287148), (16, 0.1877978891134262), (17, 0.020070068538188934), (18, 0.021571239456534386), (19, 0.01953330636024475), (20, 0.02042771317064762), (21, 0.02520466223359108), (22, 0.022610940039157867), (23, 0.04145980253815651), (24, 0.06920783966779709), (25, 0.1323905885219574), (26, 0.10595063120126724), (27, 0.014828345738351345)], Temporal MSE: [(0, 1.1442838513175957e-05), (1, 5.980005880701356e-06), (2, 3.259302684455179e-05), (3, 0.00017932023911271244), (4, 0.0002144600875908509), (5, 0.0004698218253906816), (6, 0.0010063264053314924), (7, 0.0016893985448405147), (8, 0.001785001833923161), (9, 0.003078399458900094), (10, 0.005116700194776058), (11, 0.006577855441719294), (12, 0.007607382256537676), (13, 0.011402185074985027), (14, 0.01667734421789646), (15, 0.015323172323405743), (16, 0.016040004789829254), (17, 0.015365956351161003), (18, 0.012355400249361992), (19, 0.013536888174712658), (20, 0.0318647138774395), (21, 0.025424586609005928), (22, 0.013046535663306713), (23, 0.02171732857823372), (24, 0.05230589956045151), (25, 0.07721651345491409), (26, 0.042723648250103), (27, 0.004700909834355116)]
+
+ 86%|████████▌ | 43/50 [08:09<01:16, 10.97s/it]Time step 120, Spatial MSE: [(0, 1.014676035993034e-05), (1, 5.65840418857988e-05), (2, 0.00017974372894968837), (3, 4.337618520366959e-05), (4, 2.8334752641967498e-05), (5, 7.32943371986039e-05), (6, 5.035313733969815e-05), (7, 7.428898970829323e-05), (8, 8.044188871281222e-05), (9, 5.2845527534373105e-05), (10, 6.186889368109405e-05), (11, 8.315065497299656e-05), (12, 9.014463284984231e-05), (13, 0.00016697718820068985), (14, 0.00040310522308573127), (15, 0.000254814192885533), (16, 0.010972666554152966), (17, 0.0001777374418452382), (18, 0.0002096958487527445), (19, 0.0004339782753959298), (20, 0.0003552912676241249), (21, 0.0004223543219268322), (22, 0.000458389928098768), (23, 0.001017228001728654), (24, 0.0016428242670372128), (25, 0.002804710529744625), (26, 0.0032200301066040993), (27, 0.018169470131397247)], Temporal MSE: [(0, 1.1835043551400304e-05), (1, 6.221391686267452e-06), (2, 1.111538313125493e-05), (3, 1.6754214811953716e-05), (4, 1.935931868501939e-05), (5, 3.6305151297710836e-05), (6, 4.123879625694826e-05), (7, 4.756351336254738e-05), (8, 6.050617230357602e-05), (9, 7.341588934650645e-05), (10, 7.028202526271343e-05), (11, 7.963346433825791e-05), (12, 8.33520243759267e-05), (13, 0.00018111124518327415), (14, 0.00022177238133735955), (15, 0.00013412781117949635), (16, 0.00016374181723222136), (17, 0.00015389933832921088), (18, 0.00017319340258836746), (19, 0.00020342964853625745), (20, 0.0003521853359416127), (21, 0.00032831807038746774), (22, 0.00034386717015877366), (23, 0.0005691039259545505), (24, 0.0008394530741497874), (25, 0.0014532747445628047), (26, 0.0018052535597234964), (27, 0.0010736766271293163)]
+
+ 88%|████████▊ | 44/50 [08:18<01:02, 10.40s/it]Time step 100, Spatial MSE: [(0, 1.3619075616588816e-05), (1, 6.810818740632385e-05), (2, 0.0005542298313230276), (3, 7.332057430176064e-05), (4, 0.00015340073150582612), (5, 0.00033433621865697205), (6, 0.0007143127149902284), (7, 0.0012418609112501144), (8, 0.001358029549010098), (9, 0.0015860372222959995), (10, 0.001711806864477694), (11, 0.002452633110806346), (12, 0.0027585590723901987), (13, 0.0037447114009410143), (14, 0.01188123319298029), (15, 0.009795312769711018), (16, 0.20478352904319763), (17, 0.0047591691836714745), (18, 0.00611802889034152), (19, 0.008013749495148659), (20, 0.0099231768399477), (21, 0.007876698859035969), (22, 0.007041969336569309), (23, 0.015290554612874985), (24, 0.03135424107313156), (25, 0.06573565304279327), (26, 0.06640225648880005), (27, 0.018385551869869232)], Temporal MSE: [(0, 1.6722577129257843e-05), (1, 8.479518328385893e-06), (2, 2.6326455554226413e-05), (3, 4.6313161874422804e-05), (4, 9.301219688495621e-05), (5, 0.00017479245434515178), (6, 0.00032363313948735595), (7, 0.00042003870476037264), (8, 0.0005157117266207933), (9, 0.0007723650778643787), (10, 0.00111020274925977), (11, 0.0014079173561185598), (12, 0.0016783452592790127), (13, 0.0030743731185793877), (14, 0.0066369506530463696), (15, 0.0032536748331040144), (16, 0.003079174319282174), (17, 0.003219811711460352), (18, 0.0027842815034091473), (19, 0.003006698563694954), (20, 0.0052391900680959225), (21, 0.005453955847769976), (22, 0.003585456870496273), (23, 0.007098644971847534), (24, 0.012846536934375763), (25, 0.0224434956908226), (26, 0.015542500652372837), (27, 0.0021631564013659954)]
+
+ 90%|█████████ | 45/50 [08:29<00:53, 10.73s/it]Time step 80, Spatial MSE: [(0, 1.4447576177190058e-05), (1, 9.384746954310685e-05), (2, 0.0005241361213847995), (3, 4.9721002142177895e-05), (4, 7.787808135617524e-05), (5, 0.00020989855693187565), (6, 0.00032656139228492975), (7, 0.0005421449313871562), (8, 0.0006668293499387801), (9, 0.0006778701208531857), (10, 0.0007189168827608228), (11, 0.0010284458985552192), (12, 0.0010745353065431118), (13, 0.0015033185482025146), (14, 0.0056177424266934395), (15, 0.004275464452803135), (16, 0.11929994076490402), (17, 0.0021029943600296974), (18, 0.0022818997967988253), (19, 0.003825295716524124), (20, 0.004811046179383993), (21, 0.003510646056383848), (22, 0.0028548187110573053), (23, 0.00629898626357317), (24, 0.013103842735290527), (25, 0.023632526397705078), (26, 0.024133801460266113), (27, 0.017845986410975456)], Temporal MSE: [(0, 1.9534578314051032e-05), (1, 9.941217285813764e-06), (2, 1.8459848433849402e-05), (3, 2.915789264079649e-05), (4, 4.620523759513162e-05), (5, 8.222550968639553e-05), (6, 0.00012804489233531058), (7, 0.00016745756147429347), (8, 0.0002400792291155085), (9, 0.00031493930146098137), (10, 0.00037210408481769264), (11, 0.0005088375182822347), (12, 0.0006319676758721471), (13, 0.0011848019203171134), (14, 0.0032037622295320034), (15, 0.0012588939862325788), (16, 0.0011741043999791145), (17, 0.0011678090086206794), (18, 0.0011076410301029682), (19, 0.001106304582208395), (20, 0.0018350196769461036), (21, 0.0021810124162584543), (22, 0.0014602611772716045), (23, 0.002921280451118946), (24, 0.005215717013925314), (25, 0.009005840867757797), (26, 0.006802782416343689), (27, 0.0016486356034874916)]
+
+ 92%|█████████▏| 46/50 [08:40<00:42, 10.58s/it]Time step 60, Spatial MSE: [(0, 1.941915434144903e-05), (1, 0.00013585251872427762), (2, 0.0007253932417370379), (3, 6.580213812412694e-05), (4, 0.00010015218867920339), (5, 0.00026981672272086143), (6, 0.00046322925481945276), (7, 0.0007413384737446904), (8, 0.0009382484131492674), (9, 0.0009950920939445496), (10, 0.0010140897938981652), (11, 0.0013993806205689907), (12, 0.0014636769192293286), (13, 0.002010520314797759), (14, 0.005936496891081333), (15, 0.0051802643574774265), (16, 0.09868362545967102), (17, 0.002491311402991414), (18, 0.002723289653658867), (19, 0.004041580483317375), (20, 0.005515172146260738), (21, 0.0038891835138201714), (22, 0.0033740350045263767), (23, 0.007380313239991665), (24, 0.015903277322649956), (25, 0.02648097276687622), (26, 0.026735110208392143), (27, 0.01887711137533188)], Temporal MSE: [(0, 2.632636642374564e-05), (1, 1.3486055649991613e-05), (2, 2.538062835810706e-05), (3, 3.897372153005563e-05), (4, 6.017350096954033e-05), (5, 0.00011158468259964138), (6, 0.000165295583428815), (7, 0.0002251688129035756), (8, 0.0003254689509049058), (9, 0.0004126659769099206), (10, 0.0004599228850565851), (11, 0.0006343972636386752), (12, 0.0007572349277324975), (13, 0.0015002733562141657), (14, 0.003245316445827484), (15, 0.0014506467850878835), (16, 0.0013427013764157891), (17, 0.0013112486340105534), (18, 0.001323794131167233), (19, 0.00132024718914181), (20, 0.0021134233102202415), (21, 0.0022667539305984974), (22, 0.001675651059485972), (23, 0.0033725854009389877), (24, 0.0064964136108756065), (25, 0.010748138651251793), (26, 0.008822426199913025), (27, 0.0032292022369802)]
+
+ 94%|█████████▍| 47/50 [08:52<00:33, 11.06s/it]Time step 40, Spatial MSE: [(0, 2.9115572033333592e-05), (1, 0.00021318781364243478), (2, 0.0010448468383401632), (3, 9.260905790142715e-05), (4, 0.00013825579662807286), (5, 0.000369123910786584), (6, 0.0007149695884436369), (7, 0.0010899151675403118), (8, 0.0014467558357864618), (9, 0.0015990589745342731), (10, 0.00157065293751657), (11, 0.0020141650456935167), (12, 0.0019666338339447975), (13, 0.002559286542236805), (14, 0.006008310243487358), (15, 0.00661510182544589), (16, 0.075953409075737), (17, 0.0035108369775116444), (18, 0.0035181541461497545), (19, 0.004652280826121569), (20, 0.006545270327478647), (21, 0.005321648437529802), (22, 0.004469042178243399), (23, 0.009449897333979607), (24, 0.021101750433444977), (25, 0.03221184015274048), (26, 0.03405505046248436), (27, 0.023721596226096153)], Temporal MSE: [(0, 3.869230204145424e-05), (1, 1.9758299458771944e-05), (2, 4.2673553252825513e-05), (3, 5.5780790717108175e-05), (4, 8.20510831545107e-05), (5, 0.00016267580213025212), (6, 0.0002390780282439664), (7, 0.0003340073744766414), (8, 0.00046651283628307283), (9, 0.0005833240575157106), (10, 0.000638559169601649), (11, 0.0008473120979033411), (12, 0.0009775282815098763), (13, 0.001580424141138792), (14, 0.003093376290053129), (15, 0.0017889929004013538), (16, 0.0017151316860690713), (17, 0.0016640425892546773), (18, 0.0016755455872043967), (19, 0.0017304344801232219), (20, 0.002725244965404272), (21, 0.0027007898315787315), (22, 0.002141089178621769), (23, 0.004330525174736977), (24, 0.009090432897210121), (25, 0.01430395245552063), (26, 0.013605662621557713), (27, 0.012793057598173618)]
+
+ 96%|█████████▌| 48/50 [09:06<00:23, 11.84s/it]Time step 20, Spatial MSE: [(0, 5.563714512391016e-05), (1, 0.00041442536166869104), (2, 0.0017232925165444613), (3, 0.00015838629042264074), (4, 0.0002853392797987908), (5, 0.0005983884911984205), (6, 0.0013557299971580505), (7, 0.002090390073135495), (8, 0.00268981303088367), (9, 0.0032478373032063246), (10, 0.0029900409281253815), (11, 0.0032546906732022762), (12, 0.0028897300362586975), (13, 0.004106788896024227), (14, 0.013001169078052044), (15, 0.009855744428932667), (16, 0.22179900109767914), (17, 0.006136564537882805), (18, 0.005541506223380566), (19, 0.009019803255796432), (20, 0.01260731928050518), (21, 0.009780948981642723), (22, 0.00753101147711277), (23, 0.015036246739327908), (24, 0.03384387493133545), (25, 0.051872894167900085), (26, 0.05952586606144905), (27, 0.051093559712171555)], Temporal MSE: [(0, 6.93951005814597e-05), (1, 3.5195753298467025e-05), (2, 9.50151588767767e-05), (3, 9.506967762717977e-05), (4, 0.0001380765752401203), (5, 0.00028402896714396775), (6, 0.00048069702461361885), (7, 0.0005936731467954814), (8, 0.0007968597346916795), (9, 0.0009866845794022083), (10, 0.0010153742041438818), (11, 0.0012704418040812016), (12, 0.0013789668446406722), (13, 0.002891979180276394), (14, 0.0062720272690057755), (15, 0.0027041591238230467), (16, 0.0027173056732863188), (17, 0.002598498249426484), (18, 0.0027927933260798454), (19, 0.0028297428507357836), (20, 0.004620763938874006), (21, 0.004908340983092785), (22, 0.0036231407430022955), (23, 0.007048941217362881), (24, 0.016718409955501556), (25, 0.025656448677182198), (26, 0.0314086452126503), (27, 0.12672656774520874)]
+
+ 98%|█████████▊| 49/50 [09:19<00:12, 12.24s/it]Time step 0, Spatial MSE: [(0, 0.0003944486379623413), (1, 0.0023468821309506893), (2, 0.009191160090267658), (3, 0.0009031372610479593), (4, 0.003975528292357922), (5, 0.003102887189015746), (6, 0.007318263407796621), (7, 0.012879046611487865), (8, 0.014482881873846054), (9, 0.02092566154897213), (10, 0.018707241863012314), (11, 0.01247959304600954), (12, 0.011579183861613274), (13, 0.013807559385895729), (14, 0.02245265617966652), (15, 0.033814687281847), (16, 0.16792869567871094), (17, 0.029789533466100693), (18, 0.02717258781194687), (19, 0.03181732818484306), (20, 0.04358433932065964), (21, 0.04016115888953209), (22, 0.04126279056072235), (23, 0.10644857585430145), (24, 0.19897542893886566), (25, 0.7498639225959778), (26, 1.166369915008545), (27, 1.8042784929275513)], Temporal MSE: [(0, 0.0005470088799484074), (1, 0.00022154177713673562), (2, 0.000813068007119), (3, 0.0005306490347720683), (4, 0.0010106668341904879), (5, 0.00148116541095078), (6, 0.004162702243775129), (7, 0.0032546170987188816), (8, 0.005924593657255173), (9, 0.006042442750185728), (10, 0.005894841626286507), (11, 0.004948852583765984), (12, 0.00486554391682148), (13, 0.007039303425699472), (14, 0.011413216590881348), (15, 0.009718264453113079), (16, 0.009472331963479519), (17, 0.008489100262522697), (18, 0.010071326047182083), (19, 0.012108881957828999), (20, 0.021994736045598984), (21, 0.016732564195990562), (22, 0.022374145686626434), (23, 0.03986160457134247), (24, 0.10849697887897491), (25, 0.19899123907089233), (26, 0.5018264651298523), (27, 5.080696105957031)]
+
+100%|██████████| 50/50 [09:28<00:00, 11.43s/it]
+100%|██████████| 50/50 [09:28<00:00, 11.37s/it]
+Saved video to ./samples/latte/Snow_falling_over_multiple_houses_and_trees_on_winter_landscape_against_night_sky._christmas_festivity_and_celebration_concept._0000.mp4
+Saved spatial MLP MSE outputs to ./samples/latte/Snow_falling_over_multiple_houses_and_trees_on_winter_landscape_against_night_sky._christmas_festivity_and_celebration_concept._0000_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/latte/Snow_falling_over_multiple_houses_and_trees_on_winter_landscape_against_night_sky._christmas_festivity_and_celebration_concept._0000_temporal_mlp_mse.pt
+Processing the (Time Lapse of the rising sun over a tree in an open rural landscape, with clouds in the blue sky beautifully playing with the rays of light) prompt
+
+  0%|          | 0/50 [00:00<?, ?it/s]
+  2%|▏         | 1/50 [00:04<03:49,  4.68s/it]Time step 960, Spatial MSE: [(0, 1.0291370927006938e-05), (1, 1.6777952396296314e-06), (2, 2.3058502847561613e-05), (3, 2.879216117435135e-05), (4, 8.626376802567393e-05), (5, 5.554899325943552e-05), (6, 0.00012768646411132067), (7, 0.00010363398905610666), (8, 0.0002134238020516932), (9, 0.0007059607887640595), (10, 0.0009647549013607204), (11, 0.0009894123068079352), (12, 0.0010963142849504948), (13, 0.0011610063957050443), (14, 0.002676728181540966), (15, 0.004630755167454481), (16, 0.005243709776550531), (17, 0.002549136755988002), (18, 0.0034611907321959734), (19, 0.003309345804154873), (20, 0.006982249673455954), (21, 0.002769151236861944), (22, 0.0012405890738591552), (23, 0.002154034562408924), (24, 0.005662842188030481), (25, 0.006162107922136784), (26, 0.004884403198957443), (27, 0.0020602988079190254)], Temporal MSE: [(0, 1.4467439541476779e-05), (1, 4.7820835789025296e-06), (2, 4.700708814198151e-06), (3, 9.617844625608996e-06), (4, 9.963448974303901e-06), (5, 1.8723789253272116e-05), (6, 4.357335274107754e-05), (7, 5.305467493599281e-05), (8, 4.6666766138514504e-05), (9, 0.00021130974346306175), (10, 0.0005220196326263249), (11, 0.0004956077900715172), (12, 0.0005064759170636535), (13, 0.0009119012975133955), (14, 0.0005899110692553222), (15, 0.0005933095817454159), (16, 0.000922688574064523), (17, 0.000612635281868279), (18, 0.00040114708826877177), (19, 0.00023978232638910413), (20, 0.0005283501814119518), (21, 0.0003777222300413996), (22, 0.0004506109980866313), (23, 0.0004770472296513617), (24, 0.0006275575142353773), (25, 0.0007114552427083254), (26, 0.0003657493507489562), (27, 1.1041217476304155e-05)]
+
+  4%|▍         | 2/50 [00:15<06:26,  8.06s/it]Time step 940, Spatial MSE: [(0, 1.16629335025209e-05), (1, 1.9337062440172303e-06), (2, 2.117471922247205e-05), (3, 3.2507854484720156e-05), (4, 6.87756619299762e-05), (5, 7.950952567625791e-05), (6, 0.0003451377560850233), (7, 0.00025515147717669606), (8, 0.0005988048505969346), (9, 0.0016540068900212646), (10, 0.0026996294036507607), (11, 0.003013395005837083), (12, 0.0026141968555748463), (13, 0.00260116346180439), (14, 0.004341715946793556), (15, 0.005841490812599659), (16, 0.005810907110571861), (17, 0.0035245800390839577), (18, 0.004256190732121468), (19, 0.004731772933155298), (20, 0.009705360978841782), (21, 0.003259568940848112), (22, 0.0019315851386636496), (23, 0.0028535379096865654), (24, 0.005619304720312357), (25, 0.009334924630820751), (26, 0.007190762087702751), (27, 0.0022272407077252865)], Temporal MSE: [(0, 1.5588155292789452e-05), (1, 5.07713502884144e-06), (2, 5.8265945881430525e-06), (3, 1.1435823580541182e-05), (4, 1.2684536159213167e-05), (5, 3.0297604098450392e-05), (6, 8.768297266215086e-05), (7, 9.131072874879465e-05), (8, 0.00011591619841055945), (9, 0.00043719832319766283), (10, 0.0008341147913597524), (11, 0.0009740165551193058), (12, 0.0010889549739658833), (13, 0.002009725896641612), (14, 0.0015029546339064837), (15, 0.0010967383859679103), (16, 0.0013957034097984433), (17, 0.000903816195204854), (18, 0.0006333680939860642), (19, 0.0005821384256705642), (20, 0.001295287162065506), (21, 0.0008749797125346959), (22, 0.0008671542163938284), (23, 0.0008353369194082916), (24, 0.001173202646896243), (25, 0.0012957447906956077), (26, 0.0005599925061687827), (27, 2.7299460271024145e-05)]
+
+  6%|▌         | 3/50 [00:24<06:41,  8.54s/it]Time step 920, Spatial MSE: [(0, 1.1928984349651728e-05), (1, 2.0172515178273898e-06), (2, 2.474819848430343e-05), (3, 3.132737765554339e-05), (4, 5.3410607506521046e-05), (5, 7.892421854194254e-05), (6, 0.0004879510379396379), (7, 0.00030750062433071434), (8, 0.0005099503905512393), (9, 0.0011434839107096195), (10, 0.0015376490773633122), (11, 0.0013923072256147861), (12, 0.001453357981517911), (13, 0.0014543007127940655), (14, 0.0037509289104491472), (15, 0.004432461690157652), (16, 0.018778493627905846), (17, 0.003510840702801943), (18, 0.0030409437604248524), (19, 0.003951060585677624), (20, 0.008380446583032608), (21, 0.0032041009981185198), (22, 0.001721806707791984), (23, 0.002289828145876527), (24, 0.004593018442392349), (25, 0.009858447127044201), (26, 0.007372604683041573), (27, 0.002524870913475752)], Temporal MSE: [(0, 1.7946640582522377e-05), (1, 5.862982561666286e-06), (2, 6.444381597248139e-06), (3, 1.3579057849710807e-05), (4, 1.322427488048561e-05), (5, 3.476296114968136e-05), (6, 8.49245407152921e-05), (7, 9.078289440367371e-05), (8, 0.00010694416414480656), (9, 0.0004202357667963952), (10, 0.0004926861729472876), (11, 0.0007615737849846482), (12, 0.0007014622678980231), (13, 0.0009070556261576712), (14, 0.0009453321690671146), (15, 0.0006604478694498539), (16, 0.0009597347234375775), (17, 0.0006328768795356154), (18, 0.00040309710311703384), (19, 0.0003375046653673053), (20, 0.0006428298074752092), (21, 0.000637321500107646), (22, 0.0006678087520413101), (23, 0.0006032026140019298), (24, 0.0008789940620772541), (25, 0.0011205506743863225), (26, 0.0005176365957595408), (27, 2.2153020836412907e-05)]
+
+  8%|▊         | 4/50 [00:37<08:03, 10.51s/it]Time step 900, Spatial MSE: [(0, 1.5543146218988113e-05), (1, 2.3132599835662404e-06), (2, 2.9946913855383173e-05), (3, 3.47514906025026e-05), (4, 4.438188625499606e-05), (5, 8.055921352934092e-05), (6, 0.0006196835311129689), (7, 0.000279537052847445), (8, 0.0003852292720694095), (9, 0.0007216442609205842), (10, 0.0009813670767471194), (11, 0.0009295684867538512), (12, 0.0012503236066550016), (13, 0.0012709449511021376), (14, 0.003039287170395255), (15, 0.0034959237091243267), (16, 0.005875564180314541), (17, 0.0027596859727054834), (18, 0.0027226251550018787), (19, 0.0032480729278177023), (20, 0.006820177659392357), (21, 0.0022637394722551107), (22, 0.0016990957083180547), (23, 0.0020909574814140797), (24, 0.00362547324039042), (25, 0.01030823215842247), (26, 0.00935493316501379), (27, 0.003174822311848402)], Temporal MSE: [(0, 2.2089563572080806e-05), (1, 6.8407412072701845e-06), (2, 7.940197974676266e-06), (3, 1.8072812963509932e-05), (4, 1.5261763110174797e-05), (5, 4.607038135873154e-05), (6, 7.6492884545587e-05), (7, 7.771505624987185e-05), (8, 8.951822383096442e-05), (9, 0.00040469656232744455), (10, 0.00042361250962130725), (11, 0.0006355736986733973), (12, 0.0005378212663345039), (13, 0.0006457036943174899), (14, 0.0006405283929780126), (15, 0.0005322553915902972), (16, 0.0007353763794526458), (17, 0.0005727692623622715), (18, 0.0003336863883305341), (19, 0.0002923663705587387), (20, 0.0005418651271611452), (21, 0.0005674393032677472), (22, 0.0006784012075513601), (23, 0.0005876171053387225), (24, 0.0008736629388295114), (25, 0.0012556068832054734), (26, 0.0005722676869481802), (27, 2.413800211797934e-05)]
+
+ 10%|█         | 5/50 [00:48<08:02, 10.72s/it]Time step 880, Spatial MSE: [(0, 1.2167769455118105e-05), (1, 2.2439007807406597e-06), (2, 3.754181307158433e-05), (3, 2.6518264348851517e-05), (4, 3.9330236177193e-05), (5, 7.520697545260191e-05), (6, 0.0005727165844291449), (7, 0.0002635753189679235), (8, 0.0004862745990976691), (9, 0.0008781935321167111), (10, 0.0013188078301027417), (11, 0.0012891204096376896), (12, 0.0017848159186542034), (13, 0.0018990457756444812), (14, 0.0038356035947799683), (15, 0.005397165194153786), (16, 0.007533167954534292), (17, 0.003859645687043667), (18, 0.0032321331091225147), (19, 0.0031306662131100893), (20, 0.0060943542048335075), (21, 0.002079338300973177), (22, 0.0023287986405193806), (23, 0.0030661718919873238), (24, 0.004567164462059736), (25, 0.01648593321442604), (26, 0.009449435397982597), (27, 0.002998826326802373)], Temporal MSE: [(0, 1.873678047559224e-05), (1, 6.352763648465043e-06), (2, 7.656645720999222e-06), (3, 1.5069360415509436e-05), (4, 1.558532858325634e-05), (5, 4.620305844582617e-05), (6, 8.08395489002578e-05), (7, 7.955603359732777e-05), (8, 9.962646436179057e-05), (9, 0.0004306127957534045), (10, 0.00055243622045964), (11, 0.0006650285213254392), (12, 0.0006440876168198884), (13, 0.0008116127573885024), (14, 0.0008891396573744714), (15, 0.0008519230759702623), (16, 0.0010396720608696342), (17, 0.0008835989283397794), (18, 0.0005885881837457418), (19, 0.0005556573742069304), (20, 0.001088766846805811), (21, 0.0006777294329367578), (22, 0.0007626353180967271), (23, 0.0008111410425044596), (24, 0.0014659981243312359), (25, 0.0018279618816450238), (26, 0.0008856205968186259), (27, 3.080781243625097e-05)]
+
+ 12%|█▏        | 6/50 [00:59<07:47, 10.63s/it]Time step 860, Spatial MSE: [(0, 4.485650424612686e-05), (1, 7.3885407800844405e-06), (2, 4.2154566472163424e-05), (3, 8.275822619907558e-05), (4, 0.00014970717893447727), (5, 0.00017734704306349158), (6, 0.0015933408867567778), (7, 0.0006419529090635478), (8, 0.0013200194807723165), (9, 0.0017233105609193444), (10, 0.0024462745059281588), (11, 0.0026230113580822945), (12, 0.0035101445391774178), (13, 0.0038537979125976562), (14, 0.006656809709966183), (15, 0.008989711292088032), (16, 0.00755499629303813), (17, 0.006001111585646868), (18, 0.007283526938408613), (19, 0.0067722126841545105), (20, 0.011794843710958958), (21, 0.005143808666616678), (22, 0.003996430430561304), (23, 0.004515802022069693), (24, 0.007131101563572884), (25, 0.02136288583278656), (26, 0.02832772582769394), (27, 0.01161952968686819)], Temporal MSE: [(0, 6.896010017953813e-05), (1, 1.9452547348919325e-05), (2, 2.2830283342045732e-05), (3, 3.991711128037423e-05), (4, 4.1113926272373646e-05), (5, 0.00018173728312831372), (6, 0.00017267138173338026), (7, 0.000186332588782534), (8, 0.00023236814013216645), (9, 0.0009302946273237467), (10, 0.0009670336148701608), (11, 0.0010866390075534582), (12, 0.0011802216758951545), (13, 0.0016232887282967567), (14, 0.0015672824811190367), (15, 0.001451130723580718), (16, 0.0018175640143454075), (17, 0.0013930051354691386), (18, 0.001064971787855029), (19, 0.0009519612067379057), (20, 0.0020874703768640757), (21, 0.0015537792351096869), (22, 0.002042770851403475), (23, 0.001828733249567449), (24, 0.0033837631344795227), (25, 0.004438322503119707), (26, 0.002026140224188566), (27, 0.00012909772340208292)]
+
+ 14%|█▍        | 7/50 [01:09<07:33, 10.54s/it]Time step 840, Spatial MSE: [(0, 4.90929505758686e-06), (1, 2.827422576956451e-06), (2, 5.640684685204178e-05), (3, 1.7297210433753207e-05), (4, 3.1088533432921395e-05), (5, 6.567050877492875e-05), (6, 0.0003516781725920737), (7, 0.00020269847300369292), (8, 0.0003721076063811779), (9, 0.00047409109538421035), (10, 0.0005902175325900316), (11, 0.0006026846822351217), (12, 0.0006759900134056807), (13, 0.0007400217582471669), (14, 0.0011117130052298307), (15, 0.001131586846895516), (16, 0.0008819217910058796), (17, 0.0007316719274967909), (18, 0.0007410519174300134), (19, 0.0007665685843676329), (20, 0.000812110083643347), (21, 0.000555636128410697), (22, 0.0006579401670023799), (23, 0.0009208466508425772), (24, 0.0017629247158765793), (25, 0.00617991341277957), (26, 0.0031091433484107256), (27, 0.004540258552879095)], Temporal MSE: [(0, 8.432507456745952e-06), (1, 3.0873338801029604e-06), (2, 7.213170192699181e-06), (3, 1.994378362724092e-05), (4, 1.5293491742340848e-05), (5, 2.792897430481389e-05), (6, 5.3540341468760744e-05), (7, 4.977906428393908e-05), (8, 6.536549335578457e-05), (9, 0.00020768675312865525), (10, 0.00023999121913220733), (11, 0.0002983674348797649), (12, 0.00022987573174759746), (13, 0.00032840814674273133), (14, 0.00033007387537509203), (15, 0.00027882264112122357), (16, 0.0002880778338294476), (17, 0.00022279241238720715), (18, 0.00022574487957172096), (19, 0.0002505587472114712), (20, 0.0003510502865538001), (21, 0.00036556171835400164), (22, 0.00031469634268432856), (23, 0.0003514404525049031), (24, 0.0007402502233162522), (25, 0.0009319037199020386), (26, 0.00043329843902029097), (27, 3.746019137906842e-05)]
+
+ 16%|█▌        | 8/50 [01:22<07:51, 11.22s/it]Time step 820, Spatial MSE: [(0, 1.2841161151300184e-05), (1, 2.623301725179772e-06), (2, 5.916530062677339e-05), (3, 3.4240303648402914e-05), (4, 0.0001006040329230018), (5, 8.625756890978664e-05), (6, 0.0005238121375441551), (7, 0.00025078156613744795), (8, 0.00040244151023216546), (9, 0.0004901056527160108), (10, 0.0009384815348312259), (11, 0.0007287621265277267), (12, 0.0008019370725378394), (13, 0.0008021869580261409), (14, 0.0015572498086839914), (15, 0.00162200175691396), (16, 0.0019058274338021874), (17, 0.0016425427747890353), (18, 0.001521249650977552), (19, 0.0015089118387550116), (20, 0.0028553889133036137), (21, 0.001002308796159923), (22, 0.001137137645855546), (23, 0.0013805797789245844), (24, 0.002599554369226098), (25, 0.010695576667785645), (26, 0.010129964910447598), (27, 0.003849684726446867)], Temporal MSE: [(0, 1.865886770247016e-05), (1, 6.247684723348357e-06), (2, 1.4935008948668838e-05), (3, 2.959597622975707e-05), (4, 2.5233737687813118e-05), (5, 5.969698031549342e-05), (6, 7.555291813332587e-05), (7, 6.930408562766388e-05), (8, 8.506785525241867e-05), (9, 0.00034321378916502), (10, 0.0004220662813168019), (11, 0.0004515236068982631), (12, 0.00038165791193023324), (13, 0.0005563626764342189), (14, 0.00038394343573600054), (15, 0.0003716862993314862), (16, 0.00041817367309704423), (17, 0.00035880631185136735), (18, 0.00028347104671411216), (19, 0.00026539794635027647), (20, 0.00041433729347772896), (21, 0.0005855322233401239), (22, 0.0006193217122927308), (23, 0.0005595986149273813), (24, 0.0009674992761574686), (25, 0.0015410134801641107), (26, 0.0007717108819633722), (27, 4.625129804480821e-05)]
+
+ 18%|█▊        | 9/50 [01:33<07:42, 11.28s/it]Time step 800, Spatial MSE: [(0, 1.2603101822605822e-05), (1, 3.0768946999160107e-06), (2, 7.446100062225014e-05), (3, 3.3917458495125175e-05), (4, 0.0001266199251404032), (5, 7.903169898781925e-05), (6, 0.0003104392089881003), (7, 0.00021162214397918433), (8, 0.00030474731465801597), (9, 0.0004367924702819437), (10, 0.0010355535196140409), (11, 0.0007068286067806184), (12, 0.0007162454421631992), (13, 0.0007367236539721489), (14, 0.0014791626017540693), (15, 0.0018263690872117877), (16, 0.0022047513630241156), (17, 0.0018778472440317273), (18, 0.0017827802803367376), (19, 0.0014969996409490705), (20, 0.002835272578522563), (21, 0.0012860328424721956), (22, 0.0011478678788989782), (23, 0.0014237506547942758), (24, 0.0028212082106620073), (25, 0.010832982137799263), (26, 0.009615353308618069), (27, 0.004407317843288183)], Temporal MSE: [(0, 1.8684902897803113e-05), (1, 6.003977887303336e-06), (2, 1.6246189261437394e-05), (3, 3.045230550924316e-05), (4, 2.5787745471461676e-05), (5, 4.8801237426232547e-05), (6, 7.10049134795554e-05), (7, 6.470927473856136e-05), (8, 7.662814459763467e-05), (9, 0.0002805170661304146), (10, 0.0003893535176757723), (11, 0.0003863839665427804), (12, 0.0003358326794113964), (13, 0.0005360755021683872), (14, 0.0003278460935689509), (15, 0.00037756451638415456), (16, 0.00044739380246028304), (17, 0.00042434249189682305), (18, 0.0003117471351288259), (19, 0.0002553383819758892), (20, 0.00040884449845179915), (21, 0.0006029030773788691), (22, 0.0005900943069718778), (23, 0.0005326976533979177), (24, 0.0009711227612569928), (25, 0.0016404162161052227), (26, 0.000866990361828357), (27, 5.853689071955159e-05)]
+
+ 20%|██        | 10/50 [01:44<07:19, 10.99s/it]Time step 780, Spatial MSE: [(0, 9.571605005476158e-06), (1, 2.7580190362641588e-06), (2, 6.830136408098042e-05), (3, 2.047278576355893e-05), (4, 3.026938793482259e-05), (5, 3.391184509382583e-05), (6, 5.361400326364674e-05), (7, 0.00010504696547286585), (8, 0.0001578210503794253), (9, 0.00018057874694932252), (10, 0.0007241485873237252), (11, 0.0005985572934150696), (12, 0.0005194916157051921), (13, 0.0006403847364708781), (14, 0.0011660022428259254), (15, 0.0011546839959919453), (16, 0.001240390818566084), (17, 0.0013835171703249216), (18, 0.0011373579036444426), (19, 0.0009533858974464238), (20, 0.0015750613529235125), (21, 0.0006698619108647108), (22, 0.0006072537507861853), (23, 0.001260029966942966), (24, 0.0025584986433386803), (25, 0.006907187402248383), (26, 0.0031636161729693413), (27, 0.005156833212822676)], Temporal MSE: [(0, 1.308682476519607e-05), (1, 5.1081806304864585e-06), (2, 1.090810474124737e-05), (3, 1.4070961697143503e-05), (4, 1.2949152733199298e-05), (5, 2.6595847884891555e-05), (6, 5.839210643898696e-05), (7, 0.000127820938359946), (8, 5.8005382015835494e-05), (9, 0.00023141306883189827), (10, 0.0004924791282974184), (11, 0.0003420168359298259), (12, 0.00027699218480847776), (13, 0.0004965775297023356), (14, 0.0004766264755744487), (15, 0.0005215967539697886), (16, 0.0009423077572137117), (17, 0.0006898112478666008), (18, 0.0008487201412208378), (19, 0.0003768220776692033), (20, 0.001184945460408926), (21, 0.0007721681031398475), (22, 0.0002975235693156719), (23, 0.000632311450317502), (24, 0.0010907913092523813), (25, 0.0014870772138237953), (26, 0.0005596532137133181), (27, 4.649108450394124e-05)]
+
+ 22%|██▏       | 11/50 [01:57<07:39, 11.79s/it]Time step 760, Spatial MSE: [(0, 9.633865374780726e-06), (1, 3.2270422707370017e-06), (2, 5.690493344445713e-05), (3, 1.0938477316813078e-05), (4, 1.1219064617762342e-05), (5, 6.6439197325962596e-06), (6, 1.1021541467926e-05), (7, 1.3566894267569296e-05), (8, 2.365095861023292e-05), (9, 2.6022842575912364e-05), (10, 4.9808753828983754e-05), (11, 4.535213520284742e-05), (12, 5.9398422308731824e-05), (13, 0.00010724861931521446), (14, 0.00013170980673749), (15, 0.00018590818217489868), (16, 0.0015464334283024073), (17, 0.00019057971076108515), (18, 0.00023856988991610706), (19, 0.00021843139256816357), (20, 0.0003257779171690345), (21, 0.00047126380377449095), (22, 0.00018556830764282495), (23, 0.0003488265792839229), (24, 0.0008625293266959488), (25, 0.0015402447897940874), (26, 0.0012802660930901766), (27, 0.00426446320489049)], Temporal MSE: [(0, 1.7255060811294243e-05), (1, 5.935994977335213e-06), (2, 5.300123120832723e-06), (3, 5.029718977311859e-06), (4, 2.8068961910321377e-06), (5, 5.163087280379841e-06), (6, 4.966979759046808e-06), (7, 6.958927770028822e-06), (8, 5.553261416935129e-06), (9, 2.0080879039596766e-05), (10, 2.056314224319067e-05), (11, 4.0262988477479666e-05), (12, 3.8170746847754344e-05), (13, 8.278974564746022e-05), (14, 0.00011471211473690346), (15, 9.463326568948105e-05), (16, 0.00020204525208100677), (17, 0.00010729135829024017), (18, 0.0001929926365846768), (19, 0.00010153881157748401), (20, 0.0002783022355288267), (21, 0.0002876840007957071), (22, 0.0001044078599079512), (23, 0.00018376813386566937), (24, 0.00017558102263137698), (25, 0.0006487779901362956), (26, 0.0001761996536515653), (27, 7.22290642443113e-05)]
+
+ 24%|██▍       | 12/50 [02:07<07:10, 11.33s/it]Time step 740, Spatial MSE: [(0, 2.165085606975481e-05), (1, 1.1992719919362571e-05), (2, 0.00016555216279812157), (3, 0.00014427007408812642), (4, 0.0005933455540798604), (5, 0.00034540845081210136), (6, 0.0007807048386894166), (7, 0.0010411138646304607), (8, 0.0013959745410829782), (9, 0.0021908795461058617), (10, 0.004284569062292576), (11, 0.0036821647081524134), (12, 0.003269463311880827), (13, 0.003784656524658203), (14, 0.00939200259745121), (15, 0.01610809564590454), (16, 0.025375457480549812), (17, 0.020467085763812065), (18, 0.017672870308160782), (19, 0.00709456717595458), (20, 0.015429665334522724), (21, 0.02390936389565468), (22, 0.005362359806895256), (23, 0.006271960213780403), (24, 0.015988970175385475), (25, 0.06189633160829544), (26, 0.04648321866989136), (27, 0.007169658318161964)], Temporal MSE: [(0, 4.088509376742877e-05), (1, 7.654035471205134e-06), (2, 5.94211342104245e-05), (3, 0.00010492451838217676), (4, 0.00012090821837773547), (5, 0.0002056984812952578), (6, 0.00037224384141154587), (7, 0.0003613877634052187), (8, 0.00037957343738526106), (9, 0.0007392552797682583), (10, 0.0009139669709838927), (11, 0.0011006651911884546), (12, 0.0012116980506107211), (13, 0.0020185192115604877), (14, 0.0023829976562410593), (15, 0.002935574157163501), (16, 0.0043198708444833755), (17, 0.005881631281226873), (18, 0.00515570817515254), (19, 0.002877825638279319), (20, 0.007541140541434288), (21, 0.004616509657353163), (22, 0.003361265640705824), (23, 0.002915439195930958), (24, 0.007749711629003286), (25, 0.011313463561236858), (26, 0.006393706426024437), (27, 0.0003476962447166443)]
+
+ 26%|██▌       | 13/50 [02:16<06:33, 10.64s/it]Time step 720, Spatial MSE: [(0, 8.00540601630928e-06), (1, 4.451603672350757e-06), (2, 8.978586265584454e-05), (3, 1.4593556443287525e-05), (4, 1.3405476238403935e-05), (5, 8.00811994849937e-06), (6, 1.6238227544818074e-05), (7, 1.9281169443274848e-05), (8, 2.1492600353667513e-05), (9, 3.3242071367567405e-05), (10, 4.379994061309844e-05), (11, 5.7632936659501866e-05), (12, 7.003899372648448e-05), (13, 9.974838758353144e-05), (14, 0.00012192665599286556), (15, 0.00018943949544336647), (16, 0.00018994699348695576), (17, 0.0001576717768330127), (18, 0.00034247428993694484), (19, 0.00016175622295122594), (20, 0.0003392430080566555), (21, 0.00048492089263163507), (22, 0.00020515939104370773), (23, 0.0004331791715230793), (24, 0.0009915400296449661), (25, 0.0018479214049875736), (26, 0.0015837830724194646), (27, 0.005098129156976938)], Temporal MSE: [(0, 1.4002588613948319e-05), (1, 4.893092864222126e-06), (2, 6.3701286308059935e-06), (3, 5.2507743930618744e-06), (4, 3.1207086976792198e-06), (5, 6.747560746589443e-06), (6, 8.094598342722747e-06), (7, 1.187198176921811e-05), (8, 6.94324990035966e-06), (9, 1.8281083612237126e-05), (10, 2.5839823138085194e-05), (11, 4.6507149818353355e-05), (12, 4.7288354835473e-05), (13, 7.716091931797564e-05), (14, 9.151730046141893e-05), (15, 0.00011266661749687046), (16, 0.0001503724924987182), (17, 9.411665087100118e-05), (18, 0.00023294432321563363), (19, 0.00012220094504300505), (20, 0.00025170700973831117), (21, 0.0003248990105930716), (22, 9.559587488183752e-05), (23, 0.0002151190856238827), (24, 0.00022639910457655787), (25, 0.0008041492546908557), (26, 0.00024931796360760927), (27, 7.641177217010409e-05)]
+
+ 28%|██▊       | 14/50 [02:29<06:44, 11.23s/it]Time step 700, Spatial MSE: [(0, 1.951871490746271e-05), (1, 9.089004379347898e-06), (2, 0.00014560187992174178), (3, 4.7726716729812324e-05), (4, 8.015330968191847e-05), (5, 0.00012032323138555512), (6, 0.00041736720595508814), (7, 0.000499185873195529), (8, 0.0008838434587232769), (9, 0.0007871921407058835), (10, 0.0025985524989664555), (11, 0.0028744989540427923), (12, 0.0022912390995770693), (13, 0.002808504505082965), (14, 0.004398689139634371), (15, 0.0027336771599948406), (16, 0.002889357041567564), (17, 0.0029246548656374216), (18, 0.0017410988220945), (19, 0.0020002825185656548), (20, 0.0022103239316493273), (21, 0.0012925256742164493), (22, 0.0015202985377982259), (23, 0.0029560697730630636), (24, 0.006708077155053616), (25, 0.02160653844475746), (26, 0.008041595108807087), (27, 0.009660352021455765)], Temporal MSE: [(0, 1.2518712537712418e-05), (1, 4.64445383840939e-06), (2, 2.928355934272986e-05), (3, 3.69641202269122e-05), (4, 3.536072836141102e-05), (5, 0.00012296735076233745), (6, 0.00024245718668680638), (7, 0.0004992720787413418), (8, 0.00020969886099919677), (9, 0.0005614078836515546), (10, 0.0012120429892092943), (11, 0.0010566363343968987), (12, 0.0008434436749666929), (13, 0.0016509320121258497), (14, 0.0014643871691077948), (15, 0.0012691974407061934), (16, 0.0018687593983486295), (17, 0.0017209514044225216), (18, 0.0017180847935378551), (19, 0.000920926860999316), (20, 0.002217184519395232), (21, 0.0019289524061605334), (22, 0.000893749063834548), (23, 0.0018275127513334155), (24, 0.004415176343172789), (25, 0.0059142084792256355), (26, 0.0022660112008452415), (27, 0.00015542034816462547)]
+
+ 30%|███       | 15/50 [02:42<06:46, 11.63s/it]Time step 680, Spatial MSE: [(0, 6.890104486956261e-06), (1, 1.7238442524103448e-05), (2, 0.0001543024118291214), (3, 5.246241926215589e-05), (4, 9.382858843309805e-05), (5, 0.00010335974366171286), (6, 0.00043971941340714693), (7, 0.000666137202642858), (8, 0.00026357470778748393), (9, 0.00047501560766249895), (10, 0.0005648690857924521), (11, 0.0007232700590975583), (12, 0.0007110861479304731), (13, 0.0011865270789712667), (14, 0.00293139461427927), (15, 0.004710543435066938), (16, 0.045555613934993744), (17, 0.0021168473176658154), (18, 0.0035324692726135254), (19, 0.002056586556136608), (20, 0.004717997740954161), (21, 0.00693525280803442), (22, 0.003203136147931218), (23, 0.0021509891375899315), (24, 0.0052512027323246), (25, 0.0219942107796669), (26, 0.005431113764643669), (27, 0.007470720447599888)], Temporal MSE: [(0, 1.9952678485424258e-05), (1, 7.325523256440647e-06), (2, 3.2169511541724205e-05), (3, 6.994485011091456e-05), (4, 5.935264562140219e-05), (5, 0.0001437849277863279), (6, 0.00022968306438997388), (7, 0.00012143870844738558), (8, 9.476134437136352e-05), (9, 0.00024403388670179993), (10, 0.0005174866528250277), (11, 0.0005639022565446794), (12, 0.00048782024532556534), (13, 0.001478790771216154), (14, 0.0032867328263819218), (15, 0.001364058582112193), (16, 0.0015662943478673697), (17, 0.0007720810826867819), (18, 0.0006999934557825327), (19, 0.0006359231192618608), (20, 0.0032467367127537727), (21, 0.0012306238058954477), (22, 0.0014491209294646978), (23, 0.0017844297690317035), (24, 0.004863594193011522), (25, 0.00594730768352747), (26, 0.003817102639004588), (27, 0.00017308833776041865)]
+
+ 32%|███▏      | 16/50 [02:52<06:25, 11.33s/it]Time step 660, Spatial MSE: [(0, 1.5806042938493192e-05), (1, 1.1320413250359707e-05), (2, 0.00014483147242572159), (3, 9.509681694908068e-05), (4, 0.0001598681410541758), (5, 0.00016506266547366977), (6, 0.0005153053789399564), (7, 0.0005183890461921692), (8, 0.000626585737336427), (9, 0.0009674386237747967), (10, 0.0018688080599531531), (11, 0.0020149564370512962), (12, 0.0021351734176278114), (13, 0.003512726863846183), (14, 0.006893780082464218), (15, 0.004621526226401329), (16, 0.04952819645404816), (17, 0.002610947238281369), (18, 0.0028613910544663668), (19, 0.0037893769331276417), (20, 0.0031869548838585615), (21, 0.003129596821963787), (22, 0.0025879996828734875), (23, 0.0033190669491887093), (24, 0.010233436711132526), (25, 0.024207836017012596), (26, 0.009803305380046368), (27, 0.006991133093833923)], Temporal MSE: [(0, 9.276234777644277e-06), (1, 3.675870857478003e-06), (2, 1.9869414245476946e-05), (3, 5.659562521032058e-05), (4, 7.848532550269738e-05), (5, 0.00012880914437118918), (6, 0.0001485583052271977), (7, 0.00031843793112784624), (8, 0.0002462250704411417), (9, 0.000454467604868114), (10, 0.00075443962123245), (11, 0.001092533697374165), (12, 0.0009314287453889847), (13, 0.0022370198275893927), (14, 0.003826072672381997), (15, 0.002306225709617138), (16, 0.003134465543553233), (17, 0.0018997115548700094), (18, 0.001942989300005138), (19, 0.0016379846492782235), (20, 0.003708953969180584), (21, 0.0027013192884624004), (22, 0.0017352582653984427), (23, 0.0025721851270645857), (24, 0.007723354734480381), (25, 0.020965415984392166), (26, 0.01115564163774252), (27, 0.00031185997067950666)]
+
+ 34%|███▍      | 17/50 [03:01<05:49, 10.58s/it]Time step 640, Spatial MSE: [(0, 7.228451067931019e-06), (1, 9.846630746324081e-06), (2, 0.00010492849833099172), (3, 2.3917249563965015e-05), (4, 1.3150714039511513e-05), (5, 8.515853551216424e-06), (6, 4.081609222339466e-05), (7, 4.228951729601249e-05), (8, 1.6077132386271842e-05), (9, 3.095137435593642e-05), (10, 4.571844692691229e-05), (11, 6.188842962728813e-05), (12, 6.446496990974993e-05), (13, 5.921333286096342e-05), (14, 0.00010863748320844024), (15, 0.00013426148507278413), (16, 0.00010111444134963676), (17, 0.0001384212519042194), (18, 0.0002773267915472388), (19, 0.00011235134297749028), (20, 0.00021795736392959952), (21, 0.0005194912664592266), (22, 0.0001524301478639245), (23, 0.0004093929601367563), (24, 0.0008581025176681578), (25, 0.0020404625684022903), (26, 0.00268096593208611), (27, 0.006238391622900963)], Temporal MSE: [(0, 8.770683052716777e-06), (1, 3.0336434519995237e-06), (2, 6.141634912637528e-06), (3, 4.619218088919297e-06), (4, 3.702844423969509e-06), (5, 7.881143574195448e-06), (6, 1.920298018376343e-05), (7, 2.2907956008566543e-05), (8, 8.167740816134028e-06), (9, 1.657139728195034e-05), (10, 3.355219087097794e-05), (11, 4.964548861607909e-05), (12, 3.3175791031681e-05), (13, 4.950681613991037e-05), (14, 6.695932825095952e-05), (15, 9.455953841097653e-05), (16, 0.00010176547220908105), (17, 7.611606270074844e-05), (18, 0.0001892131840577349), (19, 9.631904686102644e-05), (20, 0.00014524201105814427), (21, 0.00029492468456737697), (22, 5.969778430880979e-05), (23, 0.00019346251792740077), (24, 0.00026087125297635794), (25, 0.000787468918133527), (26, 0.0003560750337783247), (27, 5.93937402300071e-05)]
+
+ 36%|███▌      | 18/50 [03:10<05:19,  9.97s/it]Time step 620, Spatial MSE: [(0, 1.7508788005216047e-05), (1, 2.4836379452608526e-05), (2, 0.00018784300482366234), (3, 0.00010060605563921854), (4, 0.00034011827665381134), (5, 0.0006102820043452084), (6, 0.0031603737734258175), (7, 0.007996580563485622), (8, 0.008969442918896675), (9, 0.014402966015040874), (10, 0.01802303083240986), (11, 0.022592732682824135), (12, 0.02512197010219097), (13, 0.02425531856715679), (14, 0.04551311954855919), (15, 0.05346861854195595), (16, 0.07092756778001785), (17, 0.059240348637104034), (18, 0.052509207278490067), (19, 0.022695176303386688), (20, 0.042948395013809204), (21, 0.10096360743045807), (22, 0.017204245552420616), (23, 0.023135095834732056), (24, 0.0519552119076252), (25, 0.10368165373802185), (26, 0.04180409014225006), (27, 0.01015276089310646)], Temporal MSE: [(0, 1.760985287546646e-05), (1, 8.18561420601327e-06), (2, 2.9046314011793584e-05), (3, 0.0003528360975906253), (4, 0.0005188103532418609), (5, 0.0011712606064975262), (6, 0.003607883583754301), (7, 0.0020200153812766075), (8, 0.0026505040004849434), (9, 0.004259887151420116), (10, 0.005920188967138529), (11, 0.007089210674166679), (12, 0.009235708974301815), (13, 0.013420886360108852), (14, 0.015711836516857147), (15, 0.015791533514857292), (16, 0.02113886922597885), (17, 0.022729240357875824), (18, 0.02058381587266922), (19, 0.01815662533044815), (20, 0.03887101635336876), (21, 0.02452828362584114), (22, 0.011469000950455666), (23, 0.014781107194721699), (24, 0.041548553854227066), (25, 0.05754385516047478), (26, 0.022495010867714882), (27, 0.0009565380169078708)]
+
+ 38%|███▊      | 19/50 [03:19<04:59,  9.65s/it]Time step 600, Spatial MSE: [(0, 7.0595278884866275e-06), (1, 1.298796632909216e-05), (2, 0.00010701220890041441), (3, 2.9563318094005808e-05), (4, 1.1584666026465129e-05), (5, 9.417571163794491e-06), (6, 3.4615521144587547e-05), (7, 4.753509711008519e-05), (8, 1.6590451195952483e-05), (9, 3.0399432944250293e-05), (10, 4.292710582376458e-05), (11, 5.759113992098719e-05), (12, 6.170379492687061e-05), (13, 5.72784774703905e-05), (14, 8.05363233666867e-05), (15, 0.00012037743726978078), (16, 9.069660882232711e-05), (17, 0.00010156013013329357), (18, 0.00019431949476711452), (19, 0.00010378772276453674), (20, 0.00017648593347985297), (21, 0.00038785641663707793), (22, 0.00015105113561730832), (23, 0.00041942772804759443), (24, 0.0009136100416071713), (25, 0.001998692285269499), (26, 0.0033016614615917206), (27, 0.006981263402849436)], Temporal MSE: [(0, 7.392186944343848e-06), (1, 2.6740481189335696e-06), (2, 6.1523951444542035e-06), (3, 3.9123528949858155e-06), (4, 4.283099769963883e-06), (5, 7.789989467710257e-06), (6, 2.0582418073900044e-05), (7, 2.136288458132185e-05), (8, 8.456050636596046e-06), (9, 1.5302612155210227e-05), (10, 3.284729609731585e-05), (11, 4.319268191466108e-05), (12, 3.0402297852560878e-05), (13, 4.9036385462386534e-05), (14, 5.348115519154817e-05), (15, 7.73515785112977e-05), (16, 7.93833751231432e-05), (17, 6.01205465500243e-05), (18, 0.00011688868107739836), (19, 7.667729369131848e-05), (20, 9.758993110153824e-05), (21, 0.00022897876624483615), (22, 5.588979911408387e-05), (23, 0.00016720339772291481), (24, 0.00026670063380151987), (25, 0.0007708290941081941), (26, 0.0004587623116094619), (27, 5.705477451556362e-05)]
+
+ 40%|████      | 20/50 [03:28<04:43,  9.46s/it]Time step 580, Spatial MSE: [(0, 1.6448395399493165e-05), (1, 1.9078381228609942e-05), (2, 0.00017705898790154606), (3, 4.1260969737777486e-05), (4, 6.137415766716003e-05), (5, 7.220056431833655e-05), (6, 0.0005326604004949331), (7, 0.00029114671633578837), (8, 0.0005766791291534901), (9, 0.0005478333914652467), (10, 0.001085355062969029), (11, 0.001520073739811778), (12, 0.0015580463223159313), (13, 0.0018311494495719671), (14, 0.0025018362794071436), (15, 0.0014697214355692267), (16, 0.0009544583735987544), (17, 0.0008962111896835268), (18, 0.0009966084035113454), (19, 0.0006825520540587604), (20, 0.0009533681441098452), (21, 0.001616950728930533), (22, 0.0010238825343549252), (23, 0.001613399013876915), (24, 0.004667447414249182), (25, 0.008961468935012817), (26, 0.007402748800814152), (27, 0.009759706445038319)], Temporal MSE: [(0, 6.955315711820731e-06), (1, 3.559570359357167e-06), (2, 1.8333013940718956e-05), (3, 2.9483684556907974e-05), (4, 2.614843106130138e-05), (5, 8.920832624426112e-05), (6, 9.654242603573948e-05), (7, 0.00018430923228152096), (8, 0.00011676814028760418), (9, 0.0001769111113389954), (10, 0.0003140141488984227), (11, 0.00041666976176202297), (12, 0.00046125100925564766), (13, 0.0007885044906288385), (14, 0.0007488687988370657), (15, 0.0005458754021674395), (16, 0.0005155071849003434), (17, 0.0004540666413959116), (18, 0.0004354119300842285), (19, 0.0003871338558383286), (20, 0.0008100595441646874), (21, 0.001094744191505015), (22, 0.0006495405687019229), (23, 0.0009941267780959606), (24, 0.0035239888820797205), (25, 0.006600040476769209), (26, 0.0023669854272156954), (27, 0.00018780407845042646)]
+
+ 42%|████▏     | 21/50 [03:40<04:55, 10.20s/it]Time step 560, Spatial MSE: [(0, 6.446724455599906e-06), (1, 2.355810283916071e-05), (2, 0.00010291656508343294), (3, 4.621998596121557e-05), (4, 4.848667231271975e-05), (5, 9.27518995013088e-05), (6, 0.0002522596041671932), (7, 0.0005324641824699938), (8, 0.0002573530946392566), (9, 0.0003121504560112953), (10, 0.00042133699753321707), (11, 0.0005204805638641119), (12, 0.0005476208752952516), (13, 0.0007933591841720045), (14, 0.0013108095154166222), (15, 0.0026661762967705727), (16, 0.008619285188615322), (17, 0.0010287879267707467), (18, 0.0014930741162970662), (19, 0.0009209635900333524), (20, 0.002540706656873226), (21, 0.003423617919906974), (22, 0.0013079616473987699), (23, 0.0013522037770599127), (24, 0.003441835753619671), (25, 0.01010840479284525), (26, 0.002975837793201208), (27, 0.007880651392042637)], Temporal MSE: [(0, 8.34325601317687e-06), (1, 4.67479685539729e-06), (2, 2.2627898943028413e-05), (3, 7.192458724603057e-05), (4, 8.210420492105186e-05), (5, 8.208638610085472e-05), (6, 0.00013977137859910727), (7, 0.00013364233018364757), (8, 0.00011065726721426472), (9, 0.00027640003827400506), (10, 0.00048593105748295784), (11, 0.0005869492888450623), (12, 0.000708453357219696), (13, 0.0012216427130624652), (14, 0.0013469057157635689), (15, 0.0008375548059120774), (16, 0.0007500629290007055), (17, 0.0006116160657256842), (18, 0.0006079229642637074), (19, 0.0005187076749280095), (20, 0.0012992715928703547), (21, 0.0014877294888719916), (22, 0.0010513161541894078), (23, 0.0010950651485472918), (24, 0.00330696115270257), (25, 0.005953972693532705), (26, 0.0037387614138424397), (27, 0.0002443813136778772)]
+
+ 44%|████▍     | 22/50 [03:48<04:30,  9.66s/it]Time step 540, Spatial MSE: [(0, 1.5262006854754873e-05), (1, 2.0898845832562074e-05), (2, 0.0001789717935025692), (3, 6.997687160037458e-05), (4, 9.312012844020501e-05), (5, 0.00012687603884842247), (6, 0.0005397425265982747), (7, 0.0004725830804090947), (8, 0.0005251202965155244), (9, 0.0006937154103070498), (10, 0.0008033875492401421), (11, 0.0011724871583282948), (12, 0.0013344326289370656), (13, 0.0019866125658154488), (14, 0.003409571247175336), (15, 0.0037338787224143744), (16, 0.010247848927974701), (17, 0.002055291784927249), (18, 0.002329715294763446), (19, 0.0019125877879559994), (20, 0.002423179568722844), (21, 0.003548626322299242), (22, 0.002109588123857975), (23, 0.0029868995770812035), (24, 0.009918996132910252), (25, 0.0172029510140419), (26, 0.015785811468958855), (27, 0.007982106879353523)], Temporal MSE: [(0, 6.058543476683553e-06), (1, 3.390772917555296e-06), (2, 1.305467048950959e-05), (3, 3.796705641434528e-05), (4, 7.564657425973564e-05), (5, 0.00012823267024941742), (6, 0.00015599436301272362), (7, 0.00018664487288333476), (8, 0.0001775772252585739), (9, 0.00030384439742192626), (10, 0.00041100505040958524), (11, 0.0006337680388242006), (12, 0.0007337549468502402), (13, 0.0013845397625118494), (14, 0.0016313644591718912), (15, 0.0015804498689249158), (16, 0.0013301043072715402), (17, 0.0009866354521363974), (18, 0.0010133133037015796), (19, 0.0010371902026236057), (20, 0.0019225215073674917), (21, 0.0018485526088625193), (22, 0.0016025126678869128), (23, 0.002037712838500738), (24, 0.007114110514521599), (25, 0.015757925808429718), (26, 0.010052128694951534), (27, 0.0003479444421827793)]
+
+ 46%|████▌     | 23/50 [03:56<04:10,  9.29s/it]Time step 520, Spatial MSE: [(0, 6.289775228651706e-06), (1, 1.606485602678731e-05), (2, 8.175589755410329e-05), (3, 2.9124643333489075e-05), (4, 1.089795205189148e-05), (5, 9.19395824894309e-06), (6, 2.4801669496810064e-05), (7, 4.964389881934039e-05), (8, 1.4529253348882776e-05), (9, 2.354837670281995e-05), (10, 3.0165758289513178e-05), (11, 4.564231130643748e-05), (12, 4.084789543412626e-05), (13, 4.569225347950123e-05), (14, 4.66560923086945e-05), (15, 7.323027239181101e-05), (16, 7.217479287646711e-05), (17, 6.543126801261678e-05), (18, 0.00012740508827846497), (19, 7.603305857628584e-05), (20, 0.0001217260432895273), (21, 0.0001864511432358995), (22, 0.0001105280316551216), (23, 0.00028371578082442284), (24, 0.0005750920390710235), (25, 0.0014489461900666356), (26, 0.0032184135634452105), (27, 0.007048455532640219)], Temporal MSE: [(0, 5.4882493714103475e-06), (1, 2.1670568912668386e-06), (2, 4.715948307421058e-06), (3, 3.785012495427509e-06), (4, 3.965726136812009e-06), (5, 7.063494194881059e-06), (6, 1.7633821698836982e-05), (7, 1.8792001355905086e-05), (8, 7.252522664202843e-06), (9, 1.1519289728312287e-05), (10, 2.674456663953606e-05), (11, 3.2275998819386587e-05), (12, 2.1404248400358483e-05), (13, 3.516851211315952e-05), (14, 2.994378337461967e-05), (15, 4.3238735088380054e-05), (16, 5.262876584311016e-05), (17, 3.857924821204506e-05), (18, 4.9229573050979525e-05), (19, 4.896645623375662e-05), (20, 5.766272806795314e-05), (21, 0.00011960601841565222), (22, 4.582148903864436e-05), (23, 0.0001031543652061373), (24, 0.00020633624808397144), (25, 0.0005270791007205844), (26, 0.0004858973843511194), (27, 5.672542829415761e-05)]
+
+ 48%|████▊     | 24/50 [04:05<03:57,  9.13s/it]Time step 500, Spatial MSE: [(0, 1.4725983419339173e-05), (1, 2.9421495128190145e-05), (2, 0.00015306807472370565), (3, 6.10293609497603e-05), (4, 0.00017589327762834728), (5, 0.00046773639041930437), (6, 0.0013722521252930164), (7, 0.004154412541538477), (8, 0.004348911810666323), (9, 0.005646040663123131), (10, 0.006211364641785622), (11, 0.00801126379519701), (12, 0.008432164788246155), (13, 0.01022375002503395), (14, 0.01515878550708294), (15, 0.01969337835907936), (16, 0.016899263486266136), (17, 0.016547031700611115), (18, 0.020963182672858238), (19, 0.018456541001796722), (20, 0.030895987525582314), (21, 0.027888119220733643), (22, 0.014350341632962227), (23, 0.020747067406773567), (24, 0.042813632637262344), (25, 0.06532616168260574), (26, 0.03857019543647766), (27, 0.00797603651881218)], Temporal MSE: [(0, 6.703946382913273e-06), (1, 5.079537004348822e-06), (2, 2.426594619464595e-05), (3, 0.00011065035505453125), (4, 0.00027751398738473654), (5, 0.000522468879353255), (6, 0.0013476116582751274), (7, 0.0014430166920647025), (8, 0.001437594648450613), (9, 0.0024271979928016663), (10, 0.0038035204634070396), (11, 0.004457821138203144), (12, 0.004860237706452608), (13, 0.009092319756746292), (14, 0.009201337583363056), (15, 0.009506438858807087), (16, 0.01218053512275219), (17, 0.012679751962423325), (18, 0.01337346900254488), (19, 0.01224235538393259), (20, 0.022482113912701607), (21, 0.016935722902417183), (22, 0.010393943637609482), (23, 0.013158855959773064), (24, 0.03406154364347458), (25, 0.050558142364025116), (26, 0.0200189296156168), (27, 0.0009573898860253394)]
+
+ 50%|█████     | 25/50 [04:17<04:11, 10.07s/it]Time step 480, Spatial MSE: [(0, 6.2055660237092525e-06), (1, 1.7681586541584693e-05), (2, 7.013027061475441e-05), (3, 3.186562753398903e-05), (4, 1.1335226190567482e-05), (5, 1.0692046089388896e-05), (6, 1.88083213288337e-05), (7, 4.67649515485391e-05), (8, 1.4057580301596317e-05), (9, 2.0278455849620514e-05), (10, 2.5373248718096875e-05), (11, 3.7692137993872166e-05), (12, 3.494076008792035e-05), (13, 4.489593629841693e-05), (14, 4.287832052796148e-05), (15, 6.95779817760922e-05), (16, 6.635588215431198e-05), (17, 4.3118998291902244e-05), (18, 0.00014207295316737145), (19, 6.950335227884352e-05), (20, 8.334602898685262e-05), (21, 0.00018264113168697804), (22, 0.0001162561311502941), (23, 0.00022432004334405065), (24, 0.0005625627236440778), (25, 0.0011668021325021982), (26, 0.0033822553232312202), (27, 0.007491706870496273)], Temporal MSE: [(0, 4.7617891141271684e-06), (1, 2.0735890302603366e-06), (2, 4.484639703150606e-06), (3, 4.467991402634652e-06), (4, 4.03380045099766e-06), (5, 7.096493845892837e-06), (6, 1.571170287206769e-05), (7, 1.606710247870069e-05), (8, 6.77817479299847e-06), (9, 1.0894634215219412e-05), (10, 2.2891648768563755e-05), (11, 2.6743431590148248e-05), (12, 1.853841786214616e-05), (13, 3.328656021039933e-05), (14, 2.757760557869915e-05), (15, 3.900717638316564e-05), (16, 3.67258871847298e-05), (17, 2.4754843252594583e-05), (18, 3.835922689177096e-05), (19, 3.6988596548326313e-05), (20, 4.6621345973107964e-05), (21, 0.00010566214041318744), (22, 5.480026811710559e-05), (23, 9.422539733350277e-05), (24, 0.00021306012058630586), (25, 0.0004692866059485823), (26, 0.0005516842938959599), (27, 7.577441283501685e-05)]
+
+ 52%|█████▏    | 26/50 [04:26<03:48,  9.50s/it]Time step 460, Spatial MSE: [(0, 1.67392790899612e-05), (1, 2.272276469739154e-05), (2, 0.00011362729128450155), (3, 3.958214074373245e-05), (4, 6.0486490838229656e-05), (5, 6.579971523024142e-05), (6, 0.00035497205681167543), (7, 0.00022011146938893944), (8, 0.0004413437272887677), (9, 0.0004251150239724666), (10, 0.0007845261716283858), (11, 0.0011199816362932324), (12, 0.0011383940000087023), (13, 0.0014127461472526193), (14, 0.0023803571239113808), (15, 0.0011573017109185457), (16, 0.000789393438026309), (17, 0.0007557247299700975), (18, 0.001030072569847107), (19, 0.0005776744801551104), (20, 0.000804346869699657), (21, 0.0014214754337444901), (22, 0.0008688736706972122), (23, 0.0012828466715291142), (24, 0.004271956626325846), (25, 0.006984023377299309), (26, 0.013565455563366413), (27, 0.00766329700127244)], Temporal MSE: [(0, 4.946805347572081e-06), (1, 2.9964596706122393e-06), (2, 2.0089215468033217e-05), (3, 4.115267074666917e-05), (4, 3.307455699541606e-05), (5, 6.88408108544536e-05), (6, 9.306011634180322e-05), (7, 0.00010226594167761505), (8, 9.328111627837643e-05), (9, 0.0001401585031999275), (10, 0.00023652423988096416), (11, 0.00035705772461369634), (12, 0.00039329694118350744), (13, 0.0007101459777913988), (14, 0.0008121814462356269), (15, 0.0004915918689221144), (16, 0.00042294818558730185), (17, 0.00035648600896820426), (18, 0.0003736102662514895), (19, 0.0003475432167761028), (20, 0.0007045237580314279), (21, 0.0010428264504298568), (22, 0.000590818643104285), (23, 0.0008017704240046442), (24, 0.0028865118511021137), (25, 0.007769552990794182), (26, 0.00266580143943429), (27, 0.00030203472124412656)]
+
+ 54%|█████▍    | 27/50 [04:36<03:48,  9.93s/it]Time step 440, Spatial MSE: [(0, 5.5984114624152426e-06), (1, 2.335081808269024e-05), (2, 5.9101847000420094e-05), (3, 3.621190262492746e-05), (4, 5.8550162066239864e-05), (5, 0.00013316154945641756), (6, 0.000143945129821077), (7, 0.000507815508171916), (8, 0.000262937304796651), (9, 0.0002984791644848883), (10, 0.0003731864271685481), (11, 0.00048014961066655815), (12, 0.0005185476038604975), (13, 0.0008598326239734888), (14, 0.001872709603048861), (15, 0.002772475592792034), (16, 0.02011074870824814), (17, 0.0008992475341074169), (18, 0.0012015540851280093), (19, 0.0011856630444526672), (20, 0.002033442724496126), (21, 0.0029728205408900976), (22, 0.0012455732794478536), (23, 0.0011280346661806107), (24, 0.002651202492415905), (25, 0.005388747900724411), (26, 0.0027823608834296465), (27, 0.007566806394606829)], Temporal MSE: [(0, 4.687619366450235e-06), (1, 3.2378325158788357e-06), (2, 2.235580723208841e-05), (3, 0.00010929384734481573), (4, 0.0001270623761229217), (5, 8.485973376082256e-05), (6, 0.0001243371079908684), (7, 0.00012998214515391737), (8, 0.00010880957415793091), (9, 0.0002780134091153741), (10, 0.0004576973442453891), (11, 0.0005955155938863754), (12, 0.0007771622040309012), (13, 0.001607492915354669), (14, 0.00226782844401896), (15, 0.0007549019064754248), (16, 0.0007014824659563601), (17, 0.0005768274422734976), (18, 0.0005312473513185978), (19, 0.0005173461977392435), (20, 0.0011350729037076235), (21, 0.0015264968387782574), (22, 0.0009186118841171265), (23, 0.0011570442002266645), (24, 0.0031682681292295456), (25, 0.005729145370423794), (26, 0.0038525285199284554), (27, 0.00044633200741373)]
+
+ 56%|█████▌    | 28/50 [04:49<03:53, 10.62s/it]Time step 420, Spatial MSE: [(0, 1.5722500393167138e-05), (1, 2.3939595848787576e-05), (2, 9.89159889286384e-05), (3, 5.468788367579691e-05), (4, 9.195862367050722e-05), (5, 0.0001602466800250113), (6, 0.0003421829896979034), (7, 0.0005582387093454599), (8, 0.0005547840846702456), (9, 0.0006593255675397813), (10, 0.0007524797110818326), (11, 0.001169504364952445), (12, 0.0013366719940677285), (13, 0.0020280650351196527), (14, 0.004467525519430637), (15, 0.00395794864743948), (16, 0.021976780146360397), (17, 0.002091356785967946), (18, 0.0021939482539892197), (19, 0.0023273546248674393), (20, 0.0023617849219590425), (21, 0.003510245820507407), (22, 0.0025177178904414177), (23, 0.003207039088010788), (24, 0.009509537369012833), (25, 0.018641704693436623), (26, 0.030320890247821808), (27, 0.010409809648990631)], Temporal MSE: [(0, 4.434998572833138e-06), (1, 2.853405476344051e-06), (2, 1.6507598047610372e-05), (3, 3.51633207174018e-05), (4, 0.00010151580499950796), (5, 0.0001513557945145294), (6, 0.00020227907225489616), (7, 0.00022341318253893405), (8, 0.0002237924636574462), (9, 0.0003530500689521432), (10, 0.0004932733718305826), (11, 0.0008017982472665608), (12, 0.0008569497149437666), (13, 0.0017951647751033306), (14, 0.002642017789185047), (15, 0.0017127764876931906), (16, 0.0016090351855382323), (17, 0.0011552926152944565), (18, 0.0010300599969923496), (19, 0.0010928406845778227), (20, 0.0026932121254503727), (21, 0.002238360233604908), (22, 0.0019639076199382544), (23, 0.002213126514106989), (24, 0.006688377354294062), (25, 0.01659429259598255), (26, 0.008925500325858593), (27, 0.0005447330186143517)]
+
+ 58%|█████▊    | 29/50 [05:01<03:55, 11.21s/it]Time step 400, Spatial MSE: [(0, 6.103153282310814e-06), (1, 1.883147888293024e-05), (2, 4.344132685218938e-05), (3, 3.102973641944118e-05), (4, 1.0957954145851545e-05), (5, 1.1122177966171876e-05), (6, 1.4626466509071179e-05), (7, 4.580033419188112e-05), (8, 1.482305015088059e-05), (9, 1.7430278603569604e-05), (10, 2.3361195417237468e-05), (11, 3.15460201818496e-05), (12, 3.1622741516912356e-05), (13, 4.968810753780417e-05), (14, 4.5095519453752786e-05), (15, 5.92906289966777e-05), (16, 0.0010879886103793979), (17, 4.650631672120653e-05), (18, 9.505095658823848e-05), (19, 0.00010210376785835251), (20, 6.932830001460388e-05), (21, 0.00017043987463694066), (22, 0.00011215746053494513), (23, 0.00020952547492925078), (24, 0.00041997258085757494), (25, 0.0010090876603499055), (26, 0.0031908834353089333), (27, 0.008446390740573406)], Temporal MSE: [(0, 4.050453753734473e-06), (1, 2.0042079995619133e-06), (2, 4.082988652953645e-06), (3, 4.852295660384698e-06), (4, 3.940276201319648e-06), (5, 7.5750804171548225e-06), (6, 1.4107496099313721e-05), (7, 1.4201650628820062e-05), (8, 6.817260327807162e-06), (9, 1.1038438060495537e-05), (10, 2.0112318452447653e-05), (11, 2.213299012510106e-05), (12, 1.844880716816988e-05), (13, 3.599398769438267e-05), (14, 3.836922405753285e-05), (15, 3.2987762097036466e-05), (16, 4.4723779865307733e-05), (17, 2.774447057163343e-05), (18, 3.400741843506694e-05), (19, 3.6065292079001665e-05), (20, 6.210267747519538e-05), (21, 8.499489922542125e-05), (22, 6.922082684468478e-05), (23, 0.000102112433523871), (24, 0.00019332583178766072), (25, 0.0004015990416519344), (26, 0.0005648123915307224), (27, 0.00011669380182866007)]
+
+ 60%|██████    | 30/50 [05:12<03:38, 10.92s/it]Time step 380, Spatial MSE: [(0, 1.4924862625775859e-05), (1, 2.6985879230778664e-05), (2, 7.099453068803996e-05), (3, 5.252983464743011e-05), (4, 0.0001479590282542631), (5, 0.0004254306259099394), (6, 0.0011161412112414837), (7, 0.002435753121972084), (8, 0.002586950547993183), (9, 0.0038520395755767822), (10, 0.004210314713418484), (11, 0.005778227001428604), (12, 0.006735850125551224), (13, 0.00868937186896801), (14, 0.012120675295591354), (15, 0.017392506822943687), (16, 0.014276470057666302), (17, 0.012844890356063843), (18, 0.014323287643492222), (19, 0.012245205231010914), (20, 0.015351022593677044), (21, 0.016376590356230736), (22, 0.012110946699976921), (23, 0.017424214631319046), (24, 0.0361851304769516), (25, 0.04864223301410675), (26, 0.04396235942840576), (27, 0.006112166214734316)], Temporal MSE: [(0, 4.385061401990242e-06), (1, 3.2358498174289707e-06), (2, 2.0314000721555203e-05), (3, 0.00016140271327458322), (4, 0.0002548274351283908), (5, 0.000428741448558867), (6, 0.0009297870565205812), (7, 0.001182046253234148), (8, 0.001332696177996695), (9, 0.002236117608845234), (10, 0.0043182275258004665), (11, 0.004815294407308102), (12, 0.0050613791681826115), (13, 0.008638312108814716), (14, 0.010167599655687809), (15, 0.009478574618697166), (16, 0.010057871229946613), (17, 0.009642592631280422), (18, 0.00929961260408163), (19, 0.008296404033899307), (20, 0.014900737442076206), (21, 0.011870344169437885), (22, 0.009668033570051193), (23, 0.012084795162081718), (24, 0.02683871053159237), (25, 0.04105038195848465), (26, 0.015935184434056282), (27, 0.0009747084695845842)]
+
+ 62%|██████▏   | 31/50 [05:22<03:24, 10.77s/it]Time step 360, Spatial MSE: [(0, 6.270607627811842e-06), (1, 1.9957995391450822e-05), (2, 3.480665691313334e-05), (3, 3.437474151724018e-05), (4, 1.1401003575883806e-05), (5, 1.3699106602871325e-05), (6, 1.3918256627221126e-05), (7, 4.44448487542104e-05), (8, 1.5343866834882647e-05), (9, 1.725407491903752e-05), (10, 2.2270243789535016e-05), (11, 2.9544222343247384e-05), (12, 3.083875708398409e-05), (13, 6.447490159189329e-05), (14, 0.00016592713654972613), (15, 7.136843487387523e-05), (16, 0.005719699896872044), (17, 4.5437318476615474e-05), (18, 0.00013031742128077894), (19, 0.00020740213221870363), (20, 9.942773613147438e-05), (21, 0.00024728511925786734), (22, 0.00017584113811608404), (23, 0.00025082367938011885), (24, 0.0005021158722229302), (25, 0.0010674261720851064), (26, 0.002978963078930974), (27, 0.009906096383929253)], Temporal MSE: [(0, 3.804871539614396e-06), (1, 2.0629013306461275e-06), (2, 4.483119482756592e-06), (3, 6.462369583459804e-06), (4, 4.701776560978033e-06), (5, 8.273473213193938e-06), (6, 1.3896324162487872e-05), (7, 1.3473315448209178e-05), (8, 7.553660907433368e-06), (9, 1.2253822205821052e-05), (10, 1.9865248759742826e-05), (11, 2.2041756892576814e-05), (12, 2.0266517822165042e-05), (13, 4.982820973964408e-05), (14, 0.0002508728939574212), (15, 3.864905011141673e-05), (16, 4.850414552493021e-05), (17, 2.9029955840087496e-05), (18, 4.738678762805648e-05), (19, 3.901706077158451e-05), (20, 9.605120430933312e-05), (21, 0.00011991704377578571), (22, 0.00011053924390580505), (23, 0.00019354316464159638), (24, 0.0003115203871857375), (25, 0.00048620824236422777), (26, 0.0006574408616870642), (27, 0.0001829608518164605)]
+
+ 64%|██████▍   | 32/50 [05:32<03:08, 10.45s/it]Time step 340, Spatial MSE: [(0, 1.6840062016854063e-05), (1, 2.2841479221824557e-05), (2, 4.05462778871879e-05), (3, 3.731328251888044e-05), (4, 4.6412067604251206e-05), (5, 7.14440320734866e-05), (6, 0.00021737314818892628), (7, 0.00020572579524014145), (8, 0.0003858925774693489), (9, 0.0003908739017788321), (10, 0.0006606859969906509), (11, 0.0010496324393898249), (12, 0.0010898608015850186), (13, 0.0013357980642467737), (14, 0.0030190811958163977), (15, 0.0013500116765499115), (16, 0.006503598764538765), (17, 0.0007824713247828186), (18, 0.0010375999845564365), (19, 0.0007401217590086162), (20, 0.0009401561692357063), (21, 0.0013137185014784336), (22, 0.000794706167653203), (23, 0.0011687190271914005), (24, 0.0038425815291702747), (25, 0.008692617528140545), (26, 0.020827991887927055), (27, 0.007702419999986887)], Temporal MSE: [(0, 4.467047801881563e-06), (1, 2.647619567142101e-06), (2, 2.115830830007326e-05), (3, 4.586935028783046e-05), (4, 4.1893839807016775e-05), (5, 7.008295506238937e-05), (6, 0.00010474364535184577), (7, 9.46144136833027e-05), (8, 9.309533197665587e-05), (9, 0.00014917648513801396), (10, 0.0002546952455304563), (11, 0.0003861025907099247), (12, 0.00040695664938539267), (13, 0.0008029014570638537), (14, 0.0012197194155305624), (15, 0.0005589391221292317), (16, 0.0004358577716629952), (17, 0.00035823677899315953), (18, 0.00042252193088643253), (19, 0.00038349084206856787), (20, 0.0008900632383301854), (21, 0.0009544475469738245), (22, 0.0006018754793331027), (23, 0.0008276510052382946), (24, 0.0023489343002438545), (25, 0.007424670737236738), (26, 0.00284567940980196), (27, 0.0005181383457966149)]
+
+ 66%|██████▌   | 33/50 [05:40<02:46,  9.79s/it]Time step 320, Spatial MSE: [(0, 6.212557309481781e-06), (1, 2.371253867750056e-05), (2, 3.0161463655531406e-05), (3, 3.23950189340394e-05), (4, 6.378864054568112e-05), (5, 0.000173451509908773), (6, 0.00016644276911392808), (7, 0.0006056238780729473), (8, 0.00031137882615439594), (9, 0.0003201798826921731), (10, 0.000447115337010473), (11, 0.0005087134195491672), (12, 0.0005625163321383297), (13, 0.0014575897948816419), (14, 0.0028701629489660263), (15, 0.003170310752466321), (16, 0.035263173282146454), (17, 0.0009893366368487477), (18, 0.0011915612267330289), (19, 0.0018011615611612797), (20, 0.0022253100760281086), (21, 0.003287610597908497), (22, 0.0012136008590459824), (23, 0.0011139637790620327), (24, 0.002446579048410058), (25, 0.00450547132641077), (26, 0.0031500558834522963), (27, 0.010982971638441086)], Temporal MSE: [(0, 4.112890565011185e-06), (1, 2.772117113636341e-06), (2, 2.2784255634178407e-05), (3, 0.00012110595707781613), (4, 0.0001510068104835227), (5, 0.00010811153333634138), (6, 0.00016100732318591326), (7, 0.00014778028707951307), (8, 0.0001278771087527275), (9, 0.00031503927311860025), (10, 0.0004753650864586234), (11, 0.0006561095942743123), (12, 0.0008482022094540298), (13, 0.002597535727545619), (14, 0.003072811057791114), (15, 0.0008391293813474476), (16, 0.0007306595798581839), (17, 0.0005697041051462293), (18, 0.0005666419165208936), (19, 0.0005014038179069757), (20, 0.0012609742116183043), (21, 0.0013909462140873075), (22, 0.000839270418509841), (23, 0.001290920190513134), (24, 0.0030689106788486242), (25, 0.004763549659401178), (26, 0.003892606822773814), (27, 0.0009963129414245486)]
+
+ 68%|██████▊   | 34/50 [05:48<02:29,  9.34s/it]Time step 300, Spatial MSE: [(0, 1.5121800970518962e-05), (1, 2.2591559172724374e-05), (2, 3.0376679205801338e-05), (3, 4.639910548576154e-05), (4, 9.500388841843233e-05), (5, 0.00017675067647360265), (6, 0.00026690823142416775), (7, 0.0005822252132929862), (8, 0.0005278308526612818), (9, 0.0006811707280576229), (10, 0.0008408621652051806), (11, 0.0011852773604914546), (12, 0.001487445435486734), (13, 0.002769091399386525), (14, 0.005842709913849831), (15, 0.004295044112950563), (16, 0.03726605325937271), (17, 0.0021527183707803488), (18, 0.0023960070684552193), (19, 0.003059475449845195), (20, 0.0024487008340656757), (21, 0.0035032208543270826), (22, 0.0028018902521580458), (23, 0.0035075380001217127), (24, 0.009507659822702408), (25, 0.02232694998383522), (26, 0.040491875261068344), (27, 0.016931839287281036)], Temporal MSE: [(0, 4.883386736764805e-06), (1, 2.690365590751753e-06), (2, 1.6495272575411946e-05), (3, 3.7511807022383437e-05), (4, 0.00011912655463675037), (5, 0.00017788608965929598), (6, 0.00021218025358393788), (7, 0.00025529079721309245), (8, 0.00026941081159748137), (9, 0.00041253899689763784), (10, 0.0005945958546362817), (11, 0.0010423039784654975), (12, 0.0010562966344878078), (13, 0.0026988876052200794), (14, 0.003380985464900732), (15, 0.002165387384593487), (16, 0.002224977593868971), (17, 0.0012784410500898957), (18, 0.0011672652326524258), (19, 0.0011682611657306552), (20, 0.0037227582652121782), (21, 0.0025705944281071424), (22, 0.0019547278061509132), (23, 0.002471414627507329), (24, 0.006151746027171612), (25, 0.014578049071133137), (26, 0.007517291232943535), (27, 0.0009774335194379091)]
+
+ 70%|███████   | 35/50 [05:58<02:20,  9.38s/it]Time step 280, Spatial MSE: [(0, 6.87279043631861e-06), (1, 2.1049527276773006e-05), (2, 3.0926417821319774e-05), (3, 3.4212956961710006e-05), (4, 1.1981493116763886e-05), (5, 1.818921373342164e-05), (6, 1.5075634109962266e-05), (7, 4.412531779962592e-05), (8, 1.8836180970538408e-05), (9, 1.995274578803219e-05), (10, 2.511048296582885e-05), (11, 3.2406995160272345e-05), (12, 3.2777650631032884e-05), (13, 7.106650446075946e-05), (14, 6.493292312370613e-05), (15, 8.06516909506172e-05), (16, 0.002394159324467182), (17, 4.6573812142014503e-05), (18, 0.00014503880811389536), (19, 0.00014503074635285884), (20, 7.627856393810362e-05), (21, 0.00022265786537900567), (22, 0.0001553944020997733), (23, 0.0002523193252272904), (24, 0.000528582138940692), (25, 0.0009549877722747624), (26, 0.0026171107310801744), (27, 0.010984783060848713)], Temporal MSE: [(0, 4.2530973587417975e-06), (1, 2.4711255264264764e-06), (2, 5.100886028230889e-06), (3, 6.959512575122062e-06), (4, 5.654439519275911e-06), (5, 1.014110694086412e-05), (6, 1.6075784515123814e-05), (7, 1.3803734873363283e-05), (8, 1.0053374353446998e-05), (9, 1.5545918358839117e-05), (10, 2.2127356714918278e-05), (11, 2.5227000151062384e-05), (12, 2.593415410956368e-05), (13, 4.986422209185548e-05), (14, 0.00017180541181005538), (15, 6.315289647318423e-05), (16, 0.00011570935021154583), (17, 4.978674041922204e-05), (18, 4.11951587011572e-05), (19, 4.920187348034233e-05), (20, 0.00018282729433849454), (21, 0.00013081020733807236), (22, 0.00011758194887079298), (23, 0.00018035982793662697), (24, 0.0002652399998623878), (25, 0.00046449207002297044), (26, 0.0006343042477965355), (27, 0.0002694255963433534)]
+
+ 72%|███████▏  | 36/50 [06:07<02:11,  9.37s/it]Time step 260, Spatial MSE: [(0, 1.3669598956767004e-05), (1, 2.5936831661965698e-05), (2, 5.3187650337349623e-05), (3, 5.246827640803531e-05), (4, 0.000163851203978993), (5, 0.00041138261440210044), (6, 0.0011435006745159626), (7, 0.0016491560963913798), (8, 0.0021100922022014856), (9, 0.0031210174784064293), (10, 0.0030107840429991484), (11, 0.0047884415835142136), (12, 0.005626531783491373), (13, 0.00692104734480381), (14, 0.010556688532233238), (15, 0.01674002781510353), (16, 0.015860186889767647), (17, 0.013423525728285313), (18, 0.015263506211340427), (19, 0.0135246142745018), (20, 0.01736721582710743), (21, 0.015930229797959328), (22, 0.012200254015624523), (23, 0.01844079978764057), (24, 0.03624655678868294), (25, 0.05280819535255432), (26, 0.0474502332508564), (27, 0.005797327496111393)], Temporal MSE: [(0, 5.652806521538878e-06), (1, 3.2878047022677492e-06), (2, 2.6128476747544482e-05), (3, 0.00024429397308267653), (4, 0.0002539778361096978), (5, 0.0003682851092889905), (6, 0.0007945362012833357), (7, 0.0008827659185044467), (8, 0.0009594662114977837), (9, 0.0016059931367635727), (10, 0.002889998722821474), (11, 0.004044553264975548), (12, 0.004445102531462908), (13, 0.00842249859124422), (14, 0.009888744913041592), (15, 0.008973401039838791), (16, 0.009328960441052914), (17, 0.008712874725461006), (18, 0.007929589599370956), (19, 0.008084497414529324), (20, 0.015057053416967392), (21, 0.011547315865755081), (22, 0.008436019532382488), (23, 0.011468354612588882), (24, 0.027621064335107803), (25, 0.043285712599754333), (26, 0.01711837388575077), (27, 0.001506922417320311)]
+
+ 74%|███████▍  | 37/50 [06:16<01:58,  9.15s/it]Time step 240, Spatial MSE: [(0, 7.625179478054633e-06), (1, 2.392366513959132e-05), (2, 4.235665255691856e-05), (3, 3.8136156945256516e-05), (4, 1.36405615194235e-05), (5, 2.20587298827013e-05), (6, 1.6420488464063965e-05), (7, 3.961283073294908e-05), (8, 2.2113305021775886e-05), (9, 2.243207927676849e-05), (10, 2.7492897061165422e-05), (11, 3.403691516723484e-05), (12, 3.613499939092435e-05), (13, 9.33360934141092e-05), (14, 0.0003242288075853139), (15, 0.00010341373854316771), (16, 0.010727472603321075), (17, 6.835260865045711e-05), (18, 0.0001423177309334278), (19, 0.0003852302033919841), (20, 0.0001722922024782747), (21, 0.0003656021144706756), (22, 0.0002801130758598447), (23, 0.00036645823274739087), (24, 0.0006683623651042581), (25, 0.0012083420297130942), (26, 0.002659962512552738), (27, 0.01426766999065876)], Temporal MSE: [(0, 5.0667417781369295e-06), (1, 2.991390601891908e-06), (2, 6.124636911408743e-06), (3, 9.782108463696204e-06), (4, 7.772887329338118e-06), (5, 1.2574528227560222e-05), (6, 1.7831707737059332e-05), (7, 1.4691173419123515e-05), (8, 1.2410620911396109e-05), (9, 1.905659883050248e-05), (10, 2.4148050215444528e-05), (11, 2.8270478651393205e-05), (12, 2.9841363357263617e-05), (13, 8.828756108414382e-05), (14, 0.0004653568903449923), (15, 7.341165473917499e-05), (16, 7.895182352513075e-05), (17, 5.4027153964852914e-05), (18, 9.544467320665717e-05), (19, 7.291211659321561e-05), (20, 0.00022026250371709466), (21, 0.00019187525322195143), (22, 0.0001799266174202785), (23, 0.0003270304878242314), (24, 0.00046468645450659096), (25, 0.000656329735647887), (26, 0.0008149108616635203), (27, 0.00048757370677776635)]
+
+ 76%|███████▌  | 38/50 [06:24<01:46,  8.86s/it]Time step 220, Spatial MSE: [(0, 1.4038199878996238e-05), (1, 2.4704531824681908e-05), (2, 8.143253944581375e-05), (3, 4.33800887549296e-05), (4, 4.259329580236226e-05), (5, 0.00010751829540822655), (6, 0.00018944244948215783), (7, 0.00027312804013490677), (8, 0.0004549241275526583), (9, 0.0004837835440412164), (10, 0.00065727619221434), (11, 0.0011217823484912515), (12, 0.0012838146649301052), (13, 0.0013735906686633825), (14, 0.0033498737029731274), (15, 0.0015511995879933238), (16, 0.011760007590055466), (17, 0.0009140694164671004), (18, 0.0013094553723931313), (19, 0.0011272610863670707), (20, 0.0013433987041935325), (21, 0.001270579406991601), (22, 0.00095562101341784), (23, 0.0016258149407804012), (24, 0.005384230986237526), (25, 0.01579325646162033), (26, 0.02941625565290451), (27, 0.010209792293608189)], Temporal MSE: [(0, 6.845977623015642e-06), (1, 3.425246404731297e-06), (2, 2.0130180928390473e-05), (3, 5.059775139670819e-05), (4, 5.177820276003331e-05), (5, 9.943467011908069e-05), (6, 0.00013666361337527633), (7, 0.00012980848259758204), (8, 0.00012624039663933218), (9, 0.000195391068700701), (10, 0.0003427565097808838), (11, 0.0004769252263940871), (12, 0.0004953707684762776), (13, 0.0009267934365198016), (14, 0.0015076574636623263), (15, 0.0006979454774409533), (16, 0.0005573310190811753), (17, 0.0004713842645287514), (18, 0.0006618716288357973), (19, 0.0005570302600972354), (20, 0.0013922607759013772), (21, 0.001110655372031033), (22, 0.0007626770529896021), (23, 0.0011137931142002344), (24, 0.0028834533877670765), (25, 0.008719752542674541), (26, 0.003628757782280445), (27, 0.0009593883878551424)]
+
+ 78%|███████▊  | 39/50 [06:35<01:43,  9.43s/it]Time step 200, Spatial MSE: [(0, 8.128340596158523e-06), (1, 2.8887498046969995e-05), (2, 8.368308044737205e-05), (3, 3.6201039620209485e-05), (4, 6.540973845403641e-05), (5, 0.0001804378698579967), (6, 0.0002586314221844077), (7, 0.0007306287880055606), (8, 0.00041040440555661917), (9, 0.0004147533036302775), (10, 0.0006266753771342337), (11, 0.0006282514077611268), (12, 0.0007116100168786943), (13, 0.0011071407934650779), (14, 0.002033754950389266), (15, 0.00368418637663126), (16, 0.019746938720345497), (17, 0.0013691602507606149), (18, 0.001689248252660036), (19, 0.0013923299266025424), (20, 0.0022854162380099297), (21, 0.0035026641562581062), (22, 0.0013828909723088145), (23, 0.0013642716221511364), (24, 0.00290249427780509), (25, 0.005602818913757801), (26, 0.004142908379435539), (27, 0.012216365896165371)], Temporal MSE: [(0, 6.971395578148076e-06), (1, 3.9454448597098235e-06), (2, 2.20571400859626e-05), (3, 0.00012299481022637337), (4, 0.00015309893933590502), (5, 0.00012429396156221628), (6, 0.0002561366418376565), (7, 0.00020216406846884638), (8, 0.0001683500304352492), (9, 0.0004003747599199414), (10, 0.0005290335393510759), (11, 0.0008081478881649673), (12, 0.001003756420686841), (13, 0.0021489213686436415), (14, 0.002799673704430461), (15, 0.0010520152281969786), (16, 0.0008532574283890426), (17, 0.0007087680278345942), (18, 0.0006531557883135974), (19, 0.0006100513273850083), (20, 0.0016448097303509712), (21, 0.0013931184075772762), (22, 0.000876430538482964), (23, 0.0013232170604169369), (24, 0.0034233161713927984), (25, 0.004648828878998756), (26, 0.004487971309572458), (27, 0.0016761490842327476)]
+
+ 80%|████████  | 40/50 [06:45<01:38,  9.85s/it]Time step 180, Spatial MSE: [(0, 1.3006007065996528e-05), (1, 2.9383159926510416e-05), (2, 0.00018105815979652107), (3, 5.0356422434560955e-05), (4, 9.998627501772717e-05), (5, 0.00017446960555389524), (6, 0.00028277712408453226), (7, 0.0005049029714427888), (8, 0.0005578432464972138), (9, 0.0008137295371852815), (10, 0.0009949089726433158), (11, 0.0012300795642659068), (12, 0.0016295512905344367), (13, 0.0023481189273297787), (14, 0.004930361174046993), (15, 0.004893215373158455), (16, 0.02097151428461075), (17, 0.0023721472825855017), (18, 0.0030123074539005756), (19, 0.003242447506636381), (20, 0.0033649413380771875), (21, 0.00390148744918406), (22, 0.0037783263251185417), (23, 0.005009623244404793), (24, 0.013213161379098892), (25, 0.03866806626319885), (26, 0.05754154920578003), (27, 0.019817369058728218)], Temporal MSE: [(0, 8.85151166585274e-06), (1, 4.3072413973277435e-06), (2, 2.5016654035425745e-05), (3, 4.68624202767387e-05), (4, 0.0001243853912455961), (5, 0.00019431048713158816), (6, 0.00024224082881119102), (7, 0.00027077062986791134), (8, 0.00031194405164569616), (9, 0.0004914414021186531), (10, 0.0008248899248428643), (11, 0.0013340270379558206), (12, 0.0014024381525814533), (13, 0.002586734713986516), (14, 0.0034239492379128933), (15, 0.0025263612624257803), (16, 0.00212544365786016), (17, 0.0013846507063135505), (18, 0.0013726925244554877), (19, 0.0013640661491081119), (20, 0.003925125114619732), (21, 0.0026849096175283194), (22, 0.0020555821247398853), (23, 0.002826301148161292), (24, 0.00720415823161602), (25, 0.01781177893280983), (26, 0.008001754060387611), (27, 0.0011316388845443726)]
+
+ 82%|████████▏ | 41/50 [06:57<01:33, 10.38s/it]Time step 160, Spatial MSE: [(0, 9.54349889070727e-06), (1, 3.5909066355088726e-05), (2, 0.00011241809988860041), (3, 4.114659532206133e-05), (4, 2.054830292763654e-05), (5, 3.2521227694815025e-05), (6, 2.3150410925154574e-05), (7, 6.971610127948225e-05), (8, 3.698747605085373e-05), (9, 3.498600563034415e-05), (10, 4.137224823352881e-05), (11, 5.1531293138395995e-05), (12, 5.4908501624595374e-05), (13, 0.00010918295447481796), (14, 7.638498937012628e-05), (15, 9.656127804191783e-05), (16, 0.00019275961676612496), (17, 9.851370850810781e-05), (18, 0.00012468462227843702), (19, 0.00016494753072038293), (20, 0.00014648231444880366), (21, 0.0001756042765919119), (22, 0.00026035308837890625), (23, 0.0004935446777381003), (24, 0.0008666855865158141), (25, 0.0014839214272797108), (26, 0.0027105496264994144), (27, 0.01531809102743864)], Temporal MSE: [(0, 8.526416422682814e-06), (1, 4.69930773761007e-06), (2, 8.760010132391471e-06), (3, 1.1302456186967902e-05), (4, 1.188506848848192e-05), (5, 1.9092920410912484e-05), (6, 2.601792039058637e-05), (7, 2.5653516786405817e-05), (8, 2.4022125217015855e-05), (9, 3.130715049337596e-05), (10, 4.2121591832255945e-05), (11, 4.79385707876645e-05), (12, 4.985999839846045e-05), (13, 8.487422746839002e-05), (14, 6.052626122254878e-05), (15, 7.55230212234892e-05), (16, 7.741515582893044e-05), (17, 6.151109846541658e-05), (18, 6.997479067649692e-05), (19, 7.362217002082616e-05), (20, 0.00010586091957520694), (21, 0.00013708426558878273), (22, 0.00016442810010630637), (23, 0.00026314775459468365), (24, 0.000404256337787956), (25, 0.0007325113983824849), (26, 0.0009404168231412768), (27, 0.0006453364621847868)]
+
+ 84%|████████▍ | 42/50 [07:07<01:22, 10.36s/it]Time step 140, Spatial MSE: [(0, 1.385285759170074e-05), (1, 3.6549092328641564e-05), (2, 0.0004225217562634498), (3, 6.876183033455163e-05), (4, 0.00020809733541682363), (5, 0.00035688679781742394), (6, 0.000617058714851737), (7, 0.0016839496092870831), (8, 0.0020829588174819946), (9, 0.003309068502858281), (10, 0.002915286924690008), (11, 0.00504649244248867), (12, 0.004964829422533512), (13, 0.005627283360809088), (14, 0.008436656557023525), (15, 0.016067201271653175), (16, 0.012278547510504723), (17, 0.011931455694139004), (18, 0.015376511961221695), (19, 0.012461686506867409), (20, 0.0156359001994133), (21, 0.01509451400488615), (22, 0.013362196274101734), (23, 0.019632238894701004), (24, 0.04029441997408867), (25, 0.061452772468328476), (26, 0.060021333396434784), (27, 0.00841071642935276)], Temporal MSE: [(0, 1.1908039596164599e-05), (1, 6.446415227401303e-06), (2, 3.965487121604383e-05), (3, 0.00018408057803753763), (4, 0.00021435912640299648), (5, 0.00028469867538660765), (6, 0.000745398283470422), (7, 0.0008643015753477812), (8, 0.000866892805788666), (9, 0.0015353537164628506), (10, 0.002835405757650733), (11, 0.004207996651530266), (12, 0.004325836896896362), (13, 0.00751050328835845), (14, 0.00992665532976389), (15, 0.008976754732429981), (16, 0.008171049878001213), (17, 0.007794906850904226), (18, 0.007136061787605286), (19, 0.007417292334139347), (20, 0.013822573237121105), (21, 0.011563117615878582), (22, 0.008258183486759663), (23, 0.01125046331435442), (24, 0.027105674147605896), (25, 0.04246006906032562), (26, 0.019672254100441933), (27, 0.002457248279824853)]
+
+ 86%|████████▌ | 43/50 [07:21<01:19, 11.31s/it]Time step 120, Spatial MSE: [(0, 1.2127205991419032e-05), (1, 5.143765883985907e-05), (2, 0.0002039253304246813), (3, 5.43405840289779e-05), (4, 2.8723801733576693e-05), (5, 5.063737626187503e-05), (6, 3.4646895073819906e-05), (7, 7.35329813323915e-05), (8, 5.895840877201408e-05), (9, 5.145898830960505e-05), (10, 5.9234051150269806e-05), (11, 7.103505049599335e-05), (12, 7.768968498567119e-05), (13, 0.00017250332166440785), (14, 0.0002904446155298501), (15, 0.00017574986850377172), (16, 0.005643945652991533), (17, 0.00014954798098187894), (18, 0.00021054969693068415), (19, 0.00039428475429303944), (20, 0.00024815055076032877), (21, 0.00046121919876895845), (22, 0.00046221964294090867), (23, 0.0007962347590364516), (24, 0.0013827348593622446), (25, 0.002288819756358862), (26, 0.003287539817392826), (27, 0.01995764672756195)], Temporal MSE: [(0, 1.1958117283938918e-05), (1, 6.428565029636957e-06), (2, 1.2611234524229076e-05), (3, 1.749330294842366e-05), (4, 1.846315171860624e-05), (5, 2.9200084100011736e-05), (6, 3.7860809243284166e-05), (7, 3.585243393899873e-05), (8, 3.89843771699816e-05), (9, 4.93167681270279e-05), (10, 6.260306690819561e-05), (11, 7.243961590575054e-05), (12, 7.792528049321845e-05), (13, 0.00018130912212654948), (14, 0.00033467684988863766), (15, 0.00014945022121537477), (16, 0.00012172140122856945), (17, 9.867499466054142e-05), (18, 0.00013022586063016206), (19, 0.00013373303227126598), (20, 0.00025198340881615877), (21, 0.00030336304917000234), (22, 0.0002676132135093212), (23, 0.0004488476552069187), (24, 0.0007479485939256847), (25, 0.001147718750871718), (26, 0.00143130321521312), (27, 0.0011084130965173244)]
+
+ 88%|████████▊ | 44/50 [07:32<01:06, 11.14s/it]Time step 100, Spatial MSE: [(0, 1.5152198102441616e-05), (1, 5.386896009440534e-05), (2, 0.0007458923500962555), (3, 7.346901838900521e-05), (4, 0.0001295803376706317), (5, 0.00017076643416658044), (6, 0.0003784731961786747), (7, 0.0006687414715997875), (8, 0.0009323129779659212), (9, 0.0014605461619794369), (10, 0.0012980017345398664), (11, 0.0018451743526384234), (12, 0.002052697818726301), (13, 0.0023646578192710876), (14, 0.004245731048285961), (15, 0.007237319368869066), (16, 0.010261707939207554), (17, 0.004348848946392536), (18, 0.0051456415094435215), (19, 0.0043495395220816135), (20, 0.005586695391684771), (21, 0.00571204861626029), (22, 0.005704598035663366), (23, 0.008932147175073624), (24, 0.017591949552297592), (25, 0.04233568161725998), (26, 0.06619424372911453), (27, 0.015958629548549652)], Temporal MSE: [(0, 1.7058082448784262e-05), (1, 8.798481758276466e-06), (2, 2.9032109523541294e-05), (3, 4.365162385511212e-05), (4, 6.441681034630165e-05), (5, 0.00010479360935278237), (6, 0.00021671922877430916), (7, 0.00023030961165204644), (8, 0.0002697368909139186), (9, 0.0005798027850687504), (10, 0.0008745462400838733), (11, 0.001552645000629127), (12, 0.0013951443834230304), (13, 0.002250158926472068), (14, 0.0030821284744888544), (15, 0.002671027323231101), (16, 0.0022891340777277946), (17, 0.0023807839024811983), (18, 0.00222954829223454), (19, 0.002186730969697237), (20, 0.0037840385921299458), (21, 0.0034772816579788923), (22, 0.0027956876438111067), (23, 0.0042908028699457645), (24, 0.0090201236307621), (25, 0.012640804052352905), (26, 0.00944460928440094), (27, 0.0016412487020716071)]
+
+ 90%|█████████ | 45/50 [07:41<00:52, 10.50s/it]Time step 80, Spatial MSE: [(0, 1.578560477355495e-05), (1, 7.505724352085963e-05), (2, 0.000662278151139617), (3, 5.43691057828255e-05), (4, 7.435218140017241e-05), (5, 0.00011411146260797977), (6, 0.00017153688531834632), (7, 0.0003222316736355424), (8, 0.00042768416460603476), (9, 0.0005488584865815938), (10, 0.000524645671248436), (11, 0.0006687088753096759), (12, 0.0007235286757349968), (13, 0.000811588775832206), (14, 0.001598034636117518), (15, 0.0033091444056481123), (16, 0.0018718383507803082), (17, 0.0019528395496308804), (18, 0.002215680433437228), (19, 0.002003343775868416), (20, 0.002663606544956565), (21, 0.0027914363890886307), (22, 0.0027618459425866604), (23, 0.004867212846875191), (24, 0.009205943904817104), (25, 0.02066180855035782), (26, 0.028901835903525352), (27, 0.018267428502440453)], Temporal MSE: [(0, 1.962530041055288e-05), (1, 9.987426892621443e-06), (2, 1.816295480239205e-05), (3, 2.5201008611475118e-05), (4, 3.158117033308372e-05), (5, 4.852436904911883e-05), (6, 0.00010172833572141826), (7, 9.90615226328373e-05), (8, 0.0001297073031309992), (9, 0.0002545094466768205), (10, 0.0003395874227862805), (11, 0.0006475226255133748), (12, 0.0005929407197982073), (13, 0.0008110231137834489), (14, 0.0012470438377931714), (15, 0.0011571473442018032), (16, 0.0009512142860330641), (17, 0.000949392793700099), (18, 0.0009853815427049994), (19, 0.0009620899800211191), (20, 0.0016515420284122229), (21, 0.0016083830269053578), (22, 0.0013380252057686448), (23, 0.0022803477477282286), (24, 0.004330071620643139), (25, 0.006344366818666458), (26, 0.00467356201261282), (27, 0.0010380340972915292)]
+
+ 92%|█████████▏| 46/50 [07:51<00:41, 10.45s/it]Time step 60, Spatial MSE: [(0, 2.0849205611739308e-05), (1, 0.00010588928125798702), (2, 0.0009247842826880515), (3, 7.559036021120846e-05), (4, 0.00010856086737476289), (5, 0.00015766847354825586), (6, 0.0002431180328130722), (7, 0.00043059579911641777), (8, 0.0005633321707136929), (9, 0.0007518253405578434), (10, 0.0006760046235285699), (11, 0.0008007941069081426), (12, 0.0009237880585715175), (13, 0.0010058371117338538), (14, 0.0019157422939315438), (15, 0.004068902228027582), (16, 0.0023240637965500355), (17, 0.002376621589064598), (18, 0.0029478834476321936), (19, 0.002710245316848159), (20, 0.0037387136835604906), (21, 0.003472929121926427), (22, 0.0037669676821678877), (23, 0.007022779900580645), (24, 0.013318696990609169), (25, 0.02945985086262226), (26, 0.036896321922540665), (27, 0.01964103803038597)], Temporal MSE: [(0, 2.6383982913102955e-05), (1, 1.3485277122526895e-05), (2, 2.5964438464143313e-05), (3, 3.295311762485653e-05), (4, 4.302717570681125e-05), (5, 6.873148959130049e-05), (6, 0.0001365762873319909), (7, 0.00013965349353384227), (8, 0.00019543292000889778), (9, 0.00036171230021864176), (10, 0.00043824047315865755), (11, 0.0008292323327623308), (12, 0.0007411738624796271), (13, 0.0010437374003231525), (14, 0.0016339668072760105), (15, 0.0014937251107767224), (16, 0.0012181834317743778), (17, 0.0011877869255840778), (18, 0.0013193282065913081), (19, 0.0012439163401722908), (20, 0.002175549976527691), (21, 0.0021388407330960035), (22, 0.001737186685204506), (23, 0.0031380774453282356), (24, 0.005865979474037886), (25, 0.008616373874247074), (26, 0.006728520151227713), (27, 0.00213945796713233)]
+
+ 94%|█████████▍| 47/50 [08:02<00:31, 10.51s/it]Time step 40, Spatial MSE: [(0, 3.079565067309886e-05), (1, 0.00016416893049608916), (2, 0.0013336552074179053), (3, 0.00011088137398473918), (4, 0.00016721838619560003), (5, 0.00023825223615858704), (6, 0.000400234799599275), (7, 0.000609382928814739), (8, 0.0007973331375978887), (9, 0.0010407711379230022), (10, 0.0009459661669097841), (11, 0.001026381622068584), (12, 0.0011545238085091114), (13, 0.0013013356365263462), (14, 0.00238274154253304), (15, 0.005246396642178297), (16, 0.0030934715177863836), (17, 0.0031974930316209793), (18, 0.004043556284159422), (19, 0.003615675261244178), (20, 0.004918861202895641), (21, 0.0048986454494297504), (22, 0.0053241620771586895), (23, 0.010406456887722015), (24, 0.020203741267323494), (25, 0.04171972721815109), (26, 0.05050522834062576), (27, 0.02270469069480896)], Temporal MSE: [(0, 3.867781924782321e-05), (1, 1.956643973244354e-05), (2, 4.390675894683227e-05), (3, 4.569211523630656e-05), (4, 6.29991918685846e-05), (5, 0.00010698068217607215), (6, 0.0002120364224538207), (7, 0.00021941724116913974), (8, 0.0003243796818424016), (9, 0.0005177510902285576), (10, 0.0005839867517352104), (11, 0.0010421370388939977), (12, 0.0009552314877510071), (13, 0.0013792249374091625), (14, 0.002195837674662471), (15, 0.001975489314645529), (16, 0.0016042417846620083), (17, 0.0015080883167684078), (18, 0.0016929253470152617), (19, 0.0016435426659882069), (20, 0.002853083424270153), (21, 0.00287827430292964), (22, 0.002290587406605482), (23, 0.00444658100605011), (24, 0.00830749049782753), (25, 0.012294630520045757), (26, 0.01071656122803688), (27, 0.007780771236866713)]
+
+ 96%|█████████▌| 48/50 [08:13<00:21, 10.84s/it]Time step 20, Spatial MSE: [(0, 5.8176967286271974e-05), (1, 0.00031785634928382933), (2, 0.0021953959949314594), (3, 0.00019568846619222313), (4, 0.0002931539493147284), (5, 0.00043226496200077236), (6, 0.0008860246744006872), (7, 0.0010975186014547944), (8, 0.0014309262624010444), (9, 0.001816075760871172), (10, 0.0016031184932217002), (11, 0.001525952247902751), (12, 0.0016612927429378033), (13, 0.00200066389515996), (14, 0.0033439649268984795), (15, 0.007539020851254463), (16, 0.004814485553652048), (17, 0.004904531873762608), (18, 0.005945759359747171), (19, 0.005418472457677126), (20, 0.00718813156709075), (21, 0.008068734779953957), (22, 0.008263309486210346), (23, 0.016761554405093193), (24, 0.0349501334130764), (25, 0.06298196315765381), (26, 0.07882240414619446), (27, 0.03438372164964676)], Temporal MSE: [(0, 6.940363527974114e-05), (1, 3.4428325307089835e-05), (2, 9.247937850886956e-05), (3, 7.707838085480034e-05), (4, 0.00011095323134213686), (5, 0.00020552870410028845), (6, 0.0004138470103498548), (7, 0.0004363999469205737), (8, 0.0006824680604040623), (9, 0.000868865055963397), (10, 0.0008807404665276408), (11, 0.0013707764446735382), (12, 0.0013673728099092841), (13, 0.0020111622288823128), (14, 0.003138594329357147), (15, 0.002866128459572792), (16, 0.002281693508848548), (17, 0.0022040759213268757), (18, 0.0025896814186125994), (19, 0.0024609798565506935), (20, 0.0040612611919641495), (21, 0.004213002976030111), (22, 0.003568907966837287), (23, 0.007095005828887224), (24, 0.014790932647883892), (25, 0.02196592651307583), (26, 0.0238486398011446), (27, 0.07191947847604752)]
+
+ 98%|█████████▊| 49/50 [08:25<00:11, 11.03s/it]Time step 0, Spatial MSE: [(0, 0.00040899578016251326), (1, 0.0017691167304292321), (2, 0.009792687371373177), (3, 0.0010772717650979757), (4, 0.001604409422725439), (5, 0.0022245405707508326), (6, 0.003945947624742985), (7, 0.005721864756196737), (8, 0.007788224145770073), (9, 0.009555251337587833), (10, 0.00851038470864296), (11, 0.005448815878480673), (12, 0.006345560774207115), (13, 0.0069321137852966785), (14, 0.010089246556162834), (15, 0.019571470096707344), (16, 0.015027742832899094), (17, 0.013520400039851665), (18, 0.016811301931738853), (19, 0.021112605929374695), (20, 0.0206502303481102), (21, 0.027496572583913803), (22, 0.03447730839252472), (23, 0.0602700375020504), (24, 0.16075727343559265), (25, 0.6736361980438232), (26, 0.8511895537376404), (27, 2.4109127521514893)], Temporal MSE: [(0, 0.0005446782452054322), (1, 0.00021486295736394823), (2, 0.0006899273139424622), (3, 0.00045972142834216356), (4, 0.0006893014651723206), (5, 0.0010051163844764233), (6, 0.0024586289655417204), (7, 0.002330171409994364), (8, 0.0036895934026688337), (9, 0.0033240406773984432), (10, 0.0037321283016353846), (11, 0.0040514967404305935), (12, 0.004438894800841808), (13, 0.005731544457376003), (14, 0.008204204961657524), (15, 0.00806442927569151), (16, 0.005975133739411831), (17, 0.005947021767497063), (18, 0.00873466394841671), (19, 0.007743417285382748), (20, 0.013393275439739227), (21, 0.012142171151936054), (22, 0.014829648658633232), (23, 0.026564642786979675), (24, 0.07682987302541733), (25, 0.1446557193994522), (26, 0.3698791563510895), (27, 5.507018566131592)]
+
+100%|██████████| 50/50 [08:34<00:00, 10.64s/it]
+100%|██████████| 50/50 [08:34<00:00, 10.30s/it]
+Saved video to ./samples/latte/Time_Lapse_of_the_rising_sun_over_a_tree_in_an_open_rural_landscape,_with_clouds_in_the_blue_sky_beautifully_playing_with_the_rays_of_light_0000.mp4
+Saved spatial MLP MSE outputs to ./samples/latte/Time_Lapse_of_the_rising_sun_over_a_tree_in_an_open_rural_landscape,_with_clouds_in_the_blue_sky_beautifully_playing_with_the_rays_of_light_0000_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/latte/Time_Lapse_of_the_rising_sun_over_a_tree_in_an_open_rural_landscape,_with_clouds_in_the_blue_sky_beautifully_playing_with_the_rays_of_light_0000_temporal_mlp_mse.pt
diff --git a/sample_opensora_mse.txt b/sample_opensora_mse.txt
new file mode 100644
index 00000000..dea2b1d6
--- /dev/null
+++ b/sample_opensora_mse.txt
@@ -0,0 +1,382 @@
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/diffusers/models/transformers/transformer_2d.py:34: FutureWarning: `Transformer2DModelOutput` is deprecated and will be removed in version 1.0.0. Importing `Transformer2DModelOutput` from `diffusers.models.transformer_2d` is deprecated and this will be removed in a future version. Please use `from diffusers.models.modeling_outputs import Transformer2DModelOutput`, instead.
+  deprecate("Transformer2DModelOutput", "1.0.0", deprecation_message)
+[10/01/24 14:48:44] INFO     colossalai - colossalai - INFO:
+                             /home/jin509/anaconda3/envs/opendit/lib/python3.10/
+                             site-packages/colossalai/initialize.py:60 launch
+                    INFO     colossalai - colossalai - INFO: Distributed
+                             environment is initialized, world size: 1
+
+Init SkipManager:
+    steps=30
+    cross_broadcast=True, cross_threshold=[450, 930], cross_gap=6
+    spatial_broadcast=True, spatial_threshold=[450, 930], spatial_gap=2
+    temporal_broadcast=True, temporal_threshold=[450, 930], temporal_gap=4
+    mlp_skip=False
+
+[2024-10-01 14:48:44] Inference configuration: Namespace(config='configs/opensora/sample_pab.yaml', seed=42, batch_size=1, flash_attn=True, enable_t5_speedup=True, resolution='480p', multi_resolution='STDiT2', dtype='bf16', save_dir='./samples/opensora', num_sample=1, prompt_as_path=True, verbose=2, prompt_path=None, prompt=['Sunset over the sea.', 'A dog in astronaut suit and sunglasses floating in space.', 'An epic tornado attacking above aglowing city at night.', 'Snow falling over multiple houses and trees on winter landscape against night sky. christmas festivity and celebration concept.', 'Time Lapse of the rising sun over a tree in an open rural landscape, with clouds in the blue sky beautifully playing with the rays of light'], llm_refine=False, num_frames=51, fps=24, image_size=None, frame_interval=1, aspect_ratio='9:16', watermark=False, num_sampling_steps=30, cfg_scale=7.0, loop=1, align=5, condition_frame_length=5, condition_frame_edit=0.0, reference_path=None, mask_strategy=None, aes=6.5, flow=None, camera_motion=None, spatial_broadcast=True, spatial_threshold=[450, 930], spatial_gap=2, temporal_broadcast=True, temporal_threshold=[450, 930], temporal_gap=4, cross_broadcast=True, cross_threshold=[450, 930], cross_gap=6, mlp_skip=False, mlp_temporal_broadcast_config=None, mlp_spatial_broadcast_config=None, diffusion_skip=False, diffusion_skip_timestep=None)
+
+[2024-10-01 14:48:44] Building models...
+You are using the default legacy behaviour of the <class 'transformers.models.t5.tokenization_t5.T5Tokenizer'>. This is expected, and simply means that the `legacy` (previous) behavior will be used so nothing changes for you. If you want to use the new behaviour, set `legacy=False`. This should only be set if you understand what it means, and thoroughly read the reason why this was added as explained in https://github.com/huggingface/transformers/pull/24565
+
+Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]
+Loading checkpoint shards:  50%|█████     | 1/2 [01:03<01:03, 63.82s/it]
+Loading checkpoint shards: 100%|██████████| 2/2 [02:09<00:00, 64.83s/it]
+Loading checkpoint shards: 100%|██████████| 2/2 [02:09<00:00, 64.68s/it]
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/colossalai/pipeline/schedule/_utils.py:19: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
+  _register_pytree_node(OrderedDict, _odict_flatten, _odict_unflatten)
+/home/jin509/anaconda3/envs/opendit/lib/pyt
+Loading checkpoint shards: 100%|████████�
+Loading checkpoint shards: 100%|██████████| 2/2 [02:16<00:00, 67.57s/it]
+Loading checkpoint shards: 100%|██████████| 2/2 [02:16<00:00, 68.02s/it]
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/colossalai/pipeline/schedule/_utils.py:19: UserWarning: torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.
+  _register_pytree_node(OrderedDict, _odsave_dir | ./samples/opensora/sample_pab
+Arguments saved to ./samples/opensora/sample_pab/args.json
+timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:0'), tensor([828.8770], device='cuda:0')timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv3d(
+
+  0%|          | 0/30 [00:01<?, ?it/s]
+[rank0]: Traceback (most recent call last):
+[rank0]:   File "/home/jin509/OpenDiT/scripts/opensora/sample_opensora_mse.py", line 450, in <module>
+[rank0]:     main(args)
+[rank0]:   File "/home/jin509/OpenDiT/sc/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/
+  3%|▎         | 1/30 [00:09<04:27,  9.23s/it]Time step 992, Spatial MSE: [(0, 0.0009266708511859179), (1, 0.001964372815564275), (2, 0.020445087924599648), (3, 0.021416611969470978), (4, 0.024798057973384857), (5, 0.05009352043271065), (6, 0.04161224141716957), (7, 0.07172810286283493), (8, 0.08762899786233902), (9, 0.2350788116455078), (10, 0.3339252173900604), (11, 0.38562941551208496), (12, 0.4068857431411743), (13, 0.3412085473537445), (14, 0.7005358338356018), (15, 0.6336411237716675), (16, 1.2084649801254272), (17, 0.6472557783126831), (18, 0.44402867555618286), (19, 0.4843830466270447), (20, 0.9968878626823425), (21, 0.7619172930717468), (22, 0.621554434299469), (23, 1.4418096542358398), (24, 7.050262451171875), (25, 6.062499523162842), (26, 5.6432647705078125), (27, 0.6295115351676941)], Temporal MSE: [(0, 4.37132730439771e-05), (1, 0.0004547384742181748), (2, 0.0013936946634203196), (3, 0.0020310101099312305), (4, 0.002069223904982209), (5, 0.007184454705566168), (6, 0.004656512290239334), (7, 0.008510898798704147), (8, 0.007425728254020214), (9, 0.013835558667778969), (10, 0.024980736896395683), (11, 0.031195254996418953), (12, 0.04029916226863861), (13, 0.023357659578323364), (14, 0.09227907657623291), (15, 0.14006251096725464), (16, 0.17229478061199188), (17, 0.08008652925491333), (18, 0.08565345406532288), (19, 0.09104886651039124), (20, 0.0826355516910553), (21, 0.14083971083164215), (22, 0.43182292580604553), (23, 0.42485642433166504), (24, 0.19641447067260742), (25, 0.11965048313140869), (26, 0.6062396764755249), (27, 0.8362858891487122)]
+
+  7%|▋         | 2/30 [00:21<05:14, 11.24s/it]Time step 984, Spatial MSE: [(0, 0.0004006043600384146), (1, 0.0015674657188355923), (2, 0.006948538590222597), (3, 0.005912716966122389), (4, 0.0072502014227211475), (5, 0.008970772847533226), (6, 0.014967601746320724), (7, 0.025083597749471664), (8, 0.032312605530023575), (9, 0.07322235405445099), (10, 0.1464146226644516), (11, 0.13587170839309692), (12, 0.14334699511528015), (13, 0.14076267182826996), (14, 0.2820338308811188), (15, 0.36522674560546875), (16, 0.36476871371269226), (17, 0.2858518660068512), (18, 0.3078519105911255), (19, 0.29815563559532166), (20, 0.29479435086250305), (21, 0.379610151052475), (22, 0.3598935306072235), (23, 0.41742438077926636), (24, 1.4148459434509277), (25, 1.1107865571975708), (26, 0.8324218392372131), (27, 0.13442839682102203)], Temporal MSE: [(0, 7.0442597461806145e-06), (1, 0.00017014547483995557), (2, 0.0002885904104914516), (3, 0.00048401099047623575), (4, 0.0005046654841862619), (5, 0.001635013846680522), (6, 0.001332456129603088), (7, 0.006680983118712902), (8, 0.0048481253907084465), (9, 0.006023965310305357), (10, 0.011178079061210155), (11, 0.012565084733068943), (12, 0.01089107058942318), (13, 0.009719246998429298), (14, 0.05527784302830696), (15, 0.08385003358125687), (16, 0.05810687318444252), (17, 0.038452163338661194), (18, 0.040270425379276276), (19, 0.048138607293367386), (20, 0.06920678168535233), (21, 0.08912437409162521), (22, 0.18188022077083588), (23, 0.17255337536334991), (24, 0.08186456561088562), (25, 0.05199260264635086), (26, 0.11436077207326889), (27, 0.08013878762722015)]
+
+ 10%|█         | 3/30 [00:35<05:27, 12.14s/it]Time step 976, Spatial MSE: [(0, 0.0001727823109831661), (1, 0.0006860928260721266), (2, 0.002079337602481246), (3, 0.0015498434659093618), (4, 0.0018771053291857243), (5, 0.00224782875739038), (6, 0.004517590161412954), (7, 0.008605698123574257), (8, 0.009618696756660938), (9, 0.01589665748178959), (10, 0.03976340591907501), (11, 0.03559447079896927), (12, 0.027149515226483345), (13, 0.028175532817840576), (14, 0.09069004654884338), (15, 0.05815749242901802), (16, 0.09089938551187515), (17, 0.06441894918680191), (18, 0.06571715325117111), (19, 0.06271632015705109), (20, 0.07174547761678696), (21, 0.06651832163333893), (22, 0.0719655454158783), (23, 0.07704268395900726), (24, 0.20802593231201172), (25, 0.1995108425617218), (26, 0.12982037663459778), (27, 0.04146185144782066)], Temporal MSE: [(0, 3.0767628231842536e-06), (1, 4.991464447812177e-05), (2, 0.00011014417395927012), (3, 0.00017054576892405748), (4, 0.0001959232467925176), (5, 0.0005941847921349108), (6, 0.0005628261133097112), (7, 0.0024047265760600567), (8, 0.0015033873496577144), (9, 0.0020875507034361362), (10, 0.0032916245982050896), (11, 0.0031261141411960125), (12, 0.002398868091404438), (13, 0.002356247277930379), (14, 0.010932584293186665), (15, 0.015843834728002548), (16, 0.015390238724648952), (17, 0.011027290485799313), (18, 0.010268963873386383), (19, 0.010804057121276855), (20, 0.02292189933359623), (21, 0.032459937036037445), (22, 0.0368894524872303), (23, 0.033287856727838516), (24, 0.016354206949472427), (25, 0.007656225934624672), (26, 0.01898087002336979), (27, 0.018119309097528458)]
+
+ 13%|█▎        | 4/30 [00:48<05:29, 12.68s/it]Time step 968, Spatial MSE: [(0, 0.00010960254439851269), (1, 0.00041812710696831346), (2, 0.0014024536358192563), (3, 0.0006026348564773798), (4, 0.0009371120249852538), (5, 0.0011854877229779959), (6, 0.002584119327366352), (7, 0.004766265861690044), (8, 0.004756585229188204), (9, 0.0074585494585335255), (10, 0.015636006370186806), (11, 0.016651278361678123), (12, 0.014092809520661831), (13, 0.01447084080427885), (14, 0.05387013405561447), (15, 0.04038003087043762), (16, 0.046666257083415985), (17, 0.04488399997353554), (18, 0.05084751546382904), (19, 0.04385882988572121), (20, 0.035080019384622574), (21, 0.04347064346075058), (22, 0.05180162936449051), (23, 0.048293210566043854), (24, 0.09765525907278061), (25, 0.08819663524627686), (26, 0.062382277101278305), (27, 0.03582541644573212)], Temporal MSE: [(0, 1.7567582517585834e-06), (1, 2.2844798877486028e-05), (2, 8.26762116048485e-05), (3, 8.782563963904977e-05), (4, 0.00013262232823763043), (5, 0.0003562266065273434), (6, 0.00031596660846844316), (7, 0.001102547044865787), (8, 0.0008168087806552649), (9, 0.0010255518136546016), (10, 0.0012601652415469289), (11, 0.0013915509916841984), (12, 0.0011155422544106841), (13, 0.0011673681437969208), (14, 0.0056514739990234375), (15, 0.008906176313757896), (16, 0.008293483406305313), (17, 0.007517124991863966), (18, 0.0074993004091084), (19, 0.008325088769197464), (20, 0.012617958709597588), (21, 0.024936256930232048), (22, 0.019689854234457016), (23, 0.019675038754940033), (24, 0.011224239133298397), (25, 0.004602625500410795), (26, 0.009749721735715866), (27, 0.008444169536232948)]
+
+ 17%|█▋        | 5/30 [01:02<05:26, 13.04s/it]Time step 960, Spatial MSE: [(0, 7.807557994965464e-05), (1, 0.00046529382234439254), (2, 0.0010989962611347437), (3, 0.000597073114477098), (4, 0.0006496207788586617), (5, 0.0008811640436761081), (6, 0.0019633930642157793), (7, 0.0032501493114978075), (8, 0.0034625132102519274), (9, 0.005906999576836824), (10, 0.014647521078586578), (11, 0.014449208974838257), (12, 0.013379567302763462), (13, 0.011510612443089485), (14, 0.05505785346031189), (15, 0.041723039001226425), (16, 0.05323343351483345), (17, 0.04218742251396179), (18, 0.05014714598655701), (19, 0.04473369941115379), (20, 0.039304886013269424), (21, 0.03324873745441437), (22, 0.04593143239617348), (23, 0.051350899040699005), (24, 0.07500309497117996), (25, 0.064060740172863), (26, 0.037272218614816666), (27, 0.019546428695321083)], Temporal MSE: [(0, 1.539536242489703e-06), (1, 1.2849155609728768e-05), (2, 6.214666063897312e-05), (3, 7.150188321247697e-05), (4, 0.0001291336229769513), (5, 0.00030068730120547116), (6, 0.00033650416298769414), (7, 0.0012281473027542233), (8, 0.0006525879725813866), (9, 0.0008248192607425153), (10, 0.0014843086246401072), (11, 0.0014821819495409727), (12, 0.0012269431026652455), (13, 0.00128587381914258), (14, 0.006934084463864565), (15, 0.009905199520289898), (16, 0.00912797637283802), (17, 0.008717631921172142), (18, 0.007950297556817532), (19, 0.008846523240208626), (20, 0.011944395489990711), (21, 0.021406568586826324), (22, 0.014286103658378124), (23, 0.015795107930898666), (24, 0.009501827880740166), (25, 0.003874744288623333), (26, 0.006433569360524416), (27, 0.004826321266591549)]
+
+ 20%|██        | 6/30 [01:17<05:26, 13.62s/it]Time step 952, Spatial MSE: [(0, 7.491833821404725e-05), (1, 0.00033736717887222767), (2, 0.0007922904915176332), (3, 0.0003180203784722835), (4, 0.00040328802424483), (5, 0.0005527707980945706), (6, 0.0012780980905517936), (7, 0.002014124533161521), (8, 0.0023985544685274363), (9, 0.0038997442461550236), (10, 0.010480476543307304), (11, 0.01181764341890812), (12, 0.01249717827886343), (13, 0.011255301535129547), (14, 0.05175250023603439), (15, 0.041526660323143005), (16, 0.05169491097331047), (17, 0.0411551296710968), (18, 0.05044740438461304), (19, 0.048104241490364075), (20, 0.042700596153736115), (21, 0.041923388838768005), (22, 0.051699504256248474), (23, 0.053591035306453705), (24, 0.06777022778987885), (25, 0.059386447072029114), (26, 0.03717577084898949), (27, 0.017370106652379036)], Temporal MSE: [(0, 1.155103177552519e-06), (1, 8.407658242504112e-06), (2, 4.7233508666977286e-05), (3, 4.1629573388490826e-05), (4, 6.580636545550078e-05), (5, 0.00013343969476409256), (6, 0.0001995993370655924), (7, 0.0007717765402048826), (8, 0.0004202634154353291), (9, 0.000504080846440047), (10, 0.0010434846626594663), (11, 0.0013803160982206464), (12, 0.0012947970535606146), (13, 0.0016057944158092141), (14, 0.006894640624523163), (15, 0.0109937135130167), (16, 0.009676747024059296), (17, 0.008619545958936214), (18, 0.007001226767897606), (19, 0.008823255077004433), (20, 0.015195740386843681), (21, 0.02463427744805813), (22, 0.01486996840685606), (23, 0.01656477153301239), (24, 0.010339288972318172), (25, 0.0030757852364331484), (26, 0.0057394192554056644), (27, 0.004841858521103859)]
+
+ 23%|██▎       | 7/30 [01:31<05:22, 14.00s/it]Time step 940, Spatial MSE: [(0, 7.24489291314967e-05), (1, 0.000313686701701954), (2, 0.0007126562995836139), (3, 0.000358480989234522), (4, 0.0006416052929125726), (5, 0.0008064563735388219), (6, 0.0019163156393915415), (7, 0.0037589105777442455), (8, 0.0032417860347777605), (9, 0.00538463331758976), (10, 0.014927765354514122), (11, 0.01713404804468155), (12, 0.015686118975281715), (13, 0.014841865748167038), (14, 0.05118165537714958), (15, 0.03923684358596802), (16, 0.052187684923410416), (17, 0.044145792722702026), (18, 0.0442981980741024), (19, 0.04453035071492195), (20, 0.03709688037633896), (21, 0.0405033640563488), (22, 0.0443095825612545), (23, 0.04004473239183426), (24, 0.06612729281187057), (25, 0.08519932627677917), (26, 0.047783732414245605), (27, 0.04844006150960922)], Temporal MSE: [(0, 1.3332016806089086e-06), (1, 9.380561095895246e-06), (2, 6.583630602108315e-05), (3, 4.633643402485177e-05), (4, 0.0001370745012536645), (5, 0.0001841498160501942), (6, 0.00034363308805041015), (7, 0.00146920804399997), (8, 0.0007672221399843693), (9, 0.0008992550428956747), (10, 0.0018642748473212123), (11, 0.0024451951030641794), (12, 0.002077569719403982), (13, 0.0021917736157774925), (14, 0.009085124358534813), (15, 0.012444879859685898), (16, 0.011418365873396397), (17, 0.009851033799350262), (18, 0.0071867844089865685), (19, 0.008445629850029945), (20, 0.015931081026792526), (21, 0.01938108541071415), (22, 0.015059318393468857), (23, 0.018062695860862732), (24, 0.010663241147994995), (25, 0.004695422947406769), (26, 0.01068099495023489), (27, 0.011487695388495922)]
+
+ 27%|██▋       | 8/30 [01:45<05:06, 13.95s/it]Time step 932, Spatial MSE: [(0, 7.119955989765003e-05), (1, 0.00028675212524831295), (2, 0.0004625353030860424), (3, 0.00026933461776934564), (4, 0.00022043999342713505), (5, 0.0003872669185511768), (6, 0.0008228204678744078), (7, 0.0011802359949797392), (8, 0.001794639159925282), (9, 0.0030920079443603754), (10, 0.008025375194847584), (11, 0.008147034794092178), (12, 0.007810922339558601), (13, 0.007287824060767889), (14, 0.01993078924715519), (15, 0.01634838618338108), (16, 0.02202003076672554), (17, 0.018556207418441772), (18, 0.01736723631620407), (19, 0.016203269362449646), (20, 0.01625337265431881), (21, 0.015932129696011543), (22, 0.016816506162285805), (23, 0.016324644908308983), (24, 0.023658620193600655), (25, 0.026683300733566284), (26, 0.020539240911602974), (27, 0.00960638839751482)], Temporal MSE: [(0, 1.1134844726257143e-06), (1, 4.394074039737461e-06), (2, 3.581352939363569e-05), (3, 2.9611974241561256e-05), (4, 6.0331090935505927e-05), (5, 9.05815904843621e-05), (6, 0.0001576998911332339), (7, 0.0004814566345885396), (8, 0.0002851993776857853), (9, 0.0004079692007508129), (10, 0.0009376517264172435), (11, 0.0012169694527983665), (12, 0.0007450192933902144), (13, 0.0009936566930264235), (14, 0.0035727820359170437), (15, 0.004670949187129736), (16, 0.004847222939133644), (17, 0.004173140041530132), (18, 0.0034396275877952576), (19, 0.004211949184536934), (20, 0.008950996212661266), (21, 0.008040103130042553), (22, 0.0057085612788796425), (23, 0.0069803413935005665), (24, 0.004266250412911177), (25, 0.001727919327095151), (26, 0.004005072172731161), (27, 0.0022541380021721125)]
+
+ 30%|███       | 9/30 [01:57<04:39, 13.30s/it]Time step 920, Spatial MSE: [(0, 3.0278890335466713e-05), (1, 0.0003708008152898401), (2, 0.000609726004768163), (3, 0.00015430247003678232), (4, 7.25817953934893e-05), (5, 0.00011898245429620147), (6, 0.00015545588394161314), (7, 0.0006153929862193763), (8, 0.0001837576855905354), (9, 0.00033650349359959364), (10, 0.0003101854817941785), (11, 0.0002864672860596329), (12, 0.000436040892964229), (13, 0.0005159039865247905), (14, 0.0007050050189718604), (15, 0.0013454710133373737), (16, 0.0034384452737867832), (17, 0.00132788997143507), (18, 0.0024471620563417673), (19, 0.0016935461899265647), (20, 0.0016078249318525195), (21, 0.0020757552701979876), (22, 0.0017879079096019268), (23, 0.002910945098847151), (24, 0.005708375014364719), (25, 0.014250732026994228), (26, 0.011190512217581272), (27, 0.013866313733160496)], Temporal MSE: [(0, 7.933289225547924e-07), (1, 3.539646286299103e-06), (2, 1.396911102347076e-05), (3, 1.4117569662630558e-05), (4, 1.383861672366038e-05), (5, 1.8226814063382335e-05), (6, 2.0563125872286037e-05), (7, 6.425527681130916e-05), (8, 2.5504721634206362e-05), (9, 3.890625885105692e-05), (10, 7.93784492998384e-05), (11, 8.266719669336453e-05), (12, 7.661200652364641e-05), (13, 7.066615216899663e-05), (14, 0.00024042761651799083), (15, 0.000641595630440861), (16, 0.0005099154659546912), (17, 0.000582410953938961), (18, 0.000759028538595885), (19, 0.0007954341126605868), (20, 0.0015391969354823232), (21, 0.0009067552746273577), (22, 0.000733073044102639), (23, 0.000899129721801728), (24, 0.0005714813596569002), (25, 0.0007526343688368797), (26, 0.0015172931598499417), (27, 0.001010350533761084)]
+
+ 33%|███▎      | 10/30 [02:10<04:23, 13.16s/it]Time step 908, Spatial MSE: [(0, 0.000166091529536061), (1, 0.0003747760783880949), (2, 0.0007242324063554406), (3, 0.0002913876960519701), (4, 0.0004052725271321833), (5, 0.0007809372036717832), (6, 0.0013191973557695746), (7, 0.001998354448005557), (8, 0.001514422008767724), (9, 0.0021812294144183397), (10, 0.008486056700348854), (11, 0.0062818448059260845), (12, 0.005737566389143467), (13, 0.006873187609016895), (14, 0.021637286990880966), (15, 0.009712382219731808), (16, 0.02292553335428238), (17, 0.009379059076309204), (18, 0.009110985323786736), (19, 0.0113415801897645), (20, 0.007648908067494631), (21, 0.013542670756578445), (22, 0.01163298636674881), (23, 0.010621827095746994), (24, 0.020733660086989403), (25, 0.04453950375318527), (26, 0.016548767685890198), (27, 0.01849394477903843)], Temporal MSE: [(0, 2.794477268253104e-06), (1, 4.581743723974796e-06), (2, 3.950664904550649e-05), (3, 8.736427844269201e-05), (4, 4.8264166252920404e-05), (5, 8.814658940536901e-05), (6, 0.00011145344615215436), (7, 0.0005860329838469625), (8, 0.0003770626208279282), (9, 0.0004321820742916316), (10, 0.0011011934839189053), (11, 0.0009614612208679318), (12, 0.0009641729411669075), (13, 0.0006906198104843497), (14, 0.0031227811705321074), (15, 0.004269621334969997), (16, 0.005705099552869797), (17, 0.003135703969746828), (18, 0.0019377312855795026), (19, 0.003500736551359296), (20, 0.0031091910786926746), (21, 0.005281544290482998), (22, 0.0062146857380867004), (23, 0.0054703098721802235), (24, 0.003534231334924698), (25, 0.003630605759099126), (26, 0.005111976992338896), (27, 0.003539988538250327)]
+
+ 37%|███▋      | 11/30 [02:22<04:03, 12.83s/it]Time step 892, Spatial MSE: [(0, 3.1562474759994075e-05), (1, 0.0004946729750372469), (2, 0.0006658314960077405), (3, 0.000179580194526352), (4, 6.353826029226184e-05), (5, 0.0001127565119531937), (6, 0.00013921920617576689), (7, 0.0004576848295982927), (8, 0.000183205702342093), (9, 0.00037068245001137257), (10, 0.0002721789642237127), (11, 0.0003467546193860471), (12, 0.000334849173668772), (13, 0.0006264886469580233), (14, 0.0007569960434921086), (15, 0.0011909744935110211), (16, 0.005695628933608532), (17, 0.001277548843063414), (18, 0.0020686520729213953), (19, 0.0026272526010870934), (20, 0.001960916444659233), (21, 0.0035630566999316216), (22, 0.002265702001750469), (23, 0.0038038240745663643), (24, 0.0075464555993676186), (25, 0.012239932082593441), (26, 0.012617518194019794), (27, 0.010119365528225899)], Temporal MSE: [(0, 7.803503763170738e-07), (1, 3.4286010759387864e-06), (2, 1.3833458069711924e-05), (3, 1.4055321116757113e-05), (4, 1.3875684999220539e-05), (5, 1.826144944061525e-05), (6, 1.760348277457524e-05), (7, 5.911258631385863e-05), (8, 3.007375016750302e-05), (9, 3.579455733415671e-05), (10, 7.133802137104794e-05), (11, 7.646727317478508e-05), (12, 6.882995512569323e-05), (13, 7.182767876656726e-05), (14, 0.0002595437108539045), (15, 0.0006668344722129405), (16, 0.0006494440021924675), (17, 0.0006806309102103114), (18, 0.0003749497118405998), (19, 0.0012199671473354101), (20, 0.0028366846963763237), (21, 0.001109774922952056), (22, 0.0009198836633004248), (23, 0.0008319175685755908), (24, 0.0007375697605311871), (25, 0.0007241280400194228), (26, 0.001117958570830524), (27, 0.0010280871065333486)]
+
+ 40%|████      | 12/30 [02:35<03:50, 12.80s/it]Time step 880, Spatial MSE: [(0, 0.0001565396087244153), (1, 0.0004383042687550187), (2, 0.0010288938647136092), (3, 0.0013283160515129566), (4, 0.0015410494524985552), (5, 0.002939558820798993), (6, 0.006559786852449179), (7, 0.013643840327858925), (8, 0.019947422668337822), (9, 0.029497215524315834), (10, 0.06092239171266556), (11, 0.04634528607130051), (12, 0.0423140823841095), (13, 0.04301172494888306), (14, 0.10755492001771927), (15, 0.10528909415006638), (16, 0.12610721588134766), (17, 0.105226069688797), (18, 0.11548751592636108), (19, 0.09860125929117203), (20, 0.08985016494989395), (21, 0.109283447265625), (22, 0.10421805083751678), (23, 0.11107815057039261), (24, 0.12677446007728577), (25, 0.14348818361759186), (26, 0.09171685576438904), (27, 0.035610221326351166)], Temporal MSE: [(0, 3.4129964205931174e-06), (1, 2.6403267838759348e-05), (2, 0.0005167841445654631), (3, 0.0002953198563773185), (4, 0.0007880820776335895), (5, 0.0010375302517786622), (6, 0.002322909887880087), (7, 0.0072181387804448605), (8, 0.004693426191806793), (9, 0.004820769652724266), (10, 0.009817460551857948), (11, 0.0082520991563797), (12, 0.006198963616043329), (13, 0.00824382621794939), (14, 0.025844065472483635), (15, 0.023077165707945824), (16, 0.02906443178653717), (17, 0.02679441124200821), (18, 0.02478397823870182), (19, 0.027072835713624954), (20, 0.05012089014053345), (21, 0.043510209769010544), (22, 0.0406271331012249), (23, 0.04704733565449715), (24, 0.02452198788523674), (25, 0.012475733645260334), (26, 0.01619626022875309), (27, 0.014487802051007748)]
+
+ 43%|████▎     | 13/30 [02:50<03:48, 13.46s/it]Time step 864, Spatial MSE: [(0, 2.1253097656881437e-05), (1, 0.0004509810241870582), (2, 0.0006706462590955198), (3, 0.0001910636929096654), (4, 5.8955003623850644e-05), (5, 8.922728738980368e-05), (6, 0.00013940807548351586), (7, 0.000306577974697575), (8, 0.0001272544905077666), (9, 0.00021790602477267385), (10, 0.0002997590054292232), (11, 0.0002331809519091621), (12, 0.00028539905906654894), (13, 0.0007960890070535243), (14, 0.0009494886035099626), (15, 0.001088725053705275), (16, 0.012812197208404541), (17, 0.0023349567782133818), (18, 0.00262839556671679), (19, 0.004491232801228762), (20, 0.0034521848428994417), (21, 0.01177708338946104), (22, 0.011136227287352085), (23, 0.005021822173148394), (24, 0.007523537613451481), (25, 0.013837520964443684), (26, 0.013823186047375202), (27, 0.013094465248286724)], Temporal MSE: [(0, 5.85074587888812e-07), (1, 2.79014830084634e-06), (2, 9.691505511000287e-06), (3, 1.0694362572394311e-05), (4, 1.0602625479805283e-05), (5, 1.5236108083627187e-05), (6, 1.641056951484643e-05), (7, 3.44781692547258e-05), (8, 2.130493703589309e-05), (9, 2.8190204830025323e-05), (10, 5.309881817083806e-05), (11, 5.093762592878193e-05), (12, 4.8210215027211234e-05), (13, 5.69220537727233e-05), (14, 0.00018976445426233113), (15, 0.00038246490294113755), (16, 0.0008302379865199327), (17, 0.0005874696071259677), (18, 0.00023357986356131732), (19, 0.0016056963941082358), (20, 0.00392140494659543), (21, 0.00332340644672513), (22, 0.0036402822006493807), (23, 0.0008880640380084515), (24, 0.0007989564328454435), (25, 0.0006922585307620466), (26, 0.0009500714368186891), (27, 0.0017826681723818183)]
+
+ 47%|████▋     | 14/30 [03:02<03:32, 13.27s/it]Time step 848, Spatial MSE: [(0, 0.00017113560170400888), (1, 0.0003777665551751852), (2, 0.000611045747064054), (3, 0.00026007829001173377), (4, 0.0003515677235554904), (5, 0.000936049094889313), (6, 0.0013610507594421506), (7, 0.00271411775611341), (8, 0.0014726497465744615), (9, 0.0022885138168931007), (10, 0.0075419810600578785), (11, 0.0051611303351819515), (12, 0.0041068256832659245), (13, 0.006730644032359123), (14, 0.018491236492991447), (15, 0.00918427761644125), (16, 0.02304554171860218), (17, 0.010018658824265003), (18, 0.010798759758472443), (19, 0.007549011614173651), (20, 0.01036981400102377), (21, 0.013733034022152424), (22, 0.010510589927434921), (23, 0.009913116693496704), (24, 0.02256060764193535), (25, 0.03866446763277054), (26, 0.019281556829810143), (27, 0.010074338875710964)], Temporal MSE: [(0, 2.579607325969846e-06), (1, 4.896719019598095e-06), (2, 1.850565240602009e-05), (3, 4.823458948521875e-05), (4, 6.788596510887146e-05), (5, 9.026257612276822e-05), (6, 0.00017863856919575483), (7, 0.0007853050483390689), (8, 0.0004914527526125312), (9, 0.0006033811951056123), (10, 0.0014630402438342571), (11, 0.0009929442312568426), (12, 0.0010876705637201667), (13, 0.0007078839698806405), (14, 0.003197432728484273), (15, 0.004735473543405533), (16, 0.005333151668310165), (17, 0.003768413793295622), (18, 0.0023877988569438457), (19, 0.003006890881806612), (20, 0.007924558594822884), (21, 0.004285102244466543), (22, 0.0033695686142891645), (23, 0.004176243674010038), (24, 0.003218691097572446), (25, 0.002489968203008175), (26, 0.003998082131147385), (27, 0.0031125491950660944)]
+
+ 50%|█████     | 15/30 [03:15<03:14, 12.97s/it]Time step 828, Spatial MSE: [(0, 2.0768125978065655e-05), (1, 0.0004684971063397825), (2, 0.0008269042591564357), (3, 0.00021666144311893731), (4, 4.878716936218552e-05), (5, 8.766615792410448e-05), (6, 8.712094859220088e-05), (7, 0.00030313554452732205), (8, 0.00013233757636044174), (9, 0.00021844296134077013), (10, 0.00021372217452153563), (11, 0.00019262993009760976), (12, 0.00031742561259306967), (13, 0.000685334496665746), (14, 0.0008486410370096564), (15, 0.0008501310367137194), (16, 0.012655056081712246), (17, 0.002193685621023178), (18, 0.002470605308189988), (19, 0.0022541095968335867), (20, 0.0016490845009684563), (21, 0.0033304484095424414), (22, 0.002262155059725046), (23, 0.0038744534831494093), (24, 0.006712699308991432), (25, 0.01417556218802929), (26, 0.01380118727684021), (27, 0.01735474355518818)], Temporal MSE: [(0, 5.410924472926126e-07), (1, 1.8230061868962366e-06), (2, 1.0023208233178593e-05), (3, 1.0213394489255734e-05), (4, 1.1067387276852969e-05), (5, 1.961219822987914e-05), (6, 1.731942393234931e-05), (7, 4.0963881474453956e-05), (8, 2.6990097467205487e-05), (9, 2.9983671993250027e-05), (10, 4.812782208318822e-05), (11, 4.5729466364718974e-05), (12, 6.264648254727945e-05), (13, 5.6013021094258875e-05), (14, 0.00018032619846053421), (15, 0.0002907713642343879), (16, 0.0008071522461250424), (17, 0.0004837704764213413), (18, 0.0002430728345643729), (19, 0.0010923242662101984), (20, 0.0024590056855231524), (21, 0.001119504333473742), (22, 0.0008426139829680324), (23, 0.0009170311386696994), (24, 0.0007523293024860322), (25, 0.0007326326449401677), (26, 0.0008318649488501251), (27, 0.0016341014998033643)]
+
+ 53%|█████▎    | 16/30 [03:26<02:54, 12.49s/it]Time step 808, Spatial MSE: [(0, 0.0001313002430833876), (1, 0.0007753408281132579), (2, 0.0009951923275366426), (3, 0.0010990797309204936), (4, 0.0013706553727388382), (5, 0.0023668543435633183), (6, 0.005181950516998768), (7, 0.011468369513750076), (8, 0.012089247815310955), (9, 0.019939806312322617), (10, 0.02391577884554863), (11, 0.0244075208902359), (12, 0.024241123348474503), (13, 0.026893485337495804), (14, 0.06506456434726715), (15, 0.05987035855650902), (16, 0.0816321149468422), (17, 0.07080021500587463), (18, 0.07576829940080643), (19, 0.0615769661962986), (20, 0.0616426095366478), (21, 0.07588231563568115), (22, 0.0723707526922226), (23, 0.08498157560825348), (24, 0.10147371143102646), (25, 0.12545621395111084), (26, 0.10565073788166046), (27, 0.025943078100681305)], Temporal MSE: [(0, 3.4809363569365814e-06), (1, 2.023650085902773e-05), (2, 0.00019268039613962173), (3, 0.00015279327635653317), (4, 0.0006598342442885041), (5, 0.0006747707375325263), (6, 0.0017167661571875215), (7, 0.004244884941726923), (8, 0.0027651789132505655), (9, 0.002558621810749173), (10, 0.004168764688074589), (11, 0.005007236264646053), (12, 0.004036130383610725), (13, 0.0059482925571501255), (14, 0.013143647462129593), (15, 0.014659900218248367), (16, 0.01843615621328354), (17, 0.019507890567183495), (18, 0.018453117460012436), (19, 0.02200586162507534), (20, 0.044608283787965775), (21, 0.03203544765710831), (22, 0.03188671916723251), (23, 0.033351585268974304), (24, 0.019846027716994286), (25, 0.010180240496993065), (26, 0.012179628014564514), (27, 0.015378889627754688)]
+
+ 57%|█████▋    | 17/30 [03:39<02:42, 12.48s/it]Time step 788, Spatial MSE: [(0, 1.9996856281068176e-05), (1, 0.0004174018686171621), (2, 0.0009710431331768632), (3, 0.0002426371502224356), (4, 5.1379796786932275e-05), (5, 8.192604582291096e-05), (6, 8.28506745165214e-05), (7, 0.00048047243035398424), (8, 0.00016842529294081032), (9, 0.00018134461424779147), (10, 0.00019793979299720377), (11, 0.00021615333389490843), (12, 0.0002790499711409211), (13, 0.0006541052716784179), (14, 0.0007757642888464034), (15, 0.0008155793184414506), (16, 0.00962698645889759), (17, 0.0017816451145336032), (18, 0.008240523748099804), (19, 0.002674764720723033), (20, 0.0016076762694865465), (21, 0.002734120236709714), (22, 0.002384481020271778), (23, 0.0036700027994811535), (24, 0.006473197136074305), (25, 0.016536463052034378), (26, 0.014746373519301414), (27, 0.010785500518977642)], Temporal MSE: [(0, 6.095596631894296e-07), (1, 1.7775822698240518e-06), (2, 1.0030090379586909e-05), (3, 9.421170034329407e-06), (4, 1.1032554539269768e-05), (5, 1.8726985217654146e-05), (6, 2.067418427031953e-05), (7, 3.922918040188961e-05), (8, 2.487794336047955e-05), (9, 3.689961522468366e-05), (10, 5.29991157236509e-05), (11, 4.8925703595159575e-05), (12, 4.40092699136585e-05), (13, 6.0844373365398496e-05), (14, 0.0001686570030869916), (15, 0.00024069730716291815), (16, 0.0006050564697943628), (17, 0.00034417022834531963), (18, 0.0016561088850721717), (19, 0.0021060998551547527), (20, 0.0044553023763000965), (21, 0.0015133452834561467), (22, 0.0007238549296744168), (23, 0.0007844484644010663), (24, 0.00072258006548509), (25, 0.0009173303842544556), (26, 0.0008268046658486128), (27, 0.001270268694497645)]
+
+ 60%|██████    | 18/30 [03:49<02:23, 11.95s/it]Time step 764, Spatial MSE: [(0, 0.0001739939907565713), (1, 0.0005653250264003873), (2, 0.0006500092567875981), (3, 0.00024405041767749935), (4, 0.0002481791016180068), (5, 0.0006369146285578609), (6, 0.0010216127848252654), (7, 0.0020966825541108847), (8, 0.0012228336418047547), (9, 0.0019496919121593237), (10, 0.006415071431547403), (11, 0.002732878550887108), (12, 0.002936282427981496), (13, 0.005146042909473181), (14, 0.014319978654384613), (15, 0.009339949116110802), (16, 0.01947435550391674), (17, 0.010968352667987347), (18, 0.00944012776017189), (19, 0.007470014039427042), (20, 0.006638163235038519), (21, 0.013845189474523067), (22, 0.009651326574385166), (23, 0.015227344818413258), (24, 0.025558695197105408), (25, 0.0517430417239666), (26, 0.025806942954659462), (27, 0.010244214907288551)], Temporal MSE: [(0, 2.8239571747690206e-06), (1, 5.298953510646243e-06), (2, 1.8808632376021706e-05), (3, 3.352563362568617e-05), (4, 5.018154843128286e-05), (5, 4.992877438780852e-05), (6, 0.00013047349057160318), (7, 0.00044853397412225604), (8, 0.00036019456456415355), (9, 0.0006847358890809119), (10, 0.0012587911915034056), (11, 0.0008726285886950791), (12, 0.0010623912094160914), (13, 0.0008349120034836233), (14, 0.0035863330122083426), (15, 0.004406739957630634), (16, 0.004771044012159109), (17, 0.0035079100634902716), (18, 0.0021481977310031652), (19, 0.0029709753580391407), (20, 0.005583662539720535), (21, 0.0050081596709787846), (22, 0.0037304596044123173), (23, 0.004166325554251671), (24, 0.006380094680935144), (25, 0.0017733604181557894), (26, 0.004458120558410883), (27, 0.003936386667191982)]
+
+ 63%|██████▎   | 19/30 [04:01<02:09, 11.82s/it]Time step 736, Spatial MSE: [(0, 3.051040584978182e-05), (1, 0.0006828628829680383), (2, 0.0013059234479442239), (3, 0.0003826994798146188), (4, 7.307783380383626e-05), (5, 0.00011109431216027588), (6, 0.00012714217882603407), (7, 0.0003822876897174865), (8, 0.00015498620632570237), (9, 0.0002932060742750764), (10, 0.00023028776922728866), (11, 0.00025902711786329746), (12, 0.0003665829135570675), (13, 0.0013445507502183318), (14, 0.001660036388784647), (15, 0.0015610693953931332), (16, 0.027822380885481834), (17, 0.00535346707329154), (18, 0.005048776976764202), (19, 0.002935066120699048), (20, 0.0024998860899358988), (21, 0.005627908743917942), (22, 0.003132651327177882), (23, 0.005776533391326666), (24, 0.00880763866007328), (25, 0.02313026413321495), (26, 0.021273111924529076), (27, 0.011191981844604015)], Temporal MSE: [(0, 8.631332661934721e-07), (1, 2.4697558274056064e-06), (2, 1.6058780602179468e-05), (3, 1.2987357877136674e-05), (4, 1.691110446699895e-05), (5, 2.208548175985925e-05), (6, 3.062097675865516e-05), (7, 5.207387948757969e-05), (8, 3.5937238862970844e-05), (9, 4.962996536050923e-05), (10, 6.711410969728604e-05), (11, 6.138189928606153e-05), (12, 6.093388583394699e-05), (13, 9.568176028551534e-05), (14, 0.0002765251265373081), (15, 0.0006256261840462685), (16, 0.0011422355892136693), (17, 0.0010410486720502377), (18, 0.000499480520375073), (19, 0.0015471010701730847), (20, 0.004914985969662666), (21, 0.0020136258099228144), (22, 0.001007667277008295), (23, 0.0010949020506814122), (24, 0.0010681032435968518), (25, 0.0015015698736533523), (26, 0.001200889004394412), (27, 0.002031149575486779)]
+
+ 67%|██████▋   | 20/30 [04:12<01:55, 11.57s/it]Time step 708, Spatial MSE: [(0, 8.947241440182552e-05), (1, 0.0014442092506214976), (2, 0.0016536642797291279), (3, 0.0012749363668262959), (4, 0.0013861419865861535), (5, 0.003656977554783225), (6, 0.004278410691767931), (7, 0.008878720924258232), (8, 0.010859200730919838), (9, 0.01936345361173153), (10, 0.021565988659858704), (11, 0.020473724231123924), (12, 0.02212996408343315), (13, 0.02470269799232483), (14, 0.05475478619337082), (15, 0.08255961537361145), (16, 0.09678848832845688), (17, 0.08712811768054962), (18, 0.09137756377458572), (19, 0.08130092173814774), (20, 0.08882882446050644), (21, 0.12055446952581406), (22, 0.10286549478769302), (23, 0.1133175939321518), (24, 0.1256517767906189), (25, 0.17337289452552795), (26, 0.15135088562965393), (27, 0.025682909414172173)], Temporal MSE: [(0, 5.519706064660568e-06), (1, 2.0226350898155943e-05), (2, 0.00011726695811375976), (3, 0.00019328310736455023), (4, 0.0007062568911351264), (5, 0.0010233721695840359), (6, 0.0016671051271259785), (7, 0.0034161247313022614), (8, 0.0022517696488648653), (9, 0.002702794736251235), (10, 0.0043049887754023075), (11, 0.005053940694779158), (12, 0.004272962920367718), (13, 0.005766340997070074), (14, 0.017053278163075447), (15, 0.018507450819015503), (16, 0.023474078625440598), (17, 0.025985684245824814), (18, 0.02627132087945938), (19, 0.031505439430475235), (20, 0.07690846920013428), (21, 0.049760352820158005), (22, 0.0385311059653759), (23, 0.0330667644739151), (24, 0.027942325919866562), (25, 0.017935816198587418), (26, 0.023853113874793053), (27, 0.02849547564983368)]
+
+ 70%|███████   | 21/30 [04:24<01:47, 11.91s/it]Time step 676, Spatial MSE: [(0, 3.398603439563885e-05), (1, 0.0009418388362973928), (2, 0.0016788395587354898), (3, 0.0005214047851040959), (4, 9.509054507361725e-05), (5, 0.00014514852955471724), (6, 0.0001562717225169763), (7, 0.00039523010491393507), (8, 0.00018550772801972926), (9, 0.00022087340767029673), (10, 0.0002591387019492686), (11, 0.00031329988269135356), (12, 0.0003644375246949494), (13, 0.0015917276032269), (14, 0.001987458672374487), (15, 0.0018048147903755307), (16, 0.029330555349588394), (17, 0.005563876125961542), (18, 0.004907253198325634), (19, 0.002416034694761038), (20, 0.001782983192242682), (21, 0.0038816812448203564), (22, 0.0032934790942817926), (23, 0.006269518751651049), (24, 0.00858910009264946), (25, 0.030752815306186676), (26, 0.026852816343307495), (27, 0.013719729147851467)], Temporal MSE: [(0, 1.1196406148883398e-06), (1, 3.2727948564570397e-06), (2, 1.7604030290385708e-05), (3, 1.6166410205187276e-05), (4, 2.1656011085724458e-05), (5, 3.349818507558666e-05), (6, 4.462359720491804e-05), (7, 7.05463025951758e-05), (8, 4.5111119106877595e-05), (9, 6.861510337330401e-05), (10, 9.780010441318154e-05), (11, 7.702975563006476e-05), (12, 6.435053364839405e-05), (13, 0.0001129509910242632), (14, 0.00040910992538556457), (15, 0.0005931564955972135), (16, 0.0013831114629283547), (17, 0.0009678892092779279), (18, 0.00038275509723462164), (19, 0.0014548269100487232), (20, 0.0037984633818268776), (21, 0.001372152823023498), (22, 0.0010510471183806658), (23, 0.00130612647626549), (24, 0.0012207975378260016), (25, 0.002398915123194456), (26, 0.0013950763968750834), (27, 0.0027708872221410275)]
+
+ 73%|███████▎  | 22/30 [04:35<01:32, 11.58s/it]Time step 636, Spatial MSE: [(0, 0.00010581436072243378), (1, 0.0010746546322479844), (2, 0.001348278601653874), (3, 0.00042596831917762756), (4, 0.0003612462605815381), (5, 0.0008891221368685365), (6, 0.001278855255804956), (7, 0.004088866990059614), (8, 0.002318138489499688), (9, 0.002742898417636752), (10, 0.012293591164052486), (11, 0.00587344029918313), (12, 0.0048633781261742115), (13, 0.008814406581223011), (14, 0.015482199378311634), (15, 0.014058682136237621), (16, 0.03841095045208931), (17, 0.016362883150577545), (18, 0.017609847709536552), (19, 0.014142442494630814), (20, 0.012421335093677044), (21, 0.02545698918402195), (22, 0.015502517111599445), (23, 0.02643137238919735), (24, 0.05164335295557976), (25, 0.13007506728172302), (26, 0.06535496562719345), (27, 0.03176114335656166)], Temporal MSE: [(0, 3.030781726920395e-06), (1, 7.433970040437998e-06), (2, 2.840403249138035e-05), (3, 5.5830587371019647e-05), (4, 0.0001536029885755852), (5, 0.00010337927233194932), (6, 0.00042081461288034916), (7, 0.0009984427597373724), (8, 0.0009532331023365259), (9, 0.001290150685235858), (10, 0.0037916461005806923), (11, 0.0018536103889346123), (12, 0.002190419239923358), (13, 0.001283389632590115), (14, 0.004317632410675287), (15, 0.006815032102167606), (16, 0.00803695060312748), (17, 0.006663689389824867), (18, 0.00494504114612937), (19, 0.00581274414435029), (20, 0.006610107142478228), (21, 0.009485944174230099), (22, 0.0054342192597687244), (23, 0.005883398465812206), (24, 0.007714760489761829), (25, 0.0042768181301653385), (26, 0.008761639706790447), (27, 0.007847518660128117)]
+
+ 77%|███████▋  | 23/30 [04:47<01:21, 11.58s/it]Time step 596, Spatial MSE: [(0, 4.532820821623318e-05), (1, 0.0012697149068117142), (2, 0.0026726489886641502), (3, 0.0008644910412840545), (4, 0.00014471853501163423), (5, 0.0002451776817906648), (6, 0.00025704895961098373), (7, 0.0004658098623622209), (8, 0.0003028618230018765), (9, 0.0003328206657897681), (10, 0.00040391794755123556), (11, 0.00043149286648258567), (12, 0.0005331034772098064), (13, 0.002368920249864459), (14, 0.002873383229598403), (15, 0.0022860837634652853), (16, 0.034925639629364014), (17, 0.0066292607225477695), (18, 0.006255300249904394), (19, 0.005422188900411129), (20, 0.005002870224416256), (21, 0.015467767603695393), (22, 0.013875368051230907), (23, 0.014592807739973068), (24, 0.015152509324252605), (25, 0.053807228803634644), (26, 0.04890967905521393), (27, 0.02486397698521614)], Temporal MSE: [(0, 2.4516411940567195e-06), (1, 5.4632159844913986e-06), (2, 2.804043651849497e-05), (3, 2.4359036615351215e-05), (4, 3.8862868677824736e-05), (5, 5.1212977268733084e-05), (6, 7.978030043886974e-05), (7, 0.00011493574129417539), (8, 8.035930659389123e-05), (9, 0.00011468896991573274), (10, 0.00013909385597798973), (11, 0.00011380104842828587), (12, 0.00010954460594803095), (13, 0.0002001309476327151), (14, 0.0004150574095547199), (15, 0.0007404027273878455), (16, 0.001805810141377151), (17, 0.0014183998573571444), (18, 0.0007339560543186963), (19, 0.0026246372144669294), (20, 0.009386817924678326), (21, 0.0048860423266887665), (22, 0.0029657927807420492), (23, 0.002149159088730812), (24, 0.002111193025484681), (25, 0.004166485741734505), (26, 0.002345779910683632), (27, 0.0062392232939600945)]
+
+ 80%|████████  | 24/30 [04:58<01:08, 11.41s/it]Time step 548, Spatial MSE: [(0, 0.00013728807971347123), (1, 0.0034570051357150078), (2, 0.004381751641631126), (3, 0.0021973643451929092), (4, 0.0022276174277067184), (5, 0.006139347329735756), (6, 0.006999131757766008), (7, 0.012428359128534794), (8, 0.014475786127150059), (9, 0.019120879471302032), (10, 0.026021679863333702), (11, 0.021926041692495346), (12, 0.025191225111484528), (13, 0.02840343676507473), (14, 0.05875236541032791), (15, 0.08263640850782394), (16, 0.10584316402673721), (17, 0.08259575068950653), (18, 0.10062895715236664), (19, 0.07674916833639145), (20, 0.08904480934143066), (21, 0.13438646495342255), (22, 0.1357794553041458), (23, 0.15761342644691467), (24, 0.21509681642055511), (25, 0.4206281900405884), (26, 0.3684559166431427), (27, 0.09144807606935501)], Temporal MSE: [(0, 1.2098555089323781e-05), (1, 3.545716390362941e-05), (2, 0.00021494664542842656), (3, 0.0003508132940623909), (4, 0.0009529016679152846), (5, 0.0017854513134807348), (6, 0.0028814144898205996), (7, 0.004224872216582298), (8, 0.003218242432922125), (9, 0.004630588926374912), (10, 0.006141536869108677), (11, 0.006099562626332045), (12, 0.006503389682620764), (13, 0.007797462400048971), (14, 0.019469639286398888), (15, 0.02213555946946144), (16, 0.027572467923164368), (17, 0.02839605323970318), (18, 0.027166126295924187), (19, 0.032851316034793854), (20, 0.08352890610694885), (21, 0.0707312673330307), (22, 0.04326946660876274), (23, 0.04913196340203285), (24, 0.07363944500684738), (25, 0.038216665387153625), (26, 0.044808026403188705), (27, 0.07945781946182251)]
+
+ 83%|████████▎ | 25/30 [05:12<01:00, 12.13s/it]Time step 492, Spatial MSE: [(0, 8.113567309919745e-05), (1, 0.00255853240378201), (2, 0.0042618499137461185), (3, 0.001644059899263084), (4, 0.00025920916232280433), (5, 0.0005421401001513004), (6, 0.0005596745177172124), (7, 0.001248164800927043), (8, 0.000731190200895071), (9, 0.0005315342568792403), (10, 0.0006668765563517809), (11, 0.0007810075767338276), (12, 0.0012203881051391363), (13, 0.002941736252978444), (14, 0.0035038066562265158), (15, 0.0025722747668623924), (16, 0.04408378154039383), (17, 0.007283768150955439), (18, 0.018267951905727386), (19, 0.005971826612949371), (20, 0.007549780886620283), (21, 0.02306612953543663), (22, 0.012064807116985321), (23, 0.02166447788476944), (24, 0.029864491894841194), (25, 0.1139032393693924), (26, 0.10925944149494171), (27, 0.044018395245075226)], Temporal MSE: [(0, 3.94315020457725e-06), (1, 9.63428283284884e-06), (2, 5.371234510676004e-05), (3, 4.9649774155113846e-05), (4, 8.526949386578053e-05), (5, 0.00012013500963803381), (6, 0.0002013923804042861), (7, 0.0003145684895571321), (8, 0.00016479214536957443), (9, 0.0002406966086709872), (10, 0.00028097149333916605), (11, 0.00021083789761178195), (12, 0.00020930419850628823), (13, 0.00038883567322045565), (14, 0.000633863965049386), (15, 0.0009607013780623674), (16, 0.0021348653826862574), (17, 0.0018638077890500426), (18, 0.002831835299730301), (19, 0.005988954566419125), (20, 0.02087218686938286), (21, 0.00821774359792471), (22, 0.0029924476984888315), (23, 0.0036574460100382566), (24, 0.004947854671627283), (25, 0.009124444797635078), (26, 0.005246873013675213), (27, 0.01478618010878563)]
+
+ 87%|████████▋ | 26/30 [05:23<00:47, 11.82s/it]Time step 426, Spatial MSE: [(0, 0.0003876523405779153), (1, 0.005344975274056196), (2, 0.005945901852101088), (3, 0.002249107463285327), (4, 0.0017296592704951763), (5, 0.0030563983600586653), (6, 0.005927606951445341), (7, 0.009417667053639889), (8, 0.009751967154443264), (9, 0.011639303527772427), (10, 0.019708232954144478), (11, 0.014637772925198078), (12, 0.01706935092806816), (13, 0.017143376171588898), (14, 0.035237010568380356), (15, 0.04483021795749664), (16, 0.07286158204078674), (17, 0.045884765684604645), (18, 0.05783654376864433), (19, 0.05086248368024826), (20, 0.058750223368406296), (21, 0.1085190623998642), (22, 0.10265008360147476), (23, 0.1675163358449936), (24, 0.2711549401283264), (25, 0.6334402561187744), (26, 0.5022031664848328), (27, 0.16474348306655884)], Temporal MSE: [(0, 1.6588030121056363e-05), (1, 3.972842387156561e-05), (2, 0.000220050715142861), (3, 0.00033533255918882787), (4, 0.0006761503173038363), (5, 0.0009588069515302777), (6, 0.0028602806851267815), (7, 0.003618288552388549), (8, 0.0019045354565605521), (9, 0.0036367871798574924), (10, 0.004268106073141098), (11, 0.0036924616433680058), (12, 0.0034717863891273737), (13, 0.0045830486342310905), (14, 0.01278496254235506), (15, 0.014777989126741886), (16, 0.01905887946486473), (17, 0.01641751639544964), (18, 0.016157489269971848), (19, 0.021492280066013336), (20, 0.03836370259523392), (21, 0.057860132306814194), (22, 0.03508758172392845), (23, 0.03660421445965767), (24, 0.08140912652015686), (25, 0.02951396442949772), (26, 0.07016880065202713), (27, 0.10540783405303955)]
+
+ 90%|█████████ | 27/30 [05:35<00:35, 11.86s/it]Time step 350, Spatial MSE: [(0, 0.00019871034601237625), (1, 0.0070135025307536125), (2, 0.00771060585975647), (3, 0.002313601318746805), (4, 0.0009564069914631546), (5, 0.0022267932072281837), (6, 0.0032613237854093313), (7, 0.00605567405000329), (8, 0.0052472250536084175), (9, 0.005127204582095146), (10, 0.010227181017398834), (11, 0.007311658468097448), (12, 0.009771804325282574), (13, 0.008744411170482635), (14, 0.01742197759449482), (15, 0.02076227404177189), (16, 0.025425804778933525), (17, 0.02074321173131466), (18, 0.024773839861154556), (19, 0.03627246618270874), (20, 0.04234285280108452), (21, 0.08232623338699341), (22, 0.099664106965065), (23, 0.13940010964870453), (24, 0.2141021490097046), (25, 0.4569512605667114), (26, 0.4134604036808014), (27, 0.22672314941883087)], Temporal MSE: [(0, 1.3914655028202105e-05), (1, 4.3324620492057875e-05), (2, 0.00017592091171536595), (3, 0.0002820659428834915), (4, 0.0005348958075046539), (5, 0.0007085184333845973), (6, 0.002215431071817875), (7, 0.0025531838182359934), (8, 0.0012827381724491715), (9, 0.002573395147919655), (10, 0.0029354251455515623), (11, 0.002121780999004841), (12, 0.0021294825710356236), (13, 0.002701766090467572), (14, 0.006806845776736736), (15, 0.006269821431487799), (16, 0.008632400073111057), (17, 0.00857488252222538), (18, 0.007867248728871346), (19, 0.01267306413501501), (20, 0.040209993720054626), (21, 0.05439985916018486), (22, 0.024917403236031532), (23, 0.02948216162621975), (24, 0.08077776432037354), (25, 0.024985404685139656), (26, 0.0590444914996624), (27, 0.11138077080249786)]
+
+ 93%|█████████▎| 28/30 [05:48<00:24, 12.23s/it]Time step 258, Spatial MSE: [(0, 0.00039188945083878934), (1, 0.011899636127054691), (2, 0.012392991222441196), (3, 0.0043101548217237), (4, 0.0017210636287927628), (5, 0.004150171298533678), (6, 0.0064783296547830105), (7, 0.01449836790561676), (8, 0.01061449944972992), (9, 0.009617363102734089), (10, 0.02018958143889904), (11, 0.013235235586762428), (12, 0.01923728920519352), (13, 0.01600763574242592), (14, 0.032172735780477524), (15, 0.03868183493614197), (16, 0.04703903943300247), (17, 0.03551596775650978), (18, 0.050682272762060165), (19, 0.054921701550483704), (20, 0.07226093858480453), (21, 0.1614818572998047), (22, 0.16675850749015808), (23, 0.2735297679901123), (24, 0.4537988305091858), (25, 1.0557820796966553), (26, 0.8891826868057251), (27, 0.4954376816749573)], Temporal MSE: [(0, 3.3296011679340154e-05), (1, 8.676163270138204e-05), (2, 0.0003381789429113269), (3, 0.0005134356906637549), (4, 0.001046127756126225), (5, 0.0015677555929869413), (6, 0.004459516145288944), (7, 0.005949531216174364), (8, 0.0027791571337729692), (9, 0.004923786967992783), (10, 0.005803010892122984), (11, 0.004456839989870787), (12, 0.00411896500736475), (13, 0.004752294160425663), (14, 0.011521080508828163), (15, 0.011594102717936039), (16, 0.01372089609503746), (17, 0.013760014437139034), (18, 0.013914760202169418), (19, 0.024110592901706696), (20, 0.06254348158836365), (21, 0.10929884761571884), (22, 0.04981252923607826), (23, 0.05866857245564461), (24, 0.1814156025648117), (25, 0.06894156336784363), (26, 0.14540983736515045), (27, 0.23570774495601654)]
+
+ 97%|█████████▋| 29/30 [06:01<00:12, 12.56s/it]Time step 143, Spatial MSE: [(0, 0.0008617169805802405), (1, 0.023438412696123123), (2, 0.023562854155898094), (3, 0.00886488240212202), (4, 0.003879025112837553), (5, 0.009579367004334927), (6, 0.014395623467862606), (7, 0.03449651971459389), (8, 0.023026417940855026), (9, 0.020763475447893143), (10, 0.039708688855171204), (11, 0.031206533312797546), (12, 0.043770089745521545), (13, 0.03170138597488403), (14, 0.07529879361391068), (15, 0.07608652859926224), (16, 0.0806422084569931), (17, 0.06656362861394882), (18, 0.09458597004413605), (19, 0.11650088429450989), (20, 0.17913199961185455), (21, 0.3029972016811371), (22, 0.3384348154067993), (23, 0.5880626440048218), (24, 1.2596023082733154), (25, 2.668867588043213), (26, 2.0173895359039307), (27, 1.1363914012908936)], Temporal MSE: [(0, 8.939651161199436e-05), (1, 0.00020733063865918666), (2, 0.0007518675993196666), (3, 0.0010208627209067345), (4, 0.0027272128500044346), (5, 0.0037651262246072292), (6, 0.011691391468048096), (7, 0.012575628235936165), (8, 0.005996612831950188), (9, 0.007503945380449295), (10, 0.010732823051512241), (11, 0.008282795548439026), (12, 0.0072469995357096195), (13, 0.0079015688970685), (14, 0.02048412710428238), (15, 0.022296415641903877), (16, 0.025562744587659836), (17, 0.02889304794371128), (18, 0.0273185633122921), (19, 0.04908120259642601), (20, 0.1802283525466919), (21, 0.2679189145565033), (22, 0.08796325325965881), (23, 0.13426680862903595), (24, 0.43247997760772705), (25, 0.290322870016098), (26, 0.34836429357528687), (27, 0.4321860373020172)]
+
+100%|██████████| 30/30 [06:14<00:00, 12.57s/it]
+100%|██████████| 30/30 [06:14<00:00, 12.48s/it]
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:456: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv2d(input, weight, bias, self.stride,
+[2024-10-01 14:58:51] Prompt: sunset over the sea. aesthetic score: 6.5.
+Saved to ./samples/opensora/sample_pab/Sunset over the sea..mp4
+Saved video to ./samples/opensora/sample_pab/Sunset over the sea..mp4
+Saved spatial MLP MSE outputs to ./samples/opensora/sample_pab/Sunset over the sea._spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora/sample_pab/Sunset over the sea._temporal_mlp_mse.pt
+timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:0'), tensor([828.8770], device='cuda:0'), tensor([809.0977], device='cuda:0'), tensor([787.4170], device='cuda:0'), tensor([763.5468], device='cuda:0'), tensor([737.1379], device='cuda:0'), tensor([707.7626], device='cuda:0'), tensor([674.8911], device='cuda:0'), tensor([637.8601], device='cuda:0'), tensor([595.8265], device='cuda:0'), tensor([547.7032], device='cuda:0'), tensor([492.0635], device='cuda:0'), tensor([426.9973], device='cuda:0'), tensor([349.8871], device='cuda:0'), tensor([257.0481], device='cuda:0'), tensor([143.1210], device='cuda:0')]
+
+  0%|          | 0/30 [00:00<?, ?it/s]
+  3%|▎         | 1/30 [00:03<01:48,  3.74s/it]Time step 992, Spatial MSE: [(0, 0.000837838975712657), (1, 0.001510637579485774), (2, 0.029183808714151382), (3, 0.019378023222088814), (4, 0.025165032595396042), (5, 0.04448821395635605), (6, 0.03928728029131889), (7, 0.06091400980949402), (8, 0.08677217364311218), (9, 0.1571597307920456), (10, 0.5048191547393799), (11, 0.36188173294067383), (12, 0.3810146152973175), (13, 0.36479872465133667), (14, 0.5362206697463989), (15, 0.5748317241668701), (16, 1.3480043411254883), (17, 0.7403931021690369), (18, 0.446037620306015), (19, 0.615665853023529), (20, 1.1111985445022583), (21, 0.5585929751396179), (22, 0.6665892004966736), (23, 1.2828869819641113), (24, 6.450198650360107), (25, 6.220454692840576), (26, 4.818821907043457), (27, 0.5050948262214661)], Temporal MSE: [(0, 4.3012834794353694e-05), (1, 0.00041159274405799806), (2, 0.0014190191868692636), (3, 0.002663613762706518), (4, 0.0024933223612606525), (5, 0.007840985432267189), (6, 0.004387391731142998), (7, 0.009008470922708511), (8, 0.007996853440999985), (9, 0.012767650187015533), (10, 0.052772730588912964), (11, 0.05236710608005524), (12, 0.054183319211006165), (13, 0.026453373953700066), (14, 0.11455726623535156), (15, 0.11714506149291992), (16, 0.17676419019699097), (17, 0.07270685583353043), (18, 0.08040221780538559), (19, 0.10212740302085876), (20, 0.0755871906876564), (21, 0.15760034322738647), (22, 0.6149211525917053), (23, 0.4770071804523468), (24, 0.2676725387573242), (25, 0.1726607233285904), (26, 0.43822842836380005), (27, 0.6378535032272339)]
+
+  7%|▋         | 2/30 [00:11<02:50,  6.10s/it]Time step 984, Spatial MSE: [(0, 0.00037538958713412285), (1, 0.0016428956296294928), (2, 0.006344865076243877), (3, 0.0063879480585455894), (4, 0.005665645003318787), (5, 0.00830729492008686), (6, 0.016525978222489357), (7, 0.036694083362817764), (8, 0.051777876913547516), (9, 0.10155420005321503), (10, 0.21351079642772675), (11, 0.1891539990901947), (12, 0.21430256962776184), (13, 0.2186707705259323), (14, 0.34092977643013), (15, 0.38679078221321106), (16, 0.49472761154174805), (17, 0.35238903760910034), (18, 0.30611616373062134), (19, 0.28104180097579956), (20, 0.48033034801483154), (21, 0.3551613390445709), (22, 0.2918616533279419), (23, 0.5098559260368347), (24, 1.1236387491226196), (25, 1.641386866569519), (26, 1.436800479888916), (27, 0.21242254972457886)], Temporal MSE: [(0, 7.66164612286957e-06), (1, 0.0001583183038746938), (2, 0.00030560677987523377), (3, 0.0005590917426161468), (4, 0.000645261665340513), (5, 0.0015664169332012534), (6, 0.0016839354066178203), (7, 0.007925319485366344), (8, 0.004280831664800644), (9, 0.007576565258204937), (10, 0.014207680709660053), (11, 0.020902467891573906), (12, 0.019802836701273918), (13, 0.015281747095286846), (14, 0.05549582839012146), (15, 0.057708222419023514), (16, 0.08543431758880615), (17, 0.054198093712329865), (18, 0.045428112149238586), (19, 0.07059189677238464), (20, 0.11753951758146286), (21, 0.1455405354499817), (22, 0.27381807565689087), (23, 0.22756975889205933), (24, 0.11897614598274231), (25, 0.06674908846616745), (26, 0.16077357530593872), (27, 0.11396657675504684)]
+
+ 10%|█         | 3/30 [00:20<03:19,  7.38s/it]Time step 976, Spatial MSE: [(0, 0.0001788770459825173), (1, 0.0008346469257958233), (2, 0.0024950176011770964), (3, 0.0022078731562942266), (4, 0.0021507160272449255), (5, 0.002679667202755809), (6, 0.009602970443665981), (7, 0.013490322977304459), (8, 0.017745375633239746), (9, 0.03568267822265625), (10, 0.0668494924902916), (11, 0.06859330087900162), (12, 0.087290458381176), (13, 0.09221071004867554), (14, 0.1683785617351532), (15, 0.17788998782634735), (16, 0.2795793116092682), (17, 0.22480306029319763), (18, 0.23731964826583862), (19, 0.23097045719623566), (20, 0.359264999628067), (21, 0.3042568862438202), (22, 0.30636104941368103), (23, 0.2769472002983093), (24, 0.6744514107704163), (25, 0.6441519856452942), (26, 0.40043905377388), (27, 0.09116052091121674)], Temporal MSE: [(0, 3.72285398952954e-06), (1, 5.483625500346534e-05), (2, 0.00012674974277615547), (3, 0.00023030434385873377), (4, 0.00024157861480489373), (5, 0.0010271604405716062), (6, 0.0005493321223184466), (7, 0.0029915301129221916), (8, 0.0018160607432946563), (9, 0.0024869386106729507), (10, 0.005538645200431347), (11, 0.006942086853086948), (12, 0.007169945631176233), (13, 0.007141494192183018), (14, 0.021153198555111885), (15, 0.028406480327248573), (16, 0.04551537707448006), (17, 0.03414582833647728), (18, 0.03265761584043503), (19, 0.04056418314576149), (20, 0.05325201153755188), (21, 0.06585172563791275), (22, 0.11267738044261932), (23, 0.11020150035619736), (24, 0.051136430352926254), (25, 0.0281146802008152), (26, 0.04532521218061447), (27, 0.03531644493341446)]
+
+ 13%|█▎        | 4/30 [00:29<03:33,  8.21s/it]Time step 968, Spatial MSE: [(0, 0.00010011097765527666), (1, 0.00046430478687398136), (2, 0.0013764870818704367), (3, 0.0008226757636293769), (4, 0.0011467640288174152), (5, 0.001741341664455831), (6, 0.005068378057330847), (7, 0.009686004370450974), (8, 0.014242413453757763), (9, 0.029011981561779976), (10, 0.05185003951191902), (11, 0.06177489086985588), (12, 0.06984422355890274), (13, 0.07420051842927933), (14, 0.17649830877780914), (15, 0.1788836121559143), (16, 0.20387063920497894), (17, 0.1769854575395584), (18, 0.18082295358181), (19, 0.1610051840543747), (20, 0.14167727530002594), (21, 0.14855562150478363), (22, 0.15628355741500854), (23, 0.1369013488292694), (24, 0.22751939296722412), (25, 0.20404702425003052), (26, 0.1510751098394394), (27, 0.038292378187179565)], Temporal MSE: [(0, 1.8890193587139947e-06), (1, 2.240601315861568e-05), (2, 7.519002247136086e-05), (3, 0.00010669534822227433), (4, 0.00016338306886609644), (5, 0.0005347802652977407), (6, 0.00040951446862891316), (7, 0.001234904513694346), (8, 0.0012868769699707627), (9, 0.0018576637376099825), (10, 0.004231521859765053), (11, 0.005602280609309673), (12, 0.0057624694891273975), (13, 0.005762485787272453), (14, 0.02081027440726757), (15, 0.03027157485485077), (16, 0.03551826626062393), (17, 0.031642768532037735), (18, 0.029123198240995407), (19, 0.03461718186736107), (20, 0.04816820099949837), (21, 0.0565691739320755), (22, 0.06441519409418106), (23, 0.06133052706718445), (24, 0.03080943413078785), (25, 0.010968233458697796), (26, 0.020284509286284447), (27, 0.012999339960515499)]
+
+ 17%|█▋        | 5/30 [00:39<03:37,  8.68s/it]Time step 960, Spatial MSE: [(0, 6.56252886983566e-05), (1, 0.000564633752219379), (2, 0.0011872781906276941), (3, 0.0007838023593649268), (4, 0.0008738146279938519), (5, 0.0014246146893128753), (6, 0.004495142959058285), (7, 0.008289378136396408), (8, 0.014443283900618553), (9, 0.025643333792686462), (10, 0.049059487879276276), (11, 0.056355927139520645), (12, 0.06132938340306282), (13, 0.06800087541341782), (14, 0.17437689006328583), (15, 0.246259868144989), (16, 0.2790673077106476), (17, 0.2366856038570404), (18, 0.2658005654811859), (19, 0.22037054598331451), (20, 0.2237914502620697), (21, 0.2430441975593567), (22, 0.30230581760406494), (23, 0.2727840840816498), (24, 0.3233460783958435), (25, 0.3092189431190491), (26, 0.17403392493724823), (27, 0.03717954084277153)], Temporal MSE: [(0, 1.6457896663268912e-06), (1, 1.2604550647665747e-05), (2, 6.0479298554128036e-05), (3, 8.959606930147856e-05), (4, 0.00013226710143499076), (5, 0.0004527673590928316), (6, 0.00036430254112929106), (7, 0.0013506780378520489), (8, 0.0014052509795874357), (9, 0.0018190022092312574), (10, 0.004489569459110498), (11, 0.005792457144707441), (12, 0.005827442277222872), (13, 0.00787500012665987), (14, 0.02513349987566471), (15, 0.041112020611763), (16, 0.04639822617173195), (17, 0.0483446829020977), (18, 0.05426602438092232), (19, 0.06136900559067726), (20, 0.07814855128526688), (21, 0.1066761463880539), (22, 0.108971506357193), (23, 0.08894457668066025), (24, 0.055386848747730255), (25, 0.03504238650202751), (26, 0.02870400995016098), (27, 0.017885494977235794)]
+
+ 20%|██        | 6/30 [00:48<03:35,  8.99s/it]Time step 952, Spatial MSE: [(0, 5.6031709391390905e-05), (1, 0.00040489842649549246), (2, 0.0008981054415926337), (3, 0.00046552400453947484), (4, 0.0006616601021960378), (5, 0.001127465395256877), (6, 0.0032012315932661295), (7, 0.006393880117684603), (8, 0.011265534907579422), (9, 0.01990344561636448), (10, 0.030257776379585266), (11, 0.04833630099892616), (12, 0.0525711365044117), (13, 0.06633118540048599), (14, 0.1406567096710205), (15, 0.22424064576625824), (16, 0.22697214782238007), (17, 0.24281544983386993), (18, 0.2528653144836426), (19, 0.2532675266265869), (20, 0.32567623257637024), (21, 0.3579097092151642), (22, 0.36642488837242126), (23, 0.3762335479259491), (24, 0.4646390974521637), (25, 0.5000202059745789), (26, 0.2437630444765091), (27, 0.05633078143000603)], Temporal MSE: [(0, 1.2147628467573668e-06), (1, 9.160522495221812e-06), (2, 4.721313962363638e-05), (3, 5.877410512766801e-05), (4, 0.00010127775021828711), (5, 0.0002942402788903564), (6, 0.0003044752520509064), (7, 0.0010981272207573056), (8, 0.0011679567396640778), (9, 0.0014596986584365368), (10, 0.003596679074689746), (11, 0.004970469977706671), (12, 0.005263093858957291), (13, 0.00650088069960475), (14, 0.025358302518725395), (15, 0.0365133062005043), (16, 0.04337942972779274), (17, 0.05086487904191017), (18, 0.05884018540382385), (19, 0.07750040292739868), (20, 0.1424623727798462), (21, 0.17283077538013458), (22, 0.1414167732000351), (23, 0.14080177247524261), (24, 0.08431826531887054), (25, 0.03510994464159012), (26, 0.03428124636411667), (27, 0.03538454324007034)]
+
+ 23%|██▎       | 7/30 [00:58<03:28,  9.07s/it]Time step 940, Spatial MSE: [(0, 6.248099816730246e-05), (1, 0.00040764297591522336), (2, 0.0008306208183057606), (3, 0.0003990714903920889), (4, 0.0007301807636395097), (5, 0.0014118836261332035), (6, 0.004027252551168203), (7, 0.012928777374327183), (8, 0.014341632835566998), (9, 0.026158848777413368), (10, 0.06712748855352402), (11, 0.06142235919833183), (12, 0.06447925418615341), (13, 0.06554607301950455), (14, 0.1917385458946228), (15, 0.2597897946834564), (16, 0.19875407218933105), (17, 0.21305309236049652), (18, 0.23263965547084808), (19, 0.246658593416214), (20, 0.30264371633529663), (21, 0.4263991415500641), (22, 0.5238738059997559), (23, 0.5833764672279358), (24, 0.7077668309211731), (25, 0.6409664154052734), (26, 0.31334438920021057), (27, 0.15519528090953827)], Temporal MSE: [(0, 1.3365157656153315e-06), (1, 1.0313745406165253e-05), (2, 5.065299410489388e-05), (3, 4.849983815802261e-05), (4, 0.00016432913253083825), (5, 0.0002383154205745086), (6, 0.0006364368600770831), (7, 0.0035605200100690126), (8, 0.003113672835752368), (9, 0.002500197384506464), (10, 0.007401128299534321), (11, 0.010447544977068901), (12, 0.007380720693618059), (13, 0.009554407559335232), (14, 0.03193914517760277), (15, 0.04467802494764328), (16, 0.05232030153274536), (17, 0.053288783878088), (18, 0.06759646534919739), (19, 0.08567943423986435), (20, 0.17154055833816528), (21, 0.23062680661678314), (22, 0.20487025380134583), (23, 0.192755326628685), (24, 0.13402238488197327), (25, 0.05615655332803726), (26, 0.04247545823454857), (27, 0.07024727016687393)]
+
+ 27%|██▋       | 8/30 [01:07<03:20,  9.13s/it]Time step 932, Spatial MSE: [(0, 4.9221431254409254e-05), (1, 0.0003964707429986447), (2, 0.0005812150775454938), (3, 0.0003290875465609133), (4, 0.00033522347803227603), (5, 0.0006528531084768474), (6, 0.0016554933972656727), (7, 0.004002972040325403), (8, 0.004997382406145334), (9, 0.00894857756793499), (10, 0.016266468912363052), (11, 0.02064543217420578), (12, 0.022414958104491234), (13, 0.020388081669807434), (14, 0.04249805584549904), (15, 0.04905031993985176), (16, 0.061697717756032944), (17, 0.06310274451971054), (18, 0.08792956918478012), (19, 0.10018345713615417), (20, 0.1352088749408722), (21, 0.17652426660060883), (22, 0.18859103322029114), (23, 0.20318710803985596), (24, 0.26053208112716675), (25, 0.23618879914283752), (26, 0.11347747594118118), (27, 0.04890047386288643)], Temporal MSE: [(0, 9.467501627113961e-07), (1, 4.4994285417487845e-06), (2, 2.5376053599757142e-05), (3, 3.153597208438441e-05), (4, 6.911401578690857e-05), (5, 9.591886191628873e-05), (6, 0.00025083418586291373), (7, 0.001157409860752523), (8, 0.0009912061505019665), (9, 0.0009783358545973897), (10, 0.0024690970312803984), (11, 0.004202938638627529), (12, 0.002403289545327425), (13, 0.00281347194686532), (14, 0.00920264795422554), (15, 0.011079014278948307), (16, 0.014877974055707455), (17, 0.01749984733760357), (18, 0.025516729801893234), (19, 0.035266097635030746), (20, 0.08112694323062897), (21, 0.10615044832229614), (22, 0.06833107769489288), (23, 0.07040931284427643), (24, 0.04964420944452286), (25, 0.013746364042162895), (26, 0.014752591960132122), (27, 0.024647492915391922)]
+
+ 30%|███       | 9/30 [01:17<03:17,  9.40s/it]Time step 920, Spatial MSE: [(0, 2.973572554765269e-05), (1, 0.00040253609768114984), (2, 0.0006068822694942355), (3, 0.00017948441382031888), (4, 7.128697325242683e-05), (5, 0.00011431502207415178), (6, 0.00014696498692501336), (7, 0.0006540316389873624), (8, 0.0002130299835698679), (9, 0.00046538750757463276), (10, 0.00044474986498244107), (11, 0.00040451993118040264), (12, 0.0005721682100556791), (13, 0.0007158584194257855), (14, 0.001106885145418346), (15, 0.00155691709369421), (16, 0.010345129296183586), (17, 0.002747342921793461), (18, 0.004005515016615391), (19, 0.0024768896400928497), (20, 0.0018530662637203932), (21, 0.002800679998472333), (22, 0.002249028068035841), (23, 0.003267063060775399), (24, 0.0067779021337628365), (25, 0.014600101858377457), (26, 0.011852046474814415), (27, 0.012989399023354053)], Temporal MSE: [(0, 8.06123239271983e-07), (1, 3.608930683185463e-06), (2, 1.3298378689796664e-05), (3, 1.1957987226196565e-05), (4, 1.410739332641242e-05), (5, 1.6277446775347926e-05), (6, 3.467114584054798e-05), (7, 9.805846639210358e-05), (8, 4.98541776323691e-05), (9, 9.443643648410216e-05), (10, 0.00016987290291581303), (11, 0.00011522712156875059), (12, 0.00011631595407379791), (13, 0.00010389316594228148), (14, 0.0003029506478924304), (15, 0.0009198684711009264), (16, 0.0008621569140814245), (17, 0.00111003580968827), (18, 0.0010977905476465821), (19, 0.0011973499786108732), (20, 0.0022300798445940018), (21, 0.0011882191756740212), (22, 0.0009837549878284335), (23, 0.0014063987182453275), (24, 0.0008108284673653543), (25, 0.0008700593607500196), (26, 0.001105615054257214), (27, 0.0010074671590700746)]
+
+ 33%|███▎      | 10/30 [01:25<03:02,  9.11s/it]Time step 908, Spatial MSE: [(0, 0.00011445621930761263), (1, 0.0005093941581435502), (2, 0.0007477166363969445), (3, 0.00027088256319984794), (4, 0.0003300183452665806), (5, 0.0007483453373424709), (6, 0.0022275024093687534), (7, 0.00198911945335567), (8, 0.0022628633305430412), (9, 0.0028313612565398216), (10, 0.008971575647592545), (11, 0.008196109905838966), (12, 0.006547793745994568), (13, 0.006757963914424181), (14, 0.02300334721803665), (15, 0.014428962022066116), (16, 0.024131901562213898), (17, 0.014637244865298271), (18, 0.01814449392259121), (19, 0.02175970934331417), (20, 0.016987310722470284), (21, 0.02629712037742138), (22, 0.025670470669865608), (23, 0.018792517483234406), (24, 0.030519427731633186), (25, 0.04353485256433487), (26, 0.019567275419831276), (27, 0.01844518445432186)], Temporal MSE: [(0, 2.7104529181087855e-06), (1, 4.20863761974033e-06), (2, 4.409543180372566e-05), (3, 5.716682426282205e-05), (4, 3.9962913433555514e-05), (5, 0.00012886020704172552), (6, 0.00012436047836672515), (7, 0.000802272348664701), (8, 0.000504222814925015), (9, 0.0006532651023007929), (10, 0.0022436606232076883), (11, 0.0013157576322555542), (12, 0.0015648887492716312), (13, 0.0010478405747562647), (14, 0.0057328734546899796), (15, 0.005767410155385733), (16, 0.008766810409724712), (17, 0.005835381802171469), (18, 0.004931222181767225), (19, 0.008860668167471886), (20, 0.007518406957387924), (21, 0.009888753294944763), (22, 0.010593899525702), (23, 0.011870096437633038), (24, 0.006254459265619516), (25, 0.0034389144275337458), (26, 0.006068933755159378), (27, 0.006534603890031576)]
+
+ 37%|███▋      | 11/30 [01:33<02:46,  8.78s/it]Time step 892, Spatial MSE: [(0, 3.0290106224128976e-05), (1, 0.0005348274717107415), (2, 0.0007336533744819462), (3, 0.00020113104255869985), (4, 6.245540134841576e-05), (5, 0.00011461689427960664), (6, 0.0001282411685679108), (7, 0.0005056801601313055), (8, 0.00022114248713478446), (9, 0.0004089366120751947), (10, 0.00033470362541265786), (11, 0.0004845364601351321), (12, 0.00040892991819418967), (13, 0.0007575755007565022), (14, 0.0010410662507638335), (15, 0.0013524730456992984), (16, 0.009478320367634296), (17, 0.0020976730156689882), (18, 0.0032612821087241173), (19, 0.0033622155897319317), (20, 0.002306809648871422), (21, 0.004274034406989813), (22, 0.003060382092371583), (23, 0.004293012898415327), (24, 0.009137453511357307), (25, 0.013572141528129578), (26, 0.014652026817202568), (27, 0.00962853990495205)], Temporal MSE: [(0, 7.443796334882791e-07), (1, 3.540890247677453e-06), (2, 1.389467161061475e-05), (3, 1.2304942174523603e-05), (4, 1.39734920594492e-05), (5, 1.681483263382688e-05), (6, 2.3686850909143686e-05), (7, 7.95118830865249e-05), (8, 5.039289317210205e-05), (9, 7.403568451991305e-05), (10, 0.00013812432007398456), (11, 0.00010201879922533408), (12, 8.747965330258012e-05), (13, 0.00010066917457152158), (14, 0.0002953417715616524), (15, 0.0007773118559271097), (16, 0.0008606144110672176), (17, 0.0010826366487890482), (18, 0.0006654653116129339), (19, 0.0016607021680101752), (20, 0.003856194671243429), (21, 0.0013466712553054094), (22, 0.001082102651707828), (23, 0.0013541565276682377), (24, 0.000999955809675157), (25, 0.0008583726012147963), (26, 0.001317808753810823), (27, 0.0010760442819446325)]
+
+ 40%|████      | 12/30 [01:41<02:29,  8.30s/it]Time step 880, Spatial MSE: [(0, 0.00012748896551784128), (1, 0.0006676285993307829), (2, 0.0018668194534257054), (3, 0.0015062305610626936), (4, 0.003214749041944742), (5, 0.007342333905398846), (6, 0.017339278012514114), (7, 0.05336695536971092), (8, 0.059996139258146286), (9, 0.08306905627250671), (10, 0.17465364933013916), (11, 0.1611688882112503), (12, 0.14291362464427948), (13, 0.1508844643831253), (14, 0.2845765948295593), (15, 0.4129176437854767), (16, 0.5248290300369263), (17, 0.5189958810806274), (18, 0.7009329795837402), (19, 0.7212214469909668), (20, 0.8944072723388672), (21, 1.0601979494094849), (22, 0.972542941570282), (23, 1.3141049146652222), (24, 1.7862459421157837), (25, 1.2203763723373413), (26, 0.6690888404846191), (27, 0.21068543195724487)], Temporal MSE: [(0, 3.6931814975105226e-06), (1, 3.0092676752246916e-05), (2, 0.00041639170376583934), (3, 0.00042016481165774167), (4, 0.0009427678305655718), (5, 0.0016085458919405937), (6, 0.0034670261666178703), (7, 0.015417840331792831), (8, 0.01015824731439352), (9, 0.011135387234389782), (10, 0.02804975025355816), (11, 0.027742089703679085), (12, 0.01806209795176983), (13, 0.02341911941766739), (14, 0.059527438133955), (15, 0.10103985667228699), (16, 0.13995781540870667), (17, 0.12954804301261902), (18, 0.17957207560539246), (19, 0.2431456744670868), (20, 0.4609300494194031), (21, 0.5718129873275757), (22, 0.37398675084114075), (23, 0.41159915924072266), (24, 0.3869493901729584), (25, 0.07694166898727417), (26, 0.08026359975337982), (27, 0.11958392709493637)]
+
+ 43%|████▎     | 13/30 [01:49<02:23,  8.43s/it]Time step 864, Spatial MSE: [(0, 2.1442072466015816e-05), (1, 0.0004881548520643264), (2, 0.000686194165609777), (3, 0.00022928013640921563), (4, 5.643629265250638e-05), (5, 9.109455277211964e-05), (6, 0.0001310396910412237), (7, 0.00034771079663187265), (8, 0.00015855033416301012), (9, 0.00024026048777159303), (10, 0.00031084567308425903), (11, 0.00033940610592253506), (12, 0.0003355835215188563), (13, 0.0006016307161189616), (14, 0.0009796199155971408), (15, 0.0009327275911346078), (16, 0.012007283978164196), (17, 0.002001250861212611), (18, 0.0030454606749117374), (19, 0.005542931612581015), (20, 0.003713177051395178), (21, 0.013934008777141571), (22, 0.018701117485761642), (23, 0.00569829810410738), (24, 0.009489788673818111), (25, 0.015400392934679985), (26, 0.01572432555258274), (27, 0.014067539945244789)], Temporal MSE: [(0, 5.58675992579083e-07), (1, 2.7664464141707867e-06), (2, 1.0034231308964081e-05), (3, 9.968762242351659e-06), (4, 1.0699990525608882e-05), (5, 1.4337620086735114e-05), (6, 2.1709714928874746e-05), (7, 5.2204388339305297e-05), (8, 3.719177038874477e-05), (9, 5.233114643488079e-05), (10, 9.537186269881204e-05), (11, 6.773600034648553e-05), (12, 6.121191108832136e-05), (13, 6.781148840673268e-05), (14, 0.00024067694903351367), (15, 0.0005633055698126554), (16, 0.0011511105112731457), (17, 0.0010847033699974418), (18, 0.0004585848655551672), (19, 0.0025860462337732315), (20, 0.0070528388023376465), (21, 0.00445443345233798), (22, 0.005674128886312246), (23, 0.0012302191462367773), (24, 0.0014428941067308187), (25, 0.0008959242841228843), (26, 0.000998315867036581), (27, 0.002460018964484334)]
+
+ 47%|████▋     | 14/30 [01:56<02:06,  7.89s/it]Time step 848, Spatial MSE: [(0, 0.00014332799764815718), (1, 0.0005082606221549213), (2, 0.0006569046527147293), (3, 0.000259202002780512), (4, 0.00027109665097668767), (5, 0.0008983627776615322), (6, 0.0025236329529434443), (7, 0.0027943160384893417), (8, 0.0022204488050192595), (9, 0.003130717668682337), (10, 0.009065095335245132), (11, 0.007473602890968323), (12, 0.005213127937167883), (13, 0.006825353018939495), (14, 0.019808538258075714), (15, 0.016765279695391655), (16, 0.02026514895260334), (17, 0.013537844642996788), (18, 0.019805684685707092), (19, 0.024466978386044502), (20, 0.018749993294477463), (21, 0.021364951506257057), (22, 0.02810586430132389), (23, 0.01910746470093727), (24, 0.032861948013305664), (25, 0.0397370383143425), (26, 0.025896888226270676), (27, 0.01070837490260601)], Temporal MSE: [(0, 1.9846647774102166e-06), (1, 4.341706244304078e-06), (2, 2.0579351257765666e-05), (3, 4.4781583710573614e-05), (4, 7.746709889033809e-05), (5, 0.00014992928481660783), (6, 0.0002720363554544747), (7, 0.0014708759263157845), (8, 0.0007310776272788644), (9, 0.0011738219764083624), (10, 0.0029845836106687784), (11, 0.00163324736058712), (12, 0.0019257679814472795), (13, 0.0014387309784069657), (14, 0.006196819711476564), (15, 0.008293398655951023), (16, 0.010865282267332077), (17, 0.008891290053725243), (18, 0.00539707625284791), (19, 0.011448592878878117), (20, 0.010537320747971535), (21, 0.012366856448352337), (22, 0.008827976882457733), (23, 0.008457615971565247), (24, 0.006265399046242237), (25, 0.002544817514717579), (26, 0.006152825430035591), (27, 0.004501642193645239)]
+
+ 50%|█████     | 15/30 [02:03<01:53,  7.57s/it]Time step 828, Spatial MSE: [(0, 2.0265235434635542e-05), (1, 0.0005694120191037655), (2, 0.0008378545753657818), (3, 0.0002672116388566792), (4, 4.6831446525175124e-05), (5, 8.579045970691368e-05), (6, 8.350938151124865e-05), (7, 0.0003346098819747567), (8, 0.00015004347369540483), (9, 0.00022490030096378177), (10, 0.00022278651886153966), (11, 0.00023792131105437875), (12, 0.00037254096241667867), (13, 0.000461316027212888), (14, 0.0007105313125066459), (15, 0.0006319349631667137), (16, 0.009255553595721722), (17, 0.0015841069398447871), (18, 0.0024295945186167955), (19, 0.002798785688355565), (20, 0.0018952788086608052), (21, 0.004025731701403856), (22, 0.003103049471974373), (23, 0.004424676764756441), (24, 0.008339866995811462), (25, 0.015517586842179298), (26, 0.015498070977628231), (27, 0.016511261463165283)], Temporal MSE: [(0, 5.342168378774659e-07), (1, 2.036626256085583e-06), (2, 1.0159717930946499e-05), (3, 8.874204468156677e-06), (4, 1.0688517249946017e-05), (5, 1.7154237866634503e-05), (6, 2.1298557840054855e-05), (7, 5.835231422679499e-05), (8, 4.433926733327098e-05), (9, 4.5588076318381354e-05), (10, 7.167801231844351e-05), (11, 5.8771671319846064e-05), (12, 7.392496627289802e-05), (13, 6.597310857614502e-05), (14, 0.000195395274204202), (15, 0.00035983900306746364), (16, 0.0008276262087747455), (17, 0.0007885299855843186), (18, 0.0003989977412857115), (19, 0.001547124469652772), (20, 0.004172988701611757), (21, 0.0016012419946491718), (22, 0.001008879509754479), (23, 0.0015913551906123757), (24, 0.0012893154053017497), (25, 0.000942687620408833), (26, 0.0008516735979355872), (27, 0.002122288104146719)]
+
+ 53%|█████▎    | 16/30 [02:10<01:43,  7.36s/it]Time step 808, Spatial MSE: [(0, 0.00011414737673476338), (1, 0.000997345894575119), (2, 0.0013287471374496818), (3, 0.0015048390487208962), (4, 0.0027444090228527784), (5, 0.00746190408244729), (6, 0.012545379810035229), (7, 0.036493051797151566), (8, 0.03581671789288521), (9, 0.04604082182049751), (10, 0.0688917338848114), (11, 0.06766576319932938), (12, 0.052233267575502396), (13, 0.061778489500284195), (14, 0.1061418280005455), (15, 0.14266854524612427), (16, 0.1931406557559967), (17, 0.1875700205564499), (18, 0.2077489197254181), (19, 0.2714979350566864), (20, 0.34574416279792786), (21, 0.4088684022426605), (22, 0.44180816411972046), (23, 0.5735325217247009), (24, 0.7110137343406677), (25, 0.6547002196311951), (26, 0.41661542654037476), (27, 0.11042246967554092)], Temporal MSE: [(0, 2.8394010769261513e-06), (1, 2.5636329155531712e-05), (2, 0.0001832524430938065), (3, 0.0003170450800098479), (4, 0.0008752650464884937), (5, 0.0017887897556647658), (6, 0.0029990358743816614), (7, 0.008562915027141571), (8, 0.006196349859237671), (9, 0.0067948210053145885), (10, 0.01129402406513691), (11, 0.012956582009792328), (12, 0.009596357122063637), (13, 0.01136060431599617), (14, 0.027701696380972862), (15, 0.03764226287603378), (16, 0.05262212082743645), (17, 0.053829316049814224), (18, 0.08050493150949478), (19, 0.11872807145118713), (20, 0.22929023206233978), (21, 0.2280580997467041), (22, 0.1948237270116806), (23, 0.19466719031333923), (24, 0.16656169295310974), (25, 0.03870174661278725), (26, 0.04246003180742264), (27, 0.07696246355772018)]
+
+ 57%|█████▋    | 17/30 [02:19<01:42,  7.86s/it]Time step 788, Spatial MSE: [(0, 2.0256111383787356e-05), (1, 0.0005205589695833623), (2, 0.0009409105987288058), (3, 0.00030353423790074885), (4, 4.934279422741383e-05), (5, 8.478333620587364e-05), (6, 8.603458991274238e-05), (7, 0.0004361117898952216), (8, 0.0001955947809619829), (9, 0.00017298688180744648), (10, 0.00020120598492212594), (11, 0.0002678200544323772), (12, 0.00032802781788632274), (13, 0.00046938558807596564), (14, 0.0006781714037060738), (15, 0.0005707218660973012), (16, 0.008115381933748722), (17, 0.0011841453379020095), (18, 0.0095317168161273), (19, 0.003344006137922406), (20, 0.0019407649524509907), (21, 0.003740783780813217), (22, 0.002904278226196766), (23, 0.0045446716248989105), (24, 0.007748816628009081), (25, 0.01707249879837036), (26, 0.01644555851817131), (27, 0.010030529461801052)], Temporal MSE: [(0, 5.831921612298174e-07), (1, 1.881251819213503e-06), (2, 1.1098989489255473e-05), (3, 9.758676242199726e-06), (4, 1.1030612768081482e-05), (5, 1.833249189076014e-05), (6, 2.3325243091676384e-05), (7, 5.709615652449429e-05), (8, 4.290920333005488e-05), (9, 4.238932888256386e-05), (10, 7.809498492861167e-05), (11, 5.6203956773970276e-05), (12, 6.193693116074428e-05), (13, 7.097781053744256e-05), (14, 0.00018322651158086956), (15, 0.0003459335712250322), (16, 0.0010721537983044982), (17, 0.0007253428921103477), (18, 0.0027665472589433193), (19, 0.002690886380150914), (20, 0.008929628878831863), (21, 0.0019413451664149761), (22, 0.0010029466357082129), (23, 0.00122950435616076), (24, 0.0011671099346131086), (25, 0.0011013014009222388), (26, 0.0009608928230591118), (27, 0.0013969654683023691)]
+
+ 60%|██████    | 18/30 [02:25<01:30,  7.52s/it]Time step 764, Spatial MSE: [(0, 0.0001434815931133926), (1, 0.0007654578657820821), (2, 0.0007845604559406638), (3, 0.000265144684817642), (4, 0.0002259214670630172), (5, 0.000650543428491801), (6, 0.0014779928606003523), (7, 0.0021852932404726744), (8, 0.0017560842679813504), (9, 0.002581019653007388), (10, 0.006467490456998348), (11, 0.0051216669380664825), (12, 0.004711503628641367), (13, 0.006214568391442299), (14, 0.01645226962864399), (15, 0.015115096233785152), (16, 0.021270152181386948), (17, 0.013192241080105305), (18, 0.016682634130120277), (19, 0.0243289303034544), (20, 0.01677405647933483), (21, 0.028781531378626823), (22, 0.026512518525123596), (23, 0.025512728840112686), (24, 0.038919322192668915), (25, 0.056882623583078384), (26, 0.030905330553650856), (27, 0.011699432507157326)], Temporal MSE: [(0, 2.0245608993718633e-06), (1, 4.761541276820935e-06), (2, 1.995065395021811e-05), (3, 4.323585380916484e-05), (4, 5.390040678321384e-05), (5, 8.615365368314087e-05), (6, 0.00022833973343949765), (7, 0.0009337119990959764), (8, 0.0005577058182097971), (9, 0.0008962529245764017), (10, 0.0024411133490502834), (11, 0.0014462287072092295), (12, 0.001680646208114922), (13, 0.0011158192064613104), (14, 0.004814159125089645), (15, 0.006132269743829966), (16, 0.008537896908819675), (17, 0.00798637792468071), (18, 0.005613849963992834), (19, 0.008993656374514103), (20, 0.011596648953855038), (21, 0.014692481607198715), (22, 0.009686144068837166), (23, 0.008634324185550213), (24, 0.01122585590928793), (25, 0.002843749476596713), (26, 0.005669888108968735), (27, 0.006379606202244759)]
+
+ 63%|██████▎   | 19/30 [02:33<01:22,  7.54s/it]Time step 736, Spatial MSE: [(0, 3.0413968488574028e-05), (1, 0.0008811237057670951), (2, 0.001356078195385635), (3, 0.00048772210720926523), (4, 7.655748777324334e-05), (5, 0.00012127877562306821), (6, 0.00012141008483013138), (7, 0.00036245869705453515), (8, 0.00018362524861004204), (9, 0.0002502728020772338), (10, 0.00023654240067116916), (11, 0.0003265419218223542), (12, 0.0003827738109976053), (13, 0.0005371784791350365), (14, 0.0009025553008541465), (15, 0.00099461548961699), (16, 0.01200978271663189), (17, 0.002474294975399971), (18, 0.0029822702053934336), (19, 0.0033580735325813293), (20, 0.002443955047056079), (21, 0.006671281065791845), (22, 0.0041936906054615974), (23, 0.006345055531710386), (24, 0.010527007281780243), (25, 0.023480087518692017), (26, 0.023285243660211563), (27, 0.012183663435280323)], Temporal MSE: [(0, 7.367775083366723e-07), (1, 2.7728935947379796e-06), (2, 1.696377330517862e-05), (3, 1.3953663255961146e-05), (4, 1.6877320376806892e-05), (5, 2.1273232050589286e-05), (6, 4.080722283106297e-05), (7, 6.65584666421637e-05), (8, 5.1796996558550745e-05), (9, 6.545231735799462e-05), (10, 8.357333717867732e-05), (11, 7.33439446776174e-05), (12, 8.348854316864163e-05), (13, 9.38464145292528e-05), (14, 0.00026412378065288067), (15, 0.0006324038840830326), (16, 0.0012694952310994267), (17, 0.001435017678886652), (18, 0.000621938263066113), (19, 0.0020000897347927094), (20, 0.008546402677893639), (21, 0.002631292212754488), (22, 0.0011933475034311414), (23, 0.0016595381312072277), (24, 0.001697447500191629), (25, 0.001726442133076489), (26, 0.0012793905334547162), (27, 0.0024615295697003603)]
+
+ 67%|██████▋   | 20/30 [02:40<01:13,  7.36s/it]Time step 708, Spatial MSE: [(0, 7.379084854619578e-05), (1, 0.0017084258142858744), (2, 0.0017049630405381322), (3, 0.0014686238719150424), (4, 0.002287618350237608), (5, 0.006483034230768681), (6, 0.008500805124640465), (7, 0.017289983108639717), (8, 0.023524805903434753), (9, 0.02797004021704197), (10, 0.032128024846315384), (11, 0.033875636756420135), (12, 0.033590804785490036), (13, 0.03899652510881424), (14, 0.06930156797170639), (15, 0.08575989305973053), (16, 0.12085078656673431), (17, 0.11085833609104156), (18, 0.13418349623680115), (19, 0.1862666755914688), (20, 0.2227754294872284), (21, 0.3119644522666931), (22, 0.3721872568130493), (23, 0.42146575450897217), (24, 0.5062047243118286), (25, 0.4819166660308838), (26, 0.3340160548686981), (27, 0.0699300765991211)], Temporal MSE: [(0, 3.9262981772481e-06), (1, 2.8320786441327073e-05), (2, 0.00014507380547001958), (3, 0.00025315259699709713), (4, 0.0009194431477226317), (5, 0.0017173157539218664), (6, 0.0029091890901327133), (7, 0.00660510640591383), (8, 0.004863535054028034), (9, 0.005545367952436209), (10, 0.008007636293768883), (11, 0.00856560468673706), (12, 0.006862425245344639), (13, 0.008126216009259224), (14, 0.020154230296611786), (15, 0.02651127241551876), (16, 0.03482797369360924), (17, 0.04124986752867699), (18, 0.0642566829919815), (19, 0.088727205991745), (20, 0.21470731496810913), (21, 0.18128274381160736), (22, 0.13768048584461212), (23, 0.15209753811359406), (24, 0.12964177131652832), (25, 0.02993883192539215), (26, 0.038895636796951294), (27, 0.06433437019586563)]
+
+ 70%|███████   | 21/30 [02:49<01:10,  7.88s/it]Time step 676, Spatial MSE: [(0, 3.347607344039716e-05), (1, 0.0011783523950725794), (2, 0.0018287200946360826), (3, 0.0006546357180923223), (4, 9.957954898709431e-05), (5, 0.00015391026681754738), (6, 0.00015064737817738205), (7, 0.0004354463017079979), (8, 0.0002106279571307823), (9, 0.00020034375484101474), (10, 0.00025429611559957266), (11, 0.0002771291183307767), (12, 0.00032183135044761), (13, 0.00043182517401874065), (14, 0.0007773812394589186), (15, 0.001267621060833335), (16, 0.0058141425251960754), (17, 0.0015239766798913479), (18, 0.0015935228439047933), (19, 0.002085026353597641), (20, 0.001812290633097291), (21, 0.004863101523369551), (22, 0.004197690170258284), (23, 0.007121602073311806), (24, 0.010199534706771374), (25, 0.03119564615190029), (26, 0.030526185408234596), (27, 0.014314034953713417)], Temporal MSE: [(0, 1.0339591653973912e-06), (1, 3.556544243110693e-06), (2, 1.9471368432277814e-05), (3, 1.8941866073873825e-05), (4, 2.2405405616154894e-05), (5, 3.421375367906876e-05), (6, 5.359348142519593e-05), (7, 9.04443659237586e-05), (8, 6.307514559011906e-05), (9, 7.432122947648168e-05), (10, 0.00010570216545602307), (11, 8.151766087394208e-05), (12, 8.35791724966839e-05), (13, 0.00010389873204985633), (14, 0.0003851843357551843), (15, 0.000711020256858319), (16, 0.0007261094870045781), (17, 0.0010178176453337073), (18, 0.0005219659069553018), (19, 0.0014480773825198412), (20, 0.006194552406668663), (21, 0.002005230402573943), (22, 0.001267059938982129), (23, 0.001775588607415557), (24, 0.001707676099613309), (25, 0.002529729390516877), (26, 0.0014583001611754298), (27, 0.0032441699877381325)]
+
+ 73%|███████▎  | 22/30 [02:57<01:02,  7.77s/it]Time step 636, Spatial MSE: [(0, 9.038360440172255e-05), (1, 0.0016279853880405426), (2, 0.0015194056322798133), (3, 0.00043253027251921594), (4, 0.00033406331203877926), (5, 0.000901554711163044), (6, 0.001953855389729142), (7, 0.0036188855301588774), (8, 0.002362586557865143), (9, 0.0027225131634622812), (10, 0.006952276453375816), (11, 0.006125088315457106), (12, 0.005325667560100555), (13, 0.0066142906434834), (14, 0.017386924475431442), (15, 0.02241048589348793), (16, 0.027575984597206116), (17, 0.0186692513525486), (18, 0.023382050916552544), (19, 0.04177479445934296), (20, 0.023787161335349083), (21, 0.03315247222781181), (22, 0.02419513277709484), (23, 0.032843783497810364), (24, 0.06519903242588043), (25, 0.11442338675260544), (26, 0.05626783147454262), (27, 0.02830308862030506)], Temporal MSE: [(0, 2.2761273612559307e-06), (1, 6.5457779783173464e-06), (2, 3.180852945661172e-05), (3, 6.51156369713135e-05), (4, 0.0001616087683942169), (5, 0.00014147238107398152), (6, 0.0006542312330566347), (7, 0.0018657803302630782), (8, 0.0010824556229636073), (9, 0.0017625420587137341), (10, 0.0044435253366827965), (11, 0.0023741424083709717), (12, 0.002659026999026537), (13, 0.001885535311885178), (14, 0.007569937035441399), (15, 0.01210759300738573), (16, 0.016299806535243988), (17, 0.014326386153697968), (18, 0.007829262875020504), (19, 0.016979089006781578), (20, 0.012274393811821938), (21, 0.0191989503800869), (22, 0.009573851712048054), (23, 0.00894281454384327), (24, 0.01025791559368372), (25, 0.004178350325673819), (26, 0.009564672596752644), (27, 0.008744274266064167)]
+
+ 77%|███████▋  | 23/30 [03:04<00:52,  7.54s/it]Time step 596, Spatial MSE: [(0, 4.544546754914336e-05), (1, 0.0018134985584765673), (2, 0.0028111329302191734), (3, 0.0010507606202736497), (4, 0.00016854930436238647), (5, 0.0002704673388507217), (6, 0.0002467586600687355), (7, 0.0003896283742506057), (8, 0.00031714607030153275), (9, 0.00028296481468714774), (10, 0.00042269035475328565), (11, 0.00042316902545280755), (12, 0.0004454566224012524), (13, 0.0007134214974939823), (14, 0.0010842460906133056), (15, 0.0011169809149578214), (16, 0.007106472738087177), (17, 0.0025677732191979885), (18, 0.002152766566723585), (19, 0.005471004173159599), (20, 0.0046692113392055035), (21, 0.017717396840453148), (22, 0.024629361927509308), (23, 0.01615848019719124), (24, 0.018169589340686798), (25, 0.053865205496549606), (26, 0.05364534258842468), (27, 0.026934927329421043)], Temporal MSE: [(0, 2.197838966822019e-06), (1, 6.386643235600786e-06), (2, 3.0255514502641745e-05), (3, 2.799131652864162e-05), (4, 4.0183407691074535e-05), (5, 5.1268405513837934e-05), (6, 9.164166112896055e-05), (7, 0.00012923954636789858), (8, 9.97610404738225e-05), (9, 0.00012340585817582905), (10, 0.00014595661195926368), (11, 0.0001378419838147238), (12, 0.00014027116412762552), (13, 0.00015618566249031574), (14, 0.0003520913887768984), (15, 0.0006477733259089291), (16, 0.001167031587101519), (17, 0.0018418219406157732), (18, 0.0009909386280924082), (19, 0.0032545626163482666), (20, 0.013967848382890224), (21, 0.006479439325630665), (22, 0.005250825546681881), (23, 0.002673147013410926), (24, 0.0028590920846909285), (25, 0.004446230363100767), (26, 0.002691885456442833), (27, 0.008177668787539005)]
+
+ 80%|████████  | 24/30 [03:10<00:43,  7.21s/it]Time step 548, Spatial MSE: [(0, 0.00010725058382377028), (1, 0.0039025309961289167), (2, 0.0038400343619287014), (3, 0.0025174799375236034), (4, 0.0028995356988161802), (5, 0.007808297406882048), (6, 0.010889336466789246), (7, 0.01599907875061035), (8, 0.02434496209025383), (9, 0.026605620980262756), (10, 0.028741447255015373), (11, 0.03272578492760658), (12, 0.03157263994216919), (13, 0.03876014053821564), (14, 0.06288103014230728), (15, 0.07299633324146271), (16, 0.10605976730585098), (17, 0.08614218235015869), (18, 0.10556034743785858), (19, 0.1492040455341339), (20, 0.1824965924024582), (21, 0.22529448568820953), (22, 0.2859169542789459), (23, 0.3589237332344055), (24, 0.47925490140914917), (25, 0.6432200074195862), (26, 0.4834979474544525), (27, 0.08457329124212265)], Temporal MSE: [(0, 1.0171073881792836e-05), (1, 5.408355355029926e-05), (2, 0.00027297023916617036), (3, 0.00042823984404094517), (4, 0.0012581164482980967), (5, 0.0023935618810355663), (6, 0.004247893579304218), (7, 0.0077459500171244144), (8, 0.005565186496824026), (9, 0.006472927052527666), (10, 0.008871573023498058), (11, 0.009503294713795185), (12, 0.008234422653913498), (13, 0.008701536804437637), (14, 0.021441249176859856), (15, 0.02607632987201214), (16, 0.031569886952638626), (17, 0.035459015518426895), (18, 0.05351459980010986), (19, 0.08005774021148682), (20, 0.17012320458889008), (21, 0.15499943494796753), (22, 0.10338077694177628), (23, 0.1311028003692627), (24, 0.13164009153842926), (25, 0.049214474856853485), (26, 0.05641781538724899), (27, 0.10076076537370682)]
+
+ 83%|████████▎ | 25/30 [03:19<00:38,  7.77s/it]Time step 492, Spatial MSE: [(0, 8.550686470698565e-05), (1, 0.0035707212518900633), (2, 0.0046167257241904736), (3, 0.0020723084453493357), (4, 0.0003075534477829933), (5, 0.000588243652600795), (6, 0.0005077971727587283), (7, 0.0010111061856150627), (8, 0.0006725647253915668), (9, 0.0006339683895930648), (10, 0.000598515325691551), (11, 0.0007269683992490172), (12, 0.0010078431805595756), (13, 0.0013969794381409883), (14, 0.0011197470594197512), (15, 0.001712044351734221), (16, 0.004406357649713755), (17, 0.0018443275475874543), (18, 0.01322075817734003), (19, 0.005290984641760588), (20, 0.006685963831841946), (21, 0.025211086496710777), (22, 0.013264463283121586), (23, 0.023495668545365334), (24, 0.031677260994911194), (25, 0.1125873327255249), (26, 0.11363515257835388), (27, 0.04484321177005768)], Temporal MSE: [(0, 4.0246382013719995e-06), (1, 1.1736836313502863e-05), (2, 5.900419273530133e-05), (3, 6.370097980834544e-05), (4, 8.641357271699235e-05), (5, 0.00012665388931054622), (6, 0.0002068801550194621), (7, 0.00033503887243568897), (8, 0.00020678223518189043), (9, 0.0002643649058882147), (10, 0.00023054672055877745), (11, 0.00022848154185339808), (12, 0.00026329068350605667), (13, 0.00030477854306809604), (14, 0.0005214366246946156), (15, 0.0009203579975292087), (16, 0.0011683671036735177), (17, 0.0014314033323898911), (18, 0.003161632688716054), (19, 0.005445447284728289), (20, 0.03318541869521141), (21, 0.009832000359892845), (22, 0.00321174249984324), (23, 0.005083298776298761), (24, 0.005125644616782665), (25, 0.00989999808371067), (26, 0.004696515388786793), (27, 0.01506782229989767)]
+
+ 87%|████████▋ | 26/30 [03:26<00:30,  7.57s/it]Time step 426, Spatial MSE: [(0, 0.0002439056697767228), (1, 0.0065732984803617), (2, 0.005705295130610466), (3, 0.001997273648157716), (4, 0.0017559771658852696), (5, 0.004322586581110954), (6, 0.006306617986410856), (7, 0.009757445193827152), (8, 0.011774273589253426), (9, 0.012266469188034534), (10, 0.016801318153738976), (11, 0.015879495069384575), (12, 0.01576446183025837), (13, 0.01830112747848034), (14, 0.046009793877601624), (15, 0.05144178122282028), (16, 0.06813295930624008), (17, 0.0509202815592289), (18, 0.05853480100631714), (19, 0.08929233998060226), (20, 0.09078136831521988), (21, 0.12169116735458374), (22, 0.14220014214515686), (23, 0.26188281178474426), (24, 0.4342649281024933), (25, 0.7392207980155945), (26, 0.49751991033554077), (27, 0.09382139891386032)], Temporal MSE: [(0, 1.3623883205582388e-05), (1, 5.0136113713961095e-05), (2, 0.0002526064054109156), (3, 0.00025076139718294144), (4, 0.0008276323787868023), (5, 0.0015592852141708136), (6, 0.0039909109473228455), (7, 0.005767309106886387), (8, 0.002956679556518793), (9, 0.003948077093809843), (10, 0.005790005438029766), (11, 0.0056829024106264114), (12, 0.004857145249843597), (13, 0.004861049819737673), (14, 0.01627478562295437), (15, 0.01913413591682911), (16, 0.02045120671391487), (17, 0.01943795755505562), (18, 0.028447089716792107), (19, 0.035145848989486694), (20, 0.05344099551439285), (21, 0.07977812737226486), (22, 0.05602189898490906), (23, 0.061941083520650864), (24, 0.09003402292728424), (25, 0.037315238267183304), (26, 0.06399115175008774), (27, 0.10340505093336105)]
+
+ 90%|█████████ | 27/30 [03:36<00:24,  8.08s/it]Time step 350, Spatial MSE: [(0, 0.0001909304701257497), (1, 0.009408675134181976), (2, 0.007844535633921623), (3, 0.0021882441360503435), (4, 0.001096869818866253), (5, 0.002949788933619857), (6, 0.003749449271708727), (7, 0.0053613935597240925), (8, 0.005386587232351303), (9, 0.005140693858265877), (10, 0.007417724467813969), (11, 0.0069926087744534016), (12, 0.007608219049870968), (13, 0.009110759012401104), (14, 0.0228126160800457), (15, 0.026019679382443428), (16, 0.028833914548158646), (17, 0.025276700034737587), (18, 0.027121305465698242), (19, 0.049017954617738724), (20, 0.05007011443376541), (21, 0.07499024271965027), (22, 0.10827329754829407), (23, 0.1794106364250183), (24, 0.26720455288887024), (25, 0.4299839437007904), (26, 0.36048632860183716), (27, 0.16440406441688538)], Temporal MSE: [(0, 1.4261580872698687e-05), (1, 5.185435657040216e-05), (2, 0.00020364060765132308), (3, 0.0001937060442287475), (4, 0.00047156529035419226), (5, 0.0008582871523685753), (6, 0.0022973506711423397), (7, 0.0027849774342030287), (8, 0.0014372485456988215), (9, 0.00198298878967762), (10, 0.0027926433831453323), (11, 0.0024092800449579954), (12, 0.00241128564812243), (13, 0.0024778852239251137), (14, 0.008340435102581978), (15, 0.008524533361196518), (16, 0.010003115050494671), (17, 0.010285916738212109), (18, 0.012575637549161911), (19, 0.016938116401433945), (20, 0.043809354305267334), (21, 0.055693119764328), (22, 0.030075641348958015), (23, 0.03674846887588501), (24, 0.06402374058961868), (25, 0.025786349549889565), (26, 0.05684203281998634), (27, 0.09184173494577408)]
+
+ 93%|█████████▎| 28/30 [03:44<00:16,  8.32s/it]Time step 258, Spatial MSE: [(0, 0.00036651978734880686), (1, 0.014960937201976776), (2, 0.0134915541857481), (3, 0.004263365175575018), (4, 0.001989407930523157), (5, 0.005439510568976402), (6, 0.006998188327997923), (7, 0.011971509084105492), (8, 0.009577448479831219), (9, 0.00882044155150652), (10, 0.013935774564743042), (11, 0.011961023323237896), (12, 0.013914165087044239), (13, 0.015508981421589851), (14, 0.040926579385995865), (15, 0.050241682678461075), (16, 0.052041009068489075), (17, 0.04626525565981865), (18, 0.06093578040599823), (19, 0.0763707160949707), (20, 0.08432064205408096), (21, 0.13710372149944305), (22, 0.1584840714931488), (23, 0.33880117535591125), (24, 0.517184853553772), (25, 0.9213084578514099), (26, 0.7409994006156921), (27, 0.37081342935562134)], Temporal MSE: [(0, 3.586952880141325e-05), (1, 0.00010055021266452968), (2, 0.00035090019810013473), (3, 0.0003903410688508302), (4, 0.0009436115506105125), (5, 0.0018308231374248862), (6, 0.0045111277140676975), (7, 0.006383298430591822), (8, 0.002802554750815034), (9, 0.0037594398017972708), (10, 0.00524876406416297), (11, 0.004538275301456451), (12, 0.004497654270380735), (13, 0.004245155490934849), (14, 0.014169934205710888), (15, 0.01454167626798153), (16, 0.017145661637187004), (17, 0.016723597422242165), (18, 0.02103656716644764), (19, 0.03018656000494957), (20, 0.055186253041028976), (21, 0.09686639159917831), (22, 0.05576552823185921), (23, 0.06291724741458893), (24, 0.13793016970157623), (25, 0.07160547375679016), (26, 0.13639521598815918), (27, 0.20049992203712463)]
+
+ 97%|█████████▋| 29/30 [03:54<00:08,  8.61s/it]Time step 143, Spatial MSE: [(0, 0.0008806612459011376), (1, 0.02847950905561447), (2, 0.02511337585747242), (3, 0.009355884045362473), (4, 0.004518439061939716), (5, 0.011542201042175293), (6, 0.014444585889577866), (7, 0.02957649528980255), (8, 0.019770508632063866), (9, 0.016884859651327133), (10, 0.03285973519086838), (11, 0.02821706049144268), (12, 0.030680475756525993), (13, 0.02951302006840706), (14, 0.07738114148378372), (15, 0.104590505361557), (16, 0.09110336005687714), (17, 0.09722113609313965), (18, 0.11248362809419632), (19, 0.16622941195964813), (20, 0.20237697660923004), (21, 0.28241756558418274), (22, 0.33573055267333984), (23, 0.6868709325790405), (24, 1.4500638246536255), (25, 2.478437900543213), (26, 1.8170429468154907), (27, 0.9542791843414307)], Temporal MSE: [(0, 0.00010440567712066695), (1, 0.00021776778157800436), (2, 0.0007632972556166351), (3, 0.0008782237418927252), (4, 0.0024213313590735197), (5, 0.004271386191248894), (6, 0.010929892770946026), (7, 0.01394362561404705), (8, 0.005735835060477257), (9, 0.007051939144730568), (10, 0.011414662003517151), (11, 0.010067097842693329), (12, 0.009312826208770275), (13, 0.00949709489941597), (14, 0.0264719370752573), (15, 0.02872348204255104), (16, 0.038725849241018295), (17, 0.03477820008993149), (18, 0.04263576865196228), (19, 0.06179631128907204), (20, 0.13243111968040466), (21, 0.21753913164138794), (22, 0.1068824827671051), (23, 0.1537286788225174), (24, 0.37813565135002136), (25, 0.30661073327064514), (26, 0.3224276900291443), (27, 0.46119678020477295)]
+
+100%|██████████| 30/30 [04:03<00:00,  8.72s/it]
+100%|██████████| 30/30 [04:03<00:00,  8.10s/it]
+[2024-10-01 15:03:01] Prompt: a dog in astronaut suit and sunglasses floating in space. aesthetic score: 6.5.
+Saved to ./samples/opensora/sample_pab/A dog in astronaut suit and sunglasses f.mp4
+Saved video to ./samples/opensora/sample_pab/A dog in astronaut suit and sunglasses f.mp4
+Saved spatial MLP MSE outputs to ./samples/opensora/sample_pab/A dog in astronaut suit and sunglasses f_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora/sample_pab/A dog in astronaut suit and sunglasses f_temporal_mlp_mse.pt
+timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:0'), tensor([828.8770], device='cuda:0'), tensor([809.0977], device='cuda:0'), tensor([787.4170], device='cuda:0'), tensor([763.5468], device='cuda:0'), tensor([737.1379], device='cuda:0'), tensor([707.7626], device='cuda:0'), tensor([674.8911], device='cuda:0'), tensor([637.8601], device='cuda:0'), tensor([595.8265], device='cuda:0'), tensor([547.7032], device='cuda:0'), tensor([492.0635], device='cuda:0'), tensor([426.9973], device='cuda:0'), tensor([349.8871], device='cuda:0'), tensor([257.0481], device='cuda:0'), tensor([143.1210], device='cuda:0')]
+
+  0%|          | 0/30 [00:00<?, ?it/s]
+  3%|▎         | 1/30 [00:03<01:48,  3.75s/it]Time step 992, Spatial MSE: [(0, 0.0009826644090935588), (1, 0.0015938361175358295), (2, 0.042121462523937225), (3, 0.02673138864338398), (4, 0.027373753488063812), (5, 0.058088574558496475), (6, 0.04488351196050644), (7, 0.06641557067632675), (8, 0.09693694859743118), (9, 0.18721871078014374), (10, 0.29922574758529663), (11, 0.3259481191635132), (12, 0.3876131474971771), (13, 0.34777945280075073), (14, 0.386199414730072), (15, 0.3575621545314789), (16, 0.9648308157920837), (17, 0.5148411393165588), (18, 0.3929464519023895), (19, 0.467255175113678), (20, 0.7310891151428223), (21, 0.5374534726142883), (22, 0.5024734139442444), (23, 1.1787619590759277), (24, 5.71518611907959), (25, 5.121818542480469), (26, 4.163567066192627), (27, 0.7164974212646484)], Temporal MSE: [(0, 4.802617695531808e-05), (1, 0.00042058571125380695), (2, 0.001467064837925136), (3, 0.0023475352209061384), (4, 0.002704013604670763), (5, 0.012725678272545338), (6, 0.005764514673501253), (7, 0.009095205925405025), (8, 0.008685636334121227), (9, 0.015592941083014011), (10, 0.03928139805793762), (11, 0.040484469383955), (12, 0.04747796803712845), (13, 0.02204746939241886), (14, 0.05358533933758736), (15, 0.07456111162900925), (16, 0.16314001381397247), (17, 0.05929147079586983), (18, 0.07579532265663147), (19, 0.10635372996330261), (20, 0.1517179310321808), (21, 0.15238164365291595), (22, 0.42685672640800476), (23, 0.4363986551761627), (24, 0.3417843282222748), (25, 0.2169264853000641), (26, 0.5021067261695862), (27, 0.7552639245986938)]
+
+  7%|▋         | 2/30 [00:11<02:51,  6.11s/it]Time step 984, Spatial MSE: [(0, 0.00042149098590016365), (1, 0.00148525380063802), (2, 0.006480132695287466), (3, 0.008520718663930893), (4, 0.011084607802331448), (5, 0.020657721906900406), (6, 0.028422275558114052), (7, 0.03839047998189926), (8, 0.07278874516487122), (9, 0.1614285558462143), (10, 0.30339470505714417), (11, 0.29468390345573425), (12, 0.2691798508167267), (13, 0.24845150113105774), (14, 0.39139845967292786), (15, 0.401341050863266), (16, 0.7285784482955933), (17, 0.6490938067436218), (18, 0.6006825566291809), (19, 0.7630776166915894), (20, 1.2522685527801514), (21, 0.7711606621742249), (22, 0.7537096738815308), (23, 1.06324303150177), (24, 3.2476799488067627), (25, 3.701446533203125), (26, 2.7724037170410156), (27, 0.29136455059051514)], Temporal MSE: [(0, 8.477046321786474e-06), (1, 0.00015108671505004168), (2, 0.00029781030025333166), (3, 0.00048819888615980744), (4, 0.0007642362033948302), (5, 0.0015784089919179678), (6, 0.002835542429238558), (7, 0.007082501892000437), (8, 0.005884467624127865), (9, 0.00875243078917265), (10, 0.025745894759893417), (11, 0.02145235985517502), (12, 0.01978873461484909), (13, 0.01809808425605297), (14, 0.05166846886277199), (15, 0.0765732005238533), (16, 0.09616583585739136), (17, 0.07173284888267517), (18, 0.05555430054664612), (19, 0.08898677676916122), (20, 0.15573304891586304), (21, 0.22027142345905304), (22, 0.4621333181858063), (23, 0.33229392766952515), (24, 0.2212623506784439), (25, 0.11589835584163666), (26, 0.23588809370994568), (27, 0.4321473240852356)]
+
+ 10%|█         | 3/30 [00:20<03:17,  7.33s/it]Time step 976, Spatial MSE: [(0, 0.00015336331853177398), (1, 0.0006216293550096452), (2, 0.0018829216714948416), (3, 0.001544506405480206), (4, 0.001564358244650066), (5, 0.002618364989757538), (6, 0.004962182603776455), (7, 0.010810772888362408), (8, 0.008766102604568005), (9, 0.021245485171675682), (10, 0.026581019163131714), (11, 0.03631715103983879), (12, 0.03270844370126724), (13, 0.03055417723953724), (14, 0.06116124615073204), (15, 0.06362449377775192), (16, 0.08780785650014877), (17, 0.06145700812339783), (18, 0.06031078100204468), (19, 0.04725087061524391), (20, 0.08658777922391891), (21, 0.06124704331159592), (22, 0.05246401205658913), (23, 0.05606619268655777), (24, 0.17717622220516205), (25, 0.2572416067123413), (26, 0.17763780057430267), (27, 0.06821934133768082)], Temporal MSE: [(0, 3.041455329366727e-06), (1, 5.14489620400127e-05), (2, 0.00011496864317450672), (3, 0.00012133639393141493), (4, 0.00028484230278991163), (5, 0.0005199707811698318), (6, 0.0006647397531196475), (7, 0.0037756438832730055), (8, 0.0016282035503536463), (9, 0.0018898691050708294), (10, 0.002523945178836584), (11, 0.003381128190085292), (12, 0.003261722158640623), (13, 0.0029590651392936707), (14, 0.01135585829615593), (15, 0.013771274127066135), (16, 0.012212591245770454), (17, 0.009206471964716911), (18, 0.007557014934718609), (19, 0.011513215489685535), (20, 0.014495196752250195), (21, 0.024988828226923943), (22, 0.030945951119065285), (23, 0.03624027594923973), (24, 0.019223224371671677), (25, 0.015010557137429714), (26, 0.023803047835826874), (27, 0.03061041049659252)]
+
+ 13%|█▎        | 4/30 [00:29<03:25,  7.90s/it]Time step 968, Spatial MSE: [(0, 9.439918358111754e-05), (1, 0.000379890319891274), (2, 0.0011350985150784254), (3, 0.0006242848467081785), (4, 0.0008362907683476806), (5, 0.0012630976270884275), (6, 0.0026634775567799807), (7, 0.007043500896543264), (8, 0.006097034551203251), (9, 0.01498443353921175), (10, 0.022842135280370712), (11, 0.02436588518321514), (12, 0.02276080846786499), (13, 0.021762002259492874), (14, 0.05840260162949562), (15, 0.05435919761657715), (16, 0.05283595621585846), (17, 0.05003209784626961), (18, 0.04891776666045189), (19, 0.032944727689027786), (20, 0.042887989431619644), (21, 0.04282194375991821), (22, 0.02997012622654438), (23, 0.03310088813304901), (24, 0.07254929840564728), (25, 0.118197001516819), (26, 0.10636322945356369), (27, 0.03880079463124275)], Temporal MSE: [(0, 1.7040034663295955e-06), (1, 2.1999641830916516e-05), (2, 7.112971070455387e-05), (3, 7.027926039882004e-05), (4, 0.00013982347445562482), (5, 0.0002553972299210727), (6, 0.00033750085276551545), (7, 0.0013773672981187701), (8, 0.0008377443300560117), (9, 0.0009799335384741426), (10, 0.0017691776156425476), (11, 0.0024519481230527163), (12, 0.0023483247496187687), (13, 0.0020149319898337126), (14, 0.009848791174590588), (15, 0.009714781306684017), (16, 0.00809292122721672), (17, 0.0075364066287875175), (18, 0.006639313418418169), (19, 0.007751127704977989), (20, 0.010610262863337994), (21, 0.018268289044499397), (22, 0.017253631725907326), (23, 0.016338421031832695), (24, 0.009715263731777668), (25, 0.012136113829910755), (26, 0.011044475249946117), (27, 0.0100288400426507)]
+
+ 17%|█▋        | 5/30 [00:37<03:21,  8.08s/it]Time step 960, Spatial MSE: [(0, 6.150140688987449e-05), (1, 0.0004956214688718319), (2, 0.0010287101613357663), (3, 0.0006519025773741305), (4, 0.0006459100404754281), (5, 0.0010938356863334775), (6, 0.0022785947658121586), (7, 0.007147985510528088), (8, 0.006452299654483795), (9, 0.014480656944215298), (10, 0.03079707734286785), (11, 0.02883915789425373), (12, 0.02370530739426613), (13, 0.02415752410888672), (14, 0.11465094238519669), (15, 0.0882720947265625), (16, 0.09421029686927795), (17, 0.07339563220739365), (18, 0.06508838385343552), (19, 0.05040021240711212), (20, 0.06755458563566208), (21, 0.048732466995716095), (22, 0.04239502176642418), (23, 0.04272999241948128), (24, 0.07130485773086548), (25, 0.1086956113576889), (26, 0.08048500120639801), (27, 0.020634744316339493)], Temporal MSE: [(0, 1.5238264268191415e-06), (1, 1.2799395335605368e-05), (2, 5.849607623531483e-05), (3, 6.278374348767102e-05), (4, 0.00011688457016134635), (5, 0.0002259458851767704), (6, 0.00036203558556735516), (7, 0.0013775143306702375), (8, 0.000794340914580971), (9, 0.0011411580489948392), (10, 0.003466737223789096), (11, 0.0039441874250769615), (12, 0.0042046597227454185), (13, 0.003736633574590087), (14, 0.02282056212425232), (15, 0.017609866335988045), (16, 0.017508869990706444), (17, 0.018019992858171463), (18, 0.01376771368086338), (19, 0.01297669019550085), (20, 0.01890135370194912), (21, 0.027106501162052155), (22, 0.022398078814148903), (23, 0.019612036645412445), (24, 0.011795259080827236), (25, 0.015235979110002518), (26, 0.01004492212086916), (27, 0.007663646247237921)]
+
+ 20%|██        | 6/30 [00:45<03:12,  8.01s/it]Time step 952, Spatial MSE: [(0, 5.361554940463975e-05), (1, 0.00034791254438459873), (2, 0.0007800057646818459), (3, 0.00034645653795450926), (4, 0.00043823797022923827), (5, 0.000823954411316663), (6, 0.0015679739881306887), (7, 0.005251678172498941), (8, 0.005336354020982981), (9, 0.010330386459827423), (10, 0.02513865940272808), (11, 0.023198438808321953), (12, 0.018214574083685875), (13, 0.019378483295440674), (14, 0.10921107232570648), (15, 0.0786089077591896), (16, 0.0782180204987526), (17, 0.06114114448428154), (18, 0.05320853367447853), (19, 0.0439169779419899), (20, 0.05279826000332832), (21, 0.049809351563453674), (22, 0.04171392694115639), (23, 0.044588472694158554), (24, 0.06509406119585037), (25, 0.09598885476589203), (26, 0.07007868587970734), (27, 0.01732233539223671)], Temporal MSE: [(0, 1.1617497648330755e-06), (1, 8.647622053103987e-06), (2, 4.814756539417431e-05), (3, 5.223751941230148e-05), (4, 7.198601815616712e-05), (5, 0.0002094157534884289), (6, 0.0002866822178475559), (7, 0.0009173686848953366), (8, 0.0005930574261583388), (9, 0.0008826762204989791), (10, 0.003432224038988352), (11, 0.003607276128605008), (12, 0.0042685032822191715), (13, 0.003917216323316097), (14, 0.020679911598563194), (15, 0.017303554341197014), (16, 0.01746818795800209), (17, 0.01770246960222721), (18, 0.012672134675085545), (19, 0.01266069058328867), (20, 0.02251124195754528), (21, 0.02614591270685196), (22, 0.023370714858174324), (23, 0.027262233197689056), (24, 0.0187698882073164), (25, 0.01736902818083763), (26, 0.0095700453966856), (27, 0.008272431790828705)]
+
+ 23%|██▎       | 7/30 [00:53<03:06,  8.10s/it]Time step 940, Spatial MSE: [(0, 6.03765802225098e-05), (1, 0.0003484303306322545), (2, 0.0007011254783719778), (3, 0.0003679518122226), (4, 0.000702829216606915), (5, 0.0007610588800162077), (6, 0.0016188243171200156), (7, 0.005166557617485523), (8, 0.004094735253602266), (9, 0.007613821886479855), (10, 0.013308119960129261), (11, 0.011832012794911861), (12, 0.011130031198263168), (13, 0.014105569571256638), (14, 0.03298427164554596), (15, 0.03620309755206108), (16, 0.03685212880373001), (17, 0.03653682395815849), (18, 0.038350678980350494), (19, 0.031394533812999725), (20, 0.034004390239715576), (21, 0.04166116192936897), (22, 0.03581806644797325), (23, 0.03221210464835167), (24, 0.052099235355854034), (25, 0.08197928220033646), (26, 0.05232112854719162), (27, 0.027782581746578217)], Temporal MSE: [(0, 1.328697067037865e-06), (1, 9.793664503376931e-06), (2, 4.9321188271278515e-05), (3, 3.404663584660739e-05), (4, 9.027819760376588e-05), (5, 0.00023287955264095217), (6, 0.00024312375171575695), (7, 0.0007846899097785354), (8, 0.0005763032822869718), (9, 0.000610783405136317), (10, 0.0013721493305638433), (11, 0.0015887029003351927), (12, 0.0019014296121895313), (13, 0.001966731157153845), (14, 0.007185835856944323), (15, 0.007457448169589043), (16, 0.00846868846565485), (17, 0.010024257935583591), (18, 0.008075481280684471), (19, 0.008514247834682465), (20, 0.013674296438694), (21, 0.014139431528747082), (22, 0.013996320776641369), (23, 0.014317010529339314), (24, 0.008157608099281788), (25, 0.009285075590014458), (26, 0.006459754891693592), (27, 0.0068793040700256824)]
+
+ 27%|██▋       | 8/30 [01:01<02:57,  8.08s/it]Time step 932, Spatial MSE: [(0, 4.853667633142322e-05), (1, 0.00031508266692981124), (2, 0.0005127409240230918), (3, 0.0003036720445379615), (4, 0.00025586754782125354), (5, 0.00038740658783353865), (6, 0.0007469973061233759), (7, 0.0014276938745751977), (8, 0.0018404799047857523), (9, 0.0031796260736882687), (10, 0.004565428011119366), (11, 0.0059287892654538155), (12, 0.005688931327313185), (13, 0.006937946658581495), (14, 0.01532474160194397), (15, 0.015802979469299316), (16, 0.01729877106845379), (17, 0.017110414803028107), (18, 0.01748303510248661), (19, 0.01441105641424656), (20, 0.015485313721001148), (21, 0.015427463687956333), (22, 0.01639517955482006), (23, 0.014762557111680508), (24, 0.021789006888866425), (25, 0.035373225808143616), (26, 0.023533906787633896), (27, 0.010664153844118118)], Temporal MSE: [(0, 1.0337821549910586e-06), (1, 4.092561084689805e-06), (2, 2.7788821171270683e-05), (3, 5.216390491113998e-05), (4, 5.276425144984387e-05), (5, 0.00018651496793609113), (6, 0.00012780670658685267), (7, 0.00023429925204254687), (8, 0.00021679862402379513), (9, 0.00029061356326565146), (10, 0.0005890976754017174), (11, 0.000805178249720484), (12, 0.0008520016563124955), (13, 0.0009414622909389436), (14, 0.003150458447635174), (15, 0.003293232759460807), (16, 0.0037334284279495478), (17, 0.004574484657496214), (18, 0.0036860848776996136), (19, 0.003828203771263361), (20, 0.009557767771184444), (21, 0.00886134896427393), (22, 0.006819413509219885), (23, 0.00686602434143424), (24, 0.004641806706786156), (25, 0.007036679890006781), (26, 0.00323438155464828), (27, 0.00419794162735343)]
+
+ 30%|███       | 9/30 [01:10<02:56,  8.40s/it]Time step 920, Spatial MSE: [(0, 3.0216368031688035e-05), (1, 0.0003685164556372911), (2, 0.0005997284897603095), (3, 0.00017745789955370128), (4, 7.215950608951971e-05), (5, 0.00010081209620693699), (6, 0.0001623231073608622), (7, 0.0006160105112940073), (8, 0.00019599153893068433), (9, 0.0003391832287888974), (10, 0.0003569031541701406), (11, 0.00037936202716082335), (12, 0.0005563823506236076), (13, 0.0005239958991296589), (14, 0.0008322870125994086), (15, 0.0015881715808063745), (16, 0.0035579134710133076), (17, 0.0014381251530721784), (18, 0.0031437783036381006), (19, 0.0019905189983546734), (20, 0.0018025476019829512), (21, 0.002308507915586233), (22, 0.0017429308500140905), (23, 0.002671527210623026), (24, 0.00552708376199007), (25, 0.01288172509521246), (26, 0.010599687695503235), (27, 0.014165942557156086)], Temporal MSE: [(0, 8.098504622466862e-07), (1, 3.6141743748885347e-06), (2, 1.3456602573569398e-05), (3, 1.2734055417240597e-05), (4, 1.3647652849613223e-05), (5, 1.6308287740685046e-05), (6, 2.7447591492091306e-05), (7, 8.263249037554488e-05), (8, 3.478249709587544e-05), (9, 6.628547998843715e-05), (10, 0.00011106756574008614), (11, 9.611529821995646e-05), (12, 9.321337711298838e-05), (13, 8.22142101242207e-05), (14, 0.00023506073921453208), (15, 0.00082941516302526), (16, 0.0006832476938143373), (17, 0.0006535588763654232), (18, 0.001042622490786016), (19, 0.0012870789505541325), (20, 0.0017385308165103197), (21, 0.0009458628483116627), (22, 0.0006358418613672256), (23, 0.001175218727439642), (24, 0.0004985707346349955), (25, 0.0008150623179972172), (26, 0.0008490268373861909), (27, 0.0008265842334367335)]
+
+ 33%|███▎      | 10/30 [01:17<02:36,  7.80s/it]Time step 908, Spatial MSE: [(0, 9.509242227068171e-05), (1, 0.00047617952805012465), (2, 0.0007057977491058409), (3, 0.00025420685415156186), (4, 0.00040827057091519237), (5, 0.0005807100096717477), (6, 0.0009898844873532653), (7, 0.0016265589511021972), (8, 0.0015149138635024428), (9, 0.0024337698705494404), (10, 0.007334948517382145), (11, 0.004343520384281874), (12, 0.004828630015254021), (13, 0.004914384800940752), (14, 0.012014327570796013), (15, 0.009330361150205135), (16, 0.01622314192354679), (17, 0.00793695729225874), (18, 0.00831589289009571), (19, 0.0081692049279809), (20, 0.008873005397617817), (21, 0.01889447122812271), (22, 0.010176638141274452), (23, 0.010336729697883129), (24, 0.01691971905529499), (25, 0.038321737200021744), (26, 0.01482096966356039), (27, 0.014873327687382698)], Temporal MSE: [(0, 3.0455610158242052e-06), (1, 4.732666639029048e-06), (2, 5.184069595998153e-05), (3, 6.208147533470765e-05), (4, 4.413690840010531e-05), (5, 8.824731776257977e-05), (6, 8.494553185300902e-05), (7, 0.0004852493875660002), (8, 0.00041118351509794593), (9, 0.0003974122228100896), (10, 0.0008742761565372348), (11, 0.0006954980199225247), (12, 0.0010379822924733162), (13, 0.0005801091901957989), (14, 0.003115977393463254), (15, 0.002637816360220313), (16, 0.004021028988063335), (17, 0.002408882137387991), (18, 0.002690480323508382), (19, 0.0043357969261705875), (20, 0.0036470431368798018), (21, 0.004829904064536095), (22, 0.004729737062007189), (23, 0.006244862452149391), (24, 0.0038233515806496143), (25, 0.0029334204737097025), (26, 0.0048484355211257935), (27, 0.00365748954936862)]
+
+ 37%|███▋      | 11/30 [01:24<02:22,  7.50s/it]Time step 892, Spatial MSE: [(0, 3.029717117897235e-05), (1, 0.0004979415098205209), (2, 0.0007244907901622355), (3, 0.00019249423348810524), (4, 6.157048483146355e-05), (5, 9.985359793063253e-05), (6, 0.00013311783550307155), (7, 0.00044543997501023114), (8, 0.00020168803166598082), (9, 0.0003257238713558763), (10, 0.0003055417619179934), (11, 0.0004542501410469413), (12, 0.00038456919719465077), (13, 0.0004988949513062835), (14, 0.0008164707105606794), (15, 0.001307509490288794), (16, 0.0044824532233178616), (17, 0.0013117193011566997), (18, 0.001953810453414917), (19, 0.00303854001685977), (20, 0.0022596754133701324), (21, 0.0035993538331240416), (22, 0.0022938810288906097), (23, 0.0036888294853270054), (24, 0.00762246223166585), (25, 0.011141521856188774), (26, 0.012105639092624187), (27, 0.009779606945812702)], Temporal MSE: [(0, 7.343475090237916e-07), (1, 3.594108875404345e-06), (2, 1.4131085663393606e-05), (3, 1.2776934454450384e-05), (4, 1.3902014870836865e-05), (5, 1.735224577714689e-05), (6, 1.886952668428421e-05), (7, 6.496803689515218e-05), (8, 3.597532122512348e-05), (9, 5.1961629651486874e-05), (10, 0.00010089411080116406), (11, 8.295302541228011e-05), (12, 7.803873450029641e-05), (13, 7.316079427255318e-05), (14, 0.0002483875723555684), (15, 0.0006043986650183797), (16, 0.0007117661880329251), (17, 0.0006135260337032378), (18, 0.0005252906703390181), (19, 0.0018730007577687502), (20, 0.0028776959516108036), (21, 0.0010630199685692787), (22, 0.0007483293302357197), (23, 0.0010498821502551436), (24, 0.0006213642773218453), (25, 0.0007542569073848426), (26, 0.0008659164886921644), (27, 0.0008115506498143077)]
+
+ 40%|████      | 12/30 [01:30<02:10,  7.24s/it]Time step 880, Spatial MSE: [(0, 0.00010591484897304326), (1, 0.0004694298841059208), (2, 0.0016742725856602192), (3, 0.0011635585688054562), (4, 0.0018767419969663024), (5, 0.0029704526532441378), (6, 0.005148344207555056), (7, 0.012286431156098843), (8, 0.016024690121412277), (9, 0.026945998892188072), (10, 0.03611517697572708), (11, 0.03960656747221947), (12, 0.03823994845151901), (13, 0.04878726974129677), (14, 0.10314806550741196), (15, 0.1191357746720314), (16, 0.12716302275657654), (17, 0.10412541776895523), (18, 0.10791091620922089), (19, 0.10273478180170059), (20, 0.09657490998506546), (21, 0.11962651461362839), (22, 0.1264427900314331), (23, 0.12867847084999084), (24, 0.17469072341918945), (25, 0.2488117665052414), (26, 0.165242999792099), (27, 0.033838097006082535)], Temporal MSE: [(0, 2.9252394142531557e-06), (1, 2.825870251399465e-05), (2, 0.0005052629858255386), (3, 0.0005659870221279562), (4, 0.0006429996574297547), (5, 0.0015201978385448456), (6, 0.0020147343166172504), (7, 0.003871078137308359), (8, 0.002731101354584098), (9, 0.003036285052075982), (10, 0.005193457473069429), (11, 0.006361373234540224), (12, 0.006759589537978172), (13, 0.007445731665939093), (14, 0.02451108582317829), (15, 0.027229880914092064), (16, 0.02649739384651184), (17, 0.027747459709644318), (18, 0.024291567504405975), (19, 0.028789082542061806), (20, 0.054808616638183594), (21, 0.0629361942410469), (22, 0.046646129339933395), (23, 0.054698146879673004), (24, 0.03280166909098625), (25, 0.039206612855196), (26, 0.018753716722130775), (27, 0.030748853459954262)]
+
+ 43%|████▎     | 13/30 [01:39<02:08,  7.58s/it]Time step 864, Spatial MSE: [(0, 2.147401391994208e-05), (1, 0.0004332208773121238), (2, 0.0006766290753148496), (3, 0.00021900788124185055), (4, 5.686954318662174e-05), (5, 7.938456838019192e-05), (6, 0.0001474883029004559), (7, 0.0002713091089390218), (8, 0.00014139682753011584), (9, 0.00019676791271194816), (10, 0.0003301821416243911), (11, 0.0003250276786275208), (12, 0.00034788253833539784), (13, 0.0005052448832429945), (14, 0.0008248378289863467), (15, 0.0009781424887478352), (16, 0.0057327826507389545), (17, 0.0012116720899939537), (18, 0.0013648528838530183), (19, 0.00475630396977067), (20, 0.0034758730325847864), (21, 0.011316174641251564), (22, 0.010487333871424198), (23, 0.004775235429406166), (24, 0.007324282079935074), (25, 0.013030742295086384), (26, 0.012604977935552597), (27, 0.013147342018783092)], Temporal MSE: [(0, 5.780819378742308e-07), (1, 2.711472461669473e-06), (2, 1.0014266081270762e-05), (3, 1.0267916877637617e-05), (4, 1.0667241440387443e-05), (5, 1.5505143892369233e-05), (6, 1.7109674445237033e-05), (7, 3.752925476874225e-05), (8, 2.4884468075470068e-05), (9, 3.6231998819857836e-05), (10, 7.938449562061578e-05), (11, 6.067720823921263e-05), (12, 5.249835885479115e-05), (13, 5.256706936052069e-05), (14, 0.00024910399224609137), (15, 0.00040791972423903644), (16, 0.0008609601645730436), (17, 0.000485891941934824), (18, 0.0002506015880499035), (19, 0.0020901698153465986), (20, 0.003673097351565957), (21, 0.0030586409848183393), (22, 0.003109870245680213), (23, 0.0007880594930611551), (24, 0.0006751224282197654), (25, 0.0008183644968084991), (26, 0.0007112413877621293), (27, 0.0013038943288847804)]
+
+ 47%|████▋     | 14/30 [01:45<01:55,  7.24s/it]Time step 848, Spatial MSE: [(0, 0.00012302456889301538), (1, 0.0004394315183162689), (2, 0.0006017314153723419), (3, 0.00024176390434149653), (4, 0.00032029824797064066), (5, 0.0007125171832740307), (6, 0.001189786591567099), (7, 0.0016252055065706372), (8, 0.0014525933656841516), (9, 0.00241361022926867), (10, 0.007668148260563612), (11, 0.0041643595322966576), (12, 0.004355946555733681), (13, 0.005234408192336559), (14, 0.013494880869984627), (15, 0.014076583087444305), (16, 0.018799511715769768), (17, 0.00840887613594532), (18, 0.011150052770972252), (19, 0.010814104229211807), (20, 0.01118440367281437), (21, 0.010907895863056183), (22, 0.011286710388958454), (23, 0.010296966880559921), (24, 0.019597485661506653), (25, 0.0323801152408123), (26, 0.018594877794384956), (27, 0.011116418987512589)], Temporal MSE: [(0, 2.5055883270397317e-06), (1, 4.69940505354316e-06), (2, 2.292390308866743e-05), (3, 5.0266880862182006e-05), (4, 7.455697050318122e-05), (5, 0.00018899681163020432), (6, 0.0001670449273660779), (7, 0.0009775827638804913), (8, 0.000654240429867059), (9, 0.0006519628805108368), (10, 0.0018824231810867786), (11, 0.0011811621952801943), (12, 0.001724732923321426), (13, 0.000944194383919239), (14, 0.005373872350901365), (15, 0.007517422083765268), (16, 0.00845480989664793), (17, 0.0056164683774113655), (18, 0.004595837090164423), (19, 0.008325236849486828), (20, 0.010168683715164661), (21, 0.006196166388690472), (22, 0.0040143695659935474), (23, 0.005159294232726097), (24, 0.0032355652656406164), (25, 0.0019468258833512664), (26, 0.004247927572578192), (27, 0.0028757916297763586)]
+
+ 50%|█████     | 15/30 [01:53<01:50,  7.39s/it]Time step 828, Spatial MSE: [(0, 1.9355973563506268e-05), (1, 0.00048663723282516), (2, 0.000817656225990504), (3, 0.0002608614740893245), (4, 4.471141437534243e-05), (5, 7.500940409954637e-05), (6, 8.443255501333624e-05), (7, 0.0002893189084716141), (8, 0.00012671522563323379), (9, 0.0001805633510230109), (10, 0.00021563246264122427), (11, 0.0002342494553886354), (12, 0.00035679887514561415), (13, 0.00037376495311036706), (14, 0.0006242422387003899), (15, 0.0006648741546086967), (16, 0.003550231223925948), (17, 0.000888713460881263), (18, 0.0010301601141691208), (19, 0.0026302230544388294), (20, 0.0017649559304118156), (21, 0.0032065745908766985), (22, 0.0022506334353238344), (23, 0.003588914405554533), (24, 0.006482634227722883), (25, 0.013180438429117203), (26, 0.011829703114926815), (27, 0.017565352842211723)], Temporal MSE: [(0, 5.427534119917254e-07), (1, 1.9964295461250003e-06), (2, 9.800049156183377e-06), (3, 8.737004463910125e-06), (4, 1.046000761562027e-05), (5, 1.7963500795303844e-05), (6, 1.6543286619707942e-05), (7, 4.254282248439267e-05), (8, 2.3598195184604265e-05), (9, 3.380673661013134e-05), (10, 5.689177487511188e-05), (11, 4.778347647516057e-05), (12, 6.223577656783164e-05), (13, 4.776400965056382e-05), (14, 0.00016894823056645691), (15, 0.00029611398349516094), (16, 0.000537662417627871), (17, 0.00039529503555968404), (18, 0.00023856788175180554), (19, 0.0015366243897005916), (20, 0.0023785566445440054), (21, 0.001104485709220171), (22, 0.0006693326868116856), (23, 0.0010887682437896729), (24, 0.0006250038859434426), (25, 0.00078302260953933), (26, 0.0006077402504161), (27, 0.0010947005357593298)]
+
+ 53%|█████▎    | 16/30 [02:00<01:42,  7.30s/it]Time step 808, Spatial MSE: [(0, 9.715570922708139e-05), (1, 0.0007649129838682711), (2, 0.0008922307752072811), (3, 0.0009733223705552518), (4, 0.0014681665925309062), (5, 0.0029175658710300922), (6, 0.004783587995916605), (7, 0.009141488932073116), (8, 0.013325309380888939), (9, 0.02188889868557453), (10, 0.027767477557063103), (11, 0.032832346856594086), (12, 0.029771534726023674), (13, 0.033956557512283325), (14, 0.08933307230472565), (15, 0.11402380466461182), (16, 0.10099964588880539), (17, 0.12224329262971878), (18, 0.12626031041145325), (19, 0.1151663064956665), (20, 0.10453478246927261), (21, 0.12355153262615204), (22, 0.15343435108661652), (23, 0.14626052975654602), (24, 0.17291590571403503), (25, 0.22445203363895416), (26, 0.1712362915277481), (27, 0.0330142043530941)], Temporal MSE: [(0, 2.8667698188655777e-06), (1, 2.3168016923591495e-05), (2, 0.0001922440860653296), (3, 0.0004590865282807499), (4, 0.0005884628626517951), (5, 0.0014782488578930497), (6, 0.0019145285477861762), (7, 0.0036834452766925097), (8, 0.0024803834967315197), (9, 0.00289977272041142), (10, 0.0047577861696481705), (11, 0.0056564658880233765), (12, 0.005723684560507536), (13, 0.006662198342382908), (14, 0.0213444996625185), (15, 0.02815309725701809), (16, 0.028909888118505478), (17, 0.039899639785289764), (18, 0.03135574609041214), (19, 0.037715453654527664), (20, 0.06786279380321503), (21, 0.07852239161729813), (22, 0.053987860679626465), (23, 0.05920291692018509), (24, 0.03849688544869423), (25, 0.024707144126296043), (26, 0.016735341399908066), (27, 0.03142837807536125)]
+
+ 57%|█████▋    | 17/30 [02:08<01:39,  7.65s/it]Time step 788, Spatial MSE: [(0, 1.999501182581298e-05), (1, 0.0004473910084925592), (2, 0.0009601078345440328), (3, 0.00030259962659329176), (4, 4.47246857220307e-05), (5, 7.008993270574138e-05), (6, 7.996979547897354e-05), (7, 0.00036580872256308794), (8, 0.00016412265540566295), (9, 0.00015976002032402903), (10, 0.00019820993475150317), (11, 0.0002626570640131831), (12, 0.0003083083429373801), (13, 0.00040994546725414693), (14, 0.0006773190107196569), (15, 0.0006312938639894128), (16, 0.0049457554705441), (17, 0.0011795511236414313), (18, 0.00712653947994113), (19, 0.002884550951421261), (20, 0.001622198848053813), (21, 0.0026095861103385687), (22, 0.0023875944316387177), (23, 0.0035383705981075764), (24, 0.006095863878726959), (25, 0.016034360975027084), (26, 0.01322214026004076), (27, 0.010100215673446655)], Temporal MSE: [(0, 5.675631200574571e-07), (1, 1.8932951206807047e-06), (2, 1.0738172932178713e-05), (3, 1.0399137863714714e-05), (4, 1.0744713108579163e-05), (5, 2.026281799771823e-05), (6, 1.953112041519489e-05), (7, 3.606963946367614e-05), (8, 2.4110097001539543e-05), (9, 3.278478470747359e-05), (10, 6.465752085205168e-05), (11, 4.8856578359846026e-05), (12, 4.5562457671621814e-05), (13, 5.486961890710518e-05), (14, 0.00020172176300548017), (15, 0.0002123580634361133), (16, 0.00060503144050017), (17, 0.000315710436552763), (18, 0.001853073830716312), (19, 0.0024879095144569874), (20, 0.004460978787392378), (21, 0.0013989908620715141), (22, 0.0006380881532095373), (23, 0.0008467663428746164), (24, 0.0006167524261400104), (25, 0.0010670229094102979), (26, 0.000772313098423183), (27, 0.0008615806582383811)]
+
+ 60%|██████    | 18/30 [02:16<01:31,  7.64s/it]Time step 764, Spatial MSE: [(0, 0.0001300215517403558), (1, 0.0006270869635045528), (2, 0.0007547324639745057), (3, 0.00024420651607215405), (4, 0.00020885858975816518), (5, 0.0005189061630517244), (6, 0.0008204976329579949), (7, 0.0013354868860915303), (8, 0.0012576893204823136), (9, 0.002314011799171567), (10, 0.005554765462875366), (11, 0.0031605029944330454), (12, 0.004429491702467203), (13, 0.005250703543424606), (14, 0.01468243170529604), (15, 0.014796409755945206), (16, 0.019348019734025), (17, 0.009864999912679195), (18, 0.010761124081909657), (19, 0.014276635833084583), (20, 0.010345511138439178), (21, 0.012798259034752846), (22, 0.013733483850955963), (23, 0.016198890283703804), (24, 0.02414800599217415), (25, 0.04672051593661308), (26, 0.023275677114725113), (27, 0.010508891195058823)], Temporal MSE: [(0, 2.6470377179066418e-06), (1, 5.57865996597684e-06), (2, 2.0970768673578277e-05), (3, 3.2645642932038754e-05), (4, 4.592249388224445e-05), (5, 0.00014616624685004354), (6, 0.00013673919602297246), (7, 0.0006980548496358097), (8, 0.00044747008359991014), (9, 0.0005600154981948435), (10, 0.0017780981725081801), (11, 0.001398045918904245), (12, 0.001968214986845851), (13, 0.0011684555793181062), (14, 0.005380087997764349), (15, 0.006431278772652149), (16, 0.006858671549707651), (17, 0.004782320931553841), (18, 0.007505456451326609), (19, 0.009672633372247219), (20, 0.007847867906093597), (21, 0.008033137768507004), (22, 0.005247682798653841), (23, 0.006906475406140089), (24, 0.00649618124589324), (25, 0.0022694936487823725), (26, 0.004580333363264799), (27, 0.005483245011419058)]
+
+ 63%|██████▎   | 19/30 [02:25<01:27,  7.93s/it]Time step 736, Spatial MSE: [(0, 2.872337972803507e-05), (1, 0.0007639742107130587), (2, 0.0013826512731611729), (3, 0.00048021331895142794), (4, 7.025980448815972e-05), (5, 9.986494114855304e-05), (6, 0.00011669505329336971), (7, 0.0003028700884897262), (8, 0.00015183880168478936), (9, 0.0002340789942536503), (10, 0.00025604854454286397), (11, 0.00036789485602639616), (12, 0.0004098247445654124), (13, 0.0004981318488717079), (14, 0.0008805664838291705), (15, 0.0011678121518343687), (16, 0.006348222494125366), (17, 0.0018183154752478004), (18, 0.001982309389859438), (19, 0.0031856815330684185), (20, 0.002203685697168112), (21, 0.004857573192566633), (22, 0.00309884175658226), (23, 0.004985024221241474), (24, 0.00818789191544056), (25, 0.02062276192009449), (26, 0.017736664041876793), (27, 0.011590633541345596)], Temporal MSE: [(0, 7.653004558960674e-07), (1, 2.8000342808809364e-06), (2, 1.655722371651791e-05), (3, 1.5029809219413437e-05), (4, 1.6513538867002353e-05), (5, 2.4110544472932816e-05), (6, 3.0067283660173416e-05), (7, 4.458759576664306e-05), (8, 3.2183572329813614e-05), (9, 4.490279025048949e-05), (10, 7.379827002296224e-05), (11, 6.635827594436705e-05), (12, 6.795230729039758e-05), (13, 7.447852840414271e-05), (14, 0.0003022408054675907), (15, 0.0005999549757689238), (16, 0.0008622687892057002), (17, 0.0007508546113967896), (18, 0.0005312928115017712), (19, 0.0018619209295138717), (20, 0.004622180946171284), (21, 0.0018326853169128299), (22, 0.0007712184451520443), (23, 0.001123597496189177), (24, 0.0008785545360296965), (25, 0.0016712270444259048), (26, 0.0009821276180446148), (27, 0.0014346352545544505)]
+
+ 67%|██████▋   | 20/30 [02:34<01:23,  8.38s/it]Time step 708, Spatial MSE: [(0, 6.782508717151359e-05), (1, 0.0013034400762990117), (2, 0.001243468257598579), (3, 0.001469548442400992), (4, 0.001531515852548182), (5, 0.0046812319196760654), (6, 0.0049015008844435215), (7, 0.007882541976869106), (8, 0.012972320429980755), (9, 0.018038872629404068), (10, 0.020752357318997383), (11, 0.02418331801891327), (12, 0.02758166380226612), (13, 0.029721425846219063), (14, 0.06450669467449188), (15, 0.07239021360874176), (16, 0.08195221424102783), (17, 0.07259180396795273), (18, 0.0885503813624382), (19, 0.08439262211322784), (20, 0.08427786082029343), (21, 0.11667878925800323), (22, 0.1259133219718933), (23, 0.12458793073892593), (24, 0.15359579026699066), (25, 0.23861195147037506), (26, 0.16472284495830536), (27, 0.02828434854745865)], Temporal MSE: [(0, 4.261847152520204e-06), (1, 2.802473318297416e-05), (2, 0.0001644104631850496), (3, 0.0010144903790205717), (4, 0.0008642636239528656), (5, 0.002615860430523753), (6, 0.0030171643011271954), (7, 0.0036731925792992115), (8, 0.0029359550680965185), (9, 0.004361739382147789), (10, 0.005299106240272522), (11, 0.006849012803286314), (12, 0.0059209903702139854), (13, 0.0061991410329937935), (14, 0.021610163152217865), (15, 0.02329954318702221), (16, 0.025190891698002815), (17, 0.02613387256860733), (18, 0.026342378929257393), (19, 0.031515005975961685), (20, 0.0590619258582592), (21, 0.06323698908090591), (22, 0.03948062285780907), (23, 0.04619117081165314), (24, 0.041293174028396606), (25, 0.03485138341784477), (26, 0.02018336020410061), (27, 0.03348328545689583)]
+
+ 70%|███████   | 21/30 [02:44<01:19,  8.85s/it]Time step 676, Spatial MSE: [(0, 3.0449184123426676e-05), (1, 0.0009834772208705544), (2, 0.001884161843918264), (3, 0.0006575463921763003), (4, 9.086721547646448e-05), (5, 0.0001245064486283809), (6, 0.00014699946041218936), (7, 0.0003201246727257967), (8, 0.00017140066483989358), (9, 0.00018783776613418013), (10, 0.00025744060985744), (11, 0.0003034847031813115), (12, 0.00037134578451514244), (13, 0.0005267211818136275), (14, 0.0009014878305606544), (15, 0.001149884657934308), (16, 0.006634208839386702), (17, 0.001410498283803463), (18, 0.001911604544147849), (19, 0.002260778332129121), (20, 0.0014946794835850596), (21, 0.0030300263315439224), (22, 0.003133679972961545), (23, 0.0055841319262981415), (24, 0.007628964260220528), (25, 0.029581844806671143), (26, 0.023701781406998634), (27, 0.013475598767399788)], Temporal MSE: [(0, 1.0672413282009074e-06), (1, 3.540922079992015e-06), (2, 1.8533566617406905e-05), (3, 2.4897946786950342e-05), (4, 2.290226984769106e-05), (5, 3.867125997203402e-05), (6, 4.650575647247024e-05), (7, 6.6874832555186e-05), (8, 4.150109089096077e-05), (9, 5.918503666180186e-05), (10, 9.541770850773901e-05), (11, 7.358188304351643e-05), (12, 7.063947123242542e-05), (13, 8.26537434477359e-05), (14, 0.00038686933112330735), (15, 0.00048698490718379617), (16, 0.0007082172087393701), (17, 0.0005958707188256085), (18, 0.0003207252884749323), (19, 0.0016176372300833464), (20, 0.003163684858009219), (21, 0.0011792643927037716), (22, 0.0008341945940628648), (23, 0.001327931066043675), (24, 0.0009559456375427544), (25, 0.0026299995370209217), (26, 0.0012163221836090088), (27, 0.0018288419814780354)]
+
+ 73%|███████▎  | 22/30 [02:52<01:09,  8.69s/it]Time step 636, Spatial MSE: [(0, 8.884839917300269e-05), (1, 0.0011850459268316627), (2, 0.0014848688151687384), (3, 0.0003967272932641208), (4, 0.00026739222812466323), (5, 0.0007987424614839256), (6, 0.0013375174021348357), (7, 0.0026319283060729504), (8, 0.001787420827895403), (9, 0.002714733360335231), (10, 0.008026114664971828), (11, 0.004296166356652975), (12, 0.005312157794833183), (13, 0.006592049729079008), (14, 0.018825536593794823), (15, 0.026059800758957863), (16, 0.03532370924949646), (17, 0.016332078725099564), (18, 0.01850861683487892), (19, 0.03089798241853714), (20, 0.017879437655210495), (21, 0.027360418811440468), (22, 0.019279561936855316), (23, 0.024360759183764458), (24, 0.04446272552013397), (25, 0.09590687602758408), (26, 0.04239613190293312), (27, 0.02437220700085163)], Temporal MSE: [(0, 2.4451549052173505e-06), (1, 8.462996447633486e-06), (2, 2.9371560231084004e-05), (3, 7.415122672682628e-05), (4, 0.00013393996050581336), (5, 0.0002586146292742342), (6, 0.00038318667793646455), (7, 0.001730239950120449), (8, 0.0009941532043740153), (9, 0.0012267015408724546), (10, 0.0038864193484187126), (11, 0.0025056772865355015), (12, 0.003813770366832614), (13, 0.0018450760981068015), (14, 0.010918088257312775), (15, 0.016908369958400726), (16, 0.018269874155521393), (17, 0.011890679597854614), (18, 0.008107944391667843), (19, 0.016784431412816048), (20, 0.010091738775372505), (21, 0.010188089683651924), (22, 0.008002359420061111), (23, 0.008250515908002853), (24, 0.008017092011868954), (25, 0.0032298278529196978), (26, 0.007085667923092842), (27, 0.005616535432636738)]
+
+ 77%|███████▋  | 23/30 [03:01<01:00,  8.67s/it]Time step 596, Spatial MSE: [(0, 3.9674254367128015e-05), (1, 0.001413350342772901), (2, 0.0028661596588790417), (3, 0.001033651060424745), (4, 0.00014871767780277878), (5, 0.00020033259352203459), (6, 0.00025887208175845444), (7, 0.00035523378755897284), (8, 0.00026960266404785216), (9, 0.00025952814030461013), (10, 0.0003982136840932071), (11, 0.00046453296090476215), (12, 0.0004907887778244913), (13, 0.001022716169245541), (14, 0.0016915415180847049), (15, 0.0015383393038064241), (16, 0.016734527423977852), (17, 0.0033693790901452303), (18, 0.004354847129434347), (19, 0.0052086785435676575), (20, 0.004297838546335697), (21, 0.011997104622423649), (22, 0.013711933977901936), (23, 0.012423683889210224), (24, 0.013337346725165844), (25, 0.05172620341181755), (26, 0.042859066277742386), (27, 0.025436388328671455)], Temporal MSE: [(0, 2.3509489892603597e-06), (1, 5.983802566333907e-06), (2, 3.0414525099331513e-05), (3, 4.0164050005842e-05), (4, 4.161779725109227e-05), (5, 5.656779831042513e-05), (6, 7.647152233403176e-05), (7, 0.0001067703269654885), (8, 7.057654875097796e-05), (9, 9.423994924873114e-05), (10, 0.00013089165440760553), (11, 0.0001288275670958683), (12, 0.0001219665864482522), (13, 0.00015069948858581483), (14, 0.0004813902487512678), (15, 0.0006873604143038392), (16, 0.0014298503519967198), (17, 0.0011354919988662004), (18, 0.0006141565390862525), (19, 0.0030016452074050903), (20, 0.006866433657705784), (21, 0.003987119533121586), (22, 0.0026394713204354048), (23, 0.001959825400263071), (24, 0.0016449644463136792), (25, 0.004523040261119604), (26, 0.002421073615550995), (27, 0.004621571395546198)]
+
+ 80%|████████  | 24/30 [03:10<00:52,  8.79s/it]Time step 548, Spatial MSE: [(0, 0.00010579497029539198), (1, 0.002727504586800933), (2, 0.00393613101914525), (3, 0.0022795216646045446), (4, 0.0021679613273590803), (5, 0.005855630151927471), (6, 0.006630100775510073), (7, 0.010300339199602604), (8, 0.015793347731232643), (9, 0.02127358876168728), (10, 0.028501903638243675), (11, 0.03350154682993889), (12, 0.035666536539793015), (13, 0.03637171909213066), (14, 0.10941813141107559), (15, 0.113444022834301), (16, 0.122264564037323), (17, 0.10257257521152496), (18, 0.11905539035797119), (19, 0.12535136938095093), (20, 0.12466393411159515), (21, 0.17657673358917236), (22, 0.23018525540828705), (23, 0.18826521933078766), (24, 0.24065515398979187), (25, 0.5174386501312256), (26, 0.3247546851634979), (27, 0.057328250259160995)], Temporal MSE: [(0, 1.0049101547338068e-05), (1, 5.167670678929426e-05), (2, 0.00017630428192205727), (3, 0.0010685115121304989), (4, 0.0013891024282202125), (5, 0.0034522952046245337), (6, 0.00549671333283186), (7, 0.005718606058508158), (8, 0.00418581860139966), (9, 0.006640988867729902), (10, 0.007783621549606323), (11, 0.01022180076688528), (12, 0.009218459017574787), (13, 0.009105721488595009), (14, 0.0375622920691967), (15, 0.03688031807541847), (16, 0.04180686175823212), (17, 0.03925160691142082), (18, 0.03364172577857971), (19, 0.04368393123149872), (20, 0.09439757466316223), (21, 0.10471779108047485), (22, 0.06001080572605133), (23, 0.0662066861987114), (24, 0.07821150869131088), (25, 0.06751782447099686), (26, 0.042572520673274994), (27, 0.05573149770498276)]
+
+ 83%|████████▎ | 25/30 [03:22<00:48,  9.69s/it]Time step 492, Spatial MSE: [(0, 6.952826515771449e-05), (1, 0.00270290719345212), (2, 0.004629991948604584), (3, 0.002050834009423852), (4, 0.0002589280193205923), (5, 0.00041946041164919734), (6, 0.0004585050046443939), (7, 0.001045917277224362), (8, 0.0005943129654042423), (9, 0.000547426228877157), (10, 0.0006303220870904624), (11, 0.000727712525986135), (12, 0.0011408152058720589), (13, 0.0014666932402178645), (14, 0.0018560101743787527), (15, 0.0021584925707429647), (16, 0.019630786031484604), (17, 0.003903832519426942), (18, 0.012970556505024433), (19, 0.005716562271118164), (20, 0.005931837484240532), (21, 0.01566639170050621), (22, 0.01071259006857872), (23, 0.018184959888458252), (24, 0.024501396343111992), (25, 0.10717813670635223), (26, 0.08903416246175766), (27, 0.044233016669750214)], Temporal MSE: [(0, 4.159758645982947e-06), (1, 9.873770068224985e-06), (2, 6.148573447717354e-05), (3, 9.920257434714586e-05), (4, 9.511779353488237e-05), (5, 0.0001358505105599761), (6, 0.00020244778716005385), (7, 0.0002878799568861723), (8, 0.0001517225755378604), (9, 0.00020215038966853172), (10, 0.00023891875753179193), (11, 0.00022007471125107259), (12, 0.00021941857994534075), (13, 0.00026425582473166287), (14, 0.0007200362160801888), (15, 0.0008329068659804761), (16, 0.0016923091607168317), (17, 0.00133738131262362), (18, 0.003045021090656519), (19, 0.005925843026489019), (20, 0.016781499609351158), (21, 0.0060525625012815), (22, 0.002415427239611745), (23, 0.0037157947663217783), (24, 0.0035637200344353914), (25, 0.009495748206973076), (26, 0.004064579494297504), (27, 0.009404491633176804)]
+
+ 87%|████████▋ | 26/30 [03:32<00:39,  9.85s/it]Time step 426, Spatial MSE: [(0, 0.00021266323165036738), (1, 0.00458354689180851), (2, 0.0066292486153542995), (3, 0.0017421022057533264), (4, 0.0013562676031142473), (5, 0.0025909114629030228), (6, 0.005631789565086365), (7, 0.01187109760940075), (8, 0.011318355798721313), (9, 0.014287762343883514), (10, 0.03310312330722809), (11, 0.02211698703467846), (12, 0.024346742779016495), (13, 0.021879613399505615), (14, 0.07364547997713089), (15, 0.08773967623710632), (16, 0.09186311066150665), (17, 0.07140286266803741), (18, 0.07778078317642212), (19, 0.09863060712814331), (20, 0.07321775704622269), (21, 0.11220751702785492), (22, 0.13431376218795776), (23, 0.14219267666339874), (24, 0.23226231336593628), (25, 0.6185866594314575), (26, 0.3670441806316376), (27, 0.09933706372976303)], Temporal MSE: [(0, 1.1459323104645591e-05), (1, 4.123920007259585e-05), (2, 0.00015100954624358565), (3, 0.0002470930339768529), (4, 0.0009231855510734022), (5, 0.0008968045585788786), (6, 0.00441311951726675), (7, 0.008721201680600643), (8, 0.003605607897043228), (9, 0.006030161865055561), (10, 0.01060553453862667), (11, 0.010024390183389187), (12, 0.007885720580816269), (13, 0.006508928257972002), (14, 0.03797720745205879), (15, 0.03848882392048836), (16, 0.03774772956967354), (17, 0.03324015066027641), (18, 0.023367110639810562), (19, 0.029772333800792694), (20, 0.04943424090743065), (21, 0.06690891087055206), (22, 0.03844597935676575), (23, 0.04105782136321068), (24, 0.07566231489181519), (25, 0.06663431227207184), (26, 0.04960610345005989), (27, 0.061920344829559326)]
+
+ 90%|█████████ | 27/30 [03:44<00:31, 10.55s/it]Time step 350, Spatial MSE: [(0, 0.0001467174879508093), (1, 0.006633477285504341), (2, 0.008952120319008827), (3, 0.0019219372188672423), (4, 0.0009057327988557518), (5, 0.001846682163886726), (6, 0.002922915853559971), (7, 0.006926411762833595), (8, 0.0054168677888810635), (9, 0.005411680322140455), (10, 0.01255236566066742), (11, 0.009558808989822865), (12, 0.010910218581557274), (13, 0.008817445486783981), (14, 0.029714995995163918), (15, 0.03435773402452469), (16, 0.04002323001623154), (17, 0.028979269787669182), (18, 0.03164291009306908), (19, 0.048150066286325455), (20, 0.046749528497457504), (21, 0.07135248929262161), (22, 0.09682903438806534), (23, 0.10627700388431549), (24, 0.1677672415971756), (25, 0.36566704511642456), (26, 0.271716833114624), (27, 0.16123053431510925)], Temporal MSE: [(0, 1.142871224146802e-05), (1, 3.714318518177606e-05), (2, 0.00016347192286048084), (3, 0.0002529021294321865), (4, 0.0005209220107644796), (5, 0.0005831952439621091), (6, 0.0024745489936321974), (7, 0.003726070746779442), (8, 0.001383563387207687), (9, 0.002262907102704048), (10, 0.003980411682277918), (11, 0.0035189324989914894), (12, 0.00303770718164742), (13, 0.0027084355242550373), (14, 0.013405333273112774), (15, 0.01223946176469326), (16, 0.013708471320569515), (17, 0.013499721884727478), (18, 0.009458085522055626), (19, 0.012955458834767342), (20, 0.03949925675988197), (21, 0.05113216117024422), (22, 0.021910488605499268), (23, 0.02785911038517952), (24, 0.06196983531117439), (25, 0.04712384566664696), (26, 0.03917277976870537), (27, 0.0642361268401146)]
+
+ 93%|█████████▎| 28/30 [03:55<00:21, 10.80s/it]Time step 258, Spatial MSE: [(0, 0.00029254911351017654), (1, 0.0104539068415761), (2, 0.014795317314565182), (3, 0.003689109580591321), (4, 0.001599493552930653), (5, 0.0035501173697412014), (6, 0.0052112615667283535), (7, 0.017036348581314087), (8, 0.011584361083805561), (9, 0.009177792817354202), (10, 0.02300165221095085), (11, 0.014997180551290512), (12, 0.01889304630458355), (13, 0.01341413427144289), (14, 0.049960944801568985), (15, 0.05537152662873268), (16, 0.06326762586832047), (17, 0.049532558768987656), (18, 0.061187759041786194), (19, 0.06759338080883026), (20, 0.07045520842075348), (21, 0.1266094297170639), (22, 0.15765470266342163), (23, 0.20270778238773346), (24, 0.3636024594306946), (25, 0.8268663883209229), (26, 0.5820115804672241), (27, 0.3396100103855133)], Temporal MSE: [(0, 2.7572705221245997e-05), (1, 7.199376705102623e-05), (2, 0.000303420441923663), (3, 0.0004381912585813552), (4, 0.0009824265725910664), (5, 0.0013695373199880123), (6, 0.004384647589176893), (7, 0.008096711710095406), (8, 0.0023387230467051268), (9, 0.0035081878304481506), (10, 0.006376349367201328), (11, 0.005365137942135334), (12, 0.004546343348920345), (13, 0.004117575008422136), (14, 0.018844597041606903), (15, 0.017643488943576813), (16, 0.01961602456867695), (17, 0.01923997513949871), (18, 0.014334163628518581), (19, 0.022624513134360313), (20, 0.06421664357185364), (21, 0.09397050738334656), (22, 0.041026242077350616), (23, 0.0545988455414772), (24, 0.14596207439899445), (25, 0.11127682775259018), (26, 0.10049208998680115), (27, 0.1416262835264206)]
+
+ 97%|█████████▋| 29/30 [04:07<00:10, 10.97s/it]Time step 143, Spatial MSE: [(0, 0.0007044906378723681), (1, 0.021269507706165314), (2, 0.026872562244534492), (3, 0.00767405005171895), (4, 0.003714492078870535), (5, 0.008457318879663944), (6, 0.012087787501513958), (7, 0.0385499931871891), (8, 0.025504417717456818), (9, 0.02215549349784851), (10, 0.05371478199958801), (11, 0.03801471367478371), (12, 0.04570776969194412), (13, 0.027854548767209053), (14, 0.11016503721475601), (15, 0.11281795054674149), (16, 0.11785517632961273), (17, 0.11036117374897003), (18, 0.12268248945474625), (19, 0.15085332095623016), (20, 0.20249833166599274), (21, 0.26271507143974304), (22, 0.35405364632606506), (23, 0.5058078765869141), (24, 1.2284131050109863), (25, 2.5011422634124756), (26, 1.6982710361480713), (27, 0.9655839204788208)], Temporal MSE: [(0, 7.423086935887113e-05), (1, 0.00016887426318135113), (2, 0.0007425045478157699), (3, 0.0009266496053896844), (4, 0.0026063560508191586), (5, 0.004006299190223217), (6, 0.010964922606945038), (7, 0.017713584005832672), (8, 0.005510445684194565), (9, 0.00646414328366518), (10, 0.013281114399433136), (11, 0.011631217785179615), (12, 0.009598391130566597), (13, 0.00824698805809021), (14, 0.033684007823467255), (15, 0.03192143887281418), (16, 0.0337340272963047), (17, 0.035136837512254715), (18, 0.028225652873516083), (19, 0.04880990833044052), (20, 0.1896020919084549), (21, 0.258351594209671), (22, 0.09279654920101166), (23, 0.14399830996990204), (24, 0.4387490153312683), (25, 0.3810785114765167), (26, 0.2939298748970032), (27, 0.3177001178264618)]
+
+100%|██████████| 30/30 [04:20<00:00, 11.58s/it]
+100%|██████████| 30/30 [04:20<00:00,  8.68s/it]
+[2024-10-01 15:07:29] Prompt: an epic tornado attacking above aglowing city at night. aesthetic score: 6.5.
+Saved to ./samples/opensora/sample_pab/An epic tornado attacking above aglowing.mp4
+Saved video to ./samples/opensora/sample_pab/An epic tornado attacking above aglowing.mp4
+Saved spatial MLP MSE outputs to ./samples/opensora/sample_pab/An epic tornado attacking above aglowing_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora/sample_pab/An epic tornado attacking above aglowing_temporal_mlp_mse.pt
+timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:0'), tensor([828.8770], device='cuda:0'), tensor([809.0977], device='cuda:0'), tensor([787.4170], device='cuda:0'), tensor([763.5468], device='cuda:0'), tensor([737.1379], device='cuda:0'), tensor([707.7626], device='cuda:0'), tensor([674.8911], device='cuda:0'), tensor([637.8601], device='cuda:0'), tensor([595.8265], device='cuda:0'), tensor([547.7032], device='cuda:0'), tensor([492.0635], device='cuda:0'), tensor([426.9973], device='cuda:0'), tensor([349.8871], device='cuda:0'), tensor([257.0481], device='cuda:0'), tensor([143.1210], device='cuda:0')]
+
+  0%|          | 0/30 [00:00<?, ?it/s]
+  3%|▎         | 1/30 [00:03<01:47,  3.71s/it]Time step 992, Spatial MSE: [(0, 0.0012019777204841375), (1, 0.0017087494488805532), (2, 0.043298520147800446), (3, 0.0327879823744297), (4, 0.04637246951460838), (5, 0.066470205783844), (6, 0.05611204728484154), (7, 0.0902031809091568), (8, 0.09273550659418106), (9, 0.22718805074691772), (10, 0.6661713123321533), (11, 0.3452557623386383), (12, 0.5020514726638794), (13, 0.4278261363506317), (14, 0.5806346535682678), (15, 0.5143808126449585), (16, 1.712443470954895), (17, 0.8488966226577759), (18, 0.643622636795044), (19, 1.140391230583191), (20, 2.3422157764434814), (21, 1.672239065170288), (22, 1.3384039402008057), (23, 3.1720991134643555), (24, 9.975135803222656), (25, 12.271929740905762), (26, 9.829469680786133), (27, 0.6087422370910645)], Temporal MSE: [(0, 5.08981647726614e-05), (1, 0.0004792749823536724), (2, 0.0018633212894201279), (3, 0.0026304377242922783), (4, 0.002526305615901947), (5, 0.010911211371421814), (6, 0.008221187628805637), (7, 0.01168273389339447), (8, 0.015256869606673717), (9, 0.017251137644052505), (10, 0.06576897203922272), (11, 0.0423940047621727), (12, 0.08074844628572464), (13, 0.034510668367147446), (14, 0.0971907451748848), (15, 0.1089475080370903), (16, 0.1822521984577179), (17, 0.08697491884231567), (18, 0.09058475494384766), (19, 0.1049029752612114), (20, 0.08934223651885986), (21, 0.1716337949037552), (22, 0.6535666584968567), (23, 0.7713021636009216), (24, 0.4765889346599579), (25, 0.2266893833875656), (26, 1.9026237726211548), (27, 0.9258289933204651)]
+
+  7%|▋         | 2/30 [00:16<04:16,  9.17s/it]Time step 984, Spatial MSE: [(0, 0.00036623020423576236), (1, 0.0013349868822842836), (2, 0.006851848214864731), (3, 0.006057983264327049), (4, 0.00654349010437727), (5, 0.014122339896857738), (6, 0.019085118547081947), (7, 0.04586661234498024), (8, 0.04771190509200096), (9, 0.12614086270332336), (10, 0.182937890291214), (11, 0.22355440258979797), (12, 0.17450356483459473), (13, 0.18611712753772736), (14, 0.2991132438182831), (15, 0.299681693315506), (16, 0.37175819277763367), (17, 0.2228280007839203), (18, 0.23273088037967682), (19, 0.21684850752353668), (20, 0.38034388422966003), (21, 0.24781690537929535), (22, 0.24775579571723938), (23, 0.5053055286407471), (24, 1.4578174352645874), (25, 1.9139554500579834), (26, 1.8126920461654663), (27, 0.3687625825405121)], Temporal MSE: [(0, 7.930624633445404e-06), (1, 0.00015180623449850827), (2, 0.000433482404332608), (3, 0.0005352367297746241), (4, 0.0007102476665750146), (5, 0.000981612945906818), (6, 0.002516159089282155), (7, 0.005543453618884087), (8, 0.004025815054774284), (9, 0.006816530600190163), (10, 0.015079767443239689), (11, 0.01505480520427227), (12, 0.014285636134445667), (13, 0.013546819798648357), (14, 0.040392253547906876), (15, 0.04050082713365555), (16, 0.05263569578528404), (17, 0.0363868810236454), (18, 0.032170526683330536), (19, 0.07803487032651901), (20, 0.06458604335784912), (21, 0.09972413629293442), (22, 0.24742022156715393), (23, 0.25072696805000305), (24, 0.15504631400108337), (25, 0.12449093908071518), (26, 0.3992815315723419), (27, 0.201817587018013)]
+
+ 10%|█         | 3/30 [00:26<04:18,  9.59s/it]Time step 976, Spatial MSE: [(0, 0.00017755533917807043), (1, 0.0008353301091119647), (2, 0.0024057941045612097), (3, 0.0020341784693300724), (4, 0.0023812875151634216), (5, 0.003571336390450597), (6, 0.01058225054293871), (7, 0.01297518890351057), (8, 0.020528066903352737), (9, 0.03348463401198387), (10, 0.0708877369761467), (11, 0.06948915123939514), (12, 0.06922650337219238), (13, 0.07847348600625992), (14, 0.12947706878185272), (15, 0.13775911927223206), (16, 0.19396129250526428), (17, 0.13816483318805695), (18, 0.14766578376293182), (19, 0.1426815688610077), (20, 0.2919110655784607), (21, 0.21634119749069214), (22, 0.15911456942558289), (23, 0.1615326702594757), (24, 0.40995290875434875), (25, 0.5736903548240662), (26, 0.24994215369224548), (27, 0.07057256996631622)], Temporal MSE: [(0, 3.5961636513093254e-06), (1, 5.728191899834201e-05), (2, 0.00012624659575521946), (3, 0.000162861542776227), (4, 0.0002527307078707963), (5, 0.0009445249452255666), (6, 0.000732957327272743), (7, 0.001839366857893765), (8, 0.0015859002014622092), (9, 0.002385962288826704), (10, 0.006203066557645798), (11, 0.006505119614303112), (12, 0.006143540143966675), (13, 0.005815382115542889), (14, 0.01985262706875801), (15, 0.02118169702589512), (16, 0.02661878801882267), (17, 0.021762555465102196), (18, 0.02199081890285015), (19, 0.025880077853798866), (20, 0.05196874961256981), (21, 0.05766952037811279), (22, 0.061758894473314285), (23, 0.0605391226708889), (24, 0.03498583287000656), (25, 0.07126490771770477), (26, 0.041501693427562714), (27, 0.03322138637304306)]
+
+ 13%|█▎        | 4/30 [00:39<04:38, 10.72s/it]Time step 968, Spatial MSE: [(0, 0.00010766331251943484), (1, 0.0005043090786784887), (2, 0.0013718536356464028), (3, 0.0006893092067912221), (4, 0.0009065234917216003), (5, 0.0015481942100450397), (6, 0.004256044514477253), (7, 0.005499379709362984), (8, 0.009235422126948833), (9, 0.015146782621741295), (10, 0.025208916515111923), (11, 0.030707895755767822), (12, 0.030938677489757538), (13, 0.03354409337043762), (14, 0.059291765093803406), (15, 0.07217136025428772), (16, 0.1053171157836914), (17, 0.08619298785924911), (18, 0.09131849557161331), (19, 0.07025942951440811), (20, 0.08169271051883698), (21, 0.07708466053009033), (22, 0.06883114576339722), (23, 0.06840965896844864), (24, 0.12987454235553741), (25, 0.2055639922618866), (26, 0.09873124957084656), (27, 0.027405161410570145)], Temporal MSE: [(0, 1.8582639995656791e-06), (1, 2.436422983009834e-05), (2, 7.263998122652993e-05), (3, 8.079355757217854e-05), (4, 0.00013452203711494803), (5, 0.0006603094516322017), (6, 0.000372762733604759), (7, 0.001066736993379891), (8, 0.0008970557246357203), (9, 0.0012648750562220812), (10, 0.0033633667044341564), (11, 0.003023267025128007), (12, 0.002812041435390711), (13, 0.002841424196958542), (14, 0.010406444780528545), (15, 0.013935168273746967), (16, 0.016593551263213158), (17, 0.015004036948084831), (18, 0.012650524266064167), (19, 0.015445208176970482), (20, 0.027014924213290215), (21, 0.030087726190686226), (22, 0.025745807215571404), (23, 0.025867318734526634), (24, 0.015313098207116127), (25, 0.03910759091377258), (26, 0.014397888444364071), (27, 0.010026146657764912)]
+
+ 17%|█▋        | 5/30 [00:49<04:27, 10.71s/it]Time step 960, Spatial MSE: [(0, 7.458626350853592e-05), (1, 0.000594788754824549), (2, 0.001179135637357831), (3, 0.0007882192730903625), (4, 0.0007316838600672781), (5, 0.0014017268549650908), (6, 0.0034205964766442776), (7, 0.0056872498244047165), (8, 0.009555336087942123), (9, 0.01621474139392376), (10, 0.028865568339824677), (11, 0.03768092766404152), (12, 0.03699798509478569), (13, 0.04478026181459427), (14, 0.10196629911661148), (15, 0.12244570255279541), (16, 0.16478759050369263), (17, 0.1365334391593933), (18, 0.14581380784511566), (19, 0.12348771840333939), (20, 0.1412898749113083), (21, 0.10338107496500015), (22, 0.11551493406295776), (23, 0.10610691457986832), (24, 0.13742823898792267), (25, 0.22565221786499023), (26, 0.10524825751781464), (27, 0.02094516158103943)], Temporal MSE: [(0, 1.5201104588413727e-06), (1, 1.2431042705429718e-05), (2, 5.799140490125865e-05), (3, 7.356148853432387e-05), (4, 0.00012384311412461102), (5, 0.0007040298078209162), (6, 0.00046854547690600157), (7, 0.0014924127608537674), (8, 0.0011403579264879227), (9, 0.0017502279952168465), (10, 0.005082959309220314), (11, 0.004950877744704485), (12, 0.004492048639804125), (13, 0.004919365514069796), (14, 0.02500869706273079), (15, 0.02931029163300991), (16, 0.0332099087536335), (17, 0.029751505702733994), (18, 0.022024761885404587), (19, 0.025462789461016655), (20, 0.04598643630743027), (21, 0.0414423868060112), (22, 0.03602253273129463), (23, 0.03962380439043045), (24, 0.019861234351992607), (25, 0.04551885277032852), (26, 0.01497639436274767), (27, 0.01221972331404686)]
+
+ 20%|██        | 6/30 [01:00<04:16, 10.68s/it]Time step 952, Spatial MSE: [(0, 6.827058678027242e-05), (1, 0.0004578150692395866), (2, 0.0008898694068193436), (3, 0.0005246179061941803), (4, 0.0007937759510241449), (5, 0.0014662870671600103), (6, 0.004240565933287144), (7, 0.006875838153064251), (8, 0.012269077822566032), (9, 0.021857431158423424), (10, 0.0414232462644577), (11, 0.0557413212954998), (12, 0.0512290894985199), (13, 0.05324845761060715), (14, 0.13800738751888275), (15, 0.18423263728618622), (16, 0.2137753665447235), (17, 0.17018654942512512), (18, 0.17188552021980286), (19, 0.14811664819717407), (20, 0.13958457112312317), (21, 0.12873749434947968), (22, 0.15076944231987), (23, 0.14098672568798065), (24, 0.18898579478263855), (25, 0.28262844681739807), (26, 0.16897764801979065), (27, 0.02782716229557991)], Temporal MSE: [(0, 1.2355027365629212e-06), (1, 9.427314580534585e-06), (2, 4.5097720430931076e-05), (3, 7.098248170223087e-05), (4, 0.00011906450527021661), (5, 0.0004294120008125901), (6, 0.0004620673134922981), (7, 0.001555066555738449), (8, 0.0012129206443205476), (9, 0.002186692552641034), (10, 0.0057219392620027065), (11, 0.006679965648800135), (12, 0.0060772765427827835), (13, 0.005343986209481955), (14, 0.0224820114672184), (15, 0.03301471844315529), (16, 0.03926180675625801), (17, 0.03661743178963661), (18, 0.024196399375796318), (19, 0.030262969434261322), (20, 0.05504385009407997), (21, 0.051655661314725876), (22, 0.05443938076496124), (23, 0.06064470112323761), (24, 0.029196547344326973), (25, 0.03783124312758446), (26, 0.018886970356106758), (27, 0.01868470385670662)]
+
+ 23%|██▎       | 7/30 [01:13<04:23, 11.45s/it]Time step 940, Spatial MSE: [(0, 7.259891572175547e-05), (1, 0.00041481610969640315), (2, 0.0007597288931719959), (3, 0.00042262382339686155), (4, 0.0007740580476820469), (5, 0.0014737116871401668), (6, 0.0037831501103937626), (7, 0.008360776118934155), (8, 0.011394643224775791), (9, 0.02066187933087349), (10, 0.051734551787376404), (11, 0.05398251488804817), (12, 0.048035573214292526), (13, 0.051880162209272385), (14, 0.13391856849193573), (15, 0.17609570920467377), (16, 0.191470667719841), (17, 0.17268668115139008), (18, 0.19645144045352936), (19, 0.19527243077754974), (20, 0.20287302136421204), (21, 0.17357848584651947), (22, 0.2132626622915268), (23, 0.20634232461452484), (24, 0.26140832901000977), (25, 0.32142937183380127), (26, 0.18202047049999237), (27, 0.04565369337797165)], Temporal MSE: [(0, 1.3566611869464396e-06), (1, 9.917024726746604e-06), (2, 5.205823254073039e-05), (3, 4.2183346522506326e-05), (4, 0.00013658960233442485), (5, 0.00038186597521416843), (6, 0.0005450525204651058), (7, 0.0021768363658338785), (8, 0.0013077246258035302), (9, 0.0023759936448186636), (10, 0.0064528463408350945), (11, 0.00740768201649189), (12, 0.006167290266603231), (13, 0.005298304837197065), (14, 0.02209061197936535), (15, 0.03160781413316727), (16, 0.03478332236409187), (17, 0.031525563448667526), (18, 0.029752135276794434), (19, 0.0413563996553421), (20, 0.07197463512420654), (21, 0.0823931097984314), (22, 0.06962738931179047), (23, 0.08235104382038116), (24, 0.03983522579073906), (25, 0.040434494614601135), (26, 0.02181808091700077), (27, 0.02707667648792267)]
+
+ 27%|██▋       | 8/30 [01:25<04:14, 11.57s/it]Time step 932, Spatial MSE: [(0, 6.453649257309735e-05), (1, 0.00039724214002490044), (2, 0.0005184452165849507), (3, 0.0003637340269051492), (4, 0.000339114194503054), (5, 0.0006435260293073952), (6, 0.0015750202583149076), (7, 0.003566577797755599), (8, 0.0057897185906767845), (9, 0.01045914739370346), (10, 0.025078389793634415), (11, 0.027340585365891457), (12, 0.02568187192082405), (13, 0.029223326593637466), (14, 0.07403713464736938), (15, 0.0891745537519455), (16, 0.09880129992961884), (17, 0.0970335453748703), (18, 0.1495056301355362), (19, 0.17692819237709045), (20, 0.20231036841869354), (21, 0.1879146844148636), (22, 0.24017062783241272), (23, 0.2878691554069519), (24, 0.3022354245185852), (25, 0.2671111226081848), (26, 0.12928061187267303), (27, 0.029606519266963005)], Temporal MSE: [(0, 9.694616665001377e-07), (1, 4.394095867610304e-06), (2, 2.6167110263486393e-05), (3, 5.240391328698024e-05), (4, 6.00340063101612e-05), (5, 0.00017916702199727297), (6, 0.00023748526291456074), (7, 0.0009244694956578314), (8, 0.0005299464683048427), (9, 0.0010142462560907006), (10, 0.0021502694580703974), (11, 0.0032746160868555307), (12, 0.002973361639305949), (13, 0.00282542803324759), (14, 0.011582410894334316), (15, 0.014336608350276947), (16, 0.01765068806707859), (17, 0.02035059966146946), (18, 0.03614484891295433), (19, 0.047768693417310715), (20, 0.06553632020950317), (21, 0.09326153248548508), (22, 0.08994602411985397), (23, 0.081145279109478), (24, 0.055156633257865906), (25, 0.025947250425815582), (26, 0.0191233791410923), (27, 0.019123537465929985)]
+
+ 30%|███       | 9/30 [01:39<04:20, 12.43s/it]Time step 920, Spatial MSE: [(0, 2.9730023015872575e-05), (1, 0.0003774450160562992), (2, 0.0005437011714093387), (3, 0.00015253554738592356), (4, 6.928293441887945e-05), (5, 0.00010592673788778484), (6, 0.00014431295858230442), (7, 0.0006223066593520343), (8, 0.00021020117856096476), (9, 0.00044754965347237885), (10, 0.00043915078276768327), (11, 0.0003972345730289817), (12, 0.0006553918356075883), (13, 0.0007387230289168656), (14, 0.0010181550169363618), (15, 0.0019377710996195674), (16, 0.006186034996062517), (17, 0.002033447613939643), (18, 0.003437657840549946), (19, 0.0027113938704133034), (20, 0.0017948599997907877), (21, 0.0022525712847709656), (22, 0.0021013556979596615), (23, 0.0031728334724903107), (24, 0.00663309870287776), (25, 0.014093607664108276), (26, 0.010653166100382805), (27, 0.012775720097124577)], Temporal MSE: [(0, 8.271177875940339e-07), (1, 3.62517562280118e-06), (2, 1.3147534446034115e-05), (3, 1.1839006219815928e-05), (4, 1.4801314137002919e-05), (5, 1.8667036783881485e-05), (6, 4.059577258885838e-05), (7, 0.0001040858551277779), (8, 5.356093606678769e-05), (9, 9.351912012789398e-05), (10, 0.00011899159289896488), (11, 0.00012148112728027627), (12, 0.00011690308747347444), (13, 8.896701183402911e-05), (14, 0.00029340499895624816), (15, 0.0010950539726763964), (16, 0.0009200060158036649), (17, 0.0008441293612122536), (18, 0.0011395118199288845), (19, 0.0015118950977921486), (20, 0.0018290698062628508), (21, 0.0010922126239165664), (22, 0.0007231998606584966), (23, 0.0013269418850541115), (24, 0.0006903026951476932), (25, 0.0009646713151596487), (26, 0.0012400370324030519), (27, 0.0008910749456845224)]
+
+ 33%|███▎      | 10/30 [01:49<03:54, 11.74s/it]Time step 908, Spatial MSE: [(0, 0.0001589477324159816), (1, 0.0005100114503875375), (2, 0.0006980177131481469), (3, 0.00026025823899544775), (4, 0.00035798054886981845), (5, 0.0007698294939473271), (6, 0.0016071473946794868), (7, 0.002151371678337455), (8, 0.002082566265016794), (9, 0.0031641507521271706), (10, 0.0075220209546387196), (11, 0.005737255327403545), (12, 0.006214497145265341), (13, 0.006831277161836624), (14, 0.020128117874264717), (15, 0.015972521156072617), (16, 0.027367593720555305), (17, 0.015899615362286568), (18, 0.017936984077095985), (19, 0.01777833327651024), (20, 0.017041238024830818), (21, 0.03213971108198166), (22, 0.02190004102885723), (23, 0.01873273029923439), (24, 0.028297778218984604), (25, 0.041932251304388046), (26, 0.018667055293917656), (27, 0.01422170177102089)], Temporal MSE: [(0, 2.460802306814003e-06), (1, 4.764306140714325e-06), (2, 4.959361467626877e-05), (3, 6.176836905069649e-05), (4, 5.037340451963246e-05), (5, 0.0001784160267561674), (6, 0.00015136826550588012), (7, 0.0009610832785256207), (8, 0.000529724988155067), (9, 0.0006803650758229196), (10, 0.0013684093719348311), (11, 0.001036404399201274), (12, 0.0013406324433162808), (13, 0.0008015168714337051), (14, 0.00436047650873661), (15, 0.00458641117438674), (16, 0.006626828573644161), (17, 0.0041847918182611465), (18, 0.004685236141085625), (19, 0.006731182802468538), (20, 0.006568692624568939), (21, 0.009656347334384918), (22, 0.009647554717957973), (23, 0.011712984181940556), (24, 0.00576635031029582), (25, 0.0029560558032244444), (26, 0.005696933716535568), (27, 0.005809905473142862)]
+
+ 37%|███▋      | 11/30 [02:01<03:40, 11.61s/it]Time step 892, Spatial MSE: [(0, 3.056843706872314e-05), (1, 0.0004931410658173263), (2, 0.0006804225849919021), (3, 0.00017295288853347301), (4, 6.184310041135177e-05), (5, 0.00010785568883875385), (6, 0.00012165920634288341), (7, 0.0005296653253026307), (8, 0.00021582299086730927), (9, 0.0003852508671116084), (10, 0.00036236978485248983), (11, 0.00045640708412975073), (12, 0.0005046729929745197), (13, 0.0009621118078939617), (14, 0.0011349705746397376), (15, 0.0018167397938668728), (16, 0.01195723656564951), (17, 0.0023748406674712896), (18, 0.00350949470885098), (19, 0.003859925549477339), (20, 0.0023792951833456755), (21, 0.0035711927339434624), (22, 0.0029448079876601696), (23, 0.004382514860481024), (24, 0.009051679633557796), (25, 0.0128545006737113), (26, 0.013570159673690796), (27, 0.00955620314925909)], Temporal MSE: [(0, 7.707790246058721e-07), (1, 3.589264224501676e-06), (2, 1.3304664207680617e-05), (3, 1.2312923900026362e-05), (4, 1.4245246347854845e-05), (5, 1.9498238543746993e-05), (6, 2.7113386749988422e-05), (7, 8.237466681748629e-05), (8, 5.245327702141367e-05), (9, 7.04226185916923e-05), (10, 0.00010508068226044998), (11, 0.00010979797662002966), (12, 8.682724728714675e-05), (13, 9.193639561999589e-05), (14, 0.0003191737050656229), (15, 0.0009468590724281967), (16, 0.0010889708064496517), (17, 0.0009587521199136972), (18, 0.0006760733085684478), (19, 0.00197775149717927), (20, 0.003356573637574911), (21, 0.0012959910091012716), (22, 0.0009013603557832539), (23, 0.001229917979799211), (24, 0.0008875842904672027), (25, 0.0009036740520969033), (26, 0.0015288020949810743), (27, 0.000984520767815411)]
+
+ 40%|████      | 12/30 [02:10<03:18, 11.01s/it]Time step 880, Spatial MSE: [(0, 0.0001768292859196663), (1, 0.0006601089262403548), (2, 0.0018446446629241109), (3, 0.0013995945919305086), (4, 0.002537432825192809), (5, 0.007460171822458506), (6, 0.015411497093737125), (7, 0.06218390539288521), (8, 0.09291768819093704), (9, 0.12018606066703796), (10, 0.2848523259162903), (11, 0.21614882349967957), (12, 0.14584191143512726), (13, 0.14500336349010468), (14, 0.39977511763572693), (15, 0.498250812292099), (16, 0.5337214469909668), (17, 0.5725091695785522), (18, 0.5434544086456299), (19, 0.5116189122200012), (20, 0.5895302891731262), (21, 0.6061674356460571), (22, 0.6286939978599548), (23, 0.7972052693367004), (24, 1.0295788049697876), (25, 0.9513813853263855), (26, 0.589335560798645), (27, 0.1370854675769806)], Temporal MSE: [(0, 3.6337355595605914e-06), (1, 3.0500097636831924e-05), (2, 0.00044782960321754217), (3, 0.0006791511550545692), (4, 0.0007911747670732439), (5, 0.0026007001288235188), (6, 0.004399839323014021), (7, 0.017442012205719948), (8, 0.00884441751986742), (9, 0.010631733573973179), (10, 0.022116227075457573), (11, 0.025681642815470695), (12, 0.020253470167517662), (13, 0.01878536306321621), (14, 0.07591940462589264), (15, 0.090732641518116), (16, 0.11217281222343445), (17, 0.12682949006557465), (18, 0.10730580985546112), (19, 0.14078456163406372), (20, 0.2726151645183563), (21, 0.3073088526725769), (22, 0.2366500347852707), (23, 0.2869439125061035), (24, 0.21568018198013306), (25, 0.07469291239976883), (26, 0.08004838228225708), (27, 0.07636535912752151)]
+
+ 43%|████▎     | 13/30 [02:20<02:59, 10.58s/it]Time step 864, Spatial MSE: [(0, 2.1919764549238607e-05), (1, 0.00044996722135692835), (2, 0.0006462846649810672), (3, 0.00019053995492868125), (4, 5.680415779352188e-05), (5, 8.258979505626485e-05), (6, 0.00011964463192271069), (7, 0.00035027938429266214), (8, 0.0001460585481254384), (9, 0.000219644934986718), (10, 0.00030403866549022496), (11, 0.0003011483349837363), (12, 0.0003860224096570164), (13, 0.0006562714115716517), (14, 0.0008825685945339501), (15, 0.000977117451839149), (16, 0.010838061571121216), (17, 0.0020225869957357645), (18, 0.002427668310701847), (19, 0.005827223416417837), (20, 0.003600148484110832), (21, 0.011843173764646053), (22, 0.015051474794745445), (23, 0.005804389249533415), (24, 0.009151156060397625), (25, 0.015121503733098507), (26, 0.014960614033043385), (27, 0.01361082773655653)], Temporal MSE: [(0, 5.656318649016612e-07), (1, 2.847680661943741e-06), (2, 9.584082363289781e-06), (3, 1.0200246833846904e-05), (4, 1.0997111530741677e-05), (5, 1.8254166207043454e-05), (6, 2.50972734647803e-05), (7, 5.8414374507265165e-05), (8, 3.85446946893353e-05), (9, 5.5345502914860845e-05), (10, 7.441771595040336e-05), (11, 6.924585613887757e-05), (12, 5.829159272252582e-05), (13, 5.984392191749066e-05), (14, 0.00025656819343566895), (15, 0.00043231758172623813), (16, 0.0011644442565739155), (17, 0.0007514820317737758), (18, 0.000331383926095441), (19, 0.0024506638292223215), (20, 0.00486728036776185), (21, 0.0034857122227549553), (22, 0.0041977036744356155), (23, 0.0012398459948599339), (24, 0.0011608783388510346), (25, 0.0009470141376368701), (26, 0.001103412127122283), (27, 0.0019940074998885393)]
+
+ 47%|████▋     | 14/30 [02:31<02:51, 10.70s/it]Time step 848, Spatial MSE: [(0, 0.00019081974460277706), (1, 0.0004914766177535057), (2, 0.000608909351285547), (3, 0.00024738776846788824), (4, 0.0003077146830037236), (5, 0.0009959028102457523), (6, 0.002343488158658147), (7, 0.0028850322123616934), (8, 0.0025809244252741337), (9, 0.003915824927389622), (10, 0.007902572862803936), (11, 0.007294136565178633), (12, 0.007156216073781252), (13, 0.008466177619993687), (14, 0.01926233060657978), (15, 0.02393011935055256), (16, 0.03682997450232506), (17, 0.019124848768115044), (18, 0.02412739396095276), (19, 0.031332097947597504), (20, 0.02614920772612095), (21, 0.033797070384025574), (22, 0.027974598109722137), (23, 0.022542381659150124), (24, 0.03447232022881508), (25, 0.04620804265141487), (26, 0.02839900553226471), (27, 0.011468224227428436)], Temporal MSE: [(0, 2.09859126698575e-06), (1, 5.077385139884427e-06), (2, 2.328587470401544e-05), (3, 5.8634846936911345e-05), (4, 0.00010706762986956164), (5, 0.0002585813926998526), (6, 0.00035906018456444144), (7, 0.001991186523810029), (8, 0.0010736464755609632), (9, 0.00157649302855134), (10, 0.0032638420816510916), (11, 0.0024268622510135174), (12, 0.003387613222002983), (13, 0.001826025196351111), (14, 0.009219775907695293), (15, 0.016976675018668175), (16, 0.021723492071032524), (17, 0.01293371431529522), (18, 0.008700116537511349), (19, 0.021034710109233856), (20, 0.017697958275675774), (21, 0.01485529262572527), (22, 0.0088425287976861), (23, 0.010950111784040928), (24, 0.006780195981264114), (25, 0.0028074588626623154), (26, 0.006612851284444332), (27, 0.004491771105676889)]
+
+ 50%|█████     | 15/30 [02:42<02:40, 10.68s/it]Time step 828, Spatial MSE: [(0, 2.0247991415089928e-05), (1, 0.0005246700602583587), (2, 0.0007872406276874244), (3, 0.0002131619257852435), (4, 4.561432797345333e-05), (5, 8.065125439316034e-05), (6, 8.02224240032956e-05), (7, 0.0003370897320564836), (8, 0.00014324663789011538), (9, 0.0001887068065116182), (10, 0.00023745778889860958), (11, 0.00022928813996259123), (12, 0.0004552783793769777), (13, 0.0005973224760964513), (14, 0.0007130953599698842), (15, 0.0008015698404051363), (16, 0.007609464228153229), (17, 0.0014769387198612094), (18, 0.0018230039859190583), (19, 0.003119454951956868), (20, 0.0018351313192397356), (21, 0.0033638563472777605), (22, 0.003000566503033042), (23, 0.004584147594869137), (24, 0.00805535539984703), (25, 0.015179477632045746), (26, 0.014411992393434048), (27, 0.015620493330061436)], Temporal MSE: [(0, 5.662172384290898e-07), (1, 2.1405321604106575e-06), (2, 9.46965428738622e-06), (3, 9.153302926279139e-06), (4, 1.0932637451332994e-05), (5, 1.9186327335773967e-05), (6, 2.3736605726298876e-05), (7, 6.108679372118786e-05), (8, 3.7737285310868174e-05), (9, 5.000165037927218e-05), (10, 5.965432501398027e-05), (11, 5.9651192714227363e-05), (12, 8.224749763030559e-05), (13, 6.433494854718447e-05), (14, 0.0001793838746380061), (15, 0.0003349531616549939), (16, 0.0008274615393020213), (17, 0.0005814139149151742), (18, 0.0003219068457838148), (19, 0.0016366352792829275), (20, 0.0028320467099547386), (21, 0.0013127877609804273), (22, 0.0008263367344625294), (23, 0.001502499682828784), (24, 0.0009749565506353974), (25, 0.0009052440873347223), (26, 0.0009516536956652999), (27, 0.0016149643342942)]
+
+ 53%|█████▎    | 16/30 [02:52<02:28, 10.59s/it]Time step 808, Spatial MSE: [(0, 0.0001356365974061191), (1, 0.0009738566004671156), (2, 0.0011512083001434803), (3, 0.0011998035479336977), (4, 0.0019122057128697634), (5, 0.007308355998247862), (6, 0.010838273912668228), (7, 0.03969709947705269), (8, 0.056865282356739044), (9, 0.060435373336076736), (10, 0.07138152420520782), (11, 0.07797674089670181), (12, 0.07574045658111572), (13, 0.07790309935808182), (14, 0.17351602017879486), (15, 0.22236725687980652), (16, 0.27519986033439636), (17, 0.25588637590408325), (18, 0.31443771719932556), (19, 0.356498122215271), (20, 0.41084256768226624), (21, 0.44913968443870544), (22, 0.5492855906486511), (23, 0.6729869246482849), (24, 0.7986234426498413), (25, 0.8517496585845947), (26, 0.5218895673751831), (27, 0.18387851119041443)], Temporal MSE: [(0, 3.2701791496947408e-06), (1, 2.7241965653956868e-05), (2, 0.0001713580422801897), (3, 0.000483568204799667), (4, 0.00088800786761567), (5, 0.0021691853180527687), (6, 0.003815134521573782), (7, 0.012527716346085072), (8, 0.007546462584286928), (9, 0.008602526970207691), (10, 0.014399204403162003), (11, 0.018552269786596298), (12, 0.015969736501574516), (13, 0.015122771263122559), (14, 0.04637525975704193), (15, 0.06291143596172333), (16, 0.08014003187417984), (17, 0.09915199130773544), (18, 0.09538213908672333), (19, 0.14194652438163757), (20, 0.34057027101516724), (21, 0.30180880427360535), (22, 0.22044254839420319), (23, 0.25113484263420105), (24, 0.21342825889587402), (25, 0.06142878904938698), (26, 0.06874197721481323), (27, 0.10452091693878174)]
+
+ 57%|█████▋    | 17/30 [03:02<02:16, 10.46s/it]Time step 788, Spatial MSE: [(0, 2.1101061065564863e-05), (1, 0.0004976601921953261), (2, 0.0009174575097858906), (3, 0.0002460344403516501), (4, 4.855899896938354e-05), (5, 8.025462011573836e-05), (6, 8.370772411581129e-05), (7, 0.0003960580506827682), (8, 0.0001720715081319213), (9, 0.00015957809227984399), (10, 0.00020071474136784673), (11, 0.00023865688126534224), (12, 0.0004250006750226021), (13, 0.0004285965405870229), (14, 0.0005894167115911841), (15, 0.0006134877912700176), (16, 0.005782962776720524), (17, 0.0009870337089523673), (18, 0.008055506274104118), (19, 0.003346661338582635), (20, 0.0019682992715388536), (21, 0.0032099499367177486), (22, 0.002963701030239463), (23, 0.0045319730415940285), (24, 0.007444707676768303), (25, 0.01725940965116024), (26, 0.01539162453263998), (27, 0.010453718714416027)], Temporal MSE: [(0, 6.290578653533885e-07), (1, 2.1608022962027462e-06), (2, 1.0886155905609485e-05), (3, 1.1073139830841683e-05), (4, 1.1949872714467347e-05), (5, 2.401875826762989e-05), (6, 2.7729636713047512e-05), (7, 5.520038030226715e-05), (8, 4.132866524741985e-05), (9, 4.283260932425037e-05), (10, 6.419757846742868e-05), (11, 5.300181874190457e-05), (12, 5.7007473515113816e-05), (13, 6.439073331421241e-05), (14, 0.00018089207878801972), (15, 0.00025462862686254084), (16, 0.0008704752544872463), (17, 0.0004283062880858779), (18, 0.002182914176955819), (19, 0.00271195312961936), (20, 0.006241383031010628), (21, 0.0018180873012170196), (22, 0.0008448619628325105), (23, 0.001273426110856235), (24, 0.0009603826329112053), (25, 0.0011715692235156894), (26, 0.001162541564553976), (27, 0.001367834978736937)]
+
+ 60%|██████    | 18/30 [03:12<02:04, 10.40s/it]Time step 764, Spatial MSE: [(0, 0.0001800965255824849), (1, 0.0007373286643996835), (2, 0.000751557934563607), (3, 0.0002616662241052836), (4, 0.00026094645727425814), (5, 0.0009028638596646488), (6, 0.001699003390967846), (7, 0.0026960582472383976), (8, 0.002371885348111391), (9, 0.003714824328199029), (10, 0.007551510818302631), (11, 0.006591756362468004), (12, 0.007732009049504995), (13, 0.009086063131690025), (14, 0.02158207818865776), (15, 0.03017248585820198), (16, 0.050483208149671555), (17, 0.02140442654490471), (18, 0.02569660171866417), (19, 0.037056826055049896), (20, 0.026765616610646248), (21, 0.03399672359228134), (22, 0.032112061977386475), (23, 0.03220422938466072), (24, 0.04389588534832001), (25, 0.06951308250427246), (26, 0.03780478984117508), (27, 0.01175861619412899)], Temporal MSE: [(0, 2.322782847841154e-06), (1, 5.508583853952587e-06), (2, 2.0907711586914957e-05), (3, 5.7844477851176634e-05), (4, 7.262910366989672e-05), (5, 0.0002484172291588038), (6, 0.00032149211619980633), (7, 0.0016407504444941878), (8, 0.0009312132024206221), (9, 0.0015346223954111338), (10, 0.00351159181445837), (11, 0.002546987496316433), (12, 0.003747980110347271), (13, 0.001971675083041191), (14, 0.011076557449996471), (15, 0.01894945651292801), (16, 0.02393842488527298), (17, 0.013108989223837852), (18, 0.008642749860882759), (19, 0.013663089834153652), (20, 0.019086532294750214), (21, 0.017773684114217758), (22, 0.011144362390041351), (23, 0.012498983182013035), (24, 0.014299165457487106), (25, 0.004390763118863106), (26, 0.006925489753484726), (27, 0.007870538160204887)]
+
+ 63%|██████▎   | 19/30 [03:26<02:05, 11.36s/it]Time step 736, Spatial MSE: [(0, 3.027711863978766e-05), (1, 0.0008335225866176188), (2, 0.0013468440156430006), (3, 0.0003926053468603641), (4, 7.771111268084496e-05), (5, 0.0001191307746921666), (6, 0.0001246946194441989), (7, 0.00034620510996319354), (8, 0.00017255966668017209), (9, 0.00021089018264319748), (10, 0.00022722136054653674), (11, 0.0003078325535170734), (12, 0.0005369162536226213), (13, 0.0006312795449048281), (14, 0.000893946853466332), (15, 0.0011155165266245604), (16, 0.010141310282051563), (17, 0.002026068279519677), (18, 0.002961329184472561), (19, 0.0035325600765645504), (20, 0.0023990334011614323), (21, 0.005268123000860214), (22, 0.004056280013173819), (23, 0.006493870634585619), (24, 0.010267816483974457), (25, 0.023509854450821877), (26, 0.021558642387390137), (27, 0.012063036672770977)], Temporal MSE: [(0, 8.116762728604954e-07), (1, 3.2022683171817334e-06), (2, 1.629060716368258e-05), (3, 1.6173436961253174e-05), (4, 1.84526652446948e-05), (5, 2.966879947052803e-05), (6, 4.967837958247401e-05), (7, 6.947367364773527e-05), (8, 5.2646519179688767e-05), (9, 6.936731369933113e-05), (10, 7.516956247854978e-05), (11, 7.595103670610115e-05), (12, 8.941921259975061e-05), (13, 9.283117105951533e-05), (14, 0.0002705000515561551), (15, 0.0006928024813532829), (16, 0.0011744559742510319), (17, 0.0010251000057905912), (18, 0.0006076754070818424), (19, 0.002103931736201048), (20, 0.005270665045827627), (21, 0.0022102785296738148), (22, 0.0010070562129840255), (23, 0.001712238765321672), (24, 0.0013933645095676184), (25, 0.0016777542186900973), (26, 0.001463005319237709), (27, 0.0022203479893505573)]
+
+ 67%|██████▋   | 20/30 [03:35<01:47, 10.71s/it]Time step 708, Spatial MSE: [(0, 7.506375550292432e-05), (1, 0.0017359982011839747), (2, 0.001575741800479591), (3, 0.001411031698808074), (4, 0.002065825741738081), (5, 0.006760349031537771), (6, 0.009560720063745975), (7, 0.01890862174332142), (8, 0.029918836429715157), (9, 0.032844025641679764), (10, 0.0337994359433651), (11, 0.04814239218831062), (12, 0.0573505200445652), (13, 0.05954336002469063), (14, 0.159574955701828), (15, 0.21056467294692993), (16, 0.2677692174911499), (17, 0.24583902955055237), (18, 0.2923738360404968), (19, 0.31815871596336365), (20, 0.32901373505592346), (21, 0.37621989846229553), (22, 0.43164294958114624), (23, 0.5141634345054626), (24, 0.5786771178245544), (25, 0.6940761804580688), (26, 0.45638859272003174), (27, 0.09845872223377228)], Temporal MSE: [(0, 4.638017799152294e-06), (1, 3.260774974478409e-05), (2, 0.00013506495452020317), (3, 0.0005283314385451376), (4, 0.0011495556682348251), (5, 0.0021412214264273643), (6, 0.005109604448080063), (7, 0.010781445540487766), (8, 0.006010169163346291), (9, 0.008201553486287594), (10, 0.010604110546410084), (11, 0.013576637953519821), (12, 0.012921660207211971), (13, 0.012887571938335896), (14, 0.04528617486357689), (15, 0.06234220042824745), (16, 0.08023440837860107), (17, 0.09522988647222519), (18, 0.08381578326225281), (19, 0.1236352026462555), (20, 0.2559455335140228), (21, 0.21926656365394592), (22, 0.1588934063911438), (23, 0.18397745490074158), (24, 0.1477978527545929), (25, 0.06112757697701454), (26, 0.05429277941584587), (27, 0.0896846354007721)]
+
+ 70%|███████   | 21/30 [03:47<01:38, 10.89s/it]Time step 676, Spatial MSE: [(0, 3.339280010550283e-05), (1, 0.0011186482151970267), (2, 0.001874840003438294), (3, 0.0005308236577548087), (4, 0.000102339377917815), (5, 0.00015528166841249913), (6, 0.0001510060828877613), (7, 0.00041165880975313485), (8, 0.00019840392633341253), (9, 0.00019736868853215128), (10, 0.00027060971478931606), (11, 0.00026960764080286026), (12, 0.0004742788732983172), (13, 0.0005721268826164305), (14, 0.0009118336020037532), (15, 0.0010813144035637379), (16, 0.008001215755939484), (17, 0.0014211074449121952), (18, 0.0022945194505155087), (19, 0.0022280849516391754), (20, 0.001919531961902976), (21, 0.0038811375852674246), (22, 0.0040037743747234344), (23, 0.006925446446985006), (24, 0.009505984373390675), (25, 0.03333057463169098), (26, 0.028106579557061195), (27, 0.013857993297278881)], Temporal MSE: [(0, 1.1821274483736488e-06), (1, 4.062047082697973e-06), (2, 1.955063817149494e-05), (3, 2.466905971232336e-05), (4, 2.6555084332358092e-05), (5, 4.432870264281519e-05), (6, 6.689013389404863e-05), (7, 9.515165584161878e-05), (8, 6.39656136627309e-05), (9, 8.095581870293245e-05), (10, 9.712731116451323e-05), (11, 8.149709174176678e-05), (12, 9.018948185257614e-05), (13, 9.793809294933453e-05), (14, 0.0003593132714740932), (15, 0.0005915173096582294), (16, 0.0007704570889472961), (17, 0.0007645143778063357), (18, 0.00040268563316203654), (19, 0.0014841489028185606), (20, 0.0034654191695153713), (21, 0.001632377621717751), (22, 0.0010981211671605706), (23, 0.0019027116941288114), (24, 0.00135360739659518), (25, 0.00254827830940485), (26, 0.0017786342650651932), (27, 0.0029517491348087788)]
+
+ 73%|███████▎  | 22/30 [03:58<01:28, 11.04s/it]Time step 636, Spatial MSE: [(0, 9.092291293200105e-05), (1, 0.0015327103901654482), (2, 0.0014915099600329995), (3, 0.00040651639574207366), (4, 0.00034723561839200556), (5, 0.001103892340324819), (6, 0.002039809711277485), (7, 0.0035873078741133213), (8, 0.0027166311629116535), (9, 0.004107434302568436), (10, 0.008535243570804596), (11, 0.007963070645928383), (12, 0.009865057654678822), (13, 0.011151427403092384), (14, 0.029459526762366295), (15, 0.05009622499346733), (16, 0.07135752588510513), (17, 0.029734719544649124), (18, 0.033544670790433884), (19, 0.06369329988956451), (20, 0.03492078185081482), (21, 0.04477103054523468), (22, 0.030003154650330544), (23, 0.038754820823669434), (24, 0.06497561186552048), (25, 0.12005253881216049), (26, 0.06486082077026367), (27, 0.028114721179008484)], Temporal MSE: [(0, 2.3014392809272977e-06), (1, 7.177052793849725e-06), (2, 3.0513483579852618e-05), (3, 9.203272202285007e-05), (4, 0.0002092952054226771), (5, 0.0003642638912424445), (6, 0.0008100591949187219), (7, 0.003232290269806981), (8, 0.0018069290090352297), (9, 0.0027623935602605343), (10, 0.006907550152391195), (11, 0.0050395638681948185), (12, 0.007113123778253794), (13, 0.003401558380573988), (14, 0.021266940981149673), (15, 0.04153956472873688), (16, 0.05267397686839104), (17, 0.02608592063188553), (18, 0.014635034836828709), (19, 0.038211263716220856), (20, 0.02533252350986004), (21, 0.02379036135971546), (22, 0.010345284827053547), (23, 0.0133651252835989), (24, 0.011777403764426708), (25, 0.006496966816484928), (26, 0.01059931330382824), (27, 0.00994313508272171)]
+
+ 77%|███████▋  | 23/30 [04:08<01:15, 10.73s/it]Time step 596, Spatial MSE: [(0, 4.5816766942152753e-05), (1, 0.0017446884885430336), (2, 0.0029031147714704275), (3, 0.0008796796319074929), (4, 0.00017259438754990697), (5, 0.0002752753207460046), (6, 0.0002636952558532357), (7, 0.0003891190281137824), (8, 0.00030691191204823554), (9, 0.00027697303448803723), (10, 0.0003805875312536955), (11, 0.0003806498716585338), (12, 0.0005783425294794142), (13, 0.0010082012740895152), (14, 0.0015505100600421429), (15, 0.001172750722616911), (16, 0.019514726474881172), (17, 0.003272776957601309), (18, 0.004631511867046356), (19, 0.00546655710786581), (20, 0.00434525404125452), (21, 0.014992574229836464), (22, 0.020027006044983864), (23, 0.015416789799928665), (24, 0.01660538651049137), (25, 0.05702570080757141), (26, 0.04779922962188721), (27, 0.027082320302724838)], Temporal MSE: [(0, 2.365200543863466e-06), (1, 7.075261692079948e-06), (2, 3.0517847335431725e-05), (3, 3.6487122997641563e-05), (4, 4.6316676161950454e-05), (5, 6.418861448764801e-05), (6, 0.00010682160791475326), (7, 0.00015599263133481145), (8, 0.00010288892372045666), (9, 0.0001254875969607383), (10, 0.00013113814929965883), (11, 0.00013124635734129697), (12, 0.0001470211282139644), (13, 0.0001573041663505137), (14, 0.00040416745468974113), (15, 0.0006671544979326427), (16, 0.0017560336273163557), (17, 0.001594316796399653), (18, 0.0008465314749628305), (19, 0.0031168546993285418), (20, 0.007704612798988819), (21, 0.0053527760319411755), (22, 0.004090283066034317), (23, 0.0028124002274125814), (24, 0.00233554863370955), (25, 0.004267544019967318), (26, 0.0029558732639998198), (27, 0.007270378060638905)]
+
+ 80%|████████  | 24/30 [04:19<01:04, 10.74s/it]Time step 548, Spatial MSE: [(0, 0.00010781409946503118), (1, 0.00385073060169816), (2, 0.003738043364137411), (3, 0.0021823104470968246), (4, 0.003122893860563636), (5, 0.009851312264800072), (6, 0.011745958589017391), (7, 0.019942814484238625), (8, 0.030408641323447227), (9, 0.03177355229854584), (10, 0.056559473276138306), (11, 0.07674247026443481), (12, 0.07129371911287308), (13, 0.06524093449115753), (14, 0.1766687035560608), (15, 0.19781789183616638), (16, 0.24058301746845245), (17, 0.2069447934627533), (18, 0.25411906838417053), (19, 0.2754637598991394), (20, 0.29954037070274353), (21, 0.3620005249977112), (22, 0.4216383397579193), (23, 0.4844883382320404), (24, 0.545994222164154), (25, 0.9397621154785156), (26, 0.6300696134567261), (27, 0.10034532845020294)], Temporal MSE: [(0, 1.2493798749346752e-05), (1, 7.275195821421221e-05), (2, 0.0002545977768022567), (3, 0.0006967152585275471), (4, 0.0016050835838541389), (5, 0.003767278976738453), (6, 0.007014330010861158), (7, 0.009262453764677048), (8, 0.006261593196541071), (9, 0.008732572197914124), (10, 0.014880717732012272), (11, 0.01697017066180706), (12, 0.015289849601686), (13, 0.014778721146285534), (14, 0.048898015171289444), (15, 0.06190408021211624), (16, 0.07988939434289932), (17, 0.08151137828826904), (18, 0.07641882449388504), (19, 0.1250811219215393), (20, 0.21886290609836578), (21, 0.21793410181999207), (22, 0.1382567286491394), (23, 0.17088086903095245), (24, 0.1659146249294281), (25, 0.1540674865245819), (26, 0.06606844067573547), (27, 0.1305365413427353)]
+
+ 83%|████████▎ | 25/30 [04:31<00:56, 11.24s/it]Time step 492, Spatial MSE: [(0, 8.163603342836723e-05), (1, 0.003393501276150346), (2, 0.004856135696172714), (3, 0.0018364809220656753), (4, 0.0003133474965579808), (5, 0.0005896790535189211), (6, 0.000540650449693203), (7, 0.0010439485777169466), (8, 0.0006486722268164158), (9, 0.0006665154360234737), (10, 0.0006127325468696654), (11, 0.0007073920569382608), (12, 0.0014728577807545662), (13, 0.0015933694085106254), (14, 0.0014850333100184798), (15, 0.0016542666126042604), (16, 0.012546388432383537), (17, 0.0026791603304445744), (18, 0.013657873496413231), (19, 0.0058592515997588634), (20, 0.006766229402273893), (21, 0.02146899700164795), (22, 0.013800771906971931), (23, 0.022875703871250153), (24, 0.030789416283369064), (25, 0.11924075335264206), (26, 0.1046409010887146), (27, 0.04531978443264961)], Temporal MSE: [(0, 4.595172413246473e-06), (1, 1.2263094504305627e-05), (2, 6.071696407161653e-05), (3, 8.190212975023314e-05), (4, 0.00010307440243195742), (5, 0.0001510092115495354), (6, 0.00024853815557435155), (7, 0.00035520008532330394), (8, 0.00021445995662361383), (9, 0.00025451110559515655), (10, 0.0002443552657496184), (11, 0.00022981836809776723), (12, 0.00027986682835035026), (13, 0.00030128611251711845), (14, 0.0005495968507602811), (15, 0.0008728128741495311), (16, 0.0015896561089903116), (17, 0.0013242304557934403), (18, 0.0033596171997487545), (19, 0.005460715387016535), (20, 0.02136201784014702), (21, 0.00855014007538557), (22, 0.003074321895837784), (23, 0.005666127894073725), (24, 0.004700321704149246), (25, 0.009106905199587345), (26, 0.00518501503393054), (27, 0.014924413524568081)]
+
+ 87%|████████▋ | 26/30 [04:41<00:43, 10.97s/it]Time step 426, Spatial MSE: [(0, 0.0003076221328228712), (1, 0.006284063681960106), (2, 0.005792234558612108), (3, 0.0017606762703508139), (4, 0.0019021467305719852), (5, 0.005004050210118294), (6, 0.0077775889076292515), (7, 0.01634938083589077), (8, 0.014394683763384819), (9, 0.016910241916775703), (10, 0.02744726650416851), (11, 0.028752829879522324), (12, 0.02918054908514023), (13, 0.02910539321601391), (14, 0.05987264961004257), (15, 0.0950486958026886), (16, 0.11852699518203735), (17, 0.091453418135643), (18, 0.11094687134027481), (19, 0.13486719131469727), (20, 0.13709469139575958), (21, 0.18628321588039398), (22, 0.19611918926239014), (23, 0.2544638514518738), (24, 0.37275058031082153), (25, 0.8251082897186279), (26, 0.5541279911994934), (27, 0.10575851052999496)], Temporal MSE: [(0, 1.4131804164208006e-05), (1, 6.0563961596926674e-05), (2, 0.0002547064796090126), (3, 0.00030128154321573675), (4, 0.0008772335713729262), (5, 0.0014242491452023387), (6, 0.005449745804071426), (7, 0.009972172789275646), (8, 0.0043150512501597404), (9, 0.006285015493631363), (10, 0.01198180578649044), (11, 0.010482911951839924), (12, 0.00891907885670662), (13, 0.0067053986713290215), (14, 0.028291940689086914), (15, 0.03565644472837448), (16, 0.046488843858242035), (17, 0.041365426033735275), (18, 0.03595753014087677), (19, 0.04656349495053291), (20, 0.07027247548103333), (21, 0.09881140291690826), (22, 0.06059110164642334), (23, 0.06896954029798508), (24, 0.10113120824098587), (25, 0.11711879074573517), (26, 0.06631273031234741), (27, 0.11207406967878342)]
+
+ 90%|█████████ | 27/30 [04:53<00:33, 11.20s/it]Time step 350, Spatial MSE: [(0, 0.00019918393809348345), (1, 0.00903516635298729), (2, 0.007805353030562401), (3, 0.0020680383313447237), (4, 0.0011394082102924585), (5, 0.0031793678645044565), (6, 0.004087022505700588), (7, 0.007647464983165264), (8, 0.005802582949399948), (9, 0.006167904939502478), (10, 0.009039690718054771), (11, 0.009507742710411549), (12, 0.010546766221523285), (13, 0.012015406973659992), (14, 0.02094632014632225), (15, 0.03321528062224388), (16, 0.03677539527416229), (17, 0.033029086887836456), (18, 0.04025806859135628), (19, 0.05844055861234665), (20, 0.06119377538561821), (21, 0.09338460117578506), (22, 0.12678839266300201), (23, 0.15496157109737396), (24, 0.23983174562454224), (25, 0.48971039056777954), (26, 0.3892243206501007), (27, 0.17307095229625702)], Temporal MSE: [(0, 1.5177830391621683e-05), (1, 5.5293665354838595e-05), (2, 0.00022170103329699486), (3, 0.0002304303488926962), (4, 0.0004760892770718783), (5, 0.0008089735638350248), (6, 0.002708979183807969), (7, 0.004096533637493849), (8, 0.001812984119169414), (9, 0.002268544165417552), (10, 0.003852039109915495), (11, 0.0031965586822479963), (12, 0.003151515731588006), (13, 0.0027196919545531273), (14, 0.009850415401160717), (15, 0.011733590625226498), (16, 0.015510949306190014), (17, 0.015331357717514038), (18, 0.013698328286409378), (19, 0.01817791908979416), (20, 0.04427720606327057), (21, 0.05844198539853096), (22, 0.0311884768307209), (23, 0.03734912350773811), (24, 0.07755016535520554), (25, 0.06835074722766876), (26, 0.055643316358327866), (27, 0.10000798851251602)]
+
+ 93%|█████████▎| 28/30 [05:05<00:22, 11.44s/it]Time step 258, Spatial MSE: [(0, 0.00043886806815862656), (1, 0.014363158494234085), (2, 0.01328191626816988), (3, 0.004064491949975491), (4, 0.0020736963488161564), (5, 0.005789248738437891), (6, 0.0070636156015098095), (7, 0.016183115541934967), (8, 0.009453115053474903), (9, 0.009220096282660961), (10, 0.015374592505395412), (11, 0.013574550859630108), (12, 0.016373347491025925), (13, 0.017781225964426994), (14, 0.032019540667533875), (15, 0.05083495005965233), (16, 0.06817664206027985), (17, 0.050302259624004364), (18, 0.07343580573797226), (19, 0.08035044372081757), (20, 0.09272531419992447), (21, 0.1585259586572647), (22, 0.17272305488586426), (23, 0.26954683661460876), (24, 0.48064276576042175), (25, 1.018856406211853), (26, 0.7439261674880981), (27, 0.35735440254211426)], Temporal MSE: [(0, 3.905569246853702e-05), (1, 0.00010455308802193031), (2, 0.0003831663925666362), (3, 0.0004332819953560829), (4, 0.0009780540131032467), (5, 0.0017310517141595483), (6, 0.004704752936959267), (7, 0.009114507585763931), (8, 0.003194384276866913), (9, 0.003736149752512574), (10, 0.006169431377202272), (11, 0.005075798369944096), (12, 0.004925570450723171), (13, 0.0040903231129050255), (14, 0.0147241884842515), (15, 0.017175568267703056), (16, 0.021837100386619568), (17, 0.02177458256483078), (18, 0.01999484747648239), (19, 0.02806154265999794), (20, 0.0645546019077301), (21, 0.09649507701396942), (22, 0.050801776349544525), (23, 0.06320694088935852), (24, 0.17127621173858643), (25, 0.1540265828371048), (26, 0.12295866757631302), (27, 0.1961187869310379)]
+
+ 97%|█████████▋| 29/30 [05:18<00:11, 11.78s/it]Time step 143, Spatial MSE: [(0, 0.001058694557286799), (1, 0.02704860456287861), (2, 0.024400554597377777), (3, 0.00888145063072443), (4, 0.004465209320187569), (5, 0.011819928884506226), (6, 0.013691949658095837), (7, 0.03662261739373207), (8, 0.018364863470196724), (9, 0.01719096302986145), (10, 0.03915717452764511), (11, 0.031443968415260315), (12, 0.03454258665442467), (13, 0.031502023339271545), (14, 0.07962963730096817), (15, 0.11364997923374176), (16, 0.14114253222942352), (17, 0.1118442565202713), (18, 0.13466785848140717), (19, 0.17430958151817322), (20, 0.22152839601039886), (21, 0.2899969220161438), (22, 0.32232823967933655), (23, 0.5472574830055237), (24, 1.4634604454040527), (25, 2.7538883686065674), (26, 1.7691905498504639), (27, 0.869491457939148)], Temporal MSE: [(0, 0.00011260031897109002), (1, 0.00021521908638533205), (2, 0.000796944776084274), (3, 0.0009240306681022048), (4, 0.0025048686657100916), (5, 0.00402654055505991), (6, 0.010879458859562874), (7, 0.019281184300780296), (8, 0.006285420618951321), (9, 0.006593390367925167), (10, 0.014302474446594715), (11, 0.01215061079710722), (12, 0.010380689054727554), (13, 0.007658409886062145), (14, 0.03190824016928673), (15, 0.03430921956896782), (16, 0.04402923211455345), (17, 0.0456220842897892), (18, 0.03564029186964035), (19, 0.053765058517456055), (20, 0.18513290584087372), (21, 0.21955378353595734), (22, 0.09346309304237366), (23, 0.15266670286655426), (24, 0.4756668508052826), (25, 0.48464328050613403), (26, 0.28695544600486755), (27, 0.39167460799217224)]
+
+100%|██████████| 30/30 [05:30<00:00, 11.80s/it]
+100%|██████████| 30/30 [05:30<00:00, 11.00s/it]
+[2024-10-01 15:13:06] Prompt: snow falling over multiple houses and trees on winter landscape against night sky. christmas festivity and celebration concept. aesthetic score: 6.5.
+Saved to ./samples/opensora/sample_pab/Snow falling over multiple houses and tr.mp4
+Saved video to ./samples/opensora/sample_pab/Snow falling over multiple houses and tr.mp4
+Saved spatial MLP MSE outputs to ./samples/opensora/sample_pab/Snow falling over multiple houses and tr_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora/sample_pab/Snow falling over multiple houses and tr_temporal_mlp_mse.pt
+timesteps: [tensor([1000.], device='cuda:0'), tensor([992.9313], device='cuda:0'), tensor([985.4678], device='cuda:0'), tensor([977.5753], device='cuda:0'), tensor([969.2160], device='cuda:0'), tensor([960.3470], device='cuda:0'), tensor([950.9203], device='cuda:0'), tensor([940.8815], device='cuda:0'), tensor([930.1691], device='cuda:0'), tensor([918.7130], device='cuda:0'), tensor([906.4327], device='cuda:0'), tensor([893.2362], device='cuda:0'), tensor([879.0170], device='cuda:0'), tensor([863.6513], device='cuda:0'), tensor([846.9945], device='cuda:0'), tensor([828.8770], device='cuda:0'), tensor([809.0977], device='cuda:0'), tensor([787.4170], device='cuda:0'), tensor([763.5468], device='cuda:0'), tensor([737.1379], device='cuda:0'), tensor([707.7626], device='cuda:0'), tensor([674.8911], device='cuda:0'), tensor([637.8601], device='cuda:0'), tensor([595.8265], device='cuda:0'), tensor([547.7032], device='cuda:0'), tensor([492.0635], device='cuda:0'), tensor([426.9973], device='cuda:0'), tensor([349.8871], device='cuda:0'), tensor([257.0481], device='cuda:0'), tensor([143.1210], device='cuda:0')]
+
+  0%|          | 0/30 [00:00<?, ?it/s]
+  3%|▎         | 1/30 [00:03<01:48,  3.75s/it]Time step 992, Spatial MSE: [(0, 0.0009034814429469407), (1, 0.0018998270388692617), (2, 0.014445044100284576), (3, 0.021470127627253532), (4, 0.025045622140169144), (5, 0.04280921444296837), (6, 0.04432007297873497), (7, 0.06651034206151962), (8, 0.08734379708766937), (9, 0.19936446845531464), (10, 0.3888300657272339), (11, 0.34386616945266724), (12, 0.36684489250183105), (13, 0.32413581013679504), (14, 0.7708515524864197), (15, 0.5985571146011353), (16, 1.2180510759353638), (17, 0.6783449649810791), (18, 0.45242053270339966), (19, 0.5873988270759583), (20, 0.8689930438995361), (21, 0.6541589498519897), (22, 0.5959876775741577), (23, 1.3141224384307861), (24, 7.015657424926758), (25, 6.165170669555664), (26, 6.01273775100708), (27, 0.487280011177063)], Temporal MSE: [(0, 4.576309220283292e-05), (1, 0.00042874625069089234), (2, 0.0014869810547679663), (3, 0.0018123692134395242), (4, 0.002646995009854436), (5, 0.007088537327945232), (6, 0.00825102161616087), (7, 0.011654362082481384), (8, 0.010471867397427559), (9, 0.01513741910457611), (10, 0.03210221976041794), (11, 0.03306734189391136), (12, 0.03838055208325386), (13, 0.027130544185638428), (14, 0.09120739251375198), (15, 0.1371115893125534), (16, 0.18083930015563965), (17, 0.07312297821044922), (18, 0.08373244851827621), (19, 0.09780039638280869), (20, 0.1087692528963089), (21, 0.1510612815618515), (22, 0.5293027758598328), (23, 0.48894041776657104), (24, 0.353137344121933), (25, 0.1491704136133194), (26, 0.6347561478614807), (27, 0.665677547454834)]
+
+  7%|▋         | 2/30 [00:16<04:19,  9.28s/it]Time step 984, Spatial MSE: [(0, 0.00036458740942180157), (1, 0.001317496644333005), (2, 0.005865172483026981), (3, 0.005922127515077591), (4, 0.007372791413217783), (5, 0.010126126930117607), (6, 0.014862448908388615), (7, 0.026061084121465683), (8, 0.037709638476371765), (9, 0.07926461845636368), (10, 0.15633907914161682), (11, 0.19859570264816284), (12, 0.19767004251480103), (13, 0.17647767066955566), (14, 0.32113346457481384), (15, 0.2937110662460327), (16, 0.3424801230430603), (17, 0.3022148013114929), (18, 0.28235650062561035), (19, 0.27327847480773926), (20, 0.3369826674461365), (21, 0.31935903429985046), (22, 0.28025519847869873), (23, 0.3304067552089691), (24, 1.0641058683395386), (25, 1.3776681423187256), (26, 1.273385763168335), (27, 0.1606261283159256)], Temporal MSE: [(0, 6.942830168554792e-06), (1, 0.00014477352669928223), (2, 0.0003179325140081346), (3, 0.0005097099347040057), (4, 0.000604836386628449), (5, 0.0012652910081669688), (6, 0.0013769547222182155), (7, 0.005454662721604109), (8, 0.004026819486171007), (9, 0.0057366047985851765), (10, 0.013305273838341236), (11, 0.014631739817559719), (12, 0.013964383862912655), (13, 0.012757665477693081), (14, 0.045696843415498734), (15, 0.059064313769340515), (16, 0.0571233332157135), (17, 0.03936736658215523), (18, 0.03811033070087433), (19, 0.04874478653073311), (20, 0.06820098310709), (21, 0.10739099979400635), (22, 0.23278070986270905), (23, 0.19165050983428955), (24, 0.10490591824054718), (25, 0.05399481952190399), (26, 0.16020245850086212), (27, 0.10184919834136963)]
+
+ 10%|█         | 3/30 [00:26<04:17,  9.53s/it]Time step 976, Spatial MSE: [(0, 0.00016801044694148004), (1, 0.0007467159884981811), (2, 0.0021623950451612473), (3, 0.002146713435649872), (4, 0.0036848997697234154), (5, 0.005162544082850218), (6, 0.009611217305064201), (7, 0.01760050281882286), (8, 0.029385823756456375), (9, 0.04696415364742279), (10, 0.08217813074588776), (11, 0.09237085282802582), (12, 0.10759510099887848), (13, 0.12507203221321106), (14, 0.20485301315784454), (15, 0.19373950362205505), (16, 0.2527734935283661), (17, 0.2078486531972885), (18, 0.23270222544670105), (19, 0.24011163413524628), (20, 0.34352347254753113), (21, 0.27305638790130615), (22, 0.1907566636800766), (23, 0.24148832261562347), (24, 0.5370064973831177), (25, 0.897817075252533), (26, 0.5802884101867676), (27, 0.09550143033266068)], Temporal MSE: [(0, 3.2124949029821437e-06), (1, 5.062513810116798e-05), (2, 0.00011967137106694281), (3, 0.00020949887402821332), (4, 0.000263775116764009), (5, 0.0005630180821754038), (6, 0.0010131358867511153), (7, 0.002765381010249257), (8, 0.002013057004660368), (9, 0.0030387050937861204), (10, 0.007239467464387417), (11, 0.011356046423316002), (12, 0.009475570172071457), (13, 0.00866658240556717), (14, 0.0357208326458931), (15, 0.03723887726664543), (16, 0.04486418515443802), (17, 0.034124456346035004), (18, 0.027486108243465424), (19, 0.03736647218465805), (20, 0.06728187948465347), (21, 0.0770234614610672), (22, 0.1322050243616104), (23, 0.11872918903827667), (24, 0.06525849550962448), (25, 0.05521872267127037), (26, 0.06348206102848053), (27, 0.045247141271829605)]
+
+ 13%|█▎        | 4/30 [00:37<04:16,  9.87s/it]Time step 968, Spatial MSE: [(0, 9.654593304730952e-05), (1, 0.0004292460507713258), (2, 0.0011900124372914433), (3, 0.0006714680348522961), (4, 0.0009911204688251019), (5, 0.0014366141986101866), (6, 0.0033648977987468243), (7, 0.005034149624407291), (8, 0.007766650523990393), (9, 0.014327977783977985), (10, 0.019481930881738663), (11, 0.024568334221839905), (12, 0.02337251603603363), (13, 0.02712804451584816), (14, 0.05430188030004501), (15, 0.05560969561338425), (16, 0.06305485963821411), (17, 0.06118632107973099), (18, 0.056185267865657806), (19, 0.04973495751619339), (20, 0.047924596816301346), (21, 0.052481792867183685), (22, 0.046930134296417236), (23, 0.04696695879101753), (24, 0.09707578271627426), (25, 0.1320425570011139), (26, 0.08730882406234741), (27, 0.03439868986606598)], Temporal MSE: [(0, 1.714018026177655e-06), (1, 2.26614938583225e-05), (2, 7.769616058794782e-05), (3, 9.905337356030941e-05), (4, 0.00012829461775254458), (5, 0.0005172242526896298), (6, 0.00031548686092719436), (7, 0.0008473668131045997), (8, 0.0007289639906957746), (9, 0.0009256555931642652), (10, 0.001510990085080266), (11, 0.0022123202215880156), (12, 0.0019427952356636524), (13, 0.0019679737742990255), (14, 0.006402514409273863), (15, 0.008046690374612808), (16, 0.009438131004571915), (17, 0.00837702490389347), (18, 0.008272353559732437), (19, 0.009452302008867264), (20, 0.01737673580646515), (21, 0.020784834399819374), (22, 0.019032031297683716), (23, 0.021910160779953003), (24, 0.011230092495679855), (25, 0.009686345234513283), (26, 0.0115663418546319), (27, 0.008234425447881222)]
+
+ 17%|█▋        | 5/30 [00:47<04:13, 10.13s/it]Time step 960, Spatial MSE: [(0, 6.0811289586126804e-05), (1, 0.0005058854003436863), (2, 0.0009603449725545943), (3, 0.0006589259137399495), (4, 0.0006276838248595595), (5, 0.0008709240355528891), (6, 0.002228435128927231), (7, 0.0034374301321804523), (8, 0.004034264944493771), (9, 0.00785973109304905), (10, 0.012099872343242168), (11, 0.014850622043013573), (12, 0.013059715740382671), (13, 0.013754836283624172), (14, 0.0455472432076931), (15, 0.047504834830760956), (16, 0.051641736179590225), (17, 0.05257430672645569), (18, 0.047339826822280884), (19, 0.04256707429885864), (20, 0.042208947241306305), (21, 0.04521765932440758), (22, 0.06905613839626312), (23, 0.060629021376371384), (24, 0.0725758969783783), (25, 0.07492014765739441), (26, 0.042481448501348495), (27, 0.02081620879471302)], Temporal MSE: [(0, 1.393808588545653e-06), (1, 1.3119064533384517e-05), (2, 6.110242975410074e-05), (3, 7.885014929343015e-05), (4, 0.00012650205462705344), (5, 0.00038176728412508965), (6, 0.0002988921187352389), (7, 0.0008582696900703013), (8, 0.0005948135512880981), (9, 0.0006280190427787602), (10, 0.0010831172112375498), (11, 0.0014927851734682918), (12, 0.0013801720924675465), (13, 0.0013743782183155417), (14, 0.006418984849005938), (15, 0.007640238385647535), (16, 0.008851653896272182), (17, 0.010211575776338577), (18, 0.01109939906746149), (19, 0.011218830943107605), (20, 0.018777497112751007), (21, 0.02701527439057827), (22, 0.02285527065396309), (23, 0.017955364659428596), (24, 0.010435700416564941), (25, 0.010932532139122486), (26, 0.007098371163010597), (27, 0.005032146815210581)]
+
+ 20%|██        | 6/30 [00:57<04:03, 10.13s/it]Time step 952, Spatial MSE: [(0, 5.095432425150648e-05), (1, 0.00035977771040052176), (2, 0.0006832531653344631), (3, 0.00033875147346407175), (4, 0.0004020181659143418), (5, 0.0005740044289268553), (6, 0.0013897091848775744), (7, 0.0027574903797358274), (8, 0.002979404991492629), (9, 0.00582730071619153), (10, 0.01244927104562521), (11, 0.013836674392223358), (12, 0.012840038165450096), (13, 0.013662816025316715), (14, 0.06073010712862015), (15, 0.062141112983226776), (16, 0.06739959120750427), (17, 0.06035764887928963), (18, 0.05576557666063309), (19, 0.04472171887755394), (20, 0.04048454761505127), (21, 0.04981214553117752), (22, 0.06721329689025879), (23, 0.06050892546772957), (24, 0.06876469403505325), (25, 0.06473628431558609), (26, 0.03892284631729126), (27, 0.01871464028954506)], Temporal MSE: [(0, 1.0579684612821438e-06), (1, 8.7497137428727e-06), (2, 4.081624501850456e-05), (3, 4.7476034524152055e-05), (4, 6.678451609332114e-05), (5, 0.0001538142969366163), (6, 0.00017995790403801948), (7, 0.0006579136243090034), (8, 0.00045874385978095233), (9, 0.0005387485143728554), (10, 0.0013148506404832006), (11, 0.0019281415734440088), (12, 0.0018882156582549214), (13, 0.0022145211696624756), (14, 0.010921711102128029), (15, 0.013325095176696777), (16, 0.012632559053599834), (17, 0.014875557273626328), (18, 0.013274935074150562), (19, 0.013322899118065834), (20, 0.020487260073423386), (21, 0.030442554503679276), (22, 0.02366027608513832), (23, 0.017157237976789474), (24, 0.010990114882588387), (25, 0.008118594996631145), (26, 0.006493722088634968), (27, 0.004109179135411978)]
+
+ 23%|██▎       | 7/30 [01:09<04:04, 10.63s/it]Time step 940, Spatial MSE: [(0, 5.9478566981852055e-05), (1, 0.0003283401019871235), (2, 0.0006801537238061428), (3, 0.0004030933487229049), (4, 0.0005581195582635701), (5, 0.000731218489818275), (6, 0.001922497060149908), (7, 0.004643882159143686), (8, 0.0036181595642119646), (9, 0.007735861465334892), (10, 0.01652614399790764), (11, 0.014216684736311436), (12, 0.014217437244951725), (13, 0.015009024180471897), (14, 0.04501868039369583), (15, 0.046853940933942795), (16, 0.050814904272556305), (17, 0.04143883287906647), (18, 0.037755534052848816), (19, 0.04019882529973984), (20, 0.038257796317338943), (21, 0.0448317676782608), (22, 0.049065496772527695), (23, 0.04612739011645317), (24, 0.0661567822098732), (25, 0.07326620817184448), (26, 0.042421821504831314), (27, 0.029574519023299217)], Temporal MSE: [(0, 1.328247094534163e-06), (1, 9.179816515825223e-06), (2, 5.6565917475381866e-05), (3, 4.07175648433622e-05), (4, 0.00012689572758972645), (5, 0.00019714669906534255), (6, 0.00029151278431527317), (7, 0.0012563701020553708), (8, 0.0008521712734363973), (9, 0.0009705651318654418), (10, 0.002153808483853936), (11, 0.0022995404433459044), (12, 0.0020464612171053886), (13, 0.0020210035145282745), (14, 0.010567955672740936), (15, 0.011258549988269806), (16, 0.010340207256376743), (17, 0.011321617290377617), (18, 0.011668205261230469), (19, 0.012058401480317116), (20, 0.016833657398819923), (21, 0.019718555733561516), (22, 0.01616809144616127), (23, 0.015795884653925896), (24, 0.009966405108571053), (25, 0.006358490791171789), (26, 0.008152876049280167), (27, 0.006665317341685295)]
+
+ 27%|██▋       | 8/30 [01:22<04:07, 11.23s/it]Time step 932, Spatial MSE: [(0, 4.215761509840377e-05), (1, 0.0003113538259640336), (2, 0.0004296828410588205), (3, 0.0002726498059928417), (4, 0.00024208734976127744), (5, 0.0003806479217018932), (6, 0.0009450454381294549), (7, 0.0016667988384142518), (8, 0.0018898952985182405), (9, 0.0034985903184860945), (10, 0.005874862894415855), (11, 0.006826518569141626), (12, 0.005846144165843725), (13, 0.006743471138179302), (14, 0.016423828899860382), (15, 0.019694482907652855), (16, 0.020631449297070503), (17, 0.018185608088970184), (18, 0.018734563142061234), (19, 0.019360698759555817), (20, 0.01906430907547474), (21, 0.021299604326486588), (22, 0.023589501157402992), (23, 0.021308202296495438), (24, 0.027975616976618767), (25, 0.02937636524438858), (26, 0.02106255292892456), (27, 0.010017863474786282)], Temporal MSE: [(0, 9.208426376972056e-07), (1, 3.749289589904947e-06), (2, 2.953453986265231e-05), (3, 3.3784348488552496e-05), (4, 5.732642239308916e-05), (5, 0.00010276517423335463), (6, 0.00013007799861952662), (7, 0.00040269558667205274), (8, 0.00031579608912579715), (9, 0.00037932643317617476), (10, 0.0008179069845937192), (11, 0.0009102844051085413), (12, 0.0006981247570365667), (13, 0.0008202095050364733), (14, 0.0040262676775455475), (15, 0.0045011090114712715), (16, 0.004463807214051485), (17, 0.005281141493469477), (18, 0.005172309931367636), (19, 0.00549125112593174), (20, 0.00922441203147173), (21, 0.011464599519968033), (22, 0.007850180380046368), (23, 0.007805585395544767), (24, 0.005184162873774767), (25, 0.0024112756364047527), (26, 0.003733712015673518), (27, 0.0022398114670068026)]
+
+ 30%|███       | 9/30 [01:32<03:51, 11.02s/it]Time step 920, Spatial MSE: [(0, 2.9048740543657914e-05), (1, 0.0003755608049686998), (2, 0.0005339890485629439), (3, 0.00018162009655497968), (4, 7.219341932795942e-05), (5, 0.00011212802201043814), (6, 0.00014090193144511431), (7, 0.0006308747106231749), (8, 0.00017578821280039847), (9, 0.00036682479549199343), (10, 0.0003243708924856037), (11, 0.0003036698908545077), (12, 0.0004996896022930741), (13, 0.0004925125394947827), (14, 0.0007647446473129094), (15, 0.0013875382719561458), (16, 0.003691648133099079), (17, 0.001719473279081285), (18, 0.0027475326787680387), (19, 0.0019065403612330556), (20, 0.0017510488396510482), (21, 0.0023835282772779465), (22, 0.0017770514823496342), (23, 0.0025631857570260763), (24, 0.005351323168724775), (25, 0.012462357059121132), (26, 0.010355857200920582), (27, 0.013518104329705238)], Temporal MSE: [(0, 8.187400339920714e-07), (1, 3.2516845749341883e-06), (2, 1.3808040421281476e-05), (3, 1.3925561688665766e-05), (4, 1.3544963621825445e-05), (5, 1.5784436982357875e-05), (6, 2.133806628989987e-05), (7, 6.71110610710457e-05), (8, 2.751180545601528e-05), (9, 5.502948260982521e-05), (10, 8.049238385865465e-05), (11, 7.4092808063142e-05), (12, 7.504197128582746e-05), (13, 7.466621900675818e-05), (14, 0.0002265881485072896), (15, 0.0006923606269992888), (16, 0.0004875505983363837), (17, 0.0007062368094921112), (18, 0.0009424392483197153), (19, 0.0009387439931742847), (20, 0.0017768904799595475), (21, 0.0009766860166564584), (22, 0.0007108885911293328), (23, 0.0008887932053767145), (24, 0.0005218797596171498), (25, 0.0007943589007481933), (26, 0.0011830112198367715), (27, 0.0008980154525488615)]
+
+ 33%|███▎      | 10/30 [01:44<03:42, 11.15s/it]Time step 908, Spatial MSE: [(0, 9.004227467812598e-05), (1, 0.0003906298370566219), (2, 0.0006886573974043131), (3, 0.0002698686148505658), (4, 0.00037535751471295953), (5, 0.0007425211952067912), (6, 0.0015218588523566723), (7, 0.002255515195429325), (8, 0.0015784867573529482), (9, 0.002920369617640972), (10, 0.006514149717986584), (11, 0.005989163648337126), (12, 0.006068720947951078), (13, 0.006710325367748737), (14, 0.021550459787249565), (15, 0.013560949824750423), (16, 0.02404944971203804), (17, 0.01214413158595562), (18, 0.011752000078558922), (19, 0.013526102527976036), (20, 0.012004410848021507), (21, 0.020021261647343636), (22, 0.016496911644935608), (23, 0.01265049446374178), (24, 0.022374330088496208), (25, 0.04143877327442169), (26, 0.014975042082369328), (27, 0.020500801503658295)], Temporal MSE: [(0, 2.4702719656488625e-06), (1, 3.774365495701204e-06), (2, 3.101949914707802e-05), (3, 6.818979454692453e-05), (4, 5.014183625462465e-05), (5, 0.0001061429429682903), (6, 0.00012153379066148773), (7, 0.0008137113181874156), (8, 0.0004668596666306257), (9, 0.000531572091858834), (10, 0.0013061054050922394), (11, 0.0010480402270331979), (12, 0.0014657689025625587), (13, 0.0007298589334823191), (14, 0.004591429140418768), (15, 0.005377530585974455), (16, 0.007426577620208263), (17, 0.004022254608571529), (18, 0.002921057865023613), (19, 0.005759872030466795), (20, 0.005032643210142851), (21, 0.0071440450847148895), (22, 0.008458851836621761), (23, 0.007804123219102621), (24, 0.004452962428331375), (25, 0.003370896680280566), (26, 0.005300118587911129), (27, 0.004092983901500702)]
+
+ 37%|███▋      | 11/30 [01:54<03:27, 10.91s/it]Time step 892, Spatial MSE: [(0, 2.9723883926635608e-05), (1, 0.0004961915547028184), (2, 0.0005975149688310921), (3, 0.00022127223201096058), (4, 6.324148853309453e-05), (5, 0.00010554292384767905), (6, 0.00012229655112605542), (7, 0.0004916397738270462), (8, 0.00017136587121058255), (9, 0.0003531911061145365), (10, 0.0002826970594469458), (11, 0.0003399394336156547), (12, 0.00037012138636782765), (13, 0.0005254263523966074), (14, 0.0007748589850962162), (15, 0.0011514781508594751), (16, 0.005001400597393513), (17, 0.0015772698679938912), (18, 0.001914838794618845), (19, 0.0028082123026251793), (20, 0.0022351399529725313), (21, 0.003932403866201639), (22, 0.0023403773084282875), (23, 0.003428185125812888), (24, 0.007285567931830883), (25, 0.010915838181972504), (26, 0.01182746421545744), (27, 0.00956172775477171)], Temporal MSE: [(0, 7.836771374059026e-07), (1, 3.1149590995482868e-06), (2, 1.3605481399281416e-05), (3, 1.3919218872615602e-05), (4, 1.3229991054686252e-05), (5, 1.6923304428928532e-05), (6, 1.7322596249869093e-05), (7, 6.218931957846507e-05), (8, 2.8877233489765786e-05), (9, 4.353658005129546e-05), (10, 7.399491005344316e-05), (11, 6.703859980916604e-05), (12, 6.41861988697201e-05), (13, 7.237327372422442e-05), (14, 0.00023527216399088502), (15, 0.000622964056674391), (16, 0.0005655093700625002), (17, 0.0007356833666563034), (18, 0.00048180544399656355), (19, 0.001367861172184348), (20, 0.0031670276075601578), (21, 0.0011378168128430843), (22, 0.0009165371302515268), (23, 0.0009438122506253421), (24, 0.0007001908961683512), (25, 0.0007157934596762061), (26, 0.0009833064395934343), (27, 0.000836221210192889)]
+
+ 40%|████      | 12/30 [02:03<03:08, 10.46s/it]Time step 880, Spatial MSE: [(0, 9.995748405344784e-05), (1, 0.000424756231950596), (2, 0.0009064009063877165), (3, 0.0014687322545796633), (4, 0.0017907030414789915), (5, 0.002898354083299637), (6, 0.007003676611930132), (7, 0.018988028168678284), (8, 0.018232446163892746), (9, 0.03246414288878441), (10, 0.0630747377872467), (11, 0.06481392681598663), (12, 0.05566546693444252), (13, 0.05749901384115219), (14, 0.16244952380657196), (15, 0.19950301945209503), (16, 0.1766795516014099), (17, 0.1889120191335678), (18, 0.18266728520393372), (19, 0.17877300083637238), (20, 0.17196376621723175), (21, 0.247264102101326), (22, 0.24699623882770538), (23, 0.23027651011943817), (24, 0.27149081230163574), (25, 0.24418684840202332), (26, 0.1513204425573349), (27, 0.04083549603819847)], Temporal MSE: [(0, 2.850408236554358e-06), (1, 2.4825574655551463e-05), (2, 0.00043989354162476957), (3, 0.0003935817803721875), (4, 0.0007481435895897448), (5, 0.0008195592672564089), (6, 0.0018968600779771805), (7, 0.005355983506888151), (8, 0.00447674049064517), (9, 0.004435308743268251), (10, 0.010983451269567013), (11, 0.011329112574458122), (12, 0.008363423869013786), (13, 0.009902900084853172), (14, 0.039666276425123215), (15, 0.04064986854791641), (16, 0.04803409054875374), (17, 0.045267559587955475), (18, 0.04233268275856972), (19, 0.050103429704904556), (20, 0.09754353016614914), (21, 0.11138270050287247), (22, 0.07379758358001709), (23, 0.07710687816143036), (24, 0.05259561166167259), (25, 0.023582296445965767), (26, 0.020996367558836937), (27, 0.01680171862244606)]
+
+ 43%|████▎     | 13/30 [02:15<03:03, 10.82s/it]Time step 864, Spatial MSE: [(0, 2.00581198441796e-05), (1, 0.00045395310735329986), (2, 0.0005792034207843244), (3, 0.00023846076510380954), (4, 5.8422341680852696e-05), (5, 8.18755361251533e-05), (6, 0.00012386197340674698), (7, 0.000315765937557444), (8, 0.00012179846817161888), (9, 0.00020400132052600384), (10, 0.0002718660107348114), (11, 0.00024535218835808337), (12, 0.00030929839704185724), (13, 0.00046909519005566835), (14, 0.0007085784454829991), (15, 0.00080651487223804), (16, 0.005679619498550892), (17, 0.001368975848890841), (18, 0.001222853665240109), (19, 0.004698526579886675), (20, 0.0036669012624770403), (21, 0.012031619437038898), (22, 0.01229893695563078), (23, 0.004584051202982664), (24, 0.007269789464771748), (25, 0.012310676276683807), (26, 0.012150991708040237), (27, 0.012557610869407654)], Temporal MSE: [(0, 5.24068923368759e-07), (1, 2.50881680585735e-06), (2, 9.455005965719465e-06), (3, 1.0559654583630618e-05), (4, 9.96781363937771e-06), (5, 1.3313379895407706e-05), (6, 1.4333658327814192e-05), (7, 3.454158650129102e-05), (8, 2.219276939285919e-05), (9, 2.9516131689888425e-05), (10, 6.271849997574463e-05), (11, 4.832843842450529e-05), (12, 4.338301732786931e-05), (13, 5.242667248239741e-05), (14, 0.00021054504031781107), (15, 0.00039151881355792284), (16, 0.0006902576424181461), (17, 0.0005987525801174343), (18, 0.00024887247127480805), (19, 0.0019263821886852384), (20, 0.0047980137169361115), (21, 0.003472080221399665), (22, 0.004100844729691744), (23, 0.0008289242396131158), (24, 0.0007962119998410344), (25, 0.0006869661738164723), (26, 0.0007387709338217974), (27, 0.0016122496454045177)]
+
+ 47%|████▋     | 14/30 [02:26<02:52, 10.75s/it]Time step 848, Spatial MSE: [(0, 0.00011189124052179977), (1, 0.0004062514053657651), (2, 0.0006165045779198408), (3, 0.000244724826188758), (4, 0.0003311919863335788), (5, 0.0008885803981684148), (6, 0.0016292963409796357), (7, 0.002437020419165492), (8, 0.0015766980359330773), (9, 0.002886291826143861), (10, 0.008277290500700474), (11, 0.00616676127538085), (12, 0.0057639372535049915), (13, 0.007832269184291363), (14, 0.02581148035824299), (15, 0.02033044397830963), (16, 0.021672314032912254), (17, 0.014300198294222355), (18, 0.014987976290285587), (19, 0.016345616430044174), (20, 0.01731104589998722), (21, 0.019801178947091103), (22, 0.01896430365741253), (23, 0.013319341465830803), (24, 0.02463880367577076), (25, 0.03569670021533966), (26, 0.0195373073220253), (27, 0.010360450483858585)], Temporal MSE: [(0, 2.471271500326111e-06), (1, 3.970480520365527e-06), (2, 1.9102810256299563e-05), (3, 4.643004649551585e-05), (4, 6.970907998038456e-05), (5, 0.00013769596989732236), (6, 0.0001898664195323363), (7, 0.0008853043545968831), (8, 0.0005706918891519308), (9, 0.000807030766736716), (10, 0.0022316002286970615), (11, 0.0013840635074302554), (12, 0.001891452819108963), (13, 0.0010146270506083965), (14, 0.0063972692005336285), (15, 0.009648028761148453), (16, 0.009413042105734348), (17, 0.006643953733146191), (18, 0.004281738772988319), (19, 0.007188145071268082), (20, 0.01081644557416439), (21, 0.007373015396296978), (22, 0.006279032211750746), (23, 0.0068807536736130714), (24, 0.004700637422502041), (25, 0.0025207153521478176), (26, 0.004990222863852978), (27, 0.0029149753972887993)]
+
+ 50%|█████     | 15/30 [02:35<02:35, 10.37s/it]Time step 828, Spatial MSE: [(0, 1.954396066139452e-05), (1, 0.0004813812847714871), (2, 0.0007120893569663167), (3, 0.00027880334528163075), (4, 4.6462544560199603e-05), (5, 8.017849177122116e-05), (6, 7.966806151671335e-05), (7, 0.00030398834496736526), (8, 0.00011836805788334459), (9, 0.00019603875989560038), (10, 0.00019912880088668317), (11, 0.0001946980191860348), (12, 0.000324591645039618), (13, 0.00037675982457585633), (14, 0.0005579218850471079), (15, 0.0005928305326960981), (16, 0.004441075026988983), (17, 0.0010375026613473892), (18, 0.0012816495727747679), (19, 0.0024319791700690985), (20, 0.0017837948398664594), (21, 0.003689444623887539), (22, 0.0022516411263495684), (23, 0.003451770870015025), (24, 0.006393676158040762), (25, 0.012347170151770115), (26, 0.011712632142007351), (27, 0.016301527619361877)], Temporal MSE: [(0, 5.298548444443441e-07), (1, 1.7353413568343967e-06), (2, 9.540261089568958e-06), (3, 9.106234756472986e-06), (4, 9.624252925277688e-06), (5, 1.7763153664418496e-05), (6, 1.4425875633605756e-05), (7, 3.873649620800279e-05), (8, 2.4553746698074974e-05), (9, 3.0626524676335976e-05), (10, 5.126480391481891e-05), (11, 4.281265864847228e-05), (12, 5.4518204706255347e-05), (13, 4.9674079491524026e-05), (14, 0.00015433701628353447), (15, 0.0002586686459835619), (16, 0.0005031383479945362), (17, 0.0004104967520106584), (18, 0.0002445725549478084), (19, 0.0012402413412928581), (20, 0.0029991264455020428), (21, 0.0012701902305707335), (22, 0.0008111280621960759), (23, 0.000967706146184355), (24, 0.0006946215871721506), (25, 0.000757858157157898), (26, 0.00065470056142658), (27, 0.0014744179788976908)]
+
+ 53%|█████▎    | 16/30 [02:43<02:13,  9.55s/it]Time step 808, Spatial MSE: [(0, 0.0001054376843967475), (1, 0.0006860038847662508), (2, 0.0008278058958239853), (3, 0.0012950720265507698), (4, 0.0015681613003835082), (5, 0.002709700260311365), (6, 0.005820801947265863), (7, 0.015026502311229706), (8, 0.015240516513586044), (9, 0.026724625378847122), (10, 0.04067626968026161), (11, 0.03993890434503555), (12, 0.03499683365225792), (13, 0.038499537855386734), (14, 0.0938761979341507), (15, 0.10572449117898941), (16, 0.10747472196817398), (17, 0.11094197630882263), (18, 0.12185219675302505), (19, 0.1295597404241562), (20, 0.1390223205089569), (21, 0.19771380722522736), (22, 0.20438426733016968), (23, 0.19510570168495178), (24, 0.20694950222969055), (25, 0.18867433071136475), (26, 0.12710091471672058), (27, 0.025485724210739136)], Temporal MSE: [(0, 2.8018589546263684e-06), (1, 2.1724887119489722e-05), (2, 0.0002119235577993095), (3, 0.00025743243168108165), (4, 0.0006124945357441902), (5, 0.0008136908290907741), (6, 0.0016590255545452237), (7, 0.003849840024486184), (8, 0.002935680327937007), (9, 0.0038686287589371204), (10, 0.007554417476058006), (11, 0.007916188798844814), (12, 0.0058919526636600494), (13, 0.007764365524053574), (14, 0.022221939638257027), (15, 0.023740414530038834), (16, 0.03421778231859207), (17, 0.031633660197257996), (18, 0.03847794607281685), (19, 0.05937284231185913), (20, 0.10528095066547394), (21, 0.08968812972307205), (22, 0.06732642650604248), (23, 0.06566555798053741), (24, 0.04425133392214775), (25, 0.012453900650143623), (26, 0.015856970101594925), (27, 0.015441929921507835)]
+
+ 57%|█████▋    | 17/30 [02:54<02:09,  9.96s/it]Time step 788, Spatial MSE: [(0, 1.921961120388005e-05), (1, 0.0004471860302146524), (2, 0.000835329876281321), (3, 0.00033306112163700163), (4, 4.805396747542545e-05), (5, 7.266183092724532e-05), (6, 7.651059422641993e-05), (7, 0.0004957060446031392), (8, 0.00015265426191035658), (9, 0.00017553687212057412), (10, 0.00020967854652553797), (11, 0.00021620586630888283), (12, 0.0003172399301547557), (13, 0.00042197489528916776), (14, 0.0006419984856620431), (15, 0.0005334964953362942), (16, 0.007051278371363878), (17, 0.0012704830151051283), (18, 0.00781684834510088), (19, 0.0028439336456358433), (20, 0.0016468717949464917), (21, 0.003151846118271351), (22, 0.0022707912139594555), (23, 0.00346354185603559), (24, 0.006115790456533432), (25, 0.014925754629075527), (26, 0.012457743287086487), (27, 0.010062032379209995)], Temporal MSE: [(0, 5.503012516783201e-07), (1, 1.6524664943062817e-06), (2, 1.0017653039540164e-05), (3, 8.732125934329815e-06), (4, 9.443327144253999e-06), (5, 1.629806683922652e-05), (6, 1.6134785255417228e-05), (7, 3.6420799006009474e-05), (8, 2.395993760728743e-05), (9, 3.165140515193343e-05), (10, 5.4367159464163706e-05), (11, 4.9915328418137506e-05), (12, 4.3885298509849235e-05), (13, 5.785646862932481e-05), (14, 0.00017115265654865652), (15, 0.0002364221727475524), (16, 0.0006558867171406746), (17, 0.00037508481182157993), (18, 0.0021165930666029453), (19, 0.0020915125496685505), (20, 0.005500416271388531), (21, 0.0016419871244579554), (22, 0.0007475444581359625), (23, 0.000851171207614243), (24, 0.0006641775835305452), (25, 0.0010432616109028459), (26, 0.0006996242445893586), (27, 0.0010352696990594268)]
+
+ 60%|██████    | 18/30 [03:04<01:59,  9.96s/it]Time step 764, Spatial MSE: [(0, 0.0001268546184292063), (1, 0.000586872803978622), (2, 0.0006869320059195161), (3, 0.0002383727114647627), (4, 0.00025069748517125845), (5, 0.000776497763581574), (6, 0.0013120599323883653), (7, 0.001594003988429904), (8, 0.0013811185490339994), (9, 0.002395007526502013), (10, 0.007293921895325184), (11, 0.00404564943164587), (12, 0.004710470326244831), (13, 0.006914243567734957), (14, 0.022588690742850304), (15, 0.018878836184740067), (16, 0.0204476285725832), (17, 0.015572766773402691), (18, 0.014918974600732327), (19, 0.017257768660783768), (20, 0.013672971166670322), (21, 0.022930817678570747), (22, 0.017941264435648918), (23, 0.018254773691296577), (24, 0.027336305007338524), (25, 0.04655546322464943), (26, 0.023121340200304985), (27, 0.00865992996841669)], Temporal MSE: [(0, 2.50043694904889e-06), (1, 5.198462986300001e-06), (2, 1.8294835172127932e-05), (3, 3.547868982423097e-05), (4, 4.35385518358089e-05), (5, 8.614854596089572e-05), (6, 0.0001366557989967987), (7, 0.00048185052582994103), (8, 0.0003953313862439245), (9, 0.0007387235527858138), (10, 0.0017392875161021948), (11, 0.0013345363549888134), (12, 0.001803138293325901), (13, 0.0011889188317582011), (14, 0.005646763369441032), (15, 0.00788916926831007), (16, 0.007932605221867561), (17, 0.005833110306411982), (18, 0.0056510018184781075), (19, 0.007059506606310606), (20, 0.009758698754012585), (21, 0.009046645835042), (22, 0.0066453684121370316), (23, 0.006626309361308813), (24, 0.006021005567163229), (25, 0.0020146253518760204), (26, 0.0033168261870741844), (27, 0.0023587660398334265)]
+
+ 63%|██████▎   | 19/30 [03:14<01:52, 10.24s/it]Time step 736, Spatial MSE: [(0, 2.7572392355068587e-05), (1, 0.0007251285715028644), (2, 0.0011486404109746218), (3, 0.0005255856667645276), (4, 6.839283742010593e-05), (5, 0.00010180005483562127), (6, 0.00010868661775020882), (7, 0.00037390069337561727), (8, 0.00014256592839956284), (9, 0.00024614844005554914), (10, 0.00024424493312835693), (11, 0.00026951971813105047), (12, 0.0003746023867279291), (13, 0.0005286941304802895), (14, 0.0009740524692460895), (15, 0.0009703278774395585), (16, 0.014296937733888626), (17, 0.0029909000732004642), (18, 0.002990877954289317), (19, 0.0030109488870948553), (20, 0.0024542161263525486), (21, 0.005998510867357254), (22, 0.0030722878873348236), (23, 0.004940933082252741), (24, 0.008261597715318203), (25, 0.019080568104982376), (26, 0.016702519729733467), (27, 0.010519825853407383)], Temporal MSE: [(0, 6.815928372816416e-07), (1, 2.34105368690507e-06), (2, 1.5756184438942e-05), (3, 1.2356322258710861e-05), (4, 1.4258735973271541e-05), (5, 1.794036870705895e-05), (6, 2.531078462197911e-05), (7, 4.742397868540138e-05), (8, 3.303559424239211e-05), (9, 4.167836232227273e-05), (10, 6.688477878924459e-05), (11, 5.573123416979797e-05), (12, 5.794886237708852e-05), (13, 7.70202386775054e-05), (14, 0.0002477977250237018), (15, 0.0006075560813769698), (16, 0.0010947359260171652), (17, 0.0009854125091806054), (18, 0.0004710899374913424), (19, 0.0015745233977213502), (20, 0.006091188173741102), (21, 0.0021743045654147863), (22, 0.0009420046117156744), (23, 0.001149171032011509), (24, 0.0009408238111063838), (25, 0.0014993824297562242), (26, 0.0009683206444606185), (27, 0.001552892499603331)]
+
+ 67%|██████▋   | 20/30 [03:24<01:39, 10.00s/it]Time step 708, Spatial MSE: [(0, 8.066907321335748e-05), (1, 0.0012115250574424863), (2, 0.0012734092306345701), (3, 0.001526110921986401), (4, 0.0015449101338163018), (5, 0.00371740385890007), (6, 0.004723658785223961), (7, 0.008880718611180782), (8, 0.011640907265245914), (9, 0.018610171973705292), (10, 0.028740186244249344), (11, 0.02712765522301197), (12, 0.028667839244008064), (13, 0.03151404857635498), (14, 0.07386838644742966), (15, 0.08656676113605499), (16, 0.10783124715089798), (17, 0.0900256484746933), (18, 0.1126098707318306), (19, 0.1252666711807251), (20, 0.11670257896184921), (21, 0.1869991421699524), (22, 0.18719598650932312), (23, 0.15786263346672058), (24, 0.16804301738739014), (25, 0.20479924976825714), (26, 0.14102619886398315), (27, 0.02484547346830368)], Temporal MSE: [(0, 4.52658377980697e-06), (1, 2.5274843210354447e-05), (2, 0.00015387116582132876), (3, 0.0004432027053553611), (4, 0.0007534447940997779), (5, 0.0015071749221533537), (6, 0.0020326937083154917), (7, 0.003007198218256235), (8, 0.0024300147779285908), (9, 0.0035745480563491583), (10, 0.005329071544110775), (11, 0.006040426902472973), (12, 0.005145837087184191), (13, 0.0073419720865786076), (14, 0.01885189674794674), (15, 0.022978367283940315), (16, 0.03186854347586632), (17, 0.03493710979819298), (18, 0.03895531967282295), (19, 0.05164727196097374), (20, 0.09577538818120956), (21, 0.08745943009853363), (22, 0.057979606091976166), (23, 0.055528804659843445), (24, 0.03990514948964119), (25, 0.015470810234546661), (26, 0.021220436319708824), (27, 0.02101663313806057)]
+
+ 70%|███████   | 21/30 [03:35<01:31, 10.22s/it]Time step 676, Spatial MSE: [(0, 2.9896595151512884e-05), (1, 0.0009969824459403753), (2, 0.0015061667654663324), (3, 0.0007424899959005415), (4, 8.657330909045413e-05), (5, 0.00012597732711583376), (6, 0.00012893062375951558), (7, 0.00041918412898667157), (8, 0.00016472401330247521), (9, 0.0001881346688605845), (10, 0.0002330683928448707), (11, 0.000256202562013641), (12, 0.000351783906808123), (13, 0.000590587267652154), (14, 0.0010920936474576592), (15, 0.001072075217962265), (16, 0.013145607896149158), (17, 0.00246137916110456), (18, 0.0026753749698400497), (19, 0.002172189299017191), (20, 0.001634355285204947), (21, 0.004286330658942461), (22, 0.003137060906738043), (23, 0.00550727779045701), (24, 0.007686501834541559), (25, 0.026240123435854912), (26, 0.02193584106862545), (27, 0.013246958144009113)], Temporal MSE: [(0, 9.033064429786464e-07), (1, 3.0978292215877445e-06), (2, 1.6961863366304897e-05), (3, 1.5630334019078873e-05), (4, 1.751006129779853e-05), (5, 2.75464935839409e-05), (6, 3.552465568645857e-05), (7, 6.485264020739123e-05), (8, 3.93966693081893e-05), (9, 5.2671563025796786e-05), (10, 8.228071965277195e-05), (11, 6.953754927963018e-05), (12, 6.26922192168422e-05), (13, 8.433943003183231e-05), (14, 0.0003666543052531779), (15, 0.000548360520042479), (16, 0.000900949933566153), (17, 0.0007417317829094827), (18, 0.00036081497091799974), (19, 0.0013483643997460604), (20, 0.004721464589238167), (21, 0.0014658595900982618), (22, 0.0009723239345476031), (23, 0.0013655113289132714), (24, 0.0009770410833880305), (25, 0.0024275491014122963), (26, 0.0011476168874651194), (27, 0.002206176519393921)]
+
+ 73%|███████▎  | 22/30 [03:44<01:20, 10.03s/it]Time step 636, Spatial MSE: [(0, 0.0001012592765619047), (1, 0.0011233810801059008), (2, 0.0013418751768767834), (3, 0.0003551766858436167), (4, 0.00028763472801074386), (5, 0.0009265586268156767), (6, 0.0014314569998532534), (7, 0.0027032566722482443), (8, 0.0020451347809284925), (9, 0.002729194238781929), (10, 0.00946889165788889), (11, 0.0064149824902415276), (12, 0.006026090122759342), (13, 0.008737667463719845), (14, 0.02883017621934414), (15, 0.02853679656982422), (16, 0.03324120119214058), (17, 0.022802140563726425), (18, 0.02408268116414547), (19, 0.029329946264624596), (20, 0.02128482423722744), (21, 0.03241259604692459), (22, 0.020258018746972084), (23, 0.027703572064638138), (24, 0.05178973078727722), (25, 0.10580353438854218), (26, 0.04911864548921585), (27, 0.020704643800854683)], Temporal MSE: [(0, 2.514398602215806e-06), (1, 7.4302201937825885e-06), (2, 2.7681020583258942e-05), (3, 4.100249134353362e-05), (4, 9.914274414768443e-05), (5, 9.823639265960082e-05), (6, 0.0002720328338909894), (7, 0.0007437086314894259), (8, 0.0007245075539685786), (9, 0.0012729305308312178), (10, 0.0030846577137708664), (11, 0.001913692569360137), (12, 0.002556602703407407), (13, 0.0015928861685097218), (14, 0.008311309851706028), (15, 0.01253422163426876), (16, 0.014495937153697014), (17, 0.009806937538087368), (18, 0.007059262599796057), (19, 0.011274089105427265), (20, 0.012724619358778), (21, 0.012073368765413761), (22, 0.007998662069439888), (23, 0.00909618753939867), (24, 0.00788145512342453), (25, 0.003478123340755701), (26, 0.008652913384139538), (27, 0.004946708679199219)]
+
+ 77%|███████▋  | 23/30 [03:55<01:11, 10.27s/it]Time step 596, Spatial MSE: [(0, 3.968470264226198e-05), (1, 0.0013683821307495236), (2, 0.002395706484094262), (3, 0.0012075778795406222), (4, 0.00013191050675231963), (5, 0.00021007297618780285), (6, 0.0002043955319095403), (7, 0.00039925301098264754), (8, 0.0002540571440476924), (9, 0.0002603042230475694), (10, 0.000369260145816952), (11, 0.0003565693914424628), (12, 0.00044147748849354684), (13, 0.000913754221983254), (14, 0.0014275223948061466), (15, 0.001096193096600473), (16, 0.013979078270494938), (17, 0.0031213220208883286), (18, 0.003044568467885256), (19, 0.0051147653721272945), (20, 0.004603260196745396), (21, 0.015851063653826714), (22, 0.015330256894230843), (23, 0.01261121965944767), (24, 0.013718150556087494), (25, 0.046910304576158524), (26, 0.039134688675403595), (27, 0.02472669817507267)], Temporal MSE: [(0, 1.7365084659104468e-06), (1, 5.309050720825326e-06), (2, 2.720373413467314e-05), (3, 2.2335274479701184e-05), (4, 2.9530730898841284e-05), (5, 4.004687434644438e-05), (6, 5.8534675190458074e-05), (7, 9.020168363349512e-05), (8, 6.579371256520972e-05), (9, 8.918436651583761e-05), (10, 0.00011560528218979016), (11, 0.00012035587860736996), (12, 0.00010202496196143329), (13, 0.00013946268882136792), (14, 0.0003620522911660373), (15, 0.0006264469702728093), (16, 0.001225723302923143), (17, 0.0012812662171199918), (18, 0.0006722966791130602), (19, 0.0025451069232076406), (20, 0.010834054090082645), (21, 0.004782320000231266), (22, 0.003357378765940666), (23, 0.0020755035802721977), (24, 0.001674414612352848), (25, 0.004323609638959169), (26, 0.0018450513016432524), (27, 0.005350182764232159)]
+
+ 80%|████████  | 24/30 [04:06<01:03, 10.53s/it]Time step 548, Spatial MSE: [(0, 0.00011885487037943676), (1, 0.00264485040679574), (2, 0.0034380187280476093), (3, 0.002291375072672963), (4, 0.002247183583676815), (5, 0.005130769219249487), (6, 0.006364154163748026), (7, 0.009907989762723446), (8, 0.0141688147559762), (9, 0.01925479620695114), (10, 0.025308100506663322), (11, 0.027646910399198532), (12, 0.03349488973617554), (13, 0.035940881818532944), (14, 0.0724923387169838), (15, 0.10076099634170532), (16, 0.0951409786939621), (17, 0.10252704471349716), (18, 0.127563938498497), (19, 0.1320929080247879), (20, 0.1376836895942688), (21, 0.19588619470596313), (22, 0.213257297873497), (23, 0.20000648498535156), (24, 0.2453009933233261), (25, 0.39383912086486816), (26, 0.28145211935043335), (27, 0.056502681225538254)], Temporal MSE: [(0, 1.0921678949671332e-05), (1, 4.504345270106569e-05), (2, 0.00015561874897684902), (3, 0.0006814283551648259), (4, 0.0011643979232758284), (5, 0.0023673989344388247), (6, 0.0037155556492507458), (7, 0.004131566733121872), (8, 0.003413223894312978), (9, 0.005084264557808638), (10, 0.006145639810711145), (11, 0.007405290380120277), (12, 0.00736375292763114), (13, 0.00926098134368658), (14, 0.022720003500580788), (15, 0.028296954929828644), (16, 0.036146339029073715), (17, 0.040856797248125076), (18, 0.041344523429870605), (19, 0.05218416079878807), (20, 0.1227293312549591), (21, 0.1072244718670845), (22, 0.06482020020484924), (23, 0.06861072033643723), (24, 0.06638530641794205), (25, 0.03785836696624756), (26, 0.037882741540670395), (27, 0.0482899434864521)]
+
+ 83%|████████▎ | 25/30 [04:23<01:01, 12.27s/it]Time step 492, Spatial MSE: [(0, 6.719477823935449e-05), (1, 0.002606129739433527), (2, 0.003774621058255434), (3, 0.0023112352937459946), (4, 0.00021647608082275838), (5, 0.0004381979815661907), (6, 0.00042681669583544135), (7, 0.0010468818945810199), (8, 0.0005892735789529979), (9, 0.000517281994689256), (10, 0.0005851287278346717), (11, 0.0006236322806216776), (12, 0.0010860746260732412), (13, 0.0013648401945829391), (14, 0.0015386580489575863), (15, 0.0015757640358060598), (16, 0.012358859181404114), (17, 0.00280223460868001), (18, 0.012135777622461319), (19, 0.005141876637935638), (20, 0.006555291824042797), (21, 0.023472601547837257), (22, 0.010352447628974915), (23, 0.01832175999879837), (24, 0.02515728585422039), (25, 0.09867969900369644), (26, 0.08378402888774872), (27, 0.0399816557765007)], Temporal MSE: [(0, 2.7439161840447923e-06), (1, 8.986236025521066e-06), (2, 5.2235289331292734e-05), (3, 5.082136340206489e-05), (4, 6.112698611104861e-05), (5, 9.499513544142246e-05), (6, 0.00014092055789660662), (7, 0.00024112574465107173), (8, 0.0001346246717730537), (9, 0.00019524770323187113), (10, 0.0002059268153971061), (11, 0.0001993700279854238), (12, 0.0001952556340256706), (13, 0.0002740862255450338), (14, 0.000536977662704885), (15, 0.000729890598449856), (16, 0.001440393622033298), (17, 0.0013059579068794847), (18, 0.0028605188708752394), (19, 0.005033737979829311), (20, 0.02511846087872982), (21, 0.007766944821923971), (22, 0.0027112283278256655), (23, 0.00402564462274313), (24, 0.003435522085055709), (25, 0.009081951342523098), (26, 0.004032575059682131), (27, 0.010517613030970097)]
+
+ 87%|████████▋ | 26/30 [04:37<00:51, 12.86s/it]Time step 426, Spatial MSE: [(0, 0.00019218222587369382), (1, 0.004633840173482895), (2, 0.005250266753137112), (3, 0.001589371357113123), (4, 0.0013768227072432637), (5, 0.002624718239530921), (6, 0.00542021170258522), (7, 0.007392091676592827), (8, 0.008666261099278927), (9, 0.011466123163700104), (10, 0.021825620904564857), (11, 0.016192298382520676), (12, 0.018127046525478363), (13, 0.019999206066131592), (14, 0.04898326098918915), (15, 0.06261825561523438), (16, 0.06273335218429565), (17, 0.05615316331386566), (18, 0.0685167983174324), (19, 0.08024841547012329), (20, 0.07265522330999374), (21, 0.10919549316167831), (22, 0.11619922518730164), (23, 0.16520576179027557), (24, 0.2514041066169739), (25, 0.5251231789588928), (26, 0.36078113317489624), (27, 0.10999365150928497)], Temporal MSE: [(0, 1.2146220797148999e-05), (1, 3.467025817371905e-05), (2, 0.0001526343694422394), (3, 0.0002501713461242616), (4, 0.0007687953184358776), (5, 0.0011519857216626406), (6, 0.0032507923897355795), (7, 0.003954088781028986), (8, 0.0019149064319208264), (9, 0.0037045441567897797), (10, 0.004930628929287195), (11, 0.004390450660139322), (12, 0.0038467724807560444), (13, 0.005254314746707678), (14, 0.015154200606048107), (15, 0.018903804942965508), (16, 0.022870216518640518), (17, 0.022315166890621185), (18, 0.02404957450926304), (19, 0.027955569326877594), (20, 0.0470120869576931), (21, 0.06307220458984375), (22, 0.037922218441963196), (23, 0.03955329209566116), (24, 0.05648845434188843), (25, 0.03195483610033989), (26, 0.05859587341547012), (27, 0.07124489545822144)]
+
+ 90%|█████████ | 27/30 [04:56<00:44, 14.82s/it]Time step 350, Spatial MSE: [(0, 0.0001501742663094774), (1, 0.006445808336138725), (2, 0.007917571812868118), (3, 0.0020073936320841312), (4, 0.0007984485127963126), (5, 0.0019027275266125798), (6, 0.0028394656255841255), (7, 0.004817246459424496), (8, 0.004264132585376501), (9, 0.004538962617516518), (10, 0.00829166080802679), (11, 0.00692552188411355), (12, 0.008211083710193634), (13, 0.008341940119862556), (14, 0.021957088261842728), (15, 0.025837423279881477), (16, 0.03536513075232506), (17, 0.025791659951210022), (18, 0.02951028384268284), (19, 0.042907923460006714), (20, 0.04401884227991104), (21, 0.07044852524995804), (22, 0.091987743973732), (23, 0.11950770020484924), (24, 0.17186491191387177), (25, 0.33766093850135803), (26, 0.2874288856983185), (27, 0.1843264102935791)], Temporal MSE: [(0, 1.136604078055825e-05), (1, 3.512661351123825e-05), (2, 0.00015079643344506621), (3, 0.00018395540246274322), (4, 0.0004748424980789423), (5, 0.0006135681178420782), (6, 0.0020828568376600742), (7, 0.0021704030223190784), (8, 0.0009924264159053564), (9, 0.0018420577980577946), (10, 0.002517735119909048), (11, 0.002002085093408823), (12, 0.00193392566870898), (13, 0.002305585192516446), (14, 0.007562996353954077), (15, 0.008504641242325306), (16, 0.01042159739881754), (17, 0.01070286612957716), (18, 0.00960603542625904), (19, 0.01344850193709135), (20, 0.04122554138302803), (21, 0.04731396958231926), (22, 0.02032497711479664), (23, 0.02524171769618988), (24, 0.04725175350904465), (25, 0.02214553765952587), (26, 0.04641896113753319), (27, 0.06733769178390503)]
+
+ 93%|█████████▎| 28/30 [05:13<00:30, 15.35s/it]Time step 258, Spatial MSE: [(0, 0.0002863820700440556), (1, 0.010285746306180954), (2, 0.012416983023285866), (3, 0.003512411843985319), (4, 0.0014142540749162436), (5, 0.0037174869794398546), (6, 0.00539505947381258), (7, 0.01200532540678978), (8, 0.009213628247380257), (9, 0.0078458646312356), (10, 0.017174381762742996), (11, 0.011378487572073936), (12, 0.015963029116392136), (13, 0.013414986431598663), (14, 0.03694240748882294), (15, 0.04414638504385948), (16, 0.04940342903137207), (17, 0.04085756838321686), (18, 0.058572590351104736), (19, 0.06331734359264374), (20, 0.0652320384979248), (21, 0.12726902961730957), (22, 0.1380212903022766), (23, 0.2249653935432434), (24, 0.3526792526245117), (25, 0.7539297342300415), (26, 0.638352632522583), (27, 0.41826289892196655)], Temporal MSE: [(0, 2.7928925192099996e-05), (1, 7.385997741948813e-05), (2, 0.00028043417842127383), (3, 0.000340960017638281), (4, 0.0009233235032297671), (5, 0.001397537300363183), (6, 0.0039002075791358948), (7, 0.005119217559695244), (8, 0.001954998355358839), (9, 0.0035379105247557163), (10, 0.004821392707526684), (11, 0.003782976884394884), (12, 0.0035852952860295773), (13, 0.003977396059781313), (14, 0.013298945501446724), (15, 0.014712115749716759), (16, 0.016562016680836678), (17, 0.016726668924093246), (18, 0.015554741024971008), (19, 0.02314656414091587), (20, 0.05915459990501404), (21, 0.08943995088338852), (22, 0.03846216946840286), (23, 0.04678739234805107), (24, 0.1122235655784607), (25, 0.05537952482700348), (26, 0.12234440445899963), (27, 0.15951955318450928)]
+
+ 97%|█████████▋| 29/30 [05:27<00:14, 14.91s/it]Time step 143, Spatial MSE: [(0, 0.0006552163977175951), (1, 0.020702261477708817), (2, 0.023331917822360992), (3, 0.007435800042003393), (4, 0.003400778165087104), (5, 0.008854605257511139), (6, 0.012494396418333054), (7, 0.029105911031365395), (8, 0.022513635456562042), (9, 0.018909195438027382), (10, 0.03830797225236893), (11, 0.026804283261299133), (12, 0.0401010625064373), (13, 0.027308929711580276), (14, 0.08171574026346207), (15, 0.08609751611948013), (16, 0.08548565953969955), (17, 0.08118493109941483), (18, 0.10576746612787247), (19, 0.13924384117126465), (20, 0.17194558680057526), (21, 0.2683711349964142), (22, 0.3079575300216675), (23, 0.5263897180557251), (24, 1.1233235597610474), (25, 2.1600687503814697), (26, 1.7108246088027954), (27, 1.0287812948226929)], Temporal MSE: [(0, 7.66803277656436e-05), (1, 0.00017740638577379286), (2, 0.0006337363156490028), (3, 0.0007476062164641917), (4, 0.0025294749066233635), (5, 0.003905882826074958), (6, 0.010073477402329445), (7, 0.012724366039037704), (8, 0.004583064466714859), (9, 0.006628519855439663), (10, 0.011004096828401089), (11, 0.008285376243293285), (12, 0.007501672487705946), (13, 0.008400787599384785), (14, 0.025955313816666603), (15, 0.02826032042503357), (16, 0.032843876630067825), (17, 0.03404368832707405), (18, 0.032333023846149445), (19, 0.04862288385629654), (20, 0.15539348125457764), (21, 0.23457729816436768), (22, 0.07975535094738007), (23, 0.12368989735841751), (24, 0.3429652452468872), (25, 0.24083110690116882), (26, 0.32483625411987305), (27, 0.35420092940330505)]
+
+100%|██████████| 30/30 [05:44<00:00, 15.70s/it]
+100%|██████████| 30/30 [05:44<00:00, 11.49s/it]
+[2024-10-01 15:18:58] Prompt: time lapse of the rising sun over a tree in an open rural landscape, with clouds in the blue sky beautifully playing with the rays of light aesthetic score: 6.5.
+Saved to ./samples/opensora/sample_pab/Time Lapse of the rising sun over a tree.mp4
+Saved video to ./samples/opensora/sample_pab/Time Lapse of the rising sun over a tree.mp4
+Saved spatial MLP MSE outputs to ./samples/opensora/sample_pab/Time Lapse of the rising sun over a tree_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora/sample_pab/Time Lapse of the rising sun over a tree_temporal_mlp_mse.pt
+[2024-10-01 15:19:01] Inference finished.
+[2024-10-01 15:19:01] Saved samples to ./samples/opensora/sample_pab
diff --git a/sample_opensora_plan_mse.txt b/sample_opensora_plan_mse.txt
new file mode 100644
index 00000000..4ae9093e
--- /dev/null
+++ b/sample_opensora_plan_mse.txt
@@ -0,0 +1,11218 @@
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/diffusers/models/transformers/transformer_2d.py:34: FutureWarning: `Transformer2DModelOutput` is deprecated and will be removed in version 1.0.0. Importing `Transformer2DModelOutput` from `diffusers.models.transformer_2d` is deprecated and this will be removed in a future version. Please use `from diffusers.models.modeling_outputs import Transformer2DModelOutput`, instead.
+  deprecate("Transformer2DModelOutput", "1.0.0", deprecation_message)
+[10/01/24 15:16:11] INFO     colossalai - colossalai - INFO:
+                             /home/jin509/anaconda3/envs/opendit/lib/python3.10/
+                             site-packages/colossalai/initialize.py:60 launch
+                    INFO     colossalai - colossalai - INFO: Distributed
+                             environment is initialized, world size: 1
+
+Init SkipManager:
+                steps=150
+                cross_broadcast=True, cross_threshold=[100, 850], cross_gap=6
+                spatial_broadcast=True, spatial_threshold=[100, 850], spatial_gap=2, spatial_layer_range=[0, 28]
+                temporal_broadcast=True, temporal_threshold=[100, 850], temporal_gap=4
+                diffusion_skip=False, diffusion_timestep_respacing=None
+
+Loading model from LanguageBind/Open-Sora-Plan-v1.1.0
+The config attributes {'in_channels': 3, 'out_channels': 3} were passed to CausalVAEModel, but are not expected and will be ignored. Please verify your config.json configuration file.
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torchvision/models/_utils.py:208: UserWarning: The parameter 'pretrained' is deprecated since 0.13 and may be removed in the future, please use 'weights' instead.
+  warnings.warn(
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torchvision/models/_utils.py:223: UserWarning: Arguments other than a weight enum or `None` for 'weights' are deprecated since 0.13 and may be removed in the future. The current behavior is equivalent to passing `weights=VGG16_Weights.IMAGENET1K_V1`. You can also use `weights=VGG16_Weights.DEFAULT` to get the most up-to-date weights.
+  warnings.warn(msg)
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.0.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.0.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.2.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.2.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.5.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.5.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.7.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.7.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.10.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.10.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.12.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.12.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.14.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.14.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.17.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.17.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.19.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.19.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.21.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.21.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.24.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.24.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.26.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.26.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.28.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for features.28.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.0.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.0.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.3.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.3.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.6.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for classifier.6.bias: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for lin0.model.1.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for lin1.model.1.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for lin2.model.1.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for lin3.model.1.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/module.py:2047: UserWarning: for lin4.model.1.weight: copying from a non-meta parameter in the checkpoint to a meta parameter in the current model, which is a no-op. (Did you mean to pass `assign=True` to assign items in the state dictionary to their corresponding key in the module instead of copying them in place?)
+  warnings.warn(f'for {key}: copying from a non-meta parameter in the checkpoint to a meta '
+loaded pretrained LPIPS loss from taming/modules/autoencoder/lpips/vgg.pth
+You are using the default legacy behaviour of the <class 'transformers.models.t5.tokenization_t5.T5Tokenizer'>. This is expected, and simply means that the `legacy` (previous) behavior will be used so nothing changes for you. If you want to use the new behaviour, set `legacy=False`. This should only be set if you understand what it means, and thoroughly read the reason why this was added as explained in https://github.com/huggingface/transformers/pull/24565
+
+Downloading shards:   0%|          | 0/2 [00:00<?, ?it/s]
+Downloading shards: 100%|██████████| 2/2 [00:34<00:00, 17.04s/it]
+Downloading shards: 100%|██████████| 2/2 [00:34<00:00, 17.04s/it]
+
+Loading checkpoint shards:   0%|          | 0/2 [00:00<?, ?it/s]
+Loading checkpoint shards:  50%|█████     | 1/2 [00:10<00:10, 10.96s/it]
+Loading checkpoint shards: 100%|██████████| 2/2 [00:21<00:00, 10.62s/it]
+Loading checkpoint shards: 100%|██████████| 2/2 [00:21<00:00, 10.67s/it]
+
+  0%|          | 0/150 [00:00<?, ?it/s]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/backends/cuda/__init__.py:342: FutureWarning: torch.backends.cuda.sdp_kernel() is deprecated. In the future, this context manager will be removed. Please see, torch.nn.attention.sdpa_kernel() for the new context manager, with updated signature.
+  warnings.warn(
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.178468664875254e-06), (1, 6.389327154465718e-06), (2, 0.0001644557632971555), (3, 2.0604567907867022e-05), (4, 8.964254448073916e-06), (5, 7.041211119940272e-06), (6, 8.684407475811895e-06), (7, 6.223340278666001e-06), (8, 6.255907919694437e-06), (9, 6.87408828525804e-06), (10, 1.0234925866825506e-05), (11, 1.1839481885544956e-05), (12, 1.3265676898299716e-05), (13, 1.767053072398994e-05), (14, 6.256129563553259e-05), (15, 0.00013637695519719273), (16, 0.00039488993934355676), (17, 5.924455035710707e-05), (18, 5.9960111684631556e-05), (19, 7.20885582268238e-05), (20, 8.953617361839861e-05), (21, 6.365979788824916e-05), (22, 8.318154868902639e-05), (23, 7.847957749618217e-05), (24, 0.0001491493749199435), (25, 0.0003716802457347512), (26, 0.0004000751068815589), (27, 0.003655975218862295)], Temporal MSE: [(0, 1.6439080354757607e-06), (1, 1.195070581161417e-06), (2, 1.6939859506237553e-06), (3, 1.5300207678592415e-06), (4, 1.1545218967512483e-06), (5, 1.522264597042522e-06), (6, 7.991680490704312e-07), (7, 1.1063775673392229e-06), (8, 7.175190717134683e-07), (9, 1.1046323606933584e-06), (10, 3.049800852750195e-06), (11, 2.2214790078578517e-06), (12, 1.6446123254354461e-06), (13, 4.65000402982696e-06), (14, 4.420430286700139e-06), (15, 6.920137820998207e-06), (16, 2.627832600410329e-06), (17, 5.804248303320492e-06), (18, 3.4169320315413643e-06), (19, 3.233144525438547e-06), (20, 4.48621676696348e-06), (21, 3.3225928746105637e-06), (22, 9.288488399761263e-06), (23, 6.009115622873651e-06), (24, 1.537053867650684e-05), (25, 1.5760166206746362e-05), (26, 2.9593002182082273e-06), (27, 1.176244859379949e-05)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1561769319712312e-08), (1, 1.411965229181078e-07), (2, 1.4375549426404177e-06), (3, 2.6285258059033367e-07), (4, 1.0856301457806694e-07), (5, 8.5726661325225e-08), (6, 2.1119329574048606e-07), (7, 1.2377364555504755e-07), (8, 1.0805997874285822e-07), (9, 1.4142402449124347e-07), (10, 1.4917483781573537e-07), (11, 1.5248977547344111e-07), (12, 1.7853396627742768e-07), (13, 1.6174199402030354e-07), (14, 2.738789532941155e-07), (15, 5.267668825581495e-07), (16, 7.276715905391029e-07), (17, 4.162398852258775e-07), (18, 8.951748782237701e-07), (19, 5.223566290624149e-07), (20, 6.551370574925386e-07), (21, 9.513670988781087e-07), (22, 1.237866513292829e-06), (23, 1.0108429933097796e-06), (24, 2.867576995413401e-06), (25, 4.7634189286327455e-06), (26, 5.553245500777848e-06), (27, 2.7636750019155443e-05)], Temporal MSE: [(0, 7.015649838848503e-09), (1, 9.349512275491634e-09), (2, 9.368168463197435e-09), (3, 1.142037753254499e-08), (4, 1.4839554474121996e-08), (5, 2.596439152569019e-08), (6, 1.3327509762461887e-08), (7, 1.7445090705336952e-08), (8, 1.6921401169156525e-08), (9, 2.292632039768705e-08), (10, 3.084866406766196e-08), (11, 3.0611456480755805e-08), (12, 2.7661437584924897e-08), (13, 4.5057284125959995e-08), (14, 4.279261744954965e-08), (15, 7.61646745672806e-08), (16, 2.8382274308569322e-08), (17, 5.399366997949073e-08), (18, 5.576986339406176e-08), (19, 4.894328498039613e-08), (20, 9.201288975191346e-08), (21, 7.079619024352723e-08), (22, 1.1960673873545602e-07), (23, 1.2879229416284943e-07), (24, 2.7133262392453616e-07), (25, 3.5233202311246714e-07), (26, 1.3238664564596547e-07), (27, 1.286577457904059e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 9.241383054359176e-07), (1, 5.684374627890065e-06), (2, 0.00015654211165383458), (3, 1.5351415640907362e-05), (4, 6.886443316034274e-06), (5, 5.533943294722121e-06), (6, 5.552629318117397e-06), (7, 6.5100580286525656e-06), (8, 6.425399078580085e-06), (9, 7.940155228425283e-06), (10, 1.241086283698678e-05), (11, 1.4132074284134433e-05), (12, 1.8210925190942362e-05), (13, 1.8847809769795276e-05), (14, 7.481357170036063e-05), (15, 0.00014458235818892717), (16, 0.00029452049056999385), (17, 4.38388560723979e-05), (18, 7.190948235802352e-05), (19, 6.534953718073666e-05), (20, 9.364950528834015e-05), (21, 6.835781096015126e-05), (22, 5.8625537349144e-05), (23, 6.821284478064626e-05), (24, 0.00018871124484576285), (25, 0.0004198308160994202), (26, 0.00020624556054826826), (27, 0.0026255124248564243)], Temporal MSE: [(0, 1.3170750889912597e-06), (1, 1.0123703759745695e-06), (2, 1.1587039807636756e-06), (3, 9.517477792542195e-07), (4, 9.277150070374773e-07), (5, 1.3259044635560713e-06), (6, 7.564363500023319e-07), (7, 1.4189662351782317e-06), (8, 7.044637300168688e-07), (9, 1.4899953839631053e-06), (10, 4.631838237401098e-06), (11, 3.3660767257970292e-06), (12, 2.2784724933444522e-06), (13, 7.516544883401366e-06), (14, 6.21852814219892e-06), (15, 9.905955266731326e-06), (16, 3.141487923130626e-06), (17, 7.934542736620642e-06), (18, 3.5830844353768043e-06), (19, 3.955411557399202e-06), (20, 5.245668489806121e-06), (21, 3.475367520877626e-06), (22, 8.556188731745351e-06), (23, 5.838062406837707e-06), (24, 1.6290654457407072e-05), (25, 1.3304329513630364e-05), (26, 2.8052145353285596e-06), (27, 7.6412961789174e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 1.0888207668813266e-08), (1, 1.2251662440121436e-07), (2, 1.2252481838004314e-06), (3, 2.2988035652815597e-07), (4, 1.0209588907628131e-07), (5, 8.077307001030931e-08), (6, 1.9268851758624805e-07), (7, 1.1730325866210478e-07), (8, 1.0222758817235444e-07), (9, 1.3328205739071564e-07), (10, 1.3970689849429618e-07), (11, 1.4126899827715533e-07), (12, 1.6597067542534205e-07), (13, 1.6028305083182204e-07), (14, 3.6095593713980634e-07), (15, 6.778459464840125e-07), (16, 3.107059683316038e-06), (17, 4.265689597104938e-07), (18, 9.168185783892113e-07), (19, 6.537800913974934e-07), (20, 7.287204653039225e-07), (21, 9.735597359394887e-07), (22, 1.2315881576796528e-06), (23, 9.677478374214843e-07), (24, 2.7474493435875047e-06), (25, 4.603492925525643e-06), (26, 5.472501470649149e-06), (27, 2.6882491511059925e-05)], Temporal MSE: [(0, 6.633151361512546e-09), (1, 8.486460423284825e-09), (2, 8.65918181602865e-09), (3, 1.056218490447236e-08), (4, 1.3829242639928907e-08), (5, 2.4349469995854633e-08), (6, 1.2551398143045844e-08), (7, 1.6450654172217583e-08), (8, 1.586342257553497e-08), (9, 2.1222007262622355e-08), (10, 2.8675582797177412e-08), (11, 2.835233203768439e-08), (12, 2.5683414506261215e-08), (13, 4.487534965846862e-08), (14, 3.975847562287527e-08), (15, 7.462234918875765e-08), (16, 2.896652695483226e-08), (17, 5.655630985756943e-08), (18, 5.52919878771263e-08), (19, 4.868687497605606e-08), (20, 8.922766170371688e-08), (21, 7.007516700241467e-08), (22, 1.2316810682477808e-07), (23, 1.3010662769374903e-07), (24, 2.6247340656482265e-07), (25, 3.429057358061982e-07), (26, 1.2829845275064145e-07), (27, 1.265933349259285e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.3443325315165566e-06), (1, 3.715558250405593e-06), (2, 9.131305705523118e-05), (3, 1.6561201846343465e-05), (4, 6.789760846004356e-06), (5, 4.797926521860063e-06), (6, 6.2314566093846224e-06), (7, 4.446857474249555e-06), (8, 5.741481345467037e-06), (9, 8.726242413104046e-06), (10, 1.1422715942899231e-05), (11, 1.1676555914164055e-05), (12, 1.406779483659193e-05), (13, 1.5372481357189827e-05), (14, 3.851760266115889e-05), (15, 6.354111974360421e-05), (16, 0.00025493555585853755), (17, 2.4305008992087096e-05), (18, 3.2989490136969835e-05), (19, 3.8149217289173976e-05), (20, 3.352834391989745e-05), (21, 2.199285154347308e-05), (22, 3.784061846090481e-05), (23, 4.191971311229281e-05), (24, 9.126881195697933e-05), (25, 0.00017198397836182266), (26, 0.00019928129040636122), (27, 0.004030435346066952)], Temporal MSE: [(0, 1.6750921076891245e-06), (1, 9.946490990841994e-07), (2, 1.7142357364718919e-06), (3, 1.4186499583956902e-06), (4, 1.11239091893367e-06), (5, 1.436839738744311e-06), (6, 8.691347375133773e-07), (7, 6.989833423176606e-07), (8, 1.2429808293745737e-06), (9, 1.639747324588825e-06), (10, 2.4394221327384003e-06), (11, 1.8434272988088196e-06), (12, 2.204266820626799e-06), (13, 3.970040324929869e-06), (14, 5.215943019720726e-06), (15, 3.787072273553349e-06), (16, 2.5036840725078946e-06), (17, 2.1685050342057366e-06), (18, 2.8223155368323205e-06), (19, 2.90378056888585e-06), (20, 3.7498257370316423e-06), (21, 2.6834061372937867e-06), (22, 3.1753963867231505e-06), (23, 4.655336852010805e-06), (24, 1.4312926396087278e-05), (25, 1.2060611879860517e-05), (26, 3.758416596610914e-06), (27, 3.6850547076028306e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.1684845091508578e-08), (1, 1.4240642087770539e-07), (2, 1.4355363191498327e-06), (3, 2.6197358238277957e-07), (4, 1.1220478057794026e-07), (5, 8.726647848789071e-08), (6, 2.118030977271701e-07), (7, 1.256801311910749e-07), (8, 1.0923938731366434e-07), (9, 1.4432571049383114e-07), (10, 1.5092643934622174e-07), (11, 1.5517959184307983e-07), (12, 1.8189474815244466e-07), (13, 1.6509999056779634e-07), (14, 2.8419876230145746e-07), (15, 5.42772397693625e-07), (16, 7.198862590485078e-07), (17, 4.208688153539697e-07), (18, 8.657075909468404e-07), (19, 5.407369485510571e-07), (20, 6.853182412669412e-07), (21, 9.703119303594576e-07), (22, 1.3321256346898736e-06), (23, 1.0323832384528941e-06), (24, 2.9354507660173113e-06), (25, 4.790704679180635e-06), (26, 5.7304710026073735e-06), (27, 2.817200220306404e-05)], Temporal MSE: [(0, 6.973806865318011e-09), (1, 9.362699060488922e-09), (2, 9.448324789218532e-09), (3, 1.1368253005628048e-08), (4, 1.5010481746458026e-08), (5, 2.6094951977029268e-08), (6, 1.3549534827461684e-08), (7, 1.7571615273936914e-08), (8, 1.7209936586937147e-08), (9, 2.3336468757406692e-08), (10, 3.125101244449979e-08), (11, 3.0856636357157186e-08), (12, 2.820382682955369e-08), (13, 4.679387188843975e-08), (14, 4.4655688924422066e-08), (15, 7.937254764556201e-08), (16, 2.9997792694302916e-08), (17, 5.583713047485617e-08), (18, 5.654439760860441e-08), (19, 5.090223709203201e-08), (20, 9.543263956857118e-08), (21, 7.135433577332151e-08), (22, 1.2425749673639075e-07), (23, 1.3151993982774002e-07), (24, 2.7689011972142907e-07), (25, 3.5750673532675137e-07), (26, 1.3600289605619764e-07), (27, 1.3370264184686675e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 3.463894699962111e-07), (1, 9.25728386391711e-07), (2, 1.9512515791575424e-05), (3, 4.43319913756568e-06), (4, 1.8939274468721123e-06), (5, 1.6353832279492053e-06), (6, 2.2595172595174517e-06), (7, 2.146771976185846e-06), (8, 3.14758176500618e-06), (9, 5.7723063946468756e-06), (10, 8.3415789049468e-06), (11, 8.791610525804572e-06), (12, 1.044808050210122e-05), (13, 1.153666107711615e-05), (14, 3.388779077795334e-05), (15, 7.304576865863055e-05), (16, 8.772976434556767e-05), (17, 1.9394557966734283e-05), (18, 2.789258178381715e-05), (19, 2.436478462186642e-05), (20, 2.207276520493906e-05), (21, 1.5757501387270167e-05), (22, 2.309504088771064e-05), (23, 2.299297193530947e-05), (24, 4.977095522917807e-05), (25, 9.82531564659439e-05), (26, 4.373333649709821e-05), (27, 0.0006735087954439223)], Temporal MSE: [(0, 4.1701571262819925e-07), (1, 2.2030572210951505e-07), (2, 2.8681486696768843e-07), (3, 2.833313885730604e-07), (4, 3.574978961751185e-07), (5, 4.816354817194224e-07), (6, 4.029750471090665e-07), (7, 5.520176387108222e-07), (8, 4.010690872746636e-07), (9, 9.702434908831492e-07), (10, 2.3084803615347482e-06), (11, 1.570118683957844e-06), (12, 1.4088989246374695e-06), (13, 3.855446720990585e-06), (14, 4.243892362865154e-06), (15, 4.699637884186814e-06), (16, 2.2248518689593766e-06), (17, 3.4868819511757465e-06), (18, 2.8336942250462016e-06), (19, 2.541334197303513e-06), (20, 4.512440227699699e-06), (21, 2.0575928374455543e-06), (22, 2.2273809463513317e-06), (23, 2.5076308247662382e-06), (24, 7.395543434540741e-06), (25, 5.231287559581688e-06), (26, 1.508740638200834e-06), (27, 2.2163269477459835e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 1.0850263798545257e-08), (1, 1.1947832945224945e-07), (2, 1.2097044645997812e-06), (3, 2.2719548553595814e-07), (4, 1.011047601195969e-07), (5, 8.018150055022488e-08), (6, 1.9050592925395904e-07), (7, 1.1671045996308749e-07), (8, 1.018311408529371e-07), (9, 1.3349237804050063e-07), (10, 1.3992820413477602e-07), (11, 1.4181033236582152e-07), (12, 1.6648820633236028e-07), (13, 1.4979308105012024e-07), (14, 2.6926312557407073e-07), (15, 5.308185109242913e-07), (16, 1.1407935289753368e-06), (17, 4.1356733504471777e-07), (18, 8.340020372088475e-07), (19, 5.365018864722515e-07), (20, 6.596109187739785e-07), (21, 9.179260587188764e-07), (22, 1.2038202612529858e-06), (23, 9.609325388737489e-07), (24, 2.7382016014598776e-06), (25, 4.559086391964229e-06), (26, 5.460275133373216e-06), (27, 2.7799716917797923e-05)], Temporal MSE: [(0, 6.5771446067230954e-09), (1, 8.387151417821315e-09), (2, 8.593883826790716e-09), (3, 1.049374009909343e-08), (4, 1.3787230912498671e-08), (5, 2.4116747709967967e-08), (6, 1.249525372060134e-08), (7, 1.637232394102739e-08), (8, 1.5963701471832792e-08), (9, 2.13986925956533e-08), (10, 2.8894183046190847e-08), (11, 2.839682089472717e-08), (12, 2.5848917672988136e-08), (13, 4.232975570062081e-08), (14, 3.961734407198492e-08), (15, 7.32692271299129e-08), (16, 2.7430859361743387e-08), (17, 5.326053198473346e-08), (18, 5.338348074701571e-08), (19, 4.707841938511592e-08), (20, 8.799272421811111e-08), (21, 6.759430704050828e-08), (22, 1.1981296665908303e-07), (23, 1.253019945579581e-07), (24, 2.60531408002862e-07), (25, 3.4392292036500294e-07), (26, 1.2952895644957607e-07), (27, 1.326061607187512e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 3.2170646591112018e-06), (1, 3.0168481316650286e-05), (2, 0.0007356178248301148), (3, 8.058496314333752e-05), (4, 3.1018607842270285e-05), (5, 2.1677544282283634e-05), (6, 2.7991109163849615e-05), (7, 1.7448532162234187e-05), (8, 1.6314648746629246e-05), (9, 1.7169448256026953e-05), (10, 2.374678115302231e-05), (11, 2.502188908692915e-05), (12, 3.187081529176794e-05), (13, 3.9037771784933284e-05), (14, 0.0001670210185693577), (15, 0.0003762846172321588), (16, 0.001371056423522532), (17, 0.00016282786964438856), (18, 0.000250427023274824), (19, 0.0002638183068484068), (20, 0.00033877964597195387), (21, 0.0001690983772277832), (22, 0.00023031249293126166), (23, 0.00022997923952061683), (24, 0.0004500007489696145), (25, 0.0010106554254889488), (26, 0.0011646209750324488), (27, 0.015183796174824238)], Temporal MSE: [(0, 4.37868902736227e-06), (1, 4.060142146045109e-06), (2, 6.083663265599171e-06), (3, 5.519004844245501e-06), (4, 4.670070666179527e-06), (5, 4.797317615157226e-06), (6, 2.22648486669641e-06), (7, 3.529031573634711e-06), (8, 2.1623961856676033e-06), (9, 2.8166389256512048e-06), (10, 6.953393040021183e-06), (11, 5.827933819091413e-06), (12, 3.813564035226591e-06), (13, 1.279056232306175e-05), (14, 1.0282570656272583e-05), (15, 1.9410303139011376e-05), (16, 6.674838459730381e-06), (17, 2.3663935280637816e-05), (18, 1.0317876331100706e-05), (19, 1.4024393749423325e-05), (20, 2.2751504729967564e-05), (21, 1.0326784831704572e-05), (22, 2.0850631699431688e-05), (23, 1.7327787645626813e-05), (24, 4.947346315020695e-05), (25, 4.9099791795015335e-05), (26, 9.768335075932555e-06), (27, 3.8814847357571125e-05)]
+
+  1%|          | 1/150 [02:06<5:14:06, 126.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 1.1773704677864316e-08), (1, 1.516492744713105e-07), (2, 1.501817337157263e-06), (3, 2.740053446359525e-07), (4, 1.114660079792884e-07), (5, 8.761015379832315e-08), (6, 2.224039405973599e-07), (7, 1.2683868533258646e-07), (8, 1.1056797433184329e-07), (9, 1.478557578593609e-07), (10, 1.5457328572665574e-07), (11, 1.5913045103843615e-07), (12, 1.8740678342510364e-07), (13, 1.688471087391008e-07), (14, 3.0516625315613055e-07), (15, 5.971867835796729e-07), (16, 1.166205379377061e-06), (17, 4.5493629841075744e-07), (18, 9.164119774141e-07), (19, 5.833863951920648e-07), (20, 7.578784106954117e-07), (21, 1.0052518746306305e-06), (22, 1.3467503094943822e-06), (23, 1.044040573106031e-06), (24, 2.9785385322611546e-06), (25, 4.9157415560330264e-06), (26, 5.7193406064470764e-06), (27, 2.8944510631845333e-05)], Temporal MSE: [(0, 7.040917182621342e-09), (1, 9.446045723393581e-09), (2, 9.576099024855012e-09), (3, 1.1514643460941443e-08), (4, 1.509778613240087e-08), (5, 2.650415531491035e-08), (6, 1.393201110033715e-08), (7, 1.8106106836057734e-08), (8, 1.760811230155923e-08), (9, 2.431287349224931e-08), (10, 3.248840485525761e-08), (11, 3.203988185873641e-08), (12, 2.896078399317048e-08), (13, 4.8448555389768444e-08), (14, 4.64425227164611e-08), (15, 8.10359281899764e-08), (16, 3.161696326969832e-08), (17, 5.9316899836403536e-08), (18, 6.159716292586381e-08), (19, 5.252016777035351e-08), (20, 9.747579809982199e-08), (21, 7.601646956345576e-08), (22, 1.2862085441156523e-07), (23, 1.3467956705426332e-07), (24, 2.8264969387237215e-07), (25, 3.7377677131189557e-07), (26, 1.398041433731123e-07), (27, 1.37417686119079e-07)]
+
+  1%|▏         | 2/150 [02:17<2:25:01, 58.80s/it] Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 3.764787834370509e-06), (1, 2.9268516300362535e-05), (2, 0.0008786935941316187), (3, 9.629225678509101e-05), (4, 3.9522881706943735e-05), (5, 2.6133715437026694e-05), (6, 3.9099366404116154e-05), (7, 2.1033807570347562e-05), (8, 1.8834803995559923e-05), (9, 1.9964942111982964e-05), (10, 2.7666441383189522e-05), (11, 2.7909445634577423e-05), (12, 3.8379635952878743e-05), (13, 4.966584310750477e-05), (14, 0.00021389346511568874), (15, 0.00042097148252651095), (16, 0.0015912757953628898), (17, 0.0002264456416014582), (18, 0.0002780658833216876), (19, 0.00028589711291715503), (20, 0.00047663439181633294), (21, 0.0003028129576705396), (22, 0.00027012702776119113), (23, 0.0002726057427935302), (24, 0.0005976737011224031), (25, 0.0015689019346609712), (26, 0.0014270400861278176), (27, 0.02059837058186531)], Temporal MSE: [(0, 5.165885795577196e-06), (1, 4.776960395247443e-06), (2, 7.419698704325128e-06), (3, 6.138991011539474e-06), (4, 5.562874775932869e-06), (5, 6.260612281039357e-06), (6, 2.5865520001389086e-06), (7, 4.05170385420206e-06), (8, 2.780320301098982e-06), (9, 3.129773858745466e-06), (10, 7.432918209815398e-06), (11, 6.553597359015839e-06), (12, 4.408065706229536e-06), (13, 1.421130400558468e-05), (14, 1.1332727808621712e-05), (15, 2.1153518900973722e-05), (16, 8.487470950058196e-06), (17, 2.6539762984612025e-05), (18, 1.2774278729921207e-05), (19, 1.5431382053066045e-05), (20, 2.7827860321849585e-05), (21, 1.1609890862018801e-05), (22, 2.9816666938131675e-05), (23, 1.9661545593407936e-05), (24, 6.09197340963874e-05), (25, 6.371697236318141e-05), (26, 1.2663410416280385e-05), (27, 4.667058965424076e-05)]
+
+  2%|▏         | 3/150 [02:28<1:30:05, 36.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 1.1191963800172289e-08), (1, 1.250482881687276e-07), (2, 1.2134917142248014e-06), (3, 2.2706470304001414e-07), (4, 1.0367057967641813e-07), (5, 7.994810857780976e-08), (6, 1.9712733489996026e-07), (7, 1.1737012783896716e-07), (8, 1.0202545297488541e-07), (9, 1.3371113993798645e-07), (10, 1.3990091929372284e-07), (11, 1.4247419244384218e-07), (12, 1.690498550033226e-07), (13, 1.7461347567859775e-07), (14, 5.574295300903032e-07), (15, 1.5514059441557038e-06), (16, 2.2136251573101617e-05), (17, 5.600363692792598e-07), (18, 1.2687371508945944e-06), (19, 1.5628015717084054e-06), (20, 1.1758328355426784e-06), (21, 1.4492362652163138e-06), (22, 1.477868977417529e-06), (23, 1.0494717344045057e-06), (24, 2.8969730010430794e-06), (25, 4.621813786798157e-06), (26, 5.395865173341008e-06), (27, 2.681664227566216e-05)], Temporal MSE: [(0, 6.678384067981824e-09), (1, 8.450514066282722e-09), (2, 8.444405175112024e-09), (3, 1.0402684047505772e-08), (4, 1.3614660510086196e-08), (5, 2.4110658358722503e-08), (6, 1.2677130456495433e-08), (7, 1.657073767091788e-08), (8, 1.614289857343465e-08), (9, 2.1716189735343505e-08), (10, 2.950526223344241e-08), (11, 2.9335270212982323e-08), (12, 2.6697694721633525e-08), (13, 4.874000225640884e-08), (14, 4.4596127679596975e-08), (15, 8.532767026281363e-08), (16, 5.7453679147556613e-08), (17, 1.011357966262949e-07), (18, 6.928441109721462e-08), (19, 6.779060868211673e-08), (20, 1.0653010207306579e-07), (21, 8.82275443814251e-08), (22, 1.7209147529229085e-07), (23, 1.6350178100310586e-07), (24, 2.8173525379315834e-07), (25, 3.6369041822581494e-07), (26, 1.3357905004340864e-07), (27, 1.2910135183119564e-07)]
+
+  3%|▎         | 4/150 [02:41<1:06:29, 27.32s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 870, Spatial MSE: [(0, 1.2093134500901215e-05), (1, 0.00010115240002050996), (2, 0.002576196799054742), (3, 0.00027811521431431174), (4, 0.000112246758362744), (5, 7.565798296127468e-05), (6, 9.040017175721005e-05), (7, 7.959781214594841e-05), (8, 5.863172191311605e-05), (9, 5.648739170283079e-05), (10, 7.216048106784001e-05), (11, 7.80484260758385e-05), (12, 0.00010311538790119812), (13, 0.00017928126908373088), (14, 0.000774506654124707), (15, 0.0013994327746331692), (16, 0.007618054747581482), (17, 0.0006298836087808013), (18, 0.0008272962877526879), (19, 0.0011140605201944709), (20, 0.0013726844917982817), (21, 0.0006497541326098144), (22, 0.0008337858016602695), (23, 0.000813318882137537), (24, 0.001695173210464418), (25, 0.004818265791982412), (26, 0.004403381142765284), (27, 0.05793527886271477)], Temporal MSE: [(0, 1.6884894648683257e-05), (1, 1.5358633390860632e-05), (2, 2.2210455426829867e-05), (3, 1.843296195147559e-05), (4, 1.6139601939357817e-05), (5, 1.7373125956510194e-05), (6, 8.569800229452085e-06), (7, 1.0955182006000541e-05), (8, 8.314645128848497e-06), (9, 1.0334744729334489e-05), (10, 2.3254180632648058e-05), (11, 1.9785909898928367e-05), (12, 1.4713232303620316e-05), (13, 5.871069151908159e-05), (14, 3.9079764974303544e-05), (15, 7.044465746730566e-05), (16, 2.854189369827509e-05), (17, 0.00010132182069355622), (18, 4.251227801432833e-05), (19, 6.188495899550617e-05), (20, 9.845159365795553e-05), (21, 5.367548510548659e-05), (22, 7.177950465120375e-05), (23, 6.463528552558273e-05), (24, 0.00019666156731545925), (25, 0.00024565288913436234), (26, 3.7434303521877155e-05), (27, 0.00014309995458461344)]
+
+  3%|▎         | 5/150 [02:55<54:54, 22.72s/it]  Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 864, Spatial MSE: [(0, 1.5433090084115975e-05), (1, 0.00013765535550191998), (2, 0.00359719293192029), (3, 0.000380476179998368), (4, 0.00016016852168831974), (5, 0.00010487532563274726), (6, 0.00016136055637616664), (7, 8.22167785372585e-05), (8, 7.316909614019096e-05), (9, 8.509504550602287e-05), (10, 0.0001212217248394154), (11, 0.0001108417782234028), (12, 0.00016177791985683143), (13, 0.0002080925041809678), (14, 0.000762014533393085), (15, 0.0015398545656353235), (16, 0.005568619351834059), (17, 0.0008340515778400004), (18, 0.001103947521187365), (19, 0.0010883321519941092), (20, 0.0016722467262297869), (21, 0.0008795331232249737), (22, 0.00102492596488446), (23, 0.0010057603940367699), (24, 0.002107220469042659), (25, 0.005860355217009783), (26, 0.005246963817626238), (27, 0.08941394835710526)], Temporal MSE: [(0, 2.1044526874902658e-05), (1, 1.9615097698988393e-05), (2, 2.7491958462633193e-05), (3, 2.46575891651446e-05), (4, 2.242773371108342e-05), (5, 2.581702938186936e-05), (6, 1.071120368578704e-05), (7, 1.9512310245772824e-05), (8, 1.1732185157597996e-05), (9, 1.4133620425127447e-05), (10, 3.629833008744754e-05), (11, 3.069724334636703e-05), (12, 2.1380457837949507e-05), (13, 6.937645957805216e-05), (14, 5.305504964780994e-05), (15, 9.727515862323344e-05), (16, 3.8868085539434105e-05), (17, 0.00013531965669244528), (18, 6.648099224548787e-05), (19, 9.419619163963944e-05), (20, 0.00018308077414985746), (21, 6.42284139757976e-05), (22, 9.595809387974441e-05), (23, 7.958143396535888e-05), (24, 0.00022167604765854776), (25, 0.00026807960239239037), (26, 4.6831908548483625e-05), (27, 0.000148736871778965)]
+
+  4%|▍         | 6/150 [03:07<45:34, 18.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 858, Spatial MSE: [(0, 3.909549559466541e-05), (1, 0.0003469606745056808), (2, 0.00866476260125637), (3, 0.0009000820573419333), (4, 0.00037777330726385117), (5, 0.00024946132907643914), (6, 0.0002867280272766948), (7, 0.0002440063253743574), (8, 0.00018052445375360548), (9, 0.00018608693790156394), (10, 0.0002580113068688661), (11, 0.00023948751913849264), (12, 0.0003843102604150772), (13, 0.000585512665566057), (14, 0.0024756477214396), (15, 0.004243904259055853), (16, 0.023479798808693886), (17, 0.0019555927719920874), (18, 0.0024064385797828436), (19, 0.0033618269953876734), (20, 0.0046911705285310745), (21, 0.0022096331231296062), (22, 0.0026933399494737387), (23, 0.002767261816188693), (24, 0.0060285525396466255), (25, 0.017666812986135483), (26, 0.014341494999825954), (27, 0.2086983025074005)], Temporal MSE: [(0, 5.607887214864604e-05), (1, 4.953932511853054e-05), (2, 6.868639320600778e-05), (3, 5.926057929173112e-05), (4, 5.3069888963364065e-05), (5, 6.229778955457732e-05), (6, 2.771131221379619e-05), (7, 4.284263195586391e-05), (8, 2.8505637601483613e-05), (9, 3.478985308902338e-05), (10, 8.157802221830934e-05), (11, 7.103041571099311e-05), (12, 5.262394552119076e-05), (13, 0.0001899365015560761), (14, 0.0001278840354643762), (15, 0.0002627428330015391), (16, 0.00010244159784633666), (17, 0.00037993473233655095), (18, 0.00016667292220517993), (19, 0.00024684591335244477), (20, 0.00044561922550201416), (21, 0.00020684642367996275), (22, 0.00023063362459652126), (23, 0.0002128282649209723), (24, 0.0006165452650748193), (25, 0.0008942572749219835), (26, 0.00013216758088674396), (27, 0.00044381077168509364)]
+
+  5%|▍         | 7/150 [03:19<39:37, 16.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 852, Spatial MSE: [(0, 4.804255695489701e-07), (1, 2.8876688702439424e-06), (2, 5.163085006643087e-05), (3, 7.462326266249875e-06), (4, 2.6597920168569544e-06), (5, 2.424809053991339e-06), (6, 5.3637600103684235e-06), (7, 8.778175470069982e-06), (8, 8.491634616802912e-06), (9, 2.1267762349452823e-05), (10, 2.7805728677776642e-05), (11, 3.7000841984990984e-05), (12, 3.280811506556347e-05), (13, 4.680813435697928e-05), (14, 9.13866824703291e-05), (15, 0.00016822740144561976), (16, 0.000621418294031173), (17, 7.090234430506825e-05), (18, 9.383144060848281e-05), (19, 0.0001063163363141939), (20, 8.979806443676353e-05), (21, 8.092943608062342e-05), (22, 9.404835873283446e-05), (23, 9.583750943420455e-05), (24, 0.0002170941443182528), (25, 0.00030730865546502173), (26, 0.0001436055317753926), (27, 0.001145030721090734)], Temporal MSE: [(0, 5.979727575322613e-07), (1, 3.7616328540934774e-07), (2, 5.408300012277323e-07), (3, 4.834217293137044e-07), (4, 4.286273735942814e-07), (5, 7.315129550988786e-07), (6, 6.177851901156828e-07), (7, 8.204107189158094e-07), (8, 9.271997214455041e-07), (9, 2.1843800368515076e-06), (10, 5.051151219959138e-06), (11, 4.387233275338076e-06), (12, 4.961459580954397e-06), (13, 1.0275680324411951e-05), (14, 1.7809670680435374e-05), (15, 1.4450886737904511e-05), (16, 1.140317090175813e-05), (17, 1.846131999627687e-05), (18, 1.389929821016267e-05), (19, 1.6991292795864865e-05), (20, 3.62315695383586e-05), (21, 1.2493467693275306e-05), (22, 9.625989150663372e-06), (23, 1.2556568435684312e-05), (24, 4.044908200739883e-05), (25, 2.98794184345752e-05), (26, 4.826153144676937e-06), (27, 2.3848160708439536e-06)]
+
+  5%|▌         | 8/150 [03:32<36:33, 15.45s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 846, Spatial MSE: [(0, 2.8507304250524612e-06), (1, 2.0369845515233465e-05), (2, 0.0006002486334182322), (3, 0.00010141260281670839), (4, 2.4119099180097692e-05), (5, 1.2426576176949311e-05), (6, 2.918557947850786e-05), (7, 1.1305757652735338e-05), (8, 1.1212458957743365e-05), (9, 1.2833830624003895e-05), (10, 1.1770253877330106e-05), (11, 2.0223453248036094e-05), (12, 2.334420059924014e-05), (13, 0.0002407894207863137), (14, 0.0007055413443595171), (15, 0.0003170739801134914), (16, 0.009366734884679317), (17, 6.690220470773056e-05), (18, 0.0002571670338511467), (19, 0.000520758971106261), (20, 0.00014704684144817293), (21, 0.0003457473940216005), (22, 0.0005222396575845778), (23, 0.00027530602528713644), (24, 0.0005369093851186335), (25, 0.0013167213182896376), (26, 0.001316905370913446), (27, 0.01628996804356575)], Temporal MSE: [(0, 3.968228156736586e-06), (1, 2.5684673801151803e-06), (2, 3.9125493458413985e-06), (3, 3.717570734806941e-06), (4, 3.2591333365417086e-06), (5, 3.887661023327382e-06), (6, 1.8644642523213406e-06), (7, 1.9295002857688814e-06), (8, 1.579226022840885e-06), (9, 2.1186101548664737e-06), (10, 3.283453906988143e-06), (11, 6.38601432001451e-06), (12, 6.967578883632086e-06), (13, 3.227948400308378e-05), (14, 1.1561164683371317e-05), (15, 1.2154536307207309e-05), (16, 1.401479312335141e-05), (17, 2.2621114112553187e-05), (18, 1.1690186511259526e-05), (19, 1.203670854010852e-05), (20, 1.758173129928764e-05), (21, 2.2642914700554684e-05), (22, 6.707877037115395e-05), (23, 9.95100854197517e-05), (24, 0.00012240167416166514), (25, 4.874735532212071e-05), (26, 2.88349183392711e-05), (27, 8.998651537694968e-06)]
+
+  6%|▌         | 9/150 [03:45<34:45, 14.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 840, Spatial MSE: [(0, 5.979137768008513e-06), (1, 0.00013808492803946137), (2, 0.0024048977065831423), (3, 0.0002013864868786186), (4, 9.844710439210758e-05), (5, 6.991876580286771e-05), (6, 5.836312993778847e-05), (7, 8.479135431116447e-05), (8, 5.893705747439526e-05), (9, 5.765926471212879e-05), (10, 8.01273636170663e-05), (11, 8.058887033257633e-05), (12, 0.00015680590877309442), (13, 0.00033335998887196183), (14, 0.0011701270705088973), (15, 0.0012372087221592665), (16, 0.011425663717091084), (17, 0.0004174733185209334), (18, 0.0004763082251884043), (19, 0.001024650759063661), (20, 0.0007229435723274946), (21, 0.0006584386574104428), (22, 0.001049703685566783), (23, 0.0009456832194700837), (24, 0.0021351955365389585), (25, 0.004360081162303686), (26, 0.0037963681388646364), (27, 0.027054371312260628)], Temporal MSE: [(0, 7.336429007409606e-06), (1, 1.3826291251461953e-05), (2, 1.3415977264230605e-05), (3, 1.0905486306000967e-05), (4, 1.2980091014469508e-05), (5, 1.4322949937195517e-05), (6, 8.02275008027209e-06), (7, 1.6467627574456856e-05), (8, 6.705403848172864e-06), (9, 1.1940648619201966e-05), (10, 4.051392534165643e-05), (11, 3.696655039675534e-05), (12, 2.1110568923177198e-05), (13, 7.852786802686751e-05), (14, 4.887416434939951e-05), (15, 7.782984903315082e-05), (16, 4.103003811906092e-05), (17, 0.00012249023711774498), (18, 5.559034616453573e-05), (19, 7.87743556429632e-05), (20, 0.00019581409287638962), (21, 7.927820843178779e-05), (22, 0.00017952431517187506), (23, 0.0001946242991834879), (24, 0.0003300951502751559), (25, 0.0001969818549696356), (26, 6.957494770176709e-05), (27, 9.076748392544687e-05)]
+
+  7%|▋         | 10/150 [03:57<32:41, 14.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 834, Spatial MSE: [(0, 1.0638626918080263e-05), (1, 7.409560930682346e-05), (2, 0.0023149363696575165), (3, 0.0003442025335971266), (4, 7.992541213752702e-05), (5, 4.3608150008367375e-05), (6, 6.052917524357326e-05), (7, 4.679723861045204e-05), (8, 3.767761882045306e-05), (9, 4.253518636687659e-05), (10, 3.735609061550349e-05), (11, 6.042706445441581e-05), (12, 7.393781561404467e-05), (13, 0.000871822121553123), (14, 0.0022466485388576984), (15, 0.0008470526081509888), (16, 0.030495699495077133), (17, 0.00023536322987638414), (18, 0.0008650613017380238), (19, 0.0017719980096444488), (20, 0.0005211957031860948), (21, 0.0012824320001527667), (22, 0.0017074280185624957), (23, 0.0009635791066102684), (24, 0.0018547747749835253), (25, 0.004273063037544489), (26, 0.006041336338967085), (27, 0.07042630761861801)], Temporal MSE: [(0, 1.4485788597085048e-05), (1, 9.21399532671785e-06), (2, 1.37497772811912e-05), (3, 1.316012549068546e-05), (4, 1.0247256795992143e-05), (5, 1.1888768312928732e-05), (6, 6.048464001651155e-06), (7, 5.831222551933024e-06), (8, 5.11149528392707e-06), (9, 6.701015081489459e-06), (10, 1.019424780679401e-05), (11, 2.347526060475502e-05), (12, 2.3807680918253027e-05), (13, 0.00010968057904392481), (14, 3.580750126275234e-05), (15, 4.1341580072185025e-05), (16, 4.12601038988214e-05), (17, 6.560483598150313e-05), (18, 4.272041769581847e-05), (19, 4.5011649490334094e-05), (20, 6.257352652028203e-05), (21, 7.027683750493452e-05), (22, 0.00021431964705698192), (23, 0.00029093213379383087), (24, 0.0004137015203014016), (25, 0.00019646662985906005), (26, 0.0001131071403506212), (27, 3.381648275535554e-05)]
+
+  7%|▋         | 11/150 [04:07<29:04, 12.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 828, Spatial MSE: [(0, 1.1537125828908756e-05), (1, 0.00027328715077601373), (2, 0.0036831169854849577), (3, 0.0002939215046353638), (4, 0.0001683329464867711), (5, 0.0001525173574918881), (6, 0.00012432431685738266), (7, 0.0003626783727668226), (8, 0.0002250010147690773), (9, 0.00034931712434627116), (10, 0.00048108890769071877), (11, 0.0006172297289595008), (12, 0.0008751429850235581), (13, 0.0019611280877143145), (14, 0.005896520335227251), (15, 0.006030331365764141), (16, 0.05267951637506485), (17, 0.002670664805918932), (18, 0.0030438245739787817), (19, 0.005261929240077734), (20, 0.006052780896425247), (21, 0.0031880082096904516), (22, 0.0038590088952332735), (23, 0.002996587660163641), (24, 0.006799120455980301), (25, 0.03086881712079048), (26, 0.01331144105643034), (27, 0.04815829545259476)], Temporal MSE: [(0, 1.6174053598660976e-05), (1, 2.498965841368772e-05), (2, 3.3301512303296477e-05), (3, 2.46934469032567e-05), (4, 2.4814726202748716e-05), (5, 4.272870864951983e-05), (6, 2.3665941625949927e-05), (7, 3.869236388709396e-05), (8, 3.4782344300765544e-05), (9, 6.378354737535119e-05), (10, 0.00016344754840247333), (11, 0.00015510570665355772), (12, 0.00016227927699219435), (13, 0.0004585708666127175), (14, 0.00047729851212352514), (15, 0.0005572165246121585), (16, 0.00034839747240766883), (17, 0.0008562390576116741), (18, 0.0005363985546864569), (19, 0.0005780506180599332), (20, 0.0012103226035833359), (21, 0.0006577031454071403), (22, 0.00039955659303814173), (23, 0.0005835534539073706), (24, 0.0013472831342369318), (25, 0.0036138000432401896), (26, 0.0003440213040448725), (27, 0.0003189478302374482)]
+
+  8%|▊         | 12/150 [04:16<26:36, 11.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 822, Spatial MSE: [(0, 1.4226462553779129e-05), (1, 0.00010436559387017041), (2, 0.002921185689046979), (3, 0.0005013434565626085), (4, 0.00011142055882373825), (5, 5.7736637245398015e-05), (6, 0.00011062159319408238), (7, 5.526070162886754e-05), (8, 4.711073779617436e-05), (9, 5.764351590187289e-05), (10, 5.332072396413423e-05), (11, 9.557814337313175e-05), (12, 0.00014553929213434458), (13, 0.0022583724930882454), (14, 0.004564029164612293), (15, 0.0014459040248766541), (16, 0.05268273502588272), (17, 0.00032284349435940385), (18, 0.0011166682234033942), (19, 0.0032328192610293627), (20, 0.0009774280479177833), (21, 0.0014556479873135686), (22, 0.0026634386740624905), (23, 0.0014307174133136868), (24, 0.0026964694261550903), (25, 0.0067157684825360775), (26, 0.008578654378652573), (27, 0.1063523143529892)], Temporal MSE: [(0, 1.8711441953200847e-05), (1, 1.2304544725338928e-05), (2, 1.653686376812402e-05), (3, 1.638443791307509e-05), (4, 1.3793101970804855e-05), (5, 1.6558778952457942e-05), (6, 8.036830877244938e-06), (7, 8.680426617502235e-06), (8, 7.0013347794883884e-06), (9, 9.370931365992874e-06), (10, 1.3454066902340855e-05), (11, 5.9852809499716386e-05), (12, 5.836436685058288e-05), (13, 0.0002469967876095325), (14, 6.352105992846191e-05), (15, 6.937543366802856e-05), (16, 6.943183689145371e-05), (17, 9.558031160850078e-05), (18, 6.944403139641508e-05), (19, 6.492749525932595e-05), (20, 8.513903594575822e-05), (21, 0.00011379117495380342), (22, 0.0003297747753094882), (23, 0.0005116692627780139), (24, 0.0006837965338490903), (25, 0.0003086088690906763), (26, 0.000181580864591524), (27, 4.910419738735072e-05)]
+
+  9%|▊         | 13/150 [04:27<26:03, 11.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 816, Spatial MSE: [(0, 1.416915711160982e-05), (1, 0.0002784730168059468), (2, 0.0042304908856749535), (3, 0.00033680492197163403), (4, 0.00019055338634643704), (5, 0.00013889593537896872), (6, 7.969584839884192e-05), (7, 0.00018365695723332465), (8, 0.00011081982665928081), (9, 0.0001330263476120308), (10, 0.0001734281104290858), (11, 0.00020628875063266605), (12, 0.0004583040135912597), (13, 0.0032747145742177963), (14, 0.008328591473400593), (15, 0.0046699512749910355), (16, 0.08425798267126083), (17, 0.0011257958831265569), (18, 0.0014883610419929028), (19, 0.006285283714532852), (20, 0.0038126546423882246), (21, 0.0020703477784991264), (22, 0.0035784130450338125), (23, 0.002650281647220254), (24, 0.006912313401699066), (25, 0.01391333807259798), (26, 0.010158215649425983), (27, 0.05749615654349327)], Temporal MSE: [(0, 1.6513098671566695e-05), (1, 2.9140857805032283e-05), (2, 2.5393906980752945e-05), (3, 2.0338362446636893e-05), (4, 2.7266029064776376e-05), (5, 2.9680102670681663e-05), (6, 1.983287620532792e-05), (7, 2.9697042918996885e-05), (8, 1.6836924260132946e-05), (9, 3.4799366403603926e-05), (10, 8.224600605899468e-05), (11, 0.00012964816414751112), (12, 0.00010683396249078214), (13, 0.00039854765054769814), (14, 0.00015188280667643994), (15, 0.00025564723182469606), (16, 0.00017393183952663094), (17, 0.000386313273338601), (18, 0.00022026260558050126), (19, 0.0002941055572591722), (20, 0.0007207951857708395), (21, 0.0003533498093020171), (22, 0.0005293029826134443), (23, 0.0007612314075231552), (24, 0.0010763178579509258), (25, 0.0009467776981182396), (26, 0.00025380091392435133), (27, 0.00030525377951562405)]
+
+  9%|▉         | 14/150 [04:37<25:09, 11.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 810, Spatial MSE: [(0, 1.9182298274245113e-05), (1, 0.00013146975834388286), (2, 0.00335578597150743), (3, 0.0006067063077352941), (4, 0.00012921167945023626), (5, 6.945189670659602e-05), (6, 0.00014329988334793597), (7, 7.184848800534382e-05), (8, 5.726108429371379e-05), (9, 7.218772225314751e-05), (10, 6.91658424329944e-05), (11, 0.00015499813889618963), (12, 0.0003141202323604375), (13, 0.005998693406581879), (14, 0.010803097859025002), (15, 0.0031368916388601065), (16, 0.1006726548075676), (17, 0.0005315771559253335), (18, 0.0018588698003441095), (19, 0.006351551041007042), (20, 0.001983675407245755), (21, 0.0025076086167246103), (22, 0.0041998522356152534), (23, 0.002078695222735405), (24, 0.0037255757488310337), (25, 0.009545782580971718), (26, 0.010018004104495049), (27, 0.15654832124710083)], Temporal MSE: [(0, 2.5607203497202136e-05), (1, 1.541435267427005e-05), (2, 2.0601113646989688e-05), (3, 1.921732291521039e-05), (4, 1.6513366063009016e-05), (5, 2.0095960280741565e-05), (6, 1.0120455954165664e-05), (7, 1.0230391126242466e-05), (8, 8.708519999345299e-06), (9, 1.1782496585510671e-05), (10, 1.7895554265123792e-05), (11, 0.0001574688940308988), (12, 0.00014655044651590288), (13, 0.0005439119995571673), (14, 0.00012397815589793026), (15, 0.00012780383985955268), (16, 0.00011937975068576634), (17, 0.00015047151828184724), (18, 0.00014795524475630373), (19, 0.00012096361024305224), (20, 0.0001588286686455831), (21, 0.00020345313532743603), (22, 0.0005296786548569798), (23, 0.0008540483540855348), (24, 0.0010719354031607509), (25, 0.00042224820936098695), (26, 0.00026552521740086377), (27, 8.510133193340153e-05)]
+
+ 10%|█         | 15/150 [04:47<23:38, 10.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 804, Spatial MSE: [(0, 1.4082178495300468e-05), (1, 0.0003689487057272345), (2, 0.004136077594012022), (3, 0.00031468155793845654), (4, 0.00017819557979237288), (5, 0.0001850244152592495), (6, 0.0001777831930667162), (7, 0.0006186353275552392), (8, 0.00044405352673493326), (9, 0.0008098097168840468), (10, 0.0009700065711513162), (11, 0.001195979188196361), (12, 0.0016338737914338708), (13, 0.008311583660542965), (14, 0.016356168314814568), (15, 0.009196290746331215), (16, 0.12411565333604813), (17, 0.00471134576946497), (18, 0.003971101250499487), (19, 0.011089730076491833), (20, 0.010849797166883945), (21, 0.004993138834834099), (22, 0.00621678726747632), (23, 0.004248721990734339), (24, 0.010322486981749535), (25, 0.05364641174674034), (26, 0.024444956332445145), (27, 0.08876059949398041)], Temporal MSE: [(0, 1.8384620489086956e-05), (1, 3.688121068989858e-05), (2, 3.857446790789254e-05), (3, 3.162167195114307e-05), (4, 2.9959386665723287e-05), (5, 6.637017213506624e-05), (6, 4.037451435578987e-05), (7, 6.269892764976248e-05), (8, 7.501221261918545e-05), (9, 0.00012689875438809395), (10, 0.00031151852454058826), (11, 0.0004347269132267684), (12, 0.00044209943735040724), (13, 0.0011319610057398677), (14, 0.000705721671693027), (15, 0.0009302284452132881), (16, 0.0005998728447593749), (17, 0.0013002336490899324), (18, 0.001016872120089829), (19, 0.0009049178916029632), (20, 0.0016154347686097026), (21, 0.0013380366144701838), (22, 0.0007978131179697812), (23, 0.0012084564659744501), (24, 0.002512388164177537), (25, 0.007488681003451347), (26, 0.0007610612665303051), (27, 0.00038832571590319276)]
+
+ 11%|█         | 16/150 [04:56<22:56, 10.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 798, Spatial MSE: [(0, 1.7966785890166648e-05), (1, 0.0001340501185040921), (2, 0.0030726988334208727), (3, 0.0005856321658939123), (4, 0.00011328698019497097), (5, 6.309992750175297e-05), (6, 0.00013605989806819707), (7, 6.435094837797806e-05), (8, 5.141801375430077e-05), (9, 6.68476932332851e-05), (10, 6.762930570403114e-05), (11, 0.00016205479914788157), (12, 0.00038318702718243003), (13, 0.008082094602286816), (14, 0.01364425290375948), (15, 0.0038817478343844414), (16, 0.1258501261472702), (17, 0.0005463968846015632), (18, 0.0019920486956834793), (19, 0.00810539536178112), (20, 0.0026236148551106453), (21, 0.002615174977108836), (22, 0.004807778634130955), (23, 0.0023143216967582703), (24, 0.004027859773486853), (25, 0.010295474901795387), (26, 0.010833012871444225), (27, 0.16717031598091125)], Temporal MSE: [(0, 2.31891553994501e-05), (1, 1.3586660315922927e-05), (2, 1.745525332808029e-05), (3, 1.6485957530676387e-05), (4, 1.3641581062984187e-05), (5, 1.8171953342971392e-05), (6, 9.28387453313917e-06), (7, 8.989365596789867e-06), (8, 7.761865163047332e-06), (9, 1.116242947318824e-05), (10, 1.7433598259231076e-05), (11, 0.00017034726624842733), (12, 0.00018717155035119504), (13, 0.0006648881826549768), (14, 0.0001497553603257984), (15, 0.00014822217053733766), (16, 0.00013843785563949496), (17, 0.0001630084589123726), (18, 0.0001942801900440827), (19, 0.00014942132111173123), (20, 0.00022832441027276218), (21, 0.00023693540424574167), (22, 0.0005964903393760324), (23, 0.0010213205823674798), (24, 0.0012683998793363571), (25, 0.0005343725788407028), (26, 0.0003179993072990328), (27, 0.00010829589155036956)]
+
+ 11%|█▏        | 17/150 [05:07<22:58, 10.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 792, Spatial MSE: [(0, 1.689613600319717e-05), (1, 0.00032719544833526015), (2, 0.004307097289711237), (3, 0.00035009923158213496), (4, 0.00015682778030168265), (5, 0.0001430703268852085), (6, 0.00011740650370484218), (7, 0.00019938344485126436), (8, 0.00012136471195844933), (9, 0.00015956196875777096), (10, 0.00019334409444127232), (11, 0.00027946283807978034), (12, 0.0006337608210742474), (13, 0.00841144286096096), (14, 0.014677973464131355), (15, 0.0051069981418550014), (16, 0.12790131568908691), (17, 0.0011880624806508422), (18, 0.0024434567894786596), (19, 0.0087794354185462), (20, 0.0035807716194540262), (21, 0.0029211873188614845), (22, 0.005405996926128864), (23, 0.0033108857460319996), (24, 0.007944267243146896), (25, 0.01959676295518875), (26, 0.012127782218158245), (27, 0.10521452128887177)], Temporal MSE: [(0, 1.8975244529428892e-05), (1, 2.9465572879416868e-05), (2, 2.489667349436786e-05), (3, 2.1242123693809845e-05), (4, 2.663736631802749e-05), (5, 3.133597783744335e-05), (6, 2.3298669475479983e-05), (7, 2.9078770239721052e-05), (8, 2.0426879927981645e-05), (9, 4.129986700718291e-05), (10, 7.84301373641938e-05), (11, 0.0002262732741655782), (12, 0.00022625309065915644), (13, 0.0007411905680783093), (14, 0.00021677535551134497), (15, 0.0003299664531368762), (16, 0.00026469893055036664), (17, 0.0005313645233400166), (18, 0.00039376935455948114), (19, 0.0004625478759407997), (20, 0.0010927981929853559), (21, 0.0005957232206128538), (22, 0.0007267256150953472), (23, 0.0012588549870997667), (24, 0.001721012289635837), (25, 0.0018887087935581803), (26, 0.00039115778054110706), (27, 0.0003573097346816212)]
+
+ 12%|█▏        | 18/150 [05:18<23:33, 10.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 786, Spatial MSE: [(0, 2.267957097501494e-05), (1, 0.0001686655596131459), (2, 0.0034284156281501055), (3, 0.0007239766418933868), (4, 0.0001290664658881724), (5, 7.269262277986854e-05), (6, 0.00018341353279538453), (7, 7.276445830939338e-05), (8, 6.047644274076447e-05), (9, 8.047037408687174e-05), (10, 8.050259202718735e-05), (11, 0.00018679296772461385), (12, 0.00045054571819491684), (13, 0.010017991997301579), (14, 0.015653837472200394), (15, 0.004246674478054047), (16, 0.12460429221391678), (17, 0.0005327904946170747), (18, 0.0015483489260077477), (19, 0.008267420344054699), (20, 0.002818932058289647), (21, 0.0020872633904218674), (22, 0.004546483512967825), (23, 0.002348073525354266), (24, 0.004052292555570602), (25, 0.010544328019022942), (26, 0.011229122057557106), (27, 0.1982041597366333)], Temporal MSE: [(0, 2.8998225388932042e-05), (1, 1.621391493245028e-05), (2, 2.0267008949304e-05), (3, 1.872669417934958e-05), (4, 1.546566272736527e-05), (5, 2.1903508240939118e-05), (6, 1.1166103831783403e-05), (7, 1.1114207154605538e-05), (8, 9.27485143620288e-06), (9, 1.3349938853934873e-05), (10, 2.1653422663803212e-05), (11, 0.00019535369938239455), (12, 0.00021843831927981228), (13, 0.0007499965140596032), (14, 0.00016656365187373012), (15, 0.00016549136489629745), (16, 0.00014751451089978218), (17, 0.00015603943029418588), (18, 0.00017239457520190626), (19, 0.00013262676657177508), (20, 0.00015475934196729213), (21, 0.00022712488134857267), (22, 0.0005635405541397631), (23, 0.0010697916150093079), (24, 0.0013409719103947282), (25, 0.0006055015837773681), (26, 0.0003512023249641061), (27, 0.0001368040539091453)]
+
+ 13%|█▎        | 19/150 [05:28<23:00, 10.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 780, Spatial MSE: [(0, 2.1890975403948687e-05), (1, 0.0005548976478166878), (2, 0.006524077616631985), (3, 0.0005320793716236949), (4, 0.00020535232033580542), (5, 0.0002671285474207252), (6, 0.0003727458242792636), (7, 0.0009304406703449786), (8, 0.0006705395644530654), (9, 0.0010629093740135431), (10, 0.0013470057165250182), (11, 0.0016018880996853113), (12, 0.002184053184464574), (13, 0.012483212165534496), (14, 0.020923009142279625), (15, 0.01034304779022932), (16, 0.14915035665035248), (17, 0.004964737221598625), (18, 0.004327073227614164), (19, 0.013057216070592403), (20, 0.011581630446016788), (21, 0.004697973374277353), (22, 0.007043148390948772), (23, 0.005473094992339611), (24, 0.012195631861686707), (25, 0.07284976541996002), (26, 0.03408251702785492), (27, 0.15661293268203735)], Temporal MSE: [(0, 3.0391600375878625e-05), (1, 5.091512866783887e-05), (2, 5.2934392442693934e-05), (3, 4.370515307527967e-05), (4, 3.693342296173796e-05), (5, 8.606803021393716e-05), (6, 6.501925963675603e-05), (7, 9.504693298367783e-05), (8, 0.00011785252718254924), (9, 0.0001784468040568754), (10, 0.0004448145627975464), (11, 0.0006417141994461417), (12, 0.0005975931417196989), (13, 0.0015352184418588877), (14, 0.0011532577918842435), (15, 0.0015101197641342878), (16, 0.0009205202222801745), (17, 0.0019516563042998314), (18, 0.0016404460184276104), (19, 0.0017828157870098948), (20, 0.003166902344673872), (21, 0.002920513041317463), (22, 0.0010784133337438107), (23, 0.0017411367734894156), (24, 0.004284614231437445), (25, 0.01451310608536005), (26, 0.0010825475910678506), (27, 0.0003957980661652982)]
+
+ 13%|█▎        | 20/150 [05:38<22:04, 10.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 774, Spatial MSE: [(0, 3.340612965985201e-05), (1, 0.0002448380400892347), (2, 0.0043084085918962955), (3, 0.0009960199240595102), (4, 0.00017063107225112617), (5, 9.986983786802739e-05), (6, 0.00026808513212017715), (7, 9.292343020206317e-05), (8, 8.150237408699468e-05), (9, 0.00011580251157283783), (10, 0.0001226906169904396), (11, 0.0003264326660428196), (12, 0.0007984680123627186), (13, 0.016979627311229706), (14, 0.025127490982413292), (15, 0.007179785519838333), (16, 0.21596290171146393), (17, 0.0008031451143324375), (18, 0.0022420615423470736), (19, 0.013909006491303444), (20, 0.0048929238691926), (21, 0.003120372537523508), (22, 0.007012155372649431), (23, 0.0037341492716223), (24, 0.006404584273695946), (25, 0.01601344533264637), (26, 0.018424415960907936), (27, 0.30646243691444397)], Temporal MSE: [(0, 4.187170270597562e-05), (1, 2.2183743567438796e-05), (2, 2.7053220037487336e-05), (3, 2.5004954295582138e-05), (4, 2.037936974375043e-05), (5, 3.055861088796519e-05), (6, 1.5376041119452566e-05), (7, 1.5147812518989667e-05), (8, 1.3284627129905857e-05), (9, 1.9451023035799153e-05), (10, 3.2265030313283205e-05), (11, 0.00046263201511465013), (12, 0.000371950794942677), (13, 0.0010847592493519187), (14, 0.00026932722539640963), (15, 0.0002752635336946696), (16, 0.00023712380789220333), (17, 0.00022302582510747015), (18, 0.0003088956291321665), (19, 0.0002219137386418879), (20, 0.00022602449462283403), (21, 0.00034716157824732363), (22, 0.000865835405420512), (23, 0.001758741564117372), (24, 0.002215723041445017), (25, 0.001109239412471652), (26, 0.000602453132160008), (27, 0.0002327723486814648)]
+
+ 14%|█▍        | 21/150 [05:49<22:30, 10.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 768, Spatial MSE: [(0, 3.054653643630445e-05), (1, 0.0006131638074293733), (2, 0.007358307018876076), (3, 0.0007168948068283498), (4, 0.00020237114222254604), (5, 0.0002291396667715162), (6, 0.0003202602965757251), (7, 0.0003590728738345206), (8, 0.0002955272502731532), (9, 0.0005397172644734383), (10, 0.0005487989401444793), (11, 0.0007649330655112863), (12, 0.001307670958340168), (13, 0.017687320709228516), (14, 0.02679971605539322), (15, 0.00880888570100069), (16, 0.22095471620559692), (17, 0.0016139060026034713), (18, 0.003391610225662589), (19, 0.01511978916823864), (20, 0.006522106472402811), (21, 0.004097553435713053), (22, 0.008074942976236343), (23, 0.004584764130413532), (24, 0.010231749154627323), (25, 0.031043872237205505), (26, 0.01782197318971157), (27, 0.21172162890434265)], Temporal MSE: [(0, 3.465394547674805e-05), (1, 4.9557209422346205e-05), (2, 3.947734148823656e-05), (3, 3.548982567735948e-05), (4, 3.725557689904235e-05), (5, 5.0842998462030664e-05), (6, 4.094870018889196e-05), (7, 4.649792390409857e-05), (8, 4.718911077361554e-05), (9, 7.247595931403339e-05), (10, 0.00011753005674108863), (11, 0.0005712616839446127), (12, 0.00047120588715188205), (13, 0.0012288071447983384), (14, 0.0004217958194203675), (15, 0.0005919379182159901), (16, 0.0005041956319473684), (17, 0.0008805823745205998), (18, 0.0007289610803127289), (19, 0.0008681546314619482), (20, 0.0016275409143418074), (21, 0.0012368395691737533), (22, 0.0010520756477490067), (23, 0.002172853099182248), (24, 0.0030134632252156734), (25, 0.004953829105943441), (26, 0.0006435367395170033), (27, 0.00048113815137185156)]
+
+ 15%|█▍        | 22/150 [05:59<22:11, 10.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 762, Spatial MSE: [(0, 4.203086064080708e-05), (1, 0.0003135925508104265), (2, 0.004450319800525904), (3, 0.001140959793701768), (4, 0.00017955877410713583), (5, 0.00010588811710476875), (6, 0.0003590567212086171), (7, 9.401215356774628e-05), (8, 8.96040364750661e-05), (9, 0.00012407540634740144), (10, 0.00013291771756485105), (11, 0.0003140131011605263), (12, 0.0008274342981167138), (13, 0.020230084657669067), (14, 0.028073547407984734), (15, 0.007863136939704418), (16, 0.22420082986354828), (17, 0.0008450455497950315), (18, 0.0021428458858281374), (19, 0.014942529611289501), (20, 0.005449620541185141), (21, 0.00300681428052485), (22, 0.006847284268587828), (23, 0.00379108521156013), (24, 0.006457912735641003), (25, 0.015391488559544086), (26, 0.019702808931469917), (27, 0.34712502360343933)], Temporal MSE: [(0, 5.013165718992241e-05), (1, 2.4686889446456917e-05), (2, 2.8684704375336878e-05), (3, 2.5802417439990677e-05), (4, 2.047469388344325e-05), (5, 3.384038427611813e-05), (6, 1.816909571061842e-05), (7, 1.6421472537331283e-05), (8, 1.4581900359189603e-05), (9, 2.163437420676928e-05), (10, 3.3894128137035295e-05), (11, 0.00038416721508838236), (12, 0.000405064900405705), (13, 0.0011426778510212898), (14, 0.000296082318527624), (15, 0.0003111011174041778), (16, 0.00024382126866839826), (17, 0.00020675650739576668), (18, 0.0003516165306791663), (19, 0.00025023252237588167), (20, 0.00024265398678835481), (21, 0.0003502782783471048), (22, 0.0008585643954575062), (23, 0.0017967288149520755), (24, 0.0022736082319170237), (25, 0.0013889985857531428), (26, 0.0006493944092653692), (27, 0.00028784165624529123)]
+
+ 15%|█▌        | 23/150 [06:12<23:22, 11.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 756, Spatial MSE: [(0, 3.8441052311100066e-05), (1, 0.0009346151491627097), (2, 0.009668789803981781), (3, 0.0007829637615941465), (4, 0.0002671196707524359), (5, 0.00039566619670949876), (6, 0.0007501147338189185), (7, 0.0012106829090043902), (8, 0.0010800502495840192), (9, 0.0014608652563765645), (10, 0.001688729738816619), (11, 0.002186706755310297), (12, 0.003047832753509283), (13, 0.022129423916339874), (14, 0.03162435442209244), (15, 0.015124130994081497), (16, 0.2242550551891327), (17, 0.006150605622678995), (18, 0.00645184051245451), (19, 0.0183798149228096), (20, 0.01368400827050209), (21, 0.007788916118443012), (22, 0.010195846669375896), (23, 0.007288612425327301), (24, 0.0184357687830925), (25, 0.09083180129528046), (26, 0.049546536058187485), (27, 0.25119149684906006)], Temporal MSE: [(0, 4.5387845602817833e-05), (1, 7.711840589763597e-05), (2, 7.008596730884165e-05), (3, 7.065928366500884e-05), (4, 5.186217822483741e-05), (5, 0.0001136856953962706), (6, 0.00012455227260943502), (7, 0.00015885387256275862), (8, 0.00017265828500967473), (9, 0.00024077072157524526), (10, 0.0005406092386692762), (11, 0.0008664130000397563), (12, 0.0008100495324470103), (13, 0.0019657087977975607), (14, 0.001718857092782855), (15, 0.002012524288147688), (16, 0.0014597532572224736), (17, 0.002734511625021696), (18, 0.0022988892160356045), (19, 0.003259511198848486), (20, 0.004036800470203161), (21, 0.004582015331834555), (22, 0.001769857481122017), (23, 0.0029717024881392717), (24, 0.0072302184998989105), (25, 0.025799153372645378), (26, 0.0018538893200457096), (27, 0.0005260155303403735)]
+
+ 16%|█▌        | 24/150 [06:24<23:53, 11.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 750, Spatial MSE: [(0, 4.854956205235794e-05), (1, 0.0003855448157992214), (2, 0.004199740011245012), (3, 0.00126337306573987), (4, 0.00018370337784290314), (5, 0.00011157956760143861), (6, 0.0005104074371047318), (7, 9.534230775898322e-05), (8, 9.829590271692723e-05), (9, 0.00013145581760909408), (10, 0.00014611141523346305), (11, 0.0003185833047609776), (12, 0.0007765622576698661), (13, 0.01821216754615307), (14, 0.02469327487051487), (15, 0.007410360965877771), (16, 0.22137491405010223), (17, 0.0008296050364151597), (18, 0.0018818583339452744), (19, 0.01455251220613718), (20, 0.005571533460170031), (21, 0.0027908459305763245), (22, 0.006240896414965391), (23, 0.0035662916488945484), (24, 0.005897715222090483), (25, 0.013466855511069298), (26, 0.019541852176189423), (27, 0.3372102975845337)], Temporal MSE: [(0, 5.610676817013882e-05), (1, 2.5266068405471742e-05), (2, 2.868168303393759e-05), (3, 2.4839029720169492e-05), (4, 1.9812994651147164e-05), (5, 3.536628355504945e-05), (6, 2.0525360014289618e-05), (7, 1.8926577467937022e-05), (8, 1.5833898942219093e-05), (9, 2.463861346768681e-05), (10, 3.6381017707753927e-05), (11, 0.0003824761661235243), (12, 0.0003596874594222754), (13, 0.0008971908246167004), (14, 0.00028417151770554483), (15, 0.0003443688037805259), (16, 0.0002320304629392922), (17, 0.00017760666378308088), (18, 0.0003906693309545517), (19, 0.0002698712924029678), (20, 0.00024909721105359495), (21, 0.00033328685094602406), (22, 0.0007906341343186796), (23, 0.001690374338068068), (24, 0.0021759802475571632), (25, 0.001819743774831295), (26, 0.0006596604362130165), (27, 0.00031712168129161)]
+
+ 17%|█▋        | 25/150 [06:34<22:37, 10.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 744, Spatial MSE: [(0, 3.770494367927313e-05), (1, 0.0005981636350043118), (2, 0.005971633363515139), (3, 0.0005622779135592282), (4, 0.00014758195902686566), (5, 0.00017495259817223996), (6, 0.0004051349242217839), (7, 0.0002210365200880915), (8, 0.0002421868994133547), (9, 0.0003506833454594016), (10, 0.0004139944794587791), (11, 0.0005957567482255399), (12, 0.0011194554390385747), (13, 0.018122348934412003), (14, 0.0246779415756464), (15, 0.008315598592162132), (16, 0.22405798733234406), (17, 0.0012551877880468965), (18, 0.003339681075885892), (19, 0.015312928706407547), (20, 0.006663098931312561), (21, 0.005445077549666166), (22, 0.00784551352262497), (23, 0.0035511190071702003), (24, 0.008873636834323406), (25, 0.022532658651471138), (26, 0.014366045594215393), (27, 0.20958931744098663)], Temporal MSE: [(0, 3.7881574826315045e-05), (1, 4.140640521654859e-05), (2, 3.543116690707393e-05), (3, 2.873123412427958e-05), (4, 2.790572398225777e-05), (5, 4.1412105929339305e-05), (6, 3.8400838093366474e-05), (7, 3.7313177017495036e-05), (8, 4.094834366696887e-05), (9, 6.078695514588617e-05), (10, 8.01125424914062e-05), (11, 0.0004387248191051185), (12, 0.0004195658548269421), (13, 0.0009375440422445536), (14, 0.00038194976514205337), (15, 0.0005489698960445821), (16, 0.0004734205431304872), (17, 0.0007081103976815939), (18, 0.0007786382338963449), (19, 0.0009280146332457662), (20, 0.0011927714804187417), (21, 0.0012746226275339723), (22, 0.0009934413246810436), (23, 0.002121907891705632), (24, 0.0031240731477737427), (25, 0.0071000210009515285), (26, 0.0006878206040710211), (27, 0.0004051928117405623)]
+
+ 17%|█▋        | 26/150 [06:43<21:49, 10.56s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 738, Spatial MSE: [(0, 3.8118996599223465e-05), (1, 0.00031017648871056736), (2, 0.0026839585043489933), (3, 0.0008151785586960614), (4, 0.00012290551967453212), (5, 7.678437395952642e-05), (6, 0.00040337047539651394), (7, 6.719520024489611e-05), (8, 6.943738117115572e-05), (9, 9.152926941169426e-05), (10, 0.00011046943109249696), (11, 0.00028201393433846533), (12, 0.0007664881413802505), (13, 0.02004384994506836), (14, 0.02460176683962345), (15, 0.007255596108734608), (16, 0.19758020341396332), (17, 0.0006580164190381765), (18, 0.0013887847308069468), (19, 0.013442262075841427), (20, 0.0052934857085347176), (21, 0.002194109372794628), (22, 0.004869461990892887), (23, 0.002649286761879921), (24, 0.00458191754296422), (25, 0.010001316666603088), (26, 0.013643112033605576), (27, 0.290109783411026)], Temporal MSE: [(0, 4.321777305449359e-05), (1, 1.8589724277262576e-05), (2, 2.050350849458482e-05), (3, 1.7250595192308538e-05), (4, 1.2833871551265474e-05), (5, 2.484738070052117e-05), (6, 1.6152383977896534e-05), (7, 1.3750155630987138e-05), (8, 1.1967312275373843e-05), (9, 1.8669366909307428e-05), (10, 2.5805058612604626e-05), (11, 0.000425821723183617), (12, 0.0003836314135696739), (13, 0.0008602029411122203), (14, 0.0002663061022758484), (15, 0.0003083625342696905), (16, 0.00020249038061592728), (17, 0.0001469371054554358), (18, 0.00035818200558423996), (19, 0.00024767607101239264), (20, 0.00021536815620493144), (21, 0.00029030145378783345), (22, 0.0006310045719146729), (23, 0.0014538443647325039), (24, 0.001863548532128334), (25, 0.0018426600145176053), (26, 0.0005680921021848917), (27, 0.0003367392055224627)]
+
+ 18%|█▊        | 27/150 [06:55<22:06, 10.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 732, Spatial MSE: [(0, 3.2514184567844495e-05), (1, 0.0006914404220879078), (2, 0.006134452763944864), (3, 0.0005455775535665452), (4, 0.00023260751913767308), (5, 0.00035031369770877063), (6, 0.0011263600317761302), (7, 0.001337917405180633), (8, 0.0012782032135874033), (9, 0.0017606583423912525), (10, 0.0022122475784271955), (11, 0.0030036771204322577), (12, 0.003839460900053382), (13, 0.023997807875275612), (14, 0.03392153978347778), (15, 0.01780172437429428), (16, 0.2187647670507431), (17, 0.006017183419317007), (18, 0.006156215909868479), (19, 0.019108977168798447), (20, 0.01223565824329853), (21, 0.01527224201709032), (22, 0.008519649505615234), (23, 0.006634269841015339), (24, 0.017961736768484116), (25, 0.042418185621500015), (26, 0.026159541681408882), (27, 0.2126893699169159)], Temporal MSE: [(0, 3.289633968961425e-05), (1, 5.03083792864345e-05), (2, 4.331245145294815e-05), (3, 5.087899262434803e-05), (4, 4.956148404744454e-05), (5, 0.00012158956815255806), (6, 0.00016170427261386067), (7, 0.0001730737858451903), (8, 0.0001756272977218032), (9, 0.00030649566906504333), (10, 0.000569186988286674), (11, 0.0010318647837266326), (12, 0.0009697230416350067), (13, 0.0019554863683879375), (14, 0.001742019783705473), (15, 0.0021128372754901648), (16, 0.0017480412498116493), (17, 0.0018855987582355738), (18, 0.0018979021115228534), (19, 0.002379746874794364), (20, 0.003489168593659997), (21, 0.0025348160415887833), (22, 0.001805078354664147), (23, 0.0030814940109848976), (24, 0.0073606413789093494), (25, 0.024274224415421486), (26, 0.0021400072146207094), (27, 0.0004373999254312366)]
+
+ 19%|█▊        | 28/150 [07:07<23:04, 11.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 726, Spatial MSE: [(0, 3.055363049497828e-05), (1, 0.00024120845773722976), (2, 0.0017132906941697001), (3, 0.0005478871753439307), (4, 8.17835025372915e-05), (5, 5.236636206973344e-05), (6, 0.0003321859985589981), (7, 4.9714984925230965e-05), (8, 4.90724814881105e-05), (9, 6.148665124783292e-05), (10, 8.013994374778122e-05), (11, 0.00019316974794492126), (12, 0.0005335752503015101), (13, 0.014719629660248756), (14, 0.01791604422032833), (15, 0.00550488568842411), (16, 0.16156727075576782), (17, 0.0005365728866308928), (18, 0.0009809323819354177), (19, 0.010746286250650883), (20, 0.004368892405182123), (21, 0.0016700419364497066), (22, 0.003465969581156969), (23, 0.001854696311056614), (24, 0.0031550461426377296), (25, 0.006801065988838673), (26, 0.009576409123837948), (27, 0.2132551372051239)], Temporal MSE: [(0, 3.26502849929966e-05), (1, 1.2188175787741784e-05), (2, 1.3757085071119945e-05), (3, 1.1107967111456674e-05), (4, 7.88076522439951e-06), (5, 1.698801679594908e-05), (6, 1.2550071915029548e-05), (7, 1.0749429748102557e-05), (8, 8.295534826174844e-06), (9, 1.334386342932703e-05), (10, 1.8355127394897863e-05), (11, 0.0002713122230488807), (12, 0.00027028529439121485), (13, 0.0006010638317093253), (14, 0.00020870442676823586), (15, 0.00026277283905074), (16, 0.00016044091898947954), (17, 0.00011925659782718867), (18, 0.00029404237284325063), (19, 0.00020531698828563094), (20, 0.00017131617642007768), (21, 0.00025497330352663994), (22, 0.00047963554970920086), (23, 0.0011211485834792256), (24, 0.0014172133523970842), (25, 0.0018256711773574352), (26, 0.0004555753548629582), (27, 0.0003049552033189684)]
+
+ 19%|█▉        | 29/150 [07:16<21:28, 10.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 720, Spatial MSE: [(0, 2.259348548250273e-05), (1, 0.00028263923013582826), (2, 0.0023715642746537924), (3, 0.00022044763318262994), (4, 6.436689000111073e-05), (5, 6.429921631934121e-05), (6, 0.00019581642118282616), (7, 8.869644807418808e-05), (8, 0.0001219116966240108), (9, 0.0001811511319829151), (10, 0.00024614864378236234), (11, 0.0003488587972242385), (12, 0.0006808530306443572), (13, 0.01492176204919815), (14, 0.01815507747232914), (15, 0.005814937874674797), (16, 0.16201511025428772), (17, 0.0006970663089305162), (18, 0.0015538044972345233), (19, 0.010964367538690567), (20, 0.004854854196310043), (21, 0.0033932561054825783), (22, 0.0043739937245845795), (23, 0.0018116050632670522), (24, 0.004509711172431707), (25, 0.0090170344337821), (26, 0.006235391832888126), (27, 0.15156865119934082)], Temporal MSE: [(0, 1.9074284864473157e-05), (1, 1.5213504411804024e-05), (2, 1.6879534086911008e-05), (3, 1.231077840202488e-05), (4, 1.1418091162340716e-05), (5, 1.766848799888976e-05), (6, 1.814424649637658e-05), (7, 1.680911736912094e-05), (8, 2.0847270207013935e-05), (9, 2.9958166123833507e-05), (10, 3.854497117572464e-05), (11, 0.0002959630801342428), (12, 0.00029351902776397765), (13, 0.0006338945822790265), (14, 0.0002548521151766181), (15, 0.00033757113851606846), (16, 0.0002688481763470918), (17, 0.0002805984695442021), (18, 0.0004405159270390868), (19, 0.00044322243775241077), (20, 0.0004836831649299711), (21, 0.000511551508679986), (22, 0.0005771915311925113), (23, 0.0013326217886060476), (24, 0.0019938761834055185), (25, 0.004017127677798271), (26, 0.0004999209195375443), (27, 0.0002708784013520926)]
+
+ 20%|██        | 30/150 [07:26<20:33, 10.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 714, Spatial MSE: [(0, 1.871189488156233e-05), (1, 0.00014861294766888022), (2, 0.0009339540265500546), (3, 0.0003223663952667266), (4, 4.674332376453094e-05), (5, 3.0082312150625512e-05), (6, 0.00021674844902008772), (7, 3.1642030080547556e-05), (8, 2.8921242119395174e-05), (9, 3.6691231798613444e-05), (10, 5.2221716032363474e-05), (11, 0.0001623809221200645), (12, 0.0004337456193752587), (13, 0.011231664568185806), (14, 0.012072729878127575), (15, 0.003621108829975128), (16, 0.10408741235733032), (17, 0.0003205470275133848), (18, 0.0005774015444330871), (19, 0.0068426840007305145), (20, 0.0027339071966707706), (21, 0.0010692140785977244), (22, 0.002156812697649002), (23, 0.0011504603317007422), (24, 0.001942049595527351), (25, 0.004114585928618908), (26, 0.005641549825668335), (27, 0.13829752802848816)], Temporal MSE: [(0, 2.0796484022866935e-05), (1, 7.4052668423973955e-06), (2, 8.379725841223262e-06), (3, 6.662543910351815e-06), (4, 4.483800239540869e-06), (5, 9.832659088715445e-06), (6, 7.825272405170836e-06), (7, 7.05960110281012e-06), (8, 5.012502697354648e-06), (9, 8.131459253490902e-06), (10, 1.150978005171055e-05), (11, 0.0002840918314177543), (12, 0.00021927441412117332), (13, 0.0004008492105640471), (14, 0.00014673176337964833), (15, 0.00016982998931780457), (16, 0.00010579542140476406), (17, 7.334387919399887e-05), (18, 0.00019369579968042672), (19, 0.0001362935727229342), (20, 0.00011497153900563717), (21, 0.00015742555842734873), (22, 0.00029391038697212934), (23, 0.0007274430245161057), (24, 0.0009094711276702583), (25, 0.0012624792288988829), (26, 0.000299280189210549), (27, 0.00019876040460076183)]
+
+ 21%|██        | 31/150 [07:36<20:34, 10.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 708, Spatial MSE: [(0, 1.4268149243434891e-05), (1, 0.00024718718486838043), (2, 0.001925514079630375), (3, 0.0002080807462334633), (4, 0.00010502264194656163), (5, 0.00018301895761396736), (6, 0.0005471865879371762), (7, 0.0006374942022375762), (8, 0.000941596576012671), (9, 0.001375282765366137), (10, 0.0016398051520809531), (11, 0.0019678061362355947), (12, 0.0023751144763082266), (13, 0.013011367991566658), (14, 0.01626463234424591), (15, 0.00908163283020258), (16, 0.1306547224521637), (17, 0.0035787816159427166), (18, 0.003587730461731553), (19, 0.010675134137272835), (20, 0.007025152910500765), (21, 0.005521940067410469), (22, 0.004428949672728777), (23, 0.003509657923132181), (24, 0.0090766167268157), (25, 0.017144132405519485), (26, 0.01001448929309845), (27, 0.10227624326944351)], Temporal MSE: [(0, 1.2836086170864291e-05), (1, 1.8366670701652765e-05), (2, 1.624382821319159e-05), (3, 2.1842584828846157e-05), (4, 3.371844650246203e-05), (5, 8.700160105945542e-05), (6, 0.00014301709597930312), (7, 0.00013973031309433281), (8, 0.00012170374975539744), (9, 0.0002343422529520467), (10, 0.0004369115049485117), (11, 0.0007643611752428114), (12, 0.0005870477179996669), (13, 0.0010287804761901498), (14, 0.0011676026042550802), (15, 0.0016690153861418366), (16, 0.0013456358574330807), (17, 0.0012718267971649766), (18, 0.0013927256222814322), (19, 0.0015083097387105227), (20, 0.00240047718398273), (21, 0.0016042919596657157), (22, 0.0010289750061929226), (23, 0.0017523841233924031), (24, 0.003941504750400782), (25, 0.011949419043958187), (26, 0.0014120140112936497), (27, 0.0002317360631423071)]
+
+ 21%|██▏       | 32/150 [07:46<19:46, 10.06s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 702, Spatial MSE: [(0, 9.030091860040557e-06), (1, 7.21462129149586e-05), (2, 0.00042049281182698905), (3, 0.00014858818030916154), (4, 2.173233042412903e-05), (5, 1.418517240381334e-05), (6, 0.00010661243868526071), (7, 1.8154969438910484e-05), (8, 1.438104209228186e-05), (9, 1.8442424334352836e-05), (10, 2.828513788699638e-05), (11, 8.704307401785627e-05), (12, 0.0002770860446617007), (13, 0.008592857979238033), (14, 0.009418933652341366), (15, 0.002803003415465355), (16, 0.0933401882648468), (17, 0.00030833258642815053), (18, 0.000537278363481164), (19, 0.006316261366009712), (20, 0.0027829818427562714), (21, 0.0008753678412176669), (22, 0.001752460841089487), (23, 0.0008066198788583279), (24, 0.0014088723110035062), (25, 0.0030345055274665356), (26, 0.0033393697813153267), (27, 0.09081486612558365)], Temporal MSE: [(0, 9.964534001483116e-06), (1, 3.432043058637646e-06), (2, 4.068926500622183e-06), (3, 3.293406507509644e-06), (4, 2.082698301819619e-06), (5, 4.898958650301211e-06), (6, 3.9567639760207385e-06), (7, 4.233125764585566e-06), (8, 2.4834885152813513e-06), (9, 4.17256933360477e-06), (10, 6.57937562209554e-06), (11, 0.0001427001116098836), (12, 0.00015788596647325903), (13, 0.0002969319757539779), (14, 0.00011443602124927565), (15, 0.00015573720156680793), (16, 9.098363807424903e-05), (17, 6.806856254115701e-05), (18, 0.00017197339911945164), (19, 0.00012600762420333922), (20, 0.0001043886732077226), (21, 0.0001498420024290681), (22, 0.00026127227465622127), (23, 0.000619136611931026), (24, 0.000706073478795588), (25, 0.0010008050594478846), (26, 0.00022428290685638785), (27, 0.00016446903464384377)]
+
+ 22%|██▏       | 33/150 [07:57<20:33, 10.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 696, Spatial MSE: [(0, 7.33069555280963e-06), (1, 8.686335786478594e-05), (2, 0.0006288576405495405), (3, 7.556096534244716e-05), (4, 2.148676139768213e-05), (5, 2.0641024093492888e-05), (6, 4.3462481698952615e-05), (7, 5.9394995332695544e-05), (8, 7.070227729855105e-05), (9, 0.00011173440725542605), (10, 0.00020151671196799725), (11, 0.0002358119236305356), (12, 0.00041814727592281997), (13, 0.008785028010606766), (14, 0.009682930074632168), (15, 0.0029521354008466005), (16, 0.09569070488214493), (17, 0.0004005018272437155), (18, 0.0007644310244359076), (19, 0.006986645981669426), (20, 0.0035419464111328125), (21, 0.0013513548765331507), (22, 0.00209756544791162), (23, 0.0008668475784361362), (24, 0.0020004408434033394), (25, 0.003751032520085573), (26, 0.003079770365729928), (27, 0.07600326836109161)], Temporal MSE: [(0, 5.867946583748562e-06), (1, 3.674719664559234e-06), (2, 5.6190033319580834e-06), (3, 4.1558605516911484e-06), (4, 3.922029009117978e-06), (5, 6.448882231779862e-06), (6, 7.551752332801698e-06), (7, 9.179401786241215e-06), (8, 9.390493687533308e-06), (9, 1.2466781299735885e-05), (10, 2.048494752671104e-05), (11, 0.0001603279815753922), (12, 0.00017211724480148405), (13, 0.00031886770739220083), (14, 0.00014651741366833448), (15, 0.00019007247465196997), (16, 0.0001345033524557948), (17, 0.00011309167166473344), (18, 0.00022065656958147883), (19, 0.00018405979790259153), (20, 0.00022609684674534947), (21, 0.00021786257275380194), (22, 0.0003131235425826162), (23, 0.0007018359028734267), (24, 0.0009820214472711086), (25, 0.0015190115664154291), (26, 0.0002600083826109767), (27, 0.000155432935571298)]
+
+ 23%|██▎       | 34/150 [08:12<22:44, 11.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 690, Spatial MSE: [(0, 4.9105665311799385e-06), (1, 3.695910345413722e-05), (2, 0.00023195537505671382), (3, 7.138284126995131e-05), (4, 1.1195616934855934e-05), (5, 7.712728802289348e-06), (6, 5.0808710511773825e-05), (7, 8.88781596586341e-06), (8, 8.0463014455745e-06), (9, 1.0317897249478847e-05), (10, 2.0375970052555203e-05), (11, 8.330302807735279e-05), (12, 0.0002761898504104465), (13, 0.007918349467217922), (14, 0.008044729009270668), (15, 0.00229651085101068), (16, 0.07022872567176819), (17, 0.00023942416009958833), (18, 0.0003731444012373686), (19, 0.004706002771854401), (20, 0.0020946485456079245), (21, 0.0007266869070008397), (22, 0.0012772553600370884), (23, 0.0006009289645589888), (24, 0.0010573324980214238), (25, 0.002120812190696597), (26, 0.0021006837487220764), (27, 0.06818757951259613)], Temporal MSE: [(0, 5.495147888723295e-06), (1, 1.9704207261384e-06), (2, 2.400851826678263e-06), (3, 1.9008850813406752e-06), (4, 1.115523218686576e-06), (5, 2.695359398785513e-06), (6, 2.1975884010316804e-06), (7, 2.2160234038892668e-06), (8, 1.3600326838059118e-06), (9, 2.3120144305721624e-06), (10, 5.051410880696494e-06), (11, 0.00016466333181597292), (12, 0.0001519130019005388), (13, 0.0002381253580097109), (14, 8.966510358732194e-05), (15, 0.00012462932500056922), (16, 7.001295307418332e-05), (17, 5.275024886941537e-05), (18, 0.0001392200356349349), (19, 0.00010079082130687311), (20, 9.261401282856241e-05), (21, 0.00010052625293610618), (22, 0.00018113577971234918), (23, 0.00045230152318254113), (24, 0.0005365806282497942), (25, 0.0007820745813660324), (26, 0.00017721134645398706), (27, 0.00012966823123861104)]
+
+ 23%|██▎       | 35/150 [08:22<21:18, 11.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 684, Spatial MSE: [(0, 4.700070348917507e-06), (1, 7.367367652477697e-05), (2, 0.0005537665565498173), (3, 7.443905633408576e-05), (4, 4.655991506297141e-05), (5, 9.330143075203523e-05), (6, 0.00014806863327976316), (7, 0.00025628480943851173), (8, 0.0004318418796174228), (9, 0.000635675503872335), (10, 0.0007160149398259819), (11, 0.0008817996131256223), (12, 0.0010733773233368993), (13, 0.008977090008556843), (14, 0.010086608119308949), (15, 0.004440560005605221), (16, 0.08229807019233704), (17, 0.0017724575009196997), (18, 0.0018873842200264335), (19, 0.006763123907148838), (20, 0.004712292458862066), (21, 0.002954146359115839), (22, 0.0025211793836206198), (23, 0.0017053202027454972), (24, 0.0038275353144854307), (25, 0.00789862871170044), (26, 0.004137907177209854), (27, 0.06254273653030396)], Temporal MSE: [(0, 5.004581907996908e-06), (1, 5.659366706822766e-06), (2, 5.836757736688014e-06), (3, 1.2443048944987822e-05), (4, 1.625480581424199e-05), (5, 3.648515121312812e-05), (6, 6.943105108803138e-05), (7, 6.328081508399919e-05), (8, 6.0189002397237346e-05), (9, 0.00011366868420736864), (10, 0.00023810482525732368), (11, 0.0004352373944129795), (12, 0.00033101977896876633), (13, 0.000529449200257659), (14, 0.0005373862222768366), (15, 0.0007556003984063864), (16, 0.0006814862717874348), (17, 0.000596537662204355), (18, 0.0006278496002778411), (19, 0.0008221007883548737), (20, 0.0011482132831588387), (21, 0.0008229963132180274), (22, 0.0005590122891589999), (23, 0.0009704878902994096), (24, 0.0018594699213281274), (25, 0.006179933436214924), (26, 0.0007348678773269057), (27, 0.00014489662135019898)]
+
+ 24%|██▍       | 36/150 [08:31<20:04, 10.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 678, Spatial MSE: [(0, 4.330986939748982e-06), (1, 2.9059969165246002e-05), (2, 0.0001737726852297783), (3, 6.62640668451786e-05), (4, 9.94901256490266e-06), (5, 6.328802101052133e-06), (6, 5.737407263950445e-05), (7, 1.0545011718932074e-05), (8, 7.06440278008813e-06), (9, 8.344217349076644e-06), (10, 1.5276042176992632e-05), (11, 5.303493526298553e-05), (12, 0.00018701616500038654), (13, 0.006161698140203953), (14, 0.00661004614084959), (15, 0.002052192809060216), (16, 0.0646592453122139), (17, 0.0001803207560442388), (18, 0.0003182293730787933), (19, 0.0043204063549637794), (20, 0.0018751659663394094), (21, 0.0006035016849637032), (22, 0.0010923424270004034), (23, 0.0005126915639266372), (24, 0.0008777379989624023), (25, 0.0017271884717047215), (26, 0.0018845993326976895), (27, 0.05832064896821976)], Temporal MSE: [(0, 4.7068078856682405e-06), (1, 1.5380304603240802e-06), (2, 1.7819594404500094e-06), (3, 1.614695975149516e-06), (4, 9.703168188934796e-07), (5, 2.3404716102959355e-06), (6, 1.7518832464702427e-06), (7, 3.007044142577797e-06), (8, 1.1916927178390324e-06), (9, 1.9321305444464087e-06), (10, 4.0274899220094085e-06), (11, 9.388202306581661e-05), (12, 0.00010658992687240243), (13, 0.0002107779000652954), (14, 8.205208723666146e-05), (15, 0.00010523851233301684), (16, 6.0986290918663144e-05), (17, 4.719711432699114e-05), (18, 0.0001200408223667182), (19, 8.969446935225278e-05), (20, 7.247050962178037e-05), (21, 9.577019955031574e-05), (22, 0.00016138216597028077), (23, 0.0004135607450734824), (24, 0.0004709856875706464), (25, 0.0008482001721858978), (26, 0.0001654220250202343), (27, 0.0001192861091112718)]
+
+ 25%|██▍       | 37/150 [08:43<21:00, 11.16s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 672, Spatial MSE: [(0, 3.578205223675468e-06), (1, 3.529768582666293e-05), (2, 0.00032362513593398035), (3, 3.582053614081815e-05), (4, 1.3827390830556396e-05), (5, 1.2539087038021535e-05), (6, 2.1403324353741482e-05), (7, 2.4102035240503028e-05), (8, 2.6802048523677513e-05), (9, 5.135675019118935e-05), (10, 9.445321484236047e-05), (11, 0.00013133134052623063), (12, 0.0002480607363395393), (13, 0.006368688307702541), (14, 0.006972279399633408), (15, 0.0023594750091433525), (16, 0.06952730566263199), (17, 0.00024576252326369286), (18, 0.00043973277206532657), (19, 0.004704331047832966), (20, 0.0024596648290753365), (21, 0.0007278670091181993), (22, 0.0012369833420962095), (23, 0.0006057258578948677), (24, 0.0012933982070535421), (25, 0.0025372824165970087), (26, 0.002010634634643793), (27, 0.05317634344100952)], Temporal MSE: [(0, 2.9862096653232584e-06), (1, 1.7378852135152556e-06), (2, 3.0491028155665845e-06), (3, 2.669542254807311e-06), (4, 2.559285803727107e-06), (5, 3.649789960036287e-06), (6, 4.196057034278056e-06), (7, 4.627856014849385e-06), (8, 4.90955153509276e-06), (9, 6.385105734807439e-06), (10, 1.0828734957613051e-05), (11, 0.00010416828445158899), (12, 0.00011920074030058458), (13, 0.0002226535143563524), (14, 9.896911797113717e-05), (15, 0.00012564098869916052), (16, 8.310374687425792e-05), (17, 7.274793460965157e-05), (18, 0.0001479798083892092), (19, 0.00011916952644241974), (20, 0.00013063471124041826), (21, 0.00014032195031177253), (22, 0.00020098038658034056), (23, 0.00047549314331263304), (24, 0.0006691162125207484), (25, 0.0012229916173964739), (26, 0.00019400280143599957), (27, 0.00011944973084609956)]
+
+ 25%|██▌       | 38/150 [08:55<20:49, 11.16s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 666, Spatial MSE: [(0, 3.1865022265264997e-06), (1, 2.3826178221497685e-05), (2, 0.00013358680007513613), (3, 4.5049324398860335e-05), (4, 7.977544555615168e-06), (5, 4.88467139803106e-06), (6, 3.760254185181111e-05), (7, 7.6826427175547e-06), (8, 5.446519935503602e-06), (9, 6.887771178298863e-06), (10, 1.5111174434423447e-05), (11, 7.229066250147298e-05), (12, 0.00021998326701577753), (13, 0.005379052832722664), (14, 0.005024624057114124), (15, 0.001329311402514577), (16, 0.04572344198822975), (17, 0.0001808630331652239), (18, 0.0002545775205362588), (19, 0.0032980029936879873), (20, 0.0016061728820204735), (21, 0.0004252455837558955), (22, 0.000789694779086858), (23, 0.0003705323033500463), (24, 0.0006682989769615233), (25, 0.001336422166787088), (26, 0.0015316351782530546), (27, 0.04241994395852089)], Temporal MSE: [(0, 3.436982069615624e-06), (1, 1.1884343393830932e-06), (2, 1.418610395376163e-06), (3, 1.2198770491522737e-06), (4, 7.048860766190046e-07), (5, 1.6643092521917424e-06), (6, 1.2658985042435233e-06), (7, 1.816227722883923e-06), (8, 9.02956628578977e-07), (9, 1.4305799140856834e-06), (10, 4.5268448047863785e-06), (11, 0.0001601619878783822), (12, 0.00011406252451706678), (13, 0.00014157549594528973), (14, 5.566305117099546e-05), (15, 8.3363258454483e-05), (16, 4.6081004256848246e-05), (17, 4.1487197449896485e-05), (18, 9.119577589444816e-05), (19, 7.245456799864769e-05), (20, 6.434738315874711e-05), (21, 7.603630365338176e-05), (22, 0.00012804607104044408), (23, 0.00029253773391246796), (24, 0.000330166716594249), (25, 0.0005573784001171589), (26, 0.00011680573516059667), (27, 9.199434134643525e-05)]
+
+ 26%|██▌       | 39/150 [09:05<20:15, 10.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 660, Spatial MSE: [(0, 2.595358182588825e-06), (1, 4.528888166532852e-05), (2, 0.00028701595147140324), (3, 4.784829434356652e-05), (4, 3.320075120427646e-05), (5, 6.043900066288188e-05), (6, 0.00010432196722831577), (7, 0.0002096240787068382), (8, 0.000328806578181684), (9, 0.0004270798817742616), (10, 0.0005308853578753769), (11, 0.0006894873804412782), (12, 0.0008610545191913843), (13, 0.0068576764315366745), (14, 0.008032066747546196), (15, 0.003996231593191624), (16, 0.08113419264554977), (17, 0.0018847069004550576), (18, 0.0026306267827749252), (19, 0.009832944720983505), (20, 0.009571240283548832), (21, 0.005918570794165134), (22, 0.0022939303889870644), (23, 0.0021103299222886562), (24, 0.004388125147670507), (25, 0.00845742505043745), (26, 0.003961196169257164), (27, 0.03859153389930725)], Temporal MSE: [(0, 2.623397676870809e-06), (1, 3.1359300010080915e-06), (2, 3.6000476484332466e-06), (3, 1.5314477423089556e-05), (4, 1.3405559002421796e-05), (5, 2.818566281348467e-05), (6, 6.565378134837374e-05), (7, 5.04399613419082e-05), (8, 4.7494999307673424e-05), (9, 7.820019527571276e-05), (10, 0.0001475624885642901), (11, 0.00033733085729181767), (12, 0.0002663738850969821), (13, 0.00038360327016562223), (14, 0.00044924355461262167), (15, 0.0006692688330076635), (16, 0.0008167336345650256), (17, 0.0007620282704010606), (18, 0.0009784669382497668), (19, 0.0022248891182243824), (20, 0.0035814682487398386), (21, 0.002495048800483346), (22, 0.0007143437396734953), (23, 0.0010899920016527176), (24, 0.0023311034310609102), (25, 0.005317618139088154), (26, 0.0006948469090275466), (27, 0.00010283065785188228)]
+
+ 27%|██▋       | 40/150 [09:18<21:19, 11.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 654, Spatial MSE: [(0, 2.609541297715623e-06), (1, 2.106374449795112e-05), (2, 0.00011263720807619393), (3, 4.30705476901494e-05), (4, 7.1208451117854565e-06), (5, 4.7311682465078775e-06), (6, 3.5423403460299596e-05), (7, 7.6226715464144945e-06), (8, 5.268576387607027e-06), (9, 7.447156804119004e-06), (10, 2.440717798890546e-05), (11, 0.0001524836989119649), (12, 0.00026978825917467475), (13, 0.005425130482763052), (14, 0.005042918957769871), (15, 0.001299321767874062), (16, 0.04531899467110634), (17, 0.00017540152475703508), (18, 0.00024845227017067373), (19, 0.003549686400219798), (20, 0.0018678157357499003), (21, 0.00030727163539268076), (22, 0.0006540943286381662), (23, 0.00033126084599643946), (24, 0.000617938581854105), (25, 0.0012878356501460075), (26, 0.0015378581592813134), (27, 0.04349800571799278)], Temporal MSE: [(0, 2.937444151029922e-06), (1, 1.1461895610409556e-06), (2, 1.3469399391397019e-06), (3, 1.2691592701230547e-06), (4, 7.232480925267737e-07), (5, 1.681733465375146e-06), (6, 1.2402537095113075e-06), (7, 1.9800406789727276e-06), (8, 8.802300612842373e-07), (9, 1.5796485968166962e-06), (10, 9.412349754711613e-06), (11, 0.0002597799466457218), (12, 0.00012274752953089774), (13, 0.00014934313367120922), (14, 5.805767068522982e-05), (15, 7.864394865464419e-05), (16, 4.694110248237848e-05), (17, 3.927758734789677e-05), (18, 8.509727194905281e-05), (19, 6.54336909065023e-05), (20, 5.163262540008873e-05), (21, 8.587069169152528e-05), (22, 0.00012572409468702972), (23, 0.0002761837386060506), (24, 0.0003185506211593747), (25, 0.0005320789059624076), (26, 0.00011427128629293293), (27, 0.0001001752752927132)]
+
+ 27%|██▋       | 41/150 [09:27<19:41, 10.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 648, Spatial MSE: [(0, 2.5137671855191e-06), (1, 2.4060787836788222e-05), (2, 0.00018305187404621392), (3, 2.4109878722811118e-05), (4, 7.872369678807445e-06), (5, 8.556740795029327e-06), (6, 1.420433272869559e-05), (7, 1.6600364688201807e-05), (8, 1.8959384760819376e-05), (9, 4.384648855193518e-05), (10, 8.26783652883023e-05), (11, 0.00021137126896064728), (12, 0.0003123003989458084), (13, 0.005475142505019903), (14, 0.0050995261408388615), (15, 0.0013402046170085669), (16, 0.04560936987400055), (17, 0.00020587819744832814), (18, 0.00032275475678034127), (19, 0.0035685414914041758), (20, 0.0019585962872952223), (21, 0.0003508946974761784), (22, 0.0007530168513767421), (23, 0.00037514325231313705), (24, 0.0009247068664990366), (25, 0.0018105480121448636), (26, 0.0018573794513940811), (27, 0.03979711979627609)], Temporal MSE: [(0, 1.9362357761565363e-06), (1, 1.0156524012927548e-06), (2, 1.941104983416153e-06), (3, 1.903672796288447e-06), (4, 1.861329792518518e-06), (5, 2.5380979877809295e-06), (6, 3.2110194752021926e-06), (7, 3.665212034320575e-06), (8, 3.521385906424257e-06), (9, 4.42664304500795e-06), (10, 1.3921245226811152e-05), (11, 0.00026356568560004234), (12, 0.00012619663903024048), (13, 0.00015682895900681615), (14, 6.88678992446512e-05), (15, 8.691601397003978e-05), (16, 6.135739386081696e-05), (17, 5.0786016799975187e-05), (18, 0.00010259554255753756), (19, 8.436752250418067e-05), (20, 9.449425851926208e-05), (21, 0.00012204911763546988), (22, 0.00015447968326043338), (23, 0.00032094691414386034), (24, 0.00047442151117138565), (25, 0.0007526209228672087), (26, 0.00014699678285978734), (27, 0.00010053874575532973)]
+
+ 28%|██▊       | 42/150 [09:37<18:46, 10.43s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 642, Spatial MSE: [(0, 2.9204454676801106e-06), (1, 1.7285532521782443e-05), (2, 0.0001101699672290124), (3, 3.6818586522713304e-05), (4, 6.499519713543123e-06), (5, 3.876222308463184e-06), (6, 3.231137088732794e-05), (7, 8.148790584527887e-06), (8, 4.803730917046778e-06), (9, 5.969155608909205e-06), (10, 1.0856293556571472e-05), (11, 5.931340274401009e-05), (12, 0.00018385070143267512), (13, 0.004584401845932007), (14, 0.0038151699118316174), (15, 0.0009939592564478517), (16, 0.031232014298439026), (17, 0.00010178191587328911), (18, 0.0001805866922950372), (19, 0.002353580668568611), (20, 0.0011182528687641025), (21, 0.0002415531489532441), (22, 0.0004926514811813831), (23, 0.0002692838024813682), (24, 0.0004983304534107447), (25, 0.0010430540423840284), (26, 0.0017091408371925354), (27, 0.03797898069024086)], Temporal MSE: [(0, 3.0450435133388964e-06), (1, 1.0519580655454774e-06), (2, 1.2456091553758597e-06), (3, 1.3250423762656283e-06), (4, 7.218052928692487e-07), (5, 1.5995392459444702e-06), (6, 1.1474048733361997e-06), (7, 1.825893264140177e-06), (8, 8.138815701386193e-07), (9, 1.3084433021504083e-06), (10, 3.328178991068853e-06), (11, 0.00013471678539644927), (12, 9.657262125983834e-05), (13, 0.00012424710439518094), (14, 4.91553291794844e-05), (15, 4.037929102196358e-05), (16, 3.5449600545689464e-05), (17, 2.3917402359074913e-05), (18, 5.7499342801747844e-05), (19, 4.3511168769327924e-05), (20, 3.2564206776442006e-05), (21, 5.1211576646892354e-05), (22, 8.345568494405597e-05), (23, 0.00020446744747459888), (24, 0.00026045500999316573), (25, 0.0004476624308153987), (26, 0.00010163443221244961), (27, 7.265737804118544e-05)]
+
+ 29%|██▊       | 43/150 [09:46<17:47,  9.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 636, Spatial MSE: [(0, 2.3880199933046242e-06), (1, 3.3105287002399564e-05), (2, 0.00021562146139331162), (3, 3.482708416413516e-05), (4, 3.089911479037255e-05), (5, 5.718597094528377e-05), (6, 0.00011845473636640236), (7, 0.00026783309294842184), (8, 0.000477240770123899), (9, 0.000730771804228425), (10, 0.0012792899506166577), (11, 0.001520809018984437), (12, 0.0016689262120053172), (13, 0.006335126236081123), (14, 0.007828948087990284), (15, 0.006335797253996134), (16, 0.03755982220172882), (17, 0.005090172868221998), (18, 0.005560101009905338), (19, 0.006388773676007986), (20, 0.007521806284785271), (21, 0.005550137721002102), (22, 0.003974047489464283), (23, 0.004175274632871151), (24, 0.007490219082683325), (25, 0.014558009803295135), (26, 0.006781844422221184), (27, 0.03134848549962044)], Temporal MSE: [(0, 2.4625971946079517e-06), (1, 2.3332493128691567e-06), (2, 3.3315961900370894e-06), (3, 1.9390121451579034e-05), (4, 1.3171208593121264e-05), (5, 4.2167197534581646e-05), (6, 0.0001281585282413289), (7, 9.389443584950641e-05), (8, 8.005458948900923e-05), (9, 0.00015576039731968194), (10, 0.0003224200336262584), (11, 0.0005925782024860382), (12, 0.0005030725733377039), (13, 0.0009788182796910405), (14, 0.001375230960547924), (15, 0.0021108954679220915), (16, 0.0024882315192371607), (17, 0.001975217368453741), (18, 0.0016249337932094932), (19, 0.00275306380353868), (20, 0.004420767072588205), (21, 0.0032059475779533386), (22, 0.0011914765927940607), (23, 0.0018081775633618236), (24, 0.004367760382592678), (25, 0.008692611940205097), (26, 0.0011353663867339492), (27, 9.063395555131137e-05)]
+
+ 29%|██▉       | 44/150 [09:57<18:22, 10.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 630, Spatial MSE: [(0, 2.9681093565159244e-06), (1, 1.5475996406166814e-05), (2, 8.46111070131883e-05), (3, 3.399920751689933e-05), (4, 5.721520665247226e-06), (5, 3.6050862490810687e-06), (6, 3.527395165292546e-05), (7, 8.6743038991699e-06), (8, 4.5047236199025065e-06), (9, 5.520560989680234e-06), (10, 8.390910807065666e-06), (11, 4.257959881215356e-05), (12, 0.00014122104039415717), (13, 0.003926887642592192), (14, 0.0031173794995993376), (15, 0.0007691331556998193), (16, 0.021020106971263885), (17, 6.793780630687252e-05), (18, 0.00011350015120115131), (19, 0.0015646660467609763), (20, 0.0006621092325076461), (21, 0.00016414493438787758), (22, 0.0003262355749029666), (23, 0.00018519193690735847), (24, 0.0003512165567371994), (25, 0.0007253300864249468), (26, 0.00129595841281116), (27, 0.029214413836598396)], Temporal MSE: [(0, 2.8257259145902935e-06), (1, 8.45557394768548e-07), (2, 9.617560863262042e-07), (3, 1.121093532674422e-06), (4, 6.260096938603965e-07), (5, 1.3614911722470424e-06), (6, 1.0933234761978383e-06), (7, 2.4310093067470007e-06), (8, 8.051197823988332e-07), (9, 1.1946350468861056e-06), (10, 2.560007487772964e-06), (11, 9.240395593224093e-05), (12, 7.773344259476289e-05), (13, 0.00010216992086498067), (14, 3.6864741559838876e-05), (15, 2.9817625545547344e-05), (16, 2.919323742389679e-05), (17, 1.5692212400608696e-05), (18, 3.9461989217670634e-05), (19, 3.020225267391652e-05), (20, 2.187598147429526e-05), (21, 3.7733581848442554e-05), (22, 5.4210984671954066e-05), (23, 0.00014567359175998718), (24, 0.0002011013712035492), (25, 0.00041024756501428783), (26, 8.175153925549239e-05), (27, 5.6462933571310714e-05)]
+
+ 30%|███       | 45/150 [10:08<18:39, 10.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 624, Spatial MSE: [(0, 2.253085085612838e-06), (1, 1.8404720321996137e-05), (2, 0.00015649925626348704), (3, 1.9959114069934003e-05), (4, 7.522824489569757e-06), (5, 8.742006684769876e-06), (6, 1.9021368643734604e-05), (7, 2.5744397134985775e-05), (8, 2.989004315168131e-05), (9, 5.808345304103568e-05), (10, 9.596353629603982e-05), (11, 0.0001408373937010765), (12, 0.00021512147213798016), (13, 0.004046499263495207), (14, 0.003237425582483411), (15, 0.0008314804290421307), (16, 0.021098129451274872), (17, 0.00010191724868491292), (18, 0.00020391587167978287), (19, 0.0016093746526166797), (20, 0.0007227390306070447), (21, 0.00022448219533544034), (22, 0.00042715726885944605), (23, 0.0002621135499794036), (24, 0.00069103465648368), (25, 0.0012915843399241567), (26, 0.0019008826930075884), (27, 0.026686329394578934)], Temporal MSE: [(0, 1.6875968640306382e-06), (1, 8.096465080598136e-07), (2, 1.8764795868264628e-06), (3, 1.9630001588666346e-06), (4, 1.9046411807721597e-06), (5, 2.5788729089981643e-06), (6, 3.8610687624895945e-06), (7, 4.250371603120584e-06), (8, 4.023707788292086e-06), (9, 5.496417998074321e-06), (10, 1.0208933417743538e-05), (11, 0.00010167528671445325), (12, 8.558109402656555e-05), (13, 0.00011296702723484486), (14, 5.4944677685853094e-05), (15, 4.278817505110055e-05), (16, 4.4088345021009445e-05), (17, 2.8516640668385662e-05), (18, 5.711332050850615e-05), (19, 5.022149707656354e-05), (20, 6.333282362902537e-05), (21, 6.926091737113893e-05), (22, 8.447020081803203e-05), (23, 0.00019385841733310372), (24, 0.00037682452239096165), (25, 0.000595416291616857), (26, 0.00011448496661614627), (27, 5.493386561283842e-05)]
+
+ 31%|███       | 46/150 [10:21<19:18, 11.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 618, Spatial MSE: [(0, 1.6946401046880055e-06), (1, 1.2352119483693969e-05), (2, 6.140123150544241e-05), (3, 2.0626875993912108e-05), (4, 3.3479791454738006e-06), (5, 2.399642653472256e-06), (6, 1.648171564738732e-05), (7, 4.148937478021253e-06), (8, 2.6735540359368315e-06), (9, 3.273692755101365e-06), (10, 5.6818066695996094e-06), (11, 3.0466770112980157e-05), (12, 0.0001032537838909775), (13, 0.002910083858296275), (14, 0.002352441195398569), (15, 0.0005872852052561939), (16, 0.015576145611703396), (17, 5.252123810350895e-05), (18, 7.004974031588063e-05), (19, 0.0011682340409606695), (20, 0.0004951349692419171), (21, 0.0001241860445588827), (22, 0.00026328442618250847), (23, 0.0001517289929324761), (24, 0.00027897677500732243), (25, 0.0005893106572329998), (26, 0.0009518533479422331), (27, 0.021146826446056366)], Temporal MSE: [(0, 1.783870061444759e-06), (1, 6.418512725758774e-07), (2, 8.258899697466404e-07), (3, 7.712289402661554e-07), (4, 3.9294729958783137e-07), (5, 8.35189780445944e-07), (6, 7.048035399748187e-07), (7, 1.0415391216156422e-06), (8, 4.976104719389696e-07), (9, 7.619483994858456e-07), (10, 1.5225866718537873e-06), (11, 6.573093560291454e-05), (12, 5.693588900612667e-05), (13, 7.637507223989815e-05), (14, 2.8147986085969023e-05), (15, 2.2117043045000173e-05), (16, 2.0203591702738777e-05), (17, 1.1623075806710403e-05), (18, 3.131050107185729e-05), (19, 2.30475179705536e-05), (20, 1.732732380332891e-05), (21, 2.647405563038774e-05), (22, 4.050084316986613e-05), (23, 0.00011141923459945247), (24, 0.00015276808699127287), (25, 0.000261014502029866), (26, 5.994121602270752e-05), (27, 4.166960570728406e-05)]
+
+ 31%|███▏      | 47/150 [10:30<18:11, 10.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 612, Spatial MSE: [(0, 1.4306370985650574e-06), (1, 2.5740924684214406e-05), (2, 0.0001300573057960719), (3, 2.8604385079233907e-05), (4, 3.0143724870868027e-05), (5, 4.990334127796814e-05), (6, 8.620115113444626e-05), (7, 0.00018266242113895714), (8, 0.0002821122470777482), (9, 0.00041021459037438035), (10, 0.0006096937577240169), (11, 0.0007319606374949217), (12, 0.0009581929771229625), (13, 0.004188227001577616), (14, 0.004962926264852285), (15, 0.003191637806594372), (16, 0.03556875139474869), (17, 0.002067276742309332), (18, 0.0019380453741177917), (19, 0.00381335592828691), (20, 0.0032489211298525333), (21, 0.0025761898141354322), (22, 0.002732124412432313), (23, 0.0025875463616102934), (24, 0.0049316962249577045), (25, 0.00975875649601221), (26, 0.0053394692949950695), (27, 0.019949404522776604)], Temporal MSE: [(0, 1.4778956938243937e-06), (1, 1.4928172049621935e-06), (2, 2.34692220146826e-06), (3, 3.0342127502080984e-05), (4, 1.5141892617975827e-05), (5, 3.153452053084038e-05), (6, 7.911113789305091e-05), (7, 5.869390952284448e-05), (8, 5.155723920324817e-05), (9, 0.00010264167212881148), (10, 0.00020128751930315048), (11, 0.000375407631509006), (12, 0.00030385126592591405), (13, 0.0005267494125291705), (14, 0.0007589948945678771), (15, 0.0009605080122128129), (16, 0.0011262207990512252), (17, 0.0008850903832353652), (18, 0.0009523921180516481), (19, 0.0011961683630943298), (20, 0.002327471738681197), (21, 0.0014999848790466785), (22, 0.0007537577184848487), (23, 0.0012584784999489784), (24, 0.00290480675175786), (25, 0.0072295633144676685), (26, 0.0010848355013877153), (27, 5.8122703194385394e-05)]
+
+ 32%|███▏      | 48/150 [10:41<18:15, 10.74s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 606, Spatial MSE: [(0, 2.196636160078924e-06), (1, 1.2168363355158363e-05), (2, 6.677991041215137e-05), (3, 2.452497028571088e-05), (4, 3.991649919043994e-06), (5, 2.6208144845440984e-06), (6, 2.1770976672996767e-05), (7, 5.961861916148337e-06), (8, 3.251856242059148e-06), (9, 3.861760887957644e-06), (10, 5.8929508668370545e-06), (11, 2.889661118388176e-05), (12, 9.652635344536975e-05), (13, 0.0026821864303201437), (14, 0.002177275251597166), (15, 0.0006350827752612531), (16, 0.018362820148468018), (17, 5.613479152088985e-05), (18, 7.916204776847735e-05), (19, 0.0012374153593555093), (20, 0.000508224475197494), (21, 0.00013309725909493864), (22, 0.00026083269040100276), (23, 0.00015311327297240496), (24, 0.0002728143590502441), (25, 0.0005790586583316326), (26, 0.0010651755146682262), (27, 0.022307012230157852)], Temporal MSE: [(0, 1.9928422716475325e-06), (1, 6.484575010290428e-07), (2, 8.426522413174098e-07), (3, 9.356365922030818e-07), (4, 4.802967055184126e-07), (5, 1.0284273912475328e-06), (6, 8.487504601362161e-07), (7, 1.605267243576236e-06), (8, 5.951190473751922e-07), (9, 8.911883924156427e-07), (10, 1.7321526684099808e-06), (11, 6.071152529329993e-05), (12, 5.2987652452429757e-05), (13, 6.956842844374478e-05), (14, 3.05664143525064e-05), (15, 2.4777369617368095e-05), (16, 2.0896557543892413e-05), (17, 1.2336827239778358e-05), (18, 3.314359855721705e-05), (19, 2.5200541131198406e-05), (20, 1.7706022845231928e-05), (21, 2.8612841560970992e-05), (22, 4.3887252104468644e-05), (23, 0.00011794643069151789), (24, 0.00015998719027265906), (25, 0.00034391364897601306), (26, 6.778728129575029e-05), (27, 4.5689372200286016e-05)]
+
+ 33%|███▎      | 49/150 [10:53<18:29, 10.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 600, Spatial MSE: [(0, 2.056129005723051e-06), (1, 1.840559343690984e-05), (2, 0.00012727525609079748), (3, 2.0910943931085058e-05), (4, 6.92544381308835e-06), (5, 8.684998647368047e-06), (6, 1.8060025468003005e-05), (7, 1.569823689351324e-05), (8, 1.861838245531544e-05), (9, 4.62336975033395e-05), (10, 7.940342038637027e-05), (11, 0.00010864878277061507), (12, 0.00016643574053887278), (13, 0.00307366787455976), (14, 0.0030096652917563915), (15, 0.0011098321992903948), (16, 0.033405598253011703), (17, 0.00011590513895498589), (18, 0.0001872484281193465), (19, 0.002147143008187413), (20, 0.0012339173117652535), (21, 0.00021005820599384606), (22, 0.0003673333558253944), (23, 0.0002572322264313698), (24, 0.0006515864515677094), (25, 0.001208499539643526), (26, 0.002093968912959099), (27, 0.020047402009367943)], Temporal MSE: [(0, 1.4767132370252511e-06), (1, 7.761768756608944e-07), (2, 1.5366389334303676e-06), (3, 1.907391151689808e-06), (4, 1.834158069868863e-06), (5, 2.366786247876007e-06), (6, 3.142540208500577e-06), (7, 3.632907464634627e-06), (8, 3.4658796721487306e-06), (9, 4.224678377795499e-06), (10, 7.668326361454092e-06), (11, 7.064634701237082e-05), (12, 6.54381510685198e-05), (13, 8.313065336551517e-05), (14, 5.130008503329009e-05), (15, 4.558010914479382e-05), (16, 3.932342951884493e-05), (17, 3.0410566978389397e-05), (18, 5.6020111514953896e-05), (19, 5.0934690079884604e-05), (20, 6.610448326682672e-05), (21, 6.900628068251535e-05), (22, 9.202931687468663e-05), (23, 0.00019083118240814656), (24, 0.00035528463195078075), (25, 0.0006047314964234829), (26, 0.00011187975906068459), (27, 4.564557093544863e-05)]
+
+ 33%|███▎      | 50/150 [11:02<17:44, 10.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 594, Spatial MSE: [(0, 1.0219202977168607e-06), (1, 9.582668099028524e-06), (2, 5.170743679627776e-05), (3, 1.6525636965525337e-05), (4, 2.8682575248240028e-06), (5, 1.9091228296019835e-06), (6, 1.2406751920934767e-05), (7, 4.008471023553284e-06), (8, 2.725022113736486e-06), (9, 3.691196980071254e-06), (10, 1.376294039800996e-05), (11, 0.0001169623647001572), (12, 0.0001782035396900028), (13, 0.003483580192551017), (14, 0.0025323498994112015), (15, 0.0005980361602269113), (16, 0.015680192038416862), (17, 5.1705566875170916e-05), (18, 7.049994746921584e-05), (19, 0.001219027442857623), (20, 0.0005522067076526582), (21, 0.00010143070539925247), (22, 0.00022944605734664947), (23, 0.00013386488717515022), (24, 0.0002749113191384822), (25, 0.0005466494476422668), (26, 0.000862156564835459), (27, 0.02159658819437027)], Temporal MSE: [(0, 1.1055147979277535e-06), (1, 4.7116083123910357e-07), (2, 6.724895342813397e-07), (3, 7.288759888979257e-07), (4, 3.875150298426888e-07), (5, 7.950943654577713e-07), (6, 7.14942700597021e-07), (7, 1.0247059663015534e-06), (8, 4.558427519896213e-07), (9, 7.815654612386425e-07), (10, 5.201294698053971e-06), (11, 0.00019365329353604466), (12, 7.450519478879869e-05), (13, 7.604136771988124e-05), (14, 2.5172650566673838e-05), (15, 2.712815512495581e-05), (16, 1.8955941413878463e-05), (17, 1.2269208127690945e-05), (18, 3.125534931314178e-05), (19, 2.50780358328484e-05), (20, 1.912713560159318e-05), (21, 2.360833423153963e-05), (22, 3.826412284979597e-05), (23, 0.00010466574167367071), (24, 0.00014513071801047772), (25, 0.00028234091587364674), (26, 6.085191125748679e-05), (27, 4.198940587230027e-05)]
+
+ 34%|███▍      | 51/150 [11:11<16:41, 10.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 588, Spatial MSE: [(0, 1.3875118156647659e-06), (1, 2.6467314455658197e-05), (2, 0.0001381805632263422), (3, 3.461333108134568e-05), (4, 2.673862036317587e-05), (5, 4.346685091149993e-05), (6, 0.0001037476904457435), (7, 0.00020947899611201137), (8, 0.0002885030990000814), (9, 0.0004274774400983006), (10, 0.0006337034865282476), (11, 0.0007670149207115173), (12, 0.0008656939608044922), (13, 0.0045363097451627254), (14, 0.004098306875675917), (15, 0.0019534307066351175), (16, 0.02263963408768177), (17, 0.0010006629163399339), (18, 0.0011158895213156939), (19, 0.002674073912203312), (20, 0.002102164551615715), (21, 0.0011104459408670664), (22, 0.0016204116400331259), (23, 0.0014947995077818632), (24, 0.0029318237211555243), (25, 0.005742309149354696), (26, 0.0033918959088623524), (27, 0.022761106491088867)], Temporal MSE: [(0, 1.833005285334366e-06), (1, 1.415250267200463e-06), (2, 2.4384219159401255e-06), (3, 1.9866616639774293e-05), (4, 1.158796476374846e-05), (5, 3.024106626980938e-05), (6, 8.138307748595253e-05), (7, 6.1351056501735e-05), (8, 4.6285415010061115e-05), (9, 8.53308811201714e-05), (10, 0.0001690871868049726), (11, 0.00040675135096535087), (12, 0.00022955842723604292), (13, 0.00033932708902284503), (14, 0.0004426505765877664), (15, 0.0004062566440552473), (16, 0.0004849807301070541), (17, 0.00038825118099339306), (18, 0.00041556009091436863), (19, 0.0004835131112486124), (20, 0.0008339978521689773), (21, 0.0005702166236005723), (22, 0.0004285427858121693), (23, 0.0007149905432015657), (24, 0.001724396599456668), (25, 0.003925425466150045), (26, 0.0005873203626833856), (27, 6.158065662020817e-05)]
+
+ 35%|███▍      | 52/150 [11:22<16:33, 10.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 582, Spatial MSE: [(0, 1.556107918077032e-06), (1, 1.1043280210287776e-05), (2, 5.970126949250698e-05), (3, 1.6113581295940094e-05), (4, 3.4896047509391792e-06), (5, 2.0789691461686743e-06), (6, 1.3043719263805542e-05), (7, 3.994659891759511e-06), (8, 2.7409785161580658e-06), (9, 3.6959525004931493e-06), (10, 5.346710622688988e-06), (11, 3.450652729952708e-05), (12, 0.00011515618825796992), (13, 0.0032130314502865076), (14, 0.002366079716011882), (15, 0.0005851562600582838), (16, 0.01825794391334057), (17, 7.166060822783038e-05), (18, 8.194370457204059e-05), (19, 0.001269449247047305), (20, 0.0006218626513145864), (21, 0.00011280104808975011), (22, 0.0002547043259255588), (23, 0.00014923785056453198), (24, 0.0002788979618344456), (25, 0.0005674405256286263), (26, 0.0009287740685977042), (27, 0.021645663306117058)], Temporal MSE: [(0, 1.5849215060370625e-06), (1, 5.986386781842157e-07), (2, 8.124594614855596e-07), (3, 9.374691671837354e-07), (4, 4.466004668302048e-07), (5, 8.484957447763009e-07), (6, 7.388690050902369e-07), (7, 1.1376426982678822e-06), (8, 5.024375582252105e-07), (9, 7.509779038628039e-07), (10, 1.7108141037169844e-06), (11, 7.368473598035052e-05), (12, 6.23813975835219e-05), (13, 7.290418579941615e-05), (14, 2.5570261641405523e-05), (15, 2.487132587702945e-05), (16, 1.9028031601919793e-05), (17, 1.6309499187627807e-05), (18, 3.8955186028033495e-05), (19, 3.2581359846517444e-05), (20, 2.4988394216052257e-05), (21, 3.202070729457773e-05), (22, 4.7204764996422455e-05), (23, 0.00011417224595788866), (24, 0.00014985880989115685), (25, 0.0003405602474231273), (26, 6.552522972924635e-05), (27, 4.7025609092088416e-05)]
+
+ 35%|███▌      | 53/150 [11:35<18:05, 11.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 576, Spatial MSE: [(0, 1.444921622351103e-06), (1, 1.3080246390018146e-05), (2, 8.083917055046186e-05), (3, 1.5623898434569128e-05), (4, 5.707982381863985e-06), (5, 8.169656211975962e-06), (6, 1.4955184269638266e-05), (7, 1.4907266631780658e-05), (8, 1.5075032024469692e-05), (9, 3.40333572239615e-05), (10, 6.579871842404827e-05), (11, 9.877480624709278e-05), (12, 0.00017616330296732485), (13, 0.0032970586325973272), (14, 0.0024729648139327765), (15, 0.0006264032563194633), (16, 0.018002936616539955), (17, 9.44303028518334e-05), (18, 0.00011570189235499129), (19, 0.0012767391745001078), (20, 0.0006537275039590895), (21, 0.00013896900054533035), (22, 0.0003107893862761557), (23, 0.00019937432080041617), (24, 0.0005208550137467682), (25, 0.0009981145849451423), (26, 0.0020106262527406216), (27, 0.020115546882152557)], Temporal MSE: [(0, 1.0061836519525968e-06), (1, 5.011773964724853e-07), (2, 1.132926399805001e-06), (3, 1.5287477026504348e-06), (4, 1.5992229691619286e-06), (5, 2.035712668657652e-06), (6, 2.7018243144993903e-06), (7, 2.8659223971772008e-06), (8, 2.818620714606368e-06), (9, 3.425672275625402e-06), (10, 6.413724804588128e-06), (11, 7.973106403369457e-05), (12, 6.792930071242154e-05), (13, 8.054913632804528e-05), (14, 3.786505476455204e-05), (15, 3.3804113627411425e-05), (16, 2.9714072297792882e-05), (17, 2.598835271783173e-05), (18, 5.126031828694977e-05), (19, 4.81302740809042e-05), (20, 5.6480606872355565e-05), (21, 5.382352901506238e-05), (22, 6.984245555941015e-05), (23, 0.00015414218069054186), (24, 0.0002882298140320927), (25, 0.0004891508724540472), (26, 0.00010513246525079012), (27, 4.777215144713409e-05)]
+
+ 36%|███▌      | 54/150 [11:45<17:03, 10.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 570, Spatial MSE: [(0, 1.0472892881807638e-06), (1, 9.259229955205228e-06), (2, 4.5072090870235115e-05), (3, 1.5074347174959257e-05), (4, 3.2029213343776064e-06), (5, 1.70880991845479e-06), (6, 1.1902693586307578e-05), (7, 3.1821750781091396e-06), (8, 2.2201988940651063e-06), (9, 3.1283771022572182e-06), (10, 4.033090135635575e-06), (11, 2.0817869881284423e-05), (12, 8.118434197967872e-05), (13, 0.002794155152514577), (14, 0.0021831211633980274), (15, 0.0005864484119229019), (16, 0.019025301560759544), (17, 6.519562884932384e-05), (18, 8.202493336284533e-05), (19, 0.0012644600355997682), (20, 0.0006166249513626099), (21, 0.00014976694365032017), (22, 0.0002642104809638113), (23, 0.00013935437891632318), (24, 0.00027200207114219666), (25, 0.0005521678831428289), (26, 0.0008280397159978747), (27, 0.0204780250787735)], Temporal MSE: [(0, 1.1447366432548733e-06), (1, 4.7156260052361176e-07), (2, 6.154530183266615e-07), (3, 7.669576689295354e-07), (4, 3.7998654534021625e-07), (5, 7.207145245047286e-07), (6, 6.11611199019535e-07), (7, 8.885949114301184e-07), (8, 4.3254465253994567e-07), (9, 6.585144092241535e-07), (10, 1.167396249002195e-06), (11, 4.0610590076539665e-05), (12, 4.976754280505702e-05), (13, 6.869547360111028e-05), (14, 2.5019940949277952e-05), (15, 2.5404629923286848e-05), (16, 1.926459117385093e-05), (17, 1.6821199096739292e-05), (18, 3.955783904530108e-05), (19, 3.40080659952946e-05), (20, 2.698286516533699e-05), (21, 3.260130688431673e-05), (22, 4.793086918652989e-05), (23, 0.00011633982649073005), (24, 0.00015061198791954666), (25, 0.00030315894400700927), (26, 6.231545557966456e-05), (27, 4.614345380105078e-05)]
+
+ 37%|███▋      | 55/150 [11:56<17:22, 10.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 564, Spatial MSE: [(0, 1.1647512110357638e-06), (1, 2.0632322048186325e-05), (2, 0.00010166235733777285), (3, 2.6846289983950555e-05), (4, 2.6562396669760346e-05), (5, 4.3807078327517956e-05), (6, 0.00010061792272608727), (7, 0.00021696100884582847), (8, 0.0002489618200343102), (9, 0.00033896247623488307), (10, 0.00048491364577785134), (11, 0.0005118323024362326), (12, 0.0006164007936604321), (13, 0.003841122379526496), (14, 0.0043471320532262325), (15, 0.002123583108186722), (16, 0.04043320566415787), (17, 0.0007740131695754826), (18, 0.0008238035952672362), (19, 0.0030601685866713524), (20, 0.002245038514956832), (21, 0.0008732557762414217), (22, 0.0012286100536584854), (23, 0.0011374970199540257), (24, 0.002342445310205221), (25, 0.004748526960611343), (26, 0.003148973686620593), (27, 0.019261809065937996)], Temporal MSE: [(0, 1.4827995755695156e-06), (1, 1.0383707831351785e-06), (2, 2.2985752821114147e-06), (3, 1.2241335753060412e-05), (4, 1.2661353139264975e-05), (5, 2.8415279302862473e-05), (6, 9.208887786371633e-05), (7, 5.903303099330515e-05), (8, 4.356565113994293e-05), (9, 7.605931023135781e-05), (10, 0.00014413153985515237), (11, 0.00020967524324078113), (12, 0.00017710227984935045), (13, 0.0002800129586830735), (14, 0.0003593640576582402), (15, 0.00031263777054846287), (16, 0.0003387113392818719), (17, 0.00028691027546301484), (18, 0.0003143433714285493), (19, 0.00039001397090032697), (20, 0.0005596614209935069), (21, 0.0004229032783769071), (22, 0.0003527615626808256), (23, 0.0005599859287030995), (24, 0.0013342829188331962), (25, 0.0028512149583548307), (26, 0.0004195994697511196), (27, 5.8403031289344653e-05)]
+
+ 37%|███▋      | 56/150 [12:08<17:42, 11.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 558, Spatial MSE: [(0, 1.3510059488908155e-06), (1, 1.0432276212668512e-05), (2, 4.949870708514936e-05), (3, 1.4494157767330762e-05), (4, 3.0864619020576356e-06), (5, 1.8062879689750844e-06), (6, 1.081390928447945e-05), (7, 3.7921663533779792e-06), (8, 2.2960614387557143e-06), (9, 2.95232234748255e-06), (10, 3.671725835374673e-06), (11, 1.4781909158045892e-05), (12, 5.708312164642848e-05), (13, 0.002103106351569295), (14, 0.00180533854290843), (15, 0.0005723979556933045), (16, 0.016534054651856422), (17, 4.897132748737931e-05), (18, 6.445516919484362e-05), (19, 0.001093969913199544), (20, 0.00045209581730887294), (21, 0.00011929643369512632), (22, 0.00021961696620564908), (23, 0.00012981999316252768), (24, 0.00022497661120723933), (25, 0.0004724159953184426), (26, 0.0008560711285099387), (27, 0.018568838015198708)], Temporal MSE: [(0, 1.292258048124495e-06), (1, 4.924380050397303e-07), (2, 6.702599080199434e-07), (3, 8.199109515771852e-07), (4, 4.042211401156237e-07), (5, 7.331361757678678e-07), (6, 6.607980935768865e-07), (7, 1.0283779374731239e-06), (8, 4.4623618578043533e-07), (9, 6.6006890619974e-07), (10, 1.148221485891554e-06), (11, 2.5398501747986302e-05), (12, 3.5504552215570584e-05), (13, 5.877572039025836e-05), (14, 2.3624848836334422e-05), (15, 2.361209408263676e-05), (16, 1.618010355741717e-05), (17, 1.1447895303717814e-05), (18, 3.28861060552299e-05), (19, 2.559055974415969e-05), (20, 1.749305556586478e-05), (21, 2.3947593945194967e-05), (22, 3.846711479127407e-05), (23, 0.00010208141611656174), (24, 0.00013214207137934864), (25, 0.0003075880522374064), (26, 6.198941264301538e-05), (27, 4.142106627114117e-05)]
+
+ 38%|███▊      | 57/150 [12:21<18:19, 11.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 552, Spatial MSE: [(0, 1.286160454583296e-06), (1, 1.1949328836635686e-05), (2, 6.901384040247649e-05), (3, 1.6133633835124783e-05), (4, 6.699402092635864e-06), (5, 9.292254617321305e-06), (6, 2.3877677449490875e-05), (7, 2.0969189790776e-05), (8, 2.2468437236966565e-05), (9, 4.921033905702643e-05), (10, 7.985668344190344e-05), (11, 0.00010597812797641382), (12, 0.0001393864949932322), (13, 0.0022015883587300777), (14, 0.0019233874045312405), (15, 0.0006208841223269701), (16, 0.01660112477838993), (17, 7.067023398121819e-05), (18, 0.00019663113926071674), (19, 0.0012452363735064864), (20, 0.0007090819999575615), (21, 0.00019411052926443517), (22, 0.0002768384001683444), (23, 0.0001922419760376215), (24, 0.0004749158688355237), (25, 0.0009156864834949374), (26, 0.002025904366746545), (27, 0.017397120594978333)], Temporal MSE: [(0, 1.0341772167521412e-06), (1, 5.142861709828139e-07), (2, 1.041523319145199e-06), (3, 1.7474053493060637e-06), (4, 1.8382344251222094e-06), (5, 2.279679847561056e-06), (6, 3.1903459785098676e-06), (7, 3.943010142393177e-06), (8, 3.38373160957417e-06), (9, 4.545877800410381e-06), (10, 7.687452125537675e-06), (11, 3.2990628824336454e-05), (12, 4.253466613590717e-05), (13, 6.85959093971178e-05), (14, 3.983259739470668e-05), (15, 3.422356166993268e-05), (16, 2.776140536298044e-05), (17, 1.9695326045621186e-05), (18, 4.496017572819255e-05), (19, 4.2995296098524705e-05), (20, 5.93821496295277e-05), (21, 4.690560672315769e-05), (22, 6.391720671672374e-05), (23, 0.00014401829685084522), (24, 0.0002616396814119071), (25, 0.00048673071432858706), (26, 0.00010365922207711264), (27, 4.292379162507132e-05)]
+
+ 39%|███▊      | 58/150 [12:36<19:10, 12.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 546, Spatial MSE: [(0, 1.3931228295405162e-06), (1, 9.063570360012818e-06), (2, 4.160620301263407e-05), (3, 1.2962505024916027e-05), (4, 2.657321147125913e-06), (5, 1.731661768644699e-06), (6, 1.018191505863797e-05), (7, 4.328536761022406e-06), (8, 2.2126471321826102e-06), (9, 2.8063241188647225e-06), (10, 3.3500864446978085e-06), (11, 1.2023089766444173e-05), (12, 4.347200592746958e-05), (13, 0.001452816417440772), (14, 0.00137144117616117), (15, 0.0005450818571262062), (16, 0.01789005473256111), (17, 4.1516468627378345e-05), (18, 6.336790829664096e-05), (19, 0.0010274021187797189), (20, 0.0004182035627309233), (21, 0.00011558066034922376), (22, 0.00022270118643064052), (23, 0.00012405279267113656), (24, 0.00020424291142262518), (25, 0.00046700824168510735), (26, 0.0007905943784862757), (27, 0.015624940395355225)], Temporal MSE: [(0, 1.2841805983043741e-06), (1, 4.624340022019169e-07), (2, 5.89876719914173e-07), (3, 7.214283641587826e-07), (4, 3.7142964970371395e-07), (5, 6.764124123037618e-07), (6, 6.405941235243517e-07), (7, 1.1599877325352281e-06), (8, 4.392011874188029e-07), (9, 6.400852612387098e-07), (10, 1.0866842785617337e-06), (11, 2.1641590137733147e-05), (12, 2.606576163088903e-05), (13, 4.36443297076039e-05), (14, 2.4408149329246953e-05), (15, 2.3185093596111983e-05), (16, 1.5457660992979072e-05), (17, 1.1314882613078225e-05), (18, 3.249422661610879e-05), (19, 2.5745344828465022e-05), (20, 1.6483611034345813e-05), (21, 2.4684595700819045e-05), (22, 3.8619255064986646e-05), (23, 0.00010404759086668491), (24, 0.00013291198411025107), (25, 0.0003427857009228319), (26, 5.881794641027227e-05), (27, 4.050246570841409e-05)]
+
+ 39%|███▉      | 59/150 [12:50<19:42, 12.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 540, Spatial MSE: [(0, 1.0720026466515264e-06), (1, 2.1001340428483672e-05), (2, 9.552259143674746e-05), (3, 2.3105360014596954e-05), (4, 2.5161636585835367e-05), (5, 4.361594983492978e-05), (6, 0.00015693396562710404), (7, 0.0003037104324903339), (8, 0.0002711827401071787), (9, 0.0003835607785731554), (10, 0.0005247751832939684), (11, 0.0005086068413220346), (12, 0.0005741846980527043), (13, 0.0023774977307766676), (14, 0.00300514861010015), (15, 0.0018627127865329385), (16, 0.03020704723894596), (17, 0.0008345377864316106), (18, 0.0008189041400328279), (19, 0.0023697176948189735), (20, 0.0018708523130044341), (21, 0.0009517068974673748), (22, 0.001207616413012147), (23, 0.0012722944375127554), (24, 0.0026228863280266523), (25, 0.005602272693067789), (26, 0.004687655717134476), (27, 0.01628364995121956)], Temporal MSE: [(0, 1.2526769523901748e-06), (1, 7.368162755483354e-07), (2, 2.2888939383847173e-06), (3, 2.2081487259129062e-05), (4, 1.424087804480223e-05), (5, 3.2445797842228785e-05), (6, 0.00013490315177477896), (7, 6.954840500839055e-05), (8, 5.022124241804704e-05), (9, 8.409295696765184e-05), (10, 0.00015102385077625513), (11, 0.00017795726307667792), (12, 0.00015026511391624808), (13, 0.00024821015540510416), (14, 0.00037588123814202845), (15, 0.00034599879290908575), (16, 0.0003707525902427733), (17, 0.0003249559667892754), (18, 0.000338046025717631), (19, 0.0005126767791807652), (20, 0.0007757809944450855), (21, 0.0005145617760717869), (22, 0.00038273356040008366), (23, 0.0005886639701202512), (24, 0.0014653772814199328), (25, 0.002726804930716753), (26, 0.00045000133104622364), (27, 5.429132215795107e-05)]
+
+ 40%|████      | 60/150 [13:02<19:25, 12.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 534, Spatial MSE: [(0, 7.750226700409257e-07), (1, 7.147092674131272e-06), (2, 3.121001645922661e-05), (3, 1.0826105608430225e-05), (4, 2.2624012672167737e-06), (5, 1.3498100770448218e-06), (6, 7.04591957401135e-06), (7, 3.2080133678391576e-06), (8, 2.0364141164463945e-06), (9, 3.22830078403058e-06), (10, 1.1770395758503582e-05), (11, 9.220733772963285e-05), (12, 0.00011754182924050838), (13, 0.002280321903526783), (14, 0.0016848036320880055), (15, 0.00037255408824421465), (16, 0.010043883696198463), (17, 3.256132185924798e-05), (18, 7.935347821330652e-05), (19, 0.0006950449896976352), (20, 0.00030443735886365175), (21, 5.852101094205864e-05), (22, 0.00012606134987436235), (23, 8.119114500004798e-05), (24, 0.0001752751413732767), (25, 0.00034218080691061914), (26, 0.0006281001842580736), (27, 0.014047293923795223)], Temporal MSE: [(0, 8.36146227811696e-07), (1, 3.542502042819251e-07), (2, 4.786999738826125e-07), (3, 6.157788448035717e-07), (4, 3.4311506169615313e-07), (5, 6.048994691809639e-07), (6, 6.487051678050193e-07), (7, 8.966089239947905e-07), (8, 3.9140624608080543e-07), (9, 8.288592994176724e-07), (10, 5.195898665988352e-06), (11, 0.00013740704162046313), (12, 4.515358159551397e-05), (13, 4.551972961053252e-05), (14, 1.5568346498184837e-05), (15, 1.6535273971385323e-05), (16, 1.2602949027495924e-05), (17, 7.376421763183316e-06), (18, 1.959316614374984e-05), (19, 1.4139418453851249e-05), (20, 1.0857274901354685e-05), (21, 1.597652953932993e-05), (22, 2.533112456148956e-05), (23, 7.052106957416981e-05), (24, 9.404918091604486e-05), (25, 0.00022497262398246676), (26, 4.8804868129082024e-05), (27, 2.928937283286359e-05)]
+
+ 41%|████      | 61/150 [13:13<18:09, 12.24s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 528, Spatial MSE: [(0, 9.5987763870653e-07), (1, 1.342831728834426e-05), (2, 5.498024984262884e-05), (3, 1.5556937796645798e-05), (4, 5.693348157365108e-06), (5, 7.740140063106082e-06), (6, 2.1116578864166513e-05), (7, 1.3850724826625083e-05), (8, 1.535560113552492e-05), (9, 3.7041820178274065e-05), (10, 7.353381079155952e-05), (11, 0.0001600632385816425), (12, 0.00018931340309791267), (13, 0.0023671656381338835), (14, 0.0017680329037830234), (15, 0.0004170034371782094), (16, 0.010090956464409828), (17, 6.733612099196762e-05), (18, 0.00011068709136452526), (19, 0.0007149373413994908), (20, 0.0003372031496837735), (21, 7.98804103396833e-05), (22, 0.00017136972746811807), (23, 0.00014612283848691732), (24, 0.0003859085845761001), (25, 0.0007556395139545202), (26, 0.001921787392348051), (27, 0.014227187260985374)], Temporal MSE: [(0, 9.209081781591522e-07), (1, 4.6713893198102596e-07), (2, 8.906629318516934e-07), (3, 1.6591694702583482e-06), (4, 1.7682000361674e-06), (5, 2.11673318517569e-06), (6, 2.848490112228319e-06), (7, 3.232865992686129e-06), (8, 3.1250126539816847e-06), (9, 3.937867859349353e-06), (10, 1.0566029231995344e-05), (11, 0.00014423139509744942), (12, 5.184194378671236e-05), (13, 5.467865412356332e-05), (14, 3.065189957851544e-05), (15, 2.5609630029066466e-05), (16, 2.2129652279545553e-05), (17, 1.477999467169866e-05), (18, 2.9703076506848447e-05), (19, 2.501202470739372e-05), (20, 2.9363931389525533e-05), (21, 3.0955874535720795e-05), (22, 4.602286571753211e-05), (23, 0.0001023954973788932), (24, 0.00019988056737929583), (25, 0.0003854201058857143), (26, 9.170803969027475e-05), (27, 3.193182419636287e-05)]
+
+ 41%|████▏     | 62/150 [13:24<17:25, 11.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 522, Spatial MSE: [(0, 1.469912149332231e-06), (1, 7.607830411870964e-06), (2, 3.4452881664037704e-05), (3, 1.0614875463943463e-05), (4, 2.2758456452720566e-06), (5, 1.620227521925699e-06), (6, 8.049491043493617e-06), (7, 4.078648089489434e-06), (8, 2.254361334053101e-06), (9, 3.34444939653622e-06), (10, 4.8389993025921285e-06), (11, 4.009082476841286e-05), (12, 9.410948405275121e-05), (13, 0.0022531121503561735), (14, 0.001630596467293799), (15, 0.0003675409243442118), (16, 0.009881380014121532), (17, 3.2581068808212876e-05), (18, 7.871237176004797e-05), (19, 0.0006753266789019108), (20, 0.00029298587469384074), (21, 6.735455099260435e-05), (22, 0.00013808254152536392), (23, 8.268242527265102e-05), (24, 0.00017190717335324734), (25, 0.0003554457507561892), (26, 0.0008709306130185723), (27, 0.014906317926943302)], Temporal MSE: [(0, 1.1503605037432862e-06), (1, 4.1670313066788367e-07), (2, 5.050539471085358e-07), (3, 6.682313937744766e-07), (4, 3.8023856063773565e-07), (5, 6.454516778831021e-07), (6, 6.357775532706e-07), (7, 1.11134306735039e-06), (8, 4.4266496956879564e-07), (9, 6.650190016443958e-07), (10, 1.8949656350741861e-06), (11, 8.723251085029915e-05), (12, 4.470834392122924e-05), (13, 4.294166137697175e-05), (14, 1.572893415868748e-05), (15, 1.6840323951328173e-05), (16, 1.2746259926643688e-05), (17, 7.553421710326802e-06), (18, 1.989887277886737e-05), (19, 1.559957854624372e-05), (20, 1.290396448894171e-05), (21, 1.737978345772717e-05), (22, 2.7507354388944805e-05), (23, 7.443575304932892e-05), (24, 9.61207042564638e-05), (25, 0.00030358214280568063), (26, 5.489274917636067e-05), (27, 3.2483498216606677e-05)]
+
+ 42%|████▏     | 63/150 [13:36<17:09, 11.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 516, Spatial MSE: [(0, 1.1883250863320427e-06), (1, 2.1035752070019953e-05), (2, 7.374145206995308e-05), (3, 1.9089016859652475e-05), (4, 2.212923573097214e-05), (5, 4.362989420769736e-05), (6, 0.00018919887952506542), (7, 0.00028889504028484225), (8, 0.00026920283562503755), (9, 0.00037141930079087615), (10, 0.00042165903141722083), (11, 0.0005022810655646026), (12, 0.0006036894046701491), (13, 0.0030237704049795866), (14, 0.003130230586975813), (15, 0.0020733196288347244), (16, 0.027716992422938347), (17, 0.0010964868124574423), (18, 0.001072224578820169), (19, 0.003294245107099414), (20, 0.0031906538642942905), (21, 0.0012462078593671322), (22, 0.0013461234048008919), (23, 0.0017469603335484862), (24, 0.0035590315237641335), (25, 0.006943459622561932), (26, 0.006626504007726908), (27, 0.014006542973220348)], Temporal MSE: [(0, 1.2684768080362119e-06), (1, 6.451664376072586e-07), (2, 2.2278093183558667e-06), (3, 3.065814598812722e-05), (4, 1.548025284137111e-05), (5, 2.94964720524149e-05), (6, 0.00012867027544416487), (7, 6.428340566344559e-05), (8, 4.699919372797012e-05), (9, 7.462135545210913e-05), (10, 0.00012918967695441097), (11, 0.0002373843890381977), (12, 0.00015864495071582496), (13, 0.00023466978746000677), (14, 0.0004044319794047624), (15, 0.00039406109135597944), (16, 0.00047265528701245785), (17, 0.00042760351789183915), (18, 0.0003806550521403551), (19, 0.0005926272133365273), (20, 0.001028332975693047), (21, 0.0007283526938408613), (22, 0.0004826118820346892), (23, 0.0006886387127451599), (24, 0.0019879406318068504), (25, 0.003216463839635253), (26, 0.0005622946773655713), (27, 4.465778329176828e-05)]
+
+ 43%|████▎     | 64/150 [13:46<16:22, 11.43s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 510, Spatial MSE: [(0, 9.43642419315438e-07), (1, 7.351048679993255e-06), (2, 3.279395605204627e-05), (3, 9.72686211753171e-06), (4, 2.35030029216432e-06), (5, 1.4393050378203043e-06), (6, 5.262520971882623e-06), (7, 3.2699358598620165e-06), (8, 1.869728976089391e-06), (9, 3.0976000289228978e-06), (10, 3.6367628126754425e-06), (11, 2.716652670642361e-05), (12, 7.928862032713369e-05), (13, 0.0021860687993466854), (14, 0.001606629230082035), (15, 0.0003774797369260341), (16, 0.009873304516077042), (17, 3.309582461952232e-05), (18, 4.248234472470358e-05), (19, 0.0007625705329701304), (20, 0.00036902373540215194), (21, 7.014906441327184e-05), (22, 0.00015248870477080345), (23, 8.30331991892308e-05), (24, 0.00016091026191134006), (25, 0.00033473840449005365), (26, 0.0006027159397490323), (27, 0.01375633105635643)], Temporal MSE: [(0, 9.719145737108192e-07), (1, 4.105463915493601e-07), (2, 5.488425927069329e-07), (3, 6.273114649957279e-07), (4, 3.825603016593959e-07), (5, 6.40620953618054e-07), (6, 7.846646781217714e-07), (7, 9.33560727389704e-07), (8, 4.2447774717402353e-07), (9, 5.995062792862882e-07), (10, 1.3800038232147926e-06), (11, 5.936849629506469e-05), (12, 4.187110243947245e-05), (13, 4.145879574934952e-05), (14, 1.56415317178471e-05), (15, 1.708689524093643e-05), (16, 1.2135275028413162e-05), (17, 7.488277788070263e-06), (18, 2.2215584976947866e-05), (19, 1.804818384698592e-05), (20, 1.339398295385763e-05), (21, 1.621311457711272e-05), (22, 2.683616003196221e-05), (23, 7.177625957410783e-05), (24, 9.261479863198474e-05), (25, 0.00024262751685455441), (26, 5.132447404321283e-05), (27, 3.000168362632394e-05)]
+
+ 43%|████▎     | 65/150 [13:59<16:54, 11.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 504, Spatial MSE: [(0, 9.612307394490927e-07), (1, 1.1280101716693025e-05), (2, 5.012748442823067e-05), (3, 1.2283167052373756e-05), (4, 5.008705102227395e-06), (5, 7.082624506438151e-06), (6, 1.9730316125787795e-05), (7, 1.53772889461834e-05), (8, 1.4501798432320356e-05), (9, 3.505185304675251e-05), (10, 7.760065636830404e-05), (11, 0.00011521523265400901), (12, 0.00016307824989780784), (13, 0.0022965548560023308), (14, 0.001769551308825612), (15, 0.0004739710711874068), (16, 0.010206272825598717), (17, 6.348267197608948e-05), (18, 7.160032691899687e-05), (19, 0.0008133927476592362), (20, 0.0004184437566436827), (21, 0.00012198018521303311), (22, 0.0002033968485193327), (23, 0.00014862400712445378), (24, 0.0004010545671917498), (25, 0.0008418017532676458), (26, 0.0028859905432909727), (27, 0.01428347360342741)], Temporal MSE: [(0, 7.979809879543609e-07), (1, 3.9930654338604654e-07), (2, 8.226721774917678e-07), (3, 1.4689044292026665e-06), (4, 1.7191931647175807e-06), (5, 2.1624809960485436e-06), (6, 2.933358700829558e-06), (7, 3.5484858926793095e-06), (8, 3.2341774840460857e-06), (9, 4.009901203971822e-06), (10, 7.315917628147872e-06), (11, 6.683708488708362e-05), (12, 4.954537143930793e-05), (13, 5.214062912273221e-05), (14, 3.490582821541466e-05), (15, 2.8480641049100086e-05), (16, 2.4364511773455888e-05), (17, 1.5529523807344958e-05), (18, 3.312322951387614e-05), (19, 2.9905973860877566e-05), (20, 3.478476355667226e-05), (21, 3.558379103196785e-05), (22, 5.224578853813e-05), (23, 0.0001076562111848034), (24, 0.00021475694666150957), (25, 0.0004345336346887052), (26, 9.55230207182467e-05), (27, 3.419858330744319e-05)]
+
+ 44%|████▍     | 66/150 [14:10<16:03, 11.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 498, Spatial MSE: [(0, 8.265800488516106e-07), (1, 6.743222911609337e-06), (2, 2.4328479412361048e-05), (3, 8.790318133833352e-06), (4, 1.9171920939697884e-06), (5, 1.2227835668454645e-06), (6, 5.000739747629268e-06), (7, 2.9066741262795404e-06), (8, 1.7146173831861233e-06), (9, 2.1551861664192984e-06), (10, 2.7024511837225873e-06), (11, 1.7432716049370356e-05), (12, 6.65979678160511e-05), (13, 0.0021530017256736755), (14, 0.0015568987000733614), (15, 0.0003686790878418833), (16, 0.00985488761216402), (17, 3.553711940185167e-05), (18, 4.026680835522711e-05), (19, 0.0007389636593870819), (20, 0.00036231285776011646), (21, 6.945905624888837e-05), (22, 0.00012562399206217378), (23, 7.797049329383299e-05), (24, 0.00015711293963249773), (25, 0.0003196026664227247), (26, 0.0006120890029706061), (27, 0.01348263118416071)], Temporal MSE: [(0, 7.546223628196458e-07), (1, 3.14840207238376e-07), (2, 4.107055531221704e-07), (3, 5.24919187228079e-07), (4, 3.25631702935425e-07), (5, 5.275499006529571e-07), (6, 5.983549726806814e-07), (7, 8.178737402886327e-07), (8, 3.6071273257221037e-07), (9, 5.072903945801954e-07), (10, 9.750768867888837e-07), (11, 3.56723612640053e-05), (12, 4.064808672410436e-05), (13, 3.955093052354641e-05), (14, 1.482485549786361e-05), (15, 1.7033988115144894e-05), (16, 1.1356583854649216e-05), (17, 7.419475423375843e-06), (18, 2.0474435586947948e-05), (19, 1.626200719329063e-05), (20, 1.1539294973772485e-05), (21, 1.7750953702488914e-05), (22, 2.5175933842547238e-05), (23, 6.751283217454329e-05), (24, 8.991093636723235e-05), (25, 0.00023852895537856966), (26, 5.0550963351270184e-05), (27, 3.025476326001808e-05)]
+
+ 45%|████▍     | 67/150 [14:18<14:39, 10.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 492, Spatial MSE: [(0, 8.298047760035843e-07), (1, 1.763011641742196e-05), (2, 5.493584103533067e-05), (3, 1.569145024404861e-05), (4, 2.0867246348643675e-05), (5, 4.0976228774525225e-05), (6, 0.00016613882326055318), (7, 0.00023530494945589453), (8, 0.0002045321452897042), (9, 0.0002742767974268645), (10, 0.00028754016966558993), (11, 0.00034240103559568524), (12, 0.0004461880889721215), (13, 0.0027192046400159597), (14, 0.002615706529468298), (15, 0.0016555129550397396), (16, 0.015954703092575073), (17, 0.001044612261466682), (18, 0.000991274369880557), (19, 0.0017794826999306679), (20, 0.0018227132968604565), (21, 0.0013248110190033913), (22, 0.001059498987160623), (23, 0.0014170786598697305), (24, 0.0030110373627394438), (25, 0.0060005453415215015), (26, 0.0060293530113995075), (27, 0.013801977969706059)], Temporal MSE: [(0, 1.0408809885120718e-06), (1, 5.363077661968418e-07), (2, 1.7576096524862805e-06), (3, 2.2390200683730654e-05), (4, 1.3896741620555986e-05), (5, 2.7560245143831708e-05), (6, 0.00013847941590938717), (7, 5.474935460370034e-05), (8, 3.849360655294731e-05), (9, 5.556727774091996e-05), (10, 9.29800808080472e-05), (11, 0.00015001049905549735), (12, 0.00013279274571686983), (13, 0.00020304064673837274), (14, 0.0003626298566814512), (15, 0.00039353856118395925), (16, 0.0004989316803403199), (17, 0.000444236007751897), (18, 0.00034861432504840195), (19, 0.0005472844932228327), (20, 0.0008536495151929557), (21, 0.0006251885206438601), (22, 0.0004169346939306706), (23, 0.0006108536035753787), (24, 0.0015915031544864178), (25, 0.0025551696307957172), (26, 0.0005187336937524378), (27, 4.425492079462856e-05)]
+
+ 45%|████▌     | 68/150 [14:28<14:09, 10.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 486, Spatial MSE: [(0, 9.199823693961662e-07), (1, 6.53972119835089e-06), (2, 2.481662158970721e-05), (3, 9.239518476533704e-06), (4, 1.7589190974831581e-06), (5, 1.4146623925626045e-06), (6, 4.891302069154335e-06), (7, 4.384914518595906e-06), (8, 1.8522920299801626e-06), (9, 2.1263765574985882e-06), (10, 2.616835445223842e-06), (11, 1.2185221748950426e-05), (12, 5.2171741117490456e-05), (13, 0.0020137059036642313), (14, 0.001511023030616343), (15, 0.00037053064443171024), (16, 0.00983563531190157), (17, 3.352579733473249e-05), (18, 4.440359043655917e-05), (19, 0.0007322406163439155), (20, 0.0003600766649469733), (21, 5.8172387070953846e-05), (22, 0.00012013701780233532), (23, 7.824406930012628e-05), (24, 0.0001458842889405787), (25, 0.00030442007118836045), (26, 0.0007562703685835004), (27, 0.013353276066482067)], Temporal MSE: [(0, 8.45119984660414e-07), (1, 3.56574759052819e-07), (2, 4.656424437143869e-07), (3, 5.805285354654188e-07), (4, 3.889064430495637e-07), (5, 6.311663014457736e-07), (6, 7.523927934016683e-07), (7, 1.0588892109808512e-06), (8, 4.2415859979882953e-07), (9, 5.784148697784985e-07), (10, 1.0319472494302318e-06), (11, 2.1301437300280668e-05), (12, 3.521935286698863e-05), (13, 3.762283449759707e-05), (14, 1.4315593944047578e-05), (15, 1.748831164150033e-05), (16, 1.0699857739382423e-05), (17, 7.092240593919996e-06), (18, 2.0725306967506185e-05), (19, 1.7400700016878545e-05), (20, 1.135417460318422e-05), (21, 1.5852850992814638e-05), (22, 2.3703989427303895e-05), (23, 6.648342969128862e-05), (24, 8.797972259344533e-05), (25, 0.00025019547319971025), (26, 5.435907587525435e-05), (27, 3.095022475463338e-05)]
+
+ 46%|████▌     | 69/150 [14:40<14:40, 10.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 480, Spatial MSE: [(0, 8.410525538238289e-07), (1, 9.647784281696659e-06), (2, 3.4879289160016924e-05), (3, 8.679291568114422e-06), (4, 4.609961706592003e-06), (5, 6.081409992475528e-06), (6, 2.2729942429577932e-05), (7, 1.2970802345080301e-05), (8, 1.716207043500617e-05), (9, 4.7700432332931086e-05), (10, 8.660042658448219e-05), (11, 0.00011963438737438992), (12, 0.00014127812755759805), (13, 0.002157040871679783), (14, 0.0018552565015852451), (15, 0.0007049624109640718), (16, 0.019681641831994057), (17, 7.807488145772368e-05), (18, 0.00011931324115721509), (19, 0.0015297194477170706), (20, 0.0012860167771577835), (21, 0.00010235578520223498), (22, 0.00016980488726403564), (23, 0.00014110571646597236), (24, 0.0003615888417698443), (25, 0.000757128931581974), (26, 0.0022127702832221985), (27, 0.01303774956613779)], Temporal MSE: [(0, 6.447880309679022e-07), (1, 3.339498562127119e-07), (2, 7.596362365802634e-07), (3, 1.37945357892022e-06), (4, 1.654875404710765e-06), (5, 2.0806462543987436e-06), (6, 2.899793116739602e-06), (7, 3.689011009555543e-06), (8, 3.302000095573021e-06), (9, 4.466074642550666e-06), (10, 8.303985850943718e-06), (11, 2.9699067454203032e-05), (12, 4.356550925876945e-05), (13, 5.015644273953512e-05), (14, 3.660759830381721e-05), (15, 3.4648444852791727e-05), (16, 2.59684838965768e-05), (17, 1.9240487745264545e-05), (18, 3.585823651519604e-05), (19, 3.5102584661217406e-05), (20, 4.399381941766478e-05), (21, 3.967951124650426e-05), (22, 5.251774928183295e-05), (23, 0.00010040312918135896), (24, 0.00019902500207535923), (25, 0.0004520866787061095), (26, 0.00010209280299022794), (27, 3.366338205523789e-05)]
+
+ 47%|████▋     | 70/150 [14:51<14:23, 10.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 474, Spatial MSE: [(0, 1.2302695040489198e-06), (1, 6.765963462385116e-06), (2, 2.4201706764870323e-05), (3, 8.52975335874362e-06), (4, 2.072771621897118e-06), (5, 1.3431970273813931e-06), (6, 5.207314188737655e-06), (7, 3.63074923370732e-06), (8, 1.913580945256399e-06), (9, 2.040930667135399e-06), (10, 2.356230879740906e-06), (11, 9.525955647404771e-06), (12, 4.2037743696710095e-05), (13, 0.0018618395552039146), (14, 0.00145444693043828), (15, 0.0003757977101486176), (16, 0.010474247857928276), (17, 3.418651976971887e-05), (18, 4.8547783080721274e-05), (19, 0.0007678964757360518), (20, 0.0003677651984617114), (21, 7.915190508356318e-05), (22, 0.00013802821922581643), (23, 7.812747935531661e-05), (24, 0.00014860424562357366), (25, 0.00031292560743167996), (26, 0.000598810613155365), (27, 0.013341700658202171)], Temporal MSE: [(0, 1.0046885563497199e-06), (1, 3.369682417542208e-07), (2, 4.393889412313001e-07), (3, 5.528463589143939e-07), (4, 3.6298018812885857e-07), (5, 5.508035769707931e-07), (6, 6.944223969185259e-07), (7, 9.543858823235496e-07), (8, 3.9197891510411864e-07), (9, 5.631231942970771e-07), (10, 9.2251400474197e-07), (11, 1.4323230061563663e-05), (12, 2.9822616852470674e-05), (13, 3.703546099131927e-05), (14, 1.4871830899210181e-05), (15, 1.8868862753151916e-05), (16, 1.1077241651946679e-05), (17, 8.026796422200277e-06), (18, 2.3745322323520668e-05), (19, 1.945747499121353e-05), (20, 1.4517643649014644e-05), (21, 1.762594547471963e-05), (22, 2.7913016310776584e-05), (23, 7.159065717132762e-05), (24, 9.166961535811424e-05), (25, 0.0002603285829536617), (26, 5.335178866516799e-05), (27, 3.32205745507963e-05)]
+
+ 47%|████▋     | 71/150 [15:02<14:26, 10.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 468, Spatial MSE: [(0, 9.909625759974006e-07), (1, 1.589086969033815e-05), (2, 5.2909021178493276e-05), (3, 1.4875095075694844e-05), (4, 1.9751385480049066e-05), (5, 3.132954952889122e-05), (6, 0.00014365598326548934), (7, 0.0002079443947877735), (8, 0.00016375895938836038), (9, 0.00023655933910049498), (10, 0.0002809171855915338), (11, 0.0003148667165078223), (12, 0.0003812876529991627), (13, 0.0024189678952097893), (14, 0.0026467465795576572), (15, 0.0018622907809913158), (16, 0.0196194089949131), (17, 0.0010077913757413626), (18, 0.0009632818400859833), (19, 0.00212198868393898), (20, 0.00264954986050725), (21, 0.002263162285089493), (22, 0.0010815231362357736), (23, 0.0015243127709254622), (24, 0.004087773617357016), (25, 0.008324878290295601), (26, 0.008973488584160805), (27, 0.013341438956558704)], Temporal MSE: [(0, 1.0536805348237976e-06), (1, 5.185416966924095e-07), (2, 1.7847519302449655e-06), (3, 1.1343936421326362e-05), (4, 1.1259162420174107e-05), (5, 2.256364496133756e-05), (6, 0.0001147243965533562), (7, 4.438664109329693e-05), (8, 3.107842348981649e-05), (9, 4.508913480094634e-05), (10, 8.0853198596742e-05), (11, 0.00010795158596010879), (12, 0.00011380879732314497), (13, 0.00019042985513806343), (14, 0.0003414098755456507), (15, 0.0004113094473723322), (16, 0.000536374282091856), (17, 0.0004716554540209472), (18, 0.00037647553835995495), (19, 0.0007090257713571191), (20, 0.001397829269990325), (21, 0.0010113652097061276), (22, 0.0004865563241764903), (23, 0.0008929740870371461), (24, 0.0026999886613339186), (25, 0.0043859658762812614), (26, 0.0007083232630975544), (27, 5.5217129556695e-05)]
+
+ 48%|████▊     | 72/150 [15:13<14:11, 10.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 462, Spatial MSE: [(0, 7.673736490687588e-07), (1, 5.480257186718518e-06), (2, 1.745947520248592e-05), (3, 7.105349141056649e-06), (4, 1.3468811630445998e-06), (5, 1.1675646192088607e-06), (6, 3.2958064366539475e-06), (7, 2.4932417090894887e-06), (8, 1.4796191862842534e-06), (9, 2.293795432706247e-06), (10, 1.8894286313297926e-06), (11, 5.76930233364692e-06), (12, 2.1789226593682542e-05), (13, 0.0012765255523845553), (14, 0.00139841518830508), (15, 0.0004284743336029351), (16, 0.014728237874805927), (17, 5.5904805776663125e-05), (18, 3.968827149947174e-05), (19, 0.001113489968702197), (20, 0.0006585445371456444), (21, 6.729151209583506e-05), (22, 0.00015645715757273138), (23, 7.95490195741877e-05), (24, 0.00015770985919516534), (25, 0.00032447357079945505), (26, 0.0005392158636823297), (27, 0.013143708929419518)], Temporal MSE: [(0, 6.369635912051308e-07), (1, 2.568981187778263e-07), (2, 3.558774892553629e-07), (3, 4.7295878857767093e-07), (4, 3.383100590781396e-07), (5, 5.34804030394298e-07), (6, 6.858196002212935e-07), (7, 7.631630296600633e-07), (8, 3.8110567857074784e-07), (9, 4.923032861370302e-07), (10, 7.900108016656304e-07), (11, 5.769097697339021e-06), (12, 1.5367422747658566e-05), (13, 3.5577351809479296e-05), (14, 1.7777454559109174e-05), (15, 2.139076059393119e-05), (16, 1.3125754776410758e-05), (17, 1.150330626842333e-05), (18, 2.6490715754334815e-05), (19, 2.2189395167515613e-05), (20, 1.5225797142193187e-05), (21, 2.474275061103981e-05), (22, 3.713964542839676e-05), (23, 8.499917748849839e-05), (24, 9.420204150956124e-05), (25, 0.00024663613294251263), (26, 5.244174099061638e-05), (27, 3.958776142098941e-05)]
+
+ 49%|████▊     | 73/150 [15:25<14:26, 11.25s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 456, Spatial MSE: [(0, 7.938578505672922e-07), (1, 7.656447451154236e-06), (2, 2.5657922378741205e-05), (3, 9.441657311981544e-06), (4, 4.266909854777623e-06), (5, 5.929510734858923e-06), (6, 1.910248647618573e-05), (7, 1.1469133823993616e-05), (8, 1.5290483133867383e-05), (9, 3.094054773100652e-05), (10, 6.674072938039899e-05), (11, 8.901046385290101e-05), (12, 0.0001080814836313948), (13, 0.001523608691059053), (14, 0.001824982580728829), (15, 0.000803670147433877), (16, 0.02611306682229042), (17, 9.850846254266798e-05), (18, 0.00010688116890378296), (19, 0.0016524858074262738), (20, 0.00107474101241678), (21, 0.00011423325486248359), (22, 0.00020870186563115567), (23, 0.0001598117087269202), (24, 0.00040102400816977024), (25, 0.0008718294557183981), (26, 0.002493307925760746), (27, 0.013327399268746376)], Temporal MSE: [(0, 6.61565650261764e-07), (1, 3.4366101431260176e-07), (2, 6.674166002085258e-07), (3, 1.4131845773590612e-06), (4, 1.8693845049710944e-06), (5, 2.3043780856824014e-06), (6, 3.473812284937594e-06), (7, 3.7051968320156448e-06), (8, 3.173749519191915e-06), (9, 4.0110680856741965e-06), (10, 7.355608431680594e-06), (11, 1.9745961253647693e-05), (12, 2.7642829081742093e-05), (13, 4.914888268103823e-05), (14, 3.9077738620107993e-05), (15, 4.0367911424255e-05), (16, 2.714980837481562e-05), (17, 2.314184712304268e-05), (18, 4.067586996825412e-05), (19, 3.656355329439975e-05), (20, 3.477755672065541e-05), (21, 5.58055799046997e-05), (22, 7.255205855472013e-05), (23, 0.00012223357043694705), (24, 0.00021469700732268393), (25, 0.0004840019973926246), (26, 0.0001040299393935129), (27, 4.420722689246759e-05)]
+
+ 49%|████▉     | 74/150 [15:37<14:40, 11.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 450, Spatial MSE: [(0, 7.124835406102648e-07), (1, 6.881315584905678e-06), (2, 1.8853266738005914e-05), (3, 7.373661901510786e-06), (4, 1.9054102722293464e-06), (5, 1.189918862110062e-06), (6, 3.2430602914246265e-06), (7, 2.520997668398195e-06), (8, 1.6305079952871893e-06), (9, 2.785908918667701e-06), (10, 5.5483014875790104e-06), (11, 1.1722610906872433e-05), (12, 2.8240498068043962e-05), (13, 0.0008131993818096817), (14, 0.001303736469708383), (15, 0.0004702538135461509), (16, 0.01957026496529579), (17, 0.00012371038610581309), (18, 5.1295974117238075e-05), (19, 0.0013852562988176942), (20, 0.0008667998481541872), (21, 6.421298894565552e-05), (22, 0.00016977857740130275), (23, 8.125520980684087e-05), (24, 0.00016377386054955423), (25, 0.0003247415879741311), (26, 0.0005185359041206539), (27, 0.013670330867171288)], Temporal MSE: [(0, 6.616507448597986e-07), (1, 2.800314575779339e-07), (2, 3.7407605191219773e-07), (3, 4.918786089547211e-07), (4, 3.6235712741472526e-07), (5, 5.405211140896427e-07), (6, 6.46789146685478e-07), (7, 7.965086297190282e-07), (8, 4.017714729798172e-07), (9, 6.320560146377829e-07), (10, 2.6586333206068957e-06), (11, 1.2915710613015108e-05), (12, 1.235785020980984e-05), (13, 3.375158485141583e-05), (14, 1.9888891984010115e-05), (15, 3.5573099012253806e-05), (16, 1.85998942470178e-05), (17, 1.5220148270600475e-05), (18, 3.0731513106729835e-05), (19, 2.6810155759449117e-05), (20, 1.818535201891791e-05), (21, 3.1449068046640605e-05), (22, 4.869877375313081e-05), (23, 9.749712626216933e-05), (24, 0.00010162532271351665), (25, 0.00024870806373655796), (26, 5.1269355026306584e-05), (27, 4.9155030865222216e-05)]
+
+ 50%|█████     | 75/150 [15:47<13:45, 11.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 444, Spatial MSE: [(0, 8.78744913279661e-07), (1, 1.4202004422259051e-05), (2, 4.057553451275453e-05), (3, 1.3377654795476701e-05), (4, 1.9038063328480348e-05), (5, 2.8936387025169097e-05), (6, 0.00017445177945774049), (7, 0.0002058928948827088), (8, 0.00016732423682697117), (9, 0.000263217487372458), (10, 0.0003118427412118763), (11, 0.0003754673234652728), (12, 0.00043345949961803854), (13, 0.0017077575903385878), (14, 0.0030844241846352816), (15, 0.002052613999694586), (16, 0.033105652779340744), (17, 0.001066430239006877), (18, 0.0011524424189701676), (19, 0.003790991846472025), (20, 0.004216212313622236), (21, 0.002149159787222743), (22, 0.0011295474832877517), (23, 0.001570004504173994), (24, 0.0034952163696289062), (25, 0.007070756983011961), (26, 0.008337614126503468), (27, 0.014681035652756691)], Temporal MSE: [(0, 9.785029533304623e-07), (1, 4.612642214851803e-07), (2, 1.8528328382672044e-06), (3, 8.639983207103796e-06), (4, 1.121717559726676e-05), (5, 2.126796607626602e-05), (6, 0.00010437773744342849), (7, 4.253484803484753e-05), (8, 2.958269396913238e-05), (9, 4.5514971134252846e-05), (10, 8.331840945174918e-05), (11, 0.00011310820264043286), (12, 0.00011355285096215084), (13, 0.00020811054855585098), (14, 0.0003512331168167293), (15, 0.0004236762470100075), (16, 0.000522649846971035), (17, 0.00046671784366481006), (18, 0.0004076896293554455), (19, 0.0007622985867783427), (20, 0.001403520000167191), (21, 0.0010356619022786617), (22, 0.000526095274835825), (23, 0.0008303620270453393), (24, 0.002211964689195156), (25, 0.003094893181696534), (26, 0.0005838926299475133), (27, 7.118673966033384e-05)]
+
+ 51%|█████     | 76/150 [15:56<12:57, 10.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 438, Spatial MSE: [(0, 1.1639340300462209e-06), (1, 6.407627097360091e-06), (2, 1.7322510757367127e-05), (3, 7.645649020560086e-06), (4, 1.5401086557176313e-06), (5, 1.2125948387620156e-06), (6, 3.491734560157056e-06), (7, 3.2147140700544696e-06), (8, 1.6961531628112425e-06), (9, 1.74350600445905e-06), (10, 1.9336596324137645e-06), (11, 4.305710262997309e-06), (12, 1.3023318388150074e-05), (13, 0.0003446776536293328), (14, 0.0006677909987047315), (15, 0.00037752321804873645), (16, 0.020237145945429802), (17, 9.443511225981638e-05), (18, 8.659074228489771e-05), (19, 0.0013862294144928455), (20, 0.0009031257941387594), (21, 0.00011537493264768273), (22, 0.00017818495689425617), (23, 8.693799463799223e-05), (24, 0.00015891357907094061), (25, 0.000331298157107085), (26, 0.0005731584969907999), (27, 0.010869932360947132)], Temporal MSE: [(0, 8.623621852166252e-07), (1, 2.850264877451991e-07), (2, 3.7817827092112566e-07), (3, 5.072071758149832e-07), (4, 3.8069680385888205e-07), (5, 5.636936180053453e-07), (6, 7.257432912410877e-07), (7, 8.828275213090819e-07), (8, 4.1915080828403006e-07), (9, 5.411882284533931e-07), (10, 9.155813813777058e-07), (11, 6.267962362471735e-06), (12, 8.50407832331257e-06), (13, 1.6082216461654752e-05), (14, 1.982683716050815e-05), (15, 3.220626967959106e-05), (16, 2.0091240003239363e-05), (17, 2.3145572413341142e-05), (18, 3.455427213339135e-05), (19, 3.062783071072772e-05), (20, 2.473029053362552e-05), (21, 3.891297819791362e-05), (22, 5.1133443776052445e-05), (23, 9.999761823564768e-05), (24, 9.677063644630834e-05), (25, 0.0002465857833158225), (26, 4.750681182486005e-05), (27, 4.518830974120647e-05)]
+
+ 51%|█████▏    | 77/150 [16:05<12:08,  9.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 432, Spatial MSE: [(0, 9.555255928717088e-07), (1, 8.731980415177532e-06), (2, 2.6345293008489534e-05), (3, 7.711548278166447e-06), (4, 3.5382754504098557e-06), (5, 5.185640475247055e-06), (6, 1.546023486298509e-05), (7, 9.621272511139978e-06), (8, 1.3514297279471066e-05), (9, 3.55796109943185e-05), (10, 5.968554978608154e-05), (11, 8.814087050268427e-05), (12, 8.611402881797403e-05), (13, 0.0004158758674748242), (14, 0.0008209468214772642), (15, 0.0005336962058208883), (16, 0.02511085569858551), (17, 0.0001284334430238232), (18, 0.0001398127933498472), (19, 0.0016609056619927287), (20, 0.0012507927604019642), (21, 0.0001557226642034948), (22, 0.00025066742091439664), (23, 0.00016974765458144248), (24, 0.0003930398088414222), (25, 0.0009422768489457667), (26, 0.002784726209938526), (27, 0.010747939348220825)], Temporal MSE: [(0, 6.888511734359781e-07), (1, 3.4611252885952126e-07), (2, 7.202764322755684e-07), (3, 1.2024838724755682e-06), (4, 1.6951406678344938e-06), (5, 2.108709622916649e-06), (6, 3.6142703265795717e-06), (7, 3.7524214349105023e-06), (8, 3.3065116440411657e-06), (9, 3.8511834645760246e-06), (10, 6.844831204944057e-06), (11, 1.3465012671076693e-05), (12, 1.6626496289973147e-05), (13, 2.7610865799942985e-05), (14, 4.044185334350914e-05), (15, 4.4622705900110304e-05), (16, 3.0673108994960785e-05), (17, 3.0329436413012445e-05), (18, 4.872535646427423e-05), (19, 4.411492045619525e-05), (20, 4.0461669414071366e-05), (21, 6.321901310002431e-05), (22, 8.080457337200642e-05), (23, 0.00012926380441058427), (24, 0.00020248102373443544), (25, 0.000454291352070868), (26, 0.00010329556971555576), (27, 4.8248530220007524e-05)]
+
+ 52%|█████▏    | 78/150 [16:16<12:24, 10.34s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 426, Spatial MSE: [(0, 5.344975306798005e-07), (1, 5.536887783819111e-06), (2, 1.284015343117062e-05), (3, 6.774739631509874e-06), (4, 1.4632903457822977e-06), (5, 1.093473883884144e-06), (6, 2.478735495969886e-06), (7, 2.3685588530497625e-06), (8, 1.3874421256332425e-06), (9, 1.6199907122427248e-06), (10, 1.754694835653936e-06), (11, 3.508440613586572e-06), (12, 8.661326319270302e-06), (13, 0.00021730916341766715), (14, 0.00033899740083143115), (15, 0.00021725890110246837), (16, 0.005507155787199736), (17, 1.9051818526349962e-05), (18, 1.9298433471703902e-05), (19, 0.0002815098559949547), (20, 0.00013760384172201157), (21, 6.116266740718856e-05), (22, 0.00013688555918633938), (23, 5.1105460443068296e-05), (24, 8.185724436771125e-05), (25, 0.00017088647291529924), (26, 0.00032726855715736747), (27, 0.004621424712240696)], Temporal MSE: [(0, 5.257418820292514e-07), (1, 2.3224107792430004e-07), (2, 3.1977907610780676e-07), (3, 4.5097797851667565e-07), (4, 3.5993619462715287e-07), (5, 5.167796643945621e-07), (6, 7.60400041599496e-07), (7, 7.514786943829677e-07), (8, 3.737808356163441e-07), (9, 4.799134103450342e-07), (10, 7.880329917497875e-07), (11, 6.301759185589617e-06), (12, 3.978711447416572e-06), (13, 1.006082402454922e-05), (14, 1.0607972399157006e-05), (15, 1.0334931175748352e-05), (16, 4.982631253369618e-06), (17, 4.030829586554319e-06), (18, 1.2228947525727563e-05), (19, 1.0722808838181663e-05), (20, 8.97359313967172e-06), (21, 1.486782366555417e-05), (22, 2.2884591089677997e-05), (23, 4.726077168015763e-05), (24, 4.959929356118664e-05), (25, 0.00012468165368773043), (26, 2.9129707399988547e-05), (27, 1.3427038538793568e-05)]
+
+ 53%|█████▎    | 79/150 [16:27<12:23, 10.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 420, Spatial MSE: [(0, 8.525759085387108e-07), (1, 1.434077239537146e-05), (2, 2.920459519373253e-05), (3, 1.2915616025566123e-05), (4, 1.5650899513275363e-05), (5, 2.275328733958304e-05), (6, 7.891240966273472e-05), (7, 0.0001272994268219918), (8, 0.00012548644735943526), (9, 0.000197355926502496), (10, 0.00021958169236313552), (11, 0.00026364062796346843), (12, 0.00032701031886972487), (13, 0.0007526708650402725), (14, 0.0015613423893228173), (15, 0.0014458491932600737), (16, 0.02768230065703392), (17, 0.0005916086956858635), (18, 0.0007021459168754518), (19, 0.0019267620518803596), (20, 0.0016255767550319433), (21, 0.0008555056410841644), (22, 0.000792036997154355), (23, 0.0009991198312491179), (24, 0.002251732163131237), (25, 0.004874574951827526), (26, 0.00672919163480401), (27, 0.00592060387134552)], Temporal MSE: [(0, 9.348209459858481e-07), (1, 5.242164888841216e-07), (2, 1.5109602600205108e-06), (3, 4.253936367604183e-06), (4, 7.383413503703196e-06), (5, 1.2940613487444352e-05), (6, 5.940576738794334e-05), (7, 2.8948585168109275e-05), (8, 2.382086677243933e-05), (9, 3.850230132229626e-05), (10, 7.452553836628795e-05), (11, 9.07135326997377e-05), (12, 8.748607797315344e-05), (13, 0.00015948136569932103), (14, 0.000277744751656428), (15, 0.00026501467800699174), (16, 0.0003288846928626299), (17, 0.0002768768463283777), (18, 0.00024508117348887026), (19, 0.0003681493108160794), (20, 0.000570711912587285), (21, 0.0004067412228323519), (22, 0.0003343612188473344), (23, 0.0005156215629540384), (24, 0.0014102502027526498), (25, 0.0023081141989678144), (26, 0.000490069214720279), (27, 3.620165807660669e-05)]
+
+ 53%|█████▎    | 80/150 [16:41<13:24, 11.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 414, Spatial MSE: [(0, 7.672754236409673e-07), (1, 6.335768830467714e-06), (2, 1.6646517906337976e-05), (3, 6.874405244161608e-06), (4, 1.4896669426889275e-06), (5, 1.1702334177243756e-06), (6, 2.3250322556123137e-06), (7, 2.0271022549422923e-06), (8, 1.4291481420514174e-06), (9, 2.2739527594239917e-06), (10, 1.5823790135982563e-06), (11, 2.749072109509143e-06), (12, 4.713981979875825e-06), (13, 0.00013743726594839245), (14, 0.00016361742746084929), (15, 0.0001498481578892097), (16, 0.0046394518576562405), (17, 2.8370541258482262e-05), (18, 1.914995846163947e-05), (19, 0.0004577895742841065), (20, 0.00034003000473603606), (21, 2.532846156100277e-05), (22, 6.544303323607892e-05), (23, 3.428310446906835e-05), (24, 7.119884685380384e-05), (25, 0.00020110042532905936), (26, 0.00040134252049028873), (27, 0.003417608793824911)], Temporal MSE: [(0, 6.076759859752201e-07), (1, 2.5580294504834455e-07), (2, 3.517453137646953e-07), (3, 4.79273978726269e-07), (4, 3.7779443573526805e-07), (5, 5.54586051748629e-07), (6, 6.940881576156244e-07), (7, 7.39716369935195e-07), (8, 3.98350124442004e-07), (9, 5.134992306921049e-07), (10, 7.475475740648108e-07), (11, 1.8018746459347312e-06), (12, 2.9983182230353123e-06), (13, 5.226564553595381e-06), (14, 9.542613042867742e-06), (15, 8.399034413741902e-06), (16, 6.696726813970599e-06), (17, 9.16500994208036e-06), (18, 8.953958968049847e-06), (19, 9.018027412821539e-06), (20, 9.38480025070021e-06), (21, 2.0231824237271212e-05), (22, 2.2531359718414024e-05), (23, 3.588280742405914e-05), (24, 2.8454975108616054e-05), (25, 7.1661539550405e-05), (26, 2.3333102944889106e-05), (27, 1.2632812286028638e-05)]
+
+ 54%|█████▍    | 81/150 [16:54<13:51, 12.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 408, Spatial MSE: [(0, 7.309465104299306e-07), (1, 7.445873961842153e-06), (2, 1.819465796870645e-05), (3, 6.320811280602356e-06), (4, 3.6228182125341846e-06), (5, 4.7549056034768e-06), (6, 1.5534933481831104e-05), (7, 1.013123164739227e-05), (8, 1.3764362847723532e-05), (9, 2.7416586817707866e-05), (10, 6.352072523441166e-05), (11, 8.520541450707242e-05), (12, 8.575899119023234e-05), (13, 0.0003166840469930321), (14, 0.000698478368576616), (15, 0.0006309036398306489), (16, 0.015464196912944317), (17, 9.221537038683891e-05), (18, 0.00015685903781559318), (19, 0.001239104545675218), (20, 0.0011440158123150468), (21, 0.00012292973406147212), (22, 0.00012687071284744889), (23, 0.00011179760622326285), (24, 0.000316563033265993), (25, 0.0008496249793097377), (26, 0.0024178314488381147), (27, 0.0029812112916260958)], Temporal MSE: [(0, 6.589116878785717e-07), (1, 3.1293521374209377e-07), (2, 6.703414214825898e-07), (3, 1.1709888667610358e-06), (4, 1.7896891222335398e-06), (5, 2.1298305910022464e-06), (6, 3.1997203677747166e-06), (7, 3.667723603939521e-06), (8, 2.901319248849177e-06), (9, 3.815162926912308e-06), (10, 6.900959306221921e-06), (11, 1.1300102414679714e-05), (12, 1.4162979823595379e-05), (13, 1.9712510038516484e-05), (14, 3.249113433412276e-05), (15, 2.2673459170619026e-05), (16, 1.907136720546987e-05), (17, 1.7990807464229874e-05), (18, 2.3361077182926238e-05), (19, 2.608124486869201e-05), (20, 3.2609626941848546e-05), (21, 4.932663432555273e-05), (22, 6.107427179813385e-05), (23, 7.561116217402741e-05), (24, 0.0001411512930644676), (25, 0.00030653923749923706), (26, 8.432895265286788e-05), (27, 1.8776370779960416e-05)]
+
+ 55%|█████▍    | 82/150 [17:07<14:01, 12.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 402, Spatial MSE: [(0, 6.656108553215745e-07), (1, 5.5939458434295375e-06), (2, 1.1640298907877877e-05), (3, 6.4084715631906874e-06), (4, 1.6270835203613387e-06), (5, 1.3394896996032912e-06), (6, 2.3457753286493244e-06), (7, 2.9643010748259258e-06), (8, 1.6417023971371236e-06), (9, 2.2281103611021535e-06), (10, 3.003568508574972e-06), (11, 6.235254659259226e-06), (12, 1.1721390364982653e-05), (13, 0.00019744105520658195), (14, 0.00037870698724873364), (15, 0.00010149112495128065), (16, 0.009938085451722145), (17, 5.651918036164716e-05), (18, 3.547432424966246e-05), (19, 0.0008886030991561711), (20, 0.0006290634628385305), (21, 4.746983904624358e-05), (22, 9.268438589060679e-05), (23, 4.417781747179106e-05), (24, 9.051894448930398e-05), (25, 0.00024664337979629636), (26, 0.00036162667674943805), (27, 0.004929594695568085)], Temporal MSE: [(0, 6.423187528525887e-07), (1, 2.9094491083014873e-07), (2, 3.45483726960083e-07), (3, 5.10793256580655e-07), (4, 3.998765407686733e-07), (5, 5.72490307604312e-07), (6, 6.890857662256167e-07), (7, 8.841311114338168e-07), (8, 4.34031846907601e-07), (9, 5.83296468903427e-07), (10, 1.867367245722562e-06), (11, 5.445838269224623e-06), (12, 6.734447651979281e-06), (13, 6.546092208736809e-06), (14, 7.809309863660019e-06), (15, 2.0665629563154653e-05), (16, 1.2940231499669608e-05), (17, 1.2917380445287563e-05), (18, 1.5920692021609284e-05), (19, 1.5491314115934074e-05), (20, 1.4180113794282079e-05), (21, 3.1958585168467835e-05), (22, 4.01795405196026e-05), (23, 5.816313932882622e-05), (24, 4.478824121179059e-05), (25, 9.465202310821041e-05), (26, 2.4759703592280857e-05), (27, 2.3769975086906925e-05)]
+
+ 55%|█████▌    | 83/150 [17:18<13:02, 11.67s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 396, Spatial MSE: [(0, 7.93101207818836e-07), (1, 1.2252907254151069e-05), (2, 2.2579355572815984e-05), (3, 1.2363875612209085e-05), (4, 1.7350905181956477e-05), (5, 2.286350354552269e-05), (6, 7.05383499735035e-05), (7, 0.00011297583114355803), (8, 0.00012050880468450487), (9, 0.0001888177648652345), (10, 0.0002147180202882737), (11, 0.00023991621856112033), (12, 0.0002661281614564359), (13, 0.0011131283827126026), (14, 0.0024690027348697186), (15, 0.0016607509460300207), (16, 0.04483084753155708), (17, 0.0005304728983901441), (18, 0.000573563389480114), (19, 0.0028600471559911966), (20, 0.0021624560467898846), (21, 0.00076270941644907), (22, 0.0006291137542575598), (23, 0.0008775456226430833), (24, 0.0016933312872424722), (25, 0.0041405316442251205), (26, 0.006787173915654421), (27, 0.005583382211625576)], Temporal MSE: [(0, 9.097066140384413e-07), (1, 3.8843600691507163e-07), (2, 1.5295061075448757e-06), (3, 3.953854047722416e-06), (4, 7.370366802206263e-06), (5, 1.292011347686639e-05), (6, 6.317989027593285e-05), (7, 2.8980904971831478e-05), (8, 2.3387927285511978e-05), (9, 3.7357986002461985e-05), (10, 7.334028487093747e-05), (11, 8.189852087525651e-05), (12, 8.600452565588057e-05), (13, 0.0001429722033208236), (14, 0.0002541368012316525), (15, 0.00023355541634373367), (16, 0.00025684869615361094), (17, 0.00023419258650392294), (18, 0.00022870714019518346), (19, 0.0002986499748658389), (20, 0.0004480101342778653), (21, 0.00037566066021099687), (22, 0.00034145964309573174), (23, 0.0004691977228503674), (24, 0.0010276081738993526), (25, 0.0017351710703223944), (26, 0.00044066162081435323), (27, 4.536637425189838e-05)]
+
+ 56%|█████▌    | 84/150 [17:32<13:39, 12.42s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 390, Spatial MSE: [(0, 7.013040885794908e-07), (1, 6.280508387135342e-06), (2, 1.3444003343465738e-05), (3, 7.200874279078562e-06), (4, 1.6932821154114208e-06), (5, 1.3837407095707022e-06), (6, 2.2854637791169807e-06), (7, 3.494799784675706e-06), (8, 1.85060810053983e-06), (9, 1.924354819493601e-06), (10, 1.902838448586408e-06), (11, 3.186561343682115e-06), (12, 8.9127415776602e-06), (13, 0.0003055022971238941), (14, 0.0005650835810229182), (15, 0.00022330242791213095), (16, 0.013442129828035831), (17, 4.745501064462587e-05), (18, 7.737176201771945e-05), (19, 0.0010638402309268713), (20, 0.0007285160245373845), (21, 4.468463157536462e-05), (22, 9.784029680304229e-05), (23, 5.617658462142572e-05), (24, 0.00011790115240728483), (25, 0.0003171636490151286), (26, 0.0004228525795042515), (27, 0.008050985634326935)], Temporal MSE: [(0, 6.545049018313875e-07), (1, 3.124062573078845e-07), (2, 3.925021871964418e-07), (3, 5.582280095950409e-07), (4, 4.6700543521183135e-07), (5, 6.702209702780237e-07), (6, 9.312263387073472e-07), (7, 1.0066819413623307e-06), (8, 5.05746072576585e-07), (9, 6.455732659560454e-07), (10, 9.133559046858863e-07), (11, 3.1194763323583174e-06), (12, 6.93763240633416e-06), (13, 1.201041777676437e-05), (14, 1.4923582966730464e-05), (15, 2.2073230866226368e-05), (16, 1.2666559086937923e-05), (17, 2.0782377760042436e-05), (18, 2.150397267541848e-05), (19, 1.7051952454494312e-05), (20, 1.2676753613050096e-05), (21, 4.1217554098693654e-05), (22, 4.4680171413347125e-05), (23, 7.398860907414928e-05), (24, 6.441552250180393e-05), (25, 0.00016751278599258512), (26, 3.823819133685902e-05), (27, 3.5765613574767485e-05)]
+
+ 57%|█████▋    | 85/150 [17:42<12:52, 11.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 384, Spatial MSE: [(0, 7.636290888513031e-07), (1, 6.51859136269195e-06), (2, 1.4241475582821295e-05), (3, 6.153707090561511e-06), (4, 3.5337866393092554e-06), (5, 5.129390501679154e-06), (6, 1.35441196107422e-05), (7, 1.0143190593225881e-05), (8, 1.4121668755251449e-05), (9, 3.268722502980381e-05), (10, 5.95027522649616e-05), (11, 9.106127254199237e-05), (12, 8.469244494335726e-05), (13, 0.0004033718432765454), (14, 0.0008154874085448682), (15, 0.0004066723049618304), (16, 0.01889793574810028), (17, 8.298674947582185e-05), (18, 0.0001417913445038721), (19, 0.0012450043577700853), (20, 0.0007628921885043383), (21, 8.334420272149146e-05), (22, 0.00015495617117267102), (23, 0.00013494440645445138), (24, 0.000372153299394995), (25, 0.0010091630974784493), (26, 0.0026825186796486378), (27, 0.007500800304114819)], Temporal MSE: [(0, 5.037770733906655e-07), (1, 2.5836692429948016e-07), (2, 5.397550921770744e-07), (3, 1.2115448271288187e-06), (4, 1.8394657672615722e-06), (5, 2.2802594230597606e-06), (6, 3.686458512675017e-06), (7, 3.7648867419193266e-06), (8, 3.3054388950404245e-06), (9, 3.8935950215091e-06), (10, 6.801919880672358e-06), (11, 1.0812736945808865e-05), (12, 1.4702968655910809e-05), (13, 2.488574136805255e-05), (14, 3.259054210502654e-05), (15, 3.5543489502742887e-05), (16, 2.5467445084359497e-05), (17, 3.0322227758006193e-05), (18, 3.3232339774258435e-05), (19, 2.863529334717896e-05), (20, 2.965713429148309e-05), (21, 6.087631481932476e-05), (22, 7.362110773101449e-05), (23, 0.00010180773097090423), (24, 0.0001732099917717278), (25, 0.0003979303000960499), (26, 0.00010406972432974726), (27, 4.2865514842560515e-05)]
+
+ 57%|█████▋    | 86/150 [17:52<12:03, 11.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 378, Spatial MSE: [(0, 8.097030672615801e-07), (1, 5.5377072385454085e-06), (2, 1.2279806469450705e-05), (3, 5.841145139129367e-06), (4, 1.2629926686713588e-06), (5, 1.1662941687973216e-06), (6, 2.089964482365758e-06), (7, 2.816501591951237e-06), (8, 1.5796406387380557e-06), (9, 1.841335006247391e-06), (10, 1.7107257690440747e-06), (11, 5.358767793950392e-06), (12, 1.9280192645965144e-05), (13, 0.0007579557714052498), (14, 0.0006616325117647648), (15, 0.0001860888150986284), (16, 0.004575763363391161), (17, 1.644669464440085e-05), (18, 5.639531809720211e-05), (19, 0.00025607357383705676), (20, 9.255891927750781e-05), (21, 2.5330464268336073e-05), (22, 6.308728916337714e-05), (23, 4.3162704969290644e-05), (24, 9.327408770332113e-05), (25, 0.00020964899158570915), (26, 0.0003863752353936434), (27, 0.006218763068318367)], Temporal MSE: [(0, 6.18380909145344e-07), (1, 2.5187330265907804e-07), (2, 3.3104490171353973e-07), (3, 4.6808062847958354e-07), (4, 4.040898602397647e-07), (5, 5.886329290660797e-07), (6, 7.450035468536953e-07), (7, 8.333296932505618e-07), (8, 4.322363054143352e-07), (9, 5.688431770067837e-07), (10, 9.72961743173073e-07), (11, 9.123346899286844e-06), (12, 1.4375800674315542e-05), (13, 1.533116665086709e-05), (14, 7.2996745075215586e-06), (15, 1.0009893230744638e-05), (16, 6.363351076288382e-06), (17, 4.083452495251549e-06), (18, 9.559242243994959e-06), (19, 7.435410225298256e-06), (20, 6.958477115404094e-06), (21, 1.0047684554592706e-05), (22, 1.558137228130363e-05), (23, 3.922777250409126e-05), (24, 5.0461523642297834e-05), (25, 0.00015018739213701338), (26, 3.758778257179074e-05), (27, 1.9440047253738157e-05)]
+
+ 58%|█████▊    | 87/150 [18:03<11:35, 11.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 372, Spatial MSE: [(0, 8.516719276485674e-07), (1, 1.2789749234798364e-05), (2, 1.823421916924417e-05), (3, 1.1514579455251805e-05), (4, 1.7882628526422195e-05), (5, 2.337225123483222e-05), (6, 7.404857751680538e-05), (7, 0.00011430729500716552), (8, 0.00011981317948084325), (9, 0.0001867659157142043), (10, 0.00019156020425725728), (11, 0.00021566258510574698), (12, 0.00024933338863775134), (13, 0.00174073432572186), (14, 0.0028172715101391077), (15, 0.0017001823289319873), (16, 0.033921148627996445), (17, 0.00044171250192448497), (18, 0.0005681996117345989), (19, 0.0036364677362143993), (20, 0.0037565960083156824), (21, 0.0007112808525562286), (22, 0.0005403649993240833), (23, 0.0008032348123379052), (24, 0.0015110167441889644), (25, 0.003841149155050516), (26, 0.008055297657847404), (27, 0.007124603260308504)], Temporal MSE: [(0, 9.645267482483177e-07), (1, 4.1010332552104956e-07), (2, 1.6565529676881852e-06), (3, 4.237563189235516e-06), (4, 7.4692520684038755e-06), (5, 1.3687719729205128e-05), (6, 5.643625991069712e-05), (7, 2.995678732986562e-05), (8, 2.3682545361225493e-05), (9, 3.536933218128979e-05), (10, 6.593953730771318e-05), (11, 8.320687629748136e-05), (12, 8.9131215645466e-05), (13, 0.00014447425201069564), (14, 0.00022484098735731095), (15, 0.00020038828370161355), (16, 0.00022282972349785268), (17, 0.00019655372307170182), (18, 0.00018577858281787485), (19, 0.00026198907289654016), (20, 0.00037940798210911453), (21, 0.00031102431239560246), (22, 0.0002917435485869646), (23, 0.0003928924852516502), (24, 0.0009368336759507656), (25, 0.001606717356480658), (26, 0.0004379161400720477), (27, 4.565964263747446e-05)]
+
+ 59%|█████▊    | 88/150 [18:14<11:21, 10.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 366, Spatial MSE: [(0, 5.924730999140593e-07), (1, 5.200939540372929e-06), (2, 1.2024689567624591e-05), (3, 6.4389078033855185e-06), (4, 1.2414759567036526e-06), (5, 1.1527238257258432e-06), (6, 1.5124014680623077e-06), (7, 2.7933963338000467e-06), (8, 1.5658466736567789e-06), (9, 1.4912726555849076e-06), (10, 1.681467551861715e-06), (11, 2.0731176846311428e-06), (12, 3.4106813018297544e-06), (13, 2.0002047676825896e-05), (14, 1.574844827700872e-05), (15, 6.032747023709817e-06), (16, 1.2418791811796837e-05), (17, 4.62218486063648e-06), (18, 1.2615033483598381e-05), (19, 1.0939211279037409e-05), (20, 5.293184585752897e-06), (21, 8.095808880170807e-06), (22, 2.1193365682847798e-05), (23, 2.239752939203754e-05), (24, 4.50826009910088e-05), (25, 0.00010731094516813755), (26, 0.0002993343223351985), (27, 0.001378360204398632)], Temporal MSE: [(0, 4.6529220298907603e-07), (1, 2.291004790322404e-07), (2, 3.158716594953148e-07), (3, 4.6057326130721776e-07), (4, 4.3266663851682097e-07), (5, 6.109434025347582e-07), (6, 8.523448400410416e-07), (7, 8.329517982019752e-07), (8, 4.346384230302647e-07), (9, 5.372581313167757e-07), (10, 8.160153583958163e-07), (11, 1.1454312698333524e-06), (12, 1.1036858040824882e-06), (13, 1.5859800441830885e-06), (14, 1.9316275938763283e-06), (15, 1.3696392215933884e-06), (16, 1.5084369806572795e-06), (17, 1.338062475042534e-06), (18, 1.3652789903062512e-06), (19, 2.1324653971532825e-06), (20, 2.8808317438233644e-06), (21, 3.1166871394816553e-06), (22, 4.402146259963047e-06), (23, 9.296584721596446e-06), (24, 1.380001413053833e-05), (25, 4.595839709509164e-05), (26, 1.9553857782739215e-05), (27, 2.1167647901165765e-06)]
+
+ 59%|█████▉    | 89/150 [18:26<11:42, 11.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 360, Spatial MSE: [(0, 7.83040320584405e-07), (1, 6.231638053577626e-06), (2, 1.0777378520288039e-05), (3, 5.79903053221642e-06), (4, 3.144157517454005e-06), (5, 4.349307801021496e-06), (6, 1.0416193617857061e-05), (7, 8.695265933056362e-06), (8, 1.2861551113019232e-05), (9, 2.9315828214748763e-05), (10, 4.1959112422773615e-05), (11, 7.052514411043376e-05), (12, 6.566602678503841e-05), (13, 7.955119508551434e-05), (14, 7.892904977779835e-05), (15, 4.623367567546666e-05), (16, 3.530425237840973e-05), (17, 2.1180072508286685e-05), (18, 5.461020555230789e-05), (19, 3.365539669175632e-05), (20, 2.831737401720602e-05), (21, 2.959716584882699e-05), (22, 6.175028829602525e-05), (23, 7.994224142748863e-05), (24, 0.00023456703638657928), (25, 0.000738849921617657), (26, 0.002440955489873886), (27, 0.0010907092364504933)], Temporal MSE: [(0, 5.181558435651823e-07), (1, 2.700649019971024e-07), (2, 5.578300488195964e-07), (3, 1.166928200291295e-06), (4, 1.8955666973852203e-06), (5, 2.2249073481361847e-06), (6, 3.5526006740838056e-06), (7, 3.5853795452567283e-06), (8, 2.993273255924578e-06), (9, 3.46484830515692e-06), (10, 5.965480795566691e-06), (11, 7.873560207372066e-06), (12, 8.13661335996585e-06), (13, 1.0996693163178861e-05), (14, 1.6397707440773956e-05), (15, 1.0243539691145997e-05), (16, 1.1057999472541269e-05), (17, 7.519007795053767e-06), (18, 9.575438525644131e-06), (19, 1.1519462532305624e-05), (20, 1.5926021660561673e-05), (21, 1.4509510037896689e-05), (22, 2.2565229301108047e-05), (23, 3.0987786885816604e-05), (24, 9.479338768869638e-05), (25, 0.0002745722886174917), (26, 9.292635513702407e-05), (27, 1.0148060937353875e-05)]
+
+ 60%|██████    | 90/150 [18:39<11:54, 11.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 354, Spatial MSE: [(0, 1.1310269201203482e-06), (1, 5.2347768360050395e-06), (2, 1.0827405276359059e-05), (3, 6.250807473406894e-06), (4, 1.5227569747366942e-06), (5, 1.6931552409005235e-06), (6, 2.4685166408744408e-06), (7, 4.737734798254678e-06), (8, 2.3407098979078e-06), (9, 1.8330968032387318e-06), (10, 1.8859993815567577e-06), (11, 2.993693897224148e-06), (12, 4.112900569452904e-06), (13, 2.755535751930438e-05), (14, 1.6591895473538898e-05), (15, 5.609672825812595e-06), (16, 1.5272689779521897e-05), (17, 5.303268608258804e-06), (18, 2.20666806853842e-05), (19, 1.0650652257027104e-05), (20, 5.801342013000976e-06), (21, 7.850037945900112e-06), (22, 1.8126038412447087e-05), (23, 2.3527689336333424e-05), (24, 4.71631319669541e-05), (25, 0.00013892454444430768), (26, 0.00036203794297762215), (27, 0.00187510100658983)], Temporal MSE: [(0, 8.108114002425282e-07), (1, 3.211087005183799e-07), (2, 3.909032102455967e-07), (3, 5.44437341432058e-07), (4, 4.940445705869934e-07), (5, 6.7515208002078e-07), (6, 9.90170178738481e-07), (7, 1.1529572248036857e-06), (8, 5.441352186608128e-07), (9, 7.187045412138104e-07), (10, 1.0917869985860307e-06), (11, 1.493619038228644e-06), (12, 1.4954299558667117e-06), (13, 2.0170416519249557e-06), (14, 1.873918563433108e-06), (15, 1.730307531033759e-06), (16, 1.8074331364914542e-06), (17, 1.7107406620198162e-06), (18, 1.9785263702942757e-06), (19, 2.560316943345242e-06), (20, 3.6258575164538343e-06), (21, 4.4739508666680194e-06), (22, 6.390365797415143e-06), (23, 1.0061376087833196e-05), (24, 1.5067251297296025e-05), (25, 6.260332884266973e-05), (26, 2.0393259546835907e-05), (27, 4.920741957903374e-06)]
+
+ 61%|██████    | 91/150 [18:51<11:42, 11.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 348, Spatial MSE: [(0, 7.302749622795091e-07), (1, 1.1608129170781467e-05), (2, 1.3432183550321497e-05), (3, 1.0450457921251655e-05), (4, 1.6534208043594845e-05), (5, 2.3426731786457822e-05), (6, 3.728218507603742e-05), (7, 6.91231616656296e-05), (8, 0.00010021420894190669), (9, 0.0001338750298600644), (10, 0.00015260180225595832), (11, 0.00018278663628734648), (12, 0.00019469276594463736), (13, 0.0009028710192069411), (14, 0.002159914467483759), (15, 0.0015760065289214253), (16, 0.035845138132572174), (17, 0.00035135963116772473), (18, 0.0004119838122278452), (19, 0.0021864480804651976), (20, 0.0017974014626815915), (21, 0.0005476968945004046), (22, 0.0004766046768054366), (23, 0.0007418873719871044), (24, 0.0013658161042258143), (25, 0.0029658586718142033), (26, 0.005619378294795752), (27, 0.0013986978447064757)], Temporal MSE: [(0, 8.6469617599505e-07), (1, 3.0762618052904145e-07), (2, 1.472358349019487e-06), (3, 3.8405814848374575e-06), (4, 6.8911563175788615e-06), (5, 1.1341526260366663e-05), (6, 3.939104135497473e-05), (7, 2.4246608518296853e-05), (8, 2.322806540178135e-05), (9, 3.395431849639863e-05), (10, 6.738063530065119e-05), (11, 6.754203786840662e-05), (12, 6.733604823239148e-05), (13, 0.00011564737360458821), (14, 0.00021761906100437045), (15, 0.0001801015459932387), (16, 0.00019550189608708024), (17, 0.00017210951773449779), (18, 0.0001542316167615354), (19, 0.00022748741321265697), (20, 0.0003339866816531867), (21, 0.0002803971874527633), (22, 0.0002829389995895326), (23, 0.0003655008622445166), (24, 0.0010169753804802895), (25, 0.00180536019615829), (26, 0.00041021264041773975), (27, 2.6008105123764835e-05)]
+
+ 61%|██████▏   | 92/150 [19:03<11:38, 12.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 342, Spatial MSE: [(0, 6.488895110123849e-07), (1, 4.626490863302024e-06), (2, 9.276423043047544e-06), (3, 6.4314658629882615e-06), (4, 1.3724060181630193e-06), (5, 1.473800693929661e-06), (6, 1.6596860632489552e-06), (7, 3.244548224756727e-06), (8, 1.9276965304015903e-06), (9, 1.7833016272561508e-06), (10, 1.7862040522231837e-06), (11, 2.504075609977008e-06), (12, 4.2589090298861265e-06), (13, 8.319447806570679e-05), (14, 0.0001694028323981911), (15, 5.016740396968089e-05), (16, 0.004721305798739195), (17, 2.13438779610442e-05), (18, 1.3855010365659837e-05), (19, 0.00039700535126030445), (20, 0.0002900463296100497), (21, 2.1680838472093455e-05), (22, 3.7246194551698864e-05), (23, 3.397132968530059e-05), (24, 7.205643487395719e-05), (25, 0.00018700164218898863), (26, 0.00029756195726804435), (27, 0.00310310791246593)], Temporal MSE: [(0, 6.071684310882119e-07), (1, 3.176129155235685e-07), (2, 3.753638111447799e-07), (3, 5.721997808905144e-07), (4, 5.156845759302087e-07), (5, 7.31864815861627e-07), (6, 1.0763370710265008e-06), (7, 1.0672870303096715e-06), (8, 5.561752232097206e-07), (9, 6.774180860702472e-07), (10, 9.902736337608076e-07), (11, 1.5483032029806054e-06), (12, 2.033888904406922e-06), (13, 4.240687758283457e-06), (14, 5.628500275633996e-06), (15, 7.542975708929589e-06), (16, 6.834267878730316e-06), (17, 1.220040303451242e-05), (18, 7.734540304227266e-06), (19, 8.028549927985296e-06), (20, 6.951998329896014e-06), (21, 2.2000662283971906e-05), (22, 2.0680608940892853e-05), (23, 3.2701005693525076e-05), (24, 2.884366040234454e-05), (25, 7.343397010117769e-05), (26, 2.3996895833988674e-05), (27, 1.138203060691012e-05)]
+
+ 62%|██████▏   | 93/150 [19:14<10:55, 11.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 336, Spatial MSE: [(0, 7.331339020311134e-07), (1, 6.299611868598731e-06), (2, 9.327443876827601e-06), (3, 6.107945864641806e-06), (4, 4.114151579415193e-06), (5, 5.959916506981244e-06), (6, 1.5564950444968417e-05), (7, 1.2024245734210126e-05), (8, 1.7471547835157253e-05), (9, 3.9078775444068015e-05), (10, 7.315888069570065e-05), (11, 0.00010563438263488933), (12, 9.069540101336315e-05), (13, 0.00023624398454558104), (14, 0.00041550403693690896), (15, 0.0003827528562396765), (16, 0.005962215829640627), (17, 6.282782123889774e-05), (18, 0.0001464516099076718), (19, 0.00041171888005919755), (20, 0.0004594173515215516), (21, 0.00011199094296898693), (22, 0.00011253094271523878), (23, 0.00013810764357913285), (24, 0.0003843241138383746), (25, 0.0011655972339212894), (26, 0.0034317923709750175), (27, 0.003159924177452922)], Temporal MSE: [(0, 5.577558113145642e-07), (1, 3.125231842204812e-07), (2, 5.424478786153486e-07), (3, 1.366630385746248e-06), (4, 2.139785920007853e-06), (5, 2.9112859465385554e-06), (6, 4.404144874570193e-06), (7, 5.089048045192612e-06), (8, 3.901253421645379e-06), (9, 5.029228759667603e-06), (10, 8.649259143567178e-06), (11, 1.1433607141952962e-05), (12, 1.2485876140999608e-05), (13, 1.8630089471116662e-05), (14, 2.5104895030381158e-05), (15, 1.968223114090506e-05), (16, 1.6407853763666935e-05), (17, 1.3250552910903934e-05), (18, 1.770902963471599e-05), (19, 2.410907472949475e-05), (20, 4.912213626084849e-05), (21, 4.2094783566426486e-05), (22, 4.52180283900816e-05), (23, 6.135492731118575e-05), (24, 0.00015166527009569108), (25, 0.00042904436122626066), (26, 0.00012213815352879465), (27, 2.1977068172418512e-05)]
+
+ 63%|██████▎   | 94/150 [19:26<11:01, 11.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 330, Spatial MSE: [(0, 7.277884606082807e-07), (1, 6.0359966482792515e-06), (2, 1.1756408639485016e-05), (3, 6.085643690312281e-06), (4, 1.6427508171545924e-06), (5, 1.3948844070910127e-06), (6, 1.793117007764522e-06), (7, 3.0383134799194522e-06), (8, 1.98368661585846e-06), (9, 2.320943394806818e-06), (10, 7.092071427905466e-06), (11, 3.0943188903620467e-05), (12, 4.909632480121218e-05), (13, 0.0008977625402621925), (14, 0.0007083043456077576), (15, 0.00019521373906172812), (16, 0.004646337125450373), (17, 1.860005249909591e-05), (18, 1.8874165107263252e-05), (19, 0.0003324379213154316), (20, 0.00021474518871400505), (21, 3.2605988963041455e-05), (22, 8.093692304100841e-05), (23, 5.1982657168991864e-05), (24, 9.366581070935354e-05), (25, 0.00018542047473601997), (26, 0.0003588605613913387), (27, 0.006936251651495695)], Temporal MSE: [(0, 6.089476300985552e-07), (1, 2.947993493762624e-07), (2, 3.8366030707948084e-07), (3, 5.324276344254031e-07), (4, 5.182012046134332e-07), (5, 7.165056672420178e-07), (6, 9.772243174666073e-07), (7, 9.967541245714528e-07), (8, 5.577425667979696e-07), (9, 9.286684985454485e-07), (10, 4.8056513151095714e-06), (11, 6.813235813751817e-05), (12, 1.8699620341067202e-05), (13, 1.5745110431453213e-05), (14, 6.829799531260505e-06), (15, 1.9439861716818996e-05), (16, 7.167266176111298e-06), (17, 8.57567192724673e-06), (18, 1.4921293768566102e-05), (19, 1.8662254660739563e-05), (20, 3.3856340451166034e-05), (21, 1.4602311239286792e-05), (22, 1.57248923642328e-05), (23, 3.660901347757317e-05), (24, 4.683685983764008e-05), (25, 0.00016436324222013354), (26, 4.164820347796194e-05), (27, 2.1291198208928108e-05)]
+
+ 63%|██████▎   | 95/150 [19:40<11:15, 12.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 324, Spatial MSE: [(0, 7.327516868826933e-07), (1, 1.0343616850150283e-05), (2, 1.3170614693081006e-05), (3, 1.0196764378633816e-05), (4, 1.63326312758727e-05), (5, 2.275509359606076e-05), (6, 5.542262442759238e-05), (7, 0.00010712184302974492), (8, 0.00010640190885169432), (9, 0.00015893463569227606), (10, 0.00015977188013494015), (11, 0.0002136958937626332), (12, 0.00023894676996860653), (13, 0.0016805949853733182), (14, 0.0029289538506418467), (15, 0.0018109531374648213), (16, 0.04129745438694954), (17, 0.0003841836878564209), (18, 0.00042284358642064035), (19, 0.0023777091410011053), (20, 0.0019868663512170315), (21, 0.0007372155669145286), (22, 0.0005147148622199893), (23, 0.0007630478357896209), (24, 0.0014380743959918618), (25, 0.004389489535242319), (26, 0.008184263482689857), (27, 0.007243899162858725)], Temporal MSE: [(0, 8.753331712796353e-07), (1, 3.7197079905126884e-07), (2, 1.562395823384577e-06), (3, 4.019011612399481e-06), (4, 7.3855449045368005e-06), (5, 1.376205818814924e-05), (6, 5.356236215448007e-05), (7, 2.938386023743078e-05), (8, 2.4006403691601008e-05), (9, 3.3413245546398684e-05), (10, 6.555022991960868e-05), (11, 0.00013048617984168231), (12, 8.062762935878709e-05), (13, 0.00012881579459644854), (14, 0.00020490289898589253), (15, 0.00018957065185531974), (16, 0.0002003845147555694), (17, 0.00017686143110040575), (18, 0.0001632032945053652), (19, 0.00023283368500415236), (20, 0.00033225640072487295), (21, 0.00031091901473701), (22, 0.00027081937878392637), (23, 0.00035897851921617985), (24, 0.0008158026612363756), (25, 0.0014634362887591124), (26, 0.0004900948842987418), (27, 4.825130235985853e-05)]
+
+ 64%|██████▍   | 96/150 [19:53<11:25, 12.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 318, Spatial MSE: [(0, 5.739904622714675e-07), (1, 4.65557513962267e-06), (2, 8.132973562169354e-06), (3, 6.3682778090878855e-06), (4, 1.5413643268402666e-06), (5, 1.5863329281273764e-06), (6, 1.3720767810809775e-06), (7, 2.7678165679390077e-06), (8, 1.8356720374868019e-06), (9, 1.756822143761383e-06), (10, 2.397948719590204e-06), (11, 4.117657226743177e-06), (12, 9.483678695687559e-06), (13, 0.0003029754152521491), (14, 0.0004412591806612909), (15, 0.0001910333230625838), (16, 0.0047775451093912125), (17, 1.3300420505402144e-05), (18, 1.7066957298084162e-05), (19, 0.00026782529312185943), (20, 0.00015763333067297935), (21, 3.7904115743003786e-05), (22, 7.136286876630038e-05), (23, 4.699250712292269e-05), (24, 7.574186020065099e-05), (25, 0.00014417448255699128), (26, 0.00027165337814949453), (27, 0.005384513642638922)], Temporal MSE: [(0, 5.266867901809746e-07), (1, 2.882748901811283e-07), (2, 3.627210105605627e-07), (3, 5.464883656713937e-07), (4, 5.446146928989037e-07), (5, 7.436958640028024e-07), (6, 9.80802269623382e-07), (7, 9.773843885341194e-07), (8, 5.890367447136668e-07), (9, 6.999901529525232e-07), (10, 1.5019984402897535e-06), (11, 3.54083749698475e-06), (12, 4.82009136248962e-06), (13, 1.2199889170005918e-05), (14, 6.4718756220827345e-06), (15, 1.154485289589502e-05), (16, 4.778071343025658e-06), (17, 4.733054083772004e-06), (18, 1.3227935596660245e-05), (19, 1.4373965314007364e-05), (20, 1.4645598639617674e-05), (21, 1.2390210940793622e-05), (22, 1.3441775990941096e-05), (23, 3.307496081106365e-05), (24, 4.0801696741255e-05), (25, 0.00014487572479993105), (26, 3.7689111195504665e-05), (27, 1.979791522899177e-05)]
+
+ 65%|██████▍   | 97/150 [20:07<11:27, 12.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 312, Spatial MSE: [(0, 6.74002080813807e-07), (1, 5.496353878697846e-06), (2, 7.347689916059608e-06), (3, 5.439951110020047e-06), (4, 3.1960576052370016e-06), (5, 5.762220553151565e-06), (6, 1.3825241694576107e-05), (7, 1.1763403563236352e-05), (8, 1.705145950836595e-05), (9, 3.2527506846236065e-05), (10, 5.0379549065837637e-05), (11, 8.283406350528821e-05), (12, 7.60059047024697e-05), (13, 0.0003741449909284711), (14, 0.0005223016487434506), (15, 0.00023656107077840716), (16, 0.004810850601643324), (17, 3.1657669751439244e-05), (18, 3.704763003042899e-05), (19, 0.0002895658544730395), (20, 0.00018849605112336576), (21, 5.91710013395641e-05), (22, 0.00011885980347869918), (23, 0.00012709468137472868), (24, 0.00033009613980539143), (25, 0.001106578391045332), (26, 0.003049512393772602), (27, 0.005572455935180187)], Temporal MSE: [(0, 5.338613959793292e-07), (1, 2.9258757194838836e-07), (2, 5.459873477775545e-07), (3, 1.3459874708132702e-06), (4, 2.1780040242447285e-06), (5, 2.811672857205849e-06), (6, 4.1928451537387446e-06), (7, 4.730980890599312e-06), (8, 3.6027608985023107e-06), (9, 4.263788468961138e-06), (10, 7.350128726102412e-06), (11, 1.1138722584291827e-05), (12, 1.2624926966964267e-05), (13, 2.325775676581543e-05), (14, 2.3286285795620643e-05), (15, 2.151464650523849e-05), (16, 1.389038243360119e-05), (17, 1.103812155633932e-05), (18, 2.2317288312478922e-05), (19, 2.4846471205819398e-05), (20, 2.8075897716917098e-05), (21, 2.4991113605210558e-05), (22, 3.500414823065512e-05), (23, 6.0150163335492834e-05), (24, 0.0001433395518688485), (25, 0.00047507803537882864), (26, 0.0001367905642837286), (27, 3.253527756896801e-05)]
+
+ 65%|██████▌   | 98/150 [20:21<11:38, 13.43s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 306, Spatial MSE: [(0, 1.0790615760924993e-06), (1, 6.020234195602825e-06), (2, 9.33810315473238e-06), (3, 5.979508841846837e-06), (4, 1.4412056543733343e-06), (5, 1.768015863490291e-06), (6, 2.1910493614996085e-06), (7, 4.221242761559552e-06), (8, 2.2295287180895684e-06), (9, 2.0042823507537832e-06), (10, 2.2637000256509054e-06), (11, 3.4335105283389566e-06), (12, 4.4587427510123234e-06), (13, 3.9837195799918845e-05), (14, 8.383439853787422e-05), (15, 7.617220398969948e-05), (16, 0.0023803107906132936), (17, 9.755824976309668e-06), (18, 2.180949923058506e-05), (19, 0.00010880400805035606), (20, 6.0316444432828575e-05), (21, 2.8231677788426168e-05), (22, 5.3312836826080456e-05), (23, 3.861474760924466e-05), (24, 6.270530866459012e-05), (25, 0.00016613716434221715), (26, 0.0003081347676925361), (27, 0.002979157492518425)], Temporal MSE: [(0, 8.006206257960002e-07), (1, 3.3302512747468427e-07), (2, 4.000854971764056e-07), (3, 5.687435304935207e-07), (4, 5.591975309471309e-07), (5, 7.646555673090916e-07), (6, 1.0686172799978522e-06), (7, 1.2105999758205144e-06), (8, 6.137063337519066e-07), (9, 7.827369472579448e-07), (10, 1.410858544659277e-06), (11, 1.7426432350475807e-06), (12, 1.6158282960532233e-06), (13, 3.170329364365898e-06), (14, 4.562567482935265e-06), (15, 5.561968009715201e-06), (16, 3.195881390638533e-06), (17, 3.4861652693507494e-06), (18, 7.384524451481411e-06), (19, 8.99888800631743e-06), (20, 1.045968383550644e-05), (21, 9.925785889208782e-06), (22, 1.0905725503107533e-05), (23, 2.3832071747165173e-05), (24, 2.9374161385931075e-05), (25, 0.00012837837857659906), (26, 2.8127591576776467e-05), (27, 1.2324260751483962e-05)]
+
+ 66%|██████▌   | 99/150 [20:35<11:25, 13.44s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 300, Spatial MSE: [(0, 9.616122724764864e-07), (1, 9.960716852219775e-06), (2, 1.1300918231427204e-05), (3, 9.766225957719143e-06), (4, 1.6677886378602125e-05), (5, 2.246487383672502e-05), (6, 4.1747342038434e-05), (7, 9.307529398938641e-05), (8, 9.921049058903009e-05), (9, 0.00016632239567115903), (10, 0.00016199845413211733), (11, 0.0001838343305280432), (12, 0.00018912968516815454), (13, 0.00023922426044009626), (14, 0.00047070099390111864), (15, 0.0005394218605943024), (16, 0.0027446644380688667), (17, 0.00031064264476299286), (18, 0.0004515060572884977), (19, 0.0006527877412736416), (20, 0.001051690080203116), (21, 0.0006762667908333242), (22, 0.00042588674114085734), (23, 0.000712468521669507), (24, 0.0012771703768521547), (25, 0.004113797098398209), (26, 0.008752585388720036), (27, 0.0030317550990730524)], Temporal MSE: [(0, 1.1258254062340711e-06), (1, 4.4779741870115686e-07), (2, 1.6764091697041295e-06), (3, 3.993066002294654e-06), (4, 6.947940164536703e-06), (5, 1.2507821338658687e-05), (6, 4.3025753257097676e-05), (7, 2.714583206397947e-05), (8, 2.289960866619367e-05), (9, 3.2733394618844613e-05), (10, 5.684642746928148e-05), (11, 5.688339297194034e-05), (12, 5.300789780449122e-05), (13, 9.07600115169771e-05), (14, 0.00017153378576040268), (15, 0.000142794699058868), (16, 0.00018002174329012632), (17, 0.0001439015759387985), (18, 0.00012664537644013762), (19, 0.00019629466987680644), (20, 0.000287433183984831), (21, 0.00024550766102038324), (22, 0.0002246599324280396), (23, 0.0002972876245621592), (24, 0.0007706941687501967), (25, 0.0012988111702725291), (26, 0.0005044911522418261), (27, 3.7959918699925765e-05)]
+
+ 67%|██████▋   | 100/150 [20:50<11:38, 13.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 294, Spatial MSE: [(0, 6.544653956552793e-07), (1, 5.6794074225763325e-06), (2, 9.427843906451017e-06), (3, 6.3943048189685214e-06), (4, 1.6065508816609508e-06), (5, 1.706529133116419e-06), (6, 1.5800636674612178e-06), (7, 3.206840347047546e-06), (8, 2.124862021446461e-06), (9, 2.207261559306062e-06), (10, 5.403954673965927e-06), (11, 1.8727178030530922e-05), (12, 3.9700873458059505e-05), (13, 0.0008641958702355623), (14, 0.0007111606537364423), (15, 0.00019023606728296727), (16, 0.0045239911414682865), (17, 1.7828768250183202e-05), (18, 1.5609790352755226e-05), (19, 0.0003259099321439862), (20, 0.00020091036276426166), (21, 3.543773709679954e-05), (22, 7.456979074049741e-05), (23, 5.13841696374584e-05), (24, 8.758036710787565e-05), (25, 0.00016386130300816149), (26, 0.0002907078596763313), (27, 0.006606312468647957)], Temporal MSE: [(0, 5.232236617302988e-07), (1, 2.834236454418715e-07), (2, 3.8044149164306873e-07), (3, 5.753355480919708e-07), (4, 6.003315320413094e-07), (5, 8.242992066698207e-07), (6, 1.1829131381091429e-06), (7, 1.1536442343640374e-06), (8, 6.713331117680355e-07), (9, 9.155232874036301e-07), (10, 3.645252490969142e-06), (11, 4.457573595573194e-05), (12, 1.72970521816751e-05), (13, 1.6075337043730542e-05), (14, 6.136286629043752e-06), (15, 1.306691046920605e-05), (16, 5.825242624268867e-06), (17, 4.970106601831503e-06), (18, 1.466239700675942e-05), (19, 1.5786697986186482e-05), (20, 1.390625493513653e-05), (21, 1.316982161370106e-05), (22, 1.485437860537786e-05), (23, 3.621705764089711e-05), (24, 4.460276250028983e-05), (25, 0.00017722729535307735), (26, 4.155514034209773e-05), (27, 2.0497329387580976e-05)]
+
+ 67%|██████▋   | 101/150 [21:03<11:06, 13.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 288, Spatial MSE: [(0, 7.131204142751812e-07), (1, 5.831128419231391e-06), (2, 7.408664259855868e-06), (3, 5.033789420849644e-06), (4, 3.143706635455601e-06), (5, 6.188055067468667e-06), (6, 1.236662774317665e-05), (7, 1.3373842193686869e-05), (8, 1.7265861970372498e-05), (9, 3.1980162020772696e-05), (10, 5.1762373914243653e-05), (11, 9.830462659010664e-05), (12, 0.00010468695109011605), (13, 0.0009300364763475955), (14, 0.0007954894681461155), (15, 0.0002301052154507488), (16, 0.00455080671235919), (17, 3.5778637538896874e-05), (18, 3.6560973967425525e-05), (19, 0.0003464241453912109), (20, 0.00023367877292912453), (21, 6.078809383325279e-05), (22, 0.00012705948029179126), (23, 0.0001352551335003227), (24, 0.000360254809493199), (25, 0.0012434752425178885), (26, 0.0035239816643297672), (27, 0.006536658853292465)], Temporal MSE: [(0, 5.790096224700392e-07), (1, 3.2583071174485667e-07), (2, 5.382510153140174e-07), (3, 1.259167675016215e-06), (4, 2.0903257791360375e-06), (5, 2.8784970709239133e-06), (6, 4.439776148501551e-06), (7, 5.347934802557575e-06), (8, 4.05828041039058e-06), (9, 4.874003479926614e-06), (10, 9.85187853075331e-06), (11, 5.251081529422663e-05), (12, 2.5795392502914183e-05), (13, 2.867337389034219e-05), (14, 2.4232860596384853e-05), (15, 2.417080213490408e-05), (16, 1.56403002620209e-05), (17, 1.1492329576867633e-05), (18, 2.300393498444464e-05), (19, 2.688820859475527e-05), (20, 2.909935574280098e-05), (21, 2.8032465706928633e-05), (22, 3.829005072475411e-05), (23, 6.369712355080992e-05), (24, 0.0001504760584793985), (25, 0.0004750866210088134), (26, 0.00015559981693513691), (27, 3.4124321246054024e-05)]
+
+ 68%|██████▊   | 102/150 [21:16<10:49, 13.53s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 282, Spatial MSE: [(0, 7.028303912193223e-07), (1, 6.267809567361837e-06), (2, 9.046528248290997e-06), (3, 5.837639491801383e-06), (4, 1.5616562905051978e-06), (5, 1.7243431784663699e-06), (6, 1.5670584616600536e-06), (7, 3.0599828733102186e-06), (8, 2.1523335362871876e-06), (9, 2.0107811451453017e-06), (10, 2.49434788202052e-06), (11, 3.7403863188956166e-06), (12, 5.002876150683733e-06), (13, 5.515488737728447e-05), (14, 6.691466842312366e-05), (15, 4.629521572496742e-05), (16, 0.0018805526196956635), (17, 9.670394319982734e-06), (18, 1.4552733773598447e-05), (19, 9.591315756551921e-05), (20, 4.692803486250341e-05), (21, 3.602753713494167e-05), (22, 7.104725955286995e-05), (23, 4.195464862277731e-05), (24, 7.147901487769559e-05), (25, 0.00014095832011662424), (26, 0.00025585704133845866), (27, 0.0019933045841753483)], Temporal MSE: [(0, 6.644066843364271e-07), (1, 3.253745717302081e-07), (2, 4.089216929514805e-07), (3, 5.687026600753597e-07), (4, 5.883148901375534e-07), (5, 8.001247806532774e-07), (6, 1.0159020575883915e-06), (7, 1.1182713706148206e-06), (8, 6.461622206188622e-07), (9, 7.790452514200297e-07), (10, 1.5031738485049573e-06), (11, 1.8602780755827553e-06), (12, 2.082318815155304e-06), (13, 3.501087348922738e-06), (14, 4.365020686236676e-06), (15, 5.66407788937795e-06), (16, 3.992306119471323e-06), (17, 4.261994490661891e-06), (18, 7.457329047610983e-06), (19, 9.713638974062633e-06), (20, 1.3175639651308302e-05), (21, 1.1061102668463718e-05), (22, 1.2069705007888842e-05), (23, 2.5924447982106358e-05), (24, 2.9981501938891597e-05), (25, 9.56844087340869e-05), (26, 2.6101231924258173e-05), (27, 6.896005288581364e-06)]
+
+ 69%|██████▊   | 103/150 [21:30<10:35, 13.52s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 276, Spatial MSE: [(0, 7.607461611769395e-07), (1, 7.445380560966441e-06), (2, 1.3808778930979315e-05), (3, 1.1357928087818436e-05), (4, 1.5124480341910385e-05), (5, 2.495182525308337e-05), (6, 3.377868051757105e-05), (7, 6.986367952777073e-05), (8, 9.446335025131702e-05), (9, 0.00011732843267964199), (10, 0.00012079230509698391), (11, 0.0001421603374183178), (12, 0.00015947881911415607), (13, 0.00039186934009194374), (14, 0.0008494994253851473), (15, 0.0007431288831867278), (16, 0.014239862561225891), (17, 0.00027035034145228565), (18, 0.000321138504659757), (19, 0.0009632788714952767), (20, 0.0008495340007357299), (21, 0.0005854687769897282), (22, 0.0004663097206503153), (23, 0.0007411752594634891), (24, 0.0012907104101032019), (25, 0.003698384389281273), (26, 0.006870950572192669), (27, 0.002135711954906583)], Temporal MSE: [(0, 8.943811735662166e-07), (1, 4.09562659342555e-07), (2, 1.8358631450610119e-06), (3, 4.774482931679813e-06), (4, 7.95673895481741e-06), (5, 1.2801830962416716e-05), (6, 3.0265528039308265e-05), (7, 2.333650627406314e-05), (8, 2.2245631043915637e-05), (9, 3.170190757373348e-05), (10, 5.6588523875689134e-05), (11, 5.232339390204288e-05), (12, 5.188119757804088e-05), (13, 8.788823470240459e-05), (14, 0.00016184645937755704), (15, 0.00012855841487180442), (16, 0.00015816553786862642), (17, 0.00013393888366408646), (18, 0.00012057634012307972), (19, 0.00019156518101226538), (20, 0.0002714121656026691), (21, 0.00023283618793357164), (22, 0.00023049300943966955), (23, 0.0003083139890804887), (24, 0.0008268082747235894), (25, 0.0014179549179971218), (26, 0.0005061848205514252), (27, 3.701563400682062e-05)]
+
+ 69%|██████▉   | 104/150 [21:44<10:32, 13.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 270, Spatial MSE: [(0, 7.231639642668597e-07), (1, 6.259248038986698e-06), (2, 9.315082024841104e-06), (3, 6.2736362451687455e-06), (4, 1.5391788110719062e-06), (5, 1.8870027815864887e-06), (6, 1.6379033240809804e-06), (7, 3.0531637094100006e-06), (8, 2.2733015612175222e-06), (9, 2.107129603245994e-06), (10, 3.817674496531254e-06), (11, 8.566252290620469e-06), (12, 2.6541989427641965e-05), (13, 0.0008086008601821959), (14, 0.0007124082185328007), (15, 0.00020862037490587682), (16, 0.0045182001776993275), (17, 2.002515429921914e-05), (18, 1.7002517779474147e-05), (19, 0.00032019452191889286), (20, 0.00019248003081884235), (21, 4.633237767848186e-05), (22, 0.0001101366724469699), (23, 6.186217069625854e-05), (24, 9.873219096334651e-05), (25, 0.0001868349645519629), (26, 0.00030309369321912527), (27, 0.0064664860256016254)], Temporal MSE: [(0, 6.133637953098514e-07), (1, 3.37166625286045e-07), (2, 4.261359549673216e-07), (3, 6.043296139068843e-07), (4, 6.624117077080882e-07), (5, 9.113858823184273e-07), (6, 1.1739375622710213e-06), (7, 1.2115956451452803e-06), (8, 7.216958692879416e-07), (9, 8.918914318201132e-07), (10, 2.3470654468837893e-06), (11, 1.6477775716339238e-05), (12, 1.5716259440523572e-05), (13, 1.6699586922186427e-05), (14, 7.806147550581954e-06), (15, 1.430138490832178e-05), (16, 5.836237050971249e-06), (17, 5.269369012239622e-06), (18, 1.5057836208143272e-05), (19, 1.6384929040214047e-05), (20, 1.563714067742694e-05), (21, 1.79104354174342e-05), (22, 1.8668468328542076e-05), (23, 4.4659343984676525e-05), (24, 5.004964623367414e-05), (25, 0.00017959304386749864), (26, 4.578744119498879e-05), (27, 2.125101673300378e-05)]
+
+ 70%|███████   | 105/150 [21:57<10:03, 13.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 264, Spatial MSE: [(0, 8.412147280978388e-07), (1, 7.28971826902125e-06), (2, 8.248823178291786e-06), (3, 5.647946181852603e-06), (4, 3.97893609260791e-06), (5, 7.595585429953644e-06), (6, 1.7247039068024606e-05), (7, 1.535721639811527e-05), (8, 1.802420592866838e-05), (9, 3.838256816379726e-05), (10, 6.53846946079284e-05), (11, 0.00011470491153886542), (12, 0.00010287077748216689), (13, 0.0008847875869832933), (14, 0.0008040476823225617), (15, 0.0002640383318066597), (16, 0.004558653570711613), (17, 4.35170040873345e-05), (18, 0.0001183128115371801), (19, 0.0004399506433401257), (20, 0.00042340136133134365), (21, 0.00011916126095457003), (22, 0.00019369597430340946), (23, 0.00018423628353048116), (24, 0.0004991020541638136), (25, 0.0019285579910501838), (26, 0.004512464627623558), (27, 0.0065329005010426044)], Temporal MSE: [(0, 8.599688499089098e-07), (1, 4.2025217794616765e-07), (2, 6.402431154128863e-07), (3, 1.3572162060881965e-06), (4, 2.1959140212857164e-06), (5, 3.4945726383739384e-06), (6, 5.340520147001371e-06), (7, 6.896770173625555e-06), (8, 5.1511497076717205e-06), (9, 6.3720390244270675e-06), (10, 1.079609319276642e-05), (11, 2.7333462639944628e-05), (12, 2.637963552842848e-05), (13, 3.235710028093308e-05), (14, 2.9539889510488138e-05), (15, 2.737646718742326e-05), (16, 1.8452519725542516e-05), (17, 1.3944355487183202e-05), (18, 2.7369907911634073e-05), (19, 3.3199667086591944e-05), (20, 3.9599137380719185e-05), (21, 3.670165460789576e-05), (22, 4.818263550987467e-05), (23, 8.08668410172686e-05), (24, 0.00018836517119780183), (25, 0.0006032316596247256), (26, 0.0001813402632251382), (27, 3.852882582577877e-05)]
+
+ 71%|███████   | 106/150 [22:11<10:00, 13.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 258, Spatial MSE: [(0, 6.584677407772688e-07), (1, 5.891342425456969e-06), (2, 9.375750778417569e-06), (3, 5.7991705944004934e-06), (4, 1.8588820012155338e-06), (5, 1.9590993360907305e-06), (6, 1.6478498992000823e-06), (7, 2.617420932438108e-06), (8, 2.3738707568554673e-06), (9, 2.483948492226773e-06), (10, 2.3237805635289988e-06), (11, 3.738578698175843e-06), (12, 4.060607807332417e-06), (13, 2.6020925361081026e-05), (14, 1.9395645722397603e-05), (15, 8.93217929842649e-06), (16, 0.00042916712118312716), (17, 6.908077466505347e-06), (18, 1.5234976672218181e-05), (19, 5.4390140576288104e-05), (20, 1.6525880710105412e-05), (21, 3.756575824809261e-05), (22, 3.1472696718992665e-05), (23, 2.71631361101754e-05), (24, 5.030737884226255e-05), (25, 0.0001041471041389741), (26, 0.00020808310364373028), (27, 0.001325056073255837)], Temporal MSE: [(0, 7.362000360444654e-07), (1, 3.8501744370478264e-07), (2, 4.783655640494544e-07), (3, 6.538211891893297e-07), (4, 6.480884735537984e-07), (5, 8.981343171399203e-07), (6, 1.1990175607934361e-06), (7, 1.2181326383142732e-06), (8, 7.562180712739064e-07), (9, 8.787197884885245e-07), (10, 1.4287355725173256e-06), (11, 1.7727908243614365e-06), (12, 1.6536688463020255e-06), (13, 2.658738594618626e-06), (14, 2.5585718503862154e-06), (15, 3.4291304018552182e-06), (16, 4.4091002564528026e-06), (17, 4.399344106786884e-06), (18, 6.061249678168679e-06), (19, 9.101567229663488e-06), (20, 1.3890528862248175e-05), (21, 7.0088926804601215e-06), (22, 8.569981218897738e-06), (23, 1.327951395069249e-05), (24, 1.7373777154716663e-05), (25, 3.158221807098016e-05), (26, 2.132505687768571e-05), (27, 3.676483402159647e-06)]
+
+ 71%|███████▏  | 107/150 [22:22<09:18, 12.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 252, Spatial MSE: [(0, 7.351960107371269e-07), (1, 7.811821888026316e-06), (2, 1.1198206266271882e-05), (3, 9.891319677990396e-06), (4, 1.6782254533609375e-05), (5, 2.3134354705689475e-05), (6, 3.358051253599115e-05), (7, 8.043387060752138e-05), (8, 9.304740524385124e-05), (9, 0.00015266692207660526), (10, 0.00012602504284586757), (11, 0.00015304956468753517), (12, 0.00017569736519362777), (13, 0.0011255400022491813), (14, 0.003253411501646042), (15, 0.0023052901960909367), (16, 0.057767439633607864), (17, 0.00032716672285459936), (18, 0.000436239322880283), (19, 0.0031628780998289585), (20, 0.0027007891330868006), (21, 0.0008151978254318237), (22, 0.0005553046357817948), (23, 0.0009533138363622129), (24, 0.0016868093516677618), (25, 0.005381891969591379), (26, 0.009215917438268661), (27, 0.0018590318504720926)], Temporal MSE: [(0, 9.741752364789136e-07), (1, 4.2535666011644935e-07), (2, 1.7168293879876728e-06), (3, 4.24353129346855e-06), (4, 7.58683154344908e-06), (5, 1.2807964594685473e-05), (6, 3.404507879167795e-05), (7, 2.6110821636393666e-05), (8, 2.2383905161404982e-05), (9, 3.125530565739609e-05), (10, 5.219475497142412e-05), (11, 5.7718461903277785e-05), (12, 6.877341365907341e-05), (13, 0.00011124218144686893), (14, 0.00017912808107212186), (15, 0.0001610757753951475), (16, 0.00018558328156359494), (17, 0.0001569343003211543), (18, 0.00014898949302732944), (19, 0.0002342557709198445), (20, 0.00035676287370733917), (21, 0.00034111819695681334), (22, 0.0002996469847857952), (23, 0.0003947295481339097), (24, 0.0008578265551477671), (25, 0.001273924601264298), (26, 0.0005798225174658), (27, 4.456683382159099e-05)]
+
+ 72%|███████▏  | 108/150 [22:32<08:22, 11.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 246, Spatial MSE: [(0, 7.630624168086797e-07), (1, 6.811565526732011e-06), (2, 9.333484740636777e-06), (3, 6.176860551931895e-06), (4, 1.5490990108446567e-06), (5, 1.8863803461499629e-06), (6, 1.5866680769249797e-06), (7, 3.4848519590013893e-06), (8, 2.4443063466605963e-06), (9, 1.7986862985708285e-06), (10, 2.115330744345556e-06), (11, 3.264669203417725e-06), (12, 6.237050911295228e-06), (13, 0.00016922202485147864), (14, 0.0002715092559810728), (15, 0.00019999560026917607), (16, 0.007230943534523249), (17, 3.17358280881308e-05), (18, 6.646733527304605e-05), (19, 0.0007362817414104939), (20, 0.0005764925153926015), (21, 5.0959042710019276e-05), (22, 0.00010607588774291798), (23, 5.379560752771795e-05), (24, 0.00011411046580178663), (25, 0.00027202925411984324), (26, 0.0002480137045495212), (27, 0.004335480742156506)], Temporal MSE: [(0, 6.279749413806712e-07), (1, 3.175848348746513e-07), (2, 4.275968876754632e-07), (3, 5.895971071367967e-07), (4, 6.863238581900077e-07), (5, 9.360197736896225e-07), (6, 1.2714267541014124e-06), (7, 1.2490382914620568e-06), (8, 7.810472766323073e-07), (9, 9.190337095787982e-07), (10, 1.2687979733527754e-06), (11, 2.6020697987405583e-06), (12, 4.063941105414415e-06), (13, 8.163182428688742e-06), (14, 1.4734931937709916e-05), (15, 1.1731158338079695e-05), (16, 1.0870161531784106e-05), (17, 1.4550106243405025e-05), (18, 1.896810863399878e-05), (19, 1.7392047084285878e-05), (20, 1.3143036994733848e-05), (21, 4.5741671783616766e-05), (22, 4.6403285523410887e-05), (23, 6.347875023493543e-05), (24, 4.877236642641947e-05), (25, 9.18796649784781e-05), (26, 3.205292159691453e-05), (27, 2.5634115445427597e-05)]
+
+ 73%|███████▎  | 109/150 [22:42<07:48, 11.43s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 240, Spatial MSE: [(0, 9.390588502355968e-07), (1, 6.3280926951847505e-06), (2, 1.0767552339530084e-05), (3, 5.63338062420371e-06), (4, 3.854858277918538e-06), (5, 8.428573892160784e-06), (6, 1.4067822121432982e-05), (7, 1.5380135664599948e-05), (8, 1.9398128642933443e-05), (9, 3.6578410799847916e-05), (10, 5.5024571338435635e-05), (11, 9.447269258089364e-05), (12, 7.677079702261835e-05), (13, 0.0003012622764799744), (14, 0.0005997405969537795), (15, 0.0002367776323808357), (16, 0.011015698313713074), (17, 7.48812235542573e-05), (18, 0.0001562332472531125), (19, 0.0006921402527950704), (20, 0.0004800418682862073), (21, 0.00010038608888862655), (22, 0.00020428707648534328), (23, 0.00020395737374201417), (24, 0.0004984866245649755), (25, 0.0018884632736444473), (26, 0.0041825855150818825), (27, 0.0047408160753548145)], Temporal MSE: [(0, 8.24945971089619e-07), (1, 4.6211809490159794e-07), (2, 6.610438276766217e-07), (3, 1.4150623428577092e-06), (4, 2.265413741042721e-06), (5, 3.4236015835631406e-06), (6, 5.622500793833751e-06), (7, 6.789399321860401e-06), (8, 5.262086688162526e-06), (9, 6.200392817845568e-06), (10, 9.837117431743536e-06), (11, 1.4822204320807941e-05), (12, 1.562599209137261e-05), (13, 2.4529303118470125e-05), (14, 3.307967199361883e-05), (15, 3.1220217351801693e-05), (16, 2.7566136850509793e-05), (17, 2.9657543564098887e-05), (18, 2.6972535124514252e-05), (19, 2.7111560484627262e-05), (20, 4.036818063468672e-05), (21, 5.978604531264864e-05), (22, 7.368656224571168e-05), (23, 0.00010002675116993487), (24, 0.00018753347103483975), (25, 0.00041077350033447146), (26, 0.00018576218280941248), (27, 4.6096578444121405e-05)]
+
+ 73%|███████▎  | 110/150 [22:52<07:19, 10.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 234, Spatial MSE: [(0, 6.782740342714533e-07), (1, 6.569082415808225e-06), (2, 9.67112646321766e-06), (3, 5.873217560292687e-06), (4, 1.889974441837694e-06), (5, 2.3274226350622484e-06), (6, 1.7738586848281557e-06), (7, 3.579990334401373e-06), (8, 2.97386031888891e-06), (9, 2.778262796709896e-06), (10, 7.782105058140587e-06), (11, 2.976846917590592e-05), (12, 5.50894474145025e-05), (13, 0.0010571841849014163), (14, 0.0011152237420901656), (15, 0.0003073240222875029), (16, 0.013651035726070404), (17, 8.181398879969493e-05), (18, 3.916508285328746e-05), (19, 0.0011523052817210555), (20, 0.0008332498255185783), (21, 0.00015183848154265434), (22, 0.00016939926717896014), (23, 9.254767064703628e-05), (24, 0.00017220256268046796), (25, 0.00034218764631077647), (26, 0.00034790305653586984), (27, 0.010410303249955177)], Temporal MSE: [(0, 7.795075589456246e-07), (1, 4.271208808859228e-07), (2, 5.059807222096424e-07), (3, 6.82267284446425e-07), (4, 7.576389293717511e-07), (5, 1.0085989288199926e-06), (6, 1.4579881053577992e-06), (7, 1.4785993016630528e-06), (8, 9.273388741348754e-07), (9, 1.283223014070245e-06), (10, 5.3903231673757546e-06), (11, 7.367267244262621e-05), (12, 2.443721859890502e-05), (13, 2.252953709103167e-05), (14, 1.528704524389468e-05), (15, 4.106179767404683e-05), (16, 2.427661274850834e-05), (17, 2.071736889774911e-05), (18, 2.5640107196522877e-05), (19, 2.5517383619444445e-05), (20, 3.459963045315817e-05), (21, 4.778950824402273e-05), (22, 6.394590309355408e-05), (23, 9.228669659933075e-05), (24, 8.363356755580753e-05), (25, 0.0001583663688506931), (26, 5.4750023991800845e-05), (27, 4.412731868796982e-05)]
+
+ 74%|███████▍  | 111/150 [23:02<06:59, 10.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 228, Spatial MSE: [(0, 7.442936862389615e-07), (1, 8.829241778585128e-06), (2, 1.4906750948284753e-05), (3, 9.973330634238664e-06), (4, 1.7101387129514478e-05), (5, 2.3701266400166787e-05), (6, 3.189338531228714e-05), (7, 7.681546412641183e-05), (8, 9.31440808926709e-05), (9, 0.0001534180046292022), (10, 0.00013231502089183778), (11, 0.00017675597337074578), (12, 0.0002068063768092543), (13, 0.0015871318755671382), (14, 0.002559901447966695), (15, 0.0014417781494557858), (16, 0.03769905865192413), (17, 0.0003860181022901088), (18, 0.0005399052752181888), (19, 0.0036538024432957172), (20, 0.003829484572634101), (21, 0.0009925891645252705), (22, 0.0006633356679230928), (23, 0.001003077020868659), (24, 0.0017794641898944974), (25, 0.006173237692564726), (26, 0.010866320692002773), (27, 0.010296812281012535)], Temporal MSE: [(0, 9.55883251663181e-07), (1, 4.808091489394428e-07), (2, 1.8798658629748388e-06), (3, 4.441154487722088e-06), (4, 7.648342034372035e-06), (5, 1.2831004823965486e-05), (6, 2.919379221566487e-05), (7, 2.687538471946027e-05), (8, 2.2629788873018697e-05), (9, 3.2302370527759194e-05), (10, 5.4969481425359845e-05), (11, 0.00012022438022540882), (12, 7.659390394110233e-05), (13, 0.00011577732220757753), (14, 0.00018209898553323), (15, 0.00018308822473045439), (16, 0.00020936995861120522), (17, 0.00016829189553391188), (18, 0.0001576797803863883), (19, 0.0002549471682868898), (20, 0.000366939086234197), (21, 0.0003512277326080948), (22, 0.0003113382263109088), (23, 0.0004296383121982217), (24, 0.0009245305554941297), (25, 0.0013771401718258858), (26, 0.0006570523837581277), (27, 8.917188097257167e-05)]
+
+ 75%|███████▍  | 112/150 [23:13<06:51, 10.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 222, Spatial MSE: [(0, 7.532131007792486e-07), (1, 5.8553728194965515e-06), (2, 1.0780368938867468e-05), (3, 6.314297479548259e-06), (4, 1.9526382857293356e-06), (5, 2.9871350761823123e-06), (6, 1.9300332496641204e-06), (7, 4.378440735308686e-06), (8, 3.28276746586198e-06), (9, 2.4355138066312065e-06), (10, 2.996196371896076e-06), (11, 6.6554030127008446e-06), (12, 1.978706131922081e-05), (13, 0.0006815516971983016), (14, 0.0007467236137017608), (15, 0.0003162180364597589), (16, 0.00960343424230814), (17, 4.205551522318274e-05), (18, 6.78251453791745e-05), (19, 0.0007482029614038765), (20, 0.0005202526808716357), (21, 5.9119796787854284e-05), (22, 0.000115478593215812), (23, 7.878357428126037e-05), (24, 0.00014663496403954923), (25, 0.00028560005011968315), (26, 0.0003111129626631737), (27, 0.008222958073019981)], Temporal MSE: [(0, 9.174582942250709e-07), (1, 4.920357241644524e-07), (2, 5.974257533125638e-07), (3, 7.713488230365328e-07), (4, 8.773752711022098e-07), (5, 1.1742208698706236e-06), (6, 1.5428058759425767e-06), (7, 1.6382113017243682e-06), (8, 1.0349102694817702e-06), (9, 1.2314551440795185e-06), (10, 1.8628279576660134e-06), (11, 1.2118360245949589e-05), (12, 1.3985893019707873e-05), (13, 1.765585329849273e-05), (14, 1.3098166164127178e-05), (15, 2.0527128071989864e-05), (16, 1.0746592124633025e-05), (17, 1.37572669700603e-05), (18, 2.1517042114282958e-05), (19, 1.8126445866073482e-05), (20, 1.4375673345057294e-05), (21, 4.4913263991475105e-05), (22, 4.824247662327252e-05), (23, 7.276316318893805e-05), (24, 7.207559247035533e-05), (25, 0.00012023927411064506), (26, 5.2806124585913494e-05), (27, 3.321702024550177e-05)]
+
+ 75%|███████▌  | 113/150 [23:25<06:47, 11.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 216, Spatial MSE: [(0, 6.904218139425211e-07), (1, 6.671202754660044e-06), (2, 1.2774703463946935e-05), (3, 5.42536326975096e-06), (4, 3.989250217273366e-06), (5, 8.239628186856862e-06), (6, 1.3817757462675218e-05), (7, 1.8452736185281537e-05), (8, 1.9838680600514635e-05), (9, 4.021598942927085e-05), (10, 5.98494807491079e-05), (11, 0.00010198438394581899), (12, 9.257310739485547e-05), (13, 0.0008571958169341087), (14, 0.0013416865840554237), (15, 0.0007302462472580373), (16, 0.017065169289708138), (17, 9.640371717978269e-05), (18, 9.720791422296315e-05), (19, 0.0009922771714627743), (20, 0.0009489706135354936), (21, 0.00011936155351577327), (22, 0.00021720236691180617), (23, 0.00024377103545702994), (24, 0.0005923006683588028), (25, 0.0022486646194010973), (26, 0.003966228570789099), (27, 0.008078517392277718)], Temporal MSE: [(0, 7.314012577808171e-07), (1, 4.325227109802654e-07), (2, 6.421567491088354e-07), (3, 1.3710753137274878e-06), (4, 2.4604139525763458e-06), (5, 3.7040415463707177e-06), (6, 6.04446040597395e-06), (7, 6.880359251226764e-06), (8, 5.47284025742556e-06), (9, 6.520346232719021e-06), (10, 9.590844456397463e-06), (11, 2.3804852389730513e-05), (12, 2.5726967578521e-05), (13, 3.7454643461387604e-05), (14, 3.542372724041343e-05), (15, 4.162008553976193e-05), (16, 2.2737649487680756e-05), (17, 1.898644222819712e-05), (18, 4.092052040505223e-05), (19, 4.619782339432277e-05), (20, 4.9886431952472776e-05), (21, 5.4829019063618034e-05), (22, 7.480567001039162e-05), (23, 0.00011003344843629748), (24, 0.00023994327057152987), (25, 0.0006068672519177198), (26, 0.00021681289945263416), (27, 5.361549847293645e-05)]
+
+ 76%|███████▌  | 114/150 [23:38<07:03, 11.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 210, Spatial MSE: [(0, 9.01054761470732e-07), (1, 7.732419362582732e-06), (2, 1.0325253242626786e-05), (3, 5.819274974783184e-06), (4, 1.930474809341831e-06), (5, 3.392722874195897e-06), (6, 2.056691755569773e-06), (7, 5.011345365346642e-06), (8, 3.708601070684381e-06), (9, 2.7111818781122565e-06), (10, 3.4233105452585733e-06), (11, 1.0668819413695019e-05), (12, 2.7525704354047775e-05), (13, 0.0007540543447248638), (14, 0.000601212028414011), (15, 0.00026642181910574436), (16, 0.006545305252075195), (17, 2.767457590380218e-05), (18, 7.740918226772919e-05), (19, 0.00032192832441069186), (20, 0.00014030872262082994), (21, 7.961184746818617e-05), (22, 0.00015376214287243783), (23, 9.3766902864445e-05), (24, 0.00015379699470940977), (25, 0.0003149264375679195), (26, 0.0003517394361551851), (27, 0.0071631912142038345)], Temporal MSE: [(0, 9.157999443232256e-07), (1, 4.964049935551884e-07), (2, 6.000522034810274e-07), (3, 7.253037210830371e-07), (4, 8.354281249012274e-07), (5, 1.1188197959199897e-06), (6, 1.4778989907426876e-06), (7, 1.8673761132959044e-06), (8, 1.0789054840643075e-06), (9, 1.2670924434132758e-06), (10, 2.443472112645395e-06), (11, 2.602596396172885e-05), (12, 1.8191032722825184e-05), (13, 1.5217001418932341e-05), (14, 1.1923953934456222e-05), (15, 2.4860581106622703e-05), (16, 9.902821147989016e-06), (17, 1.175179841084173e-05), (18, 1.7357686374452896e-05), (19, 1.2854993656219449e-05), (20, 1.1726849152182695e-05), (21, 2.18257300730329e-05), (22, 3.008433668583166e-05), (23, 7.073464075801894e-05), (24, 7.951392763061449e-05), (25, 0.00014666847710032016), (26, 5.6787539506331086e-05), (27, 3.5127715818816796e-05)]
+
+ 77%|███████▋  | 115/150 [23:52<07:14, 12.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 204, Spatial MSE: [(0, 8.108502242976101e-07), (1, 8.630904630990699e-06), (2, 2.621042040118482e-05), (3, 1.0841555194929242e-05), (4, 1.4221262063074391e-05), (5, 2.5850416932371445e-05), (6, 3.101258698734455e-05), (7, 8.015683124540374e-05), (8, 0.00010238093818770722), (9, 0.00015599858306813985), (10, 0.00011831326264655218), (11, 0.00015381653793156147), (12, 0.0001830314431572333), (13, 0.0013498659245669842), (14, 0.001589983468875289), (15, 0.0008282004273496568), (16, 0.01144244521856308), (17, 0.0003182736982125789), (18, 0.0003768775495700538), (19, 0.0009923455072566867), (20, 0.0007040058844722807), (21, 0.0007827557856217027), (22, 0.0006066215573810041), (23, 0.00099414202850312), (24, 0.00173096579965204), (25, 0.005558996926993132), (26, 0.010279571637511253), (27, 0.006082041654735804)], Temporal MSE: [(0, 1.076611852113274e-06), (1, 5.07711206410022e-07), (2, 2.1226237549853977e-06), (3, 5.09289429828641e-06), (4, 8.504248398821801e-06), (5, 1.3728633348364383e-05), (6, 2.807311284414027e-05), (7, 2.7925920221605338e-05), (8, 2.3222002710099332e-05), (9, 3.22710802720394e-05), (10, 5.256711665424518e-05), (11, 8.284628711408004e-05), (12, 7.510891009587795e-05), (13, 0.00011256013385718688), (14, 0.00017553236102685332), (15, 0.0001537906500743702), (16, 0.00018849332991521806), (17, 0.00015236440231092274), (18, 0.0001394907885696739), (19, 0.00022201861429493874), (20, 0.0003342927957419306), (21, 0.0002898501115851104), (22, 0.0002682757331058383), (23, 0.00040630457806400955), (24, 0.0009693018510006368), (25, 0.0014820315409451723), (26, 0.0007173977792263031), (27, 8.48608833621256e-05)]
+
+ 77%|███████▋  | 116/150 [24:02<06:34, 11.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 198, Spatial MSE: [(0, 8.075267601270752e-07), (1, 8.428435648966115e-06), (2, 1.1621989870036487e-05), (3, 6.057778591639362e-06), (4, 2.038856791841681e-06), (5, 3.0243152195907896e-06), (6, 2.164407533200574e-06), (7, 4.616586011252366e-06), (8, 3.746844640772906e-06), (9, 2.9483253456419334e-06), (10, 4.2504898374318145e-06), (11, 2.870994467230048e-05), (12, 6.938511796761304e-05), (13, 0.0017225249903276563), (14, 0.0015467063058167696), (15, 0.0004870643315371126), (16, 0.013181790709495544), (17, 6.738205411238596e-05), (18, 0.00010947385453619063), (19, 0.0009750163881108165), (20, 0.0005791544681414962), (21, 9.011002839542925e-05), (22, 0.000242256312048994), (23, 0.0001190203838632442), (24, 0.0002054288488579914), (25, 0.00039853467023931444), (26, 0.00040797851397655904), (27, 0.013565764762461185)], Temporal MSE: [(0, 9.107791925089259e-07), (1, 5.113555516800261e-07), (2, 6.019570264470531e-07), (3, 7.679689701944881e-07), (4, 9.793998287932482e-07), (5, 1.2239471516295453e-06), (6, 1.9651279217214324e-06), (7, 1.8890349338107626e-06), (8, 1.2045159110130044e-06), (9, 1.416815166521701e-06), (10, 2.5495505724393297e-06), (11, 8.404145773965865e-05), (12, 3.505176573526114e-05), (13, 3.6011679185321555e-05), (14, 2.0090574253117666e-05), (15, 3.1262683478416875e-05), (16, 1.4897483197273687e-05), (17, 1.4983411347202491e-05), (18, 2.887935988837853e-05), (19, 2.1008323528803885e-05), (20, 1.786076245480217e-05), (21, 4.7501514927716926e-05), (22, 5.9032732679042965e-05), (23, 0.00011035153875127435), (24, 0.00011393609020160511), (25, 0.00019937375327572227), (26, 7.665583689231426e-05), (27, 5.50264012417756e-05)]
+
+ 78%|███████▊  | 117/150 [24:12<06:10, 11.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 192, Spatial MSE: [(0, 1.0635066018949146e-06), (1, 6.799464244977571e-06), (2, 1.854554284363985e-05), (3, 5.837322532897815e-06), (4, 4.763426659337711e-06), (5, 1.0073931662191171e-05), (6, 1.6596310160821304e-05), (7, 1.987374525924679e-05), (8, 2.1876539904042147e-05), (9, 4.459087722352706e-05), (10, 7.041830394882709e-05), (11, 0.0001330501545453444), (12, 0.00014584936434403062), (13, 0.0019354968098923564), (14, 0.0021977557335048914), (15, 0.00108273362275213), (16, 0.026626290753483772), (17, 0.00011335290037095547), (18, 0.00012581003829836845), (19, 0.002070473739877343), (20, 0.0019113661255687475), (21, 0.00018846993043553084), (22, 0.00040911848191171885), (23, 0.00034155166940763593), (24, 0.0007994230254553258), (25, 0.002840875880792737), (26, 0.005344638600945473), (27, 0.012938735075294971)], Temporal MSE: [(0, 1.093913738259289e-06), (1, 6.478724117187085e-07), (2, 8.076837048065499e-07), (3, 1.503755470366741e-06), (4, 2.688790573301958e-06), (5, 4.2933666009048466e-06), (6, 7.032308076304616e-06), (7, 8.51488221087493e-06), (8, 7.025183094810927e-06), (9, 8.360206265933812e-06), (10, 1.2671458534896374e-05), (11, 9.902066813083366e-05), (12, 4.9338363169226795e-05), (13, 5.7760389609029517e-05), (14, 4.7672474465798587e-05), (15, 6.710061279591173e-05), (16, 3.306464714114554e-05), (17, 2.7592093829298392e-05), (18, 5.354702443582937e-05), (19, 5.7611152442405e-05), (20, 6.854445382487029e-05), (21, 7.073107553878799e-05), (22, 9.446227340959013e-05), (23, 0.00015461986185982823), (24, 0.00030535663245245814), (25, 0.0006266991840675473), (26, 0.0002900487161241472), (27, 8.37890911498107e-05)]
+
+ 79%|███████▊  | 118/150 [24:23<05:59, 11.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 186, Spatial MSE: [(0, 7.764926976960851e-07), (1, 8.51684217195725e-06), (2, 1.2601495654962491e-05), (3, 5.879656328033889e-06), (4, 2.095172703775461e-06), (5, 3.2198049666476436e-06), (6, 2.251888645332656e-06), (7, 4.6628974814666435e-06), (8, 3.7669001358153764e-06), (9, 3.247307631681906e-06), (10, 7.584204013255658e-06), (11, 3.539490717230365e-05), (12, 5.529357076738961e-05), (13, 0.0010724791791290045), (14, 0.0010308189084753394), (15, 0.0004356452845968306), (16, 0.009212593547999859), (17, 3.437971827224828e-05), (18, 3.4100339689757675e-05), (19, 0.0005704674986191094), (20, 0.00033709328272379935), (21, 7.684425509069115e-05), (22, 0.0002411163441138342), (23, 0.00011240797903155908), (24, 0.00016382476314902306), (25, 0.00038391471025533974), (26, 0.00035070106969214976), (27, 0.010683339089155197)], Temporal MSE: [(0, 8.676802849549858e-07), (1, 5.050335971645836e-07), (2, 6.085276140765927e-07), (3, 7.333349572036241e-07), (4, 9.763003845364437e-07), (5, 1.265690002583142e-06), (6, 2.0033776308991946e-06), (7, 1.931304723257199e-06), (8, 1.2595488669830956e-06), (9, 1.5191657212199061e-06), (10, 4.2741571633087005e-06), (11, 8.352050645044073e-05), (12, 2.3420059733325616e-05), (13, 2.5780107534956187e-05), (14, 1.9083021470578387e-05), (15, 3.934384585591033e-05), (16, 1.252377205673838e-05), (17, 1.0718428711697925e-05), (18, 2.3469296138500795e-05), (19, 2.0840596334892325e-05), (20, 2.5711817215778865e-05), (21, 3.29923459503334e-05), (22, 4.5999669964658096e-05), (23, 8.915864600567147e-05), (24, 0.00010281569848302752), (25, 0.00015003312728367746), (26, 6.165999366203323e-05), (27, 4.3037263822043315e-05)]
+
+ 79%|███████▉  | 119/150 [24:35<05:50, 11.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 180, Spatial MSE: [(0, 9.863217655947665e-07), (1, 8.725713087187614e-06), (2, 2.966233114420902e-05), (3, 1.103448994399514e-05), (4, 1.4955433471186552e-05), (5, 2.6775085643748753e-05), (6, 3.741572072613053e-05), (7, 8.861193782649934e-05), (8, 0.00010658064275048673), (9, 0.00019037969468627125), (10, 0.00012590749247465283), (11, 0.00019704298756550997), (12, 0.00021177544840611517), (13, 0.0016685770824551582), (14, 0.0026492506731301546), (15, 0.0015946115599945188), (16, 0.03400473669171333), (17, 0.00035350240068510175), (18, 0.0004458629118744284), (19, 0.002081241924315691), (20, 0.0016704583540558815), (21, 0.000977379851974547), (22, 0.0008632204844616354), (23, 0.0012972382828593254), (24, 0.0022363830357789993), (25, 0.007689469959586859), (26, 0.013223703019320965), (27, 0.011430375277996063)], Temporal MSE: [(0, 1.3837918686476769e-06), (1, 7.002682878010091e-07), (2, 2.301736458321102e-06), (3, 4.857126441493165e-06), (4, 8.763607183936983e-06), (5, 1.5016084944363683e-05), (6, 3.351584746269509e-05), (7, 3.321504846098833e-05), (8, 2.6218403945676982e-05), (9, 3.5788729292107746e-05), (10, 5.874744601896964e-05), (11, 0.00013908713299315423), (12, 8.539734699297696e-05), (13, 0.00013268308248370886), (14, 0.00020707199291791767), (15, 0.00019816364510916173), (16, 0.00022156565682962537), (17, 0.00018209291738457978), (18, 0.000174592700204812), (19, 0.0002834726183209568), (20, 0.0004308945790398866), (21, 0.000391312874853611), (22, 0.00034230074379593134), (23, 0.0004970954614691436), (24, 0.0010439300676807761), (25, 0.0015313929179683328), (26, 0.0008641331223770976), (27, 0.00011800735956057906)]
+
+ 80%|████████  | 120/150 [24:49<06:05, 12.17s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 174, Spatial MSE: [(0, 7.579335488117067e-07), (1, 8.678764970682096e-06), (2, 1.3259446859592572e-05), (3, 6.1404716689139605e-06), (4, 2.1347441361285746e-06), (5, 3.618051323428517e-06), (6, 2.4273449525935575e-06), (7, 5.636418791254982e-06), (8, 4.6254799599410035e-06), (9, 2.5156218725896906e-06), (10, 3.219761993022985e-06), (11, 4.806159267900512e-06), (12, 1.281893582927296e-05), (13, 0.00046947228838689625), (14, 0.0005995793035253882), (15, 0.00022927274403627962), (16, 0.006573311518877745), (17, 2.8629157895920798e-05), (18, 4.5061569835525006e-05), (19, 0.0003972701379097998), (20, 0.00019659052486531436), (21, 7.277326949406415e-05), (22, 0.0001765322231221944), (23, 0.00010898154141614214), (24, 0.0001474862510804087), (25, 0.0003369289042893797), (26, 0.00033608582452870905), (27, 0.006517074536532164)], Temporal MSE: [(0, 9.9139322173869e-07), (1, 5.630699888570234e-07), (2, 6.682352022835403e-07), (3, 8.156574153872498e-07), (4, 1.0974586075462867e-06), (5, 1.4146465900921612e-06), (6, 2.2460112631961238e-06), (7, 2.2301360331766773e-06), (8, 1.38332018195797e-06), (9, 1.5864374063312425e-06), (10, 1.9021398429686087e-06), (11, 5.168184998183278e-06), (12, 8.577106200391427e-06), (13, 1.4972317330830265e-05), (14, 1.1053008165617939e-05), (15, 2.1268562704790384e-05), (16, 1.1280123544565868e-05), (17, 1.1801264918176457e-05), (18, 2.1700276192859747e-05), (19, 2.3948456146172248e-05), (20, 2.990771463373676e-05), (21, 3.1768260669196025e-05), (22, 3.2275511330226436e-05), (23, 7.389661186607555e-05), (24, 8.015493222046643e-05), (25, 0.00010402331099612638), (26, 5.648604565067217e-05), (27, 2.8291762646404095e-05)]
+
+ 81%|████████  | 121/150 [25:01<05:49, 12.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 168, Spatial MSE: [(0, 9.799347253647284e-07), (1, 7.766491762595251e-06), (2, 2.4164388378267176e-05), (3, 6.617034614464501e-06), (4, 5.329917257768102e-06), (5, 1.085578423953848e-05), (6, 1.6706553651602007e-05), (7, 2.324337219761219e-05), (8, 2.4072784071904607e-05), (9, 3.9984162867767736e-05), (10, 6.588811083929613e-05), (11, 0.00010528897837502882), (12, 8.581930887885392e-05), (13, 0.0006350954063236713), (14, 0.0010823807679116726), (15, 0.0006761764525435865), (16, 0.0134405717253685), (17, 6.71177331241779e-05), (18, 0.00010450014815432951), (19, 0.0006857727421447635), (20, 0.0005161155131645501), (21, 0.00016252350178547204), (22, 0.0003601967473514378), (23, 0.00037729169707745314), (24, 0.0008378274505957961), (25, 0.0031894331332296133), (26, 0.005873604211956263), (27, 0.0067989337258040905)], Temporal MSE: [(0, 1.2924159591420903e-06), (1, 7.808409350218426e-07), (2, 9.14094414383726e-07), (3, 1.6826446653794846e-06), (4, 2.9621126031997846e-06), (5, 4.651697508961661e-06), (6, 7.879611985117663e-06), (7, 9.199732630804647e-06), (8, 7.727474439889193e-06), (9, 8.717544915270992e-06), (10, 1.1764220289478544e-05), (11, 1.869653533503879e-05), (12, 2.2088685000198893e-05), (13, 3.5854976886184886e-05), (14, 4.205202640150674e-05), (15, 4.8582784074824303e-05), (16, 2.815901825670153e-05), (17, 2.713581852731295e-05), (18, 3.817484684986994e-05), (19, 4.033141885884106e-05), (20, 5.0039405323332176e-05), (21, 5.78162980673369e-05), (22, 8.021371468203142e-05), (23, 0.00013208620657678694), (24, 0.00031765177845954895), (25, 0.0006689848378300667), (26, 0.0003253461909480393), (27, 7.079785427777097e-05)]
+
+ 81%|████████▏ | 122/150 [25:11<05:21, 11.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 162, Spatial MSE: [(0, 9.221736263498315e-07), (1, 9.594165931048337e-06), (2, 1.473691372666508e-05), (3, 5.86072746955324e-06), (4, 2.1303196717781248e-06), (5, 5.190067440707935e-06), (6, 2.6673351385397837e-06), (7, 6.862175723654218e-06), (8, 5.688994860975072e-06), (9, 2.9797702154610306e-06), (10, 3.5274097172077745e-06), (11, 5.7815832406049594e-06), (12, 9.943125405698083e-06), (13, 0.00020498356025200337), (14, 0.00031129198032431304), (15, 0.00013722653966397047), (16, 0.008052891120314598), (17, 4.8461621190654114e-05), (18, 7.849132816772908e-05), (19, 0.0004711083311121911), (20, 0.00030307075940072536), (21, 7.279700366780162e-05), (22, 0.00013249296171125025), (23, 0.00010717168333940208), (24, 0.0001602342090336606), (25, 0.000306780420942232), (26, 0.0003503018815536052), (27, 0.004170481115579605)], Temporal MSE: [(0, 1.0638966614351375e-06), (1, 6.387229518622917e-07), (2, 7.334929819080571e-07), (3, 8.070126114034792e-07), (4, 1.0878077318920987e-06), (5, 1.4155131111692754e-06), (6, 2.289315489178989e-06), (7, 2.519110466892016e-06), (8, 1.4840979929431342e-06), (9, 1.7851306211014162e-06), (10, 2.2262968286668183e-06), (11, 4.683032784669194e-06), (12, 6.150629815238062e-06), (13, 1.0828567610587925e-05), (14, 1.0832866792043205e-05), (15, 4.0036164136836305e-05), (16, 1.9675237126648426e-05), (17, 2.0223757019266486e-05), (18, 1.954377512447536e-05), (19, 1.7243812180822715e-05), (20, 2.1642292267642915e-05), (21, 3.2303258194588125e-05), (22, 4.053081283927895e-05), (23, 7.121457747416571e-05), (24, 7.352726242970675e-05), (25, 0.0001645758020458743), (26, 5.2298932132543996e-05), (27, 2.6682821044232696e-05)]
+
+ 82%|████████▏ | 123/150 [25:22<05:07, 11.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 156, Spatial MSE: [(0, 1.0731620250226115e-06), (1, 1.0720679711084813e-05), (2, 4.001641718787141e-05), (3, 1.157980841526296e-05), (4, 1.597007576492615e-05), (5, 2.8877049771836028e-05), (6, 3.529523019096814e-05), (7, 9.415017120772973e-05), (8, 0.00013149988080840558), (9, 0.00021787462173961103), (10, 0.0001334434055024758), (11, 0.0001894886081572622), (12, 0.0001902863587019965), (13, 0.0013221099507063627), (14, 0.0036939545534551144), (15, 0.0025053382851183414), (16, 0.062329746782779694), (17, 0.00041257147677242756), (18, 0.0006004095776006579), (19, 0.005846711341291666), (20, 0.006104316096752882), (21, 0.001004477497190237), (22, 0.0008544765296392143), (23, 0.0013615809148177505), (24, 0.002220385242253542), (25, 0.006924320012331009), (26, 0.012579350732266903), (27, 0.004056643228977919)], Temporal MSE: [(0, 1.5074426755745662e-06), (1, 7.79090441938024e-07), (2, 2.493245347068296e-06), (3, 4.979137884220108e-06), (4, 9.445608156966045e-06), (5, 1.5321382306865416e-05), (6, 3.315549110993743e-05), (7, 3.7989353586453944e-05), (8, 2.858954030671157e-05), (9, 3.8431007851613685e-05), (10, 6.021146691637114e-05), (11, 7.273467053892091e-05), (12, 8.21554713184014e-05), (13, 0.00013101068907417357), (14, 0.00021775081404484808), (15, 0.00022630335297435522), (16, 0.0002444720303174108), (17, 0.0001953484461409971), (18, 0.0001945351978065446), (19, 0.00029345465009100735), (20, 0.0004519273352343589), (21, 0.0004441702039912343), (22, 0.0003733264748007059), (23, 0.0005033357301726937), (24, 0.0011289570247754455), (25, 0.0018250361317768693), (26, 0.000943401304539293), (27, 0.00012494178372435272)]
+
+ 83%|████████▎ | 124/150 [25:35<05:11, 11.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 150, Spatial MSE: [(0, 9.804798537516035e-07), (1, 1.0876799024117645e-05), (2, 1.7286953152506612e-05), (3, 7.186114089563489e-06), (4, 2.814955905705574e-06), (5, 5.01310387335252e-06), (6, 3.393840643184376e-06), (7, 7.119125712051755e-06), (8, 6.586228664673399e-06), (9, 4.334536242822651e-06), (10, 7.362412361544557e-06), (11, 1.63425829668995e-05), (12, 1.9162453099852428e-05), (13, 0.00048406500718556345), (14, 0.001243491773493588), (15, 0.0004677258257288486), (16, 0.03706827759742737), (17, 0.0001751777163008228), (18, 0.00012123049236834049), (19, 0.003315029898658395), (20, 0.0026799915358424187), (21, 9.46614527492784e-05), (22, 0.0002101587742799893), (23, 0.00019251843332313), (24, 0.00032883213134482503), (25, 0.0008117055986076593), (26, 0.0004853275022469461), (27, 0.01498609408736229)], Temporal MSE: [(0, 1.2529633295343956e-06), (1, 7.640450689905265e-07), (2, 8.705023901711684e-07), (3, 1.0109833965543658e-06), (4, 1.4248038269215613e-06), (5, 1.865447075033444e-06), (6, 3.1549141112918733e-06), (7, 3.0689238883496728e-06), (8, 2.0455911453609588e-06), (9, 2.4748267151153414e-06), (10, 5.1297415666340385e-06), (11, 9.436641448701266e-06), (12, 1.3874852811568417e-05), (13, 1.869532025011722e-05), (14, 2.936467353720218e-05), (15, 0.00010468652180861682), (16, 7.343353354372084e-05), (17, 9.242628584615886e-05), (18, 5.899384996155277e-05), (19, 5.358416819944978e-05), (20, 3.739424573723227e-05), (21, 0.00020489182497840375), (22, 0.00017054524505510926), (23, 0.00019835098646581173), (24, 0.00015339654055424035), (25, 0.0003971403930336237), (26, 8.637765131425112e-05), (27, 8.679118036525324e-05)]
+
+ 83%|████████▎ | 125/150 [25:44<04:37, 11.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 144, Spatial MSE: [(0, 1.0534458851907402e-06), (1, 9.564457286614925e-06), (2, 3.0446048185694963e-05), (3, 7.275977623066865e-06), (4, 6.150502031232463e-06), (5, 1.2816420166927855e-05), (6, 1.830837754823733e-05), (7, 2.5740477212821133e-05), (8, 2.8956048481632024e-05), (9, 4.5546043111244217e-05), (10, 7.23395642125979e-05), (11, 0.00012287577555980533), (12, 9.502506873104721e-05), (13, 0.0005808785208500922), (14, 0.0015734608750790358), (15, 0.0007386330980807543), (16, 0.04288647696375847), (17, 0.0002323884837096557), (18, 0.00017724792996887118), (19, 0.0035028893034905195), (20, 0.002910903189331293), (21, 0.00017308059614151716), (22, 0.0004407490196172148), (23, 0.0004945407854393125), (24, 0.0012903419556096196), (25, 0.004190027713775635), (26, 0.006890181452035904), (27, 0.015128271654248238)], Temporal MSE: [(0, 1.4236483139029588e-06), (1, 9.002017122838879e-07), (2, 9.779550964594819e-07), (3, 1.7245845356228529e-06), (4, 3.0487365165754454e-06), (5, 5.176025297259912e-06), (6, 8.970077033154666e-06), (7, 1.0599369488772936e-05), (8, 9.153181053989101e-06), (9, 1.0466467756486963e-05), (10, 1.6143947505042888e-05), (11, 2.44292568822857e-05), (12, 2.7684111046255566e-05), (13, 3.881827797158621e-05), (14, 6.167758692754433e-05), (15, 0.000128407686133869), (16, 9.227914415532723e-05), (17, 0.00010806271893670782), (18, 8.166639599949121e-05), (19, 8.219534356612712e-05), (20, 7.35832509235479e-05), (21, 0.00023534735373687), (22, 0.00021960031881462783), (23, 0.0002698834578040987), (24, 0.0004338217549957335), (25, 0.0009983518393710256), (26, 0.0004055579483974725), (27, 0.00014437592471949756)]
+
+ 84%|████████▍ | 126/150 [25:58<04:46, 11.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 138, Spatial MSE: [(0, 9.444538022762572e-07), (1, 1.1112239917565603e-05), (2, 1.8251597794005647e-05), (3, 6.598654181289021e-06), (4, 2.6674026685213903e-06), (5, 4.648720278055407e-06), (6, 3.367161070855218e-06), (7, 6.28616044195951e-06), (8, 5.687207703886088e-06), (9, 4.294834980100859e-06), (10, 4.238166638970142e-06), (11, 9.235877769242506e-06), (12, 2.9018519853707403e-05), (13, 0.000913040479645133), (14, 0.0010860587935894728), (15, 0.0005333246081136167), (16, 0.025164354592561722), (17, 0.00013314999523572624), (18, 0.0001007525424938649), (19, 0.00209250976331532), (20, 0.0016317261615768075), (21, 0.0007533995085395873), (22, 0.0003481693856883794), (23, 0.00022613393957726657), (24, 0.00037410578806884587), (25, 0.0007634893408976495), (26, 0.00046109684626571834), (27, 0.009845669381320477)], Temporal MSE: [(0, 1.394558012179914e-06), (1, 7.993867257027887e-07), (2, 9.307933055424655e-07), (3, 9.883682423605933e-07), (4, 1.4353947790368693e-06), (5, 1.8385350131211453e-06), (6, 2.9585953598143533e-06), (7, 3.080601800320437e-06), (8, 2.1093433133501094e-06), (9, 2.3037357550492743e-06), (10, 2.898086449931725e-06), (11, 1.9741031792364083e-05), (12, 2.4106320779537782e-05), (13, 2.336031866434496e-05), (14, 3.273674155934714e-05), (15, 6.0202786698937416e-05), (16, 4.543262912193313e-05), (17, 6.412704533431679e-05), (18, 4.191383413854055e-05), (19, 3.6862078559352085e-05), (20, 3.1422561733052135e-05), (21, 0.00018155622819904238), (22, 0.00019052260904572904), (23, 0.00020942208357155323), (24, 0.00016708853945601732), (25, 0.00032970248139463365), (26, 8.81877276697196e-05), (27, 6.235168984858319e-05)]
+
+ 85%|████████▍ | 127/150 [26:09<04:27, 11.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 132, Spatial MSE: [(0, 1.134951958192687e-06), (1, 1.1898448065039702e-05), (2, 4.795056156581268e-05), (3, 1.188209262181772e-05), (4, 1.6294174201902933e-05), (5, 3.0045936000533402e-05), (6, 4.329143848735839e-05), (7, 0.00010545549594098702), (8, 0.0001363461051369086), (9, 0.00022448187519330531), (10, 0.00012811232591047883), (11, 0.00018946350610349327), (12, 0.0001842033670982346), (13, 0.0009886821499094367), (14, 0.001369653968140483), (15, 0.0010400601895526052), (16, 0.021325258538126945), (17, 0.00047659542178735137), (18, 0.0005369008285924792), (19, 0.002211008919402957), (20, 0.00193562894128263), (21, 0.00173609284684062), (22, 0.0011290227994322777), (23, 0.0017300931503996253), (24, 0.0029595596715807915), (25, 0.010367348790168762), (26, 0.017692038789391518), (27, 0.010192959569394588)], Temporal MSE: [(0, 1.8044549960904988e-06), (1, 9.90583430393599e-07), (2, 2.5747870040504495e-06), (3, 4.846459432883421e-06), (4, 9.63030743150739e-06), (5, 1.6797839634818956e-05), (6, 4.949491994921118e-05), (7, 4.592882396536879e-05), (8, 3.400843343115412e-05), (9, 4.284382885089144e-05), (10, 6.479348667198792e-05), (11, 8.68292263476178e-05), (12, 8.622745372122154e-05), (13, 0.00013433591811917722), (14, 0.00023578121908940375), (15, 0.00022924007498659194), (16, 0.0002652027178555727), (17, 0.00023753991990815848), (18, 0.00019853468984365463), (19, 0.0002948635956272483), (20, 0.0004628866445273161), (21, 0.0005537957767955959), (22, 0.0004972530878148973), (23, 0.0006607791874557734), (24, 0.0013010872062295675), (25, 0.0020938797388225794), (26, 0.0011879651574417949), (27, 0.0001979867956833914)]
+
+ 85%|████████▌ | 128/150 [26:19<04:06, 11.21s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 126, Spatial MSE: [(0, 1.0044890359495184e-06), (1, 1.3200885405240115e-05), (2, 2.188763210142497e-05), (3, 7.6208789323573e-06), (4, 3.287000254204031e-06), (5, 5.288998181640636e-06), (6, 4.023507699457696e-06), (7, 6.6191150835948065e-06), (8, 6.826684057159582e-06), (9, 4.860421995545039e-06), (10, 1.1277937119302806e-05), (11, 3.945129719795659e-05), (12, 7.492942677345127e-05), (13, 0.0015916232950985432), (14, 0.0016920940252020955), (15, 0.0006978078745305538), (16, 0.03687053173780441), (17, 0.0001522791717434302), (18, 0.00015013079973869026), (19, 0.003264673985540867), (20, 0.0025368675123900175), (21, 0.00017051452596206218), (22, 0.0003692274331115186), (23, 0.00027592157130129635), (24, 0.0004692237707786262), (25, 0.0010971024166792631), (26, 0.000574988778680563), (27, 0.017752205953001976)], Temporal MSE: [(0, 1.5260808368111611e-06), (1, 8.702161267137853e-07), (2, 1.0252163065160858e-06), (3, 1.1440300795584335e-06), (4, 1.7496033706265735e-06), (5, 2.3423649508913513e-06), (6, 3.2991044918162515e-06), (7, 3.648386837085127e-06), (8, 2.7083233362645842e-06), (9, 3.0056100968067767e-06), (10, 7.829421520000324e-06), (11, 9.679086360847577e-05), (12, 3.795791053562425e-05), (13, 3.526440559653565e-05), (14, 3.770889452425763e-05), (15, 9.028516069520265e-05), (16, 6.274980114540085e-05), (17, 9.384335135109723e-05), (18, 7.224771979963407e-05), (19, 6.119877070887014e-05), (20, 4.5231656258692965e-05), (21, 0.0002502208517398685), (22, 0.00019400837481953204), (23, 0.000252656260272488), (24, 0.00021342762920539826), (25, 0.00047217548126354814), (26, 0.00014553790970239788), (27, 0.00010322866728529334)]
+
+ 86%|████████▌ | 129/150 [26:32<04:03, 11.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 120, Spatial MSE: [(0, 1.1751743613785948e-06), (1, 1.1652209650492296e-05), (2, 3.895961708622053e-05), (3, 7.95960113464389e-06), (4, 7.594744602101855e-06), (5, 1.580379466759041e-05), (6, 2.3073140255291946e-05), (7, 3.0179073291947134e-05), (8, 3.298351657576859e-05), (9, 5.548785702558234e-05), (10, 8.284928480861709e-05), (11, 0.00014540061238221824), (12, 0.0001521425583632663), (13, 0.0016674837097525597), (14, 0.0018066804623231292), (15, 0.000827376265078783), (16, 0.036982014775276184), (17, 0.00019849515228997916), (18, 0.00028159754583612084), (19, 0.003307278733700514), (20, 0.0026047055143862963), (21, 0.00030770536977797747), (22, 0.0006729137967340648), (23, 0.0006659437203779817), (24, 0.0015328828012570739), (25, 0.005742105655372143), (26, 0.00838304590433836), (27, 0.019349319860339165)], Temporal MSE: [(0, 1.6951930774666835e-06), (1, 1.2038376553391572e-06), (2, 1.1617047448453377e-06), (3, 1.9337946923769778e-06), (4, 3.6626047403842676e-06), (5, 6.2608114603790455e-06), (6, 1.182744381367229e-05), (7, 1.381694346491713e-05), (8, 1.1722816452675033e-05), (9, 1.3714039596379735e-05), (10, 2.1608895622193813e-05), (11, 0.00011495791841298342), (12, 5.375359978643246e-05), (13, 5.849840454175137e-05), (14, 7.983767136465758e-05), (15, 0.00011150551290484145), (16, 8.52034063427709e-05), (17, 0.00011100789561169222), (18, 9.626487008063123e-05), (19, 9.333316120319068e-05), (20, 8.310843986691907e-05), (21, 0.0002880276006180793), (22, 0.00025548198027536273), (23, 0.00034182085073553026), (24, 0.000581134227104485), (25, 0.0013455548323690891), (26, 0.0005899514653719962), (27, 0.00022323567827697843)]
+
+ 87%|████████▋ | 130/150 [26:45<04:01, 12.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 114, Spatial MSE: [(0, 1.0344617749069585e-06), (1, 1.4141570318315644e-05), (2, 2.267690433654934e-05), (3, 6.612808192585362e-06), (4, 3.2171860766538884e-06), (5, 6.416750693460926e-06), (6, 4.216945853841025e-06), (7, 8.746095772949047e-06), (8, 8.148162123688962e-06), (9, 5.1466759032336995e-06), (10, 7.49300579627743e-06), (11, 1.657879511185456e-05), (12, 4.2745890823425725e-05), (13, 0.0011516764061525464), (14, 0.001278304262086749), (15, 0.00039975609979592264), (16, 0.02360272966325283), (17, 9.883813618216664e-05), (18, 6.63454266032204e-05), (19, 0.002024157205596566), (20, 0.0015328320441767573), (21, 0.00011286318476777524), (22, 0.00025852955877780914), (23, 0.00021986302454024553), (24, 0.000340828177286312), (25, 0.0007835783180780709), (26, 0.0005403258255682886), (27, 0.013155154883861542)], Temporal MSE: [(0, 1.6869196315383306e-06), (1, 1.0017868135037133e-06), (2, 1.0801164762597182e-06), (3, 1.1201193501619855e-06), (4, 1.7399285070496262e-06), (5, 2.108119133481523e-06), (6, 4.442293175088707e-06), (7, 3.827472937700804e-06), (8, 2.7839116683026077e-06), (9, 3.157286073474097e-06), (10, 5.124144536239328e-06), (11, 2.9593002182082273e-05), (12, 3.157389073749073e-05), (13, 2.8741838832502253e-05), (14, 2.354971547902096e-05), (15, 8.614307444076985e-05), (16, 5.468014569487423e-05), (17, 6.524907803395763e-05), (18, 4.3039588490501046e-05), (19, 3.894266774295829e-05), (20, 3.074948108405806e-05), (21, 0.0001559507509227842), (22, 0.00011260980681981891), (23, 0.00015966793580446392), (24, 0.00014066396397538483), (25, 0.00032355511211790144), (26, 0.00011442729737609625), (27, 7.893088331911713e-05)]
+
+ 87%|████████▋ | 131/150 [26:57<03:46, 11.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 108, Spatial MSE: [(0, 1.2703055745078018e-06), (1, 1.285805137740681e-05), (2, 6.807875615777448e-05), (3, 1.2621878340723924e-05), (4, 1.854041693150066e-05), (5, 3.542839840520173e-05), (6, 5.0923707021865994e-05), (7, 0.0001429776311852038), (8, 0.00018839986296370625), (9, 0.00026018242351710796), (10, 0.000153897242853418), (11, 0.00021893744997214526), (12, 0.00021768506849184632), (13, 0.001303219934925437), (14, 0.0016558757051825523), (15, 0.0009309959132224321), (16, 0.024278387427330017), (17, 0.0004889607662335038), (18, 0.0005182602326385677), (19, 0.0025245207361876965), (20, 0.0021209209226071835), (21, 0.0011744085932150483), (22, 0.0012594154104590416), (23, 0.0021918092388659716), (24, 0.003709737677127123), (25, 0.012293864972889423), (26, 0.022311344742774963), (27, 0.012904629111289978)], Temporal MSE: [(0, 2.2772762804379454e-06), (1, 1.2475854873628123e-06), (2, 3.0564719963876996e-06), (3, 6.13604925092659e-06), (4, 1.0764310900412966e-05), (5, 1.8371101759839803e-05), (6, 6.174407462822273e-05), (7, 5.9181591495871544e-05), (8, 4.407440428622067e-05), (9, 5.016731665818952e-05), (10, 7.955967885209247e-05), (11, 0.00011136264947708696), (12, 0.00010736552940215915), (13, 0.0001631406630622223), (14, 0.00026583822909742594), (15, 0.0002981284633278847), (16, 0.0003073851694352925), (17, 0.00026932451874017715), (18, 0.00023549798061139882), (19, 0.0003544543869793415), (20, 0.0005659495363943279), (21, 0.0006168216350488365), (22, 0.0005002579418942332), (23, 0.0007430753903463483), (24, 0.0016367053613066673), (25, 0.0027054697275161743), (26, 0.001654548803344369), (27, 0.00035342754563316703)]
+
+ 88%|████████▊ | 132/150 [27:06<03:19, 11.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 102, Spatial MSE: [(0, 1.1828709602923482e-06), (1, 1.527553831692785e-05), (2, 2.6026003979495727e-05), (3, 7.86028613219969e-06), (4, 3.7330905797716696e-06), (5, 7.189506504801102e-06), (6, 5.209884875512216e-06), (7, 9.03875297808554e-06), (8, 8.250556675193366e-06), (9, 6.8400568125071e-06), (10, 6.10387451160932e-06), (11, 1.0868423487409018e-05), (12, 2.7610074539552443e-05), (13, 0.0010575115447863936), (14, 0.0021178312599658966), (15, 0.0010563525138422847), (16, 0.07975456118583679), (17, 0.00029058230575174093), (18, 0.00014761975035071373), (19, 0.007087495643645525), (20, 0.005535234231501818), (21, 0.00025202150573022664), (22, 0.0004899693885818124), (23, 0.00038792454870417714), (24, 0.0006626298418268561), (25, 0.0016535434406250715), (26, 0.0008047505398280919), (27, 0.02981121465563774)], Temporal MSE: [(0, 1.830138785408053e-06), (1, 1.1089742883996223e-06), (2, 1.3018086519878125e-06), (3, 1.3044782463111915e-06), (4, 2.1496198314707726e-06), (5, 2.7914936708839377e-06), (6, 4.993049515178427e-06), (7, 4.913663360639475e-06), (8, 3.7254399103403557e-06), (9, 3.7665622585336678e-06), (10, 4.342300144344335e-06), (11, 2.045371184067335e-05), (12, 2.413737092865631e-05), (13, 4.211063787806779e-05), (14, 6.661568477284163e-05), (15, 0.000125883670989424), (16, 0.00010404413478681818), (17, 0.00023149271146394312), (18, 0.000128533472889103), (19, 0.00011302461643936113), (20, 8.211337990360335e-05), (21, 0.0004067129921168089), (22, 0.0003441048611421138), (23, 0.0004149111337028444), (24, 0.0003191075229551643), (25, 0.0007977805798873305), (26, 0.00019802444148808718), (27, 0.00019126306870020926)]
+
+ 89%|████████▊ | 133/150 [27:14<02:55, 10.34s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 96, Spatial MSE: [(0, 1.3622451433548122e-06), (1, 1.4013678082847036e-05), (2, 6.046561247785576e-05), (3, 1.011010226648068e-05), (4, 1.1980447197856847e-05), (5, 2.1042314983787946e-05), (6, 3.198000558768399e-05), (7, 6.604613008676097e-05), (8, 8.856549538904801e-05), (9, 0.00012711632007267326), (10, 9.282949031330645e-05), (11, 0.00014833040768280625), (12, 0.00012336719373706728), (13, 0.0010501934448257089), (14, 0.002801178488880396), (15, 0.002036013640463352), (16, 0.07232307642698288), (17, 0.00048113352386280894), (18, 0.0004469726118259132), (19, 0.00665503554046154), (20, 0.005756209138780832), (21, 0.000670888926833868), (22, 0.00109828170388937), (23, 0.001397987944073975), (24, 0.0025597065687179565), (25, 0.009071103297173977), (26, 0.016043677926063538), (27, 0.030017992481589317)], Temporal MSE: [(0, 2.2134070150059415e-06), (1, 1.4010378208695329e-06), (2, 1.8383638007435366e-06), (3, 3.0882813462085323e-06), (4, 4.521419668890303e-06), (5, 7.131461643439252e-06), (6, 3.2812698918860406e-05), (7, 2.8335223760223016e-05), (8, 2.0598889022949152e-05), (9, 2.142928133253008e-05), (10, 3.2305870263371617e-05), (11, 5.0093243771698326e-05), (12, 4.964637628290802e-05), (13, 7.885681407060474e-05), (14, 0.00014695001300424337), (15, 0.00018542888574302197), (16, 0.00017910874157678336), (17, 0.00027811137260869145), (18, 0.00017965570441447198), (19, 0.0002209462836617604), (20, 0.0002798707573674619), (21, 0.0005393006722442806), (22, 0.00044121421524323523), (23, 0.0005854974733665586), (24, 0.0008332398138009012), (25, 0.0015613381983712316), (26, 0.0006917864666320384), (27, 0.00038727349601686)]
+
+ 89%|████████▉ | 134/150 [27:24<02:41, 10.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 90, Spatial MSE: [(0, 1.2507714473031228e-06), (1, 1.5933666873024777e-05), (2, 4.224831718602218e-05), (3, 6.361023679346545e-06), (4, 6.091641807870474e-06), (5, 1.1624411854427308e-05), (6, 1.5783394701429643e-05), (7, 3.278501390013844e-05), (8, 3.9237365854205564e-05), (9, 5.1932292990386486e-05), (10, 3.5350763937458396e-05), (11, 6.0880600358359516e-05), (12, 3.6305540561443195e-05), (13, 5.099647751194425e-05), (14, 0.00010912646393990144), (15, 9.026283805724233e-05), (16, 9.39049496082589e-05), (17, 7.183434354374185e-05), (18, 9.10430753719993e-05), (19, 0.00011517682287376374), (20, 0.00011488551535876468), (21, 0.00017797407053876668), (22, 0.0002609612129162997), (23, 0.0005273546557873487), (24, 0.0009442599257454276), (25, 0.0027933018282055855), (26, 0.005594627931714058), (27, 0.002835559658706188)], Temporal MSE: [(0, 2.3134575712902006e-06), (1, 1.3929204669693718e-06), (2, 1.670350570748269e-06), (3, 1.8168667565987562e-06), (4, 2.705280621739803e-06), (5, 3.217503490304807e-06), (6, 1.874611189123243e-05), (7, 1.2473573406168725e-05), (8, 9.441469956072979e-06), (9, 9.663582204666454e-06), (10, 1.328827056568116e-05), (11, 1.4154097698337864e-05), (12, 1.2446631444618106e-05), (13, 2.1928844944341108e-05), (14, 3.944902346120216e-05), (15, 5.0477283366490155e-05), (16, 4.694114613812417e-05), (17, 3.266953717684373e-05), (18, 3.6111236113356426e-05), (19, 5.7990771892946213e-05), (20, 9.533538832329214e-05), (21, 7.429422112181783e-05), (22, 7.103230018401518e-05), (23, 0.00011774527229135856), (24, 0.00027261869399808347), (25, 0.0004020659835077822), (26, 0.0002528727927710861), (27, 0.0001490137801738456)]
+
+ 90%|█████████ | 135/150 [27:33<02:28,  9.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 84, Spatial MSE: [(0, 1.3069688975519966e-06), (1, 1.6650556062813848e-05), (2, 4.440998600330204e-05), (3, 6.596320417884272e-06), (4, 6.150593890197342e-06), (5, 1.0503291377972346e-05), (6, 1.4466733773588203e-05), (7, 2.8566439141286537e-05), (8, 3.520672180457041e-05), (9, 4.887690010946244e-05), (10, 3.031742926395964e-05), (11, 4.837519736611284e-05), (12, 3.50329682987649e-05), (13, 4.460670243133791e-05), (14, 8.611385419499129e-05), (15, 8.651425741845742e-05), (16, 7.070638821460307e-05), (17, 6.245588883757591e-05), (18, 8.157706179190427e-05), (19, 0.00010477342584636062), (20, 9.804044384509325e-05), (21, 0.00017372863658238202), (22, 0.00024232667055912316), (23, 0.00048615894047543406), (24, 0.0009335059439763427), (25, 0.0026234486140310764), (26, 0.005204150453209877), (27, 0.0027748188003897667)], Temporal MSE: [(0, 2.4553712592023658e-06), (1, 1.506094690739701e-06), (2, 1.6763102621553116e-06), (3, 2.0204035990900593e-06), (4, 2.6659160994313424e-06), (5, 3.236404609197052e-06), (6, 1.8809241737471893e-05), (7, 1.2428250556695275e-05), (8, 9.870439498627093e-06), (9, 9.001388207252603e-06), (10, 1.2253177374077495e-05), (11, 1.4148085028864443e-05), (12, 1.1374685527698603e-05), (13, 1.9427934603299946e-05), (14, 3.4862299798987806e-05), (15, 4.0955026634037495e-05), (16, 3.963665949413553e-05), (17, 2.8243084670975804e-05), (18, 3.12337651848793e-05), (19, 4.87628931296058e-05), (20, 7.280195131897926e-05), (21, 6.494452100014314e-05), (22, 6.204234523465857e-05), (23, 0.00010730526992119849), (24, 0.0002319842460565269), (25, 0.0003665101539809257), (26, 0.00023748815874569118), (27, 0.00017670054512564093)]
+
+ 91%|█████████ | 136/150 [27:46<02:28, 10.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 78, Spatial MSE: [(0, 1.3190389154260629e-06), (1, 1.782519211701583e-05), (2, 5.127902113599703e-05), (3, 6.781578122172505e-06), (4, 6.769585979782278e-06), (5, 1.1352652109053452e-05), (6, 1.4836960872344207e-05), (7, 3.282166289864108e-05), (8, 4.250493657309562e-05), (9, 4.738952702609822e-05), (10, 3.3409291063435376e-05), (11, 4.942607483826578e-05), (12, 3.368438410689123e-05), (13, 4.5072054490447044e-05), (14, 0.00010709268826758489), (15, 0.00010799991287058219), (16, 7.701436697971076e-05), (17, 6.72610622132197e-05), (18, 9.530725219519809e-05), (19, 0.00011838948557851836), (20, 0.00011088298924732953), (21, 0.00019194628112018108), (22, 0.0002584730682428926), (23, 0.0005348601844161749), (24, 0.0009635256137698889), (25, 0.0025133497547358274), (26, 0.005358059424906969), (27, 0.002864287933334708)], Temporal MSE: [(0, 2.5036572424141923e-06), (1, 1.5497653294005431e-06), (2, 1.8080463632941246e-06), (3, 2.1295834358170396e-06), (4, 2.7913222311326535e-06), (5, 3.174691755702952e-06), (6, 2.195645538449753e-05), (7, 1.3592497452918906e-05), (8, 1.021700882120058e-05), (9, 9.211556061927695e-06), (10, 1.1327716492814943e-05), (11, 1.2967371731065214e-05), (12, 1.1031751455448102e-05), (13, 2.060666520264931e-05), (14, 3.934689812012948e-05), (15, 5.3893207223154604e-05), (16, 4.33004715887364e-05), (17, 3.1205781851895154e-05), (18, 3.448304778430611e-05), (19, 5.640902600134723e-05), (20, 8.668618102092296e-05), (21, 7.861106132622808e-05), (22, 6.976155418669805e-05), (23, 0.00011867301509482786), (24, 0.00027411943301558495), (25, 0.00039411892066709697), (26, 0.0002644249179866165), (27, 0.00024258659686893225)]
+
+ 91%|█████████▏| 137/150 [27:55<02:11, 10.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 72, Spatial MSE: [(0, 1.39845724334009e-06), (1, 1.893172702693846e-05), (2, 5.009464803151786e-05), (3, 7.28444456399302e-06), (4, 6.997909167694161e-06), (5, 1.2302893082960509e-05), (6, 1.7148473489214666e-05), (7, 3.3214200811926275e-05), (8, 4.3045620259363204e-05), (9, 5.166260598343797e-05), (10, 3.229755748179741e-05), (11, 4.5219410822028294e-05), (12, 3.347557139932178e-05), (13, 6.13812735537067e-05), (14, 0.00012677095946855843), (15, 9.669612336438149e-05), (16, 7.616035873070359e-05), (17, 7.752186502330005e-05), (18, 0.00010277976980432868), (19, 0.00012665672693401575), (20, 0.00011984884622506797), (21, 0.00023960665566846728), (22, 0.000278983119642362), (23, 0.0005870774039067328), (24, 0.0010594417108222842), (25, 0.0028545050881803036), (26, 0.005935393739491701), (27, 0.003248634049668908)], Temporal MSE: [(0, 2.6386978788650595e-06), (1, 1.6612393665127456e-06), (2, 1.8746536625258159e-06), (3, 2.195082288380945e-06), (4, 3.0650296594103565e-06), (5, 3.5831733384839026e-06), (6, 2.5885157810989767e-05), (7, 1.4303556781669613e-05), (8, 1.1545103916432709e-05), (9, 9.919693184201606e-06), (10, 1.2745841559080873e-05), (11, 1.4847721104160883e-05), (12, 1.3949123967904598e-05), (13, 2.3064048946253024e-05), (14, 4.2753621528390795e-05), (15, 5.233004776528105e-05), (16, 4.859620821662247e-05), (17, 3.437956911511719e-05), (18, 3.581248165573925e-05), (19, 6.199663039296865e-05), (20, 9.683000098448247e-05), (21, 7.69281541579403e-05), (22, 7.759332947898656e-05), (23, 0.00013203448907006532), (24, 0.0002994193055201322), (25, 0.0004382456245366484), (26, 0.0002873630146495998), (27, 0.00033829358289949596)]
+
+ 92%|█████████▏| 138/150 [28:04<01:58,  9.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 66, Spatial MSE: [(0, 1.6428480194008444e-06), (1, 2.1526295313378796e-05), (2, 5.691493788617663e-05), (3, 8.014177183213178e-06), (4, 8.19203705759719e-06), (5, 1.4378847481566481e-05), (6, 2.0377463442855515e-05), (7, 3.66640197171364e-05), (8, 5.259496901999228e-05), (9, 6.151721754577011e-05), (10, 4.006452581961639e-05), (11, 5.814323594677262e-05), (12, 4.408201857586391e-05), (13, 0.00036208966048434377), (14, 0.0014487183652818203), (15, 0.0011197620769962668), (16, 0.024782616645097733), (17, 0.00012801621051039547), (18, 0.00017125115846283734), (19, 0.002174975350499153), (20, 0.0021992006804794073), (21, 0.00028308157925494015), (22, 0.00036631349939852953), (23, 0.0007606091676279902), (24, 0.001399100641719997), (25, 0.0035358124878257513), (26, 0.007053726818412542), (27, 0.0035710595548152924)], Temporal MSE: [(0, 3.3182482184201945e-06), (1, 2.0363791009003762e-06), (2, 2.393699787717196e-06), (3, 2.79680170933716e-06), (4, 3.7229631288937526e-06), (5, 4.473520675674081e-06), (6, 2.9812073989887722e-05), (7, 1.6419518942711875e-05), (8, 1.4767511856916826e-05), (9, 1.1647607607301325e-05), (10, 1.570098174852319e-05), (11, 1.8411967175779864e-05), (12, 1.962111855391413e-05), (13, 3.612846921896562e-05), (14, 6.136720912763849e-05), (15, 8.036844519665465e-05), (16, 6.53424795018509e-05), (17, 4.952822928316891e-05), (18, 6.095513526815921e-05), (19, 9.379463881487027e-05), (20, 0.00013292219955474138), (21, 0.00011769443517550826), (22, 0.00011779203487094492), (23, 0.00018914754036813974), (24, 0.00037508123205043375), (25, 0.0005814140313304961), (26, 0.00040311162592843175), (27, 0.0005428183358162642)]
+
+ 93%|█████████▎| 139/150 [28:16<01:54, 10.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 60, Spatial MSE: [(0, 1.7716943148116115e-06), (1, 2.310294621565845e-05), (2, 6.216046313056722e-05), (3, 8.801805051916745e-06), (4, 8.926445843826514e-06), (5, 1.5176304259512108e-05), (6, 2.195573688368313e-05), (7, 4.171848195255734e-05), (8, 5.469647658173926e-05), (9, 6.17919722571969e-05), (10, 4.43135759269353e-05), (11, 5.822281673317775e-05), (12, 4.323936445871368e-05), (13, 8.685404463903978e-05), (14, 0.0001871758431661874), (15, 0.00015610606351401657), (16, 9.690554725239053e-05), (17, 9.999328904086724e-05), (18, 0.00013782127643935382), (19, 0.00017380945791956037), (20, 0.00015584699576720595), (21, 0.0002754740708041936), (22, 0.0003751237818505615), (23, 0.0008076559170149267), (24, 0.0015113994013518095), (25, 0.00366982095874846), (26, 0.007726310286670923), (27, 0.0038762325420975685)], Temporal MSE: [(0, 3.4849870189646026e-06), (1, 2.192303782067029e-06), (2, 2.6992436232831096e-06), (3, 2.9369032290560426e-06), (4, 3.969570570916403e-06), (5, 4.87932129544788e-06), (6, 3.4991342545254156e-05), (7, 1.8620481569087133e-05), (8, 1.6915579180931672e-05), (9, 1.2700300430878997e-05), (10, 1.7259682863368653e-05), (11, 1.9727069229702465e-05), (12, 1.7787031538318843e-05), (13, 3.0844708817312494e-05), (14, 6.238809146452695e-05), (15, 7.279076817212626e-05), (16, 6.225504330359399e-05), (17, 4.3945354263996705e-05), (18, 4.770080704474822e-05), (19, 8.201302989618853e-05), (20, 0.00012312275066506118), (21, 0.00010631410987116396), (22, 0.00010226049926131964), (23, 0.00018194926087744534), (24, 0.00040903306216932833), (25, 0.0006228411803022027), (26, 0.00044826336670666933), (27, 0.0008142593433149159)]
+
+ 93%|█████████▎| 140/150 [28:27<01:46, 10.64s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 54, Spatial MSE: [(0, 1.917150257213507e-06), (1, 2.5228115191566758e-05), (2, 6.914280675118789e-05), (3, 9.19609192351345e-06), (4, 9.652118933445308e-06), (5, 1.696887738944497e-05), (6, 2.5243783966288902e-05), (7, 4.558464206638746e-05), (8, 6.157966708997265e-05), (9, 6.900834705447778e-05), (10, 4.6520595788024366e-05), (11, 6.308012234512717e-05), (12, 4.427901149028912e-05), (13, 0.00034664833219721913), (14, 0.0013366310158744454), (15, 0.0011086900485679507), (16, 0.024858340620994568), (17, 0.00014218271826393902), (18, 0.00017202029994223267), (19, 0.002198541536927223), (20, 0.002224160125479102), (21, 0.0003062443865928799), (22, 0.00044073633034713566), (23, 0.0008917067898437381), (24, 0.0016578655922785401), (25, 0.003854190232232213), (26, 0.007517530117183924), (27, 0.004389315843582153)], Temporal MSE: [(0, 3.727060175151564e-06), (1, 2.372350536461454e-06), (2, 2.9580573936982546e-06), (3, 3.104127472397522e-06), (4, 4.380645350465784e-06), (5, 5.362730462366017e-06), (6, 4.2684197978815064e-05), (7, 2.076218879665248e-05), (8, 1.8925957192550413e-05), (9, 1.4038374501978979e-05), (10, 1.8219396224594675e-05), (11, 2.0051724277436733e-05), (12, 2.081238926621154e-05), (13, 4.024996451335028e-05), (14, 7.262991857714951e-05), (15, 9.930749365594238e-05), (16, 7.865988300181925e-05), (17, 5.831141606904566e-05), (18, 6.536110595334321e-05), (19, 0.00011185918265255168), (20, 0.0001652438222663477), (21, 0.00014067007577978075), (22, 0.0001363666233373806), (23, 0.00022706355957780033), (24, 0.00048075529048219323), (25, 0.0007034860318526626), (26, 0.0005147699266672134), (27, 0.001360356225632131)]
+
+ 94%|█████████▍| 141/150 [28:38<01:36, 10.73s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 48, Spatial MSE: [(0, 2.109882871081936e-06), (1, 2.7636662707664073e-05), (2, 7.462787471013144e-05), (3, 9.82421261142008e-06), (4, 1.05145536508644e-05), (5, 1.7955766452359967e-05), (6, 2.5983455998357385e-05), (7, 4.578947482514195e-05), (8, 6.886744813527912e-05), (9, 7.584110426250845e-05), (10, 5.3733318054582924e-05), (11, 5.8214282034896314e-05), (12, 4.532645471044816e-05), (13, 0.00020296286675147712), (14, 0.0008241579052992165), (15, 0.0006625006790272892), (16, 0.012617107480764389), (17, 0.00013503555965144187), (18, 0.00017327186651527882), (19, 0.0012179941404610872), (20, 0.0012234804453328252), (21, 0.00031877466244623065), (22, 0.00041318440344184637), (23, 0.0009231703588739038), (24, 0.0016963667003437877), (25, 0.0036496103275567293), (26, 0.0074858469888567924), (27, 0.0047429692931473255)], Temporal MSE: [(0, 4.099382749700453e-06), (1, 2.605129566290998e-06), (2, 3.293423333161627e-06), (3, 3.345344111949089e-06), (4, 4.714764600066701e-06), (5, 5.861566933162976e-06), (6, 4.8441517719766125e-05), (7, 2.175301779061556e-05), (8, 2.1272910089464858e-05), (9, 1.423164940206334e-05), (10, 1.849694126576651e-05), (11, 2.0442366803763434e-05), (12, 1.975982013391331e-05), (13, 3.767280941247009e-05), (14, 7.276007818290964e-05), (15, 9.411191422259435e-05), (16, 7.560342783108354e-05), (17, 5.491077899932861e-05), (18, 6.024582762620412e-05), (19, 0.00010612834739731625), (20, 0.00016046960081439465), (21, 0.00013726265751756728), (22, 0.0001318144059041515), (23, 0.0002246490039397031), (24, 0.0005254135467112064), (25, 0.00074910803232342), (26, 0.0006087044021114707), (27, 0.002017854480072856)]
+
+ 95%|█████████▍| 142/150 [28:51<01:30, 11.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 42, Spatial MSE: [(0, 2.5213960270775715e-06), (1, 3.378561814315617e-05), (2, 9.045394108397886e-05), (3, 1.1733261089830194e-05), (4, 1.2501368473749608e-05), (5, 2.2062924472265877e-05), (6, 3.55571937689092e-05), (7, 5.8104964409722015e-05), (8, 8.478912786813453e-05), (9, 8.999856800073758e-05), (10, 6.604234658880159e-05), (11, 7.988788274815306e-05), (12, 5.885605787625536e-05), (13, 0.0004046240064781159), (14, 0.0017481839749962091), (15, 0.0017339939950034022), (16, 0.037497151643037796), (17, 0.00018653017468750477), (18, 0.00024098189896903932), (19, 0.0032602243591099977), (20, 0.00327214109711349), (21, 0.0003800690174102783), (22, 0.0005501359119080007), (23, 0.0011742361821234226), (24, 0.00229523959569633), (25, 0.004588164854794741), (26, 0.009514868259429932), (27, 0.0067823780700564384)], Temporal MSE: [(0, 4.9865871005749796e-06), (1, 3.1599142857885454e-06), (2, 4.198139777145116e-06), (3, 4.099480520380894e-06), (4, 5.799493919766974e-06), (5, 7.4069048423552886e-06), (6, 5.854884147993289e-05), (7, 2.8358655981719494e-05), (8, 2.9104037821525708e-05), (9, 1.8484084648662247e-05), (10, 2.4802871848805808e-05), (11, 2.8442382244975306e-05), (12, 2.9048322176095098e-05), (13, 5.129413693794049e-05), (14, 0.00010002931230701506), (15, 0.0001377246662741527), (16, 9.889809734886512e-05), (17, 7.53376298234798e-05), (18, 8.548152254661545e-05), (19, 0.00014433245814871043), (20, 0.00021688932611141354), (21, 0.00019846056238748133), (22, 0.00018166603695135564), (23, 0.00030496696126647294), (24, 0.0006569908000528812), (25, 0.0009938189759850502), (26, 0.0008450083550997078), (27, 0.004010890610516071)]
+
+ 95%|█████████▌| 143/150 [29:04<01:22, 11.85s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 36, Spatial MSE: [(0, 3.0109256385912886e-06), (1, 3.916484274668619e-05), (2, 0.00010233229113509879), (3, 1.336960576736601e-05), (4, 1.4933549209672492e-05), (5, 2.641542232595384e-05), (6, 4.252567669027485e-05), (7, 6.555048457812518e-05), (8, 0.00010404224303783849), (9, 0.00010476615716470405), (10, 8.147448534145951e-05), (11, 9.388259059051052e-05), (12, 7.456543971784413e-05), (13, 0.0006209369748830795), (14, 0.0026548774912953377), (15, 0.0023035111371427774), (16, 0.049975235015153885), (17, 0.00024639637558721006), (18, 0.00030286013497971), (19, 0.00418747216463089), (20, 0.004229821264743805), (21, 0.00046144210500642657), (22, 0.000656500575132668), (23, 0.0014193743700161576), (24, 0.002769328420981765), (25, 0.005654599983245134), (26, 0.01095197256654501), (27, 0.010174563154578209)], Temporal MSE: [(0, 5.7370039030502085e-06), (1, 3.7720383261330426e-06), (2, 5.169366886548232e-06), (3, 4.869442363997223e-06), (4, 6.9484362938965205e-06), (5, 9.29279576666886e-06), (6, 7.846488733775914e-05), (7, 3.2003728847485036e-05), (8, 3.5306071367813274e-05), (9, 2.215146378148347e-05), (10, 3.0047780455788597e-05), (11, 3.6312481825007126e-05), (12, 4.0851689846022055e-05), (13, 6.574374856427312e-05), (14, 0.0001284351310459897), (15, 0.0001550300803501159), (16, 0.00011722570343408734), (17, 9.253525786334649e-05), (18, 0.00010483619553269818), (19, 0.00017411369481123984), (20, 0.0002605376357678324), (21, 0.00023460554075427353), (22, 0.0002276317245559767), (23, 0.0003841932921204716), (24, 0.000811097037512809), (25, 0.0012390960473567247), (26, 0.0011299100005999207), (27, 0.007548702880740166)]
+
+ 96%|█████████▌| 144/150 [29:15<01:10, 11.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 30, Spatial MSE: [(0, 3.4842846616811585e-06), (1, 4.527076816884801e-05), (2, 0.0001111505480366759), (3, 1.5619998521287926e-05), (4, 1.7188325728056952e-05), (5, 2.9485581762855873e-05), (6, 5.006636638427153e-05), (7, 7.615884533151984e-05), (8, 0.00011715325672412291), (9, 0.00011369245476089418), (10, 9.475580009166151e-05), (11, 9.4323288067244e-05), (12, 7.552722672699019e-05), (13, 0.000468559330329299), (14, 0.002552175894379616), (15, 0.0027762565296143293), (16, 0.062122780829668045), (17, 0.0002754251181613654), (18, 0.0003387770848348737), (19, 0.0052784644067287445), (20, 0.0053468309342861176), (21, 0.000529558805283159), (22, 0.000743588141631335), (23, 0.0015935443807393312), (24, 0.0032678840216249228), (25, 0.0061439024284482), (26, 0.011988593265414238), (27, 0.016679175198078156)], Temporal MSE: [(0, 6.947664132894715e-06), (1, 4.47837828687625e-06), (2, 6.245323675102554e-06), (3, 5.556230462389067e-06), (4, 8.226979844039306e-06), (5, 1.1229491974518169e-05), (6, 9.366029553348199e-05), (7, 3.7884441553615034e-05), (8, 4.4575546780833974e-05), (9, 2.569538446550723e-05), (10, 3.422549707465805e-05), (11, 3.770590774365701e-05), (12, 3.6627396184485406e-05), (13, 6.744974234607071e-05), (14, 0.0001388079981552437), (15, 0.00017550315533299), (16, 0.00012485242041293532), (17, 0.00010584585834294558), (18, 0.00012018848065054044), (19, 0.00020075403153896332), (20, 0.00029431944130919874), (21, 0.00027953044627793133), (22, 0.0002595398691482842), (23, 0.00044585627620108426), (24, 0.0009818286634981632), (25, 0.0015019397251307964), (26, 0.0015558903105556965), (27, 0.014594413340091705)]
+
+ 97%|█████████▋| 145/150 [29:26<00:57, 11.42s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 24, Spatial MSE: [(0, 4.396531494421652e-06), (1, 5.4093899962026626e-05), (2, 0.0001292620290769264), (3, 1.867000901256688e-05), (4, 2.171388223359827e-05), (5, 3.563373684301041e-05), (6, 6.876220868434757e-05), (7, 7.66790471971035e-05), (8, 0.00012244981189724058), (9, 0.00014748700778000057), (10, 0.00011360382632119581), (11, 0.00011815843754447997), (12, 9.024771861732006e-05), (13, 0.0004268956836313009), (14, 0.0015236446633934975), (15, 0.0012583275092765689), (16, 0.025094512850046158), (17, 0.00026062646065838635), (18, 0.0003916680288966745), (19, 0.002254668390378356), (20, 0.0022941092029213905), (21, 0.0005774814635515213), (22, 0.0008198670693673193), (23, 0.0017991421045735478), (24, 0.003958890680223703), (25, 0.00703013502061367), (26, 0.013475010171532631), (27, 0.033334530889987946)], Temporal MSE: [(0, 8.35369064589031e-06), (1, 5.5940899983397685e-06), (2, 7.953993190312758e-06), (3, 6.409865818568505e-06), (4, 1.0451271919009741e-05), (5, 1.4083489077165723e-05), (6, 0.00011952655768254772), (7, 4.601636828738265e-05), (8, 5.9002071793656796e-05), (9, 3.291394023108296e-05), (10, 4.355251439847052e-05), (11, 4.651155904866755e-05), (12, 4.492574225878343e-05), (13, 7.657321111764759e-05), (14, 0.00014854413166176528), (15, 0.00016844183846842498), (16, 0.00013778101128991693), (17, 0.00011058849486289546), (18, 0.00011543955042725429), (19, 0.00020890103769488633), (20, 0.00031879410380497575), (21, 0.00027498771669343114), (22, 0.00026607190375216305), (23, 0.0005065573495812714), (24, 0.001213640091009438), (25, 0.0018314174376428127), (26, 0.002334512071684003), (27, 0.030735308304429054)]
+
+ 97%|█████████▋| 146/150 [29:37<00:44, 11.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 18, Spatial MSE: [(0, 5.5670702749921475e-06), (1, 7.386499783024192e-05), (2, 0.00016520089411642402), (3, 2.364897954976186e-05), (4, 2.877072438423056e-05), (5, 4.5584234612761065e-05), (6, 9.041762677952647e-05), (7, 0.00011348997941240668), (8, 0.00018914087559096515), (9, 0.0001862185454228893), (10, 0.0001642498536966741), (11, 0.00013967265840619802), (12, 0.0001126174975070171), (13, 0.0003329768078401685), (14, 0.0016475539887323976), (15, 0.0019731358624994755), (16, 0.03777575120329857), (17, 0.00040024903137236834), (18, 0.0003971819533035159), (19, 0.0033020980190485716), (20, 0.0033726010005921125), (21, 0.0006988702807575464), (22, 0.0010436270385980606), (23, 0.002228453755378723), (24, 0.005175070837140083), (25, 0.00899135135114193), (26, 0.01711656153202057), (27, 0.08013428002595901)], Temporal MSE: [(0, 1.0475139788468368e-05), (1, 6.87290321366163e-06), (2, 1.0780774573504459e-05), (3, 7.964985343278386e-06), (4, 1.3159189620637335e-05), (5, 1.8440023268340155e-05), (6, 0.0001642278366489336), (7, 6.275876512518153e-05), (8, 8.218739094445482e-05), (9, 4.4677883124677464e-05), (10, 5.594127287622541e-05), (11, 5.923334538238123e-05), (12, 5.182493623578921e-05), (13, 9.11868191906251e-05), (14, 0.00017799055785872042), (15, 0.00020627978665288538), (16, 0.00017281502368859947), (17, 0.0001493964809924364), (18, 0.00013845079229213297), (19, 0.0002502144780009985), (20, 0.0004030198324471712), (21, 0.00034314714139327407), (22, 0.0003399857669137418), (23, 0.000651257811114192), (24, 0.0016423038905486465), (25, 0.0025127725675702095), (26, 0.0037849019281566143), (27, 0.06878335028886795)]
+
+ 98%|█████████▊| 147/150 [29:46<00:31, 10.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 12, Spatial MSE: [(0, 7.568662113044411e-06), (1, 9.976403089240193e-05), (2, 0.00022717575484421104), (3, 3.2538027880946174e-05), (4, 4.021967833978124e-05), (5, 6.000491339364089e-05), (6, 0.00014307786477729678), (7, 0.00014124304288998246), (8, 0.00023314902500715107), (9, 0.0002482682466506958), (10, 0.00024254467280115932), (11, 0.00018046273908112198), (12, 0.00015816847735550255), (13, 0.0002611882227938622), (14, 0.0007063205121085048), (15, 0.0008879207307472825), (16, 0.012476221658289433), (17, 0.0004145285929553211), (18, 0.0005112366052344441), (19, 0.001567302388139069), (20, 0.0016182569088414311), (21, 0.0008724282961338758), (22, 0.0013447738019749522), (23, 0.002800135873258114), (24, 0.007127745542675257), (25, 0.013414356857538223), (26, 0.025237536057829857), (27, 0.2607242465019226)], Temporal MSE: [(0, 1.3941684301244095e-05), (1, 9.411873179487884e-06), (2, 1.5210048331937287e-05), (3, 1.0158495570067316e-05), (4, 1.8276641640113667e-05), (5, 2.5719495170051232e-05), (6, 0.00023310421966016293), (7, 8.845081902109087e-05), (8, 0.0001224119623657316), (9, 6.201693759066984e-05), (10, 7.854044815758243e-05), (11, 8.528432226739824e-05), (12, 6.893518002470955e-05), (13, 0.00011334331065881997), (14, 0.00021447357721626759), (15, 0.00022839769371785223), (16, 0.00019319077546242625), (17, 0.0001701555011095479), (18, 0.0001574407215230167), (19, 0.0002978257543873042), (20, 0.00047738684224896133), (21, 0.0003978539607487619), (22, 0.00040652096504345536), (23, 0.0008488969760946929), (24, 0.0022741921711713076), (25, 0.0033964728936553), (26, 0.007210336625576019), (27, 0.16599541902542114)]
+
+ 99%|█████████▊| 148/150 [29:55<00:20, 10.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 6, Spatial MSE: [(0, 1.3297672012413386e-05), (1, 0.0001756383862812072), (2, 0.0003457034763414413), (3, 5.644362681778148e-05), (4, 7.437386375386268e-05), (5, 0.00010008862591348588), (6, 0.00026074214838445187), (7, 0.00022396243002731353), (8, 0.00040005368646234274), (9, 0.0004332534153945744), (10, 0.0004513610037975013), (11, 0.00032221979927271605), (12, 0.00030693289590999484), (13, 0.0014786067185923457), (14, 0.006165237631648779), (15, 0.005320553667843342), (16, 0.11423464119434357), (17, 0.0008063729619607329), (18, 0.0010063707595691085), (19, 0.009881426580250263), (20, 0.010027828626334667), (21, 0.0017826808616518974), (22, 0.0025634262710809708), (23, 0.004805737640708685), (24, 0.013690420426428318), (25, 0.0322888009250164), (26, 0.0611051470041275), (27, 1.432276964187622)], Temporal MSE: [(0, 2.345023131056223e-05), (1, 1.597175833012443e-05), (2, 2.8160015062894672e-05), (3, 1.7218302673427388e-05), (4, 3.1360315915662795e-05), (5, 4.6108776587061584e-05), (6, 0.0004046912072226405), (7, 0.00015489064389839768), (8, 0.00022684960276819766), (9, 0.0001150610187323764), (10, 0.00014313263818621635), (11, 0.00016454867727588862), (12, 0.00014289958926383406), (13, 0.00022245777654461563), (14, 0.00037326160236261785), (15, 0.00043104044743813574), (16, 0.0003156730090267956), (17, 0.0002888532471843064), (18, 0.00030170060927048326), (19, 0.0005207949434407055), (20, 0.0007593372138217092), (21, 0.0006756414659321308), (22, 0.0007520179497078061), (23, 0.0015260082436725497), (24, 0.0038249939680099487), (25, 0.006326615344733), (26, 0.01886596344411373), (27, 0.49954891204833984)]
+
+ 99%|█████████▉| 149/150 [30:05<00:10, 10.25s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 0, Spatial MSE: [(0, 8.23506634333171e-05), (1, 0.0006513071712106466), (2, 0.0018150057876482606), (3, 0.000439174851635471), (4, 0.0004819110035896301), (5, 0.000568901828955859), (6, 0.001275374204851687), (7, 0.0007775356061756611), (8, 0.0014218207215890288), (9, 0.001747057307511568), (10, 0.001781304250471294), (11, 0.0012973842676728964), (12, 0.0015072987880557775), (13, 0.0030540660955011845), (14, 0.00803564116358757), (15, 0.007809244561940432), (16, 0.12809288501739502), (17, 0.0023902023676782846), (18, 0.004746998194605112), (19, 0.016272712498903275), (20, 0.016051743179559708), (21, 0.015881037339568138), (22, 0.03051028773188591), (23, 0.058154210448265076), (24, 0.12763318419456482), (25, 0.7116453051567078), (26, 1.1722790002822876), (27, 5.009771823883057)], Temporal MSE: [(0, 0.00014519953401759267), (1, 7.784728950355202e-05), (2, 0.00012221347424201667), (3, 7.237662066472694e-05), (4, 0.00020988982578273863), (5, 0.00020933302585035563), (6, 0.001226296997629106), (7, 0.0007620968390256166), (8, 0.0008956785895861685), (9, 0.0004753494868054986), (10, 0.000703325669746846), (11, 0.0008132553193718195), (12, 0.0005684020579792559), (13, 0.0008986559114418924), (14, 0.0016042107017710805), (15, 0.0012223452795296907), (16, 0.0010685864835977554), (17, 0.000989668071269989), (18, 0.0012278666254132986), (19, 0.002019760897383094), (20, 0.0026277790311723948), (21, 0.002886566100642085), (22, 0.006910383235663176), (23, 0.013639295473694801), (24, 0.03175613656640053), (25, 0.0640583261847496), (26, 0.09731539338827133), (27, 0.8625045418739319)]
+
+100%|██████████| 150/150 [30:16<00:00, 10.31s/it]
+100%|██████████| 150/150 [30:16<00:00, 12.11s/it]
+/home/jin509/anaconda3/envs/opendit/lib/python3.10/site-packages/torch/nn/modules/conv.py:605: UserWarning: Plan failed with a cudnnException: CUDNN_BACKEND_EXECUTION_PLAN_DESCRIPTOR: cudnnFinalize Descriptor Failed cudnn_status: CUDNN_STATUS_NOT_SUPPORTED (Triggered internally at ../aten/src/ATen/native/cudnn/Conv_v8.cpp:919.)
+  return F.conv3d(
+Saved spatial MLP MSE outputs to ./samples/opensora_plan/skip_mse/0_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora_plan/skip_mse/0_temporal_mlp_mse.pt
+
+  0%|          | 0/150 [00:00<?, ?it/s]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1896321439053281e-06), (1, 6.478853720182087e-06), (2, 0.00016343838069587946), (3, 2.0506829969235696e-05), (4, 9.082627002499066e-06), (5, 6.9340853769972455e-06), (6, 8.379173777939286e-06), (7, 6.040096195647493e-06), (8, 5.506716206582496e-06), (9, 5.621662239718717e-06), (10, 8.298725333588663e-06), (11, 1.1728585377568379e-05), (12, 1.3145889170118608e-05), (13, 2.229165875178296e-05), (14, 7.928547711344436e-05), (15, 0.00014473831106442958), (16, 0.0005510285845957696), (17, 6.61679805489257e-05), (18, 6.989851681282744e-05), (19, 7.447390089510009e-05), (20, 8.589673234382644e-05), (21, 7.621342228958383e-05), (22, 7.718498090980574e-05), (23, 7.242018909892067e-05), (24, 0.00014570170606020838), (25, 0.00033396194339729846), (26, 0.0004156012728344649), (27, 0.003700865199789405)], Temporal MSE: [(0, 1.6389449228881858e-06), (1, 1.1967738373641623e-06), (2, 1.6989538380585145e-06), (3, 1.5657680023650755e-06), (4, 1.1387733138690237e-06), (5, 1.5716467487436603e-06), (6, 7.860427899686329e-07), (7, 1.0917220834016916e-06), (8, 6.992315775278257e-07), (9, 8.845935326462495e-07), (10, 2.51173833021312e-06), (11, 1.7055562011591974e-06), (12, 1.5009535445642541e-06), (13, 4.423695372679504e-06), (14, 3.4402910387143493e-06), (15, 7.211599040601868e-06), (16, 3.0272212825366296e-06), (17, 6.678849786112551e-06), (18, 2.982039859489305e-06), (19, 3.0588014396926155e-06), (20, 3.7406684896268416e-06), (21, 3.2062137051980244e-06), (22, 1.1479741260700393e-05), (23, 6.5006620388885494e-06), (24, 1.7190348444273695e-05), (25, 1.4243835721572395e-05), (26, 2.830845232892898e-06), (27, 1.2065391274518333e-05)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1559884605105708e-08), (1, 1.4106586831985624e-07), (2, 1.429980557077215e-06), (3, 2.603601387818344e-07), (4, 1.0859829302489743e-07), (5, 8.583953103880049e-08), (6, 2.1264445138058363e-07), (7, 1.250117378504001e-07), (8, 1.0798547123158642e-07), (9, 1.3635334994432924e-07), (10, 1.418761286231529e-07), (11, 1.4591499564176047e-07), (12, 1.694233162652381e-07), (13, 1.5293861110876605e-07), (14, 2.872185689284379e-07), (15, 5.540763368117041e-07), (16, 2.1957705484965118e-06), (17, 4.303142588923947e-07), (18, 9.33833518956817e-07), (19, 6.276233648350171e-07), (20, 6.904253382344905e-07), (21, 9.946645604941295e-07), (22, 1.2706153711405932e-06), (23, 1.0039744893219904e-06), (24, 2.8544816359499237e-06), (25, 4.735638867714442e-06), (26, 5.536445769394049e-06), (27, 2.8150812795502134e-05)], Temporal MSE: [(0, 7.021015324681912e-09), (1, 9.394045541455398e-09), (2, 9.385164645436816e-09), (3, 1.1338905814284317e-08), (4, 1.4757127964060146e-08), (5, 2.5819030469165227e-08), (6, 1.3270138765619777e-08), (7, 1.7590563672342796e-08), (8, 1.7029456955697242e-08), (9, 2.2660335829982614e-08), (10, 3.026298855957066e-08), (11, 2.9581725513594392e-08), (12, 2.7141675573716384e-08), (13, 4.4716049529824886e-08), (14, 4.129835318167352e-08), (15, 7.588683814674368e-08), (16, 2.911021113050083e-08), (17, 5.913604539387052e-08), (18, 5.6396892489374295e-08), (19, 4.97199614812871e-08), (20, 9.131129274919658e-08), (21, 7.195146167759958e-08), (22, 1.234905369074113e-07), (23, 1.3025592693338695e-07), (24, 2.655470723311737e-07), (25, 3.48525560411872e-07), (26, 1.3230022943844233e-07), (27, 1.3379568031268718e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 9.203286595038662e-07), (1, 5.765402420365717e-06), (2, 0.0001541027013445273), (3, 1.4982843822508585e-05), (4, 6.956462129892316e-06), (5, 5.431386853160802e-06), (6, 5.208052243688144e-06), (7, 6.3162005972117186e-06), (8, 5.657427664118586e-06), (9, 6.49187995804823e-06), (10, 1.0222830496786628e-05), (11, 1.4942134839657228e-05), (12, 2.0526076696114615e-05), (13, 2.2269709006650373e-05), (14, 9.495913400314748e-05), (15, 0.00018367343000136316), (16, 0.0004394790739752352), (17, 5.2984963986091316e-05), (18, 6.874686368973926e-05), (19, 7.329369691433385e-05), (20, 9.78896496235393e-05), (21, 7.362207543337718e-05), (22, 5.511826748261228e-05), (23, 6.420431600417942e-05), (24, 0.0001900464267237112), (25, 0.00038899548235349357), (26, 0.00023020055959932506), (27, 0.002659458899870515)], Temporal MSE: [(0, 1.3147932804713491e-06), (1, 1.0191780575041776e-06), (2, 1.1550400813575834e-06), (3, 9.712640576253762e-07), (4, 9.052002951648319e-07), (5, 1.3549558843806153e-06), (6, 7.240594754875929e-07), (7, 1.4124872222964768e-06), (8, 6.853480272184242e-07), (9, 1.1875459904331365e-06), (10, 3.7707075080106733e-06), (11, 2.663682380443788e-06), (12, 2.13163821172202e-06), (13, 6.5004655880329665e-06), (14, 5.4120441745908465e-06), (15, 1.0847134944924619e-05), (16, 3.4067586511810077e-06), (17, 9.22717936191475e-06), (18, 3.3185617667186307e-06), (19, 3.945851858588867e-06), (20, 4.031713615404442e-06), (21, 3.2273842407448683e-06), (22, 1.0773485882964451e-05), (23, 6.7729192778642755e-06), (24, 1.8165090295951813e-05), (25, 1.22535393529688e-05), (26, 2.63493802776793e-06), (27, 7.926320904516615e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 1.0921614723713446e-08), (1, 1.2262002258012217e-07), (2, 1.2341798765191925e-06), (3, 2.3214462885334797e-07), (4, 1.0315167031649253e-07), (5, 8.169374154931575e-08), (6, 1.9407761442380433e-07), (7, 1.1900104368578468e-07), (8, 1.0395787342076801e-07), (9, 1.3076969196390564e-07), (10, 1.3750204175266845e-07), (11, 1.4081653887387802e-07), (12, 1.6392139912113635e-07), (13, 1.5773697725762759e-07), (14, 2.7470653662931e-07), (15, 5.059384875494288e-07), (16, 8.818378773867153e-07), (17, 4.1382560311831185e-07), (18, 8.700057492205815e-07), (19, 5.263667048893694e-07), (20, 6.627829520766682e-07), (21, 9.27489736568532e-07), (22, 1.212846655107569e-06), (23, 9.573913075655582e-07), (24, 2.7518137812876375e-06), (25, 4.570898454403505e-06), (26, 5.430177679954795e-06), (27, 2.710755870793946e-05)], Temporal MSE: [(0, 6.656254214476576e-09), (1, 8.615009150503283e-09), (2, 8.682585317387748e-09), (3, 1.0617913659416445e-08), (4, 1.3948556087939323e-08), (5, 2.451460368035896e-08), (6, 1.272414706932068e-08), (7, 1.6672343505774734e-08), (8, 1.6208600683853547e-08), (9, 2.141439203739992e-08), (10, 2.9070571727629613e-08), (11, 2.8482432412602066e-08), (12, 2.725658099222983e-08), (13, 4.414108900618885e-08), (14, 3.977529772214439e-08), (15, 7.409556701531983e-08), (16, 2.720753045082347e-08), (17, 5.399266456151963e-08), (18, 5.365075850249923e-08), (19, 4.687876398179469e-08), (20, 8.848295607322143e-08), (21, 6.744755154386439e-08), (22, 1.1893664009221538e-07), (23, 1.2284723993616353e-07), (24, 2.5666585656836105e-07), (25, 3.3941634569600865e-07), (26, 1.2777722702139727e-07), (27, 1.2728669673833792e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.3083782732792315e-06), (1, 3.664900532385218e-06), (2, 8.799409988569096e-05), (3, 1.5055462426971644e-05), (4, 6.5587228164076805e-06), (5, 4.554932729661232e-06), (6, 5.103275725559797e-06), (7, 3.875937181874178e-06), (8, 4.869778877036879e-06), (9, 6.081426818127511e-06), (10, 8.049916687014047e-06), (11, 8.45772956381552e-06), (12, 1.0520240721234586e-05), (13, 1.2351873010629788e-05), (14, 3.8604583096457645e-05), (15, 5.4536169045604765e-05), (16, 0.0002947820466943085), (17, 1.791460090316832e-05), (18, 3.118739550700411e-05), (19, 3.547657252056524e-05), (20, 2.7324678740114905e-05), (21, 1.7433460016036406e-05), (22, 3.357779860380106e-05), (23, 3.037015449081082e-05), (24, 6.838533590780571e-05), (25, 0.0001547436841065064), (26, 0.00017293209384661168), (27, 0.0039641521871089935)], Temporal MSE: [(0, 1.6666542705934262e-06), (1, 9.922187018673867e-07), (2, 1.6990791209536837e-06), (3, 1.3723789606956416e-06), (4, 1.0318085514882114e-06), (5, 1.3566133247877588e-06), (6, 7.38171877401328e-07), (7, 6.086223720558337e-07), (8, 1.1461282838354236e-06), (9, 1.1655566822810215e-06), (10, 1.5702213431723067e-06), (11, 1.2565700444611139e-06), (12, 1.6534468159079552e-06), (13, 2.9609352623083396e-06), (14, 3.829045908787521e-06), (15, 3.0323869850690244e-06), (16, 1.944103814821574e-06), (17, 2.1329692572180647e-06), (18, 1.8918773321274784e-06), (19, 2.1392522739915876e-06), (20, 3.629369302871055e-06), (21, 2.353183845116291e-06), (22, 2.7444641546026105e-06), (23, 3.725777560248389e-06), (24, 8.200129741453566e-06), (25, 1.1341963727318216e-05), (26, 3.0525500278599793e-06), (27, 3.607628741519875e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.153710194046198e-08), (1, 1.3931349940321525e-07), (2, 1.4020854450791376e-06), (3, 2.555988203312154e-07), (4, 1.1175863789958385e-07), (5, 8.631351988697133e-08), (6, 2.0945603296240733e-07), (7, 1.2448458619473968e-07), (8, 1.077561009310557e-07), (9, 1.366981621231389e-07), (10, 1.429547040743273e-07), (11, 1.4763344324819627e-07), (12, 1.7155497289422783e-07), (13, 1.695690627911972e-07), (14, 3.739827434401377e-07), (15, 6.629317681472457e-07), (16, 5.416809017333435e-06), (17, 4.7084242282835476e-07), (18, 9.840924803938833e-07), (19, 8.446971833109274e-07), (20, 7.751970656499907e-07), (21, 1.1372316066626809e-06), (22, 1.4426938150791102e-06), (23, 1.0469611879670992e-06), (24, 2.948027713500778e-06), (25, 4.815523425349966e-06), (26, 5.6487706387997605e-06), (27, 2.825525552907493e-05)], Temporal MSE: [(0, 6.9773649080673295e-09), (1, 9.157926861291799e-09), (2, 9.308062765001068e-09), (3, 1.1231012564394405e-08), (4, 1.472921340450739e-08), (5, 2.5678019710539957e-08), (6, 1.3372320140092597e-08), (7, 1.7467860047304384e-08), (8, 1.7094185622568148e-08), (9, 2.2687840939283888e-08), (10, 3.0459549549277654e-08), (11, 2.9707544868529112e-08), (12, 2.778958752003291e-08), (13, 4.9406857272060734e-08), (14, 4.3514408076816835e-08), (15, 7.860731443543045e-08), (16, 3.456779396060483e-08), (17, 7.086121200927664e-08), (18, 6.143071829001201e-08), (19, 5.5583512903467636e-08), (20, 9.67382050021115e-08), (21, 7.688925052207196e-08), (22, 1.424644864300717e-07), (23, 1.4415870452921808e-07), (24, 2.772470395484561e-07), (25, 3.546117568475893e-07), (26, 1.342154121175554e-07), (27, 1.3334926052266383e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 3.494735381082137e-07), (1, 8.86703844571457e-07), (2, 1.7740105249686167e-05), (3, 3.6202973205945455e-06), (4, 1.8142774251828087e-06), (5, 1.3884590543966624e-06), (6, 1.451989874112769e-06), (7, 1.4043606597624603e-06), (8, 1.8831555053111515e-06), (9, 2.741516254900489e-06), (10, 4.410183009895263e-06), (11, 5.148359832674032e-06), (12, 7.69920734455809e-06), (13, 7.689121048315428e-06), (14, 2.987299012602307e-05), (15, 6.633916200371459e-05), (16, 8.667288057040423e-05), (17, 1.5966821592883207e-05), (18, 1.797079312382266e-05), (19, 1.58322091010632e-05), (20, 1.630828955967445e-05), (21, 1.0181346624449361e-05), (22, 1.2697929378191475e-05), (23, 1.2688502465607598e-05), (24, 3.1790877983439714e-05), (25, 5.965985474176705e-05), (26, 4.1636179958004504e-05), (27, 0.0006811051280237734)], Temporal MSE: [(0, 4.152802830503788e-07), (1, 2.194249049125574e-07), (2, 2.7164838911630795e-07), (3, 2.791536530821759e-07), (4, 3.0806171480435296e-07), (5, 4.479592519146536e-07), (6, 3.054934722968028e-07), (7, 4.666277106935013e-07), (8, 3.05053958982171e-07), (9, 5.402691840572516e-07), (10, 1.4206889318302274e-06), (11, 8.829062494442041e-07), (12, 7.81492133228312e-07), (13, 2.1341541014407994e-06), (14, 2.3793168111296836e-06), (15, 3.5991702134197112e-06), (16, 1.3925157418270828e-06), (17, 2.831810206771479e-06), (18, 1.3243234207038768e-06), (19, 1.2517479035523138e-06), (20, 1.4439461892834515e-06), (21, 9.289326499128947e-07), (22, 2.03761737793684e-06), (23, 1.7543233070682618e-06), (24, 4.800769602297805e-06), (25, 2.9942409582872642e-06), (26, 9.070421924661787e-07), (27, 2.191549356211908e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 1.0644432002493431e-08), (1, 1.1832808866074629e-07), (2, 1.1976898122156854e-06), (3, 2.2544124078649475e-07), (4, 1.0092037427966716e-07), (5, 8.05326720865196e-08), (6, 1.9113537064185948e-07), (7, 1.1820720402511142e-07), (8, 1.0354776946996935e-07), (9, 1.2998411591524928e-07), (10, 1.3718596392209292e-07), (11, 1.4076769616622187e-07), (12, 1.6421472537331283e-07), (13, 1.888242877612356e-07), (14, 3.9194179635160253e-07), (15, 6.69438009026635e-07), (16, 3.730945991264889e-06), (17, 4.4006389998685336e-07), (18, 8.716428965271916e-07), (19, 7.334970746342151e-07), (20, 7.480348358512856e-07), (21, 9.866516847978346e-07), (22, 1.2720520317088813e-06), (23, 9.80252593762998e-07), (24, 2.777394229269703e-06), (25, 4.634012839233037e-06), (26, 5.418659384304192e-06), (27, 2.731331369432155e-05)], Temporal MSE: [(0, 6.47263753705829e-09), (1, 8.346013657956064e-09), (2, 8.560435915683229e-09), (3, 1.0507900327638708e-08), (4, 1.3784535290994882e-08), (5, 2.4070665460840246e-08), (6, 1.2562630047341372e-08), (7, 1.645806335659472e-08), (8, 1.6258745461072976e-08), (9, 2.119590369886737e-08), (10, 2.8843631483255194e-08), (11, 2.8444381072745273e-08), (12, 2.762837780778682e-08), (13, 4.9822755698869514e-08), (14, 4.1633946068486694e-08), (15, 7.659722456310192e-08), (16, 2.9748861152256723e-08), (17, 5.998603569423722e-08), (18, 5.613105358293069e-08), (19, 5.011413151123634e-08), (20, 9.035424142211923e-08), (21, 7.219868791707995e-08), (22, 1.269259399805378e-07), (23, 1.3233159279479878e-07), (24, 2.6411791509417526e-07), (25, 3.398820069833164e-07), (26, 1.2822476946894312e-07), (27, 1.2972708418601542e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 3.229404683224857e-06), (1, 3.039642069779802e-05), (2, 0.0007335548289120197), (3, 8.143799641402438e-05), (4, 3.127850141027011e-05), (5, 2.1504774849745445e-05), (6, 2.7597245207289234e-05), (7, 1.698056803434156e-05), (8, 1.4997032849350944e-05), (9, 1.5375420844065957e-05), (10, 2.1704774553654715e-05), (11, 2.678582677617669e-05), (12, 3.611812280723825e-05), (13, 5.036556103732437e-05), (14, 0.00022168323630467057), (15, 0.00044236434041522443), (16, 0.0018812183989211917), (17, 0.00019544598762877285), (18, 0.0002827034331858158), (19, 0.0003016139380633831), (20, 0.000366363296052441), (21, 0.00020297545415814966), (22, 0.00022898209863342345), (23, 0.00022944876400288194), (24, 0.0004661072453018278), (25, 0.0009755317005328834), (26, 0.0012250535655766726), (27, 0.015317236073315144)], Temporal MSE: [(0, 4.363965217635268e-06), (1, 4.072704541613348e-06), (2, 6.109279183874605e-06), (3, 5.629301085718907e-06), (4, 4.670046109822579e-06), (5, 4.909245490125613e-06), (6, 2.249584895253065e-06), (7, 3.554200930011575e-06), (8, 2.1824789655511267e-06), (9, 2.4187106646422762e-06), (10, 6.267373464652337e-06), (11, 5.01703834743239e-06), (12, 3.7983425045240438e-06), (13, 1.226343465532409e-05), (14, 8.48771924211178e-06), (15, 2.0781197235919535e-05), (16, 7.518110578530468e-06), (17, 2.5983717932831496e-05), (18, 9.657024747866672e-06), (19, 1.380881712975679e-05), (20, 1.5311794413719326e-05), (21, 9.224098903359845e-06), (22, 2.5037856175913475e-05), (23, 1.8980945242219605e-05), (24, 5.6455388403264806e-05), (25, 4.6029643272049725e-05), (26, 1.0221065167570487e-05), (27, 3.98814954678528e-05)]
+
+  1%|          | 1/150 [01:50<4:35:08, 110.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 1.1599751381652368e-08), (1, 1.4676777482236503e-07), (2, 1.4627526070398744e-06), (3, 2.638257114995213e-07), (4, 1.094991333161488e-07), (5, 8.594617639801072e-08), (6, 2.1697100294204574e-07), (7, 1.262056628092978e-07), (8, 1.0934178362731473e-07), (9, 1.3868643122805224e-07), (10, 1.4444117368839215e-07), (11, 1.48996321058803e-07), (12, 1.7334049573491939e-07), (13, 1.8494986875339237e-07), (14, 5.045776561019011e-07), (15, 8.708384484634735e-07), (16, 1.3346379091672134e-05), (17, 6.117627435742179e-07), (18, 1.2479026736400556e-06), (19, 1.3725532426178688e-06), (20, 1.0075833642986254e-06), (21, 1.4559972214556183e-06), (22, 1.644620169827249e-06), (23, 1.1096389016529429e-06), (24, 3.0872590741637396e-06), (25, 5.02747252539848e-06), (26, 5.6905741985247005e-06), (27, 2.8784617825294845e-05)], Temporal MSE: [(0, 6.9051995232882746e-09), (1, 9.2269925033861e-09), (2, 9.397728817361894e-09), (3, 1.1322954129866503e-08), (4, 1.4747701726491869e-08), (5, 2.588566161421113e-08), (6, 1.3649504637669452e-08), (7, 1.785326553260802e-08), (8, 1.744278499415941e-08), (9, 2.32430306112974e-08), (10, 3.107233936816556e-08), (11, 3.062322306845999e-08), (12, 2.867348491975008e-08), (13, 5.1250800936486485e-08), (14, 4.730645741801709e-08), (15, 8.415938879124951e-08), (16, 4.720281410186544e-08), (17, 1.0164316677219176e-07), (18, 7.437746774030529e-08), (19, 6.879339764509496e-08), (20, 1.1062869020861399e-07), (21, 9.169240655637623e-08), (22, 1.7673005459073465e-07), (23, 1.700155252137847e-07), (24, 2.976915425279003e-07), (25, 3.729759612269845e-07), (26, 1.3997460257542116e-07), (27, 1.394747073391045e-07)]
+
+  1%|▏         | 2/150 [02:02<2:08:50, 52.23s/it] Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 3.7732634154963307e-06), (1, 2.9535969588323496e-05), (2, 0.0008745297673158348), (3, 9.77742820396088e-05), (4, 3.984937211498618e-05), (5, 2.5851702957879752e-05), (6, 3.878350617014803e-05), (7, 2.047901762125548e-05), (8, 1.7501137335784733e-05), (9, 1.827503183449153e-05), (10, 2.5879719032673165e-05), (11, 3.030353764188476e-05), (12, 4.3791205825982615e-05), (13, 6.645329995080829e-05), (14, 0.00033133142278529704), (15, 0.0006216396577656269), (16, 0.0038539969827979803), (17, 0.00027692117146216333), (18, 0.0003070075181312859), (19, 0.0004530084552243352), (20, 0.0005712008569389582), (21, 0.0003614658198785037), (22, 0.00030211685225367546), (23, 0.00028310349443927407), (24, 0.000631489441730082), (25, 0.001573491608723998), (26, 0.0015112428227439523), (27, 0.02069963701069355)], Temporal MSE: [(0, 5.146400326339062e-06), (1, 4.796761913894443e-06), (2, 7.4430731729080435e-06), (3, 6.255534117372008e-06), (4, 5.542912276723655e-06), (5, 6.423608738259645e-06), (6, 2.6114919364772504e-06), (7, 4.0853024074749555e-06), (8, 2.8211848075443413e-06), (9, 2.7508197035785997e-06), (10, 6.882025445520412e-06), (11, 5.641702045977581e-06), (12, 4.578379048325587e-06), (13, 1.4574551641999278e-05), (14, 1.0370195923314895e-05), (15, 2.369994399487041e-05), (16, 1.0862625458685216e-05), (17, 3.287714207544923e-05), (18, 1.4078120329941157e-05), (19, 1.7901711544254795e-05), (20, 2.3518859961768612e-05), (21, 1.3632346963277087e-05), (22, 3.9029688196023926e-05), (23, 2.613951983221341e-05), (24, 7.099799404386431e-05), (25, 6.452186789829284e-05), (26, 1.3604119885712862e-05), (27, 4.8255682486342266e-05)]
+
+  2%|▏         | 3/150 [02:12<1:21:35, 33.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 1.0969955610562465e-08), (1, 1.248273946430345e-07), (2, 1.2054567832819885e-06), (3, 2.2558324985766376e-07), (4, 1.0344258782879479e-07), (5, 8.000706941402314e-08), (6, 1.9653005267628032e-07), (7, 1.1828893065057855e-07), (8, 1.0299233110799833e-07), (9, 1.3025938017108274e-07), (10, 1.364050206120737e-07), (11, 1.399547358005293e-07), (12, 1.6405059000135225e-07), (13, 1.782916001502599e-07), (14, 4.889409979114134e-07), (15, 9.211329938807467e-07), (16, 7.737357918813359e-06), (17, 4.863012463829364e-07), (18, 9.59577846515458e-07), (19, 1.0870634241655353e-06), (20, 9.31803469939041e-07), (21, 1.1146788665428176e-06), (22, 1.3457622571877437e-06), (23, 1.023668460220506e-06), (24, 2.862913788703736e-06), (25, 4.620054369297577e-06), (26, 5.4262632147583645e-06), (27, 2.7344471163814887e-05)], Temporal MSE: [(0, 6.528633189617494e-09), (1, 8.374393623000742e-09), (2, 8.37567348810353e-09), (3, 1.0398981231674043e-08), (4, 1.365605672987158e-08), (5, 2.3943549365412764e-08), (6, 1.2692106921008417e-08), (7, 1.660562354288686e-08), (8, 1.6312364792270273e-08), (9, 2.1535866423505468e-08), (10, 2.9368523613015896e-08), (11, 2.8999901147130913e-08), (12, 2.7571459781938756e-08), (13, 4.9681624147979164e-08), (14, 4.5020797756478714e-08), (15, 8.429733355796998e-08), (16, 3.322348263168351e-08), (17, 7.172369720365168e-08), (18, 6.363084992244694e-08), (19, 5.929668844828484e-08), (20, 1.0169665642933978e-07), (21, 7.909866184263592e-08), (22, 1.3379529661960987e-07), (23, 1.433099185987885e-07), (24, 2.7553070935937285e-07), (25, 3.6232458455742744e-07), (26, 1.3181404767692584e-07), (27, 1.3239194629477424e-07)]
+
+  3%|▎         | 4/150 [02:24<1:00:04, 24.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 870, Spatial MSE: [(0, 1.2138697456975933e-05), (1, 0.00010195153299719095), (2, 0.002567765535786748), (3, 0.00028270535403862596), (4, 0.00011342584912199527), (5, 7.538511272286996e-05), (6, 9.006081381812692e-05), (7, 7.814846321707591e-05), (8, 5.402130409493111e-05), (9, 4.900350177194923e-05), (10, 6.306129944277927e-05), (11, 7.883085345383734e-05), (12, 0.00011564677697606385), (13, 0.0001949334255186841), (14, 0.0008156239637173712), (15, 0.0014946828596293926), (16, 0.006551393307745457), (17, 0.0007119958172552288), (18, 0.0008321771747432649), (19, 0.0009642551303841174), (20, 0.0013460791669785976), (21, 0.0006911984528414905), (22, 0.0007810838287696242), (23, 0.0007851650007069111), (24, 0.0017015277408063412), (25, 0.004515858832746744), (26, 0.004590331111103296), (27, 0.058374032378196716)], Temporal MSE: [(0, 1.6839510863064788e-05), (1, 1.5434079614351504e-05), (2, 2.2292477297014557e-05), (3, 1.8777858713292517e-05), (4, 1.60949621204054e-05), (5, 1.776713179424405e-05), (6, 8.749680091568734e-06), (7, 1.1168766832270194e-05), (8, 8.544064257876016e-06), (9, 9.017275260703173e-06), (10, 2.1276811821735464e-05), (11, 1.7215483239851892e-05), (12, 1.5019106285762973e-05), (13, 5.0793525588233024e-05), (14, 3.0419374525081366e-05), (15, 7.241639104904607e-05), (16, 2.7903612135560252e-05), (17, 9.947402577381581e-05), (18, 3.544845094438642e-05), (19, 5.58428255317267e-05), (20, 6.38911296846345e-05), (21, 4.2369505536044016e-05), (22, 7.866387022659183e-05), (23, 6.192809087224305e-05), (24, 0.0002130134089384228), (25, 0.00022060080664232373), (26, 3.844053571810946e-05), (27, 0.00014720403123646975)]
+
+  3%|▎         | 5/150 [02:35<47:53, 19.81s/it]  Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 864, Spatial MSE: [(0, 1.5509294826188125e-05), (1, 0.0001390638790326193), (2, 0.003583398414775729), (3, 0.00038681188016198575), (4, 0.0001613996282685548), (5, 0.00010402155021438375), (6, 0.00015913890092633665), (7, 7.953395106596872e-05), (8, 6.862608279334381e-05), (9, 8.211492968257517e-05), (10, 0.00012245021935086697), (11, 0.0001279301941394806), (12, 0.00019938794139306992), (13, 0.0002804680843837559), (14, 0.0012253731256350875), (15, 0.002254962921142578), (16, 0.00871144887059927), (17, 0.0010825064964592457), (18, 0.0011710769031196833), (19, 0.0013729986967518926), (20, 0.002074009506031871), (21, 0.001029048697091639), (22, 0.0010231861378997564), (23, 0.001012711669318378), (24, 0.0021765483543276787), (25, 0.005783227737993002), (26, 0.005567959509789944), (27, 0.089744433760643)], Temporal MSE: [(0, 2.0984254661016166e-05), (1, 1.97422195924446e-05), (2, 2.757734364422504e-05), (3, 2.5127468688879162e-05), (4, 2.2265869120019488e-05), (5, 2.6532241463428363e-05), (6, 1.0896953426708933e-05), (7, 1.9975557734142058e-05), (8, 1.2403522305248771e-05), (9, 1.3044003935647197e-05), (10, 3.60791600542143e-05), (11, 2.8209915399202146e-05), (12, 2.379636134719476e-05), (13, 7.532645395258442e-05), (14, 4.7108856961131096e-05), (15, 0.0001106925483327359), (16, 4.680539859691635e-05), (17, 0.00016470465925522149), (18, 6.875000690342858e-05), (19, 0.0001009819534374401), (20, 0.0001506581174908206), (21, 6.232099985936657e-05), (22, 0.00010837524314410985), (23, 8.354290184797719e-05), (24, 0.00024354482593480498), (25, 0.0002695219300221652), (26, 4.9712947657098994e-05), (27, 0.00015463968156836927)]
+
+  4%|▍         | 6/150 [02:44<38:22, 15.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 858, Spatial MSE: [(0, 3.915800334652886e-05), (1, 0.0003487404028419405), (2, 0.008640573360025883), (3, 0.0009093370172195137), (4, 0.00038073211908340454), (5, 0.00025073328288272023), (6, 0.00028090013074688613), (7, 0.0002463827549945563), (8, 0.00017838914936874062), (9, 0.0002016392390942201), (10, 0.00029652751982212067), (11, 0.00031707031303085387), (12, 0.0005224486230872571), (13, 0.000744202989153564), (14, 0.002989466767758131), (15, 0.005812420975416899), (16, 0.016486091539263725), (17, 0.0026619802229106426), (18, 0.002654530107975006), (19, 0.0030915632378309965), (20, 0.005196023266762495), (21, 0.002444906858727336), (22, 0.0026691912207752466), (23, 0.0027972927782684565), (24, 0.006286704912781715), (25, 0.018052060157060623), (26, 0.015112520195543766), (27, 0.20875968039035797)], Temporal MSE: [(0, 5.588922795141116e-05), (1, 4.980919402441941e-05), (2, 6.893662794027478e-05), (3, 5.990526551613584e-05), (4, 5.2677180065074936e-05), (5, 6.439234130084515e-05), (6, 2.8645439670071937e-05), (7, 4.5413475163513795e-05), (8, 3.142310379189439e-05), (9, 3.635004395619035e-05), (10, 9.087708895094693e-05), (11, 7.002884376561269e-05), (12, 6.410471542039886e-05), (13, 0.0002023819397436455), (14, 0.00013013962598051876), (15, 0.00032471877057105303), (16, 0.0001434473233530298), (17, 0.0005078616668470204), (18, 0.00020153168588876724), (19, 0.00030158477602526546), (20, 0.0004337357822805643), (21, 0.0002049416070804), (22, 0.00023857702035456896), (23, 0.00020803224470000714), (24, 0.0006360404076986015), (25, 0.0009384757140651345), (26, 0.00014225435734260827), (27, 0.00046442760503850877)]
+
+  5%|▍         | 7/150 [02:52<32:10, 13.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 852, Spatial MSE: [(0, 4.5660044634132646e-07), (1, 2.877827000702382e-06), (2, 4.795179847860709e-05), (3, 8.0589215940563e-06), (4, 2.7233961645833915e-06), (5, 2.2081057977629825e-06), (6, 4.052175427204929e-06), (7, 7.1560757533006836e-06), (8, 5.595627044385765e-06), (9, 1.5448187696165405e-05), (10, 1.6373785911127925e-05), (11, 1.7603550077183172e-05), (12, 2.6919833544525318e-05), (13, 2.8321337595116347e-05), (14, 9.149345714831725e-05), (15, 0.00019890260591637343), (16, 0.0004429524124134332), (17, 6.406129978131503e-05), (18, 6.348790338961408e-05), (19, 7.453735452145338e-05), (20, 7.223874126793817e-05), (21, 4.3154785089427605e-05), (22, 6.809287879150361e-05), (23, 8.076392259681597e-05), (24, 0.00016584558761678636), (25, 0.0002263618807774037), (26, 0.0001579251402290538), (27, 0.001176256570033729)], Temporal MSE: [(0, 6.011977120579104e-07), (1, 3.8117858025543683e-07), (2, 5.346549301066261e-07), (3, 5.272084990792791e-07), (4, 4.0696258452044276e-07), (5, 6.573075097549008e-07), (6, 4.509619486725569e-07), (7, 6.44645979264169e-07), (8, 6.930941367500054e-07), (9, 1.4819329408055637e-06), (10, 3.7282770790625364e-06), (11, 3.3278488444921095e-06), (12, 3.6667825042968616e-06), (13, 7.2767320489219856e-06), (14, 1.44652958624647e-05), (15, 1.2180918929516338e-05), (16, 9.078720722754952e-06), (17, 2.0790359485545196e-05), (18, 8.641982276458293e-06), (19, 1.1880255442520138e-05), (20, 1.613084714335855e-05), (21, 6.9874558903393336e-06), (22, 7.500716947106412e-06), (23, 9.302443686465267e-06), (24, 3.218089113943279e-05), (25, 2.0410130673553795e-05), (26, 3.896769612765638e-06), (27, 2.4378498437727103e-06)]
+
+  5%|▌         | 8/150 [03:00<27:41, 11.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 846, Spatial MSE: [(0, 2.1920138806308387e-06), (1, 2.9044782422715798e-05), (2, 0.0006108079687692225), (3, 8.461782999802381e-05), (4, 2.8515283702290617e-05), (5, 1.8264108803123236e-05), (6, 3.126835508737713e-05), (7, 2.014072197198402e-05), (8, 1.4480133359029423e-05), (9, 1.3275887795316521e-05), (10, 1.698911000858061e-05), (11, 1.741865162330214e-05), (12, 2.5994520910899155e-05), (13, 3.0799037631368265e-05), (14, 0.00010203432611888275), (15, 0.00023132209025789052), (16, 0.0011757684405893087), (17, 8.21409557829611e-05), (18, 6.89465960022062e-05), (19, 0.00015485619951505214), (20, 0.00013305561151355505), (21, 9.006453910842538e-05), (22, 0.00014782065409235656), (23, 0.0001839388278312981), (24, 0.0004829132813028991), (25, 0.0008770246640779078), (26, 0.0011468991870060563), (27, 0.009454665705561638)], Temporal MSE: [(0, 2.871466449505533e-06), (1, 3.652571876955335e-06), (2, 4.358781097835163e-06), (3, 4.015759259345941e-06), (4, 3.6428257317311363e-06), (5, 4.8841834541235585e-06), (6, 2.278218971696333e-06), (7, 3.543134198480402e-06), (8, 1.8227096916234586e-06), (9, 2.7270818918623263e-06), (10, 8.968033398559783e-06), (11, 8.903321031539235e-06), (12, 4.6321488298417535e-06), (13, 9.6721214504214e-06), (14, 8.343087756657042e-06), (15, 1.5231163160933647e-05), (16, 6.235041837499011e-06), (17, 1.9604227418312803e-05), (18, 9.272687748307362e-06), (19, 1.5013551092124544e-05), (20, 3.636288602137938e-05), (21, 1.336478817393072e-05), (22, 2.7576135835261084e-05), (23, 2.0935469365213066e-05), (24, 6.67006679577753e-05), (25, 4.505704055191018e-05), (26, 1.806397813197691e-05), (27, 2.2031574189895764e-05)]
+
+  6%|▌         | 9/150 [03:08<24:53, 10.59s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 840, Spatial MSE: [(0, 8.383887688978575e-06), (1, 6.51163500151597e-05), (2, 0.0019041391788050532), (3, 0.000293591117952019), (4, 6.796753586968407e-05), (5, 3.498214209685102e-05), (6, 6.554881838383153e-05), (7, 3.484381522866897e-05), (8, 2.970950117742177e-05), (9, 3.299284435342997e-05), (10, 3.044585719180759e-05), (11, 4.803988122148439e-05), (12, 5.701298141502775e-05), (13, 0.0006728220032528043), (14, 0.0015142826596274972), (15, 0.0006156866438686848), (16, 0.015971003100275993), (17, 0.00015027452900540084), (18, 0.0004668437468353659), (19, 0.0009741667308844626), (20, 0.0002941031416412443), (21, 0.0005875711212866008), (22, 0.001022240030579269), (23, 0.0005975669482722878), (24, 0.0011756691383197904), (25, 0.003007422434166074), (26, 0.0044915881007909775), (27, 0.05489388108253479)], Temporal MSE: [(0, 1.1016332791768946e-05), (1, 7.64950891607441e-06), (2, 1.0526881851546932e-05), (3, 9.640576536185108e-06), (4, 8.079377039393876e-06), (5, 9.119825335801579e-06), (6, 4.6197119445423596e-06), (7, 5.4721522246836685e-06), (8, 3.787515879594139e-06), (9, 5.123532446305035e-06), (10, 8.49638854560908e-06), (11, 1.8606062440085225e-05), (12, 2.035420766333118e-05), (13, 8.292891288874671e-05), (14, 2.6845247703022324e-05), (15, 2.9685888875974342e-05), (16, 2.4632978238514625e-05), (17, 3.836409086943604e-05), (18, 2.2637257643509656e-05), (19, 2.477574162185192e-05), (20, 3.822122380370274e-05), (21, 4.353149779490195e-05), (22, 0.00012963870540261269), (23, 0.00018212117720395327), (24, 0.00027645929367281497), (25, 0.00014686089707538486), (26, 7.679678674321622e-05), (27, 2.1523812392842956e-05)]
+
+  7%|▋         | 10/150 [03:15<22:15,  9.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 834, Spatial MSE: [(0, 9.415883141628001e-06), (1, 0.0002176517155021429), (2, 0.0031631800811737776), (3, 0.00024268367269542068), (4, 0.00013863564527127892), (5, 0.00011612758680712432), (6, 8.213791443267837e-05), (7, 0.0002429368905723095), (8, 0.00014947743329685181), (9, 0.00022743713634554297), (10, 0.000314112170599401), (11, 0.0004574593622237444), (12, 0.0009104551281780005), (13, 0.0012437221594154835), (14, 0.004503331147134304), (15, 0.007231508381664753), (16, 0.022210756316781044), (17, 0.00309984409250319), (18, 0.0020179564598947763), (19, 0.0030152015388011932), (20, 0.0050607360899448395), (21, 0.002296110615134239), (22, 0.0026729742530733347), (23, 0.0024768488947302103), (24, 0.005423437338322401), (25, 0.01876792125403881), (26, 0.00970076397061348), (27, 0.035583384335041046)], Temporal MSE: [(0, 1.1425056072766893e-05), (1, 1.748675094859209e-05), (2, 2.4242768631665967e-05), (3, 1.93775867955992e-05), (4, 1.9548773707356304e-05), (5, 2.9859675123589113e-05), (6, 1.531558700662572e-05), (7, 3.063697658944875e-05), (8, 2.468752791173756e-05), (9, 4.651697599911131e-05), (10, 0.0001258709526155144), (11, 9.586607484379783e-05), (12, 0.00010882083006436005), (13, 0.0003367767494637519), (14, 0.0002949239278677851), (15, 0.0005385356489568949), (16, 0.00035883672535419464), (17, 0.0011540944688022137), (18, 0.0005210830713622272), (19, 0.0006374316290020943), (20, 0.001097261905670166), (21, 0.00048333994345739484), (22, 0.00027456574025563896), (23, 0.0003917748399544507), (24, 0.0008959655533544719), (25, 0.0020483224652707577), (26, 0.00020446364942472428), (27, 0.00026624114252626896)]
+
+  7%|▋         | 11/150 [03:23<20:36,  8.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 828, Spatial MSE: [(0, 1.3109153769619297e-05), (1, 9.786753071239218e-05), (2, 0.002765270881354809), (3, 0.0004391352122183889), (4, 0.00010246251622447744), (5, 5.4161500884220004e-05), (6, 8.997039549285546e-05), (7, 5.583819802268408e-05), (8, 4.518621426541358e-05), (9, 5.324208177626133e-05), (10, 4.946897752233781e-05), (11, 8.681020699441433e-05), (12, 0.00012573933054227382), (13, 0.0020168228074908257), (14, 0.004646511748433113), (15, 0.0016417211154475808), (16, 0.058890942484140396), (17, 0.0003777363453991711), (18, 0.0014548577601090074), (19, 0.0032917424105107784), (20, 0.0009198518237099051), (21, 0.001929195481352508), (22, 0.002828576834872365), (23, 0.0013872060226276517), (24, 0.002569731557741761), (25, 0.006631789728999138), (26, 0.00744057958945632), (27, 0.10003665089607239)], Temporal MSE: [(0, 1.786476423148997e-05), (1, 1.141820666816784e-05), (2, 1.663449074840173e-05), (3, 1.5415005691465922e-05), (4, 1.2922930181957781e-05), (5, 1.5148549209698103e-05), (6, 7.455444119841559e-06), (7, 7.972593266458716e-06), (8, 6.848973043815931e-06), (9, 8.617013918410521e-06), (10, 1.2700903425866272e-05), (11, 4.0680577512830496e-05), (12, 4.830503894481808e-05), (13, 0.00024083575408440083), (14, 7.110098522389308e-05), (15, 7.643546996405348e-05), (16, 7.533577445428818e-05), (17, 0.00010602760448819026), (18, 7.59965114411898e-05), (19, 7.19569725333713e-05), (20, 0.00010243042197544128), (21, 0.00012814986985176802), (22, 0.00035713473334908485), (23, 0.000524679955560714), (24, 0.00066185969626531), (25, 0.0002712037821765989), (26, 0.0001593063643667847), (27, 4.580502718454227e-05)]
+
+  8%|▊         | 12/150 [03:31<19:54,  8.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 822, Spatial MSE: [(0, 1.2527012586360797e-05), (1, 0.00026676905690692365), (2, 0.003857657080516219), (3, 0.00032031795126385987), (4, 0.0001940187212312594), (5, 0.00013588977162726223), (6, 7.912150613265112e-05), (7, 0.00016728958871681243), (8, 0.00013122634845785797), (9, 0.00019090536807198077), (10, 0.0002662016195245087), (11, 0.0002974133822135627), (12, 0.0005702183698303998), (13, 0.0028708414174616337), (14, 0.008059236221015453), (15, 0.0049837552942335606), (16, 0.08538907021284103), (17, 0.0013249623589217663), (18, 0.001884005032479763), (19, 0.005969297606498003), (20, 0.0033896947279572487), (21, 0.0025721341371536255), (22, 0.0037531487178057432), (23, 0.002490642946213484), (24, 0.006731299217790365), (25, 0.01344592310488224), (26, 0.008991816081106663), (27, 0.05159537494182587)], Temporal MSE: [(0, 1.4055666724743787e-05), (1, 2.595118530734908e-05), (2, 2.3205015168059617e-05), (3, 1.974762199097313e-05), (4, 2.5879615350277163e-05), (5, 2.9312135666259564e-05), (6, 1.895446803246159e-05), (7, 2.8946355087100528e-05), (8, 1.8794391507981345e-05), (9, 3.4158481867052615e-05), (10, 9.299210796598345e-05), (11, 0.00010094475146615878), (12, 0.00010075338650494814), (13, 0.0003851084620691836), (14, 0.00017564634617883712), (15, 0.00029448041459545493), (16, 0.0002092928916681558), (17, 0.000498154025990516), (18, 0.00023671054805163294), (19, 0.00036037861718796194), (20, 0.0008078329847194254), (21, 0.00042343276436440647), (22, 0.0005512113566510379), (23, 0.0007618195377290249), (24, 0.0010734561365097761), (25, 0.0007920851348899305), (26, 0.00024979098816402256), (27, 0.00029803160578012466)]
+
+  9%|▊         | 13/150 [03:39<19:33,  8.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 816, Spatial MSE: [(0, 1.6549347492400557e-05), (1, 0.00011667732178466395), (2, 0.0031425601337105036), (3, 0.0005273856222629547), (4, 0.0001155544159701094), (5, 6.201737414812669e-05), (6, 0.00010891509737120941), (7, 6.736169598298147e-05), (8, 5.163981404621154e-05), (9, 6.30119611741975e-05), (10, 6.25982356723398e-05), (11, 0.00015383941354230046), (12, 0.00034552946453914046), (13, 0.00818490982055664), (14, 0.015611952170729637), (15, 0.004435643088072538), (16, 0.14017122983932495), (17, 0.0006363385473378003), (18, 0.002173210959881544), (19, 0.008607416413724422), (20, 0.0026718468870967627), (21, 0.0029705220367759466), (22, 0.005530661903321743), (23, 0.0025120300706475973), (24, 0.004282874055206776), (25, 0.012165769934654236), (26, 0.010209355503320694), (27, 0.1759941428899765)], Temporal MSE: [(0, 2.2616261048824526e-05), (1, 1.382225764245959e-05), (2, 1.9246515876147896e-05), (3, 1.7251244571525604e-05), (4, 1.4138867300061975e-05), (5, 1.7733005734044127e-05), (6, 8.750671440793667e-06), (7, 8.910486940294504e-06), (8, 8.029258424357977e-06), (9, 1.0458446922712028e-05), (10, 1.6074944142019376e-05), (11, 0.00013328749628271908), (12, 0.0001694479287834838), (13, 0.0008480257820338011), (14, 0.00017682812176644802), (15, 0.00016070746642071754), (16, 0.00016668633907102048), (17, 0.00021964864572510123), (18, 0.00017210717487614602), (19, 0.00014678329171147197), (20, 0.0002010505268117413), (21, 0.00028540106723085046), (22, 0.0006967185181565583), (23, 0.0011544976150617003), (24, 0.0013860245235264301), (25, 0.00047505414113402367), (26, 0.00030605512438341975), (27, 0.0001036895700963214)]
+
+  9%|▉         | 14/150 [03:47<18:50,  8.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 810, Spatial MSE: [(0, 1.5832021745154634e-05), (1, 0.0003746884176507592), (2, 0.004551967605948448), (3, 0.00038082338869571686), (4, 0.00019539306231308728), (5, 0.00020897403010167181), (6, 0.000181463998160325), (7, 0.0006271010497584939), (8, 0.0004804636992048472), (9, 0.0008934021461755037), (10, 0.0012221719371154904), (11, 0.001625748467631638), (12, 0.0031828635837882757), (13, 0.010852649807929993), (14, 0.02441723458468914), (15, 0.017160926014184952), (16, 0.16932277381420135), (17, 0.0070692696608603), (18, 0.007045751437544823), (19, 0.014804613776504993), (20, 0.017044754698872566), (21, 0.006382360123097897), (22, 0.009311235509812832), (23, 0.0071252803318202496), (24, 0.017039548605680466), (25, 0.07060800492763519), (26, 0.03155647963285446), (27, 0.11992333829402924)], Temporal MSE: [(0, 2.3624341338290833e-05), (1, 3.880450458382256e-05), (2, 4.575006096274592e-05), (3, 3.598862895159982e-05), (4, 3.0761621019337326e-05), (5, 7.27316873962991e-05), (6, 4.324182009440847e-05), (7, 8.323610381921753e-05), (8, 9.797426173463464e-05), (9, 0.00016748454072512686), (10, 0.00033287322730757296), (11, 0.00038746348582208157), (12, 0.0005167845520190895), (13, 0.0017062774859368801), (14, 0.001141651184298098), (15, 0.0013053355505689979), (16, 0.0011282977648079395), (17, 0.0029601624701172113), (18, 0.0017095942748710513), (19, 0.002172687090933323), (20, 0.004047197289764881), (21, 0.002315731719136238), (22, 0.0013332015369087458), (23, 0.0019501186907291412), (24, 0.004175443202257156), (25, 0.009965506382286549), (26, 0.001166810281574726), (27, 0.0005552777438424528)]
+
+ 10%|█         | 15/150 [03:56<19:19,  8.59s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 804, Spatial MSE: [(0, 1.5148404600040521e-05), (1, 0.00011750800331356004), (2, 0.002718783915042877), (3, 0.0005246340879239142), (4, 0.00010588893201202154), (5, 5.672490442520939e-05), (6, 0.0001270580687560141), (7, 5.11254693265073e-05), (8, 4.615995203494094e-05), (9, 5.9934733144473284e-05), (10, 6.655900506302714e-05), (11, 0.00019837029685731977), (12, 0.0005215962301008403), (13, 0.011974531225860119), (14, 0.020920777693390846), (15, 0.006035157013684511), (16, 0.19277091324329376), (17, 0.0007276071119122207), (18, 0.0025559880305081606), (19, 0.011811323463916779), (20, 0.003776080906391144), (21, 0.0034114141017198563), (22, 0.006719967816025019), (23, 0.0029597640968859196), (24, 0.004900940228253603), (25, 0.01370047777891159), (26, 0.010563848540186882), (27, 0.1989712119102478)], Temporal MSE: [(0, 1.9252143829362467e-05), (1, 1.2253124623384792e-05), (2, 1.516954944236204e-05), (3, 1.48944636748638e-05), (4, 1.2654037163883913e-05), (5, 1.6527466868865304e-05), (6, 7.45005809221766e-06), (7, 8.90540104592219e-06), (8, 7.070375886542024e-06), (9, 9.90461376204621e-06), (10, 1.6467005480080843e-05), (11, 0.0002589474315755069), (12, 0.0002662762126419693), (13, 0.0011026356369256973), (14, 0.00023326057998929173), (15, 0.0001948992721736431), (16, 0.00021821007248945534), (17, 0.00026120926486328244), (18, 0.00024297345953527838), (19, 0.000187020399607718), (20, 0.00024591319379396737), (21, 0.0003615552850533277), (22, 0.0008566169999539852), (23, 0.00153124681673944), (24, 0.0017707727383822203), (25, 0.0006214421591721475), (26, 0.0003877870331052691), (27, 0.00014382839435711503)]
+
+ 11%|█         | 16/150 [04:04<18:33,  8.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 798, Spatial MSE: [(0, 1.4725645087310113e-05), (1, 0.000303668319247663), (2, 0.0043734097853302956), (3, 0.00035341939656063914), (4, 0.00016920568305067718), (5, 0.00014378556807059795), (6, 0.00010136224591406062), (7, 0.00018606703088153154), (8, 0.00012529996456578374), (9, 0.00016088578558992594), (10, 0.00023410932044498622), (11, 0.00038842030335217714), (12, 0.0009351969929412007), (13, 0.012463919818401337), (14, 0.022365888580679893), (15, 0.007654673885554075), (16, 0.1938195675611496), (17, 0.0013979962095618248), (18, 0.0029058444779366255), (19, 0.012808888219296932), (20, 0.005469026509672403), (21, 0.0037950545083731413), (22, 0.007398601621389389), (23, 0.00399730633944273), (24, 0.009182829409837723), (25, 0.022535357624292374), (26, 0.0125403618440032), (27, 0.14618976414203644)], Temporal MSE: [(0, 1.656313907005824e-05), (1, 2.7993870389764197e-05), (2, 2.4459668566123582e-05), (3, 2.087790380755905e-05), (4, 2.562458575994242e-05), (5, 3.0312385206343606e-05), (6, 2.3485568817704916e-05), (7, 2.9306211217772216e-05), (8, 2.1772171749034896e-05), (9, 4.175523645244539e-05), (10, 8.745544619159773e-05), (11, 0.00033369375159963965), (12, 0.00033839725074358284), (13, 0.0011833444004878402), (14, 0.0003184311790391803), (15, 0.0004008445830550045), (16, 0.00034904907806776464), (17, 0.0006534619606100023), (18, 0.00044843650539405644), (19, 0.0005458687082864344), (20, 0.0011747711105272174), (21, 0.0007839678437449038), (22, 0.0010089605348184705), (23, 0.0017601022263988853), (24, 0.002306740963831544), (25, 0.0016958892811089754), (26, 0.0004964678082615137), (27, 0.00036663335049524903)]
+
+ 11%|█▏        | 17/150 [04:12<18:35,  8.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 792, Spatial MSE: [(0, 1.875135421869345e-05), (1, 0.0001400788314640522), (2, 0.002934449352324009), (3, 0.000572535558603704), (4, 0.00011447220458649099), (5, 6.308204319793731e-05), (6, 0.00012385482841636986), (7, 6.263304385356605e-05), (8, 5.144833630765788e-05), (9, 6.864988972665742e-05), (10, 7.624572026543319e-05), (11, 0.00020311836851760745), (12, 0.0005573094822466373), (13, 0.014532598666846752), (14, 0.024363398551940918), (15, 0.006849297322332859), (16, 0.20607858896255493), (17, 0.000787061988376081), (18, 0.0026007243432104588), (19, 0.01303611509501934), (20, 0.004292953293770552), (21, 0.003638940630480647), (22, 0.007143758703023195), (23, 0.003243290353566408), (24, 0.005470299161970615), (25, 0.01473467517644167), (26, 0.012250480242073536), (27, 0.23245395720005035)], Temporal MSE: [(0, 2.4635368390590884e-05), (1, 1.4255270798457786e-05), (2, 1.8777267541736364e-05), (3, 1.7062588085536845e-05), (4, 1.3828175724484026e-05), (5, 1.8799593817675486e-05), (6, 8.842632269079331e-06), (7, 9.340035830973648e-06), (8, 8.591182449890766e-06), (9, 1.1729681318684015e-05), (10, 1.9113635062240064e-05), (11, 0.00023417960619553924), (12, 0.00030167901422828436), (13, 0.0012381133856251836), (14, 0.00026584844454191625), (15, 0.00023222179152071476), (16, 0.00022635218920186162), (17, 0.000252140307566151), (18, 0.0002902028791140765), (19, 0.00021395485964603722), (20, 0.0002668929228093475), (21, 0.00038231234066188335), (22, 0.0009048039792105556), (23, 0.0016576717607676983), (24, 0.001949281431734562), (25, 0.0007089183200150728), (26, 0.00044415719457902014), (27, 0.0001720511500025168)]
+
+ 12%|█▏        | 18/150 [04:20<17:55,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 786, Spatial MSE: [(0, 1.727609742374625e-05), (1, 0.0004605216672644019), (2, 0.005267051514238119), (3, 0.000455489062005654), (4, 0.00017705498612485826), (5, 0.00023174076341092587), (6, 0.0002664949861355126), (7, 0.0007239817059598863), (8, 0.0007410073885694146), (9, 0.0010690543567761779), (10, 0.001410027383826673), (11, 0.0017264416674152017), (12, 0.0026657492853701115), (13, 0.018104810267686844), (14, 0.034619905054569244), (15, 0.014835058711469173), (16, 0.288070410490036), (17, 0.004429204389452934), (18, 0.005467842798680067), (19, 0.02229738049209118), (20, 0.01706736348569393), (21, 0.00635612104088068), (22, 0.009346048347651958), (23, 0.005790275055915117), (24, 0.012077520601451397), (25, 0.06344619393348694), (26, 0.03452803194522858), (27, 0.1777222603559494)], Temporal MSE: [(0, 2.463397140672896e-05), (1, 4.4157972297398373e-05), (2, 4.621386688086204e-05), (3, 3.9844777347752824e-05), (4, 2.9773083952022716e-05), (5, 8.907922892831266e-05), (6, 5.857065843883902e-05), (7, 9.84472208074294e-05), (8, 0.00012502606841735542), (9, 0.00019308822811581194), (10, 0.0003688340657390654), (11, 0.0005532217910513282), (12, 0.0007037078612484038), (13, 0.0019531825091689825), (14, 0.0010117804631590843), (15, 0.0010441672056913376), (16, 0.0008045823778957129), (17, 0.0014601316070184112), (18, 0.00125446158926934), (19, 0.0013348625507205725), (20, 0.0023106231819838285), (21, 0.0024255532771348953), (22, 0.0014643059112131596), (23, 0.0022891093976795673), (24, 0.003965024370700121), (25, 0.010109340772032738), (26, 0.00113570352550596), (27, 0.0004902583314105868)]
+
+ 13%|█▎        | 19/150 [04:29<18:12,  8.34s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 780, Spatial MSE: [(0, 2.9167462344048545e-05), (1, 0.00021160383766982704), (2, 0.004004001151770353), (3, 0.0008671825053170323), (4, 0.00015503489703405648), (5, 8.959507977124304e-05), (6, 0.0002164949255529791), (7, 8.260551112471148e-05), (8, 7.375127461273223e-05), (9, 9.768294694367796e-05), (10, 0.0001064625903381966), (11, 0.00025440193712711334), (12, 0.000682636396959424), (13, 0.018412267789244652), (14, 0.029199285432696342), (15, 0.008307581767439842), (16, 0.24915310740470886), (17, 0.0008970845956355333), (18, 0.002559281885623932), (19, 0.015794802457094193), (20, 0.005477444734424353), (21, 0.003902906784787774), (22, 0.008201572112739086), (23, 0.0038930063601583242), (24, 0.006551245227456093), (25, 0.016843363642692566), (26, 0.015955280512571335), (27, 0.31477394700050354)], Temporal MSE: [(0, 3.624918463174254e-05), (1, 1.96556793525815e-05), (2, 2.365418549743481e-05), (3, 2.2298143449006602e-05), (4, 1.802474434953183e-05), (5, 2.6002160666394047e-05), (6, 1.2329330274951644e-05), (7, 1.4363132322614547e-05), (8, 1.2103550034225918e-05), (9, 1.64404100360116e-05), (10, 2.7046371542382985e-05), (11, 0.000301760999718681), (12, 0.0003691593592520803), (13, 0.0014420730294659734), (14, 0.0003288154839538038), (15, 0.00027497950941324234), (16, 0.00027804530691355467), (17, 0.0002716457238420844), (18, 0.0003433285455685109), (19, 0.00024305815168190747), (20, 0.00026539090322330594), (21, 0.0004344542685430497), (22, 0.0010211009066551924), (23, 0.002018962288275361), (24, 0.0024097415152937174), (25, 0.0010422641644254327), (26, 0.0006004446186125278), (27, 0.0002448530576657504)]
+
+ 13%|█▎        | 20/150 [04:37<17:55,  8.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 774, Spatial MSE: [(0, 2.681318437680602e-05), (1, 0.0004956165212206542), (2, 0.006645611021667719), (3, 0.0006030593067407608), (4, 0.00020555552328005433), (5, 0.00022908684331923723), (6, 0.0002782262163236737), (7, 0.00026047229766845703), (8, 0.000231662139412947), (9, 0.0003066755016334355), (10, 0.0003646220138762146), (11, 0.0005454270867630839), (12, 0.0011985368328168988), (13, 0.01935747265815735), (14, 0.032321348786354065), (15, 0.011141498573124409), (16, 0.2708224952220917), (17, 0.001601874828338623), (18, 0.0032702176831662655), (19, 0.018492814153432846), (20, 0.009199343621730804), (21, 0.004653198178857565), (22, 0.009466653689742088), (23, 0.005020978394895792), (24, 0.010869172401726246), (25, 0.030918532982468605), (26, 0.01966511830687523), (27, 0.22021858394145966)], Temporal MSE: [(0, 3.1429652153747156e-05), (1, 4.4902870286023244e-05), (2, 3.7622237869072706e-05), (3, 3.259064760641195e-05), (4, 3.721833854797296e-05), (5, 5.086468081572093e-05), (6, 4.512432133196853e-05), (7, 4.4217889808351174e-05), (8, 4.282407462596893e-05), (9, 7.308022031793371e-05), (10, 0.00011576266842894256), (11, 0.0004373412812128663), (12, 0.0005055508227087557), (13, 0.0015949724474921823), (14, 0.00047141493996605277), (15, 0.000550843367818743), (16, 0.0004912276635877788), (17, 0.0008163138991221786), (18, 0.000684526632539928), (19, 0.000885969668161124), (20, 0.001775345765054226), (21, 0.0013233828358352184), (22, 0.001254736795090139), (23, 0.0024327959399670362), (24, 0.0032815642189234495), (25, 0.003969103563576937), (26, 0.0006978008314035833), (27, 0.0005673468112945557)]
+
+ 14%|█▍        | 21/150 [04:44<17:22,  8.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 768, Spatial MSE: [(0, 3.782400017371401e-05), (1, 0.0003034431138075888), (2, 0.004643826279789209), (3, 0.0011298848548904061), (4, 0.0001868750259745866), (5, 0.0001102557362173684), (6, 0.00030922421137802303), (7, 9.714319457998499e-05), (8, 9.114437852986157e-05), (9, 0.0001218324905494228), (10, 0.00013528461568057537), (11, 0.0003322301199659705), (12, 0.0008689495152793825), (13, 0.02177305333316326), (14, 0.031320661306381226), (15, 0.008767285384237766), (16, 0.2545572519302368), (17, 0.0009762485860846937), (18, 0.0025792752858251333), (19, 0.01636219583451748), (20, 0.006011570803821087), (21, 0.00378370750695467), (22, 0.008150584995746613), (23, 0.00413208594545722), (24, 0.007114578038454056), (25, 0.017734255641698837), (26, 0.019371110945940018), (27, 0.3635430932044983)], Temporal MSE: [(0, 4.7351069952128455e-05), (1, 2.486903940734919e-05), (2, 2.9436439945129678e-05), (3, 2.6409034035168588e-05), (4, 2.13806597457733e-05), (5, 3.3334112231386825e-05), (6, 1.640249502088409e-05), (7, 1.7186644981848076e-05), (8, 1.5221239664242603e-05), (9, 2.1305490008671768e-05), (10, 3.494076008792035e-05), (11, 0.0004829150275327265), (12, 0.00046044072951190174), (13, 0.0014134148368611932), (14, 0.000351071561453864), (15, 0.00031547778053209186), (16, 0.00028289196779951453), (17, 0.0002559513377491385), (18, 0.00038109038723632693), (19, 0.0002707789826672524), (20, 0.00028600115911103785), (21, 0.00043936833390034735), (22, 0.0010127165587618947), (23, 0.002067347289994359), (24, 0.0025359890423715115), (25, 0.0012906738556921482), (26, 0.0006775552174076438), (27, 0.0002944222360383719)]
+
+ 15%|█▍        | 22/150 [04:52<17:06,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 762, Spatial MSE: [(0, 3.285678394604474e-05), (1, 0.0008329437696374953), (2, 0.008712430484592915), (3, 0.0007196929072961211), (4, 0.00024496775586158037), (5, 0.00039319624193012714), (6, 0.0008950578048825264), (7, 0.0009838097030296922), (8, 0.0011300673941150308), (9, 0.001896299421787262), (10, 0.0017540593398734927), (11, 0.0024171953555196524), (12, 0.003525909734889865), (13, 0.025820251554250717), (14, 0.04061689227819443), (15, 0.016205431893467903), (16, 0.3140317499637604), (17, 0.006483737379312515), (18, 0.007071738131344318), (19, 0.02566300891339779), (20, 0.02030543051660061), (21, 0.009838156402111053), (22, 0.010825281962752342), (23, 0.007421468384563923), (24, 0.016111580654978752), (25, 0.07222304493188858), (26, 0.046798594295978546), (27, 0.2591347396373749)], Temporal MSE: [(0, 4.0409220673609525e-05), (1, 7.195563375717029e-05), (2, 6.770123582100496e-05), (3, 6.63928731228225e-05), (4, 4.494903623708524e-05), (5, 0.0001349208177998662), (6, 0.0001840742479544133), (7, 0.00018679675122257322), (8, 0.0001900863426271826), (9, 0.000270524004008621), (10, 0.0005721873021684587), (11, 0.001080926158465445), (12, 0.0011751474812626839), (13, 0.0025852916296571493), (14, 0.0016124999383464456), (15, 0.00225076568312943), (16, 0.0015477261040359735), (17, 0.0025807872880250216), (18, 0.002326353220269084), (19, 0.0029460263904184103), (20, 0.006147866137325764), (21, 0.006793123669922352), (22, 0.0018918636487796903), (23, 0.0030987870413810015), (24, 0.006416213233023882), (25, 0.016105810180306435), (26, 0.001677400548942387), (27, 0.0005984149174764752)]
+
+ 15%|█▌        | 23/150 [05:02<18:22,  8.68s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 756, Spatial MSE: [(0, 4.702939259004779e-05), (1, 0.00038204147131182253), (2, 0.00457721995189786), (3, 0.00127559551037848), (4, 0.00019846780924126506), (5, 0.00011939792602788657), (6, 0.00045939860865473747), (7, 9.729882731335238e-05), (8, 0.00010488408588571474), (9, 0.00013581155508290976), (10, 0.00015868499758653343), (11, 0.0003692328173201531), (12, 0.0008788717677816749), (13, 0.020510278642177582), (14, 0.02857453189790249), (15, 0.008141318336129189), (16, 0.2634742259979248), (17, 0.000997297465801239), (18, 0.002540221903473139), (19, 0.017357517033815384), (20, 0.007102553267031908), (21, 0.0034562856890261173), (22, 0.007832873612642288), (23, 0.003988817799836397), (24, 0.00699781347066164), (25, 0.01634681597352028), (26, 0.020366238430142403), (27, 0.3648020923137665)], Temporal MSE: [(0, 5.484307257574983e-05), (1, 2.63695401372388e-05), (2, 3.008771273016464e-05), (3, 2.6215604520984925e-05), (4, 2.1240763089735992e-05), (5, 3.566812665667385e-05), (6, 1.8089040167978965e-05), (7, 2.070097798423376e-05), (8, 1.6937688997131772e-05), (9, 2.475427754689008e-05), (10, 4.461678327061236e-05), (11, 0.0005713761202059686), (12, 0.000441505981143564), (13, 0.0011793134035542607), (14, 0.0003420279244892299), (15, 0.0003659045323729515), (16, 0.00027817804948426783), (17, 0.0002425611310172826), (18, 0.00045738238259218633), (19, 0.00032136368099600077), (20, 0.0003206076507922262), (21, 0.00045693470747210085), (22, 0.00102543318644166), (23, 0.0020292969420552254), (24, 0.00246043736115098), (25, 0.0017138357507064939), (26, 0.0007091833977028728), (27, 0.00033402832923457026)]
+
+ 16%|█▌        | 24/150 [05:10<17:41,  8.43s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 750, Spatial MSE: [(0, 4.140795499552041e-05), (1, 0.0007028928957879543), (2, 0.007638408336788416), (3, 0.0007893613073974848), (4, 0.0001889341074274853), (5, 0.00024764350382611156), (6, 0.00042922887951135635), (7, 0.0002814466424752027), (8, 0.0004077967314515263), (9, 0.0005546233733184636), (10, 0.0007310665678232908), (11, 0.0009185572271235287), (12, 0.0014675387647002935), (13, 0.021047163754701614), (14, 0.029187379404902458), (15, 0.009372015483677387), (16, 0.2509554326534271), (17, 0.0016070351703092456), (18, 0.004158458672463894), (19, 0.01853751204907894), (20, 0.009217193350195885), (21, 0.006670590955764055), (22, 0.010277778841555119), (23, 0.004360510036349297), (24, 0.01174986083060503), (25, 0.026057209819555283), (26, 0.018338313326239586), (27, 0.24163740873336792)], Temporal MSE: [(0, 4.240264024701901e-05), (1, 4.949849972035736e-05), (2, 4.289361459086649e-05), (3, 3.637988629634492e-05), (4, 3.675245534395799e-05), (5, 5.668582525686361e-05), (6, 5.474007411976345e-05), (7, 4.9093636334873736e-05), (8, 5.5069689551601186e-05), (9, 7.413372804876417e-05), (10, 0.00011927040031878278), (11, 0.0006942027248442173), (12, 0.0005829418660141528), (13, 0.0013289072085171938), (14, 0.000501257018186152), (15, 0.0005402274546213448), (16, 0.0004961066879332066), (17, 0.000689047621563077), (18, 0.0008354929159395397), (19, 0.0010070607531815767), (20, 0.0014671096578240395), (21, 0.0016632485203444958), (22, 0.0012049059150740504), (23, 0.0024194710422307253), (24, 0.003490497125312686), (25, 0.005712260492146015), (26, 0.0007854800205677748), (27, 0.00048175849951803684)]
+
+ 17%|█▋        | 25/150 [05:18<17:17,  8.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 744, Spatial MSE: [(0, 4.06146100431215e-05), (1, 0.0003292374312877655), (2, 0.0031477853190153837), (3, 0.0009052221430465579), (4, 0.00013945090177003294), (5, 8.789115963736549e-05), (6, 0.0003799428523052484), (7, 7.417010783683509e-05), (8, 8.037178486119956e-05), (9, 9.907480125548318e-05), (10, 0.00012248329585418105), (11, 0.00032988705788739026), (12, 0.0008901291294023395), (13, 0.02332424744963646), (14, 0.028530949726700783), (15, 0.00775928283110261), (16, 0.23232020437717438), (17, 0.0008140600984916091), (18, 0.0018478186102584004), (19, 0.016175787895917892), (20, 0.006817843299359083), (21, 0.0027337579522281885), (22, 0.006141202989965677), (23, 0.0032034609466791153), (24, 0.005542582832276821), (25, 0.012528516352176666), (26, 0.015531574375927448), (27, 0.3259625732898712)], Temporal MSE: [(0, 4.733843888971023e-05), (1, 2.093133116432e-05), (2, 2.3588103431393392e-05), (3, 1.9528393750078976e-05), (4, 1.4576387002307456e-05), (5, 2.6009729481302202e-05), (6, 1.4903850569680799e-05), (7, 1.6123036402859725e-05), (8, 1.3219093489169609e-05), (9, 1.9445467842160724e-05), (10, 3.475220364634879e-05), (11, 0.000559011590667069), (12, 0.00046865077456459403), (13, 0.0010885850060731173), (14, 0.00032368837855756283), (15, 0.00031701260013505816), (16, 0.0002383436367381364), (17, 0.00018876726971939206), (18, 0.0004108724824618548), (19, 0.00028496523736976087), (20, 0.0002653584524523467), (21, 0.00039850902976468205), (22, 0.0008262814371846616), (23, 0.0017526858719065785), (24, 0.0021373615600168705), (25, 0.0016685781301930547), (26, 0.000637729128357023), (27, 0.00036721027572639287)]
+
+ 17%|█▋        | 26/150 [05:26<16:46,  8.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 738, Spatial MSE: [(0, 3.479116639937274e-05), (1, 0.000752757943700999), (2, 0.006864956580102444), (3, 0.0005952038336545229), (4, 0.00030545913614332676), (5, 0.0004159030504524708), (6, 0.0009276501950807869), (7, 0.0010388657683506608), (8, 0.001322327647358179), (9, 0.0017987246392294765), (10, 0.001844578655436635), (11, 0.002378357807174325), (12, 0.003293232526630163), (13, 0.02617858722805977), (14, 0.03400539979338646), (15, 0.01343771256506443), (16, 0.2583107650279999), (17, 0.005924733821302652), (18, 0.006023858208209276), (19, 0.021546943113207817), (20, 0.014280865900218487), (21, 0.010848438367247581), (22, 0.009375550784170628), (23, 0.006788735277950764), (24, 0.016255570575594902), (25, 0.04120056331157684), (26, 0.03261532261967659), (27, 0.23959650099277496)], Temporal MSE: [(0, 3.771163028432056e-05), (1, 6.225230754353106e-05), (2, 5.095392043585889e-05), (3, 6.433642556658015e-05), (4, 5.0526294216979295e-05), (5, 0.00010493645822862163), (6, 0.000265999959083274), (7, 0.00021165938233025372), (8, 0.00021128200751263648), (9, 0.00031132061849348247), (10, 0.0006414497038349509), (11, 0.0012785414000973105), (12, 0.0011913852067664266), (13, 0.002268487121909857), (14, 0.0017035027267411351), (15, 0.001980916829779744), (16, 0.0019011948024854064), (17, 0.0029148536268621683), (18, 0.0028353396337479353), (19, 0.003209153888747096), (20, 0.0067259627394378185), (21, 0.005949520971626043), (22, 0.001994945341721177), (23, 0.003174496116116643), (24, 0.007353985216468573), (25, 0.0170730073004961), (26, 0.002203080803155899), (27, 0.00044938307837583125)]
+
+ 18%|█▊        | 27/150 [05:35<17:02,  8.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 732, Spatial MSE: [(0, 3.5518649383448064e-05), (1, 0.0002898474922403693), (2, 0.00219982978887856), (3, 0.0007316712290048599), (4, 0.0001058811612892896), (5, 6.613833829760551e-05), (6, 0.00038082286482676864), (7, 6.0353195294737816e-05), (8, 6.081358515075408e-05), (9, 7.204091525636613e-05), (10, 8.468995656585321e-05), (11, 0.0001771031238604337), (12, 0.00046816025860607624), (13, 0.013260649517178535), (14, 0.01670798845589161), (15, 0.004913073964416981), (16, 0.16152063012123108), (17, 0.0005779685452580452), (18, 0.0010982840321958065), (19, 0.010557032190263271), (20, 0.0045564789324998856), (21, 0.0019424448255449533), (22, 0.003950705751776695), (23, 0.0020468609873205423), (24, 0.0034459286835044622), (25, 0.0077231391333043575), (26, 0.01054668053984642), (27, 0.21879363059997559)], Temporal MSE: [(0, 3.916106652468443e-05), (1, 1.5019753845990635e-05), (2, 1.6866011719685048e-05), (3, 1.322489606536692e-05), (4, 9.71712597674923e-06), (5, 1.9112792870146222e-05), (6, 1.212496863445267e-05), (7, 1.3843590750184376e-05), (8, 9.865804713626858e-06), (9, 1.507492288510548e-05), (10, 2.311669959453866e-05), (11, 0.00024642859352752566), (12, 0.00026090434403158724), (13, 0.000609958020504564), (14, 0.00020969579054508358), (15, 0.0002456385991536081), (16, 0.00016534930909983814), (17, 0.00012567086378112435), (18, 0.0002960778947453946), (19, 0.00020469786250032485), (20, 0.00018364665447734296), (21, 0.00028931855922564864), (22, 0.0005429920274764299), (23, 0.0011750152334570885), (24, 0.0014119544066488743), (25, 0.001673446618951857), (26, 0.0004583396948873997), (27, 0.0002890150935854763)]
+
+ 19%|█▊        | 28/150 [05:43<17:05,  8.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 726, Spatial MSE: [(0, 2.9329423341550864e-05), (1, 0.0003965449286624789), (2, 0.003423540387302637), (3, 0.00030671179411001503), (4, 9.567179222358391e-05), (5, 9.63178463280201e-05), (6, 0.00022355539840646088), (7, 0.0001234511291841045), (8, 0.00016244003199972212), (9, 0.0002304417866980657), (10, 0.0002765669487416744), (11, 0.0003487577196210623), (12, 0.0006495054112747312), (13, 0.013725392520427704), (14, 0.017697179690003395), (15, 0.005665307864546776), (16, 0.17050135135650635), (17, 0.0007987802382558584), (18, 0.002568210940808058), (19, 0.011677003465592861), (20, 0.005845272447913885), (21, 0.0041655381210148335), (22, 0.0057188477367162704), (23, 0.0022151083685457706), (24, 0.006584314163774252), (25, 0.010156214237213135), (26, 0.007827180437743664), (27, 0.15391281247138977)], Temporal MSE: [(0, 2.5738831027410924e-05), (1, 2.1301693777786568e-05), (2, 2.359444260946475e-05), (3, 1.7721524272928946e-05), (4, 1.6735993995098397e-05), (5, 2.3978003810043447e-05), (6, 2.8532869691844098e-05), (7, 2.292680073878728e-05), (8, 2.6361554773757234e-05), (9, 3.24800203088671e-05), (10, 4.91289101773873e-05), (11, 0.00028358749113976955), (12, 0.00030928675550967455), (13, 0.0006752153858542442), (14, 0.00028583771199919283), (15, 0.00032914039911702275), (16, 0.00028079835465177894), (17, 0.0003147608949802816), (18, 0.000508280354551971), (19, 0.0004902820219285786), (20, 0.0005350012215785682), (21, 0.0006288982112891972), (22, 0.000699356896802783), (23, 0.0014522757846862078), (24, 0.0020924010314047337), (25, 0.003117873566225171), (26, 0.0005220376187935472), (27, 0.0002513317740522325)]
+
+ 19%|█▉        | 29/150 [05:51<16:34,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 720, Spatial MSE: [(0, 2.251295154565014e-05), (1, 0.00016833095287438482), (2, 0.001143608009442687), (3, 0.00036470359191298485), (4, 5.547515320358798e-05), (5, 3.5777622542809695e-05), (6, 0.00020853016758337617), (7, 3.6621357139665633e-05), (8, 3.4214361221529543e-05), (9, 3.969623867305927e-05), (10, 4.6602755901403725e-05), (11, 0.00011781414650613442), (12, 0.00038301595486700535), (13, 0.012814398854970932), (14, 0.015412693843245506), (15, 0.004584995564073324), (16, 0.1441832184791565), (17, 0.00047221730346791446), (18, 0.0011758460896089673), (19, 0.00961089413613081), (20, 0.004220796748995781), (21, 0.001554689952172339), (22, 0.003114508232101798), (23, 0.0014950387412682176), (24, 0.002632893854752183), (25, 0.0057406979613006115), (26, 0.0069869766011834145), (27, 0.1707177609205246)], Temporal MSE: [(0, 2.4566290449001826e-05), (1, 8.714001523912884e-06), (2, 9.791251613933127e-06), (3, 7.730477591394447e-06), (4, 5.047261311119655e-06), (5, 1.0365615707996767e-05), (6, 7.2777374953147955e-06), (7, 8.43876478029415e-06), (8, 5.704924205929274e-06), (9, 8.439850716968067e-06), (10, 1.2645604329009075e-05), (11, 0.0001843185309553519), (12, 0.0002270032709930092), (13, 0.0005333225126378238), (14, 0.00018902745796367526), (15, 0.00019798008725047112), (16, 0.0001385300129186362), (17, 9.850542119238526e-05), (18, 0.0002676231088116765), (19, 0.00018239548080600798), (20, 0.000153674409375526), (21, 0.0002551509824115783), (22, 0.0004636978264898062), (23, 0.001014437642879784), (24, 0.001186246401630342), (25, 0.0014037134824320674), (26, 0.00037688607699237764), (27, 0.00025243780692107975)]
+
+ 20%|██        | 30/150 [05:59<16:09,  8.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 714, Spatial MSE: [(0, 1.8141457985620946e-05), (1, 0.0003387777542229742), (2, 0.002732605207711458), (3, 0.00029550143517553806), (4, 0.00016399753803852946), (5, 0.0002445931895636022), (6, 0.00039955697138793766), (7, 0.0006720824749208987), (8, 0.0010103898821398616), (9, 0.001444028690457344), (10, 0.0019329789793118834), (11, 0.0023238470312207937), (12, 0.002947416855022311), (13, 0.015557510778307915), (14, 0.020750362426042557), (15, 0.011752639897167683), (16, 0.16855387389659882), (17, 0.005075463093817234), (18, 0.0057473392225801945), (19, 0.01607600599527359), (20, 0.012731101363897324), (21, 0.01147657260298729), (22, 0.007496464531868696), (23, 0.006422798614948988), (24, 0.015416115522384644), (25, 0.02557089552283287), (26, 0.015108898282051086), (27, 0.13487669825553894)], Temporal MSE: [(0, 1.8226881365990266e-05), (1, 2.6316247385693714e-05), (2, 2.1350035240175202e-05), (3, 3.901978561771102e-05), (4, 3.8663602026645094e-05), (5, 7.210761395981535e-05), (6, 0.00022939730843063444), (7, 0.00015396176604554057), (8, 0.00014500124962069094), (9, 0.00025702870334498584), (10, 0.0005953946965746582), (11, 0.0009883265011012554), (12, 0.0009227312402799726), (13, 0.0017729154787957668), (14, 0.002010093303397298), (15, 0.0020934834610670805), (16, 0.0017799512716010213), (17, 0.001964700175449252), (18, 0.002009090967476368), (19, 0.0024656900204718113), (20, 0.004617787431925535), (21, 0.0035494149196892977), (22, 0.0016937574837356806), (23, 0.002705012448132038), (24, 0.006543450057506561), (25, 0.011384367011487484), (26, 0.002030599396675825), (27, 0.00028039744938723743)]
+
+ 21%|██        | 31/150 [06:07<16:14,  8.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 708, Spatial MSE: [(0, 1.216960390593158e-05), (1, 0.00010237263632006943), (2, 0.0005929517210461199), (3, 0.00021022182772867382), (4, 3.169860428897664e-05), (5, 1.9639024685602635e-05), (6, 0.00013137783389538527), (7, 2.1510504666366614e-05), (8, 1.9162172975484282e-05), (9, 2.223377668997273e-05), (10, 2.8871523682028055e-05), (11, 7.59317263145931e-05), (12, 0.00025931504205800593), (13, 0.00791723933070898), (14, 0.00900192465633154), (15, 0.002899083076044917), (16, 0.09702081233263016), (17, 0.00029098018421791494), (18, 0.0006032734527252614), (19, 0.006627025082707405), (20, 0.003044027602300048), (21, 0.00100562849547714), (22, 0.0018863212317228317), (23, 0.0008854783955030143), (24, 0.0015477534616366029), (25, 0.003446066752076149), (26, 0.004115170333534479), (27, 0.10070501267910004)], Temporal MSE: [(0, 1.3482587746693753e-05), (1, 4.64654658571817e-06), (2, 5.529549980565207e-06), (3, 4.11260589316953e-06), (4, 2.6467821498954436e-06), (5, 5.757579401688417e-06), (6, 4.549678124021739e-06), (7, 5.307331321091624e-06), (8, 3.328636012156494e-06), (9, 5.036847596784355e-06), (10, 7.887345418566838e-06), (11, 0.0001333524560322985), (12, 0.00015508447540923953), (13, 0.0003166894894093275), (14, 0.00012909718498121947), (15, 0.0001287104532821104), (16, 8.953517681220546e-05), (17, 6.629455310758203e-05), (18, 0.00019451633852440864), (19, 0.0001320352457696572), (20, 0.0001114224360208027), (21, 0.00017427591956220567), (22, 0.0002887130540329963), (23, 0.0006387901376001537), (24, 0.0007654817309230566), (25, 0.0009912283858284354), (26, 0.0002512385544832796), (27, 0.00016889888502191752)]
+
+ 21%|██▏       | 32/150 [06:14<15:23,  7.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 702, Spatial MSE: [(0, 1.0098839084093925e-05), (1, 0.00010433286661282182), (2, 0.0007995266350917518), (3, 8.057629020186141e-05), (4, 2.7507185222930275e-05), (5, 2.6438530767336488e-05), (6, 4.896698737866245e-05), (7, 4.641489067580551e-05), (8, 5.3522428061114624e-05), (9, 8.33605881780386e-05), (10, 0.00012998556485399604), (11, 0.00015376631927210838), (12, 0.0003362281422596425), (13, 0.008005627430975437), (14, 0.009148395620286465), (15, 0.0030182497575879097), (16, 0.09742454439401627), (17, 0.00035545590799301863), (18, 0.0010561553062871099), (19, 0.006765354890376329), (20, 0.0031941046472638845), (21, 0.0015623338986188173), (22, 0.002557053230702877), (23, 0.0009660673094913363), (24, 0.002815388375893235), (25, 0.004379332531243563), (26, 0.0033601911272853613), (27, 0.07961013913154602)], Temporal MSE: [(0, 7.868901775509585e-06), (1, 4.812157840206055e-06), (2, 7.1726262831361964e-06), (3, 5.880943717784248e-06), (4, 5.73102306589135e-06), (5, 7.484707111871103e-06), (6, 8.94770255399635e-06), (7, 7.691457540204283e-06), (8, 8.551767678000033e-06), (9, 1.1116048881376628e-05), (10, 1.8516464479034767e-05), (11, 0.00013852820848114789), (12, 0.0001653953513596207), (13, 0.00033612747211009264), (14, 0.00015544948109891266), (15, 0.00015052600065246224), (16, 0.00012835810775868595), (17, 0.00011296980665065348), (18, 0.0002409758308203891), (19, 0.00020523412968032062), (20, 0.000209696008823812), (21, 0.00024563309852965176), (22, 0.00035547208972275257), (23, 0.000744789547752589), (24, 0.0010910890996456146), (25, 0.001290218555368483), (26, 0.0003026390331797302), (27, 0.0001587975857546553)]
+
+ 22%|██▏       | 33/150 [06:22<15:04,  7.73s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 696, Spatial MSE: [(0, 8.33268313726876e-06), (1, 5.37719351996202e-05), (2, 0.0003309502499178052), (3, 0.00011528842151165009), (4, 1.6988922652672045e-05), (5, 1.1228575203858782e-05), (6, 7.282444857992232e-05), (7, 1.5568592061754316e-05), (8, 1.1206306226085871e-05), (9, 1.3000822946196422e-05), (10, 1.6213052731473e-05), (11, 4.7442521463381127e-05), (12, 0.0001694146922091022), (13, 0.0057427301071584225), (14, 0.006050382275134325), (15, 0.0018865475431084633), (16, 0.05744130536913872), (17, 0.00017253328405786306), (18, 0.0003001682925969362), (19, 0.0038478923961520195), (20, 0.001717321458272636), (21, 0.0005948746693320572), (22, 0.0011007931316271424), (23, 0.0005066155572421849), (24, 0.000905569875612855), (25, 0.0019178984221071005), (26, 0.00243319827131927), (27, 0.06568825989961624)], Temporal MSE: [(0, 8.50529431772884e-06), (1, 2.6263194286002545e-06), (2, 3.243170340283541e-06), (3, 2.4213911729020765e-06), (4, 1.4228279496819596e-06), (5, 3.2548045965086203e-06), (6, 2.6542788873484824e-06), (7, 4.040671683469554e-06), (8, 1.8994658148585586e-06), (9, 2.78159677691292e-06), (10, 4.857925887336023e-06), (11, 8.138606062857434e-05), (12, 0.00010200570977758616), (13, 0.0002079104451695457), (14, 7.996355998329818e-05), (15, 7.771011587465182e-05), (16, 5.224890628596768e-05), (17, 4.0682065446162596e-05), (18, 0.00011302406346658245), (19, 7.786770584061742e-05), (20, 6.143486825749278e-05), (21, 0.00010307542106602341), (22, 0.00015946419443935156), (23, 0.0003850579378195107), (24, 0.0004716884286608547), (25, 0.0007051450666040182), (26, 0.00016488555411342531), (27, 0.00011056169751100242)]
+
+ 23%|██▎       | 34/150 [06:30<14:59,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 690, Spatial MSE: [(0, 6.429992481571389e-06), (1, 8.261799666797742e-05), (2, 0.0006568558746948838), (3, 7.904193626018241e-05), (4, 6.051584205124527e-05), (5, 0.0001051595900207758), (6, 0.00023015488113742322), (7, 0.0004135288472753018), (8, 0.0006186570390127599), (9, 0.000918583944439888), (10, 0.0013244850561022758), (11, 0.0014180601574480534), (12, 0.001809311448596418), (13, 0.007911134511232376), (14, 0.010189459659159184), (15, 0.007038934621959925), (16, 0.08386501669883728), (17, 0.0035215276293456554), (18, 0.0038021050859242678), (19, 0.008250057697296143), (20, 0.006612577475607395), (21, 0.0060182735323905945), (22, 0.005164331290870905), (23, 0.005570878740400076), (24, 0.013649730011820793), (25, 0.0228946041315794), (26, 0.013790257275104523), (27, 0.053810395300388336)], Temporal MSE: [(0, 5.56220311409561e-06), (1, 7.129994173737941e-06), (2, 6.516782832477475e-06), (3, 1.3998082977195736e-05), (4, 2.4682625735295005e-05), (5, 4.3895492126466706e-05), (6, 0.00019177533977199346), (7, 0.00010708873014664277), (8, 9.69238972174935e-05), (9, 0.0002024991117650643), (10, 0.0005603859317488968), (11, 0.0008204001933336258), (12, 0.0006131877307780087), (13, 0.0012845484307035804), (14, 0.0015298579819500446), (15, 0.0017781832721084356), (16, 0.001507087261416018), (17, 0.0015670740976929665), (18, 0.001426296541467309), (19, 0.0019500259077176452), (20, 0.004366761073470116), (21, 0.003118177643045783), (22, 0.0013376352144405246), (23, 0.0023169273044914007), (24, 0.007021437864750624), (25, 0.012968885712325573), (26, 0.0025831181555986404), (27, 0.00012886483455076814)]
+
+ 23%|██▎       | 35/150 [06:38<14:59,  7.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 684, Spatial MSE: [(0, 3.967863904108526e-06), (1, 3.481761086732149e-05), (2, 0.00019422640616539866), (3, 6.83323887642473e-05), (4, 1.0630951692292001e-05), (5, 7.208877832454164e-06), (6, 4.3776472011813894e-05), (7, 9.836006938712671e-06), (8, 7.27592396287946e-06), (9, 8.478313247906044e-06), (10, 1.41527825689991e-05), (11, 4.3975258449791e-05), (12, 0.0001768150250427425), (13, 0.006538765039294958), (14, 0.006675972603261471), (15, 0.001916917273774743), (16, 0.06369085609912872), (17, 0.0001814126007957384), (18, 0.000335800985340029), (19, 0.004381767939776182), (20, 0.002045832108706236), (21, 0.0006436200346797705), (22, 0.0011175234103575349), (23, 0.00049016677075997), (24, 0.0009122784249484539), (25, 0.0019218453671783209), (26, 0.0021622441709041595), (27, 0.0614544041454792)], Temporal MSE: [(0, 4.9989039325737394e-06), (1, 1.731876523081155e-06), (2, 2.1471132640726864e-06), (3, 1.6169681202882202e-06), (4, 9.573386705596931e-07), (5, 2.0817346921830904e-06), (6, 1.8811167592502898e-06), (7, 2.4578225747973192e-06), (8, 1.262939122170792e-06), (9, 1.8799814824888017e-06), (10, 4.065787379659014e-06), (11, 6.982548075029626e-05), (12, 0.00011548637121450156), (13, 0.0002223047922598198), (14, 8.553406951250508e-05), (15, 8.590309153078124e-05), (16, 5.937711102887988e-05), (17, 4.793218249687925e-05), (18, 0.00012138583406340331), (19, 8.298310422105715e-05), (20, 6.501912139356136e-05), (21, 0.00010623683920130134), (22, 0.00017621506412979215), (23, 0.00040650914888828993), (24, 0.00048745222738943994), (25, 0.000639149162452668), (26, 0.00017364407540298998), (27, 0.00011643862671917304)]
+
+ 24%|██▍       | 36/150 [06:45<14:44,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 678, Spatial MSE: [(0, 3.927615580323618e-06), (1, 3.987751551903784e-05), (2, 0.00032654995447956026), (3, 4.0785205783322453e-05), (4, 1.4733484931639396e-05), (5, 1.3602203580376226e-05), (6, 2.2026388251106255e-05), (7, 3.415964238229208e-05), (8, 5.247923400020227e-05), (9, 8.308957330882549e-05), (10, 0.00015688205894548446), (11, 0.00017548426694702357), (12, 0.00030900887213647366), (13, 0.006725041661411524), (14, 0.006946264300495386), (15, 0.002074257703498006), (16, 0.06400495767593384), (17, 0.00025244380231015384), (18, 0.0006085466593503952), (19, 0.004583890084177256), (20, 0.0022791805677115917), (21, 0.0008210213272832334), (22, 0.0013953855959698558), (23, 0.000571569602470845), (24, 0.0016467218520119786), (25, 0.002547021023929119), (26, 0.002362695522606373), (27, 0.05443422868847847)], Temporal MSE: [(0, 3.235231361031765e-06), (1, 1.8105018853020738e-06), (2, 3.177712869728566e-06), (3, 3.081649310843204e-06), (4, 3.4738941394607536e-06), (5, 4.048891696584178e-06), (6, 4.734713456855388e-06), (7, 6.041108463250566e-06), (8, 6.193717581481906e-06), (9, 9.233924174623098e-06), (10, 2.0215147742419504e-05), (11, 8.839366637403145e-05), (12, 0.00013384541671257466), (13, 0.000246441806666553), (14, 0.00011942703713430092), (15, 0.00011248337250435725), (16, 8.691541734151542e-05), (17, 7.526895933551714e-05), (18, 0.0001515335898147896), (19, 0.00011830199946416542), (20, 0.0001284113386645913), (21, 0.000147035185364075), (22, 0.00022070367413107306), (23, 0.00046624004608020186), (24, 0.0006980156758800149), (25, 0.0008854498155415058), (26, 0.0002149109059246257), (27, 0.00011540912964846939)]
+
+ 25%|██▍       | 37/150 [06:53<14:51,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 672, Spatial MSE: [(0, 3.7454110497492366e-06), (1, 2.4836859665811062e-05), (2, 0.00016416262951679528), (3, 4.8558238631812856e-05), (4, 8.11938753031427e-06), (5, 5.961961505818181e-06), (6, 3.138813190162182e-05), (7, 7.502386779378867e-06), (8, 5.8861737670667935e-06), (9, 6.949130238353973e-06), (10, 9.849291018326767e-06), (11, 3.3358628570567816e-05), (12, 0.00013750142534263432), (13, 0.0055701881647109985), (14, 0.005681597162038088), (15, 0.0016487654065713286), (16, 0.04902864247560501), (17, 0.0001390858378726989), (18, 0.00030527927447110415), (19, 0.0034056208096444607), (20, 0.0015838737599551678), (21, 0.0005743481451645494), (22, 0.0008965104352682829), (23, 0.0003922844189219177), (24, 0.0007456328603439033), (25, 0.0014917404623702168), (26, 0.0017150071216747165), (27, 0.051607947796583176)], Temporal MSE: [(0, 3.973751518060453e-06), (1, 1.3796476423522108e-06), (2, 1.7339892792733735e-06), (3, 1.3440887869364815e-06), (4, 7.535217605436628e-07), (5, 1.6862697975739138e-06), (6, 1.6212687796723912e-06), (7, 2.1972764443489723e-06), (8, 1.030993189488072e-06), (9, 1.4715724319103174e-06), (10, 2.8939930416527204e-06), (11, 5.853249240317382e-05), (12, 8.69857904035598e-05), (13, 0.00019125849939882755), (14, 7.212511263787746e-05), (15, 6.609891715925187e-05), (16, 4.763526521855965e-05), (17, 3.8172085623955354e-05), (18, 9.881451114779338e-05), (19, 6.794670480303466e-05), (20, 5.3046034736325964e-05), (21, 8.741935016587377e-05), (22, 0.00014619551075156778), (23, 0.000329147296724841), (24, 0.0003965658543165773), (25, 0.0005343337543308735), (26, 0.00014643183385487646), (27, 9.889065404422581e-05)]
+
+ 25%|██▌       | 38/150 [07:02<14:53,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 666, Spatial MSE: [(0, 3.2570237635809463e-06), (1, 4.8970461648423225e-05), (2, 0.0003602745127864182), (3, 5.120649439049885e-05), (4, 4.7510471631539986e-05), (5, 0.00010004306386690587), (6, 0.0003423785383347422), (7, 0.0005116735701449215), (8, 0.0006081307074055076), (9, 0.0009678674396127462), (10, 0.0013406728394329548), (11, 0.0015085971681401134), (12, 0.0019349574577063322), (13, 0.00843263603746891), (14, 0.010881266556680202), (15, 0.007532274816185236), (16, 0.08808023482561111), (17, 0.004462964367121458), (18, 0.0051164207980036736), (19, 0.01021421980112791), (20, 0.009455620311200619), (21, 0.007000353652983904), (22, 0.007311052177101374), (23, 0.008979959413409233), (24, 0.025764940306544304), (25, 0.038548246026039124), (26, 0.020224643871188164), (27, 0.04718565568327904)], Temporal MSE: [(0, 3.337676389492117e-06), (1, 3.818020559265278e-06), (2, 4.1713087739481125e-06), (3, 1.9018079910892993e-05), (4, 2.8041895347996615e-05), (5, 6.199169729370624e-05), (6, 0.0003036513808183372), (7, 0.00012888258788734674), (8, 0.00011162750888615847), (9, 0.00020579929696395993), (10, 0.0005264725186862051), (11, 0.0007676034001633525), (12, 0.0006284213741309941), (13, 0.0012825888115912676), (14, 0.0016044488875195384), (15, 0.0016123539535328746), (16, 0.0016395599814131856), (17, 0.0017512142658233643), (18, 0.0018133417470380664), (19, 0.0028895013965666294), (20, 0.005940668750554323), (21, 0.0049717663787305355), (22, 0.0020546549931168556), (23, 0.003587193787097931), (24, 0.01154333259910345), (25, 0.017432469874620438), (26, 0.0025363133754581213), (27, 0.00013282365398481488)]
+
+ 26%|██▌       | 39/150 [07:10<14:49,  8.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 660, Spatial MSE: [(0, 2.0240674984961515e-06), (1, 1.721282569633331e-05), (2, 9.556319855619222e-05), (3, 3.358507092343643e-05), (4, 5.333138688001782e-06), (5, 3.598063585741329e-06), (6, 2.000561471504625e-05), (7, 4.8091756070789415e-06), (8, 3.831937192444457e-06), (9, 4.3379618546168786e-06), (10, 7.76919932832243e-06), (11, 2.270624099764973e-05), (12, 8.506798621965572e-05), (13, 0.00295522203668952), (14, 0.0034796756226569414), (15, 0.0011898914817720652), (16, 0.04485319182276726), (17, 0.00011578661360545084), (18, 0.00023162983416114002), (19, 0.0029294269625097513), (20, 0.001453323638997972), (21, 0.0003858015697915107), (22, 0.0006779804243706167), (23, 0.0002877616207115352), (24, 0.0005429473822005093), (25, 0.0011562510626390576), (26, 0.0011729529360309243), (27, 0.03416930511593819)], Temporal MSE: [(0, 2.3286795567400986e-06), (1, 8.505488722221344e-07), (2, 1.106004674511496e-06), (3, 8.440113674623717e-07), (4, 5.075112881058885e-07), (5, 1.0940458423647215e-06), (6, 1.07528376247501e-06), (7, 1.3665832057085936e-06), (8, 7.048972747725202e-07), (9, 1.046111833602481e-06), (10, 2.3470654468837893e-06), (11, 3.836042378679849e-05), (12, 5.07154873048421e-05), (13, 0.00010909447883022949), (14, 5.7838748034555465e-05), (15, 5.710032564820722e-05), (16, 3.7035690183984116e-05), (17, 3.391091013327241e-05), (18, 8.264367352239788e-05), (19, 5.701302507077344e-05), (20, 4.3331710912752897e-05), (21, 7.680644921492785e-05), (22, 0.00011884851119248196), (23, 0.00026428079581819475), (24, 0.00030828395392745733), (25, 0.00044156942749395967), (26, 0.0001108601936721243), (27, 7.554545300081372e-05)]
+
+ 27%|██▋       | 40/150 [07:17<14:28,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 654, Spatial MSE: [(0, 2.3873913050920237e-06), (1, 2.33923510677414e-05), (2, 0.0001921968359965831), (3, 3.047755490115378e-05), (4, 1.1348979569447692e-05), (5, 9.597068128641695e-06), (6, 1.809226523619145e-05), (7, 1.9259083273936994e-05), (8, 2.174680048483424e-05), (9, 3.415115497773513e-05), (10, 5.9343918110243976e-05), (11, 6.959769234526902e-05), (12, 0.0001331165840383619), (13, 0.003006484592333436), (14, 0.0035624902229756117), (15, 0.0012524193152785301), (16, 0.04506600275635719), (17, 0.00016139481158461422), (18, 0.00039286972605623305), (19, 0.003011765656992793), (20, 0.0015392393106594682), (21, 0.0004748548089992255), (22, 0.0008618393330834806), (23, 0.00039046903839334846), (24, 0.0011068105231970549), (25, 0.0019097131444141269), (26, 0.0019313582452014089), (27, 0.031964030116796494)], Temporal MSE: [(0, 2.301018867001403e-06), (1, 1.2789997754225624e-06), (2, 2.4209623461501906e-06), (3, 2.3469312964152778e-06), (4, 3.1947731713444227e-06), (5, 3.0902124308340717e-06), (6, 3.6516858017421328e-06), (7, 3.7709410207753535e-06), (8, 3.5125515296385856e-06), (9, 4.6682525862706825e-06), (10, 9.424479685549159e-06), (11, 4.6322616981342435e-05), (12, 5.8390822232468054e-05), (13, 0.00012076604616595432), (14, 7.426300726365298e-05), (15, 7.013532012933865e-05), (16, 5.226445500738919e-05), (17, 5.33839083800558e-05), (18, 0.0001058235066011548), (19, 8.368497219635174e-05), (20, 9.150824189418927e-05), (21, 0.0001166067595477216), (22, 0.00015959377924446017), (23, 0.00031904946081340313), (24, 0.0004899459891021252), (25, 0.000731524545699358), (26, 0.00015907976194284856), (27, 7.727959018666297e-05)]
+
+ 27%|██▋       | 41/150 [07:25<14:18,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 648, Spatial MSE: [(0, 2.8883569029858336e-06), (1, 1.8256121620652266e-05), (2, 9.935963316820562e-05), (3, 3.300372191006318e-05), (4, 6.263233899517218e-06), (5, 3.7029324175819056e-06), (6, 2.6355372028774582e-05), (7, 6.136124284239486e-06), (8, 4.434931724972557e-06), (9, 5.3807430049346294e-06), (10, 7.016403742454713e-06), (11, 2.1238127374090254e-05), (12, 8.270583930425346e-05), (13, 0.0032791488338261843), (14, 0.0033804832492023706), (15, 0.001129104639403522), (16, 0.03337518870830536), (17, 9.373008651891723e-05), (18, 0.0001506063126726076), (19, 0.0023029015865176916), (20, 0.001158959697932005), (21, 0.00028015347197651863), (22, 0.0005210181698203087), (23, 0.0002426427963655442), (24, 0.000465970893856138), (25, 0.0009374406072311103), (26, 0.0011926747392863035), (27, 0.03258812800049782)], Temporal MSE: [(0, 2.867599960154621e-06), (1, 9.173076591650897e-07), (2, 1.094417257263558e-06), (3, 8.729867886358988e-07), (4, 5.305773811414838e-07), (5, 1.1218010058655636e-06), (6, 1.1104631312264246e-06), (7, 1.6651267742417986e-06), (8, 7.605848963976314e-07), (9, 1.052071752383199e-06), (10, 1.9799231267825235e-06), (11, 3.588523031794466e-05), (12, 4.811297549167648e-05), (13, 9.669928112998605e-05), (14, 4.334270488470793e-05), (15, 4.83051517221611e-05), (16, 2.754312299657613e-05), (17, 2.452393528074026e-05), (18, 6.669055437669158e-05), (19, 4.5299206249183044e-05), (20, 3.480793020571582e-05), (21, 5.6886674428824335e-05), (22, 8.976546087069437e-05), (23, 0.0002083023136947304), (24, 0.0002515932428650558), (25, 0.00042418931843712926), (26, 0.00010233298235107213), (27, 6.590744305867702e-05)]
+
+ 28%|██▊       | 42/150 [07:32<13:44,  7.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 642, Spatial MSE: [(0, 2.791768338283873e-06), (1, 3.8233192753978074e-05), (2, 0.00024744035908952355), (3, 4.440003613126464e-05), (4, 4.549174991552718e-05), (5, 8.797841292107478e-05), (6, 0.00031562295043841004), (7, 0.00046049230149947107), (8, 0.0007040193886496127), (9, 0.0012629939010366797), (10, 0.0017111127963289618), (11, 0.0020824202802032232), (12, 0.0027369067538529634), (13, 0.007055375259369612), (14, 0.009825658053159714), (15, 0.01004105806350708), (16, 0.05604850500822067), (17, 0.006950304377824068), (18, 0.006865311414003372), (19, 0.010525903664529324), (20, 0.010946518741548061), (21, 0.008881689049303532), (22, 0.009569752030074596), (23, 0.012035791762173176), (24, 0.028831126168370247), (25, 0.04550602287054062), (26, 0.025561250746250153), (27, 0.03278262913227081)], Temporal MSE: [(0, 2.7630537715594983e-06), (1, 3.0024157240404747e-06), (2, 3.243031869715196e-06), (3, 1.1811740478151478e-05), (4, 2.487113124516327e-05), (5, 3.881199154420756e-05), (6, 0.00017859791114460677), (7, 0.00011924133286811411), (8, 0.0001500190410297364), (9, 0.00027809865423478186), (10, 0.0007244673906825483), (11, 0.000817461172118783), (12, 0.0007677955436520278), (13, 0.0014340392081066966), (14, 0.0022716147359460592), (15, 0.0022627892903983593), (16, 0.0023449426516890526), (17, 0.002465723780915141), (18, 0.002708042273297906), (19, 0.003635426051914692), (20, 0.006622041575610638), (21, 0.005492979194968939), (22, 0.002707884879782796), (23, 0.004603138193488121), (24, 0.013659435324370861), (25, 0.022216610610485077), (26, 0.0034250207245349884), (27, 0.00011693349370034412)]
+
+ 29%|██▊       | 43/150 [07:42<14:37,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 636, Spatial MSE: [(0, 1.7870401052277884e-06), (1, 1.5890091162873432e-05), (2, 7.692909275647253e-05), (3, 2.476830741215963e-05), (4, 4.6427917368419e-06), (5, 2.963102360808989e-06), (6, 1.4907284821674693e-05), (7, 4.655078555515502e-06), (8, 3.6759336126124253e-06), (9, 4.466683549253503e-06), (10, 7.009349701547762e-06), (11, 1.154899291577749e-05), (12, 3.399686102056876e-05), (13, 0.0008517299429513514), (14, 0.0016492370050400496), (15, 0.000545298564247787), (16, 0.02967148832976818), (17, 0.00016113551100715995), (18, 0.00020701350877061486), (19, 0.0023349968250840902), (20, 0.0014731819974258542), (21, 0.0003411201760172844), (22, 0.00043363013537600636), (23, 0.00017730446415953338), (24, 0.0003628340200521052), (25, 0.0008121976279653609), (26, 0.0009787757880985737), (27, 0.01663857139647007)], Temporal MSE: [(0, 2.0535946987365605e-06), (1, 7.891492259659572e-07), (2, 1.0149493618882843e-06), (3, 7.507260306738317e-07), (4, 4.626964198450878e-07), (5, 8.698236229065515e-07), (6, 1.0174403541896027e-06), (7, 1.2373807294352446e-06), (8, 6.967213721509324e-07), (9, 9.179503877021489e-07), (10, 2.3463685465685558e-06), (11, 1.1005118722096086e-05), (12, 2.1137191652087495e-05), (13, 3.0221743145375513e-05), (14, 2.6400481146993116e-05), (15, 6.451802619267255e-05), (16, 3.479659790173173e-05), (17, 3.743202250916511e-05), (18, 5.476196747622453e-05), (19, 4.2171628592768684e-05), (20, 3.670272053568624e-05), (21, 7.415143045363948e-05), (22, 0.00011022329272236675), (23, 0.00017559449770487845), (24, 0.0001544961123727262), (25, 0.00025308021577075124), (26, 6.219815986696631e-05), (27, 5.269888788461685e-05)]
+
+ 29%|██▉       | 44/150 [07:50<14:26,  8.17s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 630, Spatial MSE: [(0, 2.345145730942022e-06), (1, 2.3196909751277417e-05), (2, 0.0001656284584896639), (3, 2.7929700081585906e-05), (4, 1.141861685027834e-05), (5, 9.85252427199157e-06), (6, 2.054279866570141e-05), (7, 2.0221494196448475e-05), (8, 2.3089611204341054e-05), (9, 3.9188489608932287e-05), (10, 6.701594975311309e-05), (11, 7.071199070196599e-05), (12, 9.901810699375346e-05), (13, 0.0009407781763002276), (14, 0.00185515230987221), (15, 0.0008397827041335404), (16, 0.03305886685848236), (17, 0.00022862326295580715), (18, 0.00031143432715907693), (19, 0.002449608873575926), (20, 0.0017958974931389093), (21, 0.00044327054638415575), (22, 0.0005995522369630635), (23, 0.00030053037335164845), (24, 0.0009587235981598496), (25, 0.0017520131077617407), (26, 0.002486914861947298), (27, 0.01571953110396862)], Temporal MSE: [(0, 2.0004831640108023e-06), (1, 1.0013052360591246e-06), (2, 2.017864062509034e-06), (3, 2.09692666430783e-06), (4, 2.971287131003919e-06), (5, 2.840470415321761e-06), (6, 3.5931489037466235e-06), (7, 4.662215815187665e-06), (8, 3.915745764970779e-06), (9, 5.43880878467462e-06), (10, 1.120056003856007e-05), (11, 1.9894423530786298e-05), (12, 2.721947203099262e-05), (13, 5.3132684115553275e-05), (14, 4.978273864253424e-05), (15, 7.316563278436661e-05), (16, 5.5331791372736916e-05), (17, 5.994401726638898e-05), (18, 8.476692892145365e-05), (19, 7.918466144474223e-05), (20, 9.927200881065801e-05), (21, 0.00012206799874547869), (22, 0.0001499769714428112), (23, 0.00022555302712135017), (24, 0.000378383178031072), (25, 0.0006761166150681674), (26, 0.00012132673873566091), (27, 5.06397700519301e-05)]
+
+ 30%|███       | 45/150 [07:57<13:51,  7.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 624, Spatial MSE: [(0, 2.0193463114992483e-06), (1, 1.475666886108229e-05), (2, 8.643474575364962e-05), (3, 2.2452139091910794e-05), (4, 5.144231181475334e-06), (5, 3.1476363346882863e-06), (6, 1.3276276149554178e-05), (7, 4.491262188821565e-06), (8, 3.5354055398784112e-06), (9, 4.205868663120782e-06), (10, 4.54818064099527e-06), (11, 7.231835297716316e-06), (12, 2.1157353330636397e-05), (13, 0.0008909589378163218), (14, 0.0013213363708928227), (15, 0.0005804206593893468), (16, 0.020552169531583786), (17, 7.792764517944306e-05), (18, 0.0001309276995016262), (19, 0.0015645759413018823), (20, 0.0009391884668730199), (21, 0.0003086226060986519), (22, 0.0003075293207075447), (23, 0.0001536402851343155), (24, 0.0002978607953991741), (25, 0.000629064510576427), (26, 0.000869060168042779), (27, 0.015805838629603386)], Temporal MSE: [(0, 1.9517271994118346e-06), (1, 7.32255614366295e-07), (2, 9.993921139539452e-07), (3, 7.343875267906697e-07), (4, 4.4359163098306453e-07), (5, 9.031760441757797e-07), (6, 1.0586203416096396e-06), (7, 1.1610705996645265e-06), (8, 6.551129558829416e-07), (9, 8.666808071211562e-07), (10, 1.3899260693506221e-06), (11, 6.1329374148044735e-06), (12, 1.0842172741831746e-05), (13, 3.185131208738312e-05), (14, 1.9962084479629993e-05), (15, 2.643892730702646e-05), (16, 1.3978707102069166e-05), (17, 1.9048122339881957e-05), (18, 4.2202824261039495e-05), (19, 3.1199211662169546e-05), (20, 2.695771945582237e-05), (21, 4.720207289210521e-05), (22, 7.37712107365951e-05), (23, 0.00012817625247407705), (24, 0.00012619384506251663), (25, 0.00026103027630597353), (26, 5.778555714641698e-05), (27, 3.772511990973726e-05)]
+
+ 31%|███       | 46/150 [08:05<13:45,  7.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 618, Spatial MSE: [(0, 1.9949584384448826e-06), (1, 3.427650153753348e-05), (2, 0.0001738258870318532), (3, 3.619271228672005e-05), (4, 3.930141247110441e-05), (5, 9.127891826210544e-05), (6, 0.0003246913547627628), (7, 0.0003983433707617223), (8, 0.0006278082146309316), (9, 0.0010586755815893412), (10, 0.0013107486302033067), (11, 0.001514894189313054), (12, 0.0018236180767416954), (13, 0.004885186906903982), (14, 0.00911464262753725), (15, 0.007710674311965704), (16, 0.10547656565904617), (17, 0.004522853996604681), (18, 0.005011432338505983), (19, 0.011033007875084877), (20, 0.011332602240145206), (21, 0.0071146683767437935), (22, 0.008020530454814434), (23, 0.01101177278906107), (24, 0.02630798891186714), (25, 0.0428965762257576), (26, 0.024547839537262917), (27, 0.015558622777462006)], Temporal MSE: [(0, 1.978698946913937e-06), (1, 1.789067482604878e-06), (2, 3.2367956919188146e-06), (3, 1.7706795915728435e-05), (4, 2.5288481992902234e-05), (5, 4.5034026697976515e-05), (6, 0.00021309326984919608), (7, 0.0001219306286657229), (8, 0.00012106412032153457), (9, 0.0002131352957803756), (10, 0.00043502001790329814), (11, 0.0004632632771972567), (12, 0.000495328800752759), (13, 0.0008836100460030138), (14, 0.0013803532347083092), (15, 0.0012715624179691076), (16, 0.0015947675565257668), (17, 0.0017808334669098258), (18, 0.0016090977005660534), (19, 0.0026482369285076857), (20, 0.005215051118284464), (21, 0.004860366228967905), (22, 0.0024171024560928345), (23, 0.004141401033848524), (24, 0.013378800824284554), (25, 0.020658478140830994), (26, 0.0021568203810602427), (27, 0.00010433878196636215)]
+
+ 31%|███▏      | 47/150 [08:13<13:43,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 612, Spatial MSE: [(0, 1.1365718819433823e-06), (1, 1.1717484085238539e-05), (2, 5.199139195610769e-05), (3, 1.805501597118564e-05), (4, 3.5643579394672997e-06), (5, 2.0016468624817207e-06), (6, 1.065084961737739e-05), (7, 3.93591153624584e-06), (8, 2.674789129741839e-06), (9, 3.6223984807293164e-06), (10, 6.425762876460794e-06), (11, 1.0314522114640567e-05), (12, 2.5829209334915504e-05), (13, 0.00044024112867191434), (14, 0.001026109792292118), (15, 0.0003035549307242036), (16, 0.023038292303681374), (17, 0.00011271289258729666), (18, 0.00013162042887415737), (19, 0.0019453815184533596), (20, 0.0013305130414664745), (21, 0.00015826348681002855), (22, 0.000279067549854517), (23, 0.00011029316374333575), (24, 0.00025015181745402515), (25, 0.0005985019961372018), (26, 0.0007199702668003738), (27, 0.010878127068281174)], Temporal MSE: [(0, 1.2448389270502958e-06), (1, 5.225127779340255e-07), (2, 6.383628488038084e-07), (3, 5.430653686744336e-07), (4, 3.567776332147332e-07), (5, 6.551233582285931e-07), (6, 7.612418926328246e-07), (7, 9.715496389617329e-07), (8, 5.450004323392932e-07), (9, 7.450207704096101e-07), (10, 2.542624315537978e-06), (11, 9.024081919051241e-06), (12, 1.6936264728428796e-05), (13, 1.311268442805158e-05), (14, 1.6885551303857937e-05), (15, 5.2774350479012355e-05), (16, 2.7723001039703377e-05), (17, 2.993556518049445e-05), (18, 3.894258406944573e-05), (19, 3.123033820884302e-05), (20, 2.5789500796236098e-05), (21, 6.0648646467598155e-05), (22, 8.301096386276186e-05), (23, 0.00012755398347508162), (24, 9.905485057970509e-05), (25, 0.00018154631834477186), (26, 4.3352763896109536e-05), (27, 4.116733907721937e-05)]
+
+ 32%|███▏      | 48/150 [08:21<13:22,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 606, Spatial MSE: [(0, 1.6398623756685993e-06), (1, 1.747896931192372e-05), (2, 0.0001192138806800358), (3, 2.234377825516276e-05), (4, 8.23449681774946e-06), (5, 8.347455150214955e-06), (6, 1.6173877156688832e-05), (7, 1.945818803505972e-05), (8, 3.2073992770165205e-05), (9, 5.4065381846157834e-05), (10, 9.365858568344265e-05), (11, 0.00011123731383122504), (12, 0.00013395551650319248), (13, 0.0005413265898823738), (14, 0.001168575370684266), (15, 0.0004271074431017041), (16, 0.023344043642282486), (17, 0.00018501975864637643), (18, 0.000452533015049994), (19, 0.0023994259536266327), (20, 0.0018569453386589885), (21, 0.000352760631358251), (22, 0.00042663016938604414), (23, 0.0002305313537362963), (24, 0.0007481153006665409), (25, 0.0014295711880549788), (26, 0.0018423536093905568), (27, 0.01032413262873888)], Temporal MSE: [(0, 1.4449746004174813e-06), (1, 7.474485528291552e-07), (2, 1.6537585452169878e-06), (3, 1.871498056971177e-06), (4, 2.8402112093317555e-06), (5, 2.4460603071929654e-06), (6, 3.15402758133132e-06), (7, 4.118590368307196e-06), (8, 4.120738140045432e-06), (9, 6.158744781714631e-06), (10, 1.3521432265406474e-05), (11, 2.268900971102994e-05), (12, 3.11194198729936e-05), (13, 3.450890653766692e-05), (14, 4.547750359051861e-05), (15, 7.734812243143097e-05), (16, 5.266178777674213e-05), (17, 5.272670023259707e-05), (18, 6.558427412528545e-05), (19, 6.55327967251651e-05), (20, 8.734213042771444e-05), (21, 0.00011087633902207017), (22, 0.00012320064706727862), (23, 0.00018263827951159328), (24, 0.0002717831521295011), (25, 0.0005442176479846239), (26, 0.00010692682553781196), (27, 4.317929415265098e-05)]
+
+ 33%|███▎      | 49/150 [08:28<13:02,  7.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 600, Spatial MSE: [(0, 1.700754978628538e-06), (1, 1.2758559023495764e-05), (2, 6.723373371642083e-05), (3, 2.0841443983954377e-05), (4, 4.29072133556474e-06), (5, 2.6355055524618365e-06), (6, 1.282028551941039e-05), (7, 4.612215889210347e-06), (8, 3.1289021080738166e-06), (9, 3.8766174839111045e-06), (10, 4.131532477913424e-06), (11, 5.518608759302879e-06), (12, 1.600455289008096e-05), (13, 0.0006930859526619315), (14, 0.0010671759955585003), (15, 0.0003311137552373111), (16, 0.014954404905438423), (17, 9.376099478686228e-05), (18, 7.218609243864194e-05), (19, 0.0012155714211985469), (20, 0.0008063034620136023), (21, 0.00011759027256630361), (22, 0.00027633862919174135), (23, 0.00011795941827585921), (24, 0.00023015211627352983), (25, 0.0005254752468317747), (26, 0.0009470026707276702), (27, 0.011952526867389679)], Temporal MSE: [(0, 1.693414787951042e-06), (1, 6.652583692812186e-07), (2, 8.353234193236858e-07), (3, 6.633619022977655e-07), (4, 4.159313107265916e-07), (5, 7.606518579450494e-07), (6, 9.400229146194761e-07), (7, 1.3374574336921796e-06), (8, 6.325927301986667e-07), (9, 8.04425610567705e-07), (10, 1.257330836779147e-06), (11, 4.3855093281308655e-06), (12, 1.0928593837888911e-05), (13, 2.2786936824559234e-05), (14, 1.3308891539054457e-05), (15, 3.796171222347766e-05), (16, 2.1236486645648256e-05), (17, 1.901651376101654e-05), (18, 2.8220327294548042e-05), (19, 2.288476207468193e-05), (20, 2.0019489966216497e-05), (21, 4.196460577077232e-05), (22, 6.10753777436912e-05), (23, 0.00010084496898343787), (24, 9.013278759084642e-05), (25, 0.00019538182823453099), (26, 4.846342926612124e-05), (27, 3.0864510335959494e-05)]
+
+ 33%|███▎      | 50/150 [08:35<12:26,  7.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 594, Spatial MSE: [(0, 1.7696614804663113e-06), (1, 2.2469770556199364e-05), (2, 0.00012318963126745075), (3, 2.8748594559147023e-05), (4, 3.429488788242452e-05), (5, 7.147084397729486e-05), (6, 0.00024972594110295177), (7, 0.00030802376568317413), (8, 0.0005020300159230828), (9, 0.0008528433972969651), (10, 0.0009767136070877314), (11, 0.0011495306389406323), (12, 0.0014643797185271978), (13, 0.0037370058707892895), (14, 0.007132816128432751), (15, 0.006323458161205053), (16, 0.06684642285108566), (17, 0.004237007349729538), (18, 0.004659123718738556), (19, 0.008331229910254478), (20, 0.009349734522402287), (21, 0.006963480729609728), (22, 0.008497409522533417), (23, 0.012171871960163116), (24, 0.028993681073188782), (25, 0.05049717798829079), (26, 0.028079500421881676), (27, 0.011061957105994225)], Temporal MSE: [(0, 1.3652650068252115e-06), (1, 1.342514678981388e-06), (2, 2.390213467151625e-06), (3, 1.283266919926973e-05), (4, 2.0594721718225628e-05), (5, 2.9984066713950597e-05), (6, 0.00012798196985386312), (7, 9.211424185195938e-05), (8, 8.944945875555277e-05), (9, 0.00016282143769785762), (10, 0.00034887620131485164), (11, 0.00039329109131358564), (12, 0.0004072300798725337), (13, 0.0007404197240248322), (14, 0.0012192623689770699), (15, 0.0011634582187980413), (16, 0.0014534259680658579), (17, 0.0016432794509455562), (18, 0.0015820212429389358), (19, 0.0026301166508346796), (20, 0.005429314449429512), (21, 0.005179016385227442), (22, 0.0025035657454282045), (23, 0.004552973434329033), (24, 0.014332502149045467), (25, 0.020424112677574158), (26, 0.0025073406286537647), (27, 8.684342174092308e-05)]
+
+ 34%|███▍      | 51/150 [08:43<12:30,  7.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 588, Spatial MSE: [(0, 2.2170954707689816e-06), (1, 1.1395367437216919e-05), (2, 5.5130560213001445e-05), (3, 1.933588646352291e-05), (4, 3.4158226753788767e-06), (5, 2.3273985334526515e-06), (6, 1.1588610504986718e-05), (7, 5.022185632697074e-06), (8, 2.9947134407848353e-06), (9, 3.3226640425709775e-06), (10, 3.514855507091852e-06), (11, 5.437760592030827e-06), (12, 1.7260941604035906e-05), (13, 0.0006748063024133444), (14, 0.000774080865085125), (15, 0.00023222397430799901), (16, 0.00946727767586708), (17, 4.086656917934306e-05), (18, 0.00014087988529354334), (19, 0.0005377890192903578), (20, 0.00033214944414794445), (21, 6.323944398900494e-05), (22, 0.00017692228720989078), (23, 8.327981049660593e-05), (24, 0.00017007385031320155), (25, 0.00039806339191272855), (26, 0.0007562687969766557), (27, 0.009003294631838799)], Temporal MSE: [(0, 1.9632084331533406e-06), (1, 5.988740667817183e-07), (2, 7.390669338747102e-07), (3, 5.916498366786982e-07), (4, 4.111024054509471e-07), (5, 7.000176651672518e-07), (6, 8.976286380857346e-07), (7, 1.2734852816720377e-06), (8, 6.280149023041304e-07), (9, 7.572610343231645e-07), (10, 1.1788666824941174e-06), (11, 4.802019702765392e-06), (12, 1.2145822438469622e-05), (13, 2.8657961593125947e-05), (14, 1.7446005585952662e-05), (15, 1.1240390449529514e-05), (16, 9.703328032628633e-06), (17, 1.037400943459943e-05), (18, 1.6643796698190272e-05), (19, 1.1603561688389163e-05), (20, 1.4245058082451578e-05), (21, 3.511168688419275e-05), (22, 4.086853004992008e-05), (23, 7.500586070818827e-05), (24, 7.352443935815245e-05), (25, 0.0001788857625797391), (26, 4.380599784781225e-05), (27, 2.2764550521969795e-05)]
+
+ 35%|███▍      | 52/150 [08:50<12:18,  7.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 582, Spatial MSE: [(0, 2.0130828488618135e-06), (1, 1.6824995327624492e-05), (2, 0.00011633872054517269), (3, 1.538153810543008e-05), (4, 7.2024495239020325e-06), (5, 7.584824743389618e-06), (6, 1.1523987268446945e-05), (7, 1.5619296391378157e-05), (8, 1.915950997499749e-05), (9, 3.1819847208680585e-05), (10, 4.974722469341941e-05), (11, 7.226541492855176e-05), (12, 7.828032539691776e-05), (13, 0.0008170813089236617), (14, 0.0010753268143162131), (15, 0.0005806360859423876), (16, 0.010775483213365078), (17, 0.00011322805949021131), (18, 0.00035742943873628974), (19, 0.0005050424952059984), (20, 0.00030785033595748246), (21, 0.00014989945339038968), (22, 0.0002986790204886347), (23, 0.0002070362534141168), (24, 0.0006442199810408056), (25, 0.0013233914505690336), (26, 0.0018857596442103386), (27, 0.007583522237837315)], Temporal MSE: [(0, 1.4326590189739363e-06), (1, 6.874768132547615e-07), (2, 1.7545444279676303e-06), (3, 1.6989092728181276e-06), (4, 2.578471139713656e-06), (5, 2.2019441985321464e-06), (6, 2.97588667308446e-06), (7, 3.1786580620973837e-06), (8, 3.1234478683472844e-06), (9, 4.404450919537339e-06), (10, 8.815182809485123e-06), (11, 1.4574802662536968e-05), (12, 2.2950269340071827e-05), (13, 4.95062668051105e-05), (14, 4.021558925160207e-05), (15, 2.9148543035262264e-05), (16, 2.7656573365675285e-05), (17, 2.875026621040888e-05), (18, 3.6442364944377914e-05), (19, 3.5847824619850144e-05), (20, 6.580610352102667e-05), (21, 8.291754784295335e-05), (22, 7.576932694064453e-05), (23, 0.0001262607256649062), (24, 0.0002660181780811399), (25, 0.0005502190324477851), (26, 0.00010440921323606744), (27, 2.229406345577445e-05)]
+
+ 35%|███▌      | 53/150 [08:58<12:13,  7.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 576, Spatial MSE: [(0, 1.283196866097569e-06), (1, 8.981744940683711e-06), (2, 4.288666241336614e-05), (3, 1.3019595826335717e-05), (4, 2.4614475933049107e-06), (5, 1.6165494116648915e-06), (6, 6.280552042881027e-06), (7, 5.001976205676328e-06), (8, 2.1631474282912677e-06), (9, 2.532808821342769e-06), (10, 4.673684543377021e-06), (11, 3.206261317245662e-05), (12, 5.8640209317673e-05), (13, 0.0010331294033676386), (14, 0.0008407570421695709), (15, 0.00023530829639639705), (16, 0.004935404285788536), (17, 2.770209903246723e-05), (18, 3.238253702875227e-05), (19, 0.00037666712887585163), (20, 0.00019007061200682074), (21, 0.00011570783681236207), (22, 0.00020181250874884427), (23, 8.718244498595595e-05), (24, 0.00015495889238081872), (25, 0.0003395675157662481), (26, 0.0006783619755879045), (27, 0.00911907572299242)], Temporal MSE: [(0, 1.1709439604601357e-06), (1, 4.357093530416023e-07), (2, 5.620596539301914e-07), (3, 4.4036985968887166e-07), (4, 3.1524825772066833e-07), (5, 5.09052313191205e-07), (6, 7.22132256214536e-07), (7, 9.232178399543045e-07), (8, 4.622855840352713e-07), (9, 6.09342237112287e-07), (10, 2.296441834914731e-06), (11, 8.82245076354593e-05), (12, 2.5156834453810006e-05), (13, 2.7135158234159462e-05), (14, 1.2285442608117592e-05), (15, 1.6170426533790305e-05), (16, 7.932331755000632e-06), (17, 7.012694823060883e-06), (18, 1.2717749996227212e-05), (19, 1.2088273251720238e-05), (20, 2.1120729797985405e-05), (21, 2.3072807380231097e-05), (22, 3.283087062300183e-05), (23, 6.158377073006704e-05), (24, 7.435256702592596e-05), (25, 0.00013350890367291868), (26, 4.0002814785111696e-05), (27, 1.621762930881232e-05)]
+
+ 36%|███▌      | 54/150 [09:05<11:57,  7.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 570, Spatial MSE: [(0, 1.6181771798073896e-06), (1, 2.2114951207186095e-05), (2, 0.0001057373228832148), (3, 2.7209867766941898e-05), (4, 2.8367214326863177e-05), (5, 5.934425280429423e-05), (6, 0.0001916083856485784), (7, 0.0002596443227957934), (8, 0.0004372317634988576), (9, 0.0007415119325742126), (10, 0.0008289071847684681), (11, 0.0009767839219421148), (12, 0.0012049725046381354), (13, 0.00321235042065382), (14, 0.005437539890408516), (15, 0.00518371444195509), (16, 0.06066450849175453), (17, 0.00352077535353601), (18, 0.004611425567418337), (19, 0.009830315597355366), (20, 0.011364448815584183), (21, 0.005282720550894737), (22, 0.007524876855313778), (23, 0.009723336435854435), (24, 0.02183619886636734), (25, 0.04623178392648697), (26, 0.02375674806535244), (27, 0.009803847409784794)], Temporal MSE: [(0, 1.3481313771990244e-06), (1, 1.1099415360149578e-06), (2, 2.2264466679189354e-06), (3, 9.997810593631584e-06), (4, 1.7285503417951986e-05), (5, 2.3789965780451894e-05), (6, 8.310570410685614e-05), (7, 7.594672933919355e-05), (8, 8.103781146928668e-05), (9, 0.00013865930668544024), (10, 0.00033999126753769815), (11, 0.0004263718437869102), (12, 0.00034014275297522545), (13, 0.0006295301136560738), (14, 0.0010188741143792868), (15, 0.0009593447321094573), (16, 0.0012160701444372535), (17, 0.0013323066523298621), (18, 0.0013510218122974038), (19, 0.002024361863732338), (20, 0.004411135800182819), (21, 0.00329845049418509), (22, 0.0020300764590501785), (23, 0.003693739650771022), (24, 0.010559624992311), (25, 0.021132007241249084), (26, 0.002700727665796876), (27, 6.72278692945838e-05)]
+
+ 37%|███▋      | 55/150 [09:13<12:04,  7.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 564, Spatial MSE: [(0, 1.3957319424662273e-06), (1, 9.188601325149648e-06), (2, 4.0255534258903936e-05), (3, 1.3471615602611564e-05), (4, 2.4743435460550245e-06), (5, 1.8315230363441515e-06), (6, 6.901680990267778e-06), (7, 3.7574186535493936e-06), (8, 2.3510858682129765e-06), (9, 2.7197954750590725e-06), (10, 2.978441216328065e-06), (11, 6.751171895302832e-06), (12, 2.3620765205123462e-05), (13, 0.0006127723027020693), (14, 0.000507732795085758), (15, 0.00019283613073639572), (16, 0.00735823530703783), (17, 3.925359851564281e-05), (18, 9.036091796588153e-05), (19, 0.0005451891338452697), (20, 0.0003559761680662632), (21, 0.00011270420509390533), (22, 0.00014241543249227107), (23, 7.364015618804842e-05), (24, 0.00014050428580958396), (25, 0.0003261936653871089), (26, 0.0006138577009551227), (27, 0.006879774387925863)], Temporal MSE: [(0, 1.3289255775816855e-06), (1, 4.532044215466158e-07), (2, 6.181291496432095e-07), (3, 4.6902357553335605e-07), (4, 3.561003438790067e-07), (5, 5.95304129547003e-07), (6, 7.90789329130348e-07), (7, 1.0302712780685397e-06), (8, 5.414871111497632e-07), (9, 6.146767645986984e-07), (10, 1.0711494269344257e-06), (11, 1.4833290151727851e-05), (12, 1.3941711586085148e-05), (13, 1.6200543541344814e-05), (14, 9.858713383437134e-06), (15, 7.54760321797221e-06), (16, 6.8509307311614975e-06), (17, 7.65026288718218e-06), (18, 1.441752647224348e-05), (19, 1.2298822184675373e-05), (20, 1.299821087741293e-05), (21, 3.425816976232454e-05), (22, 3.618194386945106e-05), (23, 5.5235737818293273e-05), (24, 6.656607729382813e-05), (25, 0.00014460389502346516), (26, 3.471610034466721e-05), (27, 1.868049912445713e-05)]
+
+ 37%|███▋      | 56/150 [09:21<11:56,  7.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 558, Spatial MSE: [(0, 1.5814017615412013e-06), (1, 1.5337227523559704e-05), (2, 9.085841156775132e-05), (3, 1.4989265764597803e-05), (4, 6.270389803830767e-06), (5, 6.9979942054487765e-06), (6, 1.0553837455518078e-05), (7, 1.3665157894138247e-05), (8, 1.7344609659630805e-05), (9, 2.926839079009369e-05), (10, 4.7153251216514036e-05), (11, 6.815592496423051e-05), (12, 7.60504772188142e-05), (13, 0.0005047395243309438), (14, 0.0005734044825658202), (15, 0.00037970757693983614), (16, 0.007722493261098862), (17, 9.868726192507893e-05), (18, 0.00019359659927431494), (19, 0.0004054948512930423), (20, 0.0004374086274765432), (21, 0.00018011424981523305), (22, 0.0002565849863458425), (23, 0.00019768763741012663), (24, 0.0005752610159106553), (25, 0.0012516235001385212), (26, 0.0019327483605593443), (27, 0.006585274823009968)], Temporal MSE: [(0, 1.1413972060836386e-06), (1, 5.909406013415719e-07), (2, 1.4009780215928913e-06), (3, 1.5994930890883552e-06), (4, 2.470181470926036e-06), (5, 2.1242251477815444e-06), (6, 2.812341790559003e-06), (7, 3.0067519674048526e-06), (8, 3.0335206702147843e-06), (9, 4.202457603241783e-06), (10, 8.348568371729925e-06), (11, 2.3084610802470706e-05), (12, 2.162698183383327e-05), (13, 2.56544826697791e-05), (14, 2.6485919079277664e-05), (15, 2.359741301916074e-05), (16, 2.2153531972435303e-05), (17, 2.4125507479766384e-05), (18, 3.1074378057383e-05), (19, 3.64136831194628e-05), (20, 5.8671848819358274e-05), (21, 6.263171962928027e-05), (22, 6.903575558681041e-05), (23, 9.690103615866974e-05), (24, 0.0002220972819486633), (25, 0.00047234678640961647), (26, 9.870419307844713e-05), (27, 1.8582481061457656e-05)]
+
+ 38%|███▊      | 57/150 [09:29<11:55,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 552, Spatial MSE: [(0, 1.2325410807534354e-06), (1, 9.029430657392368e-06), (2, 3.9801121602067724e-05), (3, 1.364415766147431e-05), (4, 2.878806299122516e-06), (5, 1.8132234345102916e-06), (6, 6.278756700339727e-06), (7, 4.853567588725127e-06), (8, 2.519529061828507e-06), (9, 3.162094344588695e-06), (10, 5.850291927345097e-06), (11, 4.6446592023130506e-05), (12, 8.094321674434468e-05), (13, 0.0014032102189958096), (14, 0.0013267241884022951), (15, 0.0002731946296989918), (16, 0.013712901622056961), (17, 6.799973925808445e-05), (18, 0.00011551966599654406), (19, 0.0010342200985178351), (20, 0.000671258894726634), (21, 8.498366514686495e-05), (22, 0.00019463611533865333), (23, 9.005771426018327e-05), (24, 0.00020662885799538344), (25, 0.0004333615070208907), (26, 0.0007615428767167032), (27, 0.012224207632243633)], Temporal MSE: [(0, 1.217328190250555e-06), (1, 4.464511391688575e-07), (2, 5.935703484283295e-07), (3, 5.021576612307399e-07), (4, 3.9151808550741407e-07), (5, 6.108481329647475e-07), (6, 9.43162149269483e-07), (7, 1.1159813766425941e-06), (8, 5.66786411582143e-07), (9, 6.709013860017876e-07), (10, 1.9535168576112483e-06), (11, 9.414797386853024e-05), (12, 3.4302924177609384e-05), (13, 3.1343890441348776e-05), (14, 1.6921356291277334e-05), (15, 2.5549070414854214e-05), (16, 1.5499184883083217e-05), (17, 1.3484068404068239e-05), (18, 2.4863402359187603e-05), (19, 1.8779564925353043e-05), (20, 1.7605325410841033e-05), (21, 3.7920617614872754e-05), (22, 5.560201680054888e-05), (23, 9.421458526048809e-05), (24, 9.219712956110016e-05), (25, 0.00019792121020145714), (26, 4.937910125590861e-05), (27, 3.230305082979612e-05)]
+
+ 39%|███▊      | 58/150 [09:36<11:42,  7.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 546, Spatial MSE: [(0, 1.524908839201089e-06), (1, 2.3986318410607055e-05), (2, 0.00010563986143097281), (3, 2.765533463389147e-05), (4, 3.108290184172802e-05), (5, 5.882074037799612e-05), (6, 0.00023506596335209906), (7, 0.0002607925853226334), (8, 0.0004051289288327098), (9, 0.0006639127968810499), (10, 0.0007402874762192369), (11, 0.0008594177779741585), (12, 0.0010615643113851547), (13, 0.004241643473505974), (14, 0.007091458421200514), (15, 0.005558714270591736), (16, 0.08687806129455566), (17, 0.0031517595052719116), (18, 0.0038733782712370157), (19, 0.00882588978856802), (20, 0.009755498729646206), (21, 0.004878185223788023), (22, 0.006677743047475815), (23, 0.009197738021612167), (24, 0.021763501688838005), (25, 0.03949896991252899), (26, 0.02329861931502819), (27, 0.012478118762373924)], Temporal MSE: [(0, 1.4307883020592271e-06), (1, 1.0542546533542918e-06), (2, 2.4252683488157345e-06), (3, 1.1548672773642465e-05), (4, 1.692481600912288e-05), (5, 2.516482345527038e-05), (6, 8.839359361445531e-05), (7, 7.41303083486855e-05), (8, 6.951445539016277e-05), (9, 0.00011714523861883208), (10, 0.00027027452597394586), (11, 0.00038296199636533856), (12, 0.00032490715966559947), (13, 0.0005567875923588872), (14, 0.0009053527028299868), (15, 0.0008587495540268719), (16, 0.0011179184075444937), (17, 0.001176905701868236), (18, 0.001262615667656064), (19, 0.001930710393935442), (20, 0.004024092573672533), (21, 0.003039088798686862), (22, 0.002093396382406354), (23, 0.003910118248313665), (24, 0.01139671541750431), (25, 0.021658556535840034), (26, 0.002267322735860944), (27, 0.00010410394315840676)]
+
+ 39%|███▉      | 59/150 [09:44<11:46,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 540, Spatial MSE: [(0, 1.177337480839924e-06), (1, 8.041888577281497e-06), (2, 3.4965913073392585e-05), (3, 1.030222210829379e-05), (4, 1.9199935650249245e-06), (5, 1.4421513014895027e-06), (6, 4.022613211418502e-06), (7, 3.557376885510166e-06), (8, 1.975814939214615e-06), (9, 2.083763320115395e-06), (10, 2.312313881702721e-06), (11, 3.827422915492207e-06), (12, 1.0780086086015217e-05), (13, 0.0004227172175887972), (14, 0.0005887761362828314), (15, 0.00021333673794288188), (16, 0.0051008653827011585), (17, 2.145095459127333e-05), (18, 7.626392471138388e-05), (19, 0.0002409483859082684), (20, 9.138896712101996e-05), (21, 5.6832086556823924e-05), (22, 0.00016029774269554764), (23, 7.04131816746667e-05), (24, 0.00012552415137179196), (25, 0.000280232256045565), (26, 0.0005594273097813129), (27, 0.006020576227456331)], Temporal MSE: [(0, 9.695658036434907e-07), (1, 3.4344768096161715e-07), (2, 5.247421199783275e-07), (3, 3.9833022924540273e-07), (4, 3.403555695058458e-07), (5, 4.972106921741215e-07), (6, 7.164078397181584e-07), (7, 8.57634461226553e-07), (8, 4.806723268302449e-07), (9, 5.353923597795074e-07), (10, 9.322806135969586e-07), (11, 2.907545422203839e-06), (12, 5.774894816568121e-06), (13, 1.826753759814892e-05), (14, 1.3002341802348383e-05), (15, 6.743826361343963e-06), (16, 6.443111942644464e-06), (17, 4.5006781874690205e-06), (18, 9.690630577097181e-06), (19, 7.522706255258527e-06), (20, 1.1536181773408316e-05), (21, 1.876059832284227e-05), (22, 2.5613924663048238e-05), (23, 5.6526478147134185e-05), (24, 6.257706991164014e-05), (25, 0.00015074931434355676), (26, 3.570951594156213e-05), (27, 1.2409404916979838e-05)]
+
+ 40%|████      | 60/150 [09:53<12:15,  8.17s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 534, Spatial MSE: [(0, 1.2777818483300507e-06), (1, 9.895540642901324e-06), (2, 5.4729796829633415e-05), (3, 1.1460247151262593e-05), (4, 5.605677870335057e-06), (5, 7.152968464652076e-06), (6, 1.214107578562107e-05), (7, 1.4979232219047844e-05), (8, 2.4403019779128954e-05), (9, 4.209863254800439e-05), (10, 6.922290776856244e-05), (11, 7.988701690919697e-05), (12, 8.230012463172898e-05), (13, 0.0005015737842768431), (14, 0.0006797544774599373), (15, 0.0002953972143586725), (16, 0.0051448773592710495), (17, 7.233318319777027e-05), (18, 0.0003923531621694565), (19, 0.0008612008532509208), (20, 0.0007279592682607472), (21, 0.00023499812232330441), (22, 0.0002715712762437761), (23, 0.00019076115859206766), (24, 0.0005443654954433441), (25, 0.0011342259822413325), (26, 0.0020904559642076492), (27, 0.005450876895338297)], Temporal MSE: [(0, 7.890934057286358e-07), (1, 4.149212315951445e-07), (2, 1.0851005072254338e-06), (3, 1.3834293213221827e-06), (4, 2.319404302397743e-06), (5, 2.1194914552324917e-06), (6, 2.821790985763073e-06), (7, 3.5178270536562195e-06), (8, 3.7740687730547506e-06), (9, 5.263588718662504e-06), (10, 1.0602137990645133e-05), (11, 1.3922714060754515e-05), (12, 1.6623343981336802e-05), (13, 3.609212944866158e-05), (14, 3.4474796848371625e-05), (15, 2.5632538381614722e-05), (16, 2.6333867936045863e-05), (17, 2.0253131879144348e-05), (18, 3.473319884506054e-05), (19, 4.705870014731772e-05), (20, 7.799193554092199e-05), (21, 6.628147093579173e-05), (22, 6.340627442114055e-05), (23, 0.00010720016871346161), (24, 0.00022301996068563312), (25, 0.00047976928181014955), (26, 0.00011038324009859934), (27, 1.5071909729158506e-05)]
+
+ 41%|████      | 61/150 [10:01<12:03,  8.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 528, Spatial MSE: [(0, 1.282981202166411e-06), (1, 1.0041921086667571e-05), (2, 3.257405842305161e-05), (3, 1.1590730537136551e-05), (4, 2.5584640752640553e-06), (5, 1.5587501138725202e-06), (6, 5.820815658807987e-06), (7, 3.7746826819784474e-06), (8, 2.1054206627013627e-06), (9, 2.5870542685879627e-06), (10, 2.4136702450050507e-06), (11, 2.79908203992818e-06), (12, 5.241492544882931e-06), (13, 9.059916919795796e-05), (14, 4.651822746382095e-05), (15, 9.478160791331902e-06), (16, 2.7202504497836344e-05), (17, 9.91348861134611e-06), (18, 1.0949007446470205e-05), (19, 1.126171355281258e-05), (20, 7.970130354806315e-06), (21, 1.1808938324975315e-05), (22, 3.425641989451833e-05), (23, 3.9419934182660654e-05), (24, 7.566151907667518e-05), (25, 0.0001970541779883206), (26, 0.0005681885522790253), (27, 0.003030839143320918)], Temporal MSE: [(0, 1.1013167977580451e-06), (1, 3.9720370637041924e-07), (2, 5.081234348836006e-07), (3, 4.301729745748162e-07), (4, 3.572439766230673e-07), (5, 5.280696768750204e-07), (6, 7.123257432795072e-07), (7, 9.970053724828176e-07), (8, 5.022575919610972e-07), (9, 5.765546120528597e-07), (10, 8.958345460996497e-07), (11, 1.7007914721034467e-06), (12, 2.6400805381854298e-06), (13, 2.4516164103260962e-06), (14, 1.8067951259581605e-06), (15, 1.4373080148288864e-06), (16, 1.8151290532841813e-06), (17, 1.726213099573215e-06), (18, 1.9400517885515e-06), (19, 2.704665575947729e-06), (20, 4.329531293478794e-06), (21, 7.396205546683632e-06), (22, 6.791077794332523e-06), (23, 1.2905151379527524e-05), (24, 2.1539544832194224e-05), (25, 6.747765291947871e-05), (26, 2.1152391127543524e-05), (27, 2.0136035345785785e-06)]
+
+ 41%|████▏     | 62/150 [10:09<11:52,  8.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 522, Spatial MSE: [(0, 1.4078323147259653e-06), (1, 2.0924537238897756e-05), (2, 8.606012852396816e-05), (3, 2.2841632016934454e-05), (4, 2.749911254795734e-05), (5, 5.0803555495804176e-05), (6, 0.0001664242590777576), (7, 0.00022261064441408962), (8, 0.0003635578614193946), (9, 0.0006069595692679286), (10, 0.0006903252797201276), (11, 0.0007672595093026757), (12, 0.0009688737918622792), (13, 0.004405326675623655), (14, 0.009619214572012424), (15, 0.007284798659384251), (16, 0.15419737994670868), (17, 0.002519143046811223), (18, 0.003395618172362447), (19, 0.011738283559679985), (20, 0.01248287595808506), (21, 0.0052880230359733105), (22, 0.006110066082328558), (23, 0.00912313349545002), (24, 0.020519088953733444), (25, 0.03472062200307846), (26, 0.023000236600637436), (27, 0.0042121135629713535)], Temporal MSE: [(0, 1.2003623623968451e-06), (1, 9.050644393937546e-07), (2, 2.2057229216443375e-06), (3, 9.95095160760684e-06), (4, 1.4698245649924502e-05), (5, 2.164600846299436e-05), (6, 7.828146044630557e-05), (7, 6.446457700803876e-05), (8, 6.30471549811773e-05), (9, 0.00010326527990400791), (10, 0.00023152703943196684), (11, 0.00027187145315110683), (12, 0.00030572895775549114), (13, 0.0005412016762420535), (14, 0.0008190672378987074), (15, 0.0007656238740310073), (16, 0.000976758310571313), (17, 0.0010263104923069477), (18, 0.0012350177858024836), (19, 0.0019069148693233728), (20, 0.004037769977003336), (21, 0.003414339851588011), (22, 0.0022550446446985006), (23, 0.003970033954828978), (24, 0.011526728980243206), (25, 0.0199928916990757), (26, 0.002050376497209072), (27, 6.2752646044828e-05)]
+
+ 42%|████▏     | 63/150 [10:18<11:43,  8.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 516, Spatial MSE: [(0, 8.465589189654565e-07), (1, 8.885108400136232e-06), (2, 3.102363189100288e-05), (3, 9.731860700412653e-06), (4, 2.042052074102685e-06), (5, 1.4138174719846575e-06), (6, 3.590090273064561e-06), (7, 3.259764525864739e-06), (8, 1.860051156654663e-06), (9, 1.978966565729934e-06), (10, 2.232408405689057e-06), (11, 2.7208466235606465e-06), (12, 6.284106802922906e-06), (13, 0.00018535152776166797), (14, 0.0003130431578028947), (15, 0.00016525198589079082), (16, 0.005119152832776308), (17, 4.0715927752899006e-05), (18, 2.3929213057272136e-05), (19, 0.00041788123780861497), (20, 0.0002886223664972931), (21, 0.0002525357122067362), (22, 0.00017174251843243837), (23, 7.232526695588604e-05), (24, 0.00013159055379219353), (25, 0.00029058175277896225), (26, 0.0004947645938955247), (27, 0.0037426871713250875)], Temporal MSE: [(0, 8.811273346509552e-07), (1, 3.464952840204205e-07), (2, 5.273205374578538e-07), (3, 3.9827526165936433e-07), (4, 3.60155439693699e-07), (5, 5.20410651461134e-07), (6, 6.814200332883047e-07), (7, 8.160372431120777e-07), (8, 4.953967618348543e-07), (9, 5.262812123874028e-07), (10, 8.512834028806537e-07), (11, 2.1234054656815715e-06), (12, 4.0038344195636455e-06), (13, 7.21301057637902e-06), (14, 1.09555203380296e-05), (15, 6.983686034800485e-06), (16, 5.339000836102059e-06), (17, 9.132211744145025e-06), (18, 9.037637028086465e-06), (19, 9.044550097314641e-06), (20, 9.748368029249832e-06), (21, 3.3737178455339745e-05), (22, 3.89090710086748e-05), (23, 6.25340107944794e-05), (24, 4.6003413444850594e-05), (25, 8.826512203086168e-05), (26, 2.7025118470191956e-05), (27, 1.093709852284519e-05)]
+
+ 43%|████▎     | 64/150 [10:25<11:22,  7.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 510, Spatial MSE: [(0, 1.451601519875112e-06), (1, 1.1114349945273716e-05), (2, 6.021975423209369e-05), (3, 1.1113770597148687e-05), (4, 5.370563485485036e-06), (5, 7.013590220594779e-06), (6, 1.1294537216599565e-05), (7, 1.2981793588551227e-05), (8, 1.6461342966067605e-05), (9, 3.1388688512379304e-05), (10, 4.9765782023314387e-05), (11, 7.617868686793372e-05), (12, 6.926688365638256e-05), (13, 0.0003002574376296252), (14, 0.0005978636909276247), (15, 0.00038274057442322373), (16, 0.006449760869145393), (17, 9.965935896616429e-05), (18, 0.00020363244402687997), (19, 0.0003299795789644122), (20, 0.0002201148890890181), (21, 0.0003359283728059381), (22, 0.0002686896186787635), (23, 0.00019214620988350362), (24, 0.0004964826512150466), (25, 0.0011643368052318692), (26, 0.0019151007290929556), (27, 0.003252859693020582)], Temporal MSE: [(0, 8.966513291852607e-07), (1, 4.846490355703281e-07), (2, 1.072279701475054e-06), (3, 1.4120282685325947e-06), (4, 2.451478849252453e-06), (5, 2.1135590486665023e-06), (6, 2.891741360144806e-06), (7, 3.068401156269829e-06), (8, 3.4649178815016057e-06), (9, 4.506708592089126e-06), (10, 8.772271939960774e-06), (11, 1.1787672519858461e-05), (12, 1.386258918500971e-05), (13, 2.6355655791121535e-05), (14, 3.0487843105220236e-05), (15, 2.45031787926564e-05), (16, 2.256056177429855e-05), (17, 2.306291389686521e-05), (18, 2.780989780148957e-05), (19, 3.087891673203558e-05), (20, 5.280680488795042e-05), (21, 6.58457720419392e-05), (22, 7.109606667654589e-05), (23, 0.00010891593410633504), (24, 0.00019210707978345454), (25, 0.00042595816194079816), (26, 0.00010185919381910935), (27, 1.269668064196594e-05)]
+
+ 43%|████▎     | 65/150 [10:33<11:19,  7.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 504, Spatial MSE: [(0, 1.4630659279646352e-06), (1, 8.349413292307872e-06), (2, 3.135257793474011e-05), (3, 9.810871233639773e-06), (4, 2.153434706997359e-06), (5, 1.5790732277309871e-06), (6, 4.461768185137771e-06), (7, 4.481273208511993e-06), (8, 2.1792741335957544e-06), (9, 2.5209913019352825e-06), (10, 5.411244274000637e-06), (11, 3.140252374578267e-05), (12, 6.243642565095797e-05), (13, 0.0012486806372180581), (14, 0.001129551907069981), (15, 0.0004470528510864824), (16, 0.015023884363472462), (17, 6.898174615344033e-05), (18, 0.00022756887483410537), (19, 0.001178423990495503), (20, 0.0008284989162348211), (21, 0.00014125867164693773), (22, 0.00022587577404920012), (23, 0.00010080871288664639), (24, 0.00021069681679364294), (25, 0.0004868934047408402), (26, 0.0007626384031027555), (27, 0.011869584210216999)], Temporal MSE: [(0, 1.052334368978336e-06), (1, 3.449948451361706e-07), (2, 4.500324735090544e-07), (3, 4.009916665381752e-07), (4, 3.865775966005458e-07), (5, 5.171459065422823e-07), (6, 7.263831776072038e-07), (7, 9.84092594080721e-07), (8, 5.205961315368768e-07), (9, 5.848135629094031e-07), (10, 2.5056647245946806e-06), (11, 7.86696546128951e-05), (12, 2.6431358492118306e-05), (13, 3.058441507164389e-05), (14, 1.9557532141334377e-05), (15, 2.2983893359196372e-05), (16, 1.4987064787419513e-05), (17, 1.9216302462155e-05), (18, 3.0946219339966774e-05), (19, 2.5979144993470982e-05), (20, 2.772214611468371e-05), (21, 6.623467925237492e-05), (22, 6.873274833196774e-05), (23, 0.00011458731023594737), (24, 9.915247210301459e-05), (25, 0.0002172268577851355), (26, 5.669606980518438e-05), (27, 3.7467314541572705e-05)]
+
+ 44%|████▍     | 66/150 [10:41<11:10,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 498, Spatial MSE: [(0, 1.3535377547668759e-06), (1, 1.914126187330112e-05), (2, 6.72548048896715e-05), (3, 2.1766565623693168e-05), (4, 2.6905187041847967e-05), (5, 4.712167356046848e-05), (6, 0.00011345734674250707), (7, 0.00019909915863536298), (8, 0.0003633601882029325), (9, 0.0006103356136009097), (10, 0.0007331114029511809), (11, 0.0008007557480596006), (12, 0.0010057262843474746), (13, 0.0043500312604010105), (14, 0.007634194102138281), (15, 0.005701427813619375), (16, 0.11678285896778107), (17, 0.0021653606090694666), (18, 0.0035015512257814407), (19, 0.013670796528458595), (20, 0.01424500998109579), (21, 0.004262812901288271), (22, 0.004595090169459581), (23, 0.006882537622004747), (24, 0.015063672326505184), (25, 0.025098230689764023), (26, 0.01863565295934677), (27, 0.012015020474791527)], Temporal MSE: [(0, 1.015299517348467e-06), (1, 6.503328791040985e-07), (2, 1.938843070092844e-06), (3, 6.685500466119265e-06), (4, 1.4376814760908019e-05), (5, 1.805878673621919e-05), (6, 6.283880065893754e-05), (7, 6.221317744348198e-05), (8, 6.72975365887396e-05), (9, 0.00010726450273068622), (10, 0.00025107336114160717), (11, 0.00033898442052304745), (12, 0.0003032412496395409), (13, 0.0004966030828654766), (14, 0.0007725641480647027), (15, 0.000712727487552911), (16, 0.0008801529766060412), (17, 0.0009200184140354395), (18, 0.001037592301145196), (19, 0.0016134845791384578), (20, 0.0031172719318419695), (21, 0.002864787820726633), (22, 0.0017910513561218977), (23, 0.002995660062879324), (24, 0.008629104122519493), (25, 0.015881609171628952), (26, 0.0017111561028286815), (27, 9.447485354030505e-05)]
+
+ 45%|████▍     | 67/150 [10:49<11:07,  8.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 492, Spatial MSE: [(0, 7.429747483911342e-07), (1, 8.91948275238974e-06), (2, 2.556638719397597e-05), (3, 8.438541954092216e-06), (4, 2.1397820546553703e-06), (5, 1.1978038401139202e-06), (6, 2.6244122182106366e-06), (7, 2.314518724233494e-06), (8, 1.8770899714581901e-06), (9, 2.7197456802241504e-06), (10, 5.400943791755708e-06), (11, 3.0254428565967828e-05), (12, 5.3514082537731156e-05), (13, 0.0010062423534691334), (14, 0.0013599417870864272), (15, 0.0004928236594423652), (16, 0.03292747586965561), (17, 0.00014883566473145038), (18, 0.00011563308362383395), (19, 0.002917699282988906), (20, 0.002142014680430293), (21, 0.000231926329433918), (22, 0.00028519341140054166), (23, 0.00012699510261882097), (24, 0.0002951930509880185), (25, 0.000692903355229646), (26, 0.0005541489808820188), (27, 0.01572331227362156)], Temporal MSE: [(0, 6.987347660469823e-07), (1, 2.8908692684126436e-07), (2, 4.289867376883194e-07), (3, 3.610653607211134e-07), (4, 3.6165189953862864e-07), (5, 4.812671932086232e-07), (6, 6.299522965491633e-07), (7, 7.406897566397674e-07), (8, 5.037187520429143e-07), (9, 5.803059366371599e-07), (10, 3.336861936986679e-06), (11, 7.834035932319239e-05), (12, 2.434737325529568e-05), (13, 2.8440181267797016e-05), (14, 3.0804279958829284e-05), (15, 5.209206210565753e-05), (16, 3.518208541208878e-05), (17, 4.8412170144729316e-05), (18, 5.4327632824424654e-05), (19, 4.5357810449786484e-05), (20, 3.537842712830752e-05), (21, 0.00010841176845133305), (22, 0.00012320837413426489), (23, 0.00017046315770130605), (24, 0.00013675633817911148), (25, 0.0002845209965016693), (26, 6.0480837419163436e-05), (27, 6.949715316295624e-05)]
+
+ 45%|████▌     | 68/150 [10:57<10:47,  7.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 486, Spatial MSE: [(0, 1.258857309949235e-06), (1, 1.1562025065359194e-05), (2, 5.0561855459818617e-05), (3, 1.1651557542791124e-05), (4, 5.585497092397418e-06), (5, 7.713727427471895e-06), (6, 1.2816341950383503e-05), (7, 1.2777471965819132e-05), (8, 1.764398257364519e-05), (9, 3.229213325539604e-05), (10, 6.0629870858974755e-05), (11, 0.00010243327415082604), (12, 0.00013691656931769103), (13, 0.0016722716391086578), (14, 0.0033989630173891783), (15, 0.0025958791375160217), (16, 0.07539976388216019), (17, 0.0003720245440490544), (18, 0.00023019951186142862), (19, 0.004742234945297241), (20, 0.004282881505787373), (21, 0.00043602040386758745), (22, 0.0005441324901767075), (23, 0.00042680586921051145), (24, 0.0009741121903061867), (25, 0.0021627505775541067), (26, 0.0028824503533542156), (27, 0.014650285243988037)], Temporal MSE: [(0, 7.960549055496813e-07), (1, 4.860378908233542e-07), (2, 9.629651458453736e-07), (3, 1.4101879060035571e-06), (4, 2.4881849185476312e-06), (5, 2.2112992610345827e-06), (6, 3.0260810035542818e-06), (7, 3.5997952636535047e-06), (8, 3.925071268895408e-06), (9, 5.001732006348902e-06), (10, 1.419405271008145e-05), (11, 9.706719720270485e-05), (12, 5.3075305913807824e-05), (13, 7.545405969722196e-05), (14, 7.569069566670805e-05), (15, 0.00011188021744601429), (16, 6.452597153838724e-05), (17, 7.150819146772847e-05), (18, 9.14627526071854e-05), (19, 0.00010266208846587688), (20, 0.00012321157555561513), (21, 0.000203192001208663), (22, 0.00019511884602252394), (23, 0.0002528001496102661), (24, 0.00040187351987697184), (25, 0.0008876093779690564), (26, 0.0001596326328581199), (27, 6.718021904816851e-05)]
+
+ 46%|████▌     | 69/150 [11:05<10:36,  7.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 480, Spatial MSE: [(0, 8.430619686805585e-07), (1, 9.923890502250288e-06), (2, 2.657150980667211e-05), (3, 8.717712262296118e-06), (4, 2.4790738279989455e-06), (5, 1.3689872275790549e-06), (6, 4.152144356339704e-06), (7, 2.8108734113629907e-06), (8, 2.3359398255706765e-06), (9, 4.902021828456782e-06), (10, 2.126760227838531e-05), (11, 0.00019378814613446593), (12, 0.0003141442721243948), (13, 0.005636153742671013), (14, 0.0042350562289357185), (15, 0.0011338338954374194), (16, 0.041663046926259995), (17, 0.00016274886729661375), (18, 0.0004099226789548993), (19, 0.0026880621444433928), (20, 0.0017646802589297295), (21, 0.0002741944044828415), (22, 0.0004206770099699497), (23, 0.00019627729489002377), (24, 0.00048081562272273004), (25, 0.0009162813657894731), (26, 0.0008567121694795787), (27, 0.03311573341488838)], Temporal MSE: [(0, 7.11674033482268e-07), (1, 2.950315547423088e-07), (2, 3.975387983246037e-07), (3, 3.6779428569388983e-07), (4, 3.7206737601991335e-07), (5, 5.40938344784081e-07), (6, 6.281321134338214e-07), (7, 8.007160658962675e-07), (8, 5.696586526937608e-07), (9, 9.184800546790939e-07), (10, 1.3346409104997292e-05), (11, 0.00045080832205712795), (12, 0.00013698860129807144), (13, 0.00012336690269876271), (14, 5.313834481057711e-05), (15, 0.00012068799696862698), (16, 4.978057768312283e-05), (17, 5.536827302421443e-05), (18, 7.264578744070604e-05), (19, 4.3923275370616466e-05), (20, 3.971115802414715e-05), (21, 0.0001061942966771312), (22, 0.00013868854148313403), (23, 0.0002610663359519094), (24, 0.00025364026078023016), (25, 0.0005506958696059883), (26, 0.00012759676610585302), (27, 0.00011479471140773967)]
+
+ 47%|████▋     | 70/150 [11:16<11:41,  8.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 474, Spatial MSE: [(0, 1.1180854926351458e-06), (1, 1.9329516362631693e-05), (2, 6.010313882143237e-05), (3, 1.980458000616636e-05), (4, 2.6730549507192336e-05), (5, 4.664061634684913e-05), (6, 9.047510684467852e-05), (7, 0.00019216773216612637), (8, 0.0003898089053109288), (9, 0.000601409818045795), (10, 0.0006521149771288037), (11, 0.0008627771167084575), (12, 0.0010916440514847636), (13, 0.00744105177000165), (14, 0.007963104173541069), (15, 0.00471535325050354), (16, 0.07939691096544266), (17, 0.002439637202769518), (18, 0.003230876289308071), (19, 0.007263013627380133), (20, 0.007393255829811096), (21, 0.00400906428694725), (22, 0.004599555395543575), (23, 0.006985940970480442), (24, 0.015936926007270813), (25, 0.0273802038282156), (26, 0.021674828603863716), (27, 0.03450086712837219)], Temporal MSE: [(0, 1.0746824727903004e-06), (1, 6.302299198068795e-07), (2, 1.9456929294392467e-06), (3, 6.220643626875244e-06), (4, 1.627976234885864e-05), (5, 1.923923809954431e-05), (6, 6.399204721674323e-05), (7, 6.827346805948764e-05), (8, 7.688946061534807e-05), (9, 0.00011757929314626381), (10, 0.000260129600064829), (11, 0.0006784351426176727), (12, 0.0003828691551461816), (13, 0.0006011815275996923), (14, 0.000801311747636646), (15, 0.0008059738320298493), (16, 0.0009989205282181501), (17, 0.001028016908094287), (18, 0.0010114714968949556), (19, 0.0015348665183410048), (20, 0.0029165197629481554), (21, 0.0027306457050144672), (22, 0.001909249578602612), (23, 0.00334159261547029), (24, 0.009986924938857555), (25, 0.018181512132287025), (26, 0.0020642224699258804), (27, 0.0001996206265175715)]
+
+ 47%|████▋     | 71/150 [11:24<11:19,  8.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 468, Spatial MSE: [(0, 8.939696840570832e-07), (1, 7.190024007286411e-06), (2, 2.4862776626832783e-05), (3, 7.850401743780822e-06), (4, 1.7938033352038474e-06), (5, 1.2907177051602048e-06), (6, 2.2002350306138396e-06), (7, 3.4972847515746253e-06), (8, 1.840166191868775e-06), (9, 2.3634422632312635e-06), (10, 4.626795998774469e-06), (11, 2.11753613257315e-05), (12, 8.140241698129103e-05), (13, 0.002595504280179739), (14, 0.002073968295007944), (15, 0.0006166896782815456), (16, 0.018907345831394196), (17, 8.343257650267333e-05), (18, 0.00033861942938528955), (19, 0.001180316787213087), (20, 0.0006375020602717996), (21, 0.0005699141183868051), (22, 0.0003183952940162271), (23, 0.0001465796958655119), (24, 0.00031525694066658616), (25, 0.0006423937156796455), (26, 0.0006843502633273602), (27, 0.015938838943839073)], Temporal MSE: [(0, 7.231972745103121e-07), (1, 2.9906610166108294e-07), (2, 4.138280758070323e-07), (3, 3.7498347182918224e-07), (4, 4.0299332226823026e-07), (5, 5.337074640010542e-07), (6, 6.391166493813216e-07), (7, 8.84192729699862e-07), (8, 5.364879598346306e-07), (9, 5.924138690716063e-07), (10, 2.646088887559017e-06), (11, 5.04091294715181e-05), (12, 5.510605114977807e-05), (13, 7.170187018346041e-05), (14, 3.6263147194404155e-05), (15, 2.6652918677427806e-05), (16, 2.4434435545117594e-05), (17, 1.9534143575583585e-05), (18, 3.474901313893497e-05), (19, 2.9115228244336322e-05), (20, 3.829804336419329e-05), (21, 7.701648428337649e-05), (22, 9.667679842095822e-05), (23, 0.0001667861215537414), (24, 0.00016771633818279952), (25, 0.0003719342639669776), (26, 8.149595669237897e-05), (27, 5.4929703765083104e-05)]
+
+ 48%|████▊     | 72/150 [11:32<10:50,  8.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 462, Spatial MSE: [(0, 1.0670975143511896e-06), (1, 8.98537109605968e-06), (2, 3.312216722406447e-05), (3, 8.38974301586859e-06), (4, 5.1972592700622045e-06), (5, 7.0682604018657e-06), (6, 1.323193464486394e-05), (7, 1.343898748018546e-05), (8, 2.156519803975243e-05), (9, 4.233970321365632e-05), (10, 7.106594421202317e-05), (11, 0.00011605451436480507), (12, 0.00017910340102389455), (13, 0.00337976380251348), (14, 0.0038619069382548332), (15, 0.0019452039850875735), (16, 0.05308101326227188), (17, 0.00021063489839434624), (18, 0.0004609601164702326), (19, 0.004715382121503353), (20, 0.004099850542843342), (21, 0.0010248342296108603), (22, 0.0005136294639669359), (23, 0.00036753955646418035), (24, 0.0007826621294952929), (25, 0.0017231825040653348), (26, 0.0030497557017952204), (27, 0.015661433339118958)], Temporal MSE: [(0, 5.990655154164415e-07), (1, 3.485738773179037e-07), (2, 7.869239198043942e-07), (3, 1.2135468523410964e-06), (4, 2.390076360825333e-06), (5, 2.171789446947514e-06), (6, 2.9883342449466e-06), (7, 3.786875822697766e-06), (8, 4.576308583637001e-06), (9, 5.698950189980678e-06), (10, 1.420801072526956e-05), (11, 7.807138899806887e-05), (12, 8.662662003189325e-05), (13, 0.00011881429963978007), (14, 7.876673771534115e-05), (15, 8.396107295993716e-05), (16, 5.7080789702013135e-05), (17, 5.277050149743445e-05), (18, 7.948691927595064e-05), (19, 9.892945672618225e-05), (20, 0.00017066327563952655), (21, 0.0001594008062966168), (22, 0.00015925016487017274), (23, 0.00023271884128917009), (24, 0.00035012688022106886), (25, 0.0008178154821507633), (26, 0.00017799077613744885), (27, 5.837696517119184e-05)]
+
+ 49%|████▊     | 73/150 [11:39<10:23,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 456, Spatial MSE: [(0, 6.757271080459759e-07), (1, 6.9791412897757255e-06), (2, 1.939147296070587e-05), (3, 7.9401124821743e-06), (4, 2.076369582937332e-06), (5, 1.30413252463768e-06), (6, 2.4277014745166525e-06), (7, 3.4565846362966113e-06), (8, 1.9951041849708417e-06), (9, 3.2225038921751548e-06), (10, 1.5302764950320125e-05), (11, 0.00011855766933877021), (12, 0.00020315859001129866), (13, 0.004026664886623621), (14, 0.002747192280367017), (15, 0.0007375427521765232), (16, 0.01972929760813713), (17, 6.621559441555291e-05), (18, 5.8991507103201e-05), (19, 0.0015262754168361425), (20, 0.0007889659609645605), (21, 9.231188596459106e-05), (22, 0.00027539266739040613), (23, 0.0001326220080954954), (24, 0.0002864491252694279), (25, 0.0005329346167854965), (26, 0.0006967802182771266), (27, 0.02458290569484234)], Temporal MSE: [(0, 6.740552294104418e-07), (1, 2.9300375103957776e-07), (2, 3.7521314766308933e-07), (3, 3.7653038020835083e-07), (4, 4.176221466423158e-07), (5, 5.384961241361452e-07), (6, 7.346955612774764e-07), (7, 8.982681265479187e-07), (8, 5.574501642513496e-07), (9, 8.702375566826959e-07), (10, 1.111590609070845e-05), (11, 0.00028374791145324707), (12, 9.564609354129061e-05), (13, 8.566071483073756e-05), (14, 3.717215804499574e-05), (15, 6.766149454051629e-05), (16, 2.3099495592759922e-05), (17, 2.2426140276365913e-05), (18, 4.95708518428728e-05), (19, 4.7393448767252266e-05), (20, 8.066220470936969e-05), (21, 4.040547355543822e-05), (22, 5.9306694311089814e-05), (23, 0.0001391526748193428), (24, 0.00017560597916599363), (25, 0.0003821507270913571), (26, 0.0001027141188387759), (27, 6.534321437356994e-05)]
+
+ 49%|████▉     | 74/150 [11:47<10:05,  7.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 450, Spatial MSE: [(0, 1.1662026508929557e-06), (1, 1.628661084396299e-05), (2, 4.7078046918613836e-05), (3, 1.8945065676234663e-05), (4, 2.8205615308252163e-05), (5, 4.626233931048773e-05), (6, 0.00010946714610327035), (7, 0.00021637679310515523), (8, 0.0003909521910827607), (9, 0.0006569389952346683), (10, 0.0006557638989761472), (11, 0.0008043193956837058), (12, 0.0009833150543272495), (13, 0.005872627254575491), (14, 0.006142762023955584), (15, 0.0037371411453932524), (16, 0.04045065864920616), (17, 0.0020983265712857246), (18, 0.0022845263592898846), (19, 0.004631683696061373), (20, 0.004578486550599337), (21, 0.0034161393996328115), (22, 0.003923933487385511), (23, 0.006125574931502342), (24, 0.013682455755770206), (25, 0.022455280646681786), (26, 0.018251260742545128), (27, 0.025353483855724335)], Temporal MSE: [(0, 8.414972967329959e-07), (1, 5.544706596083415e-07), (2, 2.0510733520495705e-06), (3, 5.748272542405175e-06), (4, 1.3129397302691359e-05), (5, 1.8279657524544746e-05), (6, 6.22360166744329e-05), (7, 6.788226164644584e-05), (8, 7.012201967881992e-05), (9, 0.0001071732331183739), (10, 0.0002436807844787836), (11, 0.000515997875481844), (12, 0.00034028346999548376), (13, 0.0005605772021226585), (14, 0.000776671979110688), (15, 0.0007968638674356043), (16, 0.0009788366733118892), (17, 0.0009610308334231377), (18, 0.000893930671736598), (19, 0.0013226921437308192), (20, 0.0024187499657273293), (21, 0.0020527809392660856), (22, 0.0016267509199678898), (23, 0.0027034366503357887), (24, 0.007739735767245293), (25, 0.012439674697816372), (26, 0.001685363007709384), (27, 0.00012785142462234944)]
+
+ 50%|█████     | 75/150 [11:55<10:09,  8.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 444, Spatial MSE: [(0, 7.19142917660065e-07), (1, 7.261855444085086e-06), (2, 2.1647805624525063e-05), (3, 7.1157105594465975e-06), (4, 1.7871874433694757e-06), (5, 1.2180728390376316e-06), (6, 1.7769657461030874e-06), (7, 2.9897128115408123e-06), (8, 1.733780777612992e-06), (9, 2.339996399314259e-06), (10, 4.5479287109628785e-06), (11, 2.107357249769848e-05), (12, 9.222017251886427e-05), (13, 0.0033308251295238733), (14, 0.0027488854248076677), (15, 0.0007493235752917826), (16, 0.018081536516547203), (17, 5.9912217693636194e-05), (18, 4.945869659422897e-05), (19, 0.0013732734369114041), (20, 0.0007043529767543077), (21, 0.00012267765123397112), (22, 0.000301631836919114), (23, 0.00013572824536822736), (24, 0.0002793896710500121), (25, 0.000518516288138926), (26, 0.0006697194185107946), (27, 0.023592697456479073)], Temporal MSE: [(0, 6.272017571973265e-07), (1, 2.8449451860979025e-07), (2, 4.1554309859748173e-07), (3, 3.778166330903332e-07), (4, 4.5548688376584323e-07), (5, 5.730984184992849e-07), (6, 6.858886649752094e-07), (7, 8.551739938411629e-07), (8, 5.598164989351062e-07), (9, 5.946226337982807e-07), (10, 2.8178747015772387e-06), (11, 4.737849303637631e-05), (12, 6.37230696156621e-05), (13, 9.448820492252707e-05), (14, 4.210374390822835e-05), (15, 2.9940722015453503e-05), (16, 2.204065640398767e-05), (17, 1.3984538782096934e-05), (18, 4.367113433545455e-05), (19, 3.059241862501949e-05), (20, 2.497057539585512e-05), (21, 4.7980734962038696e-05), (22, 6.21116632828489e-05), (23, 0.00013690191553905606), (24, 0.0001706928451312706), (25, 0.0004878198087681085), (26, 0.00010173047485295683), (27, 6.385371671058238e-05)]
+
+ 51%|█████     | 76/150 [12:02<09:35,  7.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 438, Spatial MSE: [(0, 9.741519306771806e-07), (1, 9.553469681122806e-06), (2, 2.8433369152480736e-05), (3, 8.177747986337636e-06), (4, 4.077927314938279e-06), (5, 6.60948808217654e-06), (6, 1.2807079656340647e-05), (7, 1.0745416147983633e-05), (8, 1.5648873159079812e-05), (9, 2.7774960472015664e-05), (10, 4.7878023906378075e-05), (11, 8.14129234640859e-05), (12, 0.00014600019494537264), (13, 0.00339868338778615), (14, 0.0028525881934911013), (15, 0.0008054475765675306), (16, 0.018106169998645782), (17, 9.496443817624822e-05), (18, 9.161847992800176e-05), (19, 0.0014004597906023264), (20, 0.0007384712225757539), (21, 0.00015504956536460668), (22, 0.0003704236587509513), (23, 0.00022195189376361668), (24, 0.0005215341807343066), (25, 0.0012463086750358343), (26, 0.0028651850298047066), (27, 0.023226605728268623)], Temporal MSE: [(0, 7.211763772829727e-07), (1, 3.533698134106089e-07), (2, 7.51135530663305e-07), (3, 1.1313303502902272e-06), (4, 2.2740239273844054e-06), (5, 2.0230354493833147e-06), (6, 2.9261020699777873e-06), (7, 3.6310364066594047e-06), (8, 4.092954895895673e-06), (9, 4.721804543805774e-06), (10, 1.0219942851108499e-05), (11, 5.6407629017485306e-05), (12, 7.286958862096071e-05), (13, 0.00011059029202442616), (14, 5.9877693274756894e-05), (15, 4.5448614400811493e-05), (16, 3.7003028410254046e-05), (17, 2.61235090874834e-05), (18, 5.96665995544754e-05), (19, 4.954156611347571e-05), (20, 5.586213956121355e-05), (21, 7.234836084535345e-05), (22, 9.018329728860408e-05), (23, 0.00017236963321920484), (24, 0.00027273219893686473), (25, 0.0007338796276599169), (26, 0.00018128658120986074), (27, 7.068848935887218e-05)]
+
+ 51%|█████▏    | 77/150 [12:10<09:30,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 432, Spatial MSE: [(0, 7.493796942981135e-07), (1, 7.113036190276034e-06), (2, 1.998011430259794e-05), (3, 6.490539362857817e-06), (4, 1.7118024970841361e-06), (5, 1.2045195489918115e-06), (6, 1.919626129165408e-06), (7, 2.4571515950810863e-06), (8, 1.6356465266653686e-06), (9, 2.639287686179159e-06), (10, 2.616730398585787e-06), (11, 7.6242495197220705e-06), (12, 2.9421700673992746e-05), (13, 0.0012708642752841115), (14, 0.001343298819847405), (15, 0.0004591363831423223), (16, 0.011445978656411171), (17, 3.425539398449473e-05), (18, 3.0504283131449483e-05), (19, 0.0007940861396491528), (20, 0.00039607478538528085), (21, 8.388762944377959e-05), (22, 0.00018671626457944512), (23, 8.780127245699987e-05), (24, 0.00018112108227796853), (25, 0.0003570956760086119), (26, 0.0005494541837833822), (27, 0.014249193482100964)], Temporal MSE: [(0, 6.84269593875797e-07), (1, 2.9850849614376784e-07), (2, 3.8347184272424784e-07), (3, 3.6151973858977726e-07), (4, 4.260891159901803e-07), (5, 5.045910143053334e-07), (6, 6.457093491007981e-07), (7, 7.649107942597766e-07), (8, 5.244216367827903e-07), (9, 5.295669893712329e-07), (10, 1.4608688161388272e-06), (11, 1.0859846042876597e-05), (12, 2.0530545953079127e-05), (13, 5.0985236157430336e-05), (14, 2.5772438675630838e-05), (15, 1.881240496004466e-05), (16, 1.3564258551923558e-05), (17, 8.745374543650541e-06), (18, 2.7541786039364524e-05), (19, 2.0426423361641355e-05), (20, 1.6199646779568866e-05), (21, 2.7555533961276524e-05), (22, 3.908557118847966e-05), (23, 8.490390609949827e-05), (24, 0.00010749670764198527), (25, 0.00031521805794909596), (26, 6.812807259848341e-05), (27, 4.099468787899241e-05)]
+
+ 52%|█████▏    | 78/150 [12:17<09:05,  7.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 426, Spatial MSE: [(0, 1.124881691794144e-06), (1, 1.4126298992778175e-05), (2, 3.41418526659254e-05), (3, 1.6419442545156926e-05), (4, 2.7278054403723218e-05), (5, 4.589782110997476e-05), (6, 7.858723256504163e-05), (7, 0.00018275788170285523), (8, 0.0003590134729165584), (9, 0.0005194694385863841), (10, 0.0004968931316398084), (11, 0.0005405042902566493), (12, 0.0006597822066396475), (13, 0.003046409459784627), (14, 0.006330185569822788), (15, 0.004842342343181372), (16, 0.09992396831512451), (17, 0.0016420576721429825), (18, 0.0019120480865240097), (19, 0.010668129660189152), (20, 0.010577637702226639), (21, 0.0026821382343769073), (22, 0.003372844075784087), (23, 0.005095323547720909), (24, 0.011025787331163883), (25, 0.017981186509132385), (26, 0.015374290756881237), (27, 0.014803314581513405)], Temporal MSE: [(0, 7.927192200440913e-07), (1, 4.0665361211722484e-07), (2, 1.7704596757539548e-06), (3, 6.829714038758539e-06), (4, 1.718216844892595e-05), (5, 2.0297251467127353e-05), (6, 6.173791189212352e-05), (7, 7.050199928926304e-05), (8, 6.543168274220079e-05), (9, 9.624212543712929e-05), (10, 0.00020220741862431169), (11, 0.00021334824850782752), (12, 0.0002490384504199028), (13, 0.00041570913163013756), (14, 0.0006338573293760419), (15, 0.0006098863086663187), (16, 0.000803221482783556), (17, 0.0007763179601170123), (18, 0.0007605020655319095), (19, 0.001097885542549193), (20, 0.0018982273759320378), (21, 0.0017892059404402971), (22, 0.0015041736187413335), (23, 0.002228372497484088), (24, 0.006518562324345112), (25, 0.011009762063622475), (26, 0.0014377490151673555), (27, 9.48163797147572e-05)]
+
+ 53%|█████▎    | 79/150 [12:25<09:13,  7.80s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 420, Spatial MSE: [(0, 1.1407340707592084e-06), (1, 8.262249139079358e-06), (2, 2.0072573533980176e-05), (3, 7.9430365076405e-06), (4, 2.0478671558521455e-06), (5, 1.7042925719579216e-06), (6, 2.4205469344451558e-06), (7, 4.640762199414894e-06), (8, 2.3234044874698156e-06), (9, 3.552319185473607e-06), (10, 8.955908924690448e-06), (11, 2.5791243388084695e-05), (12, 3.654477040981874e-05), (13, 0.0010427054949104786), (14, 0.001944367541000247), (15, 0.0004924715030938387), (16, 0.03380398452281952), (17, 0.00016632424376439303), (18, 0.0001089670549845323), (19, 0.002902979962527752), (20, 0.002176989335566759), (21, 0.00010474013106431812), (22, 0.00022889206593390554), (23, 0.00011150533828185871), (24, 0.0002641664759721607), (25, 0.0007216842495836318), (26, 0.000648279266897589), (27, 0.01808547042310238)], Temporal MSE: [(0, 7.960002221807372e-07), (1, 3.2133880267792847e-07), (2, 3.9433965071111743e-07), (3, 4.356017768714082e-07), (4, 5.47483864465903e-07), (5, 6.559044436471595e-07), (6, 7.970227215992054e-07), (7, 1.1295068134131725e-06), (8, 6.970595904931542e-07), (9, 1.1013353287125938e-06), (10, 6.049558123777388e-06), (11, 1.3804343325318769e-05), (12, 2.7962752938037738e-05), (13, 3.670609658001922e-05), (14, 2.900922845583409e-05), (15, 9.210599819198251e-05), (16, 6.171227869344875e-05), (17, 4.172675471636467e-05), (18, 4.550390804070048e-05), (19, 3.745685171452351e-05), (20, 2.7697244149749167e-05), (21, 0.00011495184298837557), (22, 0.00012258280185051262), (23, 0.00017436737834941596), (24, 0.00013650582695845515), (25, 0.0003578549367375672), (26, 7.188898598542437e-05), (27, 8.097156387520954e-05)]
+
+ 53%|█████▎    | 80/150 [12:33<08:51,  7.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 414, Spatial MSE: [(0, 1.1810885780505487e-06), (1, 9.64326045505004e-06), (2, 2.5574001483619213e-05), (3, 7.372462277999148e-06), (4, 3.863548954541329e-06), (5, 6.610186574107502e-06), (6, 1.022446303977631e-05), (7, 1.08917038232903e-05), (8, 1.5673278539907187e-05), (9, 3.368163379491307e-05), (10, 5.57770072191488e-05), (11, 9.148589742835611e-05), (12, 9.466913616051897e-05), (13, 0.0011522873537614942), (14, 0.00231010839343071), (15, 0.0009535008575767279), (16, 0.04022704437375069), (17, 0.00024115493579301983), (18, 0.0001406784722348675), (19, 0.003139645792543888), (20, 0.0024829492904245853), (21, 0.00017540746193844825), (22, 0.0003201655636075884), (23, 0.00024222180945798755), (24, 0.0005712748970836401), (25, 0.0015310320304706693), (26, 0.0029808059334754944), (27, 0.016917824745178223)], Temporal MSE: [(0, 5.808628316117392e-07), (1, 3.363489611274417e-07), (2, 6.972656478865247e-07), (3, 1.1445063137216493e-06), (4, 2.279970203744597e-06), (5, 2.1576081508101197e-06), (6, 2.853118758139317e-06), (7, 3.794155645664432e-06), (8, 4.662431365431985e-06), (9, 5.031305136071751e-06), (10, 1.3627930457005277e-05), (11, 2.359515565331094e-05), (12, 3.677747372421436e-05), (13, 5.976594184176065e-05), (14, 5.5521566537208855e-05), (15, 0.00010718448174884543), (16, 7.667700992897153e-05), (17, 5.4174084652913734e-05), (18, 6.572218262590468e-05), (19, 6.298661173786968e-05), (20, 6.91257300786674e-05), (21, 0.0001508515706518665), (22, 0.000152776061440818), (23, 0.0002126670879079029), (24, 0.00025615611230023205), (25, 0.0006205267272889614), (26, 0.00015592183626722544), (27, 8.407753921346739e-05)]
+
+ 54%|█████▍    | 81/150 [12:40<08:49,  7.68s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 408, Spatial MSE: [(0, 8.261906714324141e-07), (1, 6.436052444769302e-06), (2, 1.5500285371672362e-05), (3, 6.078063506720355e-06), (4, 1.5293029491658672e-06), (5, 1.1070043228755821e-06), (6, 1.6661617792124161e-06), (7, 3.0029466415726347e-06), (8, 1.5655377865186892e-06), (9, 1.5979223917383933e-06), (10, 2.124437287420733e-06), (11, 3.423162070248509e-06), (12, 1.1594423995120451e-05), (13, 0.0005578385316766798), (14, 0.0010543278185650706), (15, 0.0003877662820741534), (16, 0.01936560682952404), (17, 0.00012282462557777762), (18, 5.8130244724452496e-05), (19, 0.0014709510141983628), (20, 0.0010098109487444162), (21, 0.00020700317691080272), (22, 0.00025179353542625904), (23, 9.811361087486148e-05), (24, 0.00020168471382930875), (25, 0.00046978978207334876), (26, 0.00043601173092611134), (27, 0.01050362829118967)], Temporal MSE: [(0, 6.978407895985583e-07), (1, 2.6621390247782983e-07), (2, 3.7369233041317784e-07), (3, 3.568148372323776e-07), (4, 4.3747195377363823e-07), (5, 5.53002337255748e-07), (6, 6.40170128463069e-07), (7, 8.069595196502632e-07), (8, 5.282844881548954e-07), (9, 5.21047695656307e-07), (10, 1.1270863069512416e-06), (11, 3.401682761250413e-06), (12, 8.520978553860914e-06), (13, 2.837499778252095e-05), (14, 2.4061169824562967e-05), (15, 4.1542210965417325e-05), (16, 2.6853505914914422e-05), (17, 2.0654160834965296e-05), (18, 3.0578466976294294e-05), (19, 2.458650487824343e-05), (20, 1.9256307496107183e-05), (21, 7.184341666288674e-05), (22, 9.159548062598333e-05), (23, 0.00012917909771203995), (24, 0.00010790181841002777), (25, 0.00026459377841092646), (26, 5.3946208936395124e-05), (27, 4.676552998716943e-05)]
+
+ 55%|█████▍    | 82/150 [12:48<08:34,  7.56s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 402, Spatial MSE: [(0, 1.1040549452445703e-06), (1, 1.2436911674740259e-05), (2, 2.8297381504671648e-05), (3, 1.486846485931892e-05), (4, 2.3116548618418165e-05), (5, 4.66312667413149e-05), (6, 7.763732719467953e-05), (7, 0.0001810150279197842), (8, 0.00033853884087875485), (9, 0.00045687475358136), (10, 0.0004094885371159762), (11, 0.00042824316187761724), (12, 0.0005108675104565918), (13, 0.002044089836999774), (14, 0.005037515889853239), (15, 0.0037940675392746925), (16, 0.06675303727388382), (17, 0.001299225608818233), (18, 0.0013122556265443563), (19, 0.004880944266915321), (20, 0.004647940397262573), (21, 0.0022619504015892744), (22, 0.0026742368936538696), (23, 0.004143640864640474), (24, 0.00861440971493721), (25, 0.014762036502361298), (26, 0.014131826348602772), (27, 0.011116617359220982)], Temporal MSE: [(0, 8.131929689625395e-07), (1, 4.797954034074792e-07), (2, 1.7793892084227991e-06), (3, 7.37442815079703e-06), (4, 1.6963278540060855e-05), (5, 2.2145701223053038e-05), (6, 6.234265310922638e-05), (7, 7.028200343484059e-05), (8, 6.009595745126717e-05), (9, 8.848001016303897e-05), (10, 0.0001776880380930379), (11, 0.0001710558426566422), (12, 0.00018439930863678455), (13, 0.0003165054367855191), (14, 0.0005087553872726858), (15, 0.00045966546167619526), (16, 0.0005577584961429238), (17, 0.000544683716725558), (18, 0.0005770300049334764), (19, 0.000801116693764925), (20, 0.001385740004479885), (21, 0.001311939093284309), (22, 0.0011429792502894998), (23, 0.0017940225079655647), (24, 0.005035353824496269), (25, 0.00873512588441372), (26, 0.0012087682262063026), (27, 9.98463001451455e-05)]
+
+ 55%|█████▌    | 83/150 [12:55<08:23,  7.52s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 396, Spatial MSE: [(0, 8.040966008593387e-07), (1, 7.1547860898135696e-06), (2, 1.5959096344886348e-05), (3, 7.287419521162519e-06), (4, 1.7120714801421855e-06), (5, 1.438958406652091e-06), (6, 1.7357547221763525e-06), (7, 3.0261689971666783e-06), (8, 1.9229562440159498e-06), (9, 2.5444969651289284e-06), (10, 4.904925390292192e-06), (11, 1.6338701243512332e-05), (12, 5.304199657985009e-05), (13, 0.0016814913833513856), (14, 0.00203413306735456), (15, 0.0007690253551118076), (16, 0.02822037972509861), (17, 0.00010981121158692986), (18, 0.00015375915972981602), (19, 0.0019949807319790125), (20, 0.0012999429600313306), (21, 0.00020154318190179765), (22, 0.0003466639609541744), (23, 0.00014285434735938907), (24, 0.00029191398061811924), (25, 0.0006646363181062043), (26, 0.0005709292599931359), (27, 0.01968359760940075)], Temporal MSE: [(0, 6.32486035101465e-07), (1, 2.7492981757859525e-07), (2, 3.768122098790627e-07), (3, 4.348158881839481e-07), (4, 5.688078204002522e-07), (5, 7.029996709206898e-07), (6, 8.234765118686482e-07), (7, 1.0144775615117396e-06), (8, 6.765508260286879e-07), (9, 7.625922080478631e-07), (10, 2.929927859440795e-06), (11, 3.81356367142871e-05), (12, 3.4128548577427864e-05), (13, 5.8386765886098146e-05), (14, 4.469915074878372e-05), (15, 4.299131251173094e-05), (16, 2.851120734703727e-05), (17, 3.4195258194813505e-05), (18, 4.998815347789787e-05), (19, 3.552448470145464e-05), (20, 3.000891774718184e-05), (21, 8.828787395032123e-05), (22, 0.00010958317579934373), (23, 0.00018898410780820996), (24, 0.00017410468717571348), (25, 0.00046464643673971295), (26, 9.257275814888999e-05), (27, 7.586357969557866e-05)]
+
+ 56%|█████▌    | 84/150 [13:02<08:04,  7.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 390, Spatial MSE: [(0, 1.1787077482949826e-06), (1, 9.819234946917277e-06), (2, 2.0281124307075515e-05), (3, 8.189625077648088e-06), (4, 4.624902430805378e-06), (5, 7.542831554019358e-06), (6, 1.5663057638448663e-05), (7, 1.3710135135625023e-05), (8, 2.0250636225682683e-05), (9, 3.843329250230454e-05), (10, 6.563666829606518e-05), (11, 9.506821515969932e-05), (12, 0.00012422024155966938), (13, 0.0019204963464289904), (14, 0.002842627465724945), (15, 0.001562118879519403), (16, 0.04572100192308426), (17, 0.00020587573817465454), (18, 0.0003567400563042611), (19, 0.003955139312893152), (20, 0.0031975896563380957), (21, 0.00043563262443058193), (22, 0.0005094631924293935), (23, 0.0003446777700446546), (24, 0.0007644377765245736), (25, 0.0017792836297303438), (26, 0.00328320381231606), (27, 0.01945728063583374)], Temporal MSE: [(0, 6.574979920515034e-07), (1, 3.6811866266361903e-07), (2, 7.124180569917371e-07), (3, 1.2178174984001089e-06), (4, 2.7216103717364604e-06), (5, 2.485348431946477e-06), (6, 3.610820613175747e-06), (7, 4.973148406861583e-06), (8, 5.875610440853052e-06), (9, 6.787463007640326e-06), (10, 1.3956971997686196e-05), (11, 5.2852006774628535e-05), (12, 4.941911902278662e-05), (13, 8.591679943492636e-05), (14, 7.810236274963245e-05), (15, 7.694202940911055e-05), (16, 5.2038307330803946e-05), (17, 5.438188600237481e-05), (18, 8.48722702357918e-05), (19, 9.487030911259353e-05), (20, 0.0001571232423884794), (21, 0.0001583421981194988), (22, 0.00017816430772654712), (23, 0.0002515484520699829), (24, 0.0003391076752450317), (25, 0.0008581047877669334), (26, 0.00020180230785626918), (27, 8.277898450614884e-05)]
+
+ 57%|█████▋    | 85/150 [13:10<08:03,  7.45s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 384, Spatial MSE: [(0, 5.843512553838082e-07), (1, 7.255280706885969e-06), (2, 1.5263784007402137e-05), (3, 6.326824859570479e-06), (4, 1.6856997717695776e-06), (5, 1.239740186065319e-06), (6, 1.456482095818501e-06), (7, 2.2655879092781106e-06), (8, 1.6213462004088797e-06), (9, 2.4830874281178694e-06), (10, 8.06027583166724e-06), (11, 2.4715098334127106e-05), (12, 5.265479921945371e-05), (13, 0.0013287921901792288), (14, 0.0019227922894060612), (15, 0.0006766741280443966), (16, 0.03593921661376953), (17, 0.00015420030103996396), (18, 8.408046414842829e-05), (19, 0.0025942709762603045), (20, 0.0017911484465003014), (21, 0.00014359714987222105), (22, 0.00034360826248303056), (23, 0.00015900096332188696), (24, 0.00030720786890015006), (25, 0.000695779686793685), (26, 0.0005646492354571819), (27, 0.021377375349402428)], Temporal MSE: [(0, 5.094963171359268e-07), (1, 2.354260857373447e-07), (2, 3.2845690611793543e-07), (3, 3.716847913892707e-07), (4, 5.110872507430031e-07), (5, 6.051529339856643e-07), (6, 7.035063731564151e-07), (7, 7.884987098805141e-07), (8, 5.819267130391381e-07), (9, 6.921896442690922e-07), (10, 5.039755706093274e-06), (11, 3.264485712861642e-05), (12, 3.328901220811531e-05), (13, 4.6847380872350186e-05), (14, 3.6598139558918774e-05), (15, 6.919021689100191e-05), (16, 3.9030252082739025e-05), (17, 3.69544759450946e-05), (18, 6.020802538841963e-05), (19, 5.1432045438559726e-05), (20, 5.324384619598277e-05), (21, 9.438682900508866e-05), (22, 0.00012757715012412518), (23, 0.0001965309347724542), (24, 0.00017781609494704753), (25, 0.00047098108916543424), (26, 9.568490349920467e-05), (27, 9.701893577584997e-05)]
+
+ 57%|█████▋    | 86/150 [13:17<07:54,  7.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 378, Spatial MSE: [(0, 9.660428759161732e-07), (1, 1.3728694284509402e-05), (2, 3.0047662221477367e-05), (3, 1.4747643945156597e-05), (4, 2.7239413611823693e-05), (5, 4.96401589771267e-05), (6, 7.987865683389828e-05), (7, 0.00019203151168767363), (8, 0.0003317808732390404), (9, 0.00043006771011278033), (10, 0.000394281029002741), (11, 0.0004199449613224715), (12, 0.0005354028544388711), (13, 0.0034241322427988052), (14, 0.00720906862989068), (15, 0.00457323482260108), (16, 0.1288837492465973), (17, 0.0011818327475339174), (18, 0.0013372161192819476), (19, 0.008663877844810486), (20, 0.007135216612368822), (21, 0.0021672879811376333), (22, 0.002643695566803217), (23, 0.004019100684672594), (24, 0.008145511150360107), (25, 0.013905927538871765), (26, 0.013789799995720387), (27, 0.02178354375064373)], Temporal MSE: [(0, 8.4696341673407e-07), (1, 4.1509358084113046e-07), (2, 1.9066159211433842e-06), (3, 8.112860996334348e-06), (4, 1.7162406948045827e-05), (5, 2.241189213236794e-05), (6, 6.17366677033715e-05), (7, 6.970581307541579e-05), (8, 5.833450268255547e-05), (9, 8.477380470139906e-05), (10, 0.00017001264495775104), (11, 0.00019649325986392796), (12, 0.0002205947384936735), (13, 0.00035176429082639515), (14, 0.0005141230649314821), (15, 0.0004679417470470071), (16, 0.0005101338028907776), (17, 0.0005142016452737153), (18, 0.000562682340387255), (19, 0.0007867807289585471), (20, 0.0014625195180997252), (21, 0.0014468536246567965), (22, 0.0011889829766005278), (23, 0.0018548439256846905), (24, 0.00465381657704711), (25, 0.007937727496027946), (26, 0.0011999792186543345), (27, 0.00014872674364596605)]
+
+ 58%|█████▊    | 87/150 [13:25<08:01,  7.64s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 372, Spatial MSE: [(0, 8.438101986030233e-07), (1, 7.584329523524502e-06), (2, 1.3270851923152804e-05), (3, 7.139212812035112e-06), (4, 1.8006003301707096e-06), (5, 1.6591081930528162e-06), (6, 1.838388129726809e-06), (7, 3.0998451165942242e-06), (8, 2.031328676821431e-06), (9, 2.366753960814094e-06), (10, 5.195920948608546e-06), (11, 1.6118599887704477e-05), (12, 5.2794526709476486e-05), (13, 0.0016371972160413861), (14, 0.002235063351690769), (15, 0.0006156045128591359), (16, 0.04277903959155083), (17, 0.0001999655505642295), (18, 8.771473221713677e-05), (19, 0.0036729902494698763), (20, 0.002764884615316987), (21, 0.00025799009017646313), (22, 0.00032585495500825346), (23, 0.0001326914207311347), (24, 0.0003160458290949464), (25, 0.0008391141891479492), (26, 0.000600221159402281), (27, 0.02159356325864792)], Temporal MSE: [(0, 6.824984666309319e-07), (1, 2.9796782996527327e-07), (2, 3.6053657481716073e-07), (3, 4.513836131536664e-07), (4, 6.181437015584379e-07), (5, 7.739951115581789e-07), (6, 8.149397672241321e-07), (7, 1.0723855439209729e-06), (8, 7.05841102899285e-07), (9, 7.683473199904256e-07), (10, 3.0584208161599236e-06), (11, 2.1431165805552155e-05), (12, 3.9340324292425066e-05), (13, 4.325351619627327e-05), (14, 3.758753518923186e-05), (15, 9.078533184947446e-05), (16, 5.950633203610778e-05), (17, 5.3334504627855495e-05), (18, 5.7751884014578536e-05), (19, 4.992360118194483e-05), (20, 3.619035487645306e-05), (21, 0.0001400690380251035), (22, 0.0001596700749360025), (23, 0.000239754153881222), (24, 0.000167495381901972), (25, 0.00039812090108171105), (26, 8.17462641862221e-05), (27, 0.00010160505917156115)]
+
+ 59%|█████▊    | 88/150 [13:33<07:57,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 366, Spatial MSE: [(0, 9.808023833102197e-07), (1, 7.80811024014838e-06), (2, 1.5617879398632795e-05), (3, 6.85484610585263e-06), (4, 3.556168621798861e-06), (5, 6.651519470324274e-06), (6, 1.106147828977555e-05), (7, 1.0483789083082229e-05), (8, 1.5959070879034698e-05), (9, 2.7709302230505273e-05), (10, 4.998392614652403e-05), (11, 7.168800220824778e-05), (12, 0.00010586958524072543), (13, 0.0016744773602113128), (14, 0.0025114864110946655), (15, 0.0008097456302493811), (16, 0.048512447625398636), (17, 0.00024257799668703228), (18, 0.00015183567302301526), (19, 0.0038441107608377934), (20, 0.0028823167085647583), (21, 0.0003268943983130157), (22, 0.00039830003515817225), (23, 0.00026245659682899714), (24, 0.0006299797096289694), (25, 0.0016501754289492965), (26, 0.002880464307963848), (27, 0.020843863487243652)], Temporal MSE: [(0, 4.778158313456515e-07), (1, 2.799059188873798e-07), (2, 5.995267997604969e-07), (3, 1.0950742534987512e-06), (4, 2.5693391307868296e-06), (5, 2.4553546609240584e-06), (6, 3.2863654269021936e-06), (7, 4.247479409968946e-06), (8, 4.854410690313671e-06), (9, 4.882763732894091e-06), (10, 1.0265432138112374e-05), (11, 2.9387549147941172e-05), (12, 4.5861233957111835e-05), (13, 6.0001504607498646e-05), (14, 5.729937765863724e-05), (15, 0.00010668380127754062), (16, 7.450845441780984e-05), (17, 6.68690336169675e-05), (18, 7.270096830325201e-05), (19, 6.760797259630635e-05), (20, 6.797243986511603e-05), (21, 0.00016937121108639985), (22, 0.00019843493646476418), (23, 0.00028085274971090257), (24, 0.00027400406543165445), (25, 0.0006667173001915216), (26, 0.0001893173175631091), (27, 0.00010754316463135183)]
+
+ 59%|█████▉    | 89/150 [13:41<07:54,  7.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 360, Spatial MSE: [(0, 5.394127811086946e-07), (1, 6.6567049543664325e-06), (2, 1.2897997294203378e-05), (3, 6.048920113244094e-06), (4, 1.5042278391774744e-06), (5, 1.2190809002277092e-06), (6, 1.337391154265788e-06), (7, 1.9740493826247985e-06), (8, 1.529653786747076e-06), (9, 2.157684320991393e-06), (10, 1.8575491367300856e-06), (11, 6.40842154098209e-06), (12, 3.2266492780763656e-05), (13, 0.001364756841212511), (14, 0.0010843143099918962), (15, 0.0003953494888264686), (16, 0.014816038310527802), (17, 6.812168430769816e-05), (18, 5.0361377361696213e-05), (19, 0.0012262217933312058), (20, 0.0008465586579404771), (21, 0.00017325778026133776), (22, 0.00018986896611750126), (23, 9.214928286382928e-05), (24, 0.0001946936099557206), (25, 0.00041246102773584425), (26, 0.00038175901863723993), (27, 0.010328864678740501)], Temporal MSE: [(0, 4.7479588261012395e-07), (1, 2.446069800043915e-07), (2, 3.3757919482013676e-07), (3, 3.854707131267787e-07), (4, 5.355955181585159e-07), (5, 6.381362140928104e-07), (6, 7.130602170946077e-07), (7, 8.250418659372372e-07), (8, 5.814210339849524e-07), (9, 5.490562102750118e-07), (10, 1.011178483167896e-06), (11, 1.3915638191974722e-05), (12, 2.683488855836913e-05), (13, 2.979946839332115e-05), (14, 2.119448436133098e-05), (15, 1.833265196182765e-05), (16, 1.4921803995093796e-05), (17, 2.2691217964165844e-05), (18, 2.7164727725903504e-05), (19, 2.1476113033713773e-05), (20, 1.8164751963922754e-05), (21, 5.7023396948352456e-05), (22, 7.09445375832729e-05), (23, 0.00010783787729451433), (24, 0.00010525536345085129), (25, 0.00024429953191429377), (26, 5.634009721688926e-05), (27, 3.835663665086031e-05)]
+
+ 60%|██████    | 90/150 [13:48<07:34,  7.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 354, Spatial MSE: [(0, 9.474513262830442e-07), (1, 1.2023154340567999e-05), (2, 2.448499799356796e-05), (3, 1.4523114259645808e-05), (4, 2.7463564038043842e-05), (5, 5.3678741096518934e-05), (6, 8.376391633646563e-05), (7, 0.00019309773051645607), (8, 0.00031776263494975865), (9, 0.0003882338642142713), (10, 0.0003704324481077492), (11, 0.00038228565244935453), (12, 0.00048122330917976797), (13, 0.0027453885413706303), (14, 0.004162515513598919), (15, 0.002983697922900319), (16, 0.05808154493570328), (17, 0.000960032339207828), (18, 0.0014366188552230597), (19, 0.006214306224137545), (20, 0.0060207233764231205), (21, 0.0019530585268512368), (22, 0.0021683238446712494), (23, 0.003587690880522132), (24, 0.007068244274705648), (25, 0.012065093964338303), (26, 0.013419623486697674), (27, 0.01121341623365879)], Temporal MSE: [(0, 8.725257885089377e-07), (1, 4.711549763669609e-07), (2, 1.910348146338947e-06), (3, 9.612400390324183e-06), (4, 1.952029560925439e-05), (5, 2.4924596800701693e-05), (6, 6.450082582887262e-05), (7, 6.936954014236107e-05), (8, 5.7205168559448794e-05), (9, 8.436622738372535e-05), (10, 0.0001597405644133687), (11, 0.00016790918016340584), (12, 0.00018409983022138476), (13, 0.00030454047373495996), (14, 0.0004737479903269559), (15, 0.0003739955718629062), (16, 0.00043254331103526056), (17, 0.00044796522706747055), (18, 0.0004763393953908235), (19, 0.0006926791393198073), (20, 0.0012241804506629705), (21, 0.001132243312895298), (22, 0.0009839414851740003), (23, 0.0015566436341032386), (24, 0.004183268640190363), (25, 0.0067197238095104694), (26, 0.0011203524190932512), (27, 9.268903522752225e-05)]
+
+ 61%|██████    | 91/150 [13:57<07:46,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 348, Spatial MSE: [(0, 8.230254024965689e-07), (1, 8.099639671854675e-06), (2, 1.398349013470579e-05), (3, 6.995986950641964e-06), (4, 1.7015158846334089e-06), (5, 1.4219366448742221e-06), (6, 1.7643938008404803e-06), (7, 2.691106146812672e-06), (8, 1.8820104514816194e-06), (9, 2.411898094578646e-06), (10, 5.929713552177418e-06), (11, 1.8975981220137328e-05), (12, 4.960433579981327e-05), (13, 0.001555265043862164), (14, 0.0017909364541992545), (15, 0.0005439566448330879), (16, 0.02619103528559208), (17, 0.00010351352102588862), (18, 6.624634988838807e-05), (19, 0.0021250001154839993), (20, 0.0014010737650096416), (21, 0.00010533730528550223), (22, 0.00022637905203737319), (23, 0.00011542146239662543), (24, 0.0002469537721481174), (25, 0.0004836533626075834), (26, 0.00044099532533437014), (27, 0.018703710287809372)], Temporal MSE: [(0, 6.816464974690462e-07), (1, 2.4633152406750014e-07), (2, 3.2746780220804794e-07), (3, 4.4520334085973445e-07), (4, 6.489057113867602e-07), (5, 7.561985739812371e-07), (6, 8.004387268556457e-07), (7, 9.7287556855008e-07), (8, 6.723128649355203e-07), (9, 8.676743732394243e-07), (10, 4.40697021986125e-06), (11, 3.9897589886095375e-05), (12, 2.9929462471045554e-05), (13, 4.581121538649313e-05), (14, 3.0517350751324557e-05), (15, 5.170808071852662e-05), (16, 3.3165564673254266e-05), (17, 3.0139954105834477e-05), (18, 4.900321800960228e-05), (19, 3.8362933992175385e-05), (20, 2.6832822186406702e-05), (21, 6.572473648702726e-05), (22, 8.152508962666616e-05), (23, 0.00014686807116959244), (24, 0.00014738229219801724), (25, 0.0004330160445533693), (26, 7.983828982105479e-05), (27, 7.51063198549673e-05)]
+
+ 61%|██████▏   | 92/150 [14:05<07:35,  7.85s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 342, Spatial MSE: [(0, 1.3385265447141137e-06), (1, 7.3987621362903155e-06), (2, 1.0826372090377845e-05), (3, 6.381714229064528e-06), (4, 3.604403673307388e-06), (5, 7.48197635402903e-06), (6, 1.2977668120583985e-05), (7, 1.1197843377885874e-05), (8, 1.5606623492203653e-05), (9, 2.839693843270652e-05), (10, 5.335939204087481e-05), (11, 8.09244011179544e-05), (12, 0.0001250297063961625), (13, 0.002038139384239912), (14, 0.0027699810452759266), (15, 0.0011094659566879272), (16, 0.03765491768717766), (17, 0.00017095838848035783), (18, 0.00012771230831276625), (19, 0.002798335626721382), (20, 0.0021796775981783867), (21, 0.0001796292926883325), (22, 0.000317965546855703), (23, 0.0002712559071369469), (24, 0.0005908766761422157), (25, 0.0014070926699787378), (26, 0.0027493792586028576), (27, 0.018343953415751457)], Temporal MSE: [(0, 6.439241815314745e-07), (1, 3.5967909184364544e-07), (2, 6.314624556580384e-07), (3, 1.1933111636608373e-06), (4, 2.833643748090253e-06), (5, 2.7439443783805473e-06), (6, 3.6722217373608146e-06), (7, 4.904951765638543e-06), (8, 5.223148491495522e-06), (9, 5.405814590631053e-06), (10, 1.2985859939362854e-05), (11, 6.405826570698991e-05), (12, 5.6959001085488126e-05), (13, 7.694307714700699e-05), (14, 5.954972584731877e-05), (15, 7.730575453024358e-05), (16, 4.870722841587849e-05), (17, 4.625854126061313e-05), (18, 6.639285129494965e-05), (19, 6.088950976845808e-05), (20, 6.690175359835848e-05), (21, 0.00010813344124471769), (22, 0.00013071991270408034), (23, 0.00020321209740359336), (24, 0.00027337056235410273), (25, 0.000688699830789119), (26, 0.0002025407738983631), (27, 8.452059410046786e-05)]
+
+ 62%|██████▏   | 93/150 [14:12<07:27,  7.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 336, Spatial MSE: [(0, 5.858483405063453e-07), (1, 7.152647867769701e-06), (2, 1.2199760021758266e-05), (3, 6.374501936079469e-06), (4, 1.6019952226997702e-06), (5, 1.5704179077147273e-06), (6, 1.5819746295164805e-06), (7, 2.9167883894842817e-06), (8, 1.8353949826632743e-06), (9, 2.231302460131701e-06), (10, 5.300345037539955e-06), (11, 2.8920849217684008e-05), (12, 6.813111394876614e-05), (13, 0.0013761153677478433), (14, 0.001743289758451283), (15, 0.0005301377968862653), (16, 0.036211516708135605), (17, 0.00020064052660018206), (18, 0.00011973895016126335), (19, 0.0027491943910717964), (20, 0.001975942635908723), (21, 0.00014530021871905774), (22, 0.0003255633928347379), (23, 0.00014029841986484826), (24, 0.0003231836308259517), (25, 0.0006612016004510224), (26, 0.0005167966592125595), (27, 0.01879601553082466)], Temporal MSE: [(0, 5.333905050974863e-07), (1, 2.7437016569820116e-07), (2, 3.531033598846989e-07), (3, 4.373816295810684e-07), (4, 6.487760515483387e-07), (5, 7.588021730953187e-07), (6, 8.520854066773609e-07), (7, 1.0224812285741791e-06), (8, 6.691224712085386e-07), (9, 7.305382041522535e-07), (10, 3.4366237287031254e-06), (11, 7.411283149849623e-05), (12, 3.555874354788102e-05), (13, 4.529350917437114e-05), (14, 3.608584665926173e-05), (15, 6.780218245694414e-05), (16, 5.157632767804898e-05), (17, 5.579338539973833e-05), (18, 5.724619404645637e-05), (19, 4.735096445074305e-05), (20, 3.478130020084791e-05), (21, 0.00010667992319213226), (22, 0.00013048479740973562), (23, 0.0002000848762691021), (24, 0.00015979290765244514), (25, 0.0003877760609611869), (26, 7.719681889284402e-05), (27, 8.913987403502688e-05)]
+
+ 63%|██████▎   | 94/150 [14:20<07:12,  7.73s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 330, Spatial MSE: [(0, 8.922413599066203e-07), (1, 1.2435102689778432e-05), (2, 2.256032166769728e-05), (3, 1.3608365406980738e-05), (4, 2.2780754079576582e-05), (5, 5.434422928374261e-05), (6, 9.10374365048483e-05), (7, 0.00017910412861965597), (8, 0.0002863809932023287), (9, 0.0003204970562364906), (10, 0.0003140182525385171), (11, 0.0003444391186349094), (12, 0.00044646687456406653), (13, 0.0023671381641179323), (14, 0.00385602586902678), (15, 0.0026036801282316446), (16, 0.04805104061961174), (17, 0.0009526729118078947), (18, 0.0010069699492305517), (19, 0.003970717079937458), (20, 0.0034674471244215965), (21, 0.001560868346132338), (22, 0.002066465327516198), (23, 0.0032483323011547327), (24, 0.006258493289351463), (25, 0.010873227380216122), (26, 0.0123048797249794), (27, 0.01835821568965912)], Temporal MSE: [(0, 7.438080160682148e-07), (1, 4.031863056752627e-07), (2, 1.6738486010581255e-06), (3, 1.1705300494213589e-05), (4, 2.2825515770819038e-05), (5, 2.78672141575953e-05), (6, 6.595876038772985e-05), (7, 6.478642899310216e-05), (8, 5.227604924584739e-05), (9, 7.775286212563515e-05), (10, 0.0001448195253033191), (11, 0.00022390046797227114), (12, 0.00018335079948883504), (13, 0.0002725481172092259), (14, 0.00041528051951900125), (15, 0.00035487982677295804), (16, 0.00039429255411960185), (17, 0.0004031382268294692), (18, 0.00043214205652475357), (19, 0.0006071670795790851), (20, 0.0010627112351357937), (21, 0.0009835421806201339), (22, 0.0009141711634583771), (23, 0.0014400973450392485), (24, 0.0036774093750864267), (25, 0.006059517618268728), (26, 0.0010729777859523892), (27, 0.00013629067689180374)]
+
+ 63%|██████▎   | 95/150 [14:28<07:13,  7.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 324, Spatial MSE: [(0, 7.289613108696358e-07), (1, 8.341348802787252e-06), (2, 1.4459784324571956e-05), (3, 7.169869149947772e-06), (4, 1.7995039343077224e-06), (5, 1.8029201100944192e-06), (6, 2.0181353193038376e-06), (7, 3.2277002901537344e-06), (8, 2.306594979017973e-06), (9, 3.2217783427768154e-06), (10, 9.600634257367346e-06), (11, 3.198851118213497e-05), (12, 7.856747106416151e-05), (13, 0.0019106781110167503), (14, 0.0022119530476629734), (15, 0.0006818502442911267), (16, 0.03161213546991348), (17, 0.00013882073108106852), (18, 0.00020867917919531465), (19, 0.0022396475542336702), (20, 0.0014983557630330324), (21, 0.0001695446262601763), (22, 0.0003429174539633095), (23, 0.00014534659567289054), (24, 0.0003138496831525117), (25, 0.0005962736904621124), (26, 0.0004928253474645317), (27, 0.020586887374520302)], Temporal MSE: [(0, 6.361421469591733e-07), (1, 3.070094294344017e-07), (2, 3.892867539434519e-07), (3, 5.119532033859286e-07), (4, 7.749796395728481e-07), (5, 9.257566375708848e-07), (6, 9.875167279460584e-07), (7, 1.216655618918594e-06), (8, 8.159730668921839e-07), (9, 1.1673779454213218e-06), (10, 6.964242402318632e-06), (11, 6.467571802204475e-05), (12, 4.390467802295461e-05), (13, 5.988629345665686e-05), (14, 4.373185947770253e-05), (15, 6.0170226788613945e-05), (16, 3.706291681737639e-05), (17, 3.341007686685771e-05), (18, 4.944299871567637e-05), (19, 3.746904985746369e-05), (20, 2.975449024233967e-05), (21, 9.702691750135273e-05), (22, 0.00011661516327876598), (23, 0.00020687494543381035), (24, 0.0001765611523296684), (25, 0.0004481865034904331), (26, 8.789694402366877e-05), (27, 9.126211080001667e-05)]
+
+ 64%|██████▍   | 96/150 [14:36<07:04,  7.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 318, Spatial MSE: [(0, 9.84719463303918e-07), (1, 7.015920800768072e-06), (2, 8.948238246375695e-06), (3, 6.260409918468213e-06), (4, 3.726101795109571e-06), (5, 8.040366992645431e-06), (6, 1.252937909157481e-05), (7, 1.2634068298211787e-05), (8, 1.8877726688515395e-05), (9, 3.460308653302491e-05), (10, 6.211388244992122e-05), (11, 9.801662235986441e-05), (12, 0.00013745007163379341), (13, 0.001952288206666708), (14, 0.0023095232900232077), (15, 0.0007435752195306122), (16, 0.031855255365371704), (17, 0.00016708835028111935), (18, 0.00026194530073553324), (19, 0.003469013376161456), (20, 0.0025406896602362394), (21, 0.00034518272150307894), (22, 0.00042833309271372855), (23, 0.00028738341643474996), (24, 0.0006445524631999433), (25, 0.0014407634735107422), (26, 0.0028706416487693787), (27, 0.020153021439909935)], Temporal MSE: [(0, 5.55712347249937e-07), (1, 3.1936184541336843e-07), (2, 5.894803507544566e-07), (3, 1.1382475122445612e-06), (4, 2.974788685605745e-06), (5, 2.8570325412147213e-06), (6, 3.968550117861014e-06), (7, 5.298848918755539e-06), (8, 5.5544533097418025e-06), (9, 6.55846088193357e-06), (10, 1.6093443264253438e-05), (11, 7.44934004615061e-05), (12, 5.3895069868303835e-05), (13, 7.777711289236322e-05), (14, 6.327109440462664e-05), (15, 7.588644803036004e-05), (16, 4.9786125600803643e-05), (17, 4.328164868638851e-05), (18, 7.058541086735204e-05), (19, 7.143872790038586e-05), (20, 7.960972288856283e-05), (21, 0.00012846679601352662), (22, 0.00014984916197136045), (23, 0.0002378209464950487), (24, 0.0002854083140846342), (25, 0.000708304694853723), (26, 0.0002105456660501659), (27, 0.00010071018914459273)]
+
+ 65%|██████▍   | 97/150 [14:44<07:02,  7.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 312, Spatial MSE: [(0, 5.127212716615759e-07), (1, 7.05910861142911e-06), (2, 1.1297380297037307e-05), (3, 6.256152119021863e-06), (4, 1.4479618357654545e-06), (5, 1.6966253042483004e-06), (6, 1.5602155372107518e-06), (7, 2.616429810586851e-06), (8, 1.7427621514798375e-06), (9, 2.233477516710991e-06), (10, 2.698915523069445e-06), (11, 4.93628886033548e-06), (12, 1.4448518413701095e-05), (13, 0.0006098462035879493), (14, 0.0010925163514912128), (15, 0.0005307675455696881), (16, 0.02495539002120495), (17, 9.083818440558389e-05), (18, 5.483909626491368e-05), (19, 0.0019251916091889143), (20, 0.001302893040701747), (21, 0.00011316234304104), (22, 0.0002164226898457855), (23, 0.00011244488268857822), (24, 0.0002131023647962138), (25, 0.0004956643097102642), (26, 0.00042610071250237525), (27, 0.015717539936304092)], Temporal MSE: [(0, 5.191681111682556e-07), (1, 2.710115438731009e-07), (2, 3.4918519986604224e-07), (3, 4.557229829060816e-07), (4, 6.963619512134755e-07), (5, 8.200746037800855e-07), (6, 8.597671126153728e-07), (7, 1.0138604693565867e-06), (8, 6.855741503386525e-07), (9, 6.897004141137586e-07), (10, 1.844109988269338e-06), (11, 5.785926532553276e-06), (12, 9.27009568840731e-06), (13, 3.0507762858178467e-05), (14, 2.828226388373878e-05), (15, 2.9268456273712218e-05), (16, 2.332296753593255e-05), (17, 3.417081825318746e-05), (18, 4.6562388888560236e-05), (19, 3.778978862101212e-05), (20, 3.0202292691683397e-05), (21, 6.311611650744453e-05), (22, 7.83555005909875e-05), (23, 0.00014620802539866418), (24, 0.0001271190121769905), (25, 0.00035343971103429794), (26, 6.92130415700376e-05), (27, 7.03756304574199e-05)]
+
+ 65%|██████▌   | 98/150 [14:52<06:53,  7.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 306, Spatial MSE: [(0, 9.356897407997167e-07), (1, 1.2040204637742136e-05), (2, 1.624849028303288e-05), (3, 1.1747862117772456e-05), (4, 2.0887151549686678e-05), (5, 5.453165795188397e-05), (6, 8.422127575613558e-05), (7, 0.00017711721011437476), (8, 0.0002723193902056664), (9, 0.0002875015779864043), (10, 0.000281915272353217), (11, 0.0002904829161707312), (12, 0.0003733210323844105), (13, 0.0021422700956463814), (14, 0.005268005188554525), (15, 0.0034321420826017857), (16, 0.09079551696777344), (17, 0.0007836814620532095), (18, 0.0009415002423338592), (19, 0.0058564539067447186), (20, 0.005150326061993837), (21, 0.0014636809937655926), (22, 0.001823615049943328), (23, 0.002972022630274296), (24, 0.005493553820997477), (25, 0.009469176642596722), (26, 0.011973313987255096), (27, 0.015871170908212662)], Temporal MSE: [(0, 8.586771400587168e-07), (1, 4.3519995074348117e-07), (2, 1.6227131709456444e-06), (3, 1.1490329598018434e-05), (4, 2.147752820746973e-05), (5, 2.6618419724400155e-05), (6, 6.042065433575772e-05), (7, 6.247824057936668e-05), (8, 4.780536255566403e-05), (9, 7.203748100437224e-05), (10, 0.00012988854723516852), (11, 0.0001423586072633043), (12, 0.00015326731954701245), (13, 0.0002581309818197042), (14, 0.0003841806319542229), (15, 0.00031467320513911545), (16, 0.0003457776620052755), (17, 0.00036361374077387154), (18, 0.0004028331604786217), (19, 0.0005601915763691068), (20, 0.0009686915436759591), (21, 0.000941805774345994), (22, 0.0008509107283316553), (23, 0.0013114522444084287), (24, 0.0033291096333414316), (25, 0.005328513216227293), (26, 0.0010255560046061873), (27, 0.00012154566502431408)]
+
+ 66%|██████▌   | 99/150 [15:01<06:55,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 300, Spatial MSE: [(0, 7.412211289192783e-07), (1, 8.260554750449955e-06), (2, 1.322207481280202e-05), (3, 7.1554154601471964e-06), (4, 1.9284634618088603e-06), (5, 2.0721925011457643e-06), (6, 1.9466474441287573e-06), (7, 3.4916317872557556e-06), (8, 2.425259935989743e-06), (9, 3.6251121855457313e-06), (10, 9.09190384845715e-06), (11, 1.9206991055398248e-05), (12, 3.766052032005973e-05), (13, 0.0010342743480578065), (14, 0.0023729552049189806), (15, 0.0009795234072953463), (16, 0.059346169233322144), (17, 0.0003204419044777751), (18, 0.00018998695304617286), (19, 0.00447247875854373), (20, 0.003272251458838582), (21, 0.0004040248168166727), (22, 0.0005145830800756812), (23, 0.00021556219144258648), (24, 0.00045923289144411683), (25, 0.0011426220880821347), (26, 0.0006100026075728238), (27, 0.026316456496715546)], Temporal MSE: [(0, 7.275497182490653e-07), (1, 3.7696469235015684e-07), (2, 4.54395575388844e-07), (3, 5.901357553739217e-07), (4, 9.323228482571722e-07), (5, 1.1248704367972095e-06), (6, 1.2583968782564625e-06), (7, 1.3791602668788983e-06), (8, 9.354503731628938e-07), (9, 1.0516627071410767e-06), (10, 5.38742233402445e-06), (11, 1.7021060557453893e-05), (12, 2.7062429580837488e-05), (13, 4.093549432582222e-05), (14, 5.442813926492818e-05), (15, 0.00011999448906863108), (16, 8.708882523933426e-05), (17, 9.068585495697334e-05), (18, 8.666808571433648e-05), (19, 7.238867692649364e-05), (20, 5.36658990313299e-05), (21, 0.0001866176608018577), (22, 0.00022197398357093334), (23, 0.0003497081343084574), (24, 0.00026348940446041524), (25, 0.0005948037724010646), (26, 0.00011387898121029139), (27, 0.00014047180593479425)]
+
+ 67%|██████▋   | 100/150 [15:08<06:41,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 294, Spatial MSE: [(0, 9.170993848783837e-07), (1, 9.775241778697819e-06), (2, 1.2964464076503646e-05), (3, 7.315902621485293e-06), (4, 4.376420747576049e-06), (5, 8.22578658699058e-06), (6, 1.4084791473578662e-05), (7, 1.3659105206897948e-05), (8, 1.812955815694295e-05), (9, 3.7351594073697925e-05), (10, 6.533528357977048e-05), (11, 0.00010320117144146934), (12, 0.00013238897372502834), (13, 0.0018922295421361923), (14, 0.004911566618829966), (15, 0.0028246785514056683), (16, 0.09837410598993301), (17, 0.00046498762094415724), (18, 0.0003490171511657536), (19, 0.005969573277980089), (20, 0.00484659057110548), (21, 0.0006096249562688172), (22, 0.0006961612380109727), (23, 0.0004734177200589329), (24, 0.0010161732789129019), (25, 0.002526269992813468), (26, 0.0034088497050106525), (27, 0.025169089436531067)], Temporal MSE: [(0, 5.235723961050098e-07), (1, 3.1863888239058724e-07), (2, 6.30938245649304e-07), (3, 1.2399934803397628e-06), (4, 3.2170983104151674e-06), (5, 3.183595708833309e-06), (6, 4.7131120481935795e-06), (7, 6.329200914478861e-06), (8, 6.762857537978562e-06), (9, 7.077589089021785e-06), (10, 1.6210928151849657e-05), (11, 4.7395475121447816e-05), (12, 5.9984842664562166e-05), (13, 8.652381075080484e-05), (14, 0.0001019949559122324), (15, 0.00018126817303709686), (16, 0.00010933345038210973), (17, 0.00011159922723891214), (18, 0.00012405692541506141), (19, 0.00011771926801884547), (20, 0.00016461321501992643), (21, 0.0002702621277421713), (22, 0.0003006089245900512), (23, 0.00043794323573820293), (24, 0.0004770754312630743), (25, 0.0010153765324503183), (26, 0.00025289456243626773), (27, 0.0001484951062593609)]
+
+ 67%|██████▋   | 101/150 [15:17<06:38,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 288, Spatial MSE: [(0, 6.526437914544658e-07), (1, 9.705875527288299e-06), (2, 1.556577808514703e-05), (3, 7.204682333394885e-06), (4, 2.072846427836339e-06), (5, 2.1925363853370072e-06), (6, 2.084003881464014e-06), (7, 3.4687257084442535e-06), (8, 2.857546405721223e-06), (9, 4.6983245738374535e-06), (10, 2.870375828933902e-05), (11, 0.0001298727438552305), (12, 0.00021010084310546517), (13, 0.00424992386251688), (14, 0.0046618725173175335), (15, 0.0013075758470222354), (16, 0.07083826512098312), (17, 0.00034978266921825707), (18, 0.00034221794339828193), (19, 0.005310803651809692), (20, 0.003736414248123765), (21, 0.00023662298917770386), (22, 0.0005769904237240553), (23, 0.0002617976861074567), (24, 0.0005723057547584176), (25, 0.001289453823119402), (26, 0.000818349071778357), (27, 0.041146762669086456)], Temporal MSE: [(0, 5.694412834600371e-07), (1, 3.218679296423943e-07), (2, 3.8636272847725195e-07), (3, 5.310867550178955e-07), (4, 8.26958057587035e-07), (5, 9.939935807778966e-07), (6, 1.0172302609134931e-06), (7, 1.3018294566791155e-06), (8, 8.95018501978484e-07), (9, 1.6949041992120328e-06), (10, 2.0635552573367022e-05), (11, 0.00026577329845167696), (12, 0.00010176469368161634), (13, 9.692896128399298e-05), (14, 7.01879253028892e-05), (15, 0.0002010467287618667), (16, 0.00010511232540011406), (17, 0.00010585329437162727), (18, 0.00011162953160237521), (19, 8.96535930223763e-05), (20, 9.153599239652976e-05), (21, 0.00017934154311660677), (22, 0.00023537053493782878), (23, 0.0004051168798469007), (24, 0.00033169661764986813), (25, 0.0007851081318221986), (26, 0.00017609349742997438), (27, 0.0001873721630545333)]
+
+ 68%|██████▊   | 102/150 [15:24<06:23,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 282, Spatial MSE: [(0, 8.876106107891246e-07), (1, 1.2749012967105955e-05), (2, 1.748905924614519e-05), (3, 1.2465059626265429e-05), (4, 2.2121763322502375e-05), (5, 5.56750237592496e-05), (6, 8.05575618869625e-05), (7, 0.00016623022384010255), (8, 0.00025509140687063336), (9, 0.00026863839593715966), (10, 0.0002833821636158973), (11, 0.0003863832971546799), (12, 0.0005219991435296834), (13, 0.005620053503662348), (14, 0.007548210676759481), (15, 0.003270102431997657), (16, 0.1006578579545021), (17, 0.00101742974948138), (18, 0.001163120032288134), (19, 0.010157708078622818), (20, 0.008419542573392391), (21, 0.0015617160825058818), (22, 0.001996211474761367), (23, 0.002915652235969901), (24, 0.005243503022938967), (25, 0.0094697130843997), (26, 0.011485791765153408), (27, 0.04075833037495613)], Temporal MSE: [(0, 8.6694308265578e-07), (1, 4.1911133052963123e-07), (2, 1.7594159089640016e-06), (3, 1.1387397535145283e-05), (4, 2.202691393904388e-05), (5, 2.549358396208845e-05), (6, 5.3815725550521165e-05), (7, 5.850561137776822e-05), (8, 4.3784901208709925e-05), (9, 6.745537393726408e-05), (10, 0.00013493580627255142), (11, 0.0003791854833252728), (12, 0.00022745570458937436), (13, 0.0003174009034410119), (14, 0.0003953753912355751), (15, 0.00043632742017507553), (16, 0.00040251537575386465), (17, 0.00041156611405313015), (18, 0.0004567673895508051), (19, 0.0005976122338324785), (20, 0.0009532888070680201), (21, 0.0010199007811024785), (22, 0.0009408096666447818), (23, 0.0014220610028132796), (24, 0.0031045274809002876), (25, 0.0052405535243451595), (26, 0.0011244858615100384), (27, 0.00023535404761787504)]
+
+ 69%|██████▊   | 103/150 [15:33<06:24,  8.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 276, Spatial MSE: [(0, 5.346831812857999e-07), (1, 8.173022251867224e-06), (2, 1.303714634559583e-05), (3, 7.173436188168125e-06), (4, 1.921633838719572e-06), (5, 2.3140664779930376e-06), (6, 1.9261049146734877e-06), (7, 4.417898253450403e-06), (8, 2.7455569124867907e-06), (9, 2.5639014893386047e-06), (10, 9.25104359339457e-06), (11, 6.738332012901083e-05), (12, 0.00017829456191975623), (13, 0.004804870113730431), (14, 0.004384916741400957), (15, 0.0012210699496790767), (16, 0.05079874023795128), (17, 0.0002574255340732634), (18, 0.00013299149577505887), (19, 0.0041594961658120155), (20, 0.0026854383759200573), (21, 0.00021467197802849114), (22, 0.0004510314029175788), (23, 0.0002449285821057856), (24, 0.00046204664977267385), (25, 0.0009430783102288842), (26, 0.0007141637615859509), (27, 0.0427561029791832)], Temporal MSE: [(0, 6.248710064937768e-07), (1, 3.5255393981969974e-07), (2, 4.3944729100076074e-07), (3, 6.063557975721778e-07), (4, 9.999861276810407e-07), (5, 1.1452812032075599e-06), (6, 1.403537908117869e-06), (7, 1.4393413039215375e-06), (8, 9.472148008171644e-07), (9, 1.1442024288044195e-06), (10, 6.686426786473021e-06), (11, 0.00018926830671261996), (12, 0.00010538600326981395), (13, 0.00012292635801713914), (14, 6.210565334185958e-05), (15, 0.00011281146726105362), (16, 7.150807505240664e-05), (17, 5.881543256691657e-05), (18, 9.696317283669487e-05), (19, 7.47997110011056e-05), (20, 5.298868927638978e-05), (21, 0.00012856710236519575), (22, 0.0001614500069990754), (23, 0.000301802676403895), (24, 0.0002880019601434469), (25, 0.0008528885082341731), (26, 0.00017085463332477957), (27, 0.00016448507085442543)]
+
+ 69%|██████▉   | 104/150 [15:41<06:14,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 270, Spatial MSE: [(0, 9.472688020650821e-07), (1, 8.660081221023574e-06), (2, 9.25964650377864e-06), (3, 6.290950295806397e-06), (4, 3.926458248315612e-06), (5, 9.069543011719361e-06), (6, 1.3101063814247027e-05), (7, 1.3172066246625036e-05), (8, 1.832773159549106e-05), (9, 3.4295691875740886e-05), (10, 6.612102151848376e-05), (11, 0.00012635180610232055), (12, 0.000236940904869698), (13, 0.004857833031564951), (14, 0.0045583825558424), (15, 0.0013319745194166899), (16, 0.05107106268405914), (17, 0.0002885505382437259), (18, 0.00016144396795425564), (19, 0.0041687870398163795), (20, 0.0027046804316341877), (21, 0.00024012790527194738), (22, 0.0005194966797716916), (23, 0.0003784688888117671), (24, 0.0007814553682692349), (25, 0.0019387692445889115), (26, 0.003610665909945965), (27, 0.04178408905863762)], Temporal MSE: [(0, 6.042000109118817e-07), (1, 3.584437138215435e-07), (2, 6.344930056911835e-07), (3, 1.2385492027533473e-06), (4, 3.3935214105440537e-06), (5, 3.4279705687367823e-06), (6, 4.629586783266859e-06), (7, 6.22967036179034e-06), (8, 6.0299043980194256e-06), (9, 6.647721875197021e-06), (10, 1.4827900486125145e-05), (11, 0.00019124581012874842), (12, 0.00010920936620095745), (13, 0.0001397271262248978), (14, 8.431483729509637e-05), (15, 0.0001303841854678467), (16, 8.386938861804083e-05), (17, 6.890154327265918e-05), (18, 0.00011130843631690368), (19, 9.131777187576517e-05), (20, 7.814892160240561e-05), (21, 0.00015456673281732947), (22, 0.00018873838416766375), (23, 0.00034524587681517005), (24, 0.000409524654969573), (25, 0.0011612827656790614), (26, 0.0003006580809596926), (27, 0.00017637884593568742)]
+
+ 70%|███████   | 105/150 [15:49<06:04,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 264, Spatial MSE: [(0, 1.0311916867067339e-06), (1, 8.87419082573615e-06), (2, 1.2184794286440592e-05), (3, 6.615248366870219e-06), (4, 1.7376995629092562e-06), (5, 2.3884520032879664e-06), (6, 2.191816975027905e-06), (7, 3.869427018798888e-06), (8, 2.610897126942291e-06), (9, 2.837545025613508e-06), (10, 4.703386366600171e-06), (11, 1.977369174710475e-05), (12, 8.05974705144763e-05), (13, 0.0028710144106298685), (14, 0.0028762822039425373), (15, 0.001055949367582798), (16, 0.03856419399380684), (17, 0.00011865326814586297), (18, 0.00022681354312226176), (19, 0.0026536481454968452), (20, 0.0016198079101741314), (21, 0.00038588649476878345), (22, 0.0005108323530294001), (23, 0.0002466210862621665), (24, 0.0004411736153997481), (25, 0.0009368915925733745), (26, 0.0006265718257054687), (27, 0.031666673719882965)], Temporal MSE: [(0, 8.871537033883214e-07), (1, 3.9521231087746855e-07), (2, 4.90263914798561e-07), (3, 6.141451081020932e-07), (4, 9.91075921774609e-07), (5, 1.159501834990806e-06), (6, 1.217790895680082e-06), (7, 1.518522140031564e-06), (8, 9.614157079340657e-07), (9, 1.0204367981714313e-06), (10, 3.3043088478734717e-06), (11, 4.65855082438793e-05), (12, 6.524589116452262e-05), (13, 8.768054249230772e-05), (14, 5.596468326984905e-05), (15, 7.174594065872952e-05), (16, 4.219602487864904e-05), (17, 4.179285679128952e-05), (18, 8.693933341419324e-05), (19, 6.925998604856431e-05), (20, 5.162950401427224e-05), (21, 0.00013991519517730922), (22, 0.00016800712910480797), (23, 0.00028397628921084106), (24, 0.00028122717048972845), (25, 0.0008336434257216752), (26, 0.00014803532394580543), (27, 0.00014702386397402734)]
+
+ 71%|███████   | 106/150 [15:57<05:54,  8.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 258, Spatial MSE: [(0, 1.239065568370279e-06), (1, 1.0546423254709225e-05), (2, 1.5189712939900346e-05), (3, 1.1397895832487848e-05), (4, 1.8092307072947733e-05), (5, 5.4317275498760864e-05), (6, 7.968046702444553e-05), (7, 0.00015804237045813352), (8, 0.00022543825616594404), (9, 0.00023458313080482185), (10, 0.00022649246966466308), (11, 0.0002490261977072805), (12, 0.000375027913833037), (13, 0.005305431317538023), (14, 0.008887897245585918), (15, 0.004662336781620979), (16, 0.12510176002979279), (17, 0.0007065555546432734), (18, 0.0009759741369634867), (19, 0.008133018389344215), (20, 0.006725641433149576), (21, 0.0015682289376854897), (22, 0.001960210967808962), (23, 0.002718154340982437), (24, 0.004781311843544245), (25, 0.008164242841303349), (26, 0.01084133144468069), (27, 0.031206414103507996)], Temporal MSE: [(0, 1.0650343256202177e-06), (1, 4.5997899178473745e-07), (2, 1.7478927247793763e-06), (3, 1.2489180335251149e-05), (4, 2.1145158825675026e-05), (5, 2.5295134037151e-05), (6, 5.234532000031322e-05), (7, 5.6459353800164536e-05), (8, 4.025104863103479e-05), (9, 6.0252146795392036e-05), (10, 0.00010862224735319614), (11, 0.00016710592899471521), (12, 0.00020497104560490698), (13, 0.0003087428631260991), (14, 0.00038756048888899386), (15, 0.0003214810276404023), (16, 0.0003165311354678124), (17, 0.00032714405097067356), (18, 0.0003985425573773682), (19, 0.0005144936731085181), (20, 0.0008355133468285203), (21, 0.0009023465099744499), (22, 0.0008561042486689985), (23, 0.0012764236889779568), (24, 0.0026538202073425055), (25, 0.004694322124123573), (26, 0.001038610003888607), (27, 0.00019521574722602963)]
+
+ 71%|███████▏  | 107/150 [16:06<05:59,  8.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 252, Spatial MSE: [(0, 6.603082738365629e-07), (1, 8.996706128527876e-06), (2, 1.3362098798097577e-05), (3, 7.4169984145555645e-06), (4, 1.8059231479128357e-06), (5, 2.4560908968851436e-06), (6, 2.123008698617923e-06), (7, 3.9102260416257195e-06), (8, 2.5591971279936843e-06), (9, 2.682573722267989e-06), (10, 7.018984888418345e-06), (11, 2.4228875190601684e-05), (12, 8.152320515364408e-05), (13, 0.0026863724924623966), (14, 0.0034461587201803923), (15, 0.001236627227626741), (16, 0.06569933146238327), (17, 0.0002991393848787993), (18, 0.0002317500184290111), (19, 0.005084770731627941), (20, 0.0036710496060550213), (21, 0.00022198900114744902), (22, 0.0005654196138493717), (23, 0.0002520076814107597), (24, 0.0005231807590462267), (25, 0.00126400135923177), (26, 0.0007226258749142289), (27, 0.03730439767241478)], Temporal MSE: [(0, 6.532353040711314e-07), (1, 3.718772063621145e-07), (2, 4.668816870889714e-07), (3, 6.589925192201918e-07), (4, 1.1031092981284019e-06), (5, 1.3376502465689555e-06), (6, 1.513196025371144e-06), (7, 1.6207474118346e-06), (8, 1.0611779543978628e-06), (9, 1.0770742164822877e-06), (10, 4.359034846856957e-06), (11, 5.486379814101383e-05), (12, 5.816385237267241e-05), (13, 8.282627823064104e-05), (14, 7.062077929731458e-05), (15, 0.00011496798833832145), (16, 8.079674444161355e-05), (17, 0.00010872829443542287), (18, 0.00010197022493230179), (19, 8.444309787591919e-05), (20, 6.267795106396079e-05), (21, 0.00019104790408164263), (22, 0.00022790921502746642), (23, 0.00038800309994257987), (24, 0.0002982410369440913), (25, 0.0007883064681664109), (26, 0.00015304233238566667), (27, 0.00018393236678093672)]
+
+ 72%|███████▏  | 108/150 [16:15<05:53,  8.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 246, Spatial MSE: [(0, 7.561941970379848e-07), (1, 1.0797798495332245e-05), (2, 1.1208832802367397e-05), (3, 6.449634838645579e-06), (4, 4.286481271265075e-06), (5, 8.699775207787752e-06), (6, 1.4367627045430709e-05), (7, 1.3930260138295125e-05), (8, 1.8326596546103247e-05), (9, 3.1908537494018674e-05), (10, 5.770293137175031e-05), (11, 0.0001004126388579607), (12, 0.00015946743951644748), (13, 0.0032713175751268864), (14, 0.004915150348097086), (15, 0.0025242124684154987), (16, 0.09330686926841736), (17, 0.0004251220962032676), (18, 0.0005842039827257395), (19, 0.008667010813951492), (20, 0.007157267536967993), (21, 0.0005987301119603217), (22, 0.0007495201425626874), (23, 0.0005302383215166628), (24, 0.0011940550757572055), (25, 0.0028326152823865414), (26, 0.0038727743085473776), (27, 0.03623706102371216)], Temporal MSE: [(0, 6.208761078596581e-07), (1, 3.545453353126504e-07), (2, 6.506356271529512e-07), (3, 1.2142195373598952e-06), (4, 3.391137170183356e-06), (5, 3.557759555405937e-06), (6, 5.110616712045157e-06), (7, 6.707770353386877e-06), (8, 6.330148153210757e-06), (9, 7.291097062989138e-06), (10, 1.4662295143352821e-05), (11, 7.570983871119097e-05), (12, 8.009363227756694e-05), (13, 0.00011495101352920756), (14, 0.0001121685272664763), (15, 0.0001526298583485186), (16, 0.0001022396027110517), (17, 0.00012813358625862747), (18, 0.00014264453784562647), (19, 0.00015930776135064662), (20, 0.00020479167869780213), (21, 0.0002919804828707129), (22, 0.0003141330962534994), (23, 0.0004628568422049284), (24, 0.00048388083814643323), (25, 0.0011831722222268581), (26, 0.00033681216882541776), (27, 0.00019790424266830087)]
+
+ 73%|███████▎  | 109/150 [16:23<05:47,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 240, Spatial MSE: [(0, 6.761360964446794e-07), (1, 1.035553214023821e-05), (2, 1.4036683751328383e-05), (3, 7.013287358859088e-06), (4, 2.18093396142649e-06), (5, 3.004759264513268e-06), (6, 2.3866900846769568e-06), (7, 3.9728611227474175e-06), (8, 3.1296335691877175e-06), (9, 4.024187546747271e-06), (10, 1.258019074157346e-05), (11, 5.681613038177602e-05), (12, 0.00012027996854158118), (13, 0.003072138177230954), (14, 0.0026036901399493217), (15, 0.0007956595509313047), (16, 0.029495960101485252), (17, 0.00019343342864885926), (18, 8.564045856473967e-05), (19, 0.0021798082161694765), (20, 0.0013470295816659927), (21, 0.00014620754518546164), (22, 0.00047551956959068775), (23, 0.00021018092229496688), (24, 0.0003860817232634872), (25, 0.0007471985882148147), (26, 0.0005375102628022432), (27, 0.024053292348980904)], Temporal MSE: [(0, 7.40346479233267e-07), (1, 4.3216482481511775e-07), (2, 5.257605266706378e-07), (3, 7.094112675076758e-07), (4, 1.2223836165503599e-06), (5, 1.405187731506885e-06), (6, 1.6117636505441624e-06), (7, 1.7807277572501334e-06), (8, 1.1836599469461362e-06), (9, 1.44184855344065e-06), (10, 8.396842531510629e-06), (11, 0.00013241208216641098), (12, 7.227788591990247e-05), (13, 7.245937013067305e-05), (14, 4.269905184628442e-05), (15, 7.893523434177041e-05), (16, 4.0059294406091794e-05), (17, 4.270065255695954e-05), (18, 6.0962222050875425e-05), (19, 5.730427801609039e-05), (20, 6.291134923230857e-05), (21, 8.33211379358545e-05), (22, 0.0001268940104637295), (23, 0.00022077409084886312), (24, 0.00022453376732300967), (25, 0.0005250144749879837), (26, 0.0001205167718580924), (27, 0.00010566150740487501)]
+
+ 73%|███████▎  | 110/150 [16:31<05:32,  8.32s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 234, Spatial MSE: [(0, 1.0725352694862522e-06), (1, 1.200765291287098e-05), (2, 1.7750524420989677e-05), (3, 1.2337211956037208e-05), (4, 1.912715561047662e-05), (5, 5.116407555760816e-05), (6, 7.702161383349448e-05), (7, 0.00014638097491115332), (8, 0.00020808758563362062), (9, 0.0002088910696329549), (10, 0.00021485362958628684), (11, 0.0002770432911347598), (12, 0.0004273623926565051), (13, 0.007732080295681953), (14, 0.01459918636828661), (15, 0.008010132238268852), (16, 0.23777244985103607), (17, 0.0009240276413038373), (18, 0.0011518588289618492), (19, 0.013871311210095882), (20, 0.011669186875224113), (21, 0.0016578928334638476), (22, 0.0019755542743951082), (23, 0.002694343915209174), (24, 0.0045852200128138065), (25, 0.008098459802567959), (26, 0.011568955145776272), (27, 0.024063076823949814)], Temporal MSE: [(0, 1.0230769476038404e-06), (1, 5.151292725713574e-07), (2, 1.841664129642595e-06), (3, 1.1162331247760449e-05), (4, 1.9233892089687288e-05), (5, 2.3688175133429468e-05), (6, 5.068449172540568e-05), (7, 5.6578079238533974e-05), (8, 3.6512850783765316e-05), (9, 5.520934428204782e-05), (10, 0.0001064464813680388), (11, 0.00026424237876199186), (12, 0.0002700474869925529), (13, 0.0003830940113402903), (14, 0.00045818148646503687), (15, 0.00036873394856229424), (16, 0.00033942461595870554), (17, 0.00034324481384828687), (18, 0.0004101575177628547), (19, 0.0005365146207623184), (20, 0.0008815008914098144), (21, 0.0010931816650554538), (22, 0.0009072337998077273), (23, 0.0013249858748167753), (24, 0.0024225192610174417), (25, 0.003959478344768286), (26, 0.001058231689967215), (27, 0.00017153035150840878)]
+
+ 74%|███████▍  | 111/150 [16:42<05:54,  9.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 228, Spatial MSE: [(0, 9.093953963201784e-07), (1, 1.0868632671190426e-05), (2, 1.6197274817386642e-05), (3, 8.22448328108294e-06), (4, 2.196352397731971e-06), (5, 3.177354756189743e-06), (6, 4.9632403715804685e-06), (7, 5.740456799685489e-06), (8, 3.5593566281022504e-06), (9, 2.7398968995839823e-06), (10, 5.5784094001865014e-06), (11, 1.1922446901735384e-05), (12, 2.4990282327053137e-05), (13, 0.0005535444361157715), (14, 0.0013120830990374088), (15, 0.0005253722774796188), (16, 0.03624868765473366), (17, 0.0002209509548265487), (18, 7.99533081590198e-05), (19, 0.0029545833822339773), (20, 0.0023082520347088575), (21, 0.00010128362919203937), (22, 0.0002712165587581694), (23, 0.0001382030313834548), (24, 0.0002963844744954258), (25, 0.0007518982165493071), (26, 0.0004866487579420209), (27, 0.01692933961749077)], Temporal MSE: [(0, 7.893921178947494e-07), (1, 4.2679437228798633e-07), (2, 5.441390271698765e-07), (3, 7.407240900647594e-07), (4, 1.2913405953440815e-06), (5, 1.5652367437724024e-06), (6, 1.834707404668734e-06), (7, 1.933295152412029e-06), (8, 1.2687012258538743e-06), (9, 1.300802637160814e-06), (10, 3.5167306577932322e-06), (11, 1.0915425264101941e-05), (12, 1.816176518332213e-05), (13, 2.0416082406882197e-05), (14, 2.9725692002102733e-05), (15, 7.458234904333949e-05), (16, 5.5169162806123495e-05), (17, 5.68149916944094e-05), (18, 4.7176017687888816e-05), (19, 4.3668380385497585e-05), (20, 3.23105814459268e-05), (21, 0.00012891691585537046), (22, 0.00013344429316930473), (23, 0.00020046706777065992), (24, 0.00015777992666698992), (25, 0.00033086942858062685), (26, 7.731536607025191e-05), (27, 9.006261097965762e-05)]
+
+ 75%|███████▍  | 112/150 [16:51<05:42,  9.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 222, Spatial MSE: [(0, 1.1744261882995488e-06), (1, 9.342405974166468e-06), (2, 1.4972847566241398e-05), (3, 6.551142178068403e-06), (4, 5.032850822317414e-06), (5, 1.1508197530929465e-05), (6, 1.6875799701665528e-05), (7, 1.6096919353003614e-05), (8, 2.189459883084055e-05), (9, 4.0140777855413035e-05), (10, 6.916900747455657e-05), (11, 8.763380174059421e-05), (12, 0.0001011662752716802), (13, 0.0007311349036172032), (14, 0.001779666868969798), (15, 0.001068302895873785), (16, 0.04451573267579079), (17, 0.00028363027377054095), (18, 0.00012481087469495833), (19, 0.0033011315390467644), (20, 0.0026280730962753296), (21, 0.00019745151803363115), (22, 0.0003727843868546188), (23, 0.00033713103039190173), (24, 0.0007969427388161421), (25, 0.0018331946339458227), (26, 0.003940983209758997), (27, 0.01580064371228218)], Temporal MSE: [(0, 9.531802902529307e-07), (1, 5.276178853819147e-07), (2, 7.73792521613359e-07), (3, 1.4876641216687858e-06), (4, 3.770443072426133e-06), (5, 4.062284460815135e-06), (6, 6.333739747788059e-06), (7, 8.289598554256372e-06), (8, 7.3848559623002075e-06), (9, 7.865156476327684e-06), (10, 1.2834872904932126e-05), (11, 2.571435834397562e-05), (12, 3.320326504763216e-05), (13, 4.384700150694698e-05), (14, 5.8892044762615114e-05), (15, 0.00010402811312815174), (16, 7.019862823653966e-05), (17, 6.72339228913188e-05), (18, 6.616605242015794e-05), (19, 6.952432886464521e-05), (20, 0.0001037661568261683), (21, 0.00016592710744589567), (22, 0.00017421173106413335), (23, 0.00026760593755170703), (24, 0.0003265707055106759), (25, 0.0006551401456817985), (26, 0.00025329977506771684), (27, 0.0001021451098495163)]
+
+ 75%|███████▌  | 113/150 [17:00<05:29,  8.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 216, Spatial MSE: [(0, 6.600399728995399e-07), (1, 1.0534116881899536e-05), (2, 1.4046175238036085e-05), (3, 6.5572171479288954e-06), (4, 1.8491234641260235e-06), (5, 2.828330707416171e-06), (6, 2.434138195894775e-06), (7, 3.9369697333313525e-06), (8, 2.7526273242983734e-06), (9, 3.414351112951408e-06), (10, 4.4989596972300205e-06), (11, 3.2360312616219744e-05), (12, 9.023827442433685e-05), (13, 0.0021787327714264393), (14, 0.0017327946843579412), (15, 0.0008165211766026914), (16, 0.03426347300410271), (17, 0.00015585884102620184), (18, 0.00012483556929510087), (19, 0.0030335744377225637), (20, 0.002175668254494667), (21, 0.00011139884009025991), (22, 0.00035555564681999385), (23, 0.00015993889246601611), (24, 0.0003487928188405931), (25, 0.0008204203331843019), (26, 0.0004777009889949113), (27, 0.022085966542363167)], Temporal MSE: [(0, 7.156094170568394e-07), (1, 4.0457956629325054e-07), (2, 5.088648435958021e-07), (3, 6.911036507517565e-07), (4, 1.223693629981426e-06), (5, 1.4422421372728422e-06), (6, 1.7905524600791978e-06), (7, 1.7353622752125375e-06), (8, 1.2178461474832147e-06), (9, 1.1890433597727679e-06), (10, 2.971029061882291e-06), (11, 9.804824367165565e-05), (12, 5.8182344218948856e-05), (13, 4.7671990614617243e-05), (14, 4.360745515441522e-05), (15, 5.254308780422434e-05), (16, 3.2187337637878954e-05), (17, 4.468081533559598e-05), (18, 6.760805263184011e-05), (19, 6.299326196312904e-05), (20, 7.551796443294734e-05), (21, 0.00011032883048756048), (22, 0.0001393991260556504), (23, 0.00022330008505377918), (24, 0.00019154623441863805), (25, 0.00047318558790721), (26, 0.00010723993182182312), (27, 0.00010775772534543648)]
+
+ 76%|███████▌  | 114/150 [17:07<05:05,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 210, Spatial MSE: [(0, 8.835040148369444e-07), (1, 1.1979298506048508e-05), (2, 1.654771222092677e-05), (3, 1.0434851901663933e-05), (4, 1.7102114725275896e-05), (5, 5.0932387239299715e-05), (6, 7.236136298161e-05), (7, 0.00014706660294905305), (8, 0.00020111726189497858), (9, 0.0001992339239222929), (10, 0.00019358296412974596), (11, 0.00022131737205199897), (12, 0.0003136565792374313), (13, 0.0026832171715795994), (14, 0.0028542273212224245), (15, 0.0016194189665839076), (16, 0.045643150806427), (17, 0.0006164599326439202), (18, 0.001413261634297669), (19, 0.005609766114503145), (20, 0.004913968034088612), (21, 0.0013164944248273969), (22, 0.0014306523371487856), (23, 0.0021546136122196913), (24, 0.003712931415066123), (25, 0.0070862420834600925), (26, 0.010582677088677883), (27, 0.022412458434700966)], Temporal MSE: [(0, 1.0785980748551083e-06), (1, 5.731176315748598e-07), (2, 1.8720936623140005e-06), (3, 9.200594831781927e-06), (4, 1.730668736854568e-05), (5, 2.351923831156455e-05), (6, 4.9210597353521734e-05), (7, 5.7675442803883925e-05), (8, 3.6069031921215355e-05), (9, 5.4061591072240844e-05), (10, 9.215710451826453e-05), (11, 0.00018212581926491112), (12, 0.0001443039218429476), (13, 0.00019655628420878202), (14, 0.00030702375806868076), (15, 0.0002315835445187986), (16, 0.0002602298918645829), (17, 0.00027321273228153586), (18, 0.00031704900902695954), (19, 0.0004283863818272948), (20, 0.0006782099371775985), (21, 0.0007291884976439178), (22, 0.0006734908092767), (23, 0.0009747159201651812), (24, 0.0020606182515621185), (25, 0.0034696883521974087), (26, 0.000996720977127552), (27, 0.00017576833488419652)]
+
+ 77%|███████▋  | 115/150 [17:15<04:54,  8.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 204, Spatial MSE: [(0, 7.081725925672799e-07), (1, 1.0995039701811038e-05), (2, 1.4927748452464584e-05), (3, 7.669324077141937e-06), (4, 1.913304913614411e-06), (5, 3.4104375572496792e-06), (6, 2.507312046873267e-06), (7, 5.717924068449065e-06), (8, 3.4829722608265e-06), (9, 2.399181084911106e-06), (10, 2.7528731152415276e-06), (11, 5.519288606592454e-06), (12, 2.2960644855629653e-05), (13, 0.0008506895392201841), (14, 0.0016565172700211406), (15, 0.0004868489340879023), (16, 0.03342044726014137), (17, 0.00024288006534334272), (18, 7.665515295229852e-05), (19, 0.0028357920236885548), (20, 0.00217243074439466), (21, 0.00016296494868583977), (22, 0.0003779868420679122), (23, 0.00014533977082464844), (24, 0.00030024751322343946), (25, 0.0006442285957746208), (26, 0.0004110924492124468), (27, 0.01625581830739975)], Temporal MSE: [(0, 7.333711664614384e-07), (1, 4.320005189129006e-07), (2, 5.691527462658996e-07), (3, 7.820468681529746e-07), (4, 1.4111512882664101e-06), (5, 1.6778134295236669e-06), (6, 1.8818283251675894e-06), (7, 2.067972445729538e-06), (8, 1.3722366247748141e-06), (9, 1.2916254945594119e-06), (10, 1.8819229126165737e-06), (11, 1.0868153367482591e-05), (12, 2.707696876314003e-05), (13, 3.078271765843965e-05), (14, 3.423418820602819e-05), (15, 8.840190275805071e-05), (16, 6.988074164837599e-05), (17, 5.3937186748953536e-05), (18, 4.736078335554339e-05), (19, 4.0933024138212204e-05), (20, 3.127379750367254e-05), (21, 0.00012193582369945943), (22, 0.00014800501230638474), (23, 0.0002076897071674466), (24, 0.0001612236665096134), (25, 0.0003894887922797352), (26, 8.352366421604529e-05), (27, 9.119740570895374e-05)]
+
+ 77%|███████▋  | 116/150 [17:23<04:33,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 198, Spatial MSE: [(0, 9.835553100856487e-07), (1, 1.0615371138555929e-05), (2, 1.5169185644481331e-05), (3, 6.727164418407483e-06), (4, 5.277771833789302e-06), (5, 1.1314893527014647e-05), (6, 1.746231282595545e-05), (7, 1.6507845430169255e-05), (8, 2.0651410522987135e-05), (9, 4.000685294158757e-05), (10, 6.146067607915029e-05), (11, 7.980303780641407e-05), (12, 8.74717952683568e-05), (13, 0.0008016427746042609), (14, 0.001810455578379333), (15, 0.001199021004140377), (16, 0.030428549274802208), (17, 0.0003153732104692608), (18, 0.00014216743875294924), (19, 0.002390753710642457), (20, 0.001986860064789653), (21, 0.00025177650968544185), (22, 0.00046185925020836294), (23, 0.00036902198917232454), (24, 0.0008631952223367989), (25, 0.002158538671210408), (26, 0.004146234132349491), (27, 0.015821948647499084)], Temporal MSE: [(0, 9.222042081091786e-07), (1, 5.667073992299265e-07), (2, 7.895608291619283e-07), (3, 1.5047315855554189e-06), (4, 3.9999413274927065e-06), (5, 4.451334461919032e-06), (6, 7.029469088593032e-06), (7, 9.154645340458956e-06), (8, 8.132588845910504e-06), (9, 8.71401971380692e-06), (10, 1.193195021187421e-05), (11, 2.282818968524225e-05), (12, 2.9904347684350796e-05), (13, 5.1285340305184945e-05), (14, 6.355215009534732e-05), (15, 7.010593981249258e-05), (16, 7.260234997374937e-05), (17, 5.119718116475269e-05), (18, 6.369940092554316e-05), (19, 6.782298441976309e-05), (20, 7.426334923366085e-05), (21, 0.00013640231918543577), (22, 0.00016509307897649705), (23, 0.00024296740593854338), (24, 0.0003451741358730942), (25, 0.0007508955895900726), (26, 0.0002965245512314141), (27, 0.0001116379935410805)]
+
+ 78%|███████▊  | 117/150 [17:30<04:22,  7.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 192, Spatial MSE: [(0, 6.985396225900331e-07), (1, 1.103611612052191e-05), (2, 1.55794587044511e-05), (3, 7.2846310104068834e-06), (4, 2.0930995106027694e-06), (5, 3.752114480448654e-06), (6, 2.7355613383406308e-06), (7, 4.6844543248880655e-06), (8, 3.4412350942147896e-06), (9, 3.4963952657562913e-06), (10, 3.1077806852408685e-06), (11, 6.86199700794532e-06), (12, 2.5235027351300232e-05), (13, 0.001311835483647883), (14, 0.0020854324102401733), (15, 0.0009537767036817968), (16, 0.053929515182971954), (17, 0.00032388410181738436), (18, 0.0002506823802832514), (19, 0.004244728945195675), (20, 0.0032837754115462303), (21, 0.0004527226847130805), (22, 0.0006235238397493958), (23, 0.00023227289784699678), (24, 0.0004777133581228554), (25, 0.001055944594554603), (26, 0.0005974306259304285), (27, 0.024512076750397682)], Temporal MSE: [(0, 9.547991339786677e-07), (1, 5.661862587658106e-07), (2, 6.861657766421558e-07), (3, 8.508354767400306e-07), (4, 1.521768012935354e-06), (5, 1.799891606424353e-06), (6, 2.233355871794629e-06), (7, 2.18694617615256e-06), (8, 1.5321493265219033e-06), (9, 1.4119653997113346e-06), (10, 2.111247795255622e-06), (11, 1.0874027793761343e-05), (12, 3.070004459004849e-05), (13, 4.831708429264836e-05), (14, 5.530871567316353e-05), (15, 0.00011145061580464244), (16, 7.381409523077309e-05), (17, 9.50338871916756e-05), (18, 9.034985851030797e-05), (19, 6.929069058969617e-05), (20, 5.374324246076867e-05), (21, 0.00021245074458420277), (22, 0.00026853373856283724), (23, 0.00034585880348458886), (24, 0.00026944902492687106), (25, 0.0005282540805637836), (26, 0.00011511772754602134), (27, 0.00015459569112863392)]
+
+ 79%|███████▊  | 118/150 [17:40<04:28,  8.39s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 186, Spatial MSE: [(0, 8.724904319024063e-07), (1, 1.3740371286985464e-05), (2, 2.5294992155977525e-05), (3, 1.103541308111744e-05), (4, 1.6836127542774193e-05), (5, 4.814010753761977e-05), (6, 7.026328967185691e-05), (7, 0.00013908935943618417), (8, 0.00018265222024638206), (9, 0.00017662659229245037), (10, 0.00017366606334690005), (11, 0.00019106561376247555), (12, 0.000246958079515025), (13, 0.001990703633055091), (14, 0.004911153111606836), (15, 0.003520572790876031), (16, 0.09257166087627411), (17, 0.0007904146914370358), (18, 0.0007380858878605068), (19, 0.006243317853659391), (20, 0.005657590460032225), (21, 0.0013528884155675769), (22, 0.0017328145913779736), (23, 0.0021736477501690388), (24, 0.003716509323567152), (25, 0.007251529023051262), (26, 0.010658006183803082), (27, 0.023937825113534927)], Temporal MSE: [(0, 1.043745328388468e-06), (1, 5.845274131388578e-07), (2, 1.781517767085461e-06), (3, 8.309233635372948e-06), (4, 1.6642550690448843e-05), (5, 2.3115004296414554e-05), (6, 5.560504723689519e-05), (7, 5.7066943554673344e-05), (8, 3.453744648140855e-05), (9, 5.1403076213318855e-05), (10, 9.113332635024562e-05), (11, 0.00010706992907216772), (12, 0.0001325659832218662), (13, 0.00020215254335198551), (14, 0.0003301632823422551), (15, 0.000291028234641999), (16, 0.00029176133102737367), (17, 0.0003125646326225251), (18, 0.00031478304299525917), (19, 0.0004084724932909012), (20, 0.0006396766402758658), (21, 0.0007307401392608881), (22, 0.0007157392683438957), (23, 0.0010168551234528422), (24, 0.001949169090948999), (25, 0.00327192060649395), (26, 0.0010135051561519504), (27, 0.00022641237592324615)]
+
+ 79%|███████▉  | 119/150 [17:49<04:24,  8.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 180, Spatial MSE: [(0, 8.25219160560664e-07), (1, 1.476283523516031e-05), (2, 1.9601253370638005e-05), (3, 8.45072281663306e-06), (4, 2.7295509426039644e-06), (5, 4.145970251556719e-06), (6, 3.2021175684349146e-06), (7, 5.912494543736102e-06), (8, 4.588128831528593e-06), (9, 4.623324002750451e-06), (10, 1.4592515981348697e-05), (11, 8.930369222071022e-05), (12, 0.00015878089470788836), (13, 0.003198599210008979), (14, 0.0038803028874099255), (15, 0.0010592183098196983), (16, 0.056414417922496796), (17, 0.00033720798091962934), (18, 0.000328134949086234), (19, 0.004176880232989788), (20, 0.0030846083536744118), (21, 0.00040457971044816077), (22, 0.0005715083098039031), (23, 0.00025667244335636497), (24, 0.0005298678297549486), (25, 0.0011008117580786347), (26, 0.0007016812451183796), (27, 0.03550479933619499)], Temporal MSE: [(0, 1.019761043608014e-06), (1, 6.246341968108027e-07), (2, 7.513256150559755e-07), (3, 1.0201071063420386e-06), (4, 1.8811534800988738e-06), (5, 2.1947889763396233e-06), (6, 2.597473667265149e-06), (7, 2.6897162115346873e-06), (8, 1.9373983377590775e-06), (9, 2.06005302061385e-06), (10, 8.721448466531001e-06), (11, 0.0002038235543295741), (12, 8.235467248596251e-05), (13, 8.514961518812925e-05), (14, 5.975256135570817e-05), (15, 0.0001698695996310562), (16, 0.00011130284838145599), (17, 9.850879723671824e-05), (18, 8.276833250420168e-05), (19, 6.640171341132373e-05), (20, 5.362302908906713e-05), (21, 0.0001778447040123865), (22, 0.00021852788631804287), (23, 0.00034834345569834113), (24, 0.00028978195041418076), (25, 0.0007858805474825203), (26, 0.0001683861337369308), (27, 0.00018257928604725748)]
+
+ 80%|████████  | 120/150 [17:56<04:03,  8.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 174, Spatial MSE: [(0, 9.832167506829137e-07), (1, 1.300390522374073e-05), (2, 2.288474206579849e-05), (3, 7.099351932993159e-06), (4, 6.058219696569722e-06), (5, 1.4102685781836044e-05), (6, 2.2762755179428495e-05), (7, 2.0244482584530488e-05), (8, 2.5492887289146893e-05), (9, 4.4176158553455025e-05), (10, 8.317851461470127e-05), (11, 0.00018427096074447036), (12, 0.00023995191440917552), (13, 0.0032479404471814632), (14, 0.004496668931096792), (15, 0.0017989242915064096), (16, 0.06389454752206802), (17, 0.0004375996650196612), (18, 0.0003127710660919547), (19, 0.005563077982515097), (20, 0.004204081371426582), (21, 0.0007165353745222092), (22, 0.0007049533305689692), (23, 0.000529529876075685), (24, 0.0011594704119488597), (25, 0.0027859467081725597), (26, 0.004718652926385403), (27, 0.03481253609061241)], Temporal MSE: [(0, 1.0488666930541513e-06), (1, 6.465443220804445e-07), (2, 8.675746130393236e-07), (3, 1.6721381825846038e-06), (4, 4.56783845947939e-06), (5, 5.26801386513398e-06), (6, 8.640449777885806e-06), (7, 1.1113205800938886e-05), (8, 9.357907401863486e-06), (9, 1.0773970643640496e-05), (10, 2.0769501134054735e-05), (11, 0.00022665060532744974), (12, 9.963869524654001e-05), (13, 0.00011391522275516763), (14, 9.99889089143835e-05), (15, 0.0001738077844493091), (16, 0.00011932889901800081), (17, 0.000103053665952757), (18, 0.00011305939551675692), (19, 0.00011740965419448912), (20, 0.0001475914177717641), (21, 0.00023076018260326236), (22, 0.00026881485246121883), (23, 0.0004029747215099633), (24, 0.0004941168590448797), (25, 0.0011902791447937489), (26, 0.00043106055818498135), (27, 0.00020431393932085484)]
+
+ 81%|████████  | 121/150 [18:03<03:48,  7.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 168, Spatial MSE: [(0, 8.017310051400273e-07), (1, 1.4274050954554696e-05), (2, 1.9751219952013344e-05), (3, 7.906802238721866e-06), (4, 2.7389012302592164e-06), (5, 4.382090537546901e-06), (6, 3.555723424142343e-06), (7, 6.332542852760525e-06), (8, 4.796907887794077e-06), (9, 4.281445853848709e-06), (10, 1.0774712791317143e-05), (11, 3.541226760717109e-05), (12, 8.265947690233588e-05), (13, 0.0019498870242387056), (14, 0.0030130448285490274), (15, 0.0013461158378049731), (16, 0.06291808187961578), (17, 0.0004105844418518245), (18, 0.00014938716776669025), (19, 0.004924653563648462), (20, 0.0036241591442376375), (21, 0.0006890522199682891), (22, 0.0008203532197512686), (23, 0.00032029199064709246), (24, 0.0006038882420398295), (25, 0.0012139371829107404), (26, 0.0007250483613461256), (27, 0.032350093126297)], Temporal MSE: [(0, 1.175157535726612e-06), (1, 6.70516953960032e-07), (2, 8.260106483248819e-07), (3, 1.0254453854940948e-06), (4, 1.8943434270113357e-06), (5, 2.256569132441655e-06), (6, 2.83935673905944e-06), (7, 2.8245758585399017e-06), (8, 2.064230329779093e-06), (9, 2.1260796074784594e-06), (10, 7.028881555015687e-06), (11, 7.65220174798742e-05), (12, 5.378758578444831e-05), (13, 5.464547939482145e-05), (14, 7.131812890293077e-05), (15, 0.0001342389005003497), (16, 8.403074753005058e-05), (17, 9.587758540874347e-05), (18, 0.00010469029803061858), (19, 9.239171777153388e-05), (20, 0.00010674038639990613), (21, 0.00022696009546052665), (22, 0.00031152338488027453), (23, 0.00040632180753163993), (24, 0.00033226492814719677), (25, 0.0007343448814935982), (26, 0.0001587293518241495), (27, 0.0001793581759557128)]
+
+ 81%|████████▏ | 122/150 [18:10<03:35,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 162, Spatial MSE: [(0, 9.016442845677375e-07), (1, 1.5898314813966863e-05), (2, 3.377504981472157e-05), (3, 1.1154909770993982e-05), (4, 1.774900192685891e-05), (5, 5.2124796638963744e-05), (6, 6.901199230924249e-05), (7, 0.00015275180339813232), (8, 0.00019681265985127538), (9, 0.0001884358935058117), (10, 0.0001930355792865157), (11, 0.00023163903097156435), (12, 0.00035062007373198867), (13, 0.004557360894978046), (14, 0.009835564531385899), (15, 0.00553773483261466), (16, 0.14764292538166046), (17, 0.0009887972846627235), (18, 0.0008216102723963559), (19, 0.009505685418844223), (20, 0.008812961168587208), (21, 0.001826685038395226), (22, 0.0021076935809105635), (23, 0.0024150051176548004), (24, 0.004088141489773989), (25, 0.008192770183086395), (26, 0.012552020139992237), (27, 0.030433878302574158)], Temporal MSE: [(0, 1.1236028285566135e-06), (1, 7.439106184392585e-07), (2, 2.016533017012989e-06), (3, 6.11134919381584e-06), (4, 1.5775127394590527e-05), (5, 2.4724551622057334e-05), (6, 6.075615601730533e-05), (7, 6.359952385537326e-05), (8, 3.803426443482749e-05), (9, 5.5322485422948375e-05), (10, 0.00010567533172434196), (11, 0.00019406400679145008), (12, 0.0002237074513686821), (13, 0.0002909307077061385), (14, 0.00037632888415828347), (15, 0.00032675062539055943), (16, 0.00032897948403842747), (17, 0.00033655838342383504), (18, 0.00036014014040119946), (19, 0.000463450065581128), (20, 0.0007175299688242376), (21, 0.0008357645710930228), (22, 0.0007426576339639723), (23, 0.0011118537513539195), (24, 0.00204348168335855), (25, 0.003596668830141425), (26, 0.0012072002282366157), (27, 0.00025205122074112296)]
+
+ 82%|████████▏ | 123/150 [18:18<03:29,  7.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 156, Spatial MSE: [(0, 8.338496400028816e-07), (1, 1.7732003470882773e-05), (2, 2.4585266146459617e-05), (3, 9.029513421410229e-06), (4, 3.3169974358315812e-06), (5, 4.982127848052187e-06), (6, 3.866166025545681e-06), (7, 6.894701982673723e-06), (8, 6.249734269658802e-06), (9, 6.666916760877939e-06), (10, 2.166821104765404e-05), (11, 0.0001735104015097022), (12, 0.0003029444778803736), (13, 0.006303746718913317), (14, 0.006106548942625523), (15, 0.00191381829790771), (16, 0.08260466158390045), (17, 0.0003856273542623967), (18, 0.0005079720867797732), (19, 0.005662297829985619), (20, 0.003769712522625923), (21, 0.0005501387058757246), (22, 0.0009269708534702659), (23, 0.000427678955020383), (24, 0.0008060774416662753), (25, 0.001736725214868784), (26, 0.0010537689086049795), (27, 0.06039690971374512)], Temporal MSE: [(0, 1.1684746823448222e-06), (1, 6.916329766681883e-07), (2, 8.710249517207558e-07), (3, 1.119718490372179e-06), (4, 2.0900583876937162e-06), (5, 2.549528744566487e-06), (6, 3.2445532269775867e-06), (7, 3.141208480883506e-06), (8, 2.4881917397578945e-06), (9, 2.949255758721847e-06), (10, 1.2916788364236709e-05), (11, 0.00040818325942382216), (12, 0.00014802056830376387), (13, 0.00016093904559966177), (14, 0.00010303641465725377), (15, 0.00015140244795475155), (16, 0.00010236137313768268), (17, 0.00013931328430771828), (18, 0.0001478521153330803), (19, 0.00011070181062677875), (20, 9.279210644308478e-05), (21, 0.0002556882973294705), (22, 0.000319338112603873), (23, 0.0005365474498830736), (24, 0.0004897531471215189), (25, 0.0013587820576503873), (26, 0.0003002542653121054), (27, 0.0002924945147242397)]
+
+ 83%|████████▎ | 124/150 [18:26<03:21,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 150, Spatial MSE: [(0, 1.0744668088591425e-06), (1, 1.616897679923568e-05), (2, 3.280722739873454e-05), (3, 8.252239240391646e-06), (4, 8.153122507792432e-06), (5, 1.6113412129925564e-05), (6, 2.6600810087984428e-05), (7, 2.2894546418683603e-05), (8, 3.0032513677724637e-05), (9, 5.3863808716414496e-05), (10, 0.00010744718747446314), (11, 0.0002823879476636648), (12, 0.00040469429222866893), (13, 0.00693261856213212), (14, 0.00784982554614544), (15, 0.0033917815890163183), (16, 0.10236279666423798), (17, 0.0004960802616551518), (18, 0.0006356911617331207), (19, 0.006571296136826277), (20, 0.004944457206875086), (21, 0.0007427428499795496), (22, 0.0011118167312815785), (23, 0.000799595087300986), (24, 0.0016756418626755476), (25, 0.0038923723623156548), (26, 0.005582811776548624), (27, 0.059090107679367065)], Temporal MSE: [(0, 1.2798609532183036e-06), (1, 9.026418865687447e-07), (2, 1.0565065622358816e-06), (3, 2.017384531427524e-06), (4, 5.393630090111401e-06), (5, 6.267553544603288e-06), (6, 1.1125410310341977e-05), (7, 1.3718668014917057e-05), (8, 1.2265434634173289e-05), (9, 1.3393027074926067e-05), (10, 2.7828764359583147e-05), (11, 0.00043689357698895037), (12, 0.00018109817756339908), (13, 0.00020067482546437532), (14, 0.00014693487901240587), (15, 0.00018469874339643866), (16, 0.00011524049477884546), (17, 0.00014243702753446996), (18, 0.00017864622350316495), (19, 0.00016771897207945585), (20, 0.00022845837520435452), (21, 0.00033260139753110707), (22, 0.0003908588842023164), (23, 0.0006536893197335303), (24, 0.0008174414979293942), (25, 0.0019122683443129063), (26, 0.0006439473363570869), (27, 0.00032261613523587584)]
+
+ 83%|████████▎ | 125/150 [18:34<03:15,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 144, Spatial MSE: [(0, 8.139103897519817e-07), (1, 1.8193593859905377e-05), (2, 2.3276323190657422e-05), (3, 7.78762114350684e-06), (4, 2.9999803246028023e-06), (5, 5.021775905333925e-06), (6, 4.0691570575290825e-06), (7, 8.100470040517393e-06), (8, 6.170275810291059e-06), (9, 5.9328440329409204e-06), (10, 2.0336907255114056e-05), (11, 0.00010283262236043811), (12, 0.00020080046670045704), (13, 0.005280499346554279), (14, 0.005867228843271732), (15, 0.001975986873731017), (16, 0.10303907096385956), (17, 0.0005403951508924365), (18, 0.00035098157241009176), (19, 0.007848531939089298), (20, 0.005560672376304865), (21, 0.00036183767952024937), (22, 0.0009479206055402756), (23, 0.00043972156709060073), (24, 0.00090139324311167), (25, 0.002032800344750285), (26, 0.0012076195562258363), (27, 0.06090851128101349)], Temporal MSE: [(0, 1.230846010003006e-06), (1, 7.400064419016417e-07), (2, 8.888198408385506e-07), (3, 1.1081513093813555e-06), (4, 2.160167014153558e-06), (5, 2.5218910195690114e-06), (6, 3.4338406749157002e-06), (7, 3.2839948289620224e-06), (8, 2.5690271741041215e-06), (9, 2.747476628428558e-06), (10, 1.3640150427818298e-05), (11, 0.00024015417147893459), (12, 0.00012306577991694212), (13, 0.0001362007315037772), (14, 9.90228436421603e-05), (15, 0.00021847565949428827), (16, 0.00012290629092603922), (17, 0.00017371911962982267), (18, 0.00017952959751710296), (19, 0.00015570252435281873), (20, 0.00017842209490481764), (21, 0.00029913970502093434), (22, 0.0003862075973302126), (23, 0.0006116430158726871), (24, 0.0005175704718567431), (25, 0.0013472437858581543), (26, 0.000292621785774827), (27, 0.0003192818257957697)]
+
+ 84%|████████▍ | 126/150 [18:43<03:17,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 138, Spatial MSE: [(0, 1.0638574394761235e-06), (1, 1.7447755453758873e-05), (2, 3.957768058171496e-05), (3, 1.1315706615278032e-05), (4, 1.7656717318459414e-05), (5, 5.10895079059992e-05), (6, 6.837551336502656e-05), (7, 0.00015095235721673816), (8, 0.00018748930597212166), (9, 0.00018790514150168747), (10, 0.00018435883976053447), (11, 0.00027228909311816096), (12, 0.0004108811554033309), (13, 0.0060140653513371944), (14, 0.007300058379769325), (15, 0.0034139733761548996), (16, 0.11185133457183838), (17, 0.0010244062868878245), (18, 0.0012977898586541414), (19, 0.010927091352641582), (20, 0.008877338841557503), (21, 0.0014293456915766), (22, 0.0018636500462889671), (23, 0.0024329479783773422), (24, 0.0042597027495503426), (25, 0.009358705021440983), (26, 0.013262116350233555), (27, 0.06009594351053238)], Temporal MSE: [(0, 1.5768277989991475e-06), (1, 9.136682592725265e-07), (2, 2.1707562609663e-06), (3, 4.734249159810133e-06), (4, 1.5466055629076436e-05), (5, 2.4717406631680205e-05), (6, 7.045651000225917e-05), (7, 6.854285311419517e-05), (8, 3.968355667893775e-05), (9, 5.692263221135363e-05), (10, 0.00010069685959024355), (11, 0.0003346970770508051), (12, 0.00022532981529366225), (13, 0.00029448713758029044), (14, 0.0003380063863005489), (15, 0.00033481745049357414), (16, 0.00032530465978197753), (17, 0.0003685810952447355), (18, 0.0004019032057840377), (19, 0.0004920563660562038), (20, 0.0007070210413075984), (21, 0.0008917871746234596), (22, 0.000861443520989269), (23, 0.0012824665755033493), (24, 0.002225409960374236), (25, 0.004416018258780241), (26, 0.0015248009003698826), (27, 0.00042639716411940753)]
+
+ 85%|████████▍ | 127/150 [18:51<03:08,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 132, Spatial MSE: [(0, 9.369899203193199e-07), (1, 2.0004234102088958e-05), (2, 2.467594458721578e-05), (3, 8.768994121055584e-06), (4, 3.2179138997889822e-06), (5, 5.7735082918952685e-06), (6, 4.559065473586088e-06), (7, 8.992936272989027e-06), (8, 6.464459147537127e-06), (9, 6.227858193597058e-06), (10, 6.822936938988278e-06), (11, 2.4976938220788725e-05), (12, 9.31270988075994e-05), (13, 0.003350323997437954), (14, 0.0036314199678599834), (15, 0.0011919746175408363), (16, 0.04214806109666824), (17, 0.00022148992866277695), (18, 9.770850010681897e-05), (19, 0.0031909900717437267), (20, 0.001978888176381588), (21, 0.00030540346051566303), (22, 0.0004936134791933), (23, 0.0002979289856739342), (24, 0.0004966698470525444), (25, 0.0009367694146931171), (26, 0.0007864327635616064), (27, 0.04105091467499733)], Temporal MSE: [(0, 1.4536125263475697e-06), (1, 9.092332220461685e-07), (2, 1.052856646310829e-06), (3, 1.2843925105698872e-06), (4, 2.5891808945743833e-06), (5, 3.000224296556553e-06), (6, 4.607110440701945e-06), (7, 4.007601091871038e-06), (8, 3.0873534342390485e-06), (9, 2.7651144591800403e-06), (10, 4.6721738726773765e-06), (11, 4.9684913392411545e-05), (12, 7.698409899603575e-05), (13, 0.00010342462337575853), (14, 5.917523958487436e-05), (15, 0.00010515673056943342), (16, 6.536996806971729e-05), (17, 6.752295303158462e-05), (18, 9.321817924501374e-05), (19, 7.546639972133562e-05), (20, 5.363077798392624e-05), (21, 0.00011288719542790204), (22, 0.00015283975517377257), (23, 0.00028835367993451655), (24, 0.0003041480085812509), (25, 0.0010811211541295052), (26, 0.00023662533203605562), (27, 0.00019697126117534935)]
+
+ 85%|████████▌ | 128/150 [19:00<03:06,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 126, Spatial MSE: [(0, 1.1499630545586115e-06), (1, 1.8133330740965903e-05), (2, 4.103457831661217e-05), (3, 8.669578164699487e-06), (4, 8.757831892580725e-06), (5, 1.7709744497551583e-05), (6, 2.946956556115765e-05), (7, 2.5576424377504736e-05), (8, 3.069612284889445e-05), (9, 5.260860416456126e-05), (10, 9.378315735375509e-05), (11, 0.00014209613436833024), (12, 0.00017981830751523376), (13, 0.0035230545327067375), (14, 0.004351910203695297), (15, 0.001776573364622891), (16, 0.05388577654957771), (17, 0.0002981486904900521), (18, 0.00018170288240071386), (19, 0.003715689992532134), (20, 0.0025620530359447002), (21, 0.0004364142951089889), (22, 0.0006246975972317159), (23, 0.0006358122918754816), (24, 0.0013983084354549646), (25, 0.0030413924250751734), (26, 0.005187188740819693), (27, 0.0398748479783535)], Temporal MSE: [(0, 1.6037324712669943e-06), (1, 1.0741583764684037e-06), (2, 1.2097447097403347e-06), (3, 2.1855039449292235e-06), (4, 5.756592599936994e-06), (5, 7.114847903721966e-06), (6, 1.3162040886527393e-05), (7, 1.5250802789523732e-05), (8, 1.3622168808069546e-05), (9, 1.3729356396652292e-05), (10, 2.0249479348422028e-05), (11, 6.500929885078222e-05), (12, 9.083544864552096e-05), (13, 0.00013105443213135004), (14, 9.850854257820174e-05), (15, 0.00014791077410336584), (16, 8.673286356497556e-05), (17, 8.516832895111293e-05), (18, 0.00012450784561224282), (19, 0.00011937379895243794), (20, 0.00012094000703655183), (21, 0.00016594603948760778), (22, 0.00020775104349013418), (23, 0.00039223756175488234), (24, 0.0006496028508991003), (25, 0.001715353922918439), (26, 0.0006440086290240288), (27, 0.0002459292591083795)]
+
+ 86%|████████▌ | 129/150 [19:08<02:53,  8.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 120, Spatial MSE: [(0, 1.0553799256740604e-06), (1, 2.1058944184915163e-05), (2, 2.6825149689102545e-05), (3, 8.745597369852476e-06), (4, 3.3709829949657433e-06), (5, 6.502136784547474e-06), (6, 5.4249294407782145e-06), (7, 1.074438841897063e-05), (8, 7.571851710963529e-06), (9, 7.4055710683751386e-06), (10, 1.1904390703421086e-05), (11, 7.803789776517078e-05), (12, 0.00017298964667133987), (13, 0.00395237235352397), (14, 0.0041459957137703896), (15, 0.001593290944583714), (16, 0.06332821398973465), (17, 0.00034182152012363076), (18, 0.00040628694114275277), (19, 0.004552184604108334), (20, 0.0029902763199061155), (21, 0.0006553890416398644), (22, 0.000859356950968504), (23, 0.0003974649007432163), (24, 0.000730469822883606), (25, 0.0014043900882825255), (26, 0.0009694285690784454), (27, 0.04667872563004494)], Temporal MSE: [(0, 1.5834422129046288e-06), (1, 9.863873629001318e-07), (2, 1.170058453681122e-06), (3, 1.3503046147889108e-06), (4, 2.749706709437305e-06), (5, 3.2225586892309366e-06), (6, 5.226286248216638e-06), (7, 4.551820893539116e-06), (8, 3.5563591609388823e-06), (9, 3.316474021630711e-06), (10, 8.255454304162413e-06), (11, 0.00021750498854089528), (12, 0.00010428762470837682), (13, 0.00011302872735541314), (14, 8.872905891621485e-05), (15, 0.00014228995132725686), (16, 8.207155769923702e-05), (17, 9.385651355842128e-05), (18, 0.00012961341417394578), (19, 0.00010037066385848448), (20, 7.837375596864149e-05), (21, 0.0002334720775252208), (22, 0.0003154475416522473), (23, 0.00047166363219730556), (24, 0.0004349541268311441), (25, 0.0013167575234547257), (26, 0.0002752696454990655), (27, 0.0002635020646266639)]
+
+ 87%|████████▋ | 130/150 [19:16<02:44,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 114, Spatial MSE: [(0, 1.181341644951317e-06), (1, 2.040044273599051e-05), (2, 5.01193935633637e-05), (3, 1.2532633263617754e-05), (4, 1.936273656610865e-05), (5, 5.350580977392383e-05), (6, 7.437867316184565e-05), (7, 0.00017690622189547867), (8, 0.000213065737625584), (9, 0.00019580181105993688), (10, 0.00017630099318921566), (11, 0.00025695713702589273), (12, 0.0004302361630834639), (13, 0.005699187982827425), (14, 0.009151489473879337), (15, 0.004674247466027737), (16, 0.14116977155208588), (17, 0.0008579415734857321), (18, 0.0009940923191606998), (19, 0.009320671670138836), (20, 0.008005201816558838), (21, 0.001762120402418077), (22, 0.002002800116315484), (23, 0.002517316723242402), (24, 0.004411071073263884), (25, 0.009311733767390251), (26, 0.015365714207291603), (27, 0.04676287621259689)], Temporal MSE: [(0, 1.8803120838128962e-06), (1, 1.1293312809357303e-06), (2, 2.659030087670544e-06), (3, 5.128074462845689e-06), (4, 1.6461577615700662e-05), (5, 2.7155865609529428e-05), (6, 9.712867176858708e-05), (7, 8.03084549261257e-05), (8, 4.6785837184870616e-05), (9, 6.296484207268804e-05), (10, 0.00010094395838677883), (11, 0.0003383884613867849), (12, 0.00026662845630198717), (13, 0.00031149524147622287), (14, 0.0003590002015698701), (15, 0.00033179973252117634), (16, 0.00030808919109404087), (17, 0.0003344822907820344), (18, 0.0003835654933936894), (19, 0.00046504815691150725), (20, 0.0006931610405445099), (21, 0.0008501710253767669), (22, 0.0008514230139553547), (23, 0.001304654753766954), (24, 0.002279350534081459), (25, 0.004334596451371908), (26, 0.0016658754320815206), (27, 0.00040635792538523674)]
+
+ 87%|████████▋ | 131/150 [19:24<02:35,  8.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 108, Spatial MSE: [(0, 1.127302198256075e-06), (1, 2.4826225853757933e-05), (2, 3.461951928329654e-05), (3, 1.1163982890138868e-05), (4, 4.121704478166066e-06), (5, 9.476414561504498e-06), (6, 6.588069481949788e-06), (7, 1.578217779751867e-05), (8, 1.0749895409389865e-05), (9, 6.3356305872730445e-06), (10, 8.95081120688701e-06), (11, 2.590341682662256e-05), (12, 0.00010590020247036591), (13, 0.003355116583406925), (14, 0.004829947836697102), (15, 0.001589540857821703), (16, 0.09877396374940872), (17, 0.0004574659396894276), (18, 0.0004136758507229388), (19, 0.007318445481359959), (20, 0.005381251219660044), (21, 0.0003431685909163207), (22, 0.0008148396736942232), (23, 0.0004204325668979436), (24, 0.0008291950798593462), (25, 0.001885864301584661), (26, 0.0011518503306433558), (27, 0.05370760336518288)], Temporal MSE: [(0, 1.8307190430277842e-06), (1, 1.1702968549798243e-06), (2, 1.4114135638010339e-06), (3, 1.625055006115872e-06), (4, 3.4073734695994062e-06), (5, 4.124920451431535e-06), (6, 6.1193431974970736e-06), (7, 5.892918125027791e-06), (8, 4.613598775904393e-06), (9, 4.0979994082590565e-06), (10, 6.221522198757157e-06), (11, 5.132093792781234e-05), (12, 9.261220111511648e-05), (13, 0.00011226713832002133), (14, 0.00010981447121594101), (15, 0.00022082576469983906), (16, 0.00015209842240437865), (17, 0.00020843296078965068), (18, 0.00015891533985268325), (19, 0.00013528420822694898), (20, 0.00010152103641303256), (21, 0.00031274856883101165), (22, 0.0003589980478864163), (23, 0.0005754639278165996), (24, 0.000473205087473616), (25, 0.0013731620274484158), (26, 0.0002991412184201181), (27, 0.00032752499100752175)]
+
+ 88%|████████▊ | 132/150 [19:32<02:24,  8.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 102, Spatial MSE: [(0, 1.2878664392701467e-06), (1, 2.395437695668079e-05), (2, 5.719765613321215e-05), (3, 1.0645681868481915e-05), (4, 1.3552529708249494e-05), (5, 2.7410336770117283e-05), (6, 4.430678018252365e-05), (7, 7.034468580968678e-05), (8, 9.05272172531113e-05), (9, 0.00012003695883322507), (10, 0.00011451588943600655), (11, 0.00014384780661202967), (12, 0.0002260868641315028), (13, 0.0038864261005073786), (14, 0.006569520104676485), (15, 0.003455621423199773), (16, 0.1010165587067604), (17, 0.0007015707087703049), (18, 0.0006732147303409874), (19, 0.009434507228434086), (20, 0.007818385027348995), (21, 0.000760201015509665), (22, 0.0012956493301317096), (23, 0.0012947790091857314), (24, 0.00243568723089993), (25, 0.006094742100685835), (26, 0.009814088232815266), (27, 0.05329365283250809)], Temporal MSE: [(0, 1.9063984382228227e-06), (1, 1.2231184882693924e-06), (2, 1.7896173858389375e-06), (3, 3.226077069484745e-06), (4, 6.229975497262785e-06), (5, 8.185540536942426e-06), (6, 4.199831892037764e-05), (7, 3.5175238735973835e-05), (8, 2.2011388864484616e-05), (9, 2.6394400265417062e-05), (10, 3.6004661524202675e-05), (11, 8.646795322420076e-05), (12, 0.0001341824245173484), (13, 0.00017398734053131193), (14, 0.00020334192959126085), (15, 0.0002541928261052817), (16, 0.00022064980294089764), (17, 0.0002625347697176039), (18, 0.0002480120165273547), (19, 0.0002645360946189612), (20, 0.00032592154457233846), (21, 0.00047708034981042147), (22, 0.0004927090485580266), (23, 0.0007675840170122683), (24, 0.0011038025841116905), (25, 0.0026379446499049664), (26, 0.0007684140000492334), (27, 0.0003662305825855583)]
+
+ 89%|████████▊ | 133/150 [19:42<02:25,  8.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 96, Spatial MSE: [(0, 1.1451547834440134e-06), (1, 2.935228985734284e-05), (2, 4.47964011982549e-05), (3, 7.213804110506317e-06), (4, 6.64456638332922e-06), (5, 1.412728124705609e-05), (6, 1.934037936734967e-05), (7, 3.653132444014773e-05), (8, 3.9181773900054395e-05), (9, 3.905406265403144e-05), (10, 5.192423486732878e-05), (11, 6.427962944144383e-05), (12, 9.838951518759131e-05), (13, 0.0019495420856401324), (14, 0.006882296875119209), (15, 0.004194230772554874), (16, 0.12196481227874756), (17, 0.00019650120520964265), (18, 0.0003183325461577624), (19, 0.010732068680226803), (20, 0.00998334214091301), (21, 0.0003392273501958698), (22, 0.0003276787174399942), (23, 0.0005502591375261545), (24, 0.0009321902762167156), (25, 0.001972532831132412), (26, 0.003648331854492426), (27, 0.002953195944428444)], Temporal MSE: [(0, 1.972714244402596e-06), (1, 1.3284019360071397e-06), (2, 1.7007373571686912e-06), (3, 2.1418597953015706e-06), (4, 3.7378963497758377e-06), (5, 3.671574631880503e-06), (6, 2.4585029677837156e-05), (7, 1.6669242540956475e-05), (8, 9.786057489691302e-06), (9, 1.0494733032828663e-05), (10, 1.7410877262591384e-05), (11, 4.0679529774934053e-05), (12, 8.467840234516189e-05), (13, 9.360707190353423e-05), (14, 0.00010933383600786328), (15, 9.837661491474137e-05), (16, 6.368939648382366e-05), (17, 6.610525451833382e-05), (18, 0.00010562961688265204), (19, 0.00013376906281337142), (20, 0.0001579783856868744), (21, 0.00023034648620523512), (22, 0.00020002534438390285), (23, 0.00026716437423601747), (24, 0.0004062852240167558), (25, 0.000677251722663641), (26, 0.00021558445587288588), (27, 7.392688712570816e-05)]
+
+ 89%|████████▉ | 134/150 [19:51<02:19,  8.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 90, Spatial MSE: [(0, 1.2162132634330192e-06), (1, 2.8253380150999874e-05), (2, 4.921040090266615e-05), (3, 7.069062576192664e-06), (4, 6.4019668570836075e-06), (5, 1.3304950698511675e-05), (6, 1.847377461672295e-05), (7, 3.438437852310017e-05), (8, 3.639266287791543e-05), (9, 4.132583853788674e-05), (10, 4.40465119027067e-05), (11, 5.830083682667464e-05), (12, 7.930610445328057e-05), (13, 0.001239200122654438), (14, 0.004578513093292713), (15, 0.002627202309668064), (16, 0.07334685325622559), (17, 0.0001400560577167198), (18, 0.00021015752281527966), (19, 0.00650978134945035), (20, 0.006014582701027393), (21, 0.0002681214245967567), (22, 0.00023962778504937887), (23, 0.00043152814032509923), (24, 0.0007747557829134166), (25, 0.001659685280174017), (26, 0.0030975339468568563), (27, 0.00318190292455256)], Temporal MSE: [(0, 2.212274011981208e-06), (1, 1.4158555359244929e-06), (2, 1.743755092320498e-06), (3, 1.9439764855633257e-06), (4, 3.403649543542997e-06), (5, 3.4640943340491503e-06), (6, 2.4837492674123496e-05), (7, 1.5818417523405515e-05), (8, 1.0272236977471039e-05), (9, 1.0848949386854656e-05), (10, 1.40226366056595e-05), (11, 3.398418630240485e-05), (12, 6.543788913404569e-05), (13, 7.046243990771472e-05), (14, 7.473649020539597e-05), (15, 7.336794806178659e-05), (16, 5.007098661735654e-05), (17, 5.070679253549315e-05), (18, 7.591721805511042e-05), (19, 9.290945308748633e-05), (20, 0.000117591604066547), (21, 0.00015369284665212035), (22, 0.00014301571354735643), (23, 0.00019563235400710255), (24, 0.0003246756677981466), (25, 0.0005544720916077495), (26, 0.0001712304074317217), (27, 7.51143234083429e-05)]
+
+ 90%|█████████ | 135/150 [20:00<02:13,  8.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 84, Spatial MSE: [(0, 1.296078835366643e-06), (1, 2.9429647838696837e-05), (2, 4.6843229938531294e-05), (3, 7.534276392107131e-06), (4, 6.6736397457134444e-06), (5, 1.3852747542841826e-05), (6, 2.106925967382267e-05), (7, 3.7710651668021455e-05), (8, 3.817756805801764e-05), (9, 4.603085108101368e-05), (10, 3.6286826798459515e-05), (11, 4.916826219414361e-05), (12, 6.10702481935732e-05), (13, 0.001302300370298326), (14, 0.004780123010277748), (15, 0.0026213699020445347), (16, 0.07362033426761627), (17, 0.00015560645260848105), (18, 0.00022668796009384096), (19, 0.006464662961661816), (20, 0.00592593802139163), (21, 0.0002544360759202391), (22, 0.0002496219822205603), (23, 0.0004761047021020204), (24, 0.000894378696102649), (25, 0.001792397815734148), (26, 0.003283916739746928), (27, 0.0034547881223261356)], Temporal MSE: [(0, 2.3489810700993985e-06), (1, 1.5473590337933274e-06), (2, 1.9127867290080758e-06), (3, 2.4785624646028737e-06), (4, 3.844428192678606e-06), (5, 3.957093667850131e-06), (6, 2.960210622404702e-05), (7, 1.7604599634069018e-05), (8, 1.1759042536141351e-05), (9, 1.1243790140724741e-05), (10, 1.2297487955947872e-05), (11, 2.2560407160199247e-05), (12, 5.3483385272556916e-05), (13, 6.644216045970097e-05), (14, 9.1016867372673e-05), (15, 6.818392284912989e-05), (16, 5.2379804401425645e-05), (17, 4.5550459617516026e-05), (18, 7.65631630201824e-05), (19, 9.617865725886077e-05), (20, 0.00012038527347613126), (21, 0.0001744947803672403), (22, 0.00015080708544701338), (23, 0.0002173547400161624), (24, 0.00037777595571242273), (25, 0.000620982376858592), (26, 0.0002076536911772564), (27, 8.813077147351578e-05)]
+
+ 91%|█████████ | 136/150 [20:09<02:03,  8.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 78, Spatial MSE: [(0, 1.3009897656957037e-06), (1, 3.1707342714071274e-05), (2, 5.4839914810145274e-05), (3, 7.709689270996023e-06), (4, 7.1436597863794304e-06), (5, 1.4264324818213936e-05), (6, 2.100779238389805e-05), (7, 4.344479748397134e-05), (8, 4.3628439016174525e-05), (9, 4.35472211393062e-05), (10, 4.216008892399259e-05), (11, 4.997679207008332e-05), (12, 4.5803240936947986e-05), (13, 8.238675218308344e-05), (14, 0.00017033520271070302), (15, 0.00010500646021682769), (16, 7.992669998202473e-05), (17, 6.96437491569668e-05), (18, 6.812492938479409e-05), (19, 9.138470340985805e-05), (20, 0.00011506015289342031), (21, 0.00015511181845795363), (22, 0.00016118896019179374), (23, 0.00036704656668007374), (24, 0.0007347836508415639), (25, 0.0015038397395983338), (26, 0.0031728248577564955), (27, 0.0034077647142112255)], Temporal MSE: [(0, 2.3954814878379693e-06), (1, 1.5869139815549715e-06), (2, 2.0026711808895925e-06), (3, 2.36559253607993e-06), (4, 3.862944595312001e-06), (5, 3.855337581626372e-06), (6, 3.399762499611825e-05), (7, 1.901489304145798e-05), (8, 1.2226239960000385e-05), (9, 1.1811765034508426e-05), (10, 1.215953307109885e-05), (11, 1.6604943084530532e-05), (12, 1.4036909306014422e-05), (13, 2.2158446881803684e-05), (14, 3.3057527616620064e-05), (15, 3.213615855202079e-05), (16, 2.918232712545432e-05), (17, 2.595261139504146e-05), (18, 3.412646765355021e-05), (19, 4.3040428863605484e-05), (20, 6.975916039664298e-05), (21, 6.532311817863956e-05), (22, 7.272382936207578e-05), (23, 0.00012308964505791664), (24, 0.00027922302251681685), (25, 0.00047858283505775034), (26, 0.0001837326999520883), (27, 9.225008398061618e-05)]
+
+ 91%|█████████▏| 137/150 [20:17<01:52,  8.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 72, Spatial MSE: [(0, 1.3859391856385628e-06), (1, 3.361807466717437e-05), (2, 5.512940333574079e-05), (3, 8.122450708469842e-06), (4, 7.731689038337208e-06), (5, 1.5028874258860014e-05), (6, 2.2396001440938562e-05), (7, 4.235456435708329e-05), (8, 4.574807462631725e-05), (9, 4.960082515026443e-05), (10, 4.129655644646846e-05), (11, 4.740963413496502e-05), (12, 4.0557675674790516e-05), (13, 0.000249317818088457), (14, 0.0008988805348053575), (15, 0.0007372592226602137), (16, 0.01848331093788147), (17, 0.00010180706158280373), (18, 0.00010771022061817348), (19, 0.0016481158090755343), (20, 0.0015327346045523882), (21, 0.000163592048920691), (22, 0.00019944192899856716), (23, 0.00042418859084136784), (24, 0.0008663646876811981), (25, 0.0017811496509239078), (26, 0.0035839173942804337), (27, 0.003968364093452692)], Temporal MSE: [(0, 2.528235427234904e-06), (1, 1.7001692640405963e-06), (2, 2.1193743577896385e-06), (3, 2.76325727099902e-06), (4, 4.266789346729638e-06), (5, 4.311089014663594e-06), (6, 3.829283014056273e-05), (7, 1.9667228116304614e-05), (8, 1.391795558447484e-05), (9, 1.3160473827156238e-05), (10, 1.3662592209584545e-05), (11, 1.8383434507995844e-05), (12, 1.500292455602903e-05), (13, 2.8605727493413724e-05), (14, 4.468308907235041e-05), (15, 4.233899380778894e-05), (16, 3.738484519999474e-05), (17, 3.2818454201333225e-05), (18, 4.4812790292780846e-05), (19, 5.915342262596823e-05), (20, 9.283333201892674e-05), (21, 9.521587344352156e-05), (22, 0.00010403815394965932), (23, 0.00016439140017610043), (24, 0.0003384680312592536), (25, 0.000569735246244818), (26, 0.00020953576313331723), (27, 0.00011704945791279897)]
+
+ 92%|█████████▏| 138/150 [20:25<01:41,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 66, Spatial MSE: [(0, 1.620813577574154e-06), (1, 3.8390553527278826e-05), (2, 6.140518962638453e-05), (3, 9.159517503576353e-06), (4, 8.649710252939258e-06), (5, 1.735452315188013e-05), (6, 2.8812761229346506e-05), (7, 5.001856334274635e-05), (8, 5.6223358114948496e-05), (9, 6.0030863096471876e-05), (10, 5.518869511433877e-05), (11, 6.672680319752544e-05), (12, 5.416147905634716e-05), (13, 0.00043276004726067185), (14, 0.0020095871295779943), (15, 0.001703194691799581), (16, 0.04276302084326744), (17, 0.00017132970970124006), (18, 0.0001718064013402909), (19, 0.003617828246206045), (20, 0.003516968572512269), (21, 0.000262932269833982), (22, 0.00026495379279367626), (23, 0.0005428603035397828), (24, 0.0010661237174645066), (25, 0.002158631570637226), (26, 0.0041723246686160564), (27, 0.004375176038593054)], Temporal MSE: [(0, 3.1773097362020053e-06), (1, 2.1162215944059426e-06), (2, 2.6874402010435006e-06), (3, 3.4002225675067166e-06), (4, 5.053297172707971e-06), (5, 5.288550255500013e-06), (6, 4.602387343766168e-05), (7, 2.2875268768984824e-05), (8, 1.7635073163546622e-05), (9, 1.545765917398967e-05), (10, 1.5948880900396034e-05), (11, 2.124756429111585e-05), (12, 1.8872618966270238e-05), (13, 3.7641788367182016e-05), (14, 6.245056283660233e-05), (15, 6.163866783026606e-05), (16, 4.909412018605508e-05), (17, 4.600101601681672e-05), (18, 7.122281385818496e-05), (19, 8.307265670737252e-05), (20, 0.00011848676513182), (21, 0.00013368848885875195), (22, 0.0001416549930581823), (23, 0.0002268494135932997), (24, 0.00043344072764739394), (25, 0.0007559171644970775), (26, 0.000276024715276435), (27, 0.00013555552868638188)]
+
+ 93%|█████████▎| 139/150 [20:33<01:32,  8.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 60, Spatial MSE: [(0, 1.76672904217412e-06), (1, 4.190349136479199e-05), (2, 6.808123725932091e-05), (3, 1.0113722964888439e-05), (4, 9.4767747214064e-06), (5, 1.8393022401141934e-05), (6, 3.105805444647558e-05), (7, 5.7473949709674343e-05), (8, 6.271084566833451e-05), (9, 6.269149162108079e-05), (10, 6.3234860135708e-05), (11, 7.032146822893992e-05), (12, 5.917237649555318e-05), (13, 0.0005576418479904532), (14, 0.002117766998708248), (15, 0.0015168602112680674), (16, 0.037099458277225494), (17, 0.00014916523650754243), (18, 0.00016519104246981442), (19, 0.0031242703553289175), (20, 0.0029418356716632843), (21, 0.00026653677923604846), (22, 0.0002714818110689521), (23, 0.0005721811903640628), (24, 0.0011762265348806977), (25, 0.002297096885740757), (26, 0.004555601626634598), (27, 0.004727532155811787)], Temporal MSE: [(0, 3.340757530168048e-06), (1, 2.297403852935531e-06), (2, 3.044778168259654e-06), (3, 3.6393050777405733e-06), (4, 5.429613338492345e-06), (5, 5.890180091228103e-06), (6, 5.4579799325438216e-05), (7, 2.6011639420175925e-05), (8, 2.0610652427421883e-05), (9, 1.746791895129718e-05), (10, 1.85246390174143e-05), (11, 2.7103851607535034e-05), (12, 2.6845382308238186e-05), (13, 4.6713299525436014e-05), (14, 7.44224525988102e-05), (15, 6.907017814228311e-05), (16, 5.482805136125535e-05), (17, 4.8776048060972244e-05), (18, 6.829106132499874e-05), (19, 8.813200838631019e-05), (20, 0.00012190923735033721), (21, 0.00013601376849692315), (22, 0.00014734955038875341), (23, 0.00023692379181738943), (24, 0.0004592213954310864), (25, 0.0008409630972892046), (26, 0.0003069029771722853), (27, 0.00016678508836776018)]
+
+ 93%|█████████▎| 140/150 [20:41<01:22,  8.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 54, Spatial MSE: [(0, 1.8910953940576292e-06), (1, 4.628723763744347e-05), (2, 7.571531750727445e-05), (3, 1.0589665180305019e-05), (4, 1.0360603482695296e-05), (5, 2.0579695046762936e-05), (6, 3.470913725323044e-05), (7, 6.311321340035647e-05), (8, 7.12140099494718e-05), (9, 7.579156954307109e-05), (10, 6.927893264219165e-05), (11, 8.592280209995806e-05), (12, 5.846520434715785e-05), (13, 0.000321886851452291), (14, 0.0015376571100205183), (15, 0.001419011619873345), (16, 0.03557901456952095), (17, 0.0001598674280103296), (18, 0.0001930029975483194), (19, 0.0031542957294732332), (20, 0.003076615510508418), (21, 0.0003195005701854825), (22, 0.00028779543936252594), (23, 0.0006080954917706549), (24, 0.0012370703043416142), (25, 0.0023464264813810587), (26, 0.004461871460080147), (27, 0.0050697061233222485)], Temporal MSE: [(0, 3.5708985706150997e-06), (1, 2.5131364509434206e-06), (2, 3.340339844726259e-06), (3, 3.899933290085755e-06), (4, 6.0168154050188605e-06), (5, 6.519277121697087e-06), (6, 6.61710073472932e-05), (7, 2.9082870241836645e-05), (8, 2.3679405785514973e-05), (9, 2.0482713807723485e-05), (10, 2.0354162188596092e-05), (11, 2.8177621061331593e-05), (12, 2.1420741177280433e-05), (13, 3.8889451388968155e-05), (14, 6.96380011504516e-05), (15, 6.668073910987005e-05), (16, 5.812861490994692e-05), (17, 4.767873542732559e-05), (18, 7.076052133925259e-05), (19, 8.958266698755324e-05), (20, 0.00012868335761595517), (21, 0.00014863880642224103), (22, 0.00014997921243775636), (23, 0.00024038545961957425), (24, 0.00048619863810017705), (25, 0.000848251162096858), (26, 0.0003407397889532149), (27, 0.00021282077068462968)]
+
+ 94%|█████████▍| 141/150 [20:50<01:14,  8.32s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 48, Spatial MSE: [(0, 2.105366320392932e-06), (1, 5.087874524178915e-05), (2, 8.354699093615636e-05), (3, 1.1516287486301735e-05), (4, 1.1411888408474624e-05), (5, 2.1447627659654245e-05), (6, 3.8171034248080105e-05), (7, 6.926976493559778e-05), (8, 8.357239858014509e-05), (9, 8.352322765858844e-05), (10, 8.318602340295911e-05), (11, 8.640504529466853e-05), (12, 6.819945701863617e-05), (13, 0.0007710151840001345), (14, 0.003023539436981082), (15, 0.002355844946578145), (16, 0.06142530217766762), (17, 0.00020615477114915848), (18, 0.00024116555869113654), (19, 0.005385986529290676), (20, 0.005328675732016563), (21, 0.0003448743955232203), (22, 0.0003647434350568801), (23, 0.0006798411486670375), (24, 0.0013394304551184177), (25, 0.0023725025821477175), (26, 0.004540731664747), (27, 0.005194695666432381)], Temporal MSE: [(0, 3.9322871998592746e-06), (1, 2.7644434794638073e-06), (2, 3.7659988265659194e-06), (3, 4.198193892079871e-06), (4, 6.438912350859027e-06), (5, 7.225326953630429e-06), (6, 7.813485717633739e-05), (7, 3.127133459202014e-05), (8, 2.748512451944407e-05), (9, 2.1575326172751375e-05), (10, 2.2192056349013e-05), (11, 3.097566150245257e-05), (12, 3.094185740337707e-05), (13, 4.992993126506917e-05), (14, 8.558826812077314e-05), (15, 8.914322825148702e-05), (16, 6.818542897235602e-05), (17, 5.8975729189114645e-05), (18, 8.886998693924397e-05), (19, 0.00011382906814105809), (20, 0.0001506355474703014), (21, 0.0001782915642252192), (22, 0.0001825213839765638), (23, 0.00028487257077358663), (24, 0.0005279958713799715), (25, 0.0009035486727952957), (26, 0.00040316185913980007), (27, 0.0002716111484915018)]
+
+ 95%|█████████▍| 142/150 [20:57<01:04,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 42, Spatial MSE: [(0, 2.5087201720452867e-06), (1, 6.085642598918639e-05), (2, 0.00010021268099080771), (3, 1.3788288924843073e-05), (4, 1.3531574950320646e-05), (5, 2.5638561055529863e-05), (6, 5.0091617595171556e-05), (7, 8.479835378238931e-05), (8, 0.0001058262714650482), (9, 0.00010424192441860214), (10, 0.00010791318345582113), (11, 0.00012356186925899237), (12, 7.762122550047934e-05), (13, 0.00021839438704773784), (14, 0.0005171690718270838), (15, 0.00041599819087423384), (16, 0.00587097043171525), (17, 0.00018776155775412917), (18, 0.00017831902368925512), (19, 0.0007203143322840333), (20, 0.0007494231103919446), (21, 0.0003734935016836971), (22, 0.0003380127309355885), (23, 0.0007588330772705376), (24, 0.001681828172877431), (25, 0.00286045647226274), (26, 0.005804404150694609), (27, 0.006353964563459158)], Temporal MSE: [(0, 4.782899850397371e-06), (1, 3.394063242012635e-06), (2, 4.697994882008061e-06), (3, 5.290622539178003e-06), (4, 7.82421284384327e-06), (5, 9.007175322039984e-06), (6, 9.31924587348476e-05), (7, 3.9904258301248774e-05), (8, 3.696926069096662e-05), (9, 2.8236230718903244e-05), (10, 2.8819527869927697e-05), (11, 4.031884964206256e-05), (12, 3.0201019399100915e-05), (13, 4.353658005129546e-05), (14, 7.47621088521555e-05), (15, 7.559840014437214e-05), (16, 6.731186294928193e-05), (17, 5.1018672820646316e-05), (18, 6.797846435802057e-05), (19, 9.18166188057512e-05), (20, 0.00014418021601159126), (21, 0.00013681837299373), (22, 0.0001611283078091219), (23, 0.0002863719710148871), (24, 0.0006183960358612239), (25, 0.0011354020098224282), (26, 0.0005422953981906176), (27, 0.00045960713759995997)]
+
+ 95%|█████████▌| 143/150 [21:05<00:55,  7.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 36, Spatial MSE: [(0, 2.981000079671503e-06), (1, 7.083011587383226e-05), (2, 0.00011427405843278393), (3, 1.5774638086440973e-05), (4, 1.5906402040855028e-05), (5, 3.0058801712584682e-05), (6, 5.9739268181147054e-05), (7, 0.00010276131797581911), (8, 0.00013640883844345808), (9, 0.00013071026478428394), (10, 0.000145579528179951), (11, 0.00016382506873924285), (12, 0.00012212031288072467), (13, 0.0018182523781433702), (14, 0.00787513330578804), (15, 0.005595942959189415), (16, 0.14663736522197723), (17, 0.00039979108260013163), (18, 0.0004726984479930252), (19, 0.012306718155741692), (20, 0.011947465129196644), (21, 0.0006695842021144927), (22, 0.0006486166967079043), (23, 0.0011480399407446384), (24, 0.0022841929458081722), (25, 0.003960697446018457), (26, 0.007434749975800514), (27, 0.007875220850110054)], Temporal MSE: [(0, 5.5024802350089885e-06), (1, 4.1014682210516185e-06), (2, 5.785162557003787e-06), (3, 6.103834493842442e-06), (4, 9.288151886721607e-06), (5, 1.1046224244637415e-05), (6, 0.00012988307571504265), (7, 4.639316102839075e-05), (8, 4.6654804464196786e-05), (9, 3.501602986943908e-05), (10, 3.6399407690623775e-05), (11, 6.141402263892815e-05), (12, 7.495628960896283e-05), (13, 9.975762804970145e-05), (14, 0.000154927940457128), (15, 0.00016841977776493877), (16, 0.0001205125663545914), (17, 0.00011240011372137815), (18, 0.00016640259127598256), (19, 0.00021326336718630046), (20, 0.0002772202715277672), (21, 0.00035751424729824066), (22, 0.00033446590532548726), (23, 0.0005120645510032773), (24, 0.0009182513458654284), (25, 0.0016524794045835733), (26, 0.0007940607611089945), (27, 0.0008152571390382946)]
+
+ 96%|█████████▌| 144/150 [21:13<00:47,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 30, Spatial MSE: [(0, 3.478949338386883e-06), (1, 8.120342681650072e-05), (2, 0.0001248344633495435), (3, 1.8321679817745462e-05), (4, 1.8488824935047887e-05), (5, 3.2954252674244344e-05), (6, 6.947150541236624e-05), (7, 0.00011561640712898225), (8, 0.0001578007941134274), (9, 0.00014844634279143065), (10, 0.00017170075443573296), (11, 0.00017932373157236725), (12, 0.0001298291317652911), (13, 0.0015913626411929727), (14, 0.00610722228884697), (15, 0.004377438686788082), (16, 0.11156141757965088), (17, 0.00040597273618914187), (18, 0.00046259944792836905), (19, 0.009573051705956459), (20, 0.009461658075451851), (21, 0.0006530381506308913), (22, 0.0006603148649446666), (23, 0.001268862746655941), (24, 0.0026693399995565414), (25, 0.004412819165736437), (26, 0.008671410381793976), (27, 0.009478811174631119)], Temporal MSE: [(0, 6.678550562355667e-06), (1, 4.918007107335143e-06), (2, 6.9884331423963886e-06), (3, 7.223049578897189e-06), (4, 1.0694868251448497e-05), (5, 1.3493022379407194e-05), (6, 0.00015080385492183268), (7, 5.372087252908386e-05), (8, 5.87991853535641e-05), (9, 4.212563726468943e-05), (10, 4.3976691813440993e-05), (11, 7.32736152713187e-05), (12, 7.401572656817734e-05), (13, 9.839852282311767e-05), (14, 0.00015295595221687108), (15, 0.0001710807700874284), (16, 0.00012120042083552107), (17, 0.00010651953925844282), (18, 0.00015508878277614713), (19, 0.00020661002781707793), (20, 0.00027846082230098546), (21, 0.00033167543006129563), (22, 0.00033782317768782377), (23, 0.0005587420309893787), (24, 0.0010401081526651978), (25, 0.001891141408123076), (26, 0.0010694322409108281), (27, 0.0016377029241994023)]
+
+ 97%|█████████▋| 145/150 [21:20<00:39,  7.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 24, Spatial MSE: [(0, 4.387842636788264e-06), (1, 9.617955947760493e-05), (2, 0.00014690407260786742), (3, 2.2163972971611656e-05), (4, 2.304742520209402e-05), (5, 3.878678762703203e-05), (6, 9.210083226207644e-05), (7, 0.00012200883065816015), (8, 0.0001767228968674317), (9, 0.00019980507204309106), (10, 0.0002134301175829023), (11, 0.0002389810251770541), (12, 0.00012605913798324764), (13, 0.0013298098929226398), (14, 0.005321484059095383), (15, 0.004415615927428007), (16, 0.11172781884670258), (17, 0.0004984692204743624), (18, 0.00045564596075564623), (19, 0.009391756728291512), (20, 0.00934989470988512), (21, 0.0007698246045038104), (22, 0.0007710466161370277), (23, 0.0015577219892293215), (24, 0.00341844093054533), (25, 0.005445651710033417), (26, 0.010751055553555489), (27, 0.01190162356942892)], Temporal MSE: [(0, 8.014948434720282e-06), (1, 6.157703865028452e-06), (2, 8.85596455191262e-06), (3, 8.35770242701983e-06), (4, 1.314717792411102e-05), (5, 1.6958090782281943e-05), (6, 0.00019113399321213365), (7, 6.539528112625703e-05), (8, 7.874935545260087e-05), (9, 5.397084169089794e-05), (10, 5.928854443482123e-05), (11, 8.024601265788078e-05), (12, 6.412393122445792e-05), (13, 9.910567314364016e-05), (14, 0.00017632926756050438), (15, 0.00019448963575996459), (16, 0.00014845372061245143), (17, 0.00012171295384177938), (18, 0.00017756318266037852), (19, 0.00024805133580230176), (20, 0.00032697265851311386), (21, 0.00036825932329520583), (22, 0.0003853661473840475), (23, 0.0006521027535200119), (24, 0.0013007639208808541), (25, 0.0023496788926422596), (26, 0.001652035629376769), (27, 0.0039870236068964005)]
+
+ 97%|█████████▋| 146/150 [21:28<00:30,  7.64s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 18, Spatial MSE: [(0, 5.559104465646669e-06), (1, 0.00012581280316226184), (2, 0.0001843244826886803), (3, 2.8158656277810223e-05), (4, 3.0229757612687536e-05), (5, 4.867168900091201e-05), (6, 0.00011499987886054441), (7, 0.00017429821309633553), (8, 0.00026780678308568895), (9, 0.000256808620179072), (10, 0.00031718958052806556), (11, 0.00030721601797267795), (12, 0.00016952578152995557), (13, 0.0018343671690672636), (14, 0.0069791171699762344), (15, 0.00539833027869463), (16, 0.13144001364707947), (17, 0.0007026669918559492), (18, 0.0006758830277249217), (19, 0.011083342134952545), (20, 0.011176297441124916), (21, 0.0009792724158614874), (22, 0.0010724698659032583), (23, 0.002132879802957177), (24, 0.004725325386971235), (25, 0.0075054168701171875), (26, 0.015931660309433937), (27, 0.018592463806271553)], Temporal MSE: [(0, 1.0068487426906358e-05), (1, 7.822983207006473e-06), (2, 1.1936866030737292e-05), (3, 1.051875005941838e-05), (4, 1.6354337276425213e-05), (5, 2.160649273719173e-05), (6, 0.00025625131092965603), (7, 8.586667536292225e-05), (8, 0.00010777903662528843), (9, 7.371543324552476e-05), (10, 7.788313087075949e-05), (11, 0.00010923690570052713), (12, 9.331027104053646e-05), (13, 0.00012883159797638655), (14, 0.00022449958487413824), (15, 0.00025544295203872025), (16, 0.0001949409779626876), (17, 0.0001563801633892581), (18, 0.00021537710563279688), (19, 0.00030953341047279537), (20, 0.0004220962873660028), (21, 0.0004813537816517055), (22, 0.0005059255636297166), (23, 0.0008725973893888295), (24, 0.001769994618371129), (25, 0.003261498175561428), (26, 0.002802395960316062), (27, 0.011663470417261124)]
+
+ 98%|█████████▊| 147/150 [21:35<00:22,  7.56s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 12, Spatial MSE: [(0, 7.551702765340451e-06), (1, 0.00016336646513082087), (2, 0.00025478610768914223), (3, 3.7903155316598713e-05), (4, 4.273759259376675e-05), (5, 6.439629214582965e-05), (6, 0.00017057295190170407), (7, 0.00021097855642437935), (8, 0.000337364588631317), (9, 0.000343557505402714), (10, 0.0004638734390027821), (11, 0.00042681084596551955), (12, 0.0002624226326588541), (13, 0.003547726199030876), (14, 0.012963559478521347), (15, 0.010301978327333927), (16, 0.25748851895332336), (17, 0.0011101781856268644), (18, 0.0010490725981071591), (19, 0.021593589335680008), (20, 0.022169215604662895), (21, 0.0015913194511085749), (22, 0.001689451397396624), (23, 0.003211462404578924), (24, 0.007332328706979752), (25, 0.011812088079750538), (26, 0.025592118501663208), (27, 0.03844345360994339)], Temporal MSE: [(0, 1.3413119631877635e-05), (1, 1.0731925613072235e-05), (2, 1.6526748368050903e-05), (3, 1.328009875578573e-05), (4, 2.190855775552336e-05), (5, 3.013086097780615e-05), (6, 0.00033538666320964694), (7, 0.00011713699495885521), (8, 0.00015894160605967045), (9, 9.978267189580947e-05), (10, 0.00011890654423041269), (11, 0.00019077576871495694), (12, 0.00018395840015728027), (13, 0.00020219499128870666), (14, 0.00034596878685988486), (15, 0.00041527836583554745), (16, 0.00027644250076264143), (17, 0.00023630897339899093), (18, 0.0003366877499502152), (19, 0.0005098051042295992), (20, 0.0006200271309353411), (21, 0.0007828492089174688), (22, 0.0007950612925924361), (23, 0.0013536193873733282), (24, 0.002742413431406021), (25, 0.004965709522366524), (26, 0.006032608915120363), (27, 0.04313603416085243)]
+
+ 99%|█████████▊| 148/150 [21:43<00:15,  7.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 6, Spatial MSE: [(0, 1.331548537564231e-05), (1, 0.00026823856751434505), (2, 0.00037747679743915796), (3, 6.433956150431186e-05), (4, 7.596498471684754e-05), (5, 0.00010462873615324497), (6, 0.00028089716215617955), (7, 0.000329425762174651), (8, 0.0005656113498844206), (9, 0.0005789247225038707), (10, 0.0008241098839789629), (11, 0.000759105256292969), (12, 0.0003600063209887594), (13, 0.0013191552134230733), (14, 0.004341030027717352), (15, 0.004060100298374891), (16, 0.07084067910909653), (17, 0.0014125261222943664), (18, 0.0010872121201828122), (19, 0.006814028136432171), (20, 0.007244875188916922), (21, 0.002141096629202366), (22, 0.0024178135208785534), (23, 0.00550042511895299), (24, 0.014964383095502853), (25, 0.025525398552417755), (26, 0.06502638757228851), (27, 0.1827978640794754)], Temporal MSE: [(0, 2.249392491648905e-05), (1, 1.878345210570842e-05), (2, 3.0286220862763003e-05), (3, 2.223872252216097e-05), (4, 3.6049957998329774e-05), (5, 5.169838550500572e-05), (6, 0.0005426656571216881), (7, 0.00019584746041800827), (8, 0.0002849341253750026), (9, 0.00017467906582169235), (10, 0.00022071492276154459), (11, 0.0002695034781936556), (12, 0.00017566565657034516), (13, 0.00024811207549646497), (14, 0.0005425040144473314), (15, 0.0005449266755022109), (16, 0.0003850889625027776), (17, 0.0002991749788634479), (18, 0.0003793234354816377), (19, 0.0006120478501543403), (20, 0.0008657441940158606), (21, 0.0007497037877328694), (22, 0.0010987980058416724), (23, 0.002125529106706381), (24, 0.005013765301555395), (25, 0.009821517392992973), (26, 0.018890857696533203), (27, 0.2601664066314697)]
+
+ 99%|█████████▉| 149/150 [21:51<00:07,  7.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 0, Spatial MSE: [(0, 8.744477236177772e-05), (1, 0.0009380081319250166), (2, 0.002778520341962576), (3, 0.0005016226787120104), (4, 0.000416223774664104), (5, 0.0006336686783470213), (6, 0.0008361666696146131), (7, 0.0007504603127017617), (8, 0.0014120835112407804), (9, 0.0014835152542218566), (10, 0.0018367194570600986), (11, 0.0018460000865161419), (12, 0.0017281788168475032), (13, 0.0037448937073349953), (14, 0.008066992275416851), (15, 0.0076479786075651646), (16, 0.10631826519966125), (17, 0.005186326801776886), (18, 0.004435178358107805), (19, 0.014554035849869251), (20, 0.015056024305522442), (21, 0.01212325505912304), (22, 0.01438493374735117), (23, 0.029114501550793648), (24, 0.10196729004383087), (25, 0.47162383794784546), (26, 0.7954879999160767), (27, 3.647407054901123)], Temporal MSE: [(0, 0.00013985652185510844), (1, 9.219918138114735e-05), (2, 0.0001190921975648962), (3, 0.00010127003042725846), (4, 0.0002817239728756249), (5, 0.00023264299670699984), (6, 0.000762422860134393), (7, 0.000517729262355715), (8, 0.00080661237007007), (9, 0.0005102085997350514), (10, 0.0009991327533498406), (11, 0.0012252284213900566), (12, 0.0008512779022566974), (13, 0.0012047069612890482), (14, 0.002101596910506487), (15, 0.0019295449601486325), (16, 0.0016027523670345545), (17, 0.001370109268464148), (18, 0.0016014478169381618), (19, 0.0029830620624125004), (20, 0.003579851007089019), (21, 0.003661336377263069), (22, 0.005605361424386501), (23, 0.010361749678850174), (24, 0.029446443542838097), (25, 0.07316232472658157), (26, 0.18075893819332123), (27, 2.8201282024383545)]
+
+100%|██████████| 150/150 [21:58<00:00,  7.66s/it]
+100%|██████████| 150/150 [21:58<00:00,  8.79s/it]
+Saved spatial MLP MSE outputs to ./samples/opensora_plan/skip_mse/1_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora_plan/skip_mse/1_temporal_mlp_mse.pt
+
+  0%|          | 0/150 [00:00<?, ?it/s]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1840488696179818e-06), (1, 6.533668965857942e-06), (2, 0.0001646618329687044), (3, 2.0931414837832563e-05), (4, 8.911434633773752e-06), (5, 6.671487426501699e-06), (6, 8.228103069995996e-06), (7, 5.631293788610492e-06), (8, 5.6176891121140216e-06), (9, 6.2029625951254275e-06), (10, 9.667953236203175e-06), (11, 1.1898751836270094e-05), (12, 1.5068978427734692e-05), (13, 2.5112476578215137e-05), (14, 9.382277494296432e-05), (15, 0.00014510142500512302), (16, 0.0009193935547955334), (17, 6.433895759982988e-05), (18, 8.304783841595054e-05), (19, 9.641351061873138e-05), (20, 0.00010506850230740383), (21, 8.844891272019595e-05), (22, 9.316437353845686e-05), (23, 8.396019984502345e-05), (24, 0.00016568589489907026), (25, 0.0003667851851787418), (26, 0.00041340256575495005), (27, 0.0036930327769368887)], Temporal MSE: [(0, 1.6402359506173525e-06), (1, 1.2085710068276967e-06), (2, 1.6718837514417828e-06), (3, 1.5906656471997849e-06), (4, 1.056298970070202e-06), (5, 1.575233682160615e-06), (6, 7.694083024034626e-07), (7, 9.864426147032646e-07), (8, 6.571339099536999e-07), (9, 8.795150847618061e-07), (10, 2.081196271319641e-06), (11, 1.5452154684680863e-06), (12, 1.564106923979125e-06), (13, 4.168866325926501e-06), (14, 3.884897068928694e-06), (15, 6.168395884742495e-06), (16, 3.320204086776357e-06), (17, 6.242253220989369e-06), (18, 3.4970591968885856e-06), (19, 3.4671963931032224e-06), (20, 4.6959135033830535e-06), (21, 4.067353074788116e-06), (22, 1.2006389624730218e-05), (23, 7.577572887385031e-06), (24, 1.9270499251433648e-05), (25, 1.751370837155264e-05), (26, 2.9670745789189823e-06), (27, 1.2069486729160417e-05)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1688831236256192e-08), (1, 1.444754218482558e-07), (2, 1.4475183434115024e-06), (3, 2.669312380021438e-07), (4, 1.0991303867058377e-07), (5, 8.662078698762343e-08), (6, 2.1725772114677966e-07), (7, 1.2541840987978503e-07), (8, 1.0843077546951463e-07), (9, 1.400600808665331e-07), (10, 1.4717156204824278e-07), (11, 1.510296101514541e-07), (12, 1.7757326986611588e-07), (13, 1.8928552947272692e-07), (14, 4.751131257307861e-07), (15, 6.960150926715869e-07), (16, 3.164494046359323e-06), (17, 4.601632213052653e-07), (18, 9.877679758574232e-07), (19, 7.010445415289723e-07), (20, 7.603959488733381e-07), (21, 1.0204513500866597e-06), (22, 1.307026764152397e-06), (23, 1.0331709745514672e-06), (24, 2.9328896289371187e-06), (25, 4.858265583607135e-06), (26, 5.5968057495192625e-06), (27, 2.8274587748455815e-05)], Temporal MSE: [(0, 7.074547614394078e-09), (1, 9.541526679868184e-09), (2, 9.497952646597696e-09), (3, 1.159906481973394e-08), (4, 1.4956928140463788e-08), (5, 2.6140012820974334e-08), (6, 1.3465722759065102e-08), (7, 1.7600836343945048e-08), (8, 1.7056345669175244e-08), (9, 2.2575328273433115e-08), (10, 3.067811960022482e-08), (11, 3.0156108721257624e-08), (12, 2.7489544862646653e-08), (13, 5.297237137824595e-08), (14, 4.37767511130005e-08), (15, 7.852541017427939e-08), (16, 3.0214188484478655e-08), (17, 5.9428387544357975e-08), (18, 5.947003955952823e-08), (19, 5.211170872598814e-08), (20, 9.656275068437026e-08), (21, 7.552563374702004e-08), (22, 1.29835882489715e-07), (23, 1.3721130187605013e-07), (24, 2.7817506520477764e-07), (25, 3.619600477122731e-07), (26, 1.3556832811900676e-07), (27, 1.3411609245395084e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 9.17651391318941e-07), (1, 5.777075330115622e-06), (2, 0.00015510365483351052), (3, 1.524736489955103e-05), (4, 6.845861662441166e-06), (5, 5.168995357962558e-06), (6, 5.0261396609130315e-06), (7, 5.768717073806329e-06), (8, 5.625627181871096e-06), (9, 6.839893103460781e-06), (10, 1.154274468717631e-05), (11, 1.4197650671121664e-05), (12, 2.0953862986061722e-05), (13, 2.4770630261627957e-05), (14, 9.539136954117566e-05), (15, 0.00016052660066634417), (16, 0.0005834931507706642), (17, 4.7672278014943004e-05), (18, 7.825138891348615e-05), (19, 8.464528218610212e-05), (20, 0.00011421705858083442), (21, 8.278457244159654e-05), (22, 6.506327190436423e-05), (23, 7.195780199253932e-05), (24, 0.00020817869517486542), (25, 0.00041243439773097634), (26, 0.00021688744891434908), (27, 0.0026509957388043404)], Temporal MSE: [(0, 1.3108548273521592e-06), (1, 1.0137606523130671e-06), (2, 1.144697989730048e-06), (3, 1.0041577525043976e-06), (4, 8.226597287830373e-07), (5, 1.3558310456573963e-06), (6, 6.83400571688253e-07), (7, 1.2537773272924824e-06), (8, 6.412306561287551e-07), (9, 1.1108115813840413e-06), (10, 3.1492365906160558e-06), (11, 2.2913075099495472e-06), (12, 2.0535917428787798e-06), (13, 6.014751306793187e-06), (14, 5.151141067472054e-06), (15, 8.714643627172336e-06), (16, 3.146039716739324e-06), (17, 7.178704436228145e-06), (18, 3.1983972803573124e-06), (19, 3.4866998248617165e-06), (20, 4.6461673264275305e-06), (21, 3.6065641779714497e-06), (22, 1.0144623047381174e-05), (23, 6.821069746365538e-06), (24, 1.9375167539692484e-05), (25, 1.4553963410435244e-05), (26, 3.002142875629943e-06), (27, 7.878773431002628e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 1.1035143465676356e-08), (1, 1.2372777291602688e-07), (2, 1.2304724350542529e-06), (3, 2.321934573501494e-07), (4, 1.0284553297879029e-07), (5, 8.118064442896866e-08), (6, 1.9288887642687769e-07), (7, 1.1798218224612356e-07), (8, 1.0286234441991837e-07), (9, 1.318084912327322e-07), (10, 1.3882470284443116e-07), (11, 1.423675399792046e-07), (12, 1.6715310380277515e-07), (13, 1.7234000893040502e-07), (14, 3.8295146964628657e-07), (15, 5.104648153064772e-07), (16, 8.96942367489828e-07), (17, 4.1272161865890666e-07), (18, 8.828495765555999e-07), (19, 5.292235982778948e-07), (20, 6.65633933749632e-07), (21, 9.403556191500684e-07), (22, 1.1949193776672473e-06), (23, 9.557387556924368e-07), (24, 2.7286394015391124e-06), (25, 4.560366505756974e-06), (26, 5.416331987362355e-06), (27, 2.7341699023963884e-05)], Temporal MSE: [(0, 6.736801339002341e-09), (1, 8.65346549971946e-09), (2, 8.72244676486389e-09), (3, 1.0709118924978611e-08), (4, 1.3921297892238726e-08), (5, 2.4339360749081607e-08), (6, 1.2690256845360182e-08), (7, 1.6510648848111487e-08), (8, 1.605899058176874e-08), (9, 2.1170766473233016e-08), (10, 2.9073591534256593e-08), (11, 2.8182167710610884e-08), (12, 2.592188863559386e-08), (13, 4.700971700799528e-08), (14, 3.986961871760286e-08), (15, 7.358472942087246e-08), (16, 2.7053770779161823e-08), (17, 5.2319965249125744e-08), (18, 5.3981519698709235e-08), (19, 4.654118157532139e-08), (20, 8.825244179888614e-08), (21, 6.810552832803296e-08), (22, 1.1871466654156393e-07), (23, 1.238083626731168e-07), (24, 2.5749548626663454e-07), (25, 3.433268602748285e-07), (26, 1.2933354298638733e-07), (27, 1.2825377382341685e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.313970074079407e-06), (1, 3.647566700237803e-06), (2, 8.870931196724996e-05), (3, 1.5094582522579003e-05), (4, 6.668808509857627e-06), (5, 4.485590579861309e-06), (6, 5.42354200661066e-06), (7, 4.072969659318915e-06), (8, 4.785845248989062e-06), (9, 6.064635272196028e-06), (10, 8.530008926754817e-06), (11, 8.20891455077799e-06), (12, 1.0957915037579369e-05), (13, 1.3502242836693767e-05), (14, 4.844216527999379e-05), (15, 6.087563815526664e-05), (16, 0.000566229922696948), (17, 2.1693989765481092e-05), (18, 3.169395131408237e-05), (19, 5.9246711316518486e-05), (20, 3.811960414168425e-05), (21, 3.149582335026935e-05), (22, 3.861912409774959e-05), (23, 3.118972381344065e-05), (24, 7.692582585150376e-05), (25, 0.00015676046314183623), (26, 0.00018334689957555383), (27, 0.003968806471675634)], Temporal MSE: [(0, 1.6638911120026023e-06), (1, 9.956196436178288e-07), (2, 1.7249693655685405e-06), (3, 1.3772423699265346e-06), (4, 9.95073150988901e-07), (5, 1.3170248394089867e-06), (6, 7.028596655800357e-07), (7, 5.747780846832029e-07), (8, 1.1330191682645818e-06), (9, 1.1407679494368494e-06), (10, 1.4453116818913259e-06), (11, 1.1800918855442433e-06), (12, 1.6118376606755191e-06), (13, 2.8250508421479026e-06), (14, 3.3343649192829616e-06), (15, 2.537423142712214e-06), (16, 1.820762804527476e-06), (17, 2.5333815756312106e-06), (18, 1.839266928982397e-06), (19, 2.732920165726682e-06), (20, 4.253767656337004e-06), (21, 2.504463509467314e-06), (22, 3.227203933420242e-06), (23, 4.508593065111199e-06), (24, 9.757241059560329e-06), (25, 1.1312097740301397e-05), (26, 3.6325834571471205e-06), (27, 3.5670011584443273e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.1520868703485121e-08), (1, 1.3974810997297027e-07), (2, 1.4055619885766646e-06), (3, 2.589308394362888e-07), (4, 1.1232480545686485e-07), (5, 8.658382455450919e-08), (6, 2.1137606154297828e-07), (7, 1.2441090291304135e-07), (8, 1.0794898486210514e-07), (9, 1.3913238205987e-07), (10, 1.4566670358817646e-07), (11, 1.5025794652956392e-07), (12, 1.7717746914058807e-07), (13, 1.8084020325659367e-07), (14, 4.298549640679994e-07), (15, 6.494890385511098e-07), (16, 5.8955383792635985e-06), (17, 4.928549515170744e-07), (18, 1.0486071460036328e-06), (19, 8.322791131831764e-07), (20, 7.730433253527735e-07), (21, 1.1880190413648961e-06), (22, 1.4483154018307687e-06), (23, 1.0487960935279261e-06), (24, 2.9592474675155245e-06), (25, 4.799062025995227e-06), (26, 5.699106623069383e-06), (27, 2.837411557266023e-05)], Temporal MSE: [(0, 6.936378582622638e-09), (1, 9.26395493650034e-09), (2, 9.306375226003638e-09), (3, 1.1369487573631432e-08), (4, 1.487410639100517e-08), (5, 2.58021088939131e-08), (6, 1.3414978461412375e-08), (7, 1.735200783059554e-08), (8, 1.7036608568332667e-08), (9, 2.260978781976064e-08), (10, 3.0779837345562555e-08), (11, 2.972881674168093e-08), (12, 2.7456819040594382e-08), (13, 4.9741270657932546e-08), (14, 4.379855056413362e-08), (15, 7.764048604030904e-08), (16, 3.590003316844559e-08), (17, 7.151029990382085e-08), (18, 6.146697728581785e-08), (19, 5.608535857959396e-08), (20, 9.629743402683744e-08), (21, 7.643940591606224e-08), (22, 1.439258596747095e-07), (23, 1.4635911327331996e-07), (24, 2.7947061198574374e-07), (25, 3.580516363399511e-07), (26, 1.3665706433130254e-07), (27, 1.3451513325435371e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 3.4736467569018714e-07), (1, 9.05466947642708e-07), (2, 1.842031633714214e-05), (3, 3.819040102825966e-06), (4, 1.772539462763234e-06), (5, 1.3224824897406506e-06), (6, 1.463434387005691e-06), (7, 1.4072294334255275e-06), (8, 1.91751087186276e-06), (9, 2.9734558211202966e-06), (10, 5.151020104676718e-06), (11, 5.053497261542361e-06), (12, 7.939538591017481e-06), (13, 8.288614480989054e-06), (14, 2.7909996788366698e-05), (15, 5.39320390089415e-05), (16, 0.00010147786088055), (17, 1.3383453733695205e-05), (18, 2.0685962226707488e-05), (19, 1.7110711269197054e-05), (20, 2.0960758774890564e-05), (21, 1.2325620446063112e-05), (22, 1.647103090363089e-05), (23, 1.3914239389123395e-05), (24, 3.6619567254092544e-05), (25, 6.76506242598407e-05), (26, 3.849598579108715e-05), (27, 0.0006812551291659474)], Temporal MSE: [(0, 4.14736234688462e-07), (1, 2.2272094213349192e-07), (2, 2.7577411287893483e-07), (3, 2.9566595571850485e-07), (4, 2.783557420116267e-07), (5, 4.6521492436113476e-07), (6, 2.734155657435622e-07), (7, 4.0436981407765415e-07), (8, 2.749091834175488e-07), (9, 4.935486117574328e-07), (10, 1.180988419946516e-06), (11, 7.953972840368806e-07), (12, 8.244458626904816e-07), (13, 1.979696207854431e-06), (14, 2.2146264200273436e-06), (15, 2.864474254238303e-06), (16, 1.1793486009992193e-06), (17, 1.8825804772859556e-06), (18, 1.1725501281034667e-06), (19, 1.0423120784253115e-06), (20, 1.5745252994747716e-06), (21, 1.0551908644629293e-06), (22, 1.891741931103752e-06), (23, 1.7449644929001806e-06), (24, 5.361622697819257e-06), (25, 4.294243353797356e-06), (26, 1.1567258297873195e-06), (27, 2.2538497432833537e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 1.0886224366402075e-08), (1, 1.192331779975575e-07), (2, 1.2098463457732578e-06), (3, 2.269407985977523e-07), (4, 1.0157217644746197e-07), (5, 8.033637755033851e-08), (6, 1.8910618848622107e-07), (7, 1.1858855231139387e-07), (8, 1.0302746744628166e-07), (9, 1.3200867954310525e-07), (10, 1.3834514334121195e-07), (11, 1.4213415511221683e-07), (12, 1.6706999872440065e-07), (13, 1.5985371248916636e-07), (14, 3.2651843184794416e-07), (15, 5.512849270417064e-07), (16, 2.5757217372301966e-06), (17, 4.458413798147376e-07), (18, 8.837877203404787e-07), (19, 6.258243274714914e-07), (20, 6.864873398626514e-07), (21, 9.85095994110452e-07), (22, 1.2563239124574466e-06), (23, 9.683986945674405e-07), (24, 2.7597216103458777e-06), (25, 4.579368123813765e-06), (26, 5.412544396676822e-06), (27, 2.7314225008012727e-05)], Temporal MSE: [(0, 6.57733778552938e-09), (1, 8.400808937381044e-09), (2, 8.544977170288348e-09), (3, 1.0600595068410712e-08), (4, 1.3884743133019128e-08), (5, 2.4119930941424172e-08), (6, 1.2561744533456931e-08), (7, 1.6394304580558128e-08), (8, 1.6040248240756227e-08), (9, 2.1054306742485096e-08), (10, 2.9024683101397386e-08), (11, 2.805304255559804e-08), (12, 2.571198010059561e-08), (13, 4.367367623103746e-08), (14, 4.0427089231798163e-08), (15, 7.304959837028946e-08), (16, 2.933089504608688e-08), (17, 5.8875556874227186e-08), (18, 5.57894139774362e-08), (19, 4.960368471529364e-08), (20, 9.015058566319567e-08), (21, 7.019707481958903e-08), (22, 1.2645153901758022e-07), (23, 1.3068245152680902e-07), (24, 2.63354138496652e-07), (25, 3.476748133834917e-07), (26, 1.298161436125156e-07), (27, 1.2961903905761574e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 3.224118472644477e-06), (1, 3.046360689040739e-05), (2, 0.0007350720698013902), (3, 8.248740778071806e-05), (4, 3.089853635174222e-05), (5, 2.0960336769348942e-05), (6, 2.7396594305173494e-05), (7, 1.64335870067589e-05), (8, 1.523282116977498e-05), (9, 1.6353138562408276e-05), (10, 2.293263969477266e-05), (11, 2.5344294044771232e-05), (12, 3.68641376553569e-05), (13, 6.52943563181907e-05), (14, 0.00026687385980039835), (15, 0.00046083307825028896), (16, 0.0027680499479174614), (17, 0.00018250077846460044), (18, 0.0002872974146157503), (19, 0.0003921754250768572), (20, 0.0004328801005613059), (21, 0.000244387163547799), (22, 0.0002561699948273599), (23, 0.0002447004953864962), (24, 0.0004892456345260143), (25, 0.0010059382766485214), (26, 0.001191001501865685), (27, 0.015179895795881748)], Temporal MSE: [(0, 4.363810603535967e-06), (1, 4.095974418305559e-06), (2, 6.037267212377628e-06), (3, 5.661432169290492e-06), (4, 4.451434961083578e-06), (5, 4.858444299316034e-06), (6, 2.22981157094182e-06), (7, 3.246558208047645e-06), (8, 2.0489783310040366e-06), (9, 2.375091980866273e-06), (10, 5.0578778427734505e-06), (11, 4.321895175962709e-06), (12, 3.7461229567270493e-06), (13, 1.4244122212403454e-05), (14, 9.622598554415163e-06), (15, 1.799647361622192e-05), (16, 7.319109045056393e-06), (17, 2.3547097953269258e-05), (18, 1.0350241609557997e-05), (19, 1.3619166566058993e-05), (20, 1.938148125191219e-05), (21, 1.0809045306814369e-05), (22, 2.4823730200296268e-05), (23, 2.001028406084515e-05), (24, 5.6179640523623675e-05), (25, 5.160388900549151e-05), (26, 9.954969755199272e-06), (27, 3.95321476389654e-05)]
+
+  1%|          | 1/150 [01:20<3:20:11, 80.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 1.20379466395093e-08), (1, 1.5226267180423747e-07), (2, 1.5085782933965675e-06), (3, 2.7406923663875205e-07), (4, 1.1206784478190457e-07), (5, 8.773860571409386e-08), (6, 2.2340297789469332e-07), (7, 1.2577646657518926e-07), (8, 1.0999721666848927e-07), (9, 1.4311498830466007e-07), (10, 1.5098798655799328e-07), (11, 1.5619809801137308e-07), (12, 1.844448291876688e-07), (13, 2.0424583624389925e-07), (14, 5.304436285769043e-07), (15, 7.83991822572716e-07), (16, 7.295069735846482e-06), (17, 5.331675083652954e-07), (18, 1.2097654007448e-06), (19, 1.0248309081362095e-06), (20, 9.264588811674912e-07), (21, 1.2797786439477932e-06), (22, 1.4873027112116688e-06), (23, 1.0794373110911692e-06), (24, 3.0471678655885626e-06), (25, 4.996151346858824e-06), (26, 5.780114861408947e-06), (27, 2.927277091657743e-05)], Temporal MSE: [(0, 7.1570944726317975e-09), (1, 9.484308449714263e-09), (2, 9.607180828652417e-09), (3, 1.1638797481339225e-08), (4, 1.5157270993881866e-08), (5, 2.6519890283793757e-08), (6, 1.393189119625049e-08), (7, 1.7998027956878104e-08), (8, 1.761279477818789e-08), (9, 2.3750493127749905e-08), (10, 3.2201604938109085e-08), (11, 3.149306238015015e-08), (12, 2.9176364435556934e-08), (13, 5.5532272114078296e-08), (14, 4.742896564380317e-08), (15, 8.265239159754856e-08), (16, 4.218968641112042e-08), (17, 7.40421626233001e-08), (18, 6.80929872487468e-08), (19, 5.96200493419019e-08), (20, 1.0412512096991122e-07), (21, 8.366608028609335e-08), (22, 1.505741664686866e-07), (23, 1.5186768109742843e-07), (24, 2.92192197548502e-07), (25, 3.7848320744160446e-07), (26, 1.4254396774049383e-07), (27, 1.4062075592846668e-07)]
+
+  1%|▏         | 2/150 [01:28<1:33:44, 38.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 3.773860271394369e-06), (1, 2.9540820833062753e-05), (2, 0.0008771322900429368), (3, 9.874852548819035e-05), (4, 3.943900446756743e-05), (5, 2.5225523131666705e-05), (6, 3.863235906464979e-05), (7, 2.0005238184239715e-05), (8, 1.7641143131186254e-05), (9, 1.9260380213381723e-05), (10, 2.7121877792524174e-05), (11, 2.874512574635446e-05), (12, 4.492460357141681e-05), (13, 7.086819823598489e-05), (14, 0.0002933072973974049), (15, 0.0005146242910996079), (16, 0.0034895839635282755), (17, 0.0002478235983289778), (18, 0.0002965234452858567), (19, 0.00045404519187286496), (20, 0.000592219119425863), (21, 0.00039081298746168613), (22, 0.00030888072797097266), (23, 0.00029032249585725367), (24, 0.0006426154286600649), (25, 0.0015828735195100307), (26, 0.0014572780346497893), (27, 0.02057027630507946)], Temporal MSE: [(0, 5.150494416739093e-06), (1, 4.8105493988259695e-06), (2, 7.359501069004182e-06), (3, 6.283053153310902e-06), (4, 5.292048626870383e-06), (5, 6.270085577853024e-06), (6, 2.581782382549136e-06), (7, 3.7466197682078928e-06), (8, 2.632922360135126e-06), (9, 2.7299061002850067e-06), (10, 5.585145572695183e-06), (11, 5.050028448749799e-06), (12, 4.554250153887551e-06), (13, 1.392525064147776e-05), (14, 1.0419187674415298e-05), (15, 1.9298664483358152e-05), (16, 9.43189297686331e-06), (17, 2.69347528956132e-05), (18, 1.2784003047272563e-05), (19, 1.5753717889310792e-05), (20, 2.5007506337715313e-05), (21, 1.3165765267331153e-05), (22, 3.4953478461829945e-05), (23, 2.3607655748492107e-05), (24, 6.785620644222945e-05), (25, 6.805224984418601e-05), (26, 1.3141740964783821e-05), (27, 4.7685472964076325e-05)]
+
+  2%|▏         | 3/150 [01:36<59:02, 24.10s/it]  Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 1.0897656999020455e-08), (1, 1.2413715921866242e-07), (2, 1.2033495977448183e-06), (3, 2.248296624429713e-07), (4, 1.0239349279572707e-07), (5, 7.92788199532879e-08), (6, 1.9022793651402026e-07), (7, 1.1728742776995205e-07), (8, 1.0160047025919994e-07), (9, 1.307417107909714e-07), (10, 1.3780068286450842e-07), (11, 1.4159374472910713e-07), (12, 1.6832419191814552e-07), (13, 1.615089075812648e-07), (14, 3.155219587824831e-07), (15, 5.207267008700001e-07), (16, 1.2409116152412025e-06), (17, 4.1949900264626194e-07), (18, 8.712628982721071e-07), (19, 5.562077944887278e-07), (20, 6.76750062211795e-07), (21, 9.769031521500438e-07), (22, 1.2217949461046373e-06), (23, 9.7603810900182e-07), (24, 2.7750875233323313e-06), (25, 4.492365405894816e-06), (26, 5.362481715565082e-06), (27, 2.677364318515174e-05)], Temporal MSE: [(0, 6.452773426701697e-09), (1, 8.23711765463031e-09), (2, 8.405064200189827e-09), (3, 1.0437152475617495e-08), (4, 1.3625077954770859e-08), (5, 2.3864705767095984e-08), (6, 1.2651561576149106e-08), (7, 1.6340861108687932e-08), (8, 1.604978017155645e-08), (9, 2.1368672165067437e-08), (10, 2.9429651604573337e-08), (11, 2.864594428331202e-08), (12, 2.6926725738007917e-08), (13, 4.52125838990014e-08), (14, 4.123186059246109e-08), (15, 7.725986250761707e-08), (16, 2.9002654500231984e-08), (17, 5.538378644587283e-08), (18, 5.5916597574423577e-08), (19, 4.897760774724702e-08), (20, 9.069296424968343e-08), (21, 7.075805541489899e-08), (22, 1.2021683915008907e-07), (23, 1.2740808585931518e-07), (24, 2.6116478579751856e-07), (25, 3.5601252079686674e-07), (26, 1.306098056375049e-07), (27, 1.2862005860370118e-07)]
+
+  3%|▎         | 4/150 [01:45<44:45, 18.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 870, Spatial MSE: [(0, 1.2097176295355894e-05), (1, 0.00010216389637207612), (2, 0.0025753567460924387), (3, 0.00028583899256773293), (4, 0.00011209410877199844), (5, 7.328301580855623e-05), (6, 8.930040348786861e-05), (7, 7.615134381921962e-05), (8, 5.469383540912531e-05), (9, 5.11600446770899e-05), (10, 6.595022568944842e-05), (11, 7.427457603625953e-05), (12, 0.00011381017247913405), (13, 0.0001929817080963403), (14, 0.0006664913962595165), (15, 0.0012390908086672425), (16, 0.0066401599906384945), (17, 0.0006307520088739693), (18, 0.0008709471439942718), (19, 0.0009119309834204614), (20, 0.0013115950860083103), (21, 0.0006728892913088202), (22, 0.0008213673136197031), (23, 0.0008048840099945664), (24, 0.001718722516670823), (25, 0.004646872635930777), (26, 0.004446870181709528), (27, 0.058043498545885086)], Temporal MSE: [(0, 1.6836867871461436e-05), (1, 1.5482366507058032e-05), (2, 2.206226417911239e-05), (3, 1.8795157302520238e-05), (4, 1.535868796054274e-05), (5, 1.7461996321799234e-05), (6, 8.763862751948182e-06), (7, 1.0168219887418672e-05), (8, 7.995127816684544e-06), (9, 9.028882232087199e-06), (10, 1.728172901493963e-05), (11, 1.5097816685738508e-05), (12, 1.5170970982580911e-05), (13, 4.323313623899594e-05), (14, 3.138977263006382e-05), (15, 6.127680535428226e-05), (16, 2.642267463670578e-05), (17, 8.438283839495853e-05), (18, 3.372087303432636e-05), (19, 5.0185710279038176e-05), (20, 7.29637686163187e-05), (21, 4.682180951931514e-05), (22, 7.704538438702002e-05), (23, 6.562520138686523e-05), (24, 0.00020823116938117892), (25, 0.0002467017329763621), (26, 3.72797112504486e-05), (27, 0.00014485884457826614)]
+
+  3%|▎         | 5/150 [01:54<35:56, 14.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 864, Spatial MSE: [(0, 1.547869760543108e-05), (1, 0.0001386739022564143), (2, 0.003588181221857667), (3, 0.0003887331404257566), (4, 0.0001596402726136148), (5, 0.0001012270076898858), (6, 0.0001581437245476991), (7, 7.828316302038729e-05), (8, 6.850382487755269e-05), (9, 8.220814925152808e-05), (10, 0.00011565872409846634), (11, 0.00011398659989936277), (12, 0.0001877231552498415), (13, 0.0003055544220842421), (14, 0.0013680538395419717), (15, 0.002211357234045863), (16, 0.013722151517868042), (17, 0.0009077399736270308), (18, 0.0011795182945206761), (19, 0.0016759844729676843), (20, 0.0023253606632351875), (21, 0.0009720192174427211), (22, 0.0010901789646595716), (23, 0.0010525378165766597), (24, 0.0021957235876470804), (25, 0.006008824333548546), (26, 0.0053910198621451855), (27, 0.08882279694080353)], Temporal MSE: [(0, 2.0977096937713213e-05), (1, 1.978170439542737e-05), (2, 2.732995017140638e-05), (3, 2.515549203963019e-05), (4, 2.148696694348473e-05), (5, 2.5778686904232018e-05), (6, 1.0704316991905216e-05), (7, 1.8115202692570165e-05), (8, 1.127597442973638e-05), (9, 1.2349994904070627e-05), (10, 2.8224965717527084e-05), (11, 2.3792757929186337e-05), (12, 2.01829298021039e-05), (13, 7.339224248426035e-05), (14, 4.990546585759148e-05), (15, 9.538473386783153e-05), (16, 4.157725197728723e-05), (17, 0.00014004201511852443), (18, 6.217135523911566e-05), (19, 9.099371527554467e-05), (20, 0.00017324781219940633), (21, 7.462446228601038e-05), (22, 0.00011184214963577688), (23, 9.246782428817824e-05), (24, 0.00024302191741298884), (25, 0.0003018581192009151), (26, 4.956677003065124e-05), (27, 0.0001529809960629791)]
+
+  4%|▍         | 6/150 [02:02<30:05, 12.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 858, Spatial MSE: [(0, 3.906961501343176e-05), (1, 0.00034773029619827867), (2, 0.008651754818856716), (3, 0.0009153573191724718), (4, 0.00037836740375496447), (5, 0.0002452286134939641), (6, 0.0002829107397701591), (7, 0.0002467541489750147), (8, 0.00017665885388851166), (9, 0.00019368226639926434), (10, 0.000264944217633456), (11, 0.00027808116283267736), (12, 0.0004763082251884043), (13, 0.0007862994098104537), (14, 0.0029444266110658646), (15, 0.005490333307534456), (16, 0.028729567304253578), (17, 0.002263713860884309), (18, 0.0027205164078623056), (19, 0.003920737653970718), (20, 0.005706168245524168), (21, 0.002476014196872711), (22, 0.002751237014308572), (23, 0.0028221660759299994), (24, 0.006248782854527235), (25, 0.018355967476963997), (26, 0.014679801650345325), (27, 0.20763668417930603)], Temporal MSE: [(0, 5.587815394392237e-05), (1, 4.984054612577893e-05), (2, 6.85401537339203e-05), (3, 5.964657975709997e-05), (4, 5.120859350427054e-05), (5, 6.20584687567316e-05), (6, 2.817568565660622e-05), (7, 4.0766939491732046e-05), (8, 2.8193318939884193e-05), (9, 3.374534935574047e-05), (10, 7.071688742144033e-05), (11, 5.806435365229845e-05), (12, 5.728184260078706e-05), (13, 0.00018889944476541132), (14, 0.0001278202107641846), (15, 0.0002618673024699092), (16, 0.00012350184260867536), (17, 0.00042051184573210776), (18, 0.00017859108629636467), (19, 0.0002714515430852771), (20, 0.00047522271052002907), (21, 0.0002189294173149392), (22, 0.00024924479657784104), (23, 0.00022307848848868161), (24, 0.0006200660718604922), (25, 0.0010312082013115287), (26, 0.0001419129694113508), (27, 0.0004536774358712137)]
+
+  5%|▍         | 7/150 [02:10<26:25, 11.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 852, Spatial MSE: [(0, 4.825803898711456e-07), (1, 3.0568946840503486e-06), (2, 5.2332732593640685e-05), (3, 9.042484634846915e-06), (4, 2.7987077828584006e-06), (5, 2.5657686819613446e-06), (6, 5.447480816656025e-06), (7, 7.758420906611718e-06), (8, 7.035687758616405e-06), (9, 1.829942993936129e-05), (10, 2.0451036107260734e-05), (11, 2.6397194233140908e-05), (12, 2.856133141904138e-05), (13, 4.095460462849587e-05), (14, 0.00014912631013430655), (15, 0.00027311602025292814), (16, 0.003615455236285925), (17, 5.6487962865503505e-05), (18, 6.088815280236304e-05), (19, 0.00029452258604578674), (20, 0.0002234708663308993), (21, 8.652981341583654e-05), (22, 0.00010341735469410196), (23, 7.696151442360133e-05), (24, 0.0001656243548495695), (25, 0.0002620691084302962), (26, 0.00012689291907008737), (27, 0.001147912465967238)], Temporal MSE: [(0, 5.994510274831555e-07), (1, 3.860754986817483e-07), (2, 5.533570970328583e-07), (3, 6.001057499815943e-07), (4, 4.431172726526711e-07), (5, 8.979108088169596e-07), (6, 6.316810186035582e-07), (7, 7.576977054668532e-07), (8, 7.945088214000862e-07), (9, 1.88201840956026e-06), (10, 4.331681338953786e-06), (11, 4.022280791105004e-06), (12, 4.949677986587631e-06), (13, 1.0309171557310037e-05), (14, 1.6418465747847222e-05), (15, 1.260909721167991e-05), (16, 9.241022780770436e-06), (17, 1.6524252714589238e-05), (18, 8.478647032461595e-06), (19, 1.2199107004562393e-05), (20, 2.3907357899588533e-05), (21, 1.1509760952321813e-05), (22, 1.1135994100186508e-05), (23, 1.3408804989012424e-05), (24, 3.108465170953423e-05), (25, 2.754828346951399e-05), (26, 4.544483090285212e-06), (27, 2.5615256618038984e-06)]
+
+  5%|▌         | 8/150 [02:18<24:00, 10.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 846, Spatial MSE: [(0, 2.8606314117496368e-06), (1, 2.1059806385892443e-05), (2, 0.0006047558272257447), (3, 0.0001049099228112027), (4, 2.4410710466327146e-05), (5, 1.2520466952992138e-05), (6, 2.9813349101459607e-05), (7, 1.1462325346656144e-05), (8, 1.1233219083806034e-05), (9, 1.287435406993609e-05), (10, 1.1728438948921394e-05), (11, 2.1328989532776177e-05), (12, 2.6160991183132865e-05), (13, 0.00031621468951925635), (14, 0.0008033202611841261), (15, 0.0003371287020854652), (16, 0.008254959248006344), (17, 6.527730874950066e-05), (18, 0.0002115404640790075), (19, 0.0005063480930402875), (20, 0.0001533317845314741), (21, 0.0003126561059616506), (22, 0.0005452211480587721), (23, 0.00027765758568421006), (24, 0.0005513156647793949), (25, 0.001401696470566094), (26, 0.0013269736664369702), (27, 0.016496777534484863)], Temporal MSE: [(0, 3.974798801209545e-06), (1, 2.607696160339401e-06), (2, 3.92680112781818e-06), (3, 3.6859073588857427e-06), (4, 3.3121350497822277e-06), (5, 3.892315817211056e-06), (6, 1.866029037955741e-06), (7, 1.919547457873705e-06), (8, 1.6192594785024994e-06), (9, 2.151044100173749e-06), (10, 3.2537325296289055e-06), (11, 7.165341685322346e-06), (12, 9.173760190606117e-06), (13, 3.898113209288567e-05), (14, 1.2261482879694086e-05), (15, 1.2852125109930057e-05), (16, 1.2726314707833808e-05), (17, 1.970287848962471e-05), (18, 1.1006733984686434e-05), (19, 1.1812019693024922e-05), (20, 1.927285484271124e-05), (21, 2.373882853135001e-05), (22, 6.542051414726302e-05), (23, 0.00010231106716673821), (24, 0.00013060090714134276), (25, 5.0322683819103986e-05), (26, 2.8592334274435416e-05), (27, 9.263922947866376e-06)]
+
+  6%|▌         | 9/150 [02:26<21:50,  9.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 840, Spatial MSE: [(0, 5.945856628386537e-06), (1, 0.00015248377167154104), (2, 0.0024634734727442265), (3, 0.00021084744366817176), (4, 9.774259524419904e-05), (5, 7.421204645652324e-05), (6, 6.901680171722546e-05), (7, 0.0001452497235732153), (8, 8.252004045061767e-05), (9, 0.00012524978956207633), (10, 0.00015464461466763169), (11, 0.00020649332145694643), (12, 0.00031877291621640325), (13, 0.0008048273739404976), (14, 0.0029649799689650536), (15, 0.0033815125934779644), (16, 0.028745383024215698), (17, 0.0009730045567266643), (18, 0.0009867397602647543), (19, 0.0027736532501876354), (20, 0.002673622453585267), (21, 0.0011381694348528981), (22, 0.0015506744384765625), (23, 0.0012652864679694176), (24, 0.0026172336656600237), (25, 0.00828491896390915), (26, 0.0037060617469251156), (27, 0.029772009700536728)], Temporal MSE: [(0, 7.959820322867017e-06), (1, 1.1993087355222087e-05), (2, 1.534957846160978e-05), (3, 1.295321635552682e-05), (4, 1.1756886124203447e-05), (5, 1.7302125343121588e-05), (6, 9.062146091309842e-06), (7, 1.5783924027346075e-05), (8, 9.848627996689174e-06), (9, 2.0551817215164192e-05), (10, 5.3342348110163584e-05), (11, 4.39871619164478e-05), (12, 5.4125775932334363e-05), (13, 0.00017082734848372638), (14, 0.00014026169083081186), (15, 0.00017994159134104848), (16, 0.0001160101528512314), (17, 0.0003223750100005418), (18, 0.0001607787562534213), (19, 0.00021622702479362488), (20, 0.0004953340394422412), (21, 0.00021921929146628827), (22, 0.00018591423577163368), (23, 0.00024645827943459153), (24, 0.0004635558871086687), (25, 0.0008794922614470124), (26, 9.222104563377798e-05), (27, 0.00010617545194691047)]
+
+  7%|▋         | 10/150 [02:35<21:38,  9.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 834, Spatial MSE: [(0, 1.0643281711963937e-05), (1, 7.689762423979118e-05), (2, 0.0023405798710882664), (3, 0.00035715053672902286), (4, 8.120094571495429e-05), (5, 4.416964657139033e-05), (6, 6.353442586259916e-05), (7, 4.76600653200876e-05), (8, 3.7742436688859016e-05), (9, 4.262410948285833e-05), (10, 3.802157152676955e-05), (11, 6.873194797663018e-05), (12, 9.78371172095649e-05), (13, 0.0014701292384415865), (14, 0.0030139631126075983), (15, 0.0010432120179757476), (16, 0.04007202759385109), (17, 0.0003059387963730842), (18, 0.0010151657043024898), (19, 0.0022973723243921995), (20, 0.0006506161880679429), (21, 0.0014307398814707994), (22, 0.002178617287427187), (23, 0.0011090958723798394), (24, 0.002061043633148074), (25, 0.0050354874692857265), (26, 0.006187552120536566), (27, 0.07321403175592422)], Temporal MSE: [(0, 1.4332173122966196e-05), (1, 9.265662811230868e-06), (2, 1.3734267668041866e-05), (3, 1.3141781892045401e-05), (4, 1.0474829650775064e-05), (5, 1.1963990800722968e-05), (6, 6.125027539383154e-06), (7, 5.837219759996515e-06), (8, 5.2348250392242335e-06), (9, 6.766553269699216e-06), (10, 1.0328560165362433e-05), (11, 3.2701482268748805e-05), (12, 3.959885361837223e-05), (13, 0.00017224845942109823), (14, 4.812499901163392e-05), (15, 5.453222547657788e-05), (16, 6.004555689287372e-05), (17, 8.757948671700433e-05), (18, 4.958811041433364e-05), (19, 5.077311652712524e-05), (20, 6.816500535933301e-05), (21, 9.106507786782458e-05), (22, 0.0002694547292776406), (23, 0.0003850004286505282), (24, 0.0005007021827623248), (25, 0.00021678382472600788), (26, 0.00012418274127412587), (27, 3.614440356614068e-05)]
+
+  7%|▋         | 11/150 [02:46<22:45,  9.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 828, Spatial MSE: [(0, 1.1449600606283639e-05), (1, 0.0002338214690098539), (2, 0.00352583359926939), (3, 0.0002895286597777158), (4, 0.00017461430979892612), (5, 0.00011976020323345438), (6, 7.116076449165121e-05), (7, 0.00014709883544128388), (8, 9.648984996601939e-05), (9, 0.00010917156760115176), (10, 0.00015376471856143326), (11, 0.00017415810725651681), (12, 0.0003691696038004011), (13, 0.0017830012366175652), (14, 0.0043143704533576965), (15, 0.0030957823619246483), (16, 0.05053635686635971), (17, 0.001019819756038487), (18, 0.001399899017997086), (19, 0.003432034980505705), (20, 0.0017686558421701193), (21, 0.0019064925145357847), (22, 0.0029007450211793184), (23, 0.0020267870277166367), (24, 0.005054960027337074), (25, 0.010460483841598034), (26, 0.007644227705895901), (27, 0.04323020949959755)], Temporal MSE: [(0, 1.3311095244716853e-05), (1, 2.39865985349752e-05), (2, 2.1682326405425556e-05), (3, 1.898065056593623e-05), (4, 2.3698488803347573e-05), (5, 2.7068954295828007e-05), (6, 1.732994132908061e-05), (7, 2.484823744453024e-05), (8, 1.373507529933704e-05), (9, 2.794696956698317e-05), (10, 7.81787239247933e-05), (11, 8.362288645002991e-05), (12, 7.136540079955012e-05), (13, 0.0002600251173134893), (14, 0.0001246200845343992), (15, 0.00022430229000747204), (16, 0.00013651553308591247), (17, 0.0003362775023560971), (18, 0.00015332031762227416), (19, 0.00022578502830583602), (20, 0.0005775702302344143), (21, 0.00024418585235252976), (22, 0.00044616765808314085), (23, 0.0005697825690731406), (24, 0.0007927555707283318), (25, 0.000632373325061053), (26, 0.0001838646421674639), (27, 0.0002709174295887351)]
+
+  8%|▊         | 12/150 [02:55<22:02,  9.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 822, Spatial MSE: [(0, 1.4194911273079924e-05), (1, 0.00010245062003377825), (2, 0.002880196087062359), (3, 0.0004929875140078366), (4, 0.00010913436562987044), (5, 5.673113264492713e-05), (6, 0.00010352142271585763), (7, 5.547876571654342e-05), (8, 4.573139813146554e-05), (9, 5.5345790315186605e-05), (10, 5.1362159865675494e-05), (11, 0.00010226815356872976), (12, 0.0001725496695144102), (13, 0.0030531499069184065), (14, 0.0054460130631923676), (15, 0.0015927520580589771), (16, 0.06029363349080086), (17, 0.0004025254456792027), (18, 0.0013789692893624306), (19, 0.003627433907240629), (20, 0.0010769448708742857), (21, 0.0017210482619702816), (22, 0.003001259174197912), (23, 0.00148522830568254), (24, 0.0028088793624192476), (25, 0.007136897183954716), (26, 0.008558432571589947), (27, 0.11081662774085999)], Temporal MSE: [(0, 1.888853330456186e-05), (1, 1.2381980923237279e-05), (2, 1.6645328287268057e-05), (3, 1.6328704077750444e-05), (4, 1.3820469575875904e-05), (5, 1.6239355318248272e-05), (6, 7.846770131436642e-06), (7, 8.353677912964486e-06), (8, 6.960370683373185e-06), (9, 9.041013981914148e-06), (10, 1.3343033970159013e-05), (11, 6.32518858765252e-05), (12, 7.656215893803164e-05), (13, 0.00031274292268790305), (14, 7.409812678815797e-05), (15, 8.022723341127858e-05), (16, 8.597721898695454e-05), (17, 0.00011421868839534), (18, 7.602653931826353e-05), (19, 7.408133387798443e-05), (20, 9.674138709669933e-05), (21, 0.0001320057053817436), (22, 0.00037439411971718073), (23, 0.0005644022021442652), (24, 0.0007177878869697452), (25, 0.0003130209806840867), (26, 0.00017780940106604248), (27, 5.0155551434727386e-05)]
+
+  9%|▊         | 13/150 [03:04<21:03,  9.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 816, Spatial MSE: [(0, 1.408839216310298e-05), (1, 0.00032716873101890087), (2, 0.004279074724763632), (3, 0.00035807202220894396), (4, 0.0001939656212925911), (5, 0.00018696436018217355), (6, 0.00016848687664605677), (7, 0.0006131173577159643), (8, 0.0003936343127861619), (9, 0.0007812064723111689), (10, 0.0008842412498779595), (11, 0.0012045333860442042), (12, 0.0019099822966381907), (13, 0.00448333565145731), (14, 0.010337669402360916), (15, 0.01159538235515356), (16, 0.07100783288478851), (17, 0.005017152987420559), (18, 0.00479874387383461), (19, 0.007585331331938505), (20, 0.010878967121243477), (21, 0.004356720484793186), (22, 0.006247254088521004), (23, 0.005251727066934109), (24, 0.012799669988453388), (25, 0.05818992108106613), (26, 0.025538939982652664), (27, 0.06704011559486389)], Temporal MSE: [(0, 2.1605039364658296e-05), (1, 3.458336504991166e-05), (2, 4.216088927933015e-05), (3, 3.1871673854766414e-05), (4, 2.9556806111941114e-05), (5, 6.008459968143143e-05), (6, 3.847887273877859e-05), (7, 6.031242446624674e-05), (8, 6.305240094661713e-05), (9, 0.00011354307207511738), (10, 0.0002660804311744869), (11, 0.0002512573846615851), (12, 0.0003198370977770537), (13, 0.0008537264657206833), (14, 0.0008793049491941929), (15, 0.0009094756096601486), (16, 0.0007397219887934625), (17, 0.0015648059779778123), (18, 0.0010928455740213394), (19, 0.0012099802261218429), (20, 0.0022438184823840857), (21, 0.0013311917427927256), (22, 0.0008144564344547689), (23, 0.001184228342026472), (24, 0.0030414643697440624), (25, 0.008260070346295834), (26, 0.0009081067400984466), (27, 0.00044790582614950836)]
+
+  9%|▉         | 14/150 [03:12<20:10,  8.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 810, Spatial MSE: [(0, 1.914062158903107e-05), (1, 0.0001347353681921959), (2, 0.0033875356893986464), (3, 0.0006230829167179763), (4, 0.00013081333599984646), (5, 7.039821502985433e-05), (6, 0.0001474337768740952), (7, 7.36043366487138e-05), (8, 5.739450716646388e-05), (9, 7.167409785324708e-05), (10, 6.729205051669851e-05), (11, 0.00013779723667539656), (12, 0.00024791486794129014), (13, 0.0046845972537994385), (14, 0.0089559406042099), (15, 0.0028998737689107656), (16, 0.10893934220075607), (17, 0.0005371454171836376), (18, 0.0017583498265594244), (19, 0.006301221903413534), (20, 0.0019215953070670366), (21, 0.002401353558525443), (22, 0.004554727114737034), (23, 0.002213943749666214), (24, 0.0038759273011237383), (25, 0.010249766521155834), (26, 0.010131648741662502), (27, 0.14801211655139923)], Temporal MSE: [(0, 2.517299617466051e-05), (1, 1.5328989320551045e-05), (2, 2.0498851881711744e-05), (3, 1.9185594283044338e-05), (4, 1.6914611478568986e-05), (5, 2.0017820133944042e-05), (6, 1.019963929138612e-05), (7, 1.0202136763837188e-05), (8, 8.86369525687769e-06), (9, 1.1844302207464352e-05), (10, 1.8019049093709327e-05), (11, 9.562056948198006e-05), (12, 0.00010717837722040713), (13, 0.00044598066597245634), (14, 0.00012205672828713432), (15, 0.00012724410044029355), (16, 0.0001340489397989586), (17, 0.00016111200966406614), (18, 0.00013377430150285363), (19, 0.00011376841575838625), (20, 0.00013459377805702388), (21, 0.00020807370310649276), (22, 0.0005644412594847381), (23, 0.0009564401116222143), (24, 0.0011536303209140897), (25, 0.00043766474118456244), (26, 0.0002682501799426973), (27, 7.311849913094193e-05)]
+
+ 10%|█         | 15/150 [03:19<19:08,  8.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 804, Spatial MSE: [(0, 1.3829253475705627e-05), (1, 0.0003041765885427594), (2, 0.00393252819776535), (3, 0.00031443662010133266), (4, 0.00017834494065027684), (5, 0.00013952258450444788), (6, 9.396996756549925e-05), (7, 0.0002111851063091308), (8, 0.00014617580745834857), (9, 0.0002979645796585828), (10, 0.00037458923179656267), (11, 0.0004742494784295559), (12, 0.0007377121364697814), (13, 0.005498718004673719), (14, 0.011863158084452152), (15, 0.005159873981028795), (16, 0.12670469284057617), (17, 0.001278031268157065), (18, 0.0021437888499349356), (19, 0.008259736932814121), (20, 0.0038566479925066233), (21, 0.00291442614980042), (22, 0.005285084713250399), (23, 0.0032086214050650597), (24, 0.008092653937637806), (25, 0.01764065772294998), (26, 0.010347454808652401), (27, 0.07103515416383743)], Temporal MSE: [(0, 1.5025001630419865e-05), (1, 2.7328751457389444e-05), (2, 2.1796215150970966e-05), (3, 1.9697614334290847e-05), (4, 2.792176564980764e-05), (5, 3.23157619277481e-05), (6, 2.3193395463749766e-05), (7, 2.8725387892336585e-05), (8, 2.203166877734475e-05), (9, 5.202351894695312e-05), (10, 9.083199984161183e-05), (11, 0.00014837166236247867), (12, 0.00016022972704377025), (13, 0.0005803854437544942), (14, 0.00023018373758532107), (15, 0.0003134600701741874), (16, 0.00025475837173871696), (17, 0.0004931662115268409), (18, 0.0003129196120426059), (19, 0.00041503689135424793), (20, 0.001066087163053453), (21, 0.0005482207052409649), (22, 0.0007318304851651192), (23, 0.0012006411561742425), (24, 0.0016137077473104), (25, 0.0014629759825766087), (26, 0.0003658277273643762), (27, 0.0002964594750665128)]
+
+ 11%|█         | 16/150 [03:27<18:15,  8.18s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 798, Spatial MSE: [(0, 1.7943424609256908e-05), (1, 0.00013005109212826937), (2, 0.0030333632603287697), (3, 0.0005703092901967466), (4, 0.00011002177052432671), (5, 6.171548739075661e-05), (6, 0.00012418984260875732), (7, 6.47266351734288e-05), (8, 4.982551763532683e-05), (9, 6.43875464447774e-05), (10, 6.639715866185725e-05), (11, 0.0001420684129698202), (12, 0.00030542229069396853), (13, 0.006633174140006304), (14, 0.011676446534693241), (15, 0.0034124855883419514), (16, 0.10773066431283951), (17, 0.0005184763576835394), (18, 0.0015673759626224637), (19, 0.006716915871948004), (20, 0.002215810352936387), (21, 0.002254991326481104), (22, 0.004258506000041962), (23, 0.0021021408028900623), (24, 0.0036631899420171976), (25, 0.009526576846837997), (26, 0.010416880249977112), (27, 0.15702348947525024)], Temporal MSE: [(0, 2.341231629543472e-05), (1, 1.3657619092555251e-05), (2, 1.7616544937482104e-05), (3, 1.6481692000525072e-05), (4, 1.3660564945894293e-05), (5, 1.775038799678441e-05), (6, 9.054640941030812e-06), (7, 8.591708137828391e-06), (8, 7.663797077839263e-06), (9, 1.0772963833005633e-05), (10, 1.726563277770765e-05), (11, 0.00011974878725595772), (12, 0.00013961654622107744), (13, 0.0005759592168033123), (14, 0.00013610513997264206), (15, 0.00014416960766538978), (16, 0.00012609083205461502), (17, 0.00014734858996234834), (18, 0.00014924067363608629), (19, 0.00012068204523529857), (20, 0.00015639979392290115), (21, 0.00020766790839843452), (22, 0.0005260382313281298), (23, 0.0009032828966155648), (24, 0.0011409142753109336), (25, 0.0004752888926304877), (26, 0.0002773291780613363), (27, 9.263617539545521e-05)]
+
+ 11%|█▏        | 17/150 [03:34<17:35,  7.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 792, Spatial MSE: [(0, 1.6802867321530357e-05), (1, 0.0003991503035649657), (2, 0.004539027344435453), (3, 0.00036708705010823905), (4, 0.00016725731256883591), (5, 0.00020328268874436617), (6, 0.0002848553704097867), (7, 0.0007519671344198287), (8, 0.0005587961059063673), (9, 0.0010575576452538371), (10, 0.0012406256282702088), (11, 0.0015322503168135881), (12, 0.0024449704214930534), (13, 0.009134399704635143), (14, 0.017741158604621887), (15, 0.010607698000967503), (16, 0.1381019949913025), (17, 0.004969875793904066), (18, 0.005221834406256676), (19, 0.012020955793559551), (20, 0.013782084919512272), (21, 0.0045866332948207855), (22, 0.007768684066832066), (23, 0.006479423027485609), (24, 0.013954455964267254), (25, 0.08146113902330399), (26, 0.038375139236450195), (27, 0.10127412527799606)], Temporal MSE: [(0, 2.2667985831503756e-05), (1, 4.2131960071856156e-05), (2, 4.393620838527568e-05), (3, 3.412980731809512e-05), (4, 3.15053403028287e-05), (5, 6.68445136398077e-05), (6, 5.077162495581433e-05), (7, 7.477198960259557e-05), (8, 8.870918827597052e-05), (9, 0.00014315136650111526), (10, 0.00027707451954483986), (11, 0.0003804535372182727), (12, 0.00041605523438192904), (13, 0.001107226125895977), (14, 0.0008749578264541924), (15, 0.0007630274049006402), (16, 0.0006915776175446808), (17, 0.0011857133358716965), (18, 0.00110309396404773), (19, 0.0010698989499360323), (20, 0.0017006370471790433), (21, 0.0017230429220944643), (22, 0.0011925555299967527), (23, 0.0018033906817436218), (24, 0.004435487557202578), (25, 0.013391987420618534), (26, 0.00162564089987427), (27, 0.0004904245142824948)]
+
+ 12%|█▏        | 18/150 [03:43<18:08,  8.25s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 786, Spatial MSE: [(0, 2.2628155420534313e-05), (1, 0.00017238031432498246), (2, 0.00345631199888885), (3, 0.0007420462789013982), (4, 0.00013081802171655), (5, 7.366726640611887e-05), (6, 0.00018457350961398333), (7, 7.309002103284001e-05), (8, 6.098698213463649e-05), (9, 8.093362703220919e-05), (10, 7.906038808869198e-05), (11, 0.00016404141206294298), (12, 0.0003453409590292722), (13, 0.008180896751582623), (14, 0.014506365172564983), (15, 0.004506621975451708), (16, 0.13855542242527008), (17, 0.0005731304408982396), (18, 0.001612398773431778), (19, 0.0085587864741683), (20, 0.002925619250163436), (21, 0.0023842304944992065), (22, 0.005073109176009893), (23, 0.00249229627661407), (24, 0.004194175358861685), (25, 0.01080551091581583), (26, 0.011236376129090786), (27, 0.19359634816646576)], Temporal MSE: [(0, 2.8538946935441345e-05), (1, 1.603854980203323e-05), (2, 2.0138193576713093e-05), (3, 1.874150984804146e-05), (4, 1.59782575792633e-05), (5, 2.183433753089048e-05), (6, 1.1261745385127142e-05), (7, 1.113951930165058e-05), (8, 9.462194611842278e-06), (9, 1.3679941730515566e-05), (10, 2.2014588466845453e-05), (11, 0.0001332470856141299), (12, 0.0001567933359183371), (13, 0.0007167144794948399), (14, 0.0001767075009411201), (15, 0.00016680278349667788), (16, 0.00015946704661473632), (17, 0.00016694641089998186), (18, 0.0001913442392833531), (19, 0.00014311083941720426), (20, 0.00016493182920385152), (21, 0.0002484045398887247), (22, 0.0006197043694555759), (23, 0.0011489172466099262), (24, 0.00144458026625216), (25, 0.0006446646875701845), (26, 0.00035771451075561345), (27, 0.00013144980766810477)]
+
+ 13%|█▎        | 19/150 [03:50<17:28,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 780, Spatial MSE: [(0, 2.1745883714174852e-05), (1, 0.0004562143294606358), (2, 0.005949029233306646), (3, 0.0005394567851908505), (4, 0.00019406821229495108), (5, 0.00019184949633199722), (6, 0.00018686093972064555), (7, 0.0002535407547838986), (8, 0.00018019747221842408), (9, 0.00023627291375305504), (10, 0.0002789636782836169), (11, 0.00035165983717888594), (12, 0.0006591947167180479), (13, 0.00874191801995039), (14, 0.015676159411668777), (15, 0.0054442440159618855), (16, 0.1408468782901764), (17, 0.001174910576082766), (18, 0.002285708673298359), (19, 0.009378737770020962), (20, 0.003899394301697612), (21, 0.0030371223110705614), (22, 0.006096473895013332), (23, 0.0036465926095843315), (24, 0.008069918490946293), (25, 0.022594252601265907), (26, 0.014129582792520523), (27, 0.13898137211799622)], Temporal MSE: [(0, 2.604492510727141e-05), (1, 4.0114697185344994e-05), (2, 3.3007978345267475e-05), (3, 2.868999399652239e-05), (4, 3.4199900255771354e-05), (5, 4.257784166838974e-05), (6, 3.388170080143027e-05), (7, 3.747043592738919e-05), (8, 3.205138273187913e-05), (9, 5.482581036631018e-05), (10, 9.251401934307069e-05), (11, 0.00021504538017325103), (12, 0.00022798997815698385), (13, 0.0008182205492630601), (14, 0.00026910522137768567), (15, 0.0003701777895912528), (16, 0.00029339539469219744), (17, 0.0005716198938898742), (18, 0.0004406227672006935), (19, 0.0005836605560034513), (20, 0.0012900762958452106), (21, 0.0007507877307944), (22, 0.000832780497148633), (23, 0.0015028533525764942), (24, 0.002199991373345256), (25, 0.0032287854701280594), (26, 0.00046054396079853177), (27, 0.0003336916270200163)]
+
+ 13%|█▎        | 20/150 [03:58<17:06,  7.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 774, Spatial MSE: [(0, 3.3357791835442185e-05), (1, 0.00023479647643398494), (2, 0.004246587865054607), (3, 0.00095993687864393), (4, 0.0001649070909479633), (5, 9.74513022811152e-05), (6, 0.00024202954955399036), (7, 9.122660412685946e-05), (8, 7.970583828864619e-05), (9, 0.00010914685117313638), (10, 0.00011079036630690098), (11, 0.00027606324874795973), (12, 0.0006565938820131123), (13, 0.014911500737071037), (14, 0.021887468174099922), (15, 0.006173912901431322), (16, 0.1931161731481552), (17, 0.0007941899239085615), (18, 0.0022590765729546547), (19, 0.012244513258337975), (20, 0.00426021683961153), (21, 0.0035272934474051), (22, 0.0068985107354819775), (23, 0.0035372544080018997), (24, 0.005976095329970121), (25, 0.015151835978031158), (26, 0.01752236671745777), (27, 0.2856459617614746)], Temporal MSE: [(0, 4.231626371620223e-05), (1, 2.2391206584870815e-05), (2, 2.7473768568597734e-05), (3, 2.5199960873578675e-05), (4, 2.0390782083268277e-05), (5, 2.9816328606102616e-05), (6, 1.5147173144214321e-05), (7, 1.4603650924982503e-05), (8, 1.3158256479073316e-05), (9, 1.8627250028657727e-05), (10, 3.091899634455331e-05), (11, 0.00035595218650996685), (12, 0.0003197566547896713), (13, 0.001004620105959475), (14, 0.0002475054352544248), (15, 0.0002400463417870924), (16, 0.0002261282061226666), (17, 0.00021325619309209287), (18, 0.0002830347802955657), (19, 0.00021038424165453762), (20, 0.00022993952734395862), (21, 0.00034024647902697325), (22, 0.0008425170090049505), (23, 0.0016277719987556338), (24, 0.002028831746429205), (25, 0.0009851927170529962), (26, 0.0005336542380973697), (27, 0.00019844052440021187)]
+
+ 14%|█▍        | 21/150 [04:05<16:15,  7.56s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 768, Spatial MSE: [(0, 3.065489363507368e-05), (1, 0.0007630661712028086), (2, 0.00832965224981308), (3, 0.0007073619053699076), (4, 0.00024566243519075215), (5, 0.00036668061511591077), (6, 0.000834482372738421), (7, 0.0012507822830229998), (8, 0.001250305213034153), (9, 0.002120859222486615), (10, 0.0022297408431768417), (11, 0.002407437888905406), (12, 0.0030871150083839893), (13, 0.018063100054860115), (14, 0.026487674564123154), (15, 0.011870631948113441), (16, 0.21065251529216766), (17, 0.0047193472273647785), (18, 0.006723908707499504), (19, 0.017106853425502777), (20, 0.014359145425260067), (21, 0.00697618443518877), (22, 0.010811843909323215), (23, 0.008509390987455845), (24, 0.01658600941300392), (25, 0.10677578300237656), (26, 0.06020757183432579), (27, 0.20375223457813263)], Temporal MSE: [(0, 3.875099355354905e-05), (1, 6.550378748215735e-05), (2, 6.585106166312471e-05), (3, 5.686980512109585e-05), (4, 5.4130527132656425e-05), (5, 9.932221291819587e-05), (6, 0.00012090574455214664), (7, 0.00015252735465765), (8, 0.00020190993382129818), (9, 0.00032738802838139236), (10, 0.0005017138901166618), (11, 0.0007788324146531522), (12, 0.0007401794428005815), (13, 0.001616587513126433), (14, 0.0012612763093784451), (15, 0.0011157121043652296), (16, 0.0009376263478770852), (17, 0.001051117549650371), (18, 0.0012472919188439846), (19, 0.0011632435489445925), (20, 0.0022042531054466963), (21, 0.0024444819428026676), (22, 0.0018515844130888581), (23, 0.0029769104439765215), (24, 0.006885434966534376), (25, 0.02392624132335186), (26, 0.003105729818344116), (27, 0.0005077553796581924)]
+
+ 15%|█▍        | 22/150 [04:13<16:22,  7.67s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 762, Spatial MSE: [(0, 4.1889998101396486e-05), (1, 0.0003213780000805855), (2, 0.004475751426070929), (3, 0.001165816094726324), (4, 0.000182490490260534), (5, 0.00010759851284092292), (6, 0.0003544297069311142), (7, 9.251453593606129e-05), (8, 9.312873589806259e-05), (9, 0.0001257923140656203), (10, 0.00013355357805266976), (11, 0.00031557268812321126), (12, 0.0007715094834566116), (13, 0.01814056560397148), (14, 0.025067254900932312), (15, 0.0071273003704845905), (16, 0.22271865606307983), (17, 0.0008302390342578292), (18, 0.0020682429894804955), (19, 0.014235119335353374), (20, 0.005191848613321781), (21, 0.0034270216710865498), (22, 0.007170261815190315), (23, 0.0038054771721363068), (24, 0.006372946780174971), (25, 0.014926859177649021), (26, 0.01911851391196251), (27, 0.3311683237552643)], Temporal MSE: [(0, 4.9233363824896514e-05), (1, 2.4154231141437776e-05), (2, 2.8275004297029227e-05), (3, 2.5916235244949348e-05), (4, 2.1491605366463773e-05), (5, 3.440313957980834e-05), (6, 1.8778637240757234e-05), (7, 1.697411607892718e-05), (8, 1.5133066881389823e-05), (9, 2.2760883439332247e-05), (10, 3.821142308879644e-05), (11, 0.0004189831961411983), (12, 0.0003776171070057899), (13, 0.0010804318590089679), (14, 0.00029037229251116514), (15, 0.00027671604766510427), (16, 0.00024413787468802184), (17, 0.00021148512314539403), (18, 0.0003532722475938499), (19, 0.0002495521039236337), (20, 0.00025276359519921243), (21, 0.00037006198544986546), (22, 0.0008820035145618021), (23, 0.0018085342599079013), (24, 0.0023085996508598328), (25, 0.0014287158846855164), (26, 0.0006512879044748843), (27, 0.00026695526321418583)]
+
+ 15%|█▌        | 23/150 [04:20<16:11,  7.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 756, Spatial MSE: [(0, 3.860111610265449e-05), (1, 0.0007250462076626718), (2, 0.007954485714435577), (3, 0.0008086170419119298), (4, 0.00019686835003085434), (5, 0.00023552011407446116), (6, 0.00040943335625343025), (7, 0.0002595309051685035), (8, 0.0002798614150378853), (9, 0.0003828730550594628), (10, 0.00040220163646154106), (11, 0.0005711946869269013), (12, 0.0010903584770858288), (13, 0.018674219027161598), (14, 0.02641437202692032), (15, 0.008548767305910587), (16, 0.2292976826429367), (17, 0.0012978389859199524), (18, 0.003258555196225643), (19, 0.015287344343960285), (20, 0.006678634788841009), (21, 0.005355552304536104), (22, 0.008592499420046806), (23, 0.004275075625628233), (24, 0.01033106166869402), (25, 0.02876737341284752), (26, 0.01854374073445797), (27, 0.22171683609485626)], Temporal MSE: [(0, 4.131136665819213e-05), (1, 5.324872108758427e-05), (2, 4.369978705653921e-05), (3, 3.7915426219115034e-05), (4, 3.880091026076116e-05), (5, 5.306374077918008e-05), (6, 4.62401621916797e-05), (7, 4.508335405262187e-05), (8, 4.7772075049579144e-05), (9, 5.91844400332775e-05), (10, 8.747672836761922e-05), (11, 0.0005174065008759499), (12, 0.0004685091262217611), (13, 0.0011814301833510399), (14, 0.00039352959720417857), (15, 0.0004175382782705128), (16, 0.000365688611054793), (17, 0.0005594739923253655), (18, 0.0006360071711242199), (19, 0.0007415850413963199), (20, 0.0010630942415446043), (21, 0.0009350751060992479), (22, 0.0011409152066335082), (23, 0.002300987020134926), (24, 0.003411490237340331), (25, 0.0066961003467440605), (26, 0.0007781493477523327), (27, 0.0004507033445406705)]
+
+ 16%|█▌        | 24/150 [04:29<16:36,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 750, Spatial MSE: [(0, 4.849590914091095e-05), (1, 0.00037321861600503325), (2, 0.004164919722825289), (3, 0.0012063657632097602), (4, 0.0001767075009411201), (5, 0.00010909008415183052), (6, 0.00045192177640274167), (7, 9.099488670472056e-05), (8, 9.799661347642541e-05), (9, 0.00012594618601724505), (10, 0.00013277059770189226), (11, 0.0003075714921578765), (12, 0.0008075309451669455), (13, 0.021116551011800766), (14, 0.02682468853890896), (15, 0.007320778910070658), (16, 0.2128351479768753), (17, 0.0008150599314831197), (18, 0.0018269529100507498), (19, 0.01442814338952303), (20, 0.00558797363191843), (21, 0.002920110709965229), (22, 0.006323208566755056), (23, 0.0034568733535706997), (24, 0.005814359989017248), (25, 0.013313883915543556), (26, 0.01840459182858467), (27, 0.34646108746528625)], Temporal MSE: [(0, 5.67271672480274e-05), (1, 2.5838007786660455e-05), (2, 2.9268017897265963e-05), (3, 2.567611045378726e-05), (4, 2.0111992853344418e-05), (5, 3.534975621732883e-05), (6, 2.0970875993953086e-05), (7, 1.868805702542886e-05), (8, 1.571364191477187e-05), (9, 2.4042608856689185e-05), (10, 3.843885860987939e-05), (11, 0.0004133324546273798), (12, 0.0004180253017693758), (13, 0.001092907739803195), (14, 0.0002956813550554216), (15, 0.00029371140408329666), (16, 0.00023220678849611431), (17, 0.0001910183927975595), (18, 0.00037293104105629027), (19, 0.00026316504226997495), (20, 0.0002501661074347794), (21, 0.0003669203433673829), (22, 0.0008148240740410984), (23, 0.001697059371508658), (24, 0.0021963436156511307), (25, 0.0017156152753159404), (26, 0.0006531375111080706), (27, 0.00032577195088379085)]
+
+ 17%|█▋        | 25/150 [04:36<16:10,  7.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 744, Spatial MSE: [(0, 3.810840644291602e-05), (1, 0.0008244133787229657), (2, 0.007893223315477371), (3, 0.0006464807083830237), (4, 0.0002843729453161359), (5, 0.0004070865688845515), (6, 0.001171280164271593), (7, 0.0013274955563247204), (8, 0.0012761661782860756), (9, 0.0018201059428974986), (10, 0.0019474680302664638), (11, 0.0021364805288612843), (12, 0.0029452263843268156), (13, 0.02377355843782425), (14, 0.03166944533586502), (15, 0.013208378106355667), (16, 0.233247771859169), (17, 0.004467952996492386), (18, 0.00599154457449913), (19, 0.018345393240451813), (20, 0.013397119008004665), (21, 0.008732222951948643), (22, 0.009524449706077576), (23, 0.006966753862798214), (24, 0.01573874056339264), (25, 0.074385866522789), (26, 0.05813474580645561), (27, 0.2263980209827423)], Temporal MSE: [(0, 4.063064989168197e-05), (1, 6.7823966674041e-05), (2, 5.854292612639256e-05), (3, 5.8178742619929835e-05), (4, 7.137679494917393e-05), (5, 0.0001623787684366107), (6, 0.0001937685301527381), (7, 0.0002332034200662747), (8, 0.00026909622829407454), (9, 0.000385854160413146), (10, 0.0006227870471775532), (11, 0.0008714692667126656), (12, 0.00084820477059111), (13, 0.0017527244053781033), (14, 0.0014035787899047136), (15, 0.0012238819617778063), (16, 0.0010069073177874088), (17, 0.001141274580731988), (18, 0.0012894651154056191), (19, 0.0013800858287140727), (20, 0.002684476785361767), (21, 0.002443719422444701), (22, 0.0018040887080132961), (23, 0.0030146592762321234), (24, 0.007488694041967392), (25, 0.029158517718315125), (26, 0.003932459279894829), (27, 0.00045209378004074097)]
+
+ 17%|█▋        | 26/150 [04:44<15:59,  7.74s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 738, Spatial MSE: [(0, 3.796293822233565e-05), (1, 0.0003139072796329856), (2, 0.0026804169174283743), (3, 0.0008309063850902021), (4, 0.00012504853657446802), (5, 7.782785542076454e-05), (6, 0.00038149263127706945), (7, 6.800103437853977e-05), (8, 7.37131413188763e-05), (9, 9.085995407076553e-05), (10, 0.0001090506193577312), (11, 0.00024131279496941715), (12, 0.0006327431765384972), (13, 0.016585052013397217), (14, 0.02013942040503025), (15, 0.005923219956457615), (16, 0.1856377273797989), (17, 0.0006561483605764806), (18, 0.001279702759347856), (19, 0.012245838530361652), (20, 0.0048476350493729115), (21, 0.0024675163440406322), (22, 0.004922679625451565), (23, 0.0025423632469028234), (24, 0.004326428286731243), (25, 0.009672342799603939), (26, 0.012855499982833862), (27, 0.2613356113433838)], Temporal MSE: [(0, 4.258419357938692e-05), (1, 1.802417500584852e-05), (2, 2.0234230760252103e-05), (3, 1.7685844795778394e-05), (4, 1.386608710163273e-05), (5, 2.7250413040746935e-05), (6, 1.870552296168171e-05), (7, 1.4173689123708755e-05), (8, 1.2945120943186339e-05), (9, 2.0824450984946452e-05), (10, 3.072834442718886e-05), (11, 0.0003505306667648256), (12, 0.0003212641167920083), (13, 0.0007747096242383122), (14, 0.00024435046361759305), (15, 0.0002654570562299341), (16, 0.00019288872135803103), (17, 0.00015117409930098802), (18, 0.0003366213059052825), (19, 0.00023293208505492657), (20, 0.00020745520305354148), (21, 0.00030212299316190183), (22, 0.0006479414296336472), (23, 0.0014266311191022396), (24, 0.0018029060447588563), (25, 0.0017267772927880287), (26, 0.0005479981773532927), (27, 0.00029628106858581305)]
+
+ 18%|█▊        | 27/150 [04:51<15:33,  7.59s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 732, Spatial MSE: [(0, 3.291050234111026e-05), (1, 0.00048607177450321615), (2, 0.004304526839405298), (3, 0.00047468292177654803), (4, 0.00011685478239087388), (5, 0.00011402434029150754), (6, 0.00032135582296177745), (7, 0.0001758417784003541), (8, 0.00027276610489934683), (9, 0.00045432292972691357), (10, 0.0004930611466988921), (11, 0.0006220304640009999), (12, 0.001023788470774889), (13, 0.01704765111207962), (14, 0.021306520327925682), (15, 0.007152720354497433), (16, 0.204363152384758), (17, 0.0009503941982984543), (18, 0.0025108119007200003), (19, 0.014731837436556816), (20, 0.007079394068568945), (21, 0.005875605624169111), (22, 0.006586666218936443), (23, 0.0026450389996170998), (24, 0.0076959202997386456), (25, 0.014276839792728424), (26, 0.00916742067784071), (27, 0.17724938690662384)], Temporal MSE: [(0, 2.963557926705107e-05), (1, 2.7931422664551064e-05), (2, 2.741343450907152e-05), (3, 2.2455868020188063e-05), (4, 2.2168727809912525e-05), (5, 3.22372252412606e-05), (6, 3.1453844712814316e-05), (7, 2.816469168465119e-05), (8, 3.6685789382318035e-05), (9, 4.878583422396332e-05), (10, 7.913736772025004e-05), (11, 0.00040745967999100685), (12, 0.000377538992324844), (13, 0.000836533959954977), (14, 0.0003410729405004531), (15, 0.0003525041101966053), (16, 0.00028030594694428146), (17, 0.0003279585507698357), (18, 0.0005099155241623521), (19, 0.000519771478138864), (20, 0.0005607971106655896), (21, 0.0006033155950717628), (22, 0.0008872271282598376), (23, 0.0018059457652270794), (24, 0.002873337594792247), (25, 0.005889264401048422), (26, 0.0006938956794328988), (27, 0.0002981483412440866)]
+
+ 19%|█▊        | 28/150 [05:00<15:57,  7.85s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 726, Spatial MSE: [(0, 3.0546478228643537e-05), (1, 0.0002388322609476745), (2, 0.0017120966222137213), (3, 0.000529453915078193), (4, 8.016838546609506e-05), (5, 5.1759383495664224e-05), (6, 0.0002925354056060314), (7, 4.780590461450629e-05), (8, 5.015967690269463e-05), (9, 5.911271000513807e-05), (10, 7.794635166646913e-05), (11, 0.00022073199215810746), (12, 0.0005898041417822242), (13, 0.015179083682596684), (14, 0.016849573701620102), (15, 0.005081877112388611), (16, 0.15289849042892456), (17, 0.0004870620905421674), (18, 0.0009203324443660676), (19, 0.010041559115052223), (20, 0.004101849626749754), (21, 0.0019716769456863403), (22, 0.003739706939086318), (23, 0.0018358852248638868), (24, 0.0030671628192067146), (25, 0.0067026857286691666), (26, 0.008867413736879826), (27, 0.2038096934556961)], Temporal MSE: [(0, 3.288132211309858e-05), (1, 1.252254060091218e-05), (2, 1.4023839867149945e-05), (3, 1.1743971299438272e-05), (4, 8.314455044455826e-06), (5, 1.7756099623511545e-05), (6, 1.3720402421313338e-05), (7, 1.0637887498887721e-05), (8, 8.463392077828757e-06), (9, 1.3374461559578776e-05), (10, 2.137069168384187e-05), (11, 0.0003859412099700421), (12, 0.00028671708423644304), (13, 0.0005707242526113987), (14, 0.00019128275744151324), (15, 0.0002115674433298409), (16, 0.00014672285760752857), (17, 0.00010717734403442591), (18, 0.00028193736216053367), (19, 0.00019488796533551067), (20, 0.00017087066953536123), (21, 0.0002564977912697941), (22, 0.0005116395186632872), (23, 0.0011366590624675155), (24, 0.001424237503670156), (25, 0.0016313698142766953), (26, 0.00043924269266426563), (27, 0.0002801674127113074)]
+
+ 19%|█▉        | 29/150 [05:07<15:45,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 720, Spatial MSE: [(0, 2.286882772750687e-05), (1, 0.000427643652074039), (2, 0.003556538140401244), (3, 0.0003476959827821702), (4, 0.0001718791463645175), (5, 0.00025959633057937026), (6, 0.000912938266992569), (7, 0.0008879079832695425), (8, 0.000948405999224633), (9, 0.0012977973092347383), (10, 0.0013222262496128678), (11, 0.0016209026798605919), (12, 0.002185356104746461), (13, 0.01772606186568737), (14, 0.022006146609783173), (15, 0.009955771267414093), (16, 0.19393713772296906), (17, 0.002851324388757348), (18, 0.003602922661229968), (19, 0.015066883526742458), (20, 0.010834929533302784), (21, 0.005710072349756956), (22, 0.0068598175421357155), (23, 0.005027642473578453), (24, 0.011766851879656315), (25, 0.03085358627140522), (26, 0.01997167430818081), (27, 0.14659976959228516)], Temporal MSE: [(0, 2.1421183191705495e-05), (1, 3.126370575046167e-05), (2, 2.6522855478106067e-05), (3, 3.578353062039241e-05), (4, 6.235094770090654e-05), (5, 0.00018673767044674605), (6, 0.00024614977883175015), (7, 0.00018504005856812), (8, 0.0001415738370269537), (9, 0.0002246530493721366), (10, 0.0004312498786021024), (11, 0.0008003231487236917), (12, 0.0006151526467874646), (13, 0.0011074634967371821), (14, 0.0010724621824920177), (15, 0.0010226384038105607), (16, 0.0008661604952067137), (17, 0.0008648450602777302), (18, 0.0010687187314033508), (19, 0.0013845744542777538), (20, 0.0023199832066893578), (21, 0.0017872867174446583), (22, 0.001447174814529717), (23, 0.0024975056294351816), (24, 0.005915199406445026), (25, 0.017640387639403343), (26, 0.00257023423910141), (27, 0.00029100486426614225)]
+
+ 20%|██        | 30/150 [05:15<15:22,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 714, Spatial MSE: [(0, 1.8613696738611907e-05), (1, 0.00015159016766119748), (2, 0.0009275890770368278), (3, 0.00032332484261132777), (4, 4.7459110646741465e-05), (5, 3.0531395168509334e-05), (6, 0.00019961576617788523), (7, 3.245236075599678e-05), (8, 3.0326260457513854e-05), (9, 3.515889693517238e-05), (10, 4.5759836211800575e-05), (11, 0.00010152988397749141), (12, 0.00031515987939201295), (13, 0.009745806455612183), (14, 0.0110849067568779), (15, 0.0033086452167481184), (16, 0.10616102069616318), (17, 0.00036591271054930985), (18, 0.000678084441460669), (19, 0.00734732486307621), (20, 0.003279430326074362), (21, 0.001260867458768189), (22, 0.002423947909846902), (23, 0.0011242434848099947), (24, 0.001955902436748147), (25, 0.004208280239254236), (26, 0.0053754341788589954), (27, 0.13072989881038666)], Temporal MSE: [(0, 2.0657585992012173e-05), (1, 7.263695806614123e-06), (2, 8.387529305764474e-06), (3, 7.044123776722699e-06), (4, 5.018507636123104e-06), (5, 1.135783259087475e-05), (6, 1.018307011690922e-05), (7, 7.066878879413707e-06), (8, 5.409593541116919e-06), (9, 8.608150892541744e-06), (10, 1.1901999641850125e-05), (11, 0.00014532674686051905), (12, 0.00017391472647432238), (13, 0.0003865966573357582), (14, 0.00013762708113063127), (15, 0.00014077620289754122), (16, 0.00010255700181005523), (17, 7.918873598100618e-05), (18, 0.00021347467554733157), (19, 0.0001483824453316629), (20, 0.00012153450370533392), (21, 0.0001904029049910605), (22, 0.000360409845598042), (23, 0.0007687793695367873), (24, 0.0009200868080370128), (25, 0.0011644080514088273), (26, 0.00029779362375847995), (27, 0.00019868661183863878)]
+
+ 21%|██        | 31/150 [05:22<14:54,  7.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 708, Spatial MSE: [(0, 1.4317459317680914e-05), (1, 0.00016576377674937248), (2, 0.0011889871675521135), (3, 0.00012532381515484303), (4, 3.810303678619675e-05), (5, 3.612622822402045e-05), (6, 0.0001064664320438169), (7, 6.106137880124152e-05), (8, 7.844729407224804e-05), (9, 0.00010702268627937883), (10, 0.0001229397748829797), (11, 0.00017200152797158808), (12, 0.0003864668542519212), (13, 0.009840586222708225), (14, 0.011463401839137077), (15, 0.0035988949239253998), (16, 0.11173268407583237), (17, 0.0004473599256016314), (18, 0.0012037964770570397), (19, 0.007827971130609512), (20, 0.0038183412980288267), (21, 0.0019824986811727285), (22, 0.002959222299978137), (23, 0.0012026210315525532), (24, 0.003270980203524232), (25, 0.005845670122653246), (26, 0.003946310840547085), (27, 0.09117496013641357)], Temporal MSE: [(0, 1.0831260624399874e-05), (1, 7.500663741666358e-06), (2, 9.74683825916145e-06), (3, 8.02314025349915e-06), (4, 8.781335964158643e-06), (5, 1.1946413906116504e-05), (6, 1.4759006262465846e-05), (7, 9.70807195699308e-06), (8, 1.2792560482921544e-05), (9, 1.490153226768598e-05), (10, 2.446238613629248e-05), (11, 0.00015728433209005743), (12, 0.00018331063620280474), (13, 0.00039758620550855994), (14, 0.00016560459334868938), (15, 0.00016833312110975385), (16, 0.00012776762014254928), (17, 0.00013229352771304548), (18, 0.0002708858228288591), (19, 0.00024778061197139323), (20, 0.000257226696703583), (21, 0.0002714191796258092), (22, 0.0004547468270175159), (23, 0.0009392942301928997), (24, 0.0014346602838486433), (25, 0.002741603646427393), (26, 0.0003855482500512153), (27, 0.0001835454604588449)]
+
+ 21%|██▏       | 32/150 [05:29<14:37,  7.44s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 702, Spatial MSE: [(0, 8.988918125396594e-06), (1, 6.900073640281335e-05), (2, 0.00041527432040311396), (3, 0.00014143003500066698), (4, 2.105840394506231e-05), (5, 1.3928234693594277e-05), (6, 9.226350812241435e-05), (7, 1.7279158782912418e-05), (8, 1.4137137441139203e-05), (9, 1.6202553524635732e-05), (10, 2.1770294551970437e-05), (11, 4.709701534011401e-05), (12, 0.00015061594604048878), (13, 0.004834039602428675), (14, 0.0058159553445875645), (15, 0.0019260317785665393), (16, 0.06971761584281921), (17, 0.00022507943504024297), (18, 0.00040678775985725224), (19, 0.004604149144142866), (20, 0.002111101755872369), (21, 0.0008547054603695869), (22, 0.0014302721247076988), (23, 0.0006321730324998498), (24, 0.0010707059409469366), (25, 0.002378114266321063), (26, 0.0028137739282101393), (27, 0.0669948011636734)], Temporal MSE: [(0, 9.99621806840878e-06), (1, 3.4267973205714952e-06), (2, 4.104029812879162e-06), (3, 3.4462325402273564e-06), (4, 2.2192145934241125e-06), (5, 5.279594461171655e-06), (6, 4.732018624054035e-06), (7, 4.046984031447209e-06), (8, 2.479648173903115e-06), (9, 3.7874078770983033e-06), (10, 5.735185823141364e-06), (11, 6.036919876351021e-05), (12, 8.067060844041407e-05), (13, 0.00018010300118476152), (14, 7.66500088502653e-05), (15, 9.205994865624234e-05), (16, 5.979573688819073e-05), (17, 5.090163176646456e-05), (18, 0.00013826723443344235), (19, 0.000100360055512283), (20, 8.146173786371946e-05), (21, 0.0001159197636297904), (22, 0.00022079766495153308), (23, 0.00047275025281123817), (24, 0.0005367051344364882), (25, 0.0006828068871982396), (26, 0.00016446482914034277), (27, 0.00011533582437550649)]
+
+ 22%|██▏       | 33/150 [05:37<14:33,  7.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 696, Spatial MSE: [(0, 7.418073892040411e-06), (1, 0.00012931307719554752), (2, 0.0009599925251677632), (3, 0.00012640634668059647), (4, 7.001350604696199e-05), (5, 0.00012865314783994108), (6, 0.0002680153993424028), (7, 0.0005333868903107941), (8, 0.0007703943410888314), (9, 0.0010938887717202306), (10, 0.001322283991612494), (11, 0.0022418107837438583), (12, 0.00247315620072186), (13, 0.008123387582600117), (14, 0.013375740498304367), (15, 0.010818269103765488), (16, 0.1370871216058731), (17, 0.006976548582315445), (18, 0.00618665711954236), (19, 0.01888231933116913), (20, 0.022649336606264114), (21, 0.01731513999402523), (22, 0.006980702746659517), (23, 0.006708587519824505), (24, 0.014250773936510086), (25, 0.027203597128391266), (26, 0.012808977626264095), (27, 0.05569395050406456)], Temporal MSE: [(0, 7.4067011155420914e-06), (1, 9.977313311537728e-06), (2, 8.60671752889175e-06), (3, 1.9470793631626293e-05), (4, 3.6685800296254456e-05), (5, 6.220839713932946e-05), (6, 0.00015829445328563452), (7, 0.00010623344860505313), (8, 0.00010817742440849543), (9, 0.00020061820396222174), (10, 0.00047438094043172896), (11, 0.0005640622694045305), (12, 0.0006205583922564983), (13, 0.0011303020874038339), (14, 0.0015620444901287556), (15, 0.001817202428355813), (16, 0.002140544820576906), (17, 0.0031273318454623222), (18, 0.003164140274748206), (19, 0.006201154086738825), (20, 0.013216173276305199), (21, 0.010736027732491493), (22, 0.0021558378357440233), (23, 0.0032799551263451576), (24, 0.009904082864522934), (25, 0.016575325280427933), (26, 0.0018459541024640203), (27, 0.0001527655404061079)]
+
+ 23%|██▎       | 34/150 [05:45<14:59,  7.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 690, Spatial MSE: [(0, 4.905031801172299e-06), (1, 3.928779551642947e-05), (2, 0.00023235597473103553), (3, 7.351019303314388e-05), (4, 1.1496445040393155e-05), (5, 8.187003004422877e-06), (6, 4.846728188567795e-05), (7, 9.132497325481381e-06), (8, 8.550815437047277e-06), (9, 9.900785698846448e-06), (10, 1.9731525753741153e-05), (11, 7.128932338673621e-05), (12, 0.00021913019008934498), (13, 0.006597684696316719), (14, 0.007572828326374292), (15, 0.002434871857985854), (16, 0.0769035592675209), (17, 0.0002212553663412109), (18, 0.00037115233135409653), (19, 0.005305107217282057), (20, 0.0024976138956844807), (21, 0.0006397383403964341), (22, 0.0013466208474710584), (23, 0.0005954806692898273), (24, 0.0010706669418141246), (25, 0.0022081031929701567), (26, 0.002071956405416131), (27, 0.0668940469622612)], Temporal MSE: [(0, 5.48322850590921e-06), (1, 1.972382506210124e-06), (2, 2.4513594780728454e-06), (3, 2.0055420009157388e-06), (4, 1.275180352422467e-06), (5, 3.165257339787786e-06), (6, 2.915050572482869e-06), (7, 2.234555722679943e-06), (8, 1.4857146197755355e-06), (9, 2.37489302890026e-06), (10, 4.858864485868253e-06), (11, 0.0001396461302647367), (12, 0.0001043100783135742), (13, 0.00022288058244157583), (14, 8.72200180310756e-05), (15, 0.00010800231393659487), (16, 6.0360562201822177e-05), (17, 4.893113873549737e-05), (18, 0.00015769105812069029), (19, 0.0001113019825424999), (20, 8.290683035738766e-05), (21, 0.00011035250645363703), (22, 0.00020927016157656908), (23, 0.0004883856163360178), (24, 0.0005711917765438557), (25, 0.0007452841964550316), (26, 0.00017740066687110811), (27, 0.0001357071741949767)]
+
+ 23%|██▎       | 35/150 [05:52<14:23,  7.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 684, Spatial MSE: [(0, 4.693219580076402e-06), (1, 4.748096762341447e-05), (2, 0.0003436193510424346), (3, 4.439330950845033e-05), (4, 1.5572853953926824e-05), (5, 1.332777083007386e-05), (6, 2.2912317945156246e-05), (7, 2.5309816919616424e-05), (8, 2.9044713301118463e-05), (9, 4.0643062675371766e-05), (10, 6.527452205773443e-05), (11, 0.00011442691902630031), (12, 0.00026487649301998317), (13, 0.006681268569082022), (14, 0.007660340052098036), (15, 0.002483739284798503), (16, 0.07693459838628769), (17, 0.0002579753054305911), (18, 0.0005023921257816255), (19, 0.005347960628569126), (20, 0.002596039092168212), (21, 0.0007294621900655329), (22, 0.0015315732453018427), (23, 0.0006848162156529725), (24, 0.001665536081418395), (25, 0.003154786303639412), (26, 0.0025378011632710695), (27, 0.060201674699783325)], Temporal MSE: [(0, 4.048079972562846e-06), (1, 2.2634403649135493e-06), (2, 3.4200911613879725e-06), (3, 3.296303248134791e-06), (4, 4.030442141811363e-06), (5, 4.674888259614818e-06), (6, 6.086424036766402e-06), (7, 4.147246727370657e-06), (8, 4.533118953986559e-06), (9, 5.96423387833056e-06), (10, 1.1485471986816265e-05), (11, 0.00014866435958538204), (12, 0.00011167503544129431), (13, 0.0002326023968635127), (14, 9.957700967788696e-05), (15, 0.00011858204379677773), (16, 7.138284854590893e-05), (17, 6.941571336938068e-05), (18, 0.00018257684132549912), (19, 0.00014704946079291403), (20, 0.00015016522957012057), (21, 0.00016194161435123533), (22, 0.0002569730277173221), (23, 0.0005733101861551404), (24, 0.0008286871016025543), (25, 0.0014166024047881365), (26, 0.0002412104804534465), (27, 0.00013798614963889122)]
+
+ 24%|██▍       | 36/150 [06:00<14:14,  7.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 678, Spatial MSE: [(0, 4.3069408093288075e-06), (1, 2.826660238497425e-05), (2, 0.00016973093443084508), (3, 6.327314622467384e-05), (4, 9.592196875018999e-06), (5, 6.296497758739861e-06), (6, 4.870233533438295e-05), (7, 1.0065787137136795e-05), (8, 6.9958873609721195e-06), (9, 7.61517958380864e-06), (10, 1.057417830452323e-05), (11, 2.3425616745953448e-05), (12, 8.405024709645659e-05), (13, 0.0032797646708786488), (14, 0.003677339293062687), (15, 0.0012799405958503485), (16, 0.041778139770030975), (17, 0.0001226882159244269), (18, 0.0003715587663464248), (19, 0.0027537280693650246), (20, 0.001292777364142239), (21, 0.0004680867132265121), (22, 0.0008160930010490119), (23, 0.0003537443117238581), (24, 0.000597129634115845), (25, 0.0012578557943925261), (26, 0.0015111970715224743), (27, 0.03721968084573746)], Temporal MSE: [(0, 4.7203780013660435e-06), (1, 1.5246853308781283e-06), (2, 1.8088097704094253e-06), (3, 1.6460911638205289e-06), (4, 1.0389646831754362e-06), (5, 2.5490037387498887e-06), (6, 2.0690315523097524e-06), (7, 2.904251687141368e-06), (8, 1.195122877106769e-06), (9, 1.7704520587358274e-06), (10, 2.6291274934919784e-06), (11, 3.0206176234059967e-05), (12, 5.210192102822475e-05), (13, 0.00011550244380487129), (14, 4.870650445809588e-05), (15, 6.0871017922181636e-05), (16, 3.560944242053665e-05), (17, 2.9622658985317685e-05), (18, 9.218750346917659e-05), (19, 6.919694715179503e-05), (20, 5.575004615820944e-05), (21, 7.862750499043614e-05), (22, 0.0001307882193941623), (23, 0.00028696536901406944), (24, 0.0003134427242912352), (25, 0.0005118109402246773), (26, 0.00010574254702078179), (27, 7.143404218368232e-05)]
+
+ 25%|██▍       | 37/150 [06:07<14:04,  7.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 672, Spatial MSE: [(0, 3.5665468658407917e-06), (1, 5.50534859939944e-05), (2, 0.0004646285669878125), (3, 5.937303649261594e-05), (4, 4.6404802560573444e-05), (5, 8.980534767033532e-05), (6, 0.00015880331920925528), (7, 0.0003220697399228811), (8, 0.00044417448225431144), (9, 0.0006886014016345143), (10, 0.000905886001419276), (11, 0.0010471080895513296), (12, 0.0012152710696682334), (13, 0.0058207036927342415), (14, 0.009459746070206165), (15, 0.006247413344681263), (16, 0.11418074369430542), (17, 0.0025536681059747934), (18, 0.0027357470244169235), (19, 0.009370540268719196), (20, 0.008027558214962482), (21, 0.004262334667146206), (22, 0.003817832563072443), (23, 0.003591593587771058), (24, 0.007754648569971323), (25, 0.016403518617153168), (26, 0.008699608966708183), (27, 0.03409188240766525)], Temporal MSE: [(0, 3.7624320157192415e-06), (1, 4.645904482458718e-06), (2, 4.530630576482508e-06), (3, 1.1640922821243294e-05), (4, 2.5424311388633214e-05), (5, 4.02572586608585e-05), (6, 0.00010958110942738131), (7, 7.613920752191916e-05), (8, 6.580824265256524e-05), (9, 0.0001420540502294898), (10, 0.0002720723859965801), (11, 0.000310065719531849), (12, 0.00033751403680071235), (13, 0.0006635920726694167), (14, 0.0008418926154263318), (15, 0.0009703453979454935), (16, 0.0009639989584684372), (17, 0.0011607236228883266), (18, 0.000883561500813812), (19, 0.001221212325617671), (20, 0.0020363505464047194), (21, 0.0016974005848169327), (22, 0.0009505715570412576), (23, 0.0015177910681813955), (24, 0.003992030862718821), (25, 0.007909568957984447), (26, 0.0009455293184146285), (27, 0.00010289529018336907)]
+
+ 25%|██▌       | 38/150 [06:14<13:57,  7.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 666, Spatial MSE: [(0, 3.1498816497332882e-06), (1, 2.255568324471824e-05), (2, 0.00012701857485808432), (3, 4.151892790105194e-05), (4, 6.73348813506891e-06), (5, 4.522277777141426e-06), (6, 3.036687667190563e-05), (7, 7.347661721723853e-06), (8, 4.9924033191928174e-06), (9, 5.761851753049996e-06), (10, 7.479448413505452e-06), (11, 1.426705057383515e-05), (12, 4.945639375364408e-05), (13, 0.0022559072822332382), (14, 0.002830621786415577), (15, 0.001028932281769812), (16, 0.03505746275186539), (17, 0.0001077062261174433), (18, 0.00019617924408521503), (19, 0.0024333111941814423), (20, 0.0012713255127891898), (21, 0.00039481191197410226), (22, 0.0006661920342594385), (23, 0.0002832276513800025), (24, 0.0004971674061380327), (25, 0.0011132881045341492), (26, 0.001310966326855123), (27, 0.029662683606147766)], Temporal MSE: [(0, 3.434824066061992e-06), (1, 1.121289528782654e-06), (2, 1.4088110447119107e-06), (3, 1.2174537005193997e-06), (4, 7.461861741830944e-07), (5, 1.7930535705090733e-06), (6, 1.4795534752920503e-06), (7, 1.698199753263907e-06), (8, 8.828002364680287e-07), (9, 1.282582275052846e-06), (10, 1.8927526070910972e-06), (11, 1.5891811926849186e-05), (12, 3.0122870157356374e-05), (13, 8.926993177738041e-05), (14, 4.2339797801105306e-05), (15, 4.64828044641763e-05), (16, 2.745918754953891e-05), (17, 2.6907862775260583e-05), (18, 7.568881846964359e-05), (19, 5.815557233290747e-05), (20, 4.6879231376806274e-05), (21, 6.727136496920139e-05), (22, 0.00011624854960246012), (23, 0.00023674010299146175), (24, 0.0002484329743310809), (25, 0.00039229533285833895), (26, 8.479737880406901e-05), (27, 6.392849900294095e-05)]
+
+ 26%|██▌       | 39/150 [06:22<14:00,  7.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 660, Spatial MSE: [(0, 2.5843364710453898e-06), (1, 2.5237710360670462e-05), (2, 0.00017064657004084438), (3, 2.6887215426540934e-05), (4, 1.1864590305776801e-05), (5, 9.920298907672986e-06), (6, 2.078691977658309e-05), (7, 2.0245051928213798e-05), (8, 3.186269896104932e-05), (9, 5.259926547296345e-05), (10, 8.806699770502746e-05), (11, 0.00010373110126238316), (12, 0.0001421367924194783), (13, 0.002415567869320512), (14, 0.003175796242430806), (15, 0.0012846591416746378), (16, 0.04055381938815117), (17, 0.00015518114378210157), (18, 0.00045452540507540107), (19, 0.00356752029620111), (20, 0.0024822670966386795), (21, 0.0006299666129052639), (22, 0.0008023333502933383), (23, 0.00036876313970424235), (24, 0.0009417481487616897), (25, 0.001720181666314602), (26, 0.0018401247216388583), (27, 0.02523035928606987)], Temporal MSE: [(0, 1.915866278068279e-06), (1, 1.0711418099162984e-06), (2, 1.97087479136826e-06), (3, 2.1965106498100795e-06), (4, 2.8872564143966883e-06), (5, 3.3960841392399743e-06), (6, 5.470856194733642e-06), (7, 3.6965514027542667e-06), (8, 3.856011062453035e-06), (9, 6.239777576411143e-06), (10, 1.1453083061496727e-05), (11, 2.7739632059819996e-05), (12, 4.261218782630749e-05), (13, 0.00010896133608184755), (14, 6.456459959736094e-05), (15, 6.495993147836998e-05), (16, 4.270068893674761e-05), (17, 4.591595643432811e-05), (18, 0.00010055454185931012), (19, 9.800980478757992e-05), (20, 0.00012834776134695858), (21, 0.00011629389337031171), (22, 0.00016188401787076145), (23, 0.00030385100399143994), (24, 0.00045071804197505116), (25, 0.0007762486930005252), (26, 0.00013678865798283368), (27, 6.329408643068746e-05)]
+
+ 27%|██▋       | 40/150 [06:31<14:16,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 654, Spatial MSE: [(0, 2.6038765099656302e-06), (1, 1.9226761651225388e-05), (2, 0.00010942966036964208), (3, 4.095628173672594e-05), (4, 7.503158485633321e-06), (5, 4.802009698323673e-06), (6, 3.135229053441435e-05), (7, 7.624693807883887e-06), (8, 5.069802227808395e-06), (9, 5.8288978834752925e-06), (10, 8.112479918054305e-06), (11, 1.8316024579689838e-05), (12, 6.945001950953156e-05), (13, 0.003002379322424531), (14, 0.0032111373730003834), (15, 0.0009933450492098927), (16, 0.02628561668097973), (17, 7.514155731769279e-05), (18, 9.006445907289162e-05), (19, 0.001795247895643115), (20, 0.0008527776808477938), (21, 0.00020977445819880813), (22, 0.0004919276107102633), (23, 0.0002557153347879648), (24, 0.0004284607421141118), (25, 0.0008757684845477343), (26, 0.0012804247671738267), (27, 0.029714079573750496)], Temporal MSE: [(0, 2.94232677333639e-06), (1, 1.1209998547201394e-06), (2, 1.3666734730577446e-06), (3, 1.2295668057049625e-06), (4, 7.649658186892339e-07), (5, 1.8357560520598781e-06), (6, 1.4148046147965943e-06), (7, 1.8347831201026565e-06), (8, 8.591784421696502e-07), (9, 1.257912344954093e-06), (10, 1.9505405361996964e-06), (11, 2.3096778022591025e-05), (12, 3.829180423053913e-05), (13, 8.750573033466935e-05), (14, 3.067555371671915e-05), (15, 4.876399179920554e-05), (16, 2.2550744688487612e-05), (17, 2.1451649445225485e-05), (18, 6.074620978324674e-05), (19, 4.6778121031820774e-05), (20, 3.5679200664162636e-05), (21, 3.93762493331451e-05), (22, 6.820396083639935e-05), (23, 0.0001762283209245652), (24, 0.0002217799046775326), (25, 0.0003966549411416054), (26, 8.41005748952739e-05), (27, 5.6074753956636414e-05)]
+
+ 27%|██▋       | 41/150 [06:38<13:49,  7.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 648, Spatial MSE: [(0, 2.5187155188177712e-06), (1, 3.9213384297909215e-05), (2, 0.0002439248055452481), (3, 3.9561473386129364e-05), (4, 3.712821126100607e-05), (5, 6.95645430823788e-05), (6, 0.0001360657624900341), (7, 0.0002598401624709368), (8, 0.00040093454299494624), (9, 0.0006488072103820741), (10, 0.0009924767073243856), (11, 0.000927842513192445), (12, 0.001074695959687233), (13, 0.004561677575111389), (14, 0.006373818498104811), (15, 0.0040970128029584885), (16, 0.05294569954276085), (17, 0.002019008155912161), (18, 0.00196236465126276), (19, 0.004649410955607891), (20, 0.0038190472405403852), (21, 0.0022205475252121687), (22, 0.0025137276388704777), (23, 0.0024313025642186403), (24, 0.005382601637393236), (25, 0.011718690395355225), (26, 0.0060690827667713165), (27, 0.027251549065113068)], Temporal MSE: [(0, 2.5310935143352253e-06), (1, 2.7140745260112453e-06), (2, 2.746949803622556e-06), (3, 9.286422937293537e-06), (4, 2.0602396034519188e-05), (5, 3.191499854438007e-05), (6, 8.719835750525817e-05), (7, 6.652165029663593e-05), (8, 5.6390279496554285e-05), (9, 0.00011769510456360877), (10, 0.00023268918448593467), (11, 0.0002406863059150055), (12, 0.00024172912526410073), (13, 0.0004475792229641229), (14, 0.0006393446237780154), (15, 0.0006467993371188641), (16, 0.0007340725278481841), (17, 0.000733274791855365), (18, 0.0006277604843489826), (19, 0.00078276899876073), (20, 0.0011389368446543813), (21, 0.0009466804913245142), (22, 0.0006554332212544978), (23, 0.0010252469219267368), (24, 0.0032925265841186047), (25, 0.008137863129377365), (26, 0.0007822600309737027), (27, 7.579023076687008e-05)]
+
+ 28%|██▊       | 42/150 [06:46<13:49,  7.68s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 642, Spatial MSE: [(0, 2.9006419026700314e-06), (1, 1.7286625734413974e-05), (2, 0.00010910574201261625), (3, 3.576682502171025e-05), (4, 6.683984793198761e-06), (5, 3.966407803090988e-06), (6, 2.711109300435055e-05), (7, 8.231612810050137e-06), (8, 4.868797077506315e-06), (9, 5.578066520683933e-06), (10, 6.9743309722980484e-06), (11, 1.0571879101917148e-05), (12, 2.8587393899215385e-05), (13, 0.0012459078570827842), (14, 0.0016582050593569875), (15, 0.0006478792056441307), (16, 0.018925976008176804), (17, 5.8294452173868194e-05), (18, 0.00018086860654875636), (19, 0.0011873492039740086), (20, 0.0005633958498947322), (21, 0.00022501367493532598), (22, 0.00038409174885600805), (23, 0.00021300255320966244), (24, 0.00033488855115137994), (25, 0.0007703275769017637), (26, 0.0014486568979918957), (27, 0.022471290081739426)], Temporal MSE: [(0, 3.046198571610148e-06), (1, 1.0389250064690714e-06), (2, 1.2654511465370888e-06), (3, 1.3205326467868872e-06), (4, 7.978661074048432e-07), (5, 1.8083621853293153e-06), (6, 1.3712366353502148e-06), (7, 1.7508110659036902e-06), (8, 8.335000529768877e-07), (9, 1.2910493296658387e-06), (10, 1.878983653114119e-06), (11, 8.477307346765883e-06), (12, 1.5086372513906099e-05), (13, 5.178245919523761e-05), (14, 2.3504302589572035e-05), (15, 2.958702498290222e-05), (16, 1.70630264619831e-05), (17, 1.4355897292261943e-05), (18, 4.516602348303422e-05), (19, 3.509188172756694e-05), (20, 2.6413719751872122e-05), (21, 3.288001607870683e-05), (22, 5.732286808779463e-05), (23, 0.00014005489356350154), (24, 0.00016772231901995838), (25, 0.00031252577900886536), (26, 7.088942220434546e-05), (27, 3.5488177672959864e-05)]
+
+ 29%|██▊       | 43/150 [06:55<14:37,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 636, Spatial MSE: [(0, 2.3902377961348975e-06), (1, 1.906936631712597e-05), (2, 0.00013905426021665335), (3, 1.8633059880812652e-05), (4, 9.87153089226922e-06), (5, 8.315615559695289e-06), (6, 1.2742345461447258e-05), (7, 1.740844345476944e-05), (8, 2.1760217350674793e-05), (9, 3.22371706715785e-05), (10, 5.383304232964292e-05), (11, 5.566690379055217e-05), (12, 8.17951004137285e-05), (13, 0.001472198055125773), (14, 0.0024796577636152506), (15, 0.0014890296151861548), (16, 0.03965506702661514), (17, 0.00011452969192760065), (18, 0.00025328449555672705), (19, 0.0023026447743177414), (20, 0.0016317549161612988), (21, 0.00027144572231918573), (22, 0.0005107936449348927), (23, 0.0003185769310221076), (24, 0.0008264713105745614), (25, 0.001509050838649273), (26, 0.0017717996379360557), (27, 0.015443605370819569)], Temporal MSE: [(0, 1.8948800288853818e-06), (1, 9.227047712556669e-07), (2, 1.7973076182897785e-06), (3, 2.148015937564196e-06), (4, 3.0053536193008767e-06), (5, 3.017244807779207e-06), (6, 3.584470277928631e-06), (7, 3.2654791084496537e-06), (8, 3.3291323688899865e-06), (9, 4.601726232067449e-06), (10, 7.465197086276021e-06), (11, 1.5312747564166784e-05), (12, 2.364546526223421e-05), (13, 6.65354382363148e-05), (14, 4.728823114419356e-05), (15, 5.7797416957328096e-05), (16, 3.100108733633533e-05), (17, 3.548244058038108e-05), (18, 6.636166654061526e-05), (19, 5.9635367506416515e-05), (20, 7.374955021077767e-05), (21, 9.431927173864096e-05), (22, 0.00011455864296294749), (23, 0.0002198906586272642), (24, 0.00040560439811088145), (25, 0.0007580676465295255), (26, 0.00012905338371638209), (27, 3.91539215343073e-05)]
+
+ 29%|██▉       | 44/150 [07:05<15:33,  8.80s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 630, Spatial MSE: [(0, 2.9478098895197036e-06), (1, 1.4956200175220147e-05), (2, 8.285699732368812e-05), (3, 3.1589304853696376e-05), (4, 5.576047897193348e-06), (5, 3.5265411497675814e-06), (6, 2.9154849471524358e-05), (7, 8.598781278124079e-06), (8, 4.5303936531126965e-06), (9, 5.345666522771353e-06), (10, 7.093739441188518e-06), (11, 1.1251482646912336e-05), (12, 3.379491317900829e-05), (13, 0.0010667240712791681), (14, 0.0016983570531010628), (15, 0.0006026144255883992), (16, 0.028503278270363808), (17, 0.000150274921907112), (18, 0.00014784141967538744), (19, 0.002055457094684243), (20, 0.00122714729513973), (21, 0.00015767730656079948), (22, 0.00040300015825778246), (23, 0.00018058277782984078), (24, 0.00033837693626992404), (25, 0.0007804261986166239), (26, 0.0011707113590091467), (27, 0.01937764696776867)], Temporal MSE: [(0, 2.838675982275163e-06), (1, 8.418629136031086e-07), (2, 9.834577667788835e-07), (3, 1.067007133315201e-06), (4, 6.76054582982033e-07), (5, 1.5273910776159028e-06), (6, 1.1586872687985306e-06), (7, 2.3312184112000978e-06), (8, 7.861338531256479e-07), (9, 1.1362848226781352e-06), (10, 2.145275402654079e-06), (11, 1.3397287148109172e-05), (12, 1.592096668900922e-05), (13, 4.430010085343383e-05), (14, 2.498614412616007e-05), (15, 4.9635313189355657e-05), (16, 2.599738581920974e-05), (17, 2.2820024241809733e-05), (18, 5.771090218331665e-05), (19, 4.7750469093443826e-05), (20, 3.654237661976367e-05), (21, 5.5092521506594494e-05), (22, 8.47240662551485e-05), (23, 0.00016772006347309798), (24, 0.00016237568343058228), (25, 0.0003396401589270681), (26, 6.312548066489398e-05), (27, 5.3170624596532434e-05)]
+
+ 30%|███       | 45/150 [07:13<14:43,  8.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 624, Spatial MSE: [(0, 2.259795337522519e-06), (1, 3.222364102839492e-05), (2, 0.00021792398183606565), (3, 3.3106061891885474e-05), (4, 3.3814081689342856e-05), (5, 6.292774924077094e-05), (6, 0.00016939577471930534), (7, 0.0002517730463296175), (8, 0.0003150058619212359), (9, 0.0004934593453072011), (10, 0.0008111720671877265), (11, 0.0008474148344248533), (12, 0.0009422166040167212), (13, 0.002922794781625271), (14, 0.00532707292586565), (15, 0.00404011458158493), (16, 0.07068943232297897), (17, 0.0022794550750404596), (18, 0.0027085847686976194), (19, 0.008641812950372696), (20, 0.007639652583748102), (21, 0.002908929018303752), (22, 0.0032682917080819607), (23, 0.003691583639010787), (24, 0.0082489512860775), (25, 0.015374686568975449), (26, 0.007655180990695953), (27, 0.017711983993649483)], Temporal MSE: [(0, 2.1028222363383975e-06), (1, 2.2299627744359896e-06), (2, 2.747779262790573e-06), (3, 8.868635632097721e-06), (4, 1.8413358702673577e-05), (5, 4.5672510168515146e-05), (6, 0.00012618352775461972), (7, 7.277643453562632e-05), (8, 4.967190034221858e-05), (9, 0.00010496600589249283), (10, 0.00019735847308766097), (11, 0.00021036091493442655), (12, 0.0002150533691747114), (13, 0.00040943731437437236), (14, 0.0006379689439199865), (15, 0.0006732296315021813), (16, 0.0009530832176096737), (17, 0.0009187401738017797), (18, 0.0009619983029551804), (19, 0.0012508677318692207), (20, 0.0018750581657513976), (21, 0.0014239278389140964), (22, 0.0010280102724209428), (23, 0.0014663999900221825), (24, 0.0043358574621379375), (25, 0.00967489555478096), (26, 0.0009030015207827091), (27, 7.780442683724687e-05)]
+
+ 31%|███       | 46/150 [07:21<14:29,  8.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 618, Spatial MSE: [(0, 1.6792953374533681e-06), (1, 1.3307412700669374e-05), (2, 6.249010766623542e-05), (3, 2.0231287635397166e-05), (4, 3.5100608783977805e-06), (5, 2.48978471972805e-06), (6, 1.3511546967492905e-05), (7, 4.360302227723878e-06), (8, 2.8708295758406166e-06), (9, 3.7349243484641192e-06), (10, 6.8202980401110835e-06), (11, 1.2925322153023444e-05), (12, 3.054352418985218e-05), (13, 0.0005068479222245514), (14, 0.0007621289696544409), (15, 0.0002769734419416636), (16, 0.015258613042533398), (17, 9.112928091781214e-05), (18, 8.408490975853056e-05), (19, 0.0011875444324687123), (20, 0.0007570068119093776), (21, 0.0001195908262161538), (22, 0.00022329653438646346), (23, 0.00011750585690606385), (24, 0.00022425693168770522), (25, 0.0005189250223338604), (26, 0.0007968159625306726), (27, 0.009840009734034538)], Temporal MSE: [(0, 1.7885403167383629e-06), (1, 6.453319087995624e-07), (2, 8.535288884559122e-07), (3, 7.453521675415686e-07), (4, 4.469019074804237e-07), (5, 9.914554084389238e-07), (6, 8.260415711447422e-07), (7, 1.0450955869600875e-06), (8, 5.396644837674103e-07), (9, 8.095412908915023e-07), (10, 2.3197335394797847e-06), (11, 2.1497244233614765e-05), (12, 1.5682422599638812e-05), (13, 1.7792850485420786e-05), (14, 1.4488471606455278e-05), (15, 3.541590922395699e-05), (16, 1.8471355360816233e-05), (17, 1.7335118172923103e-05), (18, 3.181717329425737e-05), (19, 2.547193980717566e-05), (20, 2.3078162485035136e-05), (21, 3.402209040359594e-05), (22, 5.2581010095309466e-05), (23, 9.317330113844946e-05), (24, 9.053928079083562e-05), (25, 0.00015440727293025702), (26, 3.6072447983315215e-05), (27, 2.8359780117170885e-05)]
+
+ 31%|███▏      | 47/150 [07:28<13:43,  7.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 612, Spatial MSE: [(0, 1.4358835187522345e-06), (1, 1.3182956536184065e-05), (2, 8.576244727009907e-05), (3, 1.4921099136699922e-05), (4, 6.564235263795126e-06), (5, 6.141478479548823e-06), (6, 9.427627446711995e-06), (7, 1.1634502698143478e-05), (8, 1.4606738659495022e-05), (9, 2.2369153157342225e-05), (10, 3.5922203096561134e-05), (11, 3.9471909985877573e-05), (12, 6.448262138292193e-05), (13, 0.000542385911103338), (14, 0.0008340136846527457), (15, 0.00030971976229920983), (16, 0.015371790155768394), (17, 0.00010707390174502507), (18, 0.00011793927114922553), (19, 0.0011952230706810951), (20, 0.0007872472051531076), (21, 0.00014272033877205104), (22, 0.00027705691172741354), (23, 0.00016983506793621927), (24, 0.0005076616653241217), (25, 0.0009133100975304842), (26, 0.0013939718483015895), (27, 0.007915642112493515)], Temporal MSE: [(0, 1.0698599908209872e-06), (1, 5.743632414123567e-07), (2, 1.1342751804477302e-06), (3, 1.5877243413342512e-06), (4, 2.3503348529629875e-06), (5, 2.2252436338021653e-06), (6, 2.6770217118610162e-06), (7, 2.3418504042638233e-06), (8, 2.370882384639117e-06), (9, 2.983328613481717e-06), (10, 6.242614745133324e-06), (11, 2.4437496904283762e-05), (12, 1.9125469407299533e-05), (13, 2.402617974439636e-05), (14, 2.3944057829794474e-05), (15, 4.363970583654009e-05), (16, 2.4588540327386e-05), (17, 2.3878063075244427e-05), (18, 4.271820580470376e-05), (19, 3.848859705612995e-05), (20, 4.6847893827361986e-05), (21, 5.6335240515181795e-05), (22, 7.729105709586293e-05), (23, 0.00013252226926852018), (24, 0.0002233349805464968), (25, 0.0003781098057515919), (26, 8.986757893580943e-05), (27, 3.092378028668463e-05)]
+
+ 32%|███▏      | 48/150 [07:37<13:55,  8.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 606, Spatial MSE: [(0, 2.1721714347222587e-06), (1, 1.2267470083315857e-05), (2, 6.594618025701493e-05), (3, 2.3199429051601328e-05), (4, 4.0589638956589624e-06), (5, 2.6295726911484962e-06), (6, 1.78825794137083e-05), (7, 6.078238129703095e-06), (8, 3.3328490189887816e-06), (9, 3.7215215797914425e-06), (10, 4.333363449404715e-06), (11, 5.033136403653771e-06), (12, 8.520941264578141e-06), (13, 0.00022044173965696245), (14, 0.00035190058406442404), (15, 0.0001965551928151399), (16, 0.005427178461104631), (17, 2.4798335289233364e-05), (18, 2.4868724722182378e-05), (19, 0.0002953586808871478), (20, 0.0001358313747914508), (21, 4.8348476411774755e-05), (22, 0.00010099558130605146), (23, 7.820038445061073e-05), (24, 0.0001256958639714867), (25, 0.00031358253909274936), (26, 0.0008230219245888293), (27, 0.008435597643256187)], Temporal MSE: [(0, 2.000309677896439e-06), (1, 6.464230182245956e-07), (2, 8.624028282611107e-07), (3, 8.350037319360126e-07), (4, 5.076097977507743e-07), (5, 1.1652650755422655e-06), (6, 9.174349884233379e-07), (7, 1.54307565480849e-06), (8, 6.099883194110589e-07), (9, 8.740106522964197e-07), (10, 1.2004087466266355e-06), (11, 2.810812020470621e-06), (12, 3.5396972180024022e-06), (13, 1.4655827726528514e-05), (14, 8.050941687542945e-06), (15, 9.818766557145864e-06), (16, 5.575830527959624e-06), (17, 5.434410923044197e-06), (18, 1.350679940514965e-05), (19, 1.1045460269087926e-05), (20, 9.467270501772873e-06), (21, 1.2110319403291214e-05), (22, 1.7468015357735567e-05), (23, 4.088961941306479e-05), (24, 5.354132008505985e-05), (25, 0.00017370865680277348), (26, 3.1643143302062526e-05), (27, 1.2927611351187807e-05)]
+
+ 33%|███▎      | 49/150 [07:45<13:41,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 600, Spatial MSE: [(0, 2.0630834569601575e-06), (1, 3.118596941931173e-05), (2, 0.00016308273188769817), (3, 3.089250094490126e-05), (4, 3.5942030081059784e-05), (5, 5.3626125009031966e-05), (6, 0.0001189231697935611), (7, 0.00019828129734378308), (8, 0.0002549119235482067), (9, 0.0004018383624497801), (10, 0.0006073280819691718), (11, 0.0005431385361589491), (12, 0.0006356024532578886), (13, 0.0013662477722391486), (14, 0.0028779690619558096), (15, 0.0025033347774297), (16, 0.041207700967788696), (17, 0.0010977688943967223), (18, 0.0013104442041367292), (19, 0.0031892918050289154), (20, 0.0028426579665392637), (21, 0.0014093900099396706), (22, 0.0016020456096157432), (23, 0.0017550463089719415), (24, 0.003951591905206442), (25, 0.007196138147264719), (26, 0.004542093724012375), (27, 0.007114093285053968)], Temporal MSE: [(0, 1.9493811578286113e-06), (1, 1.6156225228769472e-06), (2, 2.3029715521261096e-06), (3, 9.89308136922773e-06), (4, 1.720562795526348e-05), (5, 2.6829788112081587e-05), (6, 7.321629527723417e-05), (7, 4.7160843678284436e-05), (8, 3.814961382886395e-05), (9, 8.01897986093536e-05), (10, 0.0001505063846707344), (11, 0.00015069465734995902), (12, 0.00015754900232423097), (13, 0.0003139081527478993), (14, 0.00046242467942647636), (15, 0.0003704644914250821), (16, 0.0004571554600261152), (17, 0.00045473084901459515), (18, 0.00044809511746279895), (19, 0.000555646896827966), (20, 0.0009014902752824128), (21, 0.000736810325179249), (22, 0.0005661743925884366), (23, 0.000812088546808809), (24, 0.0022707025054842234), (25, 0.005029020830988884), (26, 0.0005813385359942913), (27, 3.312615808681585e-05)]
+
+ 33%|███▎      | 50/150 [07:54<13:58,  8.39s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 594, Spatial MSE: [(0, 1.0032258614955936e-06), (1, 8.799503120826557e-06), (2, 4.940102735417895e-05), (3, 1.5206884199869819e-05), (4, 2.79292862614966e-06), (5, 1.8210846519650659e-06), (6, 8.711612281331327e-06), (7, 4.212662133795675e-06), (8, 2.2915337467566133e-06), (9, 2.624665739858756e-06), (10, 3.03477531815588e-06), (11, 3.309178282506764e-06), (12, 4.934620392305078e-06), (13, 4.963179162587039e-05), (14, 2.778844464046415e-05), (15, 8.353281373274513e-06), (16, 1.4936627849237993e-05), (17, 6.662286068603862e-06), (18, 1.0707359251682647e-05), (19, 1.3619105629913975e-05), (20, 9.67863070400199e-06), (21, 1.4683198060083669e-05), (22, 2.6967269150190987e-05), (23, 4.096788325114176e-05), (24, 7.847794040571898e-05), (25, 0.00022249478206504136), (26, 0.0005922893760725856), (27, 0.0033306374680250883)], Temporal MSE: [(0, 1.1054098649765365e-06), (1, 4.521779430888273e-07), (2, 6.818476663283946e-07), (3, 6.558573204529239e-07), (4, 4.036511995764158e-07), (5, 8.635299764137017e-07), (6, 7.279112423930201e-07), (7, 9.66843799687922e-07), (8, 4.503859258875309e-07), (9, 6.277362558648747e-07), (10, 8.97701966096065e-07), (11, 1.8547534637036733e-06), (12, 1.8685564100451302e-06), (13, 1.877025511021202e-06), (14, 1.8006344362220261e-06), (15, 1.4425507970372564e-06), (16, 2.108149828927708e-06), (17, 1.6458653817608138e-06), (18, 1.6668690250298823e-06), (19, 2.2153790268930607e-06), (20, 3.40318433700304e-06), (21, 3.8051234696467873e-06), (22, 4.878109848505119e-06), (23, 8.916429578675888e-06), (24, 1.7953801943804137e-05), (25, 7.058458140818402e-05), (26, 1.4464610103459563e-05), (27, 1.1290021575405262e-06)]
+
+ 34%|███▍      | 51/150 [08:02<13:41,  8.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 588, Spatial MSE: [(0, 1.405250600328145e-06), (1, 1.512790458946256e-05), (2, 9.156025043921545e-05), (3, 1.830692053772509e-05), (4, 7.159667802625336e-06), (5, 7.166306659200927e-06), (6, 1.824295031838119e-05), (7, 1.8044285752694122e-05), (8, 2.192647662013769e-05), (9, 3.9158756408141926e-05), (10, 6.238441710593179e-05), (11, 7.100544462446123e-05), (12, 7.390203973045573e-05), (13, 0.00011799601634265855), (14, 0.00013723615847993642), (15, 0.0001293829845963046), (16, 0.003918538335710764), (17, 4.5698092435486615e-05), (18, 0.00018247691332362592), (19, 0.0004945851978845894), (20, 0.000830753764603287), (21, 0.00012693625467363745), (22, 0.00010705012391554192), (23, 0.0001294777903240174), (24, 0.0004495180619414896), (25, 0.0007300049182958901), (26, 0.0019769843202084303), (27, 0.0037084477953612804)], Temporal MSE: [(0, 1.4376868193721748e-06), (1, 6.360059501275828e-07), (2, 1.3069490023553954e-06), (3, 1.739565959724132e-06), (4, 2.3818458885216387e-06), (5, 2.575847702246392e-06), (6, 3.465171403149725e-06), (7, 3.6766314224223606e-06), (8, 3.1390848107548663e-06), (9, 5.094647349324077e-06), (10, 9.55438008531928e-06), (11, 1.1020011697837617e-05), (12, 1.115429859055439e-05), (13, 1.6747233530622907e-05), (14, 1.968518517969642e-05), (15, 1.541743586130906e-05), (16, 1.5566582078463398e-05), (17, 1.4459429621638265e-05), (18, 2.4852095521055162e-05), (19, 3.0894054361851886e-05), (20, 5.529713962459937e-05), (21, 3.849805216304958e-05), (22, 4.4139746023574844e-05), (23, 6.279616354731843e-05), (24, 0.0001994575868593529), (25, 0.0003369312616996467), (26, 6.940763705642894e-05), (27, 4.757828264700947e-06)]
+
+ 35%|███▍      | 52/150 [08:10<13:40,  8.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 582, Spatial MSE: [(0, 1.527704625914339e-06), (1, 1.0563371688476764e-05), (2, 5.8063760661752895e-05), (3, 1.4519994692818727e-05), (4, 3.403366918064421e-06), (5, 2.0247614429536043e-06), (6, 1.0002652743423823e-05), (7, 4.091833488928387e-06), (8, 2.672410801096703e-06), (9, 3.4095810406142846e-06), (10, 3.3289866223640274e-06), (11, 3.7397364849312e-06), (12, 4.794714186573401e-06), (13, 3.645772085292265e-05), (14, 2.9152923161746003e-05), (15, 1.0510477295611054e-05), (16, 1.3976507034385577e-05), (17, 1.1004319276253227e-05), (18, 1.6744052118156105e-05), (19, 2.2058999093133025e-05), (20, 1.1497929335746448e-05), (21, 2.4254550226032734e-05), (22, 3.797034514718689e-05), (23, 5.002017496735789e-05), (24, 7.619652024004608e-05), (25, 0.0002160871954401955), (26, 0.0006506196223199368), (27, 0.0038929551374167204)], Temporal MSE: [(0, 1.584782012287178e-06), (1, 5.889173166906403e-07), (2, 8.233183734773775e-07), (3, 8.034245411181473e-07), (4, 4.580406596232933e-07), (5, 9.544103249936597e-07), (6, 7.502606536036183e-07), (7, 1.0857498864424997e-06), (8, 5.02865759699489e-07), (9, 7.058612823129806e-07), (10, 1.0382485697846278e-06), (11, 1.6650106999804848e-06), (12, 1.617462316971796e-06), (13, 1.8134398942493135e-06), (14, 2.0666152522608172e-06), (15, 1.583530320203863e-06), (16, 1.9342214727657847e-06), (17, 2.244289362351992e-06), (18, 4.00772523789783e-06), (19, 3.840635145024862e-06), (20, 6.088472673582146e-06), (21, 4.600384272634983e-06), (22, 5.993765171297127e-06), (23, 1.058962425304344e-05), (24, 1.8034172171610408e-05), (25, 9.579164907336235e-05), (26, 1.7437383576179855e-05), (27, 2.162751570722321e-06)]
+
+ 35%|███▌      | 53/150 [08:18<13:11,  8.16s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 576, Spatial MSE: [(0, 1.4598088000639109e-06), (1, 2.2027583327144384e-05), (2, 0.00010825554636539891), (3, 2.505180600564927e-05), (4, 3.3037624234566465e-05), (5, 4.7552814066875726e-05), (6, 0.00010762741294456646), (7, 0.00019696853996720165), (8, 0.00024203995417337865), (9, 0.00038856620085425675), (10, 0.0005931868799962103), (11, 0.0005065143341198564), (12, 0.0005612322129309177), (13, 0.0011962027056142688), (14, 0.0026915520429611206), (15, 0.0023262423928827047), (16, 0.041620321571826935), (17, 0.0010016347514465451), (18, 0.0011938047828152776), (19, 0.003137752879410982), (20, 0.002748740604147315), (21, 0.001381640788167715), (22, 0.0014885974815115333), (23, 0.0016328439814969897), (24, 0.003669212805107236), (25, 0.006291306111961603), (26, 0.0043168822303414345), (27, 0.003274098737165332)], Temporal MSE: [(0, 1.2794620261047385e-06), (1, 1.1062013527407544e-06), (2, 1.9563819932955084e-06), (3, 9.641647011449095e-06), (4, 1.4198855751601513e-05), (5, 2.7451664209365845e-05), (6, 7.801267202012241e-05), (7, 4.471358988666907e-05), (8, 3.892074164468795e-05), (9, 7.448811084032059e-05), (10, 0.00013779319124296308), (11, 0.00014249805826693773), (12, 0.00014225294580683112), (13, 0.00025942333741113544), (14, 0.00039829203160479665), (15, 0.00029987230664119124), (16, 0.0003582346544135362), (17, 0.00038946664426475763), (18, 0.0004392663249745965), (19, 0.0005584417376667261), (20, 0.0009196141618303955), (21, 0.0007285581668838859), (22, 0.0005418626824393868), (23, 0.0007518123602494597), (24, 0.0022143537644296885), (25, 0.004581375978887081), (26, 0.0005663633346557617), (27, 2.3794043954694644e-05)]
+
+ 36%|███▌      | 54/150 [08:27<13:29,  8.44s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 570, Spatial MSE: [(0, 1.0321706440663547e-06), (1, 9.29504312807694e-06), (2, 4.45481164206285e-05), (3, 1.4269260645960458e-05), (4, 3.1520528409600956e-06), (5, 1.67960411090462e-06), (6, 8.825786608213093e-06), (7, 3.350232873344794e-06), (8, 2.2182398424774874e-06), (9, 3.0006992801645538e-06), (10, 2.9117272788425907e-06), (11, 3.252016767874011e-06), (12, 4.6053810365265235e-06), (13, 5.84919034736231e-05), (14, 3.628759804996662e-05), (15, 9.759192835190333e-06), (16, 1.9946997781516984e-05), (17, 5.367946414480684e-06), (18, 7.774931873427704e-06), (19, 1.6160884115379304e-05), (20, 9.020457582664676e-06), (21, 2.45943883783184e-05), (22, 3.14155149681028e-05), (23, 3.8348334783222526e-05), (24, 8.00821726443246e-05), (25, 0.0002187830541515723), (26, 0.0005749508854933083), (27, 0.0036016423255205154)], Temporal MSE: [(0, 1.146559384324064e-06), (1, 4.657782710637548e-07), (2, 6.300219865806866e-07), (3, 6.765752118553792e-07), (4, 4.120263383811107e-07), (5, 8.094170880212914e-07), (6, 6.617610779358074e-07), (7, 8.710451311344514e-07), (8, 4.583925203860417e-07), (9, 6.573985729119158e-07), (10, 9.082090173251345e-07), (11, 2.026680931521696e-06), (12, 1.9300373423902784e-06), (13, 2.032232487181318e-06), (14, 1.941869868460344e-06), (15, 1.3865237633581273e-06), (16, 2.2288002128334483e-06), (17, 1.561064095767506e-06), (18, 2.378405497438507e-06), (19, 2.6269610771123553e-06), (20, 4.00508861275739e-06), (21, 3.380043835932156e-06), (22, 5.967983270238619e-06), (23, 1.0812625077960547e-05), (24, 2.014969868469052e-05), (25, 5.47939125681296e-05), (26, 1.5934108887449838e-05), (27, 1.2669630677919486e-06)]
+
+ 37%|███▋      | 55/150 [08:35<12:51,  8.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 564, Spatial MSE: [(0, 1.189167846860073e-06), (1, 1.1268986781942658e-05), (2, 6.92435132805258e-05), (3, 1.4084624126553535e-05), (4, 6.009795470163226e-06), (5, 6.770110758225201e-06), (6, 1.5890760550973937e-05), (7, 1.2640729437407572e-05), (8, 1.6020510884118266e-05), (9, 2.6663399694371037e-05), (10, 3.723498957697302e-05), (11, 4.009867916465737e-05), (12, 5.02089606015943e-05), (13, 0.00011125423770863563), (14, 0.00011035709758289158), (15, 4.548813376459293e-05), (16, 5.568900451180525e-05), (17, 3.3118951250799e-05), (18, 4.8622845497448e-05), (19, 3.6733847082359716e-05), (20, 4.086739136255346e-05), (21, 5.6539272918598726e-05), (22, 8.747074753046036e-05), (23, 0.00010277423280058429), (24, 0.0003642967785708606), (25, 0.0006126068183220923), (26, 0.0015433403896167874), (27, 0.0019490921404212713)], Temporal MSE: [(0, 1.0772927225843887e-06), (1, 5.004042691325594e-07), (2, 1.109140612243209e-06), (3, 1.5922086049613426e-06), (4, 2.397959633526625e-06), (5, 2.3539607809652807e-06), (6, 2.449771045576199e-06), (7, 2.9427267236314947e-06), (8, 2.762374379017274e-06), (9, 3.5842074339598184e-06), (10, 6.041594588168664e-06), (11, 7.491588348784717e-06), (12, 7.704498784733005e-06), (13, 1.1060133147111628e-05), (14, 1.510531274107052e-05), (15, 1.0123748324986082e-05), (16, 1.0117721103597432e-05), (17, 9.495146514382213e-06), (18, 1.3947115803603083e-05), (19, 1.6762949599069543e-05), (20, 2.8199172447784804e-05), (21, 2.4883051082724705e-05), (22, 3.359283073223196e-05), (23, 5.0277751142857596e-05), (24, 0.00015933412942104042), (25, 0.0002760843199212104), (26, 7.017698953859508e-05), (27, 4.201501724310219e-06)]
+
+ 37%|███▋      | 56/150 [08:42<12:33,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 558, Spatial MSE: [(0, 1.3200893818066106e-06), (1, 1.0133840078196954e-05), (2, 4.825766518479213e-05), (3, 1.3154736734577455e-05), (4, 3.015814172613318e-06), (5, 1.7557496221343172e-06), (6, 8.362694643437862e-06), (7, 3.7933023122604936e-06), (8, 2.285791197209619e-06), (9, 2.787837274809135e-06), (10, 2.9419520615192596e-06), (11, 3.1958036288415315e-06), (12, 4.0327927308680955e-06), (13, 2.6582243663142435e-05), (14, 1.8302514945389703e-05), (15, 6.925203706487082e-06), (16, 1.0209387255599722e-05), (17, 1.0342185305489693e-05), (18, 8.442122634733096e-06), (19, 1.4428117538045626e-05), (20, 9.099991075345315e-06), (21, 1.3441939699987415e-05), (22, 2.472738196956925e-05), (23, 4.0490718674845994e-05), (24, 6.891266093589365e-05), (25, 0.00018891475338023156), (26, 0.0006136475712992251), (27, 0.003642380703240633)], Temporal MSE: [(0, 1.294444473387557e-06), (1, 4.850051027460722e-07), (2, 6.869811954857141e-07), (3, 6.987017400206241e-07), (4, 4.1388170757272746e-07), (5, 8.133684445965628e-07), (6, 6.483224979092483e-07), (7, 9.841846804192755e-07), (8, 4.4943058696844673e-07), (9, 6.218174917194119e-07), (10, 8.797745749689057e-07), (11, 1.4629699762735981e-06), (12, 1.2906211850349791e-06), (13, 1.425615778316569e-06), (14, 1.6362354244847666e-06), (15, 1.500582357039093e-06), (16, 1.775550458660291e-06), (17, 2.0142308585491264e-06), (18, 2.047584757747245e-06), (19, 2.48122819357377e-06), (20, 4.041729425807716e-06), (21, 5.099269856145838e-06), (22, 5.363672698877053e-06), (23, 8.664789675094653e-06), (24, 1.69977738551097e-05), (25, 6.373182986862957e-05), (26, 1.8574068235466257e-05), (27, 1.9896694993803976e-06)]
+
+ 38%|███▊      | 57/150 [08:49<12:00,  7.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 552, Spatial MSE: [(0, 1.293728246309911e-06), (1, 2.182826392527204e-05), (2, 9.349163883598521e-05), (3, 2.521011629141867e-05), (4, 3.1571482395520434e-05), (5, 4.3375068344175816e-05), (6, 0.00010212762572336942), (7, 0.0001699016138445586), (8, 0.0002209937374573201), (9, 0.00036346405977383256), (10, 0.00048075031372718513), (11, 0.0004473766020964831), (12, 0.0004939395003020763), (13, 0.0006687109125778079), (14, 0.0011343139922246337), (15, 0.001216055708937347), (16, 0.012325466610491276), (17, 0.0007527945563197136), (18, 0.0010724428575485945), (19, 0.001972195692360401), (20, 0.0026560055557638407), (21, 0.0010960429208353162), (22, 0.0013280666898936033), (23, 0.001533949514850974), (24, 0.0034030694514513016), (25, 0.005579716991633177), (26, 0.0044464608654379845), (27, 0.003280383301898837)], Temporal MSE: [(0, 1.3623488257508143e-06), (1, 9.556683835398871e-07), (2, 1.959434484888334e-06), (3, 1.1238458682782948e-05), (4, 1.2918372704007197e-05), (5, 2.1945197659078985e-05), (6, 5.962615614407696e-05), (7, 3.9700160414213315e-05), (8, 3.8851612771395594e-05), (9, 6.37379998806864e-05), (10, 0.00011112076026620343), (11, 0.00011168143100803718), (12, 0.00010783793550217524), (13, 0.00018878313130699098), (14, 0.00031949247932061553), (15, 0.00024370342725887895), (16, 0.00031638986547477543), (17, 0.0003406526811886579), (18, 0.000389158638427034), (19, 0.0005097785033285618), (20, 0.0008231379324570298), (21, 0.0005804635584354401), (22, 0.0005228634690865874), (23, 0.0007326177437789738), (24, 0.00227410183288157), (25, 0.004578281193971634), (26, 0.000536117353476584), (27, 1.9242253983975388e-05)]
+
+ 39%|███▊      | 58/150 [08:57<12:01,  7.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 546, Spatial MSE: [(0, 1.3762487469648477e-06), (1, 9.16193675948307e-06), (2, 4.132457252126187e-05), (3, 1.2218353731441312e-05), (4, 2.6121608698304044e-06), (5, 1.7005767176669906e-06), (6, 7.786375135765411e-06), (7, 4.3494774217833765e-06), (8, 2.2561482637684094e-06), (9, 2.7230187242821557e-06), (10, 2.752452246568282e-06), (11, 3.0180560770531883e-06), (12, 4.159142008575145e-06), (13, 3.506018765619956e-05), (14, 1.9390379748074338e-05), (15, 6.442189715016866e-06), (16, 1.3758635759586468e-05), (17, 7.144305527617689e-06), (18, 1.1111129424534738e-05), (19, 1.2757943295582663e-05), (20, 6.780162038921844e-06), (21, 1.2277258974791039e-05), (22, 1.9426621292950585e-05), (23, 3.347883466631174e-05), (24, 6.0349269915604964e-05), (25, 0.0001917701738420874), (26, 0.0005761480424553156), (27, 0.0026978508103638887)], Temporal MSE: [(0, 1.2908138842249173e-06), (1, 4.564774656046211e-07), (2, 6.159989993648196e-07), (3, 6.538627985719359e-07), (4, 4.0432448145111266e-07), (5, 7.874934340179607e-07), (6, 6.57551254334976e-07), (7, 1.1125107448606286e-06), (8, 4.635896289073571e-07), (9, 6.222317665560695e-07), (10, 9.043120599017129e-07), (11, 1.6768821069490514e-06), (12, 1.4679321793664712e-06), (13, 1.6015061419238918e-06), (14, 1.6047349618020235e-06), (15, 1.7238278360309778e-06), (16, 1.993532578126178e-06), (17, 1.7317195215582615e-06), (18, 1.6667893305566395e-06), (19, 2.404165115876822e-06), (20, 3.400758942007087e-06), (21, 5.82525444769999e-06), (22, 4.915330919175176e-06), (23, 8.863211405696347e-06), (24, 1.805522151698824e-05), (25, 8.842917304718867e-05), (26, 1.819203134800773e-05), (27, 3.206463361493661e-06)]
+
+ 39%|███▉      | 59/150 [09:05<11:33,  7.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 540, Spatial MSE: [(0, 1.0940510719592567e-06), (1, 1.111265373765491e-05), (2, 6.0103327996330336e-05), (3, 1.1367810657247901e-05), (4, 5.1760630412900355e-06), (5, 6.569651304744184e-06), (6, 1.3629770364786964e-05), (7, 1.1487606570881326e-05), (8, 1.4534606634697411e-05), (9, 2.531148675188888e-05), (10, 4.02855257561896e-05), (11, 4.5667100494029e-05), (12, 5.309924017637968e-05), (13, 7.851365808164701e-05), (14, 8.432631148025393e-05), (15, 4.138379517826252e-05), (16, 4.1759918531170115e-05), (17, 2.665058673301246e-05), (18, 6.618037150474265e-05), (19, 3.650410144473426e-05), (20, 3.988505341112614e-05), (21, 4.7776138671906665e-05), (22, 8.106361929094419e-05), (23, 0.00010478610784048215), (24, 0.00040960745536722243), (25, 0.000691272143740207), (26, 0.002157199662178755), (27, 0.00236789183691144)], Temporal MSE: [(0, 8.056855449467548e-07), (1, 3.87639317978028e-07), (2, 9.726496728035272e-07), (3, 1.3748908713751007e-06), (4, 2.0287438928789925e-06), (5, 2.085646656269091e-06), (6, 2.2886918031872483e-06), (7, 2.8456120162445586e-06), (8, 3.0058988613745896e-06), (9, 3.864783593598986e-06), (10, 6.443575330195017e-06), (11, 7.247261692100437e-06), (12, 7.880341399868485e-06), (13, 1.1442925824667327e-05), (14, 1.65257715707412e-05), (15, 1.1562558029254433e-05), (16, 1.0091292097058613e-05), (17, 9.898163625621237e-06), (18, 1.5065346815390512e-05), (19, 1.776805402187165e-05), (20, 2.726666207308881e-05), (21, 2.7089756258646958e-05), (22, 3.564196958905086e-05), (23, 4.890614218311384e-05), (24, 0.0001718864805297926), (25, 0.0003313357010483742), (26, 7.060285861371085e-05), (27, 4.925067514705006e-06)]
+
+ 40%|████      | 60/150 [09:13<11:56,  7.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 534, Spatial MSE: [(0, 7.477379995179945e-07), (1, 6.252535513340263e-06), (2, 2.8308988476055674e-05), (3, 9.0590983745642e-06), (4, 2.0876166217931313e-06), (5, 1.2310480315136374e-06), (6, 5.064322976977564e-06), (7, 3.2370239750889596e-06), (8, 1.6556939499423606e-06), (9, 2.0142067569395294e-06), (10, 2.069467427645577e-06), (11, 2.277772182424087e-06), (12, 3.1759047942614416e-06), (13, 3.2352356356568635e-05), (14, 1.8763779735309072e-05), (15, 5.574192528001731e-06), (16, 1.257348594663199e-05), (17, 3.721768507602974e-06), (18, 6.3461125137109775e-06), (19, 1.0396182005933952e-05), (20, 7.4346635301481e-06), (21, 1.2487977983255405e-05), (22, 1.9132385205011815e-05), (23, 2.5705476218718104e-05), (24, 5.237056757323444e-05), (25, 0.00015128881204873323), (26, 0.0004455893358681351), (27, 0.002170742256566882)], Temporal MSE: [(0, 8.347769835381769e-07), (1, 3.3682928801681555e-07), (2, 4.767139500927442e-07), (3, 5.221942842581484e-07), (4, 3.2762747537162795e-07), (5, 6.445885105677007e-07), (6, 5.726775498260395e-07), (7, 8.093192036540131e-07), (8, 3.493128701848036e-07), (9, 4.6181324364624743e-07), (10, 6.954766149647185e-07), (11, 1.3909834706282709e-06), (12, 1.2422462987160543e-06), (13, 1.3394209190664697e-06), (14, 1.2700377283181297e-06), (15, 1.2546657899292768e-06), (16, 1.8171655256082886e-06), (17, 1.2750952009810135e-06), (18, 1.3514739976017154e-06), (19, 1.8355475503994967e-06), (20, 2.56387079389242e-06), (21, 2.6302463993488345e-06), (22, 3.973884759034263e-06), (23, 7.5345146797189955e-06), (24, 1.4010264749231283e-05), (25, 5.175156184122898e-05), (26, 1.3231460798124317e-05), (27, 9.833609055931447e-07)]
+
+ 41%|████      | 61/150 [09:21<11:40,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 528, Spatial MSE: [(0, 9.815898920351174e-07), (1, 2.353554555156734e-05), (2, 7.40582327125594e-05), (3, 2.4191294869524427e-05), (4, 2.9374763471423648e-05), (5, 3.923426993424073e-05), (6, 7.466061651939526e-05), (7, 0.00013812152610626072), (8, 0.0001767602370819077), (9, 0.0002760779170785099), (10, 0.00029163071303628385), (11, 0.00029488574364222586), (12, 0.00033585974597372115), (13, 0.0009364247089251876), (14, 0.0020870203152298927), (15, 0.0018242110963910818), (16, 0.035563088953495026), (17, 0.0005939611000940204), (18, 0.0007784469053149223), (19, 0.0024628029204905033), (20, 0.002075033960863948), (21, 0.0009439923451282084), (22, 0.0011535889934748411), (23, 0.0014469590969383717), (24, 0.003358458401635289), (25, 0.005144312046468258), (26, 0.004206827841699123), (27, 0.0030108755454421043)], Temporal MSE: [(0, 1.234129058502731e-06), (1, 8.096243391264579e-07), (2, 1.5784174820510088e-06), (3, 1.3012562703806907e-05), (4, 1.1609554348979145e-05), (5, 1.8414209989714436e-05), (6, 4.257389446138404e-05), (7, 3.1675604986958206e-05), (8, 3.098367596976459e-05), (9, 4.9239093641517684e-05), (10, 8.654670091345906e-05), (11, 9.253158350475132e-05), (12, 9.743385453475639e-05), (13, 0.0001659718545852229), (14, 0.00027237358153797686), (15, 0.0002198749571107328), (16, 0.0002888674207497388), (17, 0.00029892154270783067), (18, 0.0003343407006468624), (19, 0.0004336059791967273), (20, 0.0006887193885631859), (21, 0.0005484722787514329), (22, 0.0005281308549456298), (23, 0.0007666957098990679), (24, 0.002343097236007452), (25, 0.004738576244562864), (26, 0.000502148293890059), (27, 2.5414796255063266e-05)]
+
+ 41%|████▏     | 62/150 [09:29<11:27,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 522, Spatial MSE: [(0, 1.4510565051750746e-06), (1, 7.502425887651043e-06), (2, 3.40928127116058e-05), (3, 9.981802577385679e-06), (4, 2.1880637177673634e-06), (5, 1.5745656583021628e-06), (6, 6.205967110872734e-06), (7, 4.2210626816086005e-06), (8, 2.205953023803886e-06), (9, 3.14966200676281e-06), (10, 2.5306164843641454e-06), (11, 3.0419435006479034e-06), (12, 4.372141120256856e-06), (13, 3.102118716924451e-05), (14, 1.5931176676531322e-05), (15, 1.2519576557679102e-05), (16, 1.4403941349883098e-05), (17, 5.035551566834329e-06), (18, 6.395393484126544e-06), (19, 1.2974196579307318e-05), (20, 7.656331035832409e-06), (21, 2.1625080989906564e-05), (22, 1.8375549188931473e-05), (23, 2.680130819499027e-05), (24, 5.449500895338133e-05), (25, 0.00017102985293604434), (26, 0.0007014317088760436), (27, 0.00340801989659667)], Temporal MSE: [(0, 1.1552626801858423e-06), (1, 4.107596680569259e-07), (2, 5.254280495137209e-07), (3, 6.395573564077495e-07), (4, 4.0986026306200074e-07), (5, 7.624963132002449e-07), (6, 6.272433665799326e-07), (7, 1.0844599955817102e-06), (8, 4.568999827370135e-07), (9, 5.939820653111383e-07), (10, 9.809874654820305e-07), (11, 1.7095208022510633e-06), (12, 1.5927546428429196e-06), (13, 1.8678956621442921e-06), (14, 1.9999358755740104e-06), (15, 2.061852001133957e-06), (16, 2.61049899563659e-06), (17, 1.8432193655826268e-06), (18, 1.7577586959305336e-06), (19, 2.9090292628097814e-06), (20, 4.095555141248042e-06), (21, 3.297026069049025e-06), (22, 5.19000332133146e-06), (23, 1.0373882105341181e-05), (24, 1.718678504403215e-05), (25, 0.00014171826478559524), (26, 2.192138163081836e-05), (27, 3.4191682516393485e-06)]
+
+ 42%|████▏     | 63/150 [09:37<11:24,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 516, Spatial MSE: [(0, 1.203362785417994e-06), (1, 1.1905293831659947e-05), (2, 4.9891386879608035e-05), (3, 1.0909850061580073e-05), (4, 5.520173090189928e-06), (5, 6.112180926720612e-06), (6, 1.6880174371181056e-05), (7, 1.3504292837751564e-05), (8, 1.7787668184610084e-05), (9, 3.481849853415042e-05), (10, 5.7401244703214616e-05), (11, 7.137696229619905e-05), (12, 6.668408605037257e-05), (13, 9.893981768982485e-05), (14, 0.00012844281445723027), (15, 5.853211405337788e-05), (16, 5.447659350465983e-05), (17, 3.285301136202179e-05), (18, 0.00018203457875642926), (19, 0.0002571625809650868), (20, 0.0005674101412296295), (21, 0.00014379981439560652), (22, 9.535201388644055e-05), (23, 0.0001211077906191349), (24, 0.0003964260686188936), (25, 0.0006872083758935332), (26, 0.002080495934933424), (27, 0.0016220214311033487)], Temporal MSE: [(0, 8.820783250484965e-07), (1, 3.988918138020381e-07), (2, 9.837838206294691e-07), (3, 1.4742868188477587e-06), (4, 2.04680509341415e-06), (5, 2.2835758954897756e-06), (6, 2.5819447273534024e-06), (7, 3.5864052279066527e-06), (8, 3.2331913644156884e-06), (9, 4.759860075864708e-06), (10, 9.10468224901706e-06), (11, 1.0399553502793424e-05), (12, 1.1013411494786851e-05), (13, 1.5358107702923007e-05), (14, 1.8863911463995464e-05), (15, 1.2397594218782615e-05), (16, 1.2020289432257414e-05), (17, 1.1087092389061581e-05), (18, 1.743680149957072e-05), (19, 2.593694989627693e-05), (20, 4.6076977014308795e-05), (21, 3.1512627174379304e-05), (22, 3.8912079617148265e-05), (23, 5.365393735701218e-05), (24, 0.0001645671291043982), (25, 0.00032001963700167835), (26, 7.855037256376818e-05), (27, 5.731707005907083e-06)]
+
+ 43%|████▎     | 64/150 [09:47<12:10,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 510, Spatial MSE: [(0, 9.092195227822231e-07), (1, 7.036999704723712e-06), (2, 3.174911762471311e-05), (3, 8.495338988723233e-06), (4, 2.2607673599850386e-06), (5, 1.3749253184869303e-06), (6, 4.378731773613254e-06), (7, 3.317046093798126e-06), (8, 1.8414259557175683e-06), (9, 2.8435408694349462e-06), (10, 2.1165976704651257e-06), (11, 2.7493715606397018e-06), (12, 3.8894945646461565e-06), (13, 1.6370426237699576e-05), (14, 9.82169422059087e-06), (15, 5.969923677184852e-06), (16, 8.588062883063685e-06), (17, 4.363277184893377e-06), (18, 1.0331824341847096e-05), (19, 1.5962879842845723e-05), (20, 7.683212970732711e-06), (21, 1.7372101865475997e-05), (22, 1.9464370780042373e-05), (23, 2.630622657306958e-05), (24, 4.768463259097189e-05), (25, 0.00014593279047403485), (26, 0.0004274296807125211), (27, 0.0020806118845939636)], Temporal MSE: [(0, 9.720685056890943e-07), (1, 4.0287011415784946e-07), (2, 5.52927588159946e-07), (3, 5.988333668938139e-07), (4, 3.918755737686297e-07), (5, 7.268525337167375e-07), (6, 7.308652811843785e-07), (7, 9.048330298355722e-07), (8, 4.1020959429260984e-07), (9, 5.291400384521694e-07), (10, 8.043638786148222e-07), (11, 1.3239593954494921e-06), (12, 1.214181224895583e-06), (13, 1.2799434898624895e-06), (14, 1.5470493508473737e-06), (15, 1.3283375892569893e-06), (16, 1.88623528174503e-06), (17, 1.6749565929785604e-06), (18, 2.8573972485901322e-06), (19, 3.1211748137138784e-06), (20, 4.165125574218109e-06), (21, 2.735346470217337e-06), (22, 4.23513392888708e-06), (23, 7.054227353364695e-06), (24, 1.3033874893153552e-05), (25, 6.223208038136363e-05), (26, 1.4920522517058998e-05), (27, 1.3442055433188216e-06)]
+
+ 43%|████▎     | 65/150 [09:53<11:21,  8.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 504, Spatial MSE: [(0, 9.700296459413948e-07), (1, 1.9409375454415567e-05), (2, 7.550997543148696e-05), (3, 2.2678592358715832e-05), (4, 2.7434207368060015e-05), (5, 3.431928053032607e-05), (6, 0.00010021865455200896), (7, 0.00016123820387292653), (8, 0.00017429000581614673), (9, 0.0002604023029562086), (10, 0.00030153480474837124), (11, 0.00029332967824302614), (12, 0.0003227568813599646), (13, 0.0007207830785773695), (14, 0.0014445037813857198), (15, 0.001114947022870183), (16, 0.017712116241455078), (17, 0.0005622678436338902), (18, 0.000657782657071948), (19, 0.0014541179407387972), (20, 0.0013886027736589313), (21, 0.0008841906092129648), (22, 0.0008309847908094525), (23, 0.0011645276099443436), (24, 0.002517098793759942), (25, 0.004135308787226677), (26, 0.005423559807240963), (27, 0.003418001113459468)], Temporal MSE: [(0, 1.04776052012312e-06), (1, 7.13128258666984e-07), (2, 1.7708541690808488e-06), (3, 1.2081089153070934e-05), (4, 9.308193511969876e-06), (5, 2.2230953618418425e-05), (6, 6.780427065677941e-05), (7, 3.790498521993868e-05), (8, 3.0060880817472935e-05), (9, 4.894439916824922e-05), (10, 8.570852514822036e-05), (11, 8.967758913058788e-05), (12, 8.883521513780579e-05), (13, 0.00015118077863007784), (14, 0.00024412930360995233), (15, 0.0002024604327743873), (16, 0.0002481923729646951), (17, 0.00024930620566010475), (18, 0.00026299277669750154), (19, 0.0003671411832328886), (20, 0.0005807580309920013), (21, 0.00043972249841317534), (22, 0.00043255568016320467), (23, 0.0006065097404643893), (24, 0.0019972091540694237), (25, 0.004063449800014496), (26, 0.0004379362508188933), (27, 3.0425535442191176e-05)]
+
+ 44%|████▍     | 66/150 [10:01<11:01,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 498, Spatial MSE: [(0, 8.106537165986083e-07), (1, 6.887762083351845e-06), (2, 2.4415992811555043e-05), (3, 8.426354725088459e-06), (4, 1.8793219851431786e-06), (5, 1.2133646123402286e-06), (6, 4.071072908118367e-06), (7, 3.032854692719411e-06), (8, 1.7068859961000271e-06), (9, 2.054217702607275e-06), (10, 1.937914703376009e-06), (11, 2.505617203496513e-06), (12, 4.311215889174491e-06), (13, 4.040469866595231e-05), (14, 2.3955512006068602e-05), (15, 8.809880455373786e-06), (16, 3.0276618417701684e-05), (17, 9.140193469647784e-06), (18, 9.011989277496468e-06), (19, 1.227741904585855e-05), (20, 6.300535005721031e-06), (21, 1.3196297913964372e-05), (22, 2.1450217900564894e-05), (23, 2.7571253667701967e-05), (24, 5.689813770004548e-05), (25, 0.00015616988821420819), (26, 0.00045177413267083466), (27, 0.001979873050004244)], Temporal MSE: [(0, 7.581364229736209e-07), (1, 3.1258613830686954e-07), (2, 4.263608559540444e-07), (3, 5.332396426638297e-07), (4, 3.5095982298116724e-07), (5, 6.336937872220005e-07), (6, 6.100455038904329e-07), (7, 8.149338555085706e-07), (8, 3.8582791717090004e-07), (9, 4.866930112257251e-07), (10, 7.417465894832276e-07), (11, 1.6711696844140533e-06), (12, 1.560338773742842e-06), (13, 1.8499874840927077e-06), (14, 1.6206211057578912e-06), (15, 1.8855072312362608e-06), (16, 2.1574032871285453e-06), (17, 2.524447609175695e-06), (18, 1.7008984514177428e-06), (19, 2.0723325633298373e-06), (20, 3.384594037925126e-06), (21, 4.192547294223914e-06), (22, 4.965287644154159e-06), (23, 8.546184290025849e-06), (24, 1.6108904674183577e-05), (25, 6.538892193930224e-05), (26, 1.5808516764082015e-05), (27, 1.4016646900927299e-06)]
+
+ 45%|████▍     | 67/150 [10:09<10:51,  7.85s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 492, Spatial MSE: [(0, 8.646088645036798e-07), (1, 9.763295565790031e-06), (2, 3.6411540349945426e-05), (3, 8.936317499319557e-06), (4, 4.4818170863436535e-06), (5, 5.831650469190208e-06), (6, 1.2806158338207752e-05), (7, 9.323024642071687e-06), (8, 1.2890110156149603e-05), (9, 2.5751020075404085e-05), (10, 2.9787968742311932e-05), (11, 4.147888103034347e-05), (12, 4.562175672617741e-05), (13, 0.00010560478403931484), (14, 0.00014164255117066205), (15, 6.891036900924519e-05), (16, 0.00010289034980814904), (17, 2.4174350983230397e-05), (18, 7.732836820650846e-05), (19, 8.319787593791261e-05), (20, 5.309861444402486e-05), (21, 4.344581611803733e-05), (22, 7.182589615695179e-05), (23, 8.726702799322084e-05), (24, 0.00029043422546237707), (25, 0.0006146571249701083), (26, 0.0015548942610621452), (27, 0.001464945380575955)], Temporal MSE: [(0, 6.689419933536556e-07), (1, 3.2426817142550135e-07), (2, 7.337333727264195e-07), (3, 1.4054104440219817e-06), (4, 1.992696297747898e-06), (5, 1.9275805698271142e-06), (6, 2.397471234871773e-06), (7, 2.594815896372893e-06), (8, 2.6380423605587566e-06), (9, 3.1715612749394495e-06), (10, 5.54979396838462e-06), (11, 6.9698689912911505e-06), (12, 7.959955837577581e-06), (13, 1.166109177574981e-05), (14, 1.661168244027067e-05), (15, 1.088172666641185e-05), (16, 9.786936061573215e-06), (17, 8.398082172789145e-06), (18, 1.164642526418902e-05), (19, 1.3699701412406284e-05), (20, 2.1787642253912054e-05), (21, 2.195855267927982e-05), (22, 3.0026105378055945e-05), (23, 4.115826232009567e-05), (24, 0.00011767187970690429), (25, 0.00028038641903549433), (26, 7.13277913746424e-05), (27, 5.574327133217594e-06)]
+
+ 45%|████▌     | 68/150 [10:17<10:41,  7.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 486, Spatial MSE: [(0, 8.866122698236722e-07), (1, 6.3722231971041765e-06), (2, 2.4123764887917787e-05), (3, 8.017536856641527e-06), (4, 1.727335757095716e-06), (5, 1.3429626051220112e-06), (6, 4.135572908126051e-06), (7, 4.673482635553228e-06), (8, 1.8612122403283138e-06), (9, 1.9964459170296323e-06), (10, 2.117753183483728e-06), (11, 2.5400313461432233e-06), (12, 4.7322055252152495e-06), (13, 5.5499924201285467e-05), (14, 3.217515768483281e-05), (15, 1.3409994608082343e-05), (16, 3.1344941817224026e-05), (17, 6.530733116960619e-06), (18, 1.2232354492880404e-05), (19, 1.4957508938095998e-05), (20, 6.815019787609344e-06), (21, 1.7325910448562354e-05), (22, 2.662502447492443e-05), (23, 3.0905706807971e-05), (24, 5.1535196689656004e-05), (25, 0.00014860424562357366), (26, 0.0005940996343269944), (27, 0.0020985063165426254)], Temporal MSE: [(0, 8.430276352555666e-07), (1, 3.5003466791749815e-07), (2, 4.679469896018418e-07), (3, 5.830136160511756e-07), (4, 3.940952524317254e-07), (5, 7.044534982014738e-07), (6, 6.938983574400481e-07), (7, 1.0419723821541993e-06), (8, 4.161344122621813e-07), (9, 5.208587481320137e-07), (10, 8.644073545838182e-07), (11, 2.0662864699261263e-06), (12, 1.971872507056105e-06), (13, 2.047822817985434e-06), (14, 2.1275791368680075e-06), (15, 1.8212043642051867e-06), (16, 1.7330560240225168e-06), (17, 1.5789743201821693e-06), (18, 1.5866778539930237e-06), (19, 2.377204964432167e-06), (20, 2.934256372100208e-06), (21, 6.0512970776471775e-06), (22, 5.203942237130832e-06), (23, 1.2153011084592436e-05), (24, 1.8663931768969633e-05), (25, 7.279223791556433e-05), (26, 1.9727080143638887e-05), (27, 2.0604097699106205e-06)]
+
+ 46%|████▌     | 69/150 [10:24<10:24,  7.72s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 480, Spatial MSE: [(0, 8.710459837857343e-07), (1, 1.7293888959102333e-05), (2, 4.848519529332407e-05), (3, 1.7846934497356415e-05), (4, 2.5794137400225736e-05), (5, 3.400211062398739e-05), (6, 6.409206980606541e-05), (7, 0.00012992775009479374), (8, 0.00016421977488789707), (9, 0.00025193116744048893), (10, 0.0002723479992710054), (11, 0.0002731214335653931), (12, 0.0002910411567427218), (13, 0.0008522220305167139), (14, 0.0020383347291499376), (15, 0.0016235106159001589), (16, 0.030198920518159866), (17, 0.0005882928380742669), (18, 0.0006912884418852627), (19, 0.0031299195252358913), (20, 0.0032792657148092985), (21, 0.00116067030467093), (22, 0.0008057332597672939), (23, 0.001233978196978569), (24, 0.0027380483224987984), (25, 0.004859226755797863), (26, 0.004824006464332342), (27, 0.0022307883482426405)], Temporal MSE: [(0, 9.48979163695185e-07), (1, 5.267622213978029e-07), (2, 1.6183339539566077e-06), (3, 1.7923604900715873e-05), (4, 9.398186193720903e-06), (5, 1.7491915059508756e-05), (6, 4.4290627556620166e-05), (7, 3.201223080395721e-05), (8, 2.6414803869556636e-05), (9, 4.3081694457214326e-05), (10, 7.34164786990732e-05), (11, 7.623378041898832e-05), (12, 7.688474579481408e-05), (13, 0.00013660472177434713), (14, 0.00023694704577792436), (15, 0.0001878010662039742), (16, 0.00021064971224404871), (17, 0.00021389409084804356), (18, 0.00026470309239812195), (19, 0.0003834289964288473), (20, 0.0006642625667154789), (21, 0.000510594982188195), (22, 0.00045636107097379863), (23, 0.0006813592044636607), (24, 0.0021983201149851084), (25, 0.004461194388568401), (26, 0.0005376730696298182), (27, 2.5880084649543278e-05)]
+
+ 47%|████▋     | 70/150 [10:32<10:23,  7.80s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 474, Spatial MSE: [(0, 1.2135840279370314e-06), (1, 6.91331160851405e-06), (2, 2.4462013243464753e-05), (3, 8.143659215420485e-06), (4, 1.9675092062243493e-06), (5, 1.319928287557559e-06), (6, 4.494515906117158e-06), (7, 3.912850843335036e-06), (8, 1.8963804677696317e-06), (9, 1.9811413949355483e-06), (10, 2.110582499881275e-06), (11, 2.9761126825178508e-06), (12, 6.726816081936704e-06), (13, 0.00011661747703328729), (14, 4.3472351535456255e-05), (15, 1.274820533581078e-05), (16, 0.0007310050423257053), (17, 1.0287325494573452e-05), (18, 2.3199216229841113e-05), (19, 7.901948265498504e-05), (20, 2.2835436539025977e-05), (21, 5.079179027234204e-05), (22, 4.609318057191558e-05), (23, 3.134712824248709e-05), (24, 5.937277455814183e-05), (25, 0.0001636232336750254), (26, 0.0004525180265773088), (27, 0.0024069726932793856)], Temporal MSE: [(0, 1.011937683870201e-06), (1, 3.34091112108581e-07), (2, 4.6055447455728427e-07), (3, 6.233813110156916e-07), (4, 3.9552622865812737e-07), (5, 6.655813535871857e-07), (6, 6.925035904714605e-07), (7, 9.721477454149863e-07), (8, 4.13243185448664e-07), (9, 5.346005877981952e-07), (10, 8.340249451066484e-07), (11, 3.140806711598998e-06), (12, 3.931545052182628e-06), (13, 3.1765193853061646e-06), (14, 3.2350637866329635e-06), (15, 5.693355888070073e-06), (16, 4.835385880141985e-06), (17, 4.444517344381893e-06), (18, 6.253004357859027e-06), (19, 7.237482805066975e-06), (20, 8.991676622827072e-06), (21, 6.9701891334261745e-06), (22, 1.1048929081880488e-05), (23, 1.7381520592607558e-05), (24, 2.3355582015938126e-05), (25, 9.118121670326218e-05), (26, 1.9627332221716642e-05), (27, 3.80349365514121e-06)]
+
+ 47%|████▋     | 71/150 [10:40<10:14,  7.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 468, Spatial MSE: [(0, 1.0073557632495067e-06), (1, 8.911108125175815e-06), (2, 3.4985252568731084e-05), (3, 8.747842002776451e-06), (4, 4.278354481357383e-06), (5, 5.94323682889808e-06), (6, 1.4771960195503198e-05), (7, 1.0718261364672799e-05), (8, 1.4104233741818462e-05), (9, 3.119548273389228e-05), (10, 4.1573082853574306e-05), (11, 5.754704761784524e-05), (12, 5.992626029183157e-05), (13, 0.0001925625983858481), (14, 0.00012845486344303936), (15, 4.918960985378362e-05), (16, 0.0007681892602704465), (17, 2.9158434699638747e-05), (18, 6.37584671494551e-05), (19, 0.00010198137897532433), (20, 5.192222670302726e-05), (21, 7.522417581640184e-05), (22, 9.876669355435297e-05), (23, 0.00010004890646087006), (24, 0.00032028224086388946), (25, 0.0007429143879562616), (26, 0.0024991563986986876), (27, 0.001509938738308847)], Temporal MSE: [(0, 6.732537940479233e-07), (1, 3.493193787562632e-07), (2, 7.054597404021479e-07), (3, 1.3467741837303038e-06), (4, 1.869323796199751e-06), (5, 2.0240217963873874e-06), (6, 2.9260656901897164e-06), (7, 3.3694486774038523e-06), (8, 3.3272301607212285e-06), (9, 4.1061825868382584e-06), (10, 7.062316399242263e-06), (11, 9.649568710301537e-06), (12, 1.2632717698579654e-05), (13, 1.5105761121958494e-05), (14, 1.9550589058781043e-05), (15, 1.6114116078824736e-05), (16, 1.3037374628765974e-05), (17, 1.2726542081509251e-05), (18, 1.727216294966638e-05), (19, 1.975230043171905e-05), (20, 3.048614780709613e-05), (21, 2.562091867730487e-05), (22, 3.551871486706659e-05), (23, 5.008053994970396e-05), (24, 0.0001357358123641461), (25, 0.00032011658186092973), (26, 7.792632095515728e-05), (27, 6.408436547644669e-06)]
+
+ 48%|████▊     | 72/150 [10:47<09:50,  7.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 462, Spatial MSE: [(0, 7.288001597771654e-07), (1, 5.427215910458472e-06), (2, 1.7053460396709852e-05), (3, 6.040962489350932e-06), (4, 1.2678639222940546e-06), (5, 1.1005524811480427e-06), (6, 2.885525418605539e-06), (7, 2.7383086944610113e-06), (8, 1.4443696727539646e-06), (9, 2.2164531401358545e-06), (10, 1.6259911035376717e-06), (11, 2.4760831820458407e-06), (12, 3.892960648954613e-06), (13, 4.467637336347252e-05), (14, 2.0958214008715004e-05), (15, 1.0390939678472932e-05), (16, 1.922925548569765e-05), (17, 4.08534060625243e-06), (18, 4.874169007962337e-06), (19, 9.427016266272403e-06), (20, 5.678764409822179e-06), (21, 1.4951525372453034e-05), (22, 1.7261812899960205e-05), (23, 2.0897497961414047e-05), (24, 4.385890133562498e-05), (25, 0.00011745205847546458), (26, 0.0003741155087482184), (27, 0.0014774935552850366)], Temporal MSE: [(0, 6.337863851513248e-07), (1, 2.5220003863068996e-07), (2, 3.523917939673993e-07), (3, 4.843138299293059e-07), (4, 3.517284881127125e-07), (5, 6.108598427090328e-07), (6, 6.652147703789524e-07), (7, 7.511685566896631e-07), (8, 3.7271874475663935e-07), (9, 4.373686977032776e-07), (10, 7.278065368154785e-07), (11, 1.8716325485002017e-06), (12, 2.1146252038306557e-06), (13, 1.8523783182899933e-06), (14, 1.851614797487855e-06), (15, 2.1418168216769118e-06), (16, 2.0333284282969544e-06), (17, 1.426094854650728e-06), (18, 1.4388868976311642e-06), (19, 2.1391174414020497e-06), (20, 3.0559444894606713e-06), (21, 2.4922935608628904e-06), (22, 4.183129476587055e-06), (23, 8.243477168434765e-06), (24, 1.392341891914839e-05), (25, 6.66168489260599e-05), (26, 1.5071758753038011e-05), (27, 1.4372027408171562e-06)]
+
+ 49%|████▊     | 73/150 [10:54<09:42,  7.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 456, Spatial MSE: [(0, 8.085429499260499e-07), (1, 1.3604717423731927e-05), (2, 3.8833699363749474e-05), (3, 1.792177681636531e-05), (4, 2.3625054382137023e-05), (5, 3.163883957313374e-05), (6, 9.798338578548282e-05), (7, 0.00015400658594444394), (8, 0.0001604013959877193), (9, 0.00024110918457154185), (10, 0.0002592222881503403), (11, 0.0002496141823939979), (12, 0.00029058472136966884), (13, 0.0004840866313315928), (14, 0.0008934028446674347), (15, 0.0009871323127299547), (16, 0.011303073726594448), (17, 0.0007554514450021088), (18, 0.0007807544316165149), (19, 0.001451774500310421), (20, 0.0023347227834165096), (21, 0.0021017640829086304), (22, 0.001021645963191986), (23, 0.001602605334483087), (24, 0.003196427132934332), (25, 0.006058068946003914), (26, 0.00589211517944932), (27, 0.0022351695224642754)], Temporal MSE: [(0, 9.489691024100466e-07), (1, 5.369113864617248e-07), (2, 1.5752423223602818e-06), (3, 1.2999619684705976e-05), (4, 8.106233508442529e-06), (5, 1.8301730960956775e-05), (6, 6.0705348005285487e-05), (7, 3.5310684324940667e-05), (8, 2.5844146875897422e-05), (9, 4.161481047049165e-05), (10, 7.085087418090552e-05), (11, 7.560729864053428e-05), (12, 7.917352195363492e-05), (13, 0.00014003437536302954), (14, 0.00023096449149306864), (15, 0.00023643042368348688), (16, 0.00030337064526975155), (17, 0.0002988744236063212), (18, 0.00035206644679419696), (19, 0.0005892068147659302), (20, 0.0011909374734386802), (21, 0.0008819843642413616), (22, 0.0005793495220132172), (23, 0.000871339172590524), (24, 0.0028088390827178955), (25, 0.005766662303358316), (26, 0.0006304241833277047), (27, 2.4975932319648564e-05)]
+
+ 49%|████▉     | 74/150 [11:03<10:01,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 450, Spatial MSE: [(0, 6.896983677506796e-07), (1, 6.812384981458308e-06), (2, 1.8462451407685876e-05), (3, 6.934291377547197e-06), (4, 1.7274363699470996e-06), (5, 1.1617285053944215e-06), (6, 2.7589126148086507e-06), (7, 2.7412388590164483e-06), (8, 1.5022271782072494e-06), (9, 2.1457096863741754e-06), (10, 1.7358553350277361e-06), (11, 2.27047894441057e-06), (12, 4.288531727070222e-06), (13, 4.824345523957163e-05), (14, 4.757316128234379e-05), (15, 7.148826989578083e-05), (16, 0.007543550804257393), (17, 3.582513818400912e-05), (18, 3.150377960992046e-05), (19, 0.0002872343175113201), (20, 0.00031258619856089354), (21, 7.749123324174434e-05), (22, 8.965857705334201e-05), (23, 4.762172466143966e-05), (24, 8.782721852185205e-05), (25, 0.000208993413252756), (26, 0.00037610187428072095), (27, 0.0036789115983992815)], Temporal MSE: [(0, 6.611713274651265e-07), (1, 2.7556191639632743e-07), (2, 3.8212519370972586e-07), (3, 5.614315909951983e-07), (4, 3.8794516399320855e-07), (5, 6.254065283428645e-07), (6, 6.506625709334912e-07), (7, 8.188697506739118e-07), (8, 4.055716260609188e-07), (9, 4.981814072380075e-07), (10, 7.797792136443604e-07), (11, 1.867015271272976e-06), (12, 2.0454433524719207e-06), (13, 1.8918379964816268e-06), (14, 3.144209358652006e-06), (15, 4.150446784478845e-06), (16, 4.460191576072248e-06), (17, 7.7112163126003e-06), (18, 1.267394964088453e-05), (19, 1.5136704860196915e-05), (20, 1.3947197658126242e-05), (21, 2.957098877232056e-05), (22, 2.1678930352209136e-05), (23, 3.723664485733025e-05), (24, 4.423243444762193e-05), (25, 8.858507499098778e-05), (26, 1.9727604012587108e-05), (27, 1.7631287846597843e-05)]
+
+ 50%|█████     | 75/150 [11:12<10:16,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 444, Spatial MSE: [(0, 9.209621794070699e-07), (1, 8.245595381595194e-06), (2, 2.8568952984642237e-05), (3, 7.277406893990701e-06), (4, 4.785983037436381e-06), (5, 5.36822108188062e-06), (6, 1.3193684935686179e-05), (7, 1.1720614566002041e-05), (8, 1.5451623767148703e-05), (9, 3.576199378585443e-05), (10, 5.5050513765309006e-05), (11, 7.748918869765475e-05), (12, 8.049359894357622e-05), (13, 0.00037601214717142284), (14, 0.0007220670813694596), (15, 0.00043120424379594624), (16, 0.012765605002641678), (17, 5.665297430823557e-05), (18, 8.767427789280191e-05), (19, 0.0007020772900432348), (20, 0.0006727435975335538), (21, 8.076278754742816e-05), (22, 0.00015021738363429904), (23, 0.0001341044990113005), (24, 0.0003593590226955712), (25, 0.0008162554586306214), (26, 0.0023143861908465624), (27, 0.003575635841116309)], Temporal MSE: [(0, 6.256036044760549e-07), (1, 3.300220896562678e-07), (2, 7.269081265803834e-07), (3, 1.4795386960031465e-06), (4, 1.917217105074087e-06), (5, 2.2395286123355618e-06), (6, 3.2655486847943394e-06), (7, 3.91808544009109e-06), (8, 3.344200649735285e-06), (9, 4.817346052732319e-06), (10, 9.750898243510164e-06), (11, 1.5708097635069862e-05), (12, 1.8761847968562506e-05), (13, 2.4308477804879658e-05), (14, 2.9904726034146734e-05), (15, 2.8926322556799278e-05), (16, 1.947516284417361e-05), (17, 1.8826767700375058e-05), (18, 2.9699311198783107e-05), (19, 3.24118445860222e-05), (20, 4.928822090732865e-05), (21, 4.438741598278284e-05), (22, 5.8546018408378586e-05), (23, 8.521677227690816e-05), (24, 0.00015716682537458837), (25, 0.0003606403188314289), (26, 9.085989586310461e-05), (27, 2.230309291917365e-05)]
+
+ 51%|█████     | 76/150 [11:20<09:51,  7.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 438, Spatial MSE: [(0, 1.123651600209996e-06), (1, 6.858078904770082e-06), (2, 1.770008384482935e-05), (3, 6.665567070740508e-06), (4, 1.5447797068190994e-06), (5, 1.152453137365228e-06), (6, 3.2847249258338707e-06), (7, 3.719405185620417e-06), (8, 1.7574989215063397e-06), (9, 1.9137064555252437e-06), (10, 4.87503257318167e-06), (11, 2.9135671866242774e-05), (12, 8.642604370834306e-05), (13, 0.0020046548452228308), (14, 0.0015815675724297762), (15, 0.0003950833634007722), (16, 0.017956484109163284), (17, 6.257779750740156e-05), (18, 7.046295650070533e-05), (19, 0.0014342244248837233), (20, 0.0009170848643407226), (21, 0.00010799077426781878), (22, 0.0001995669153984636), (23, 9.651388245401904e-05), (24, 0.00017358183686155826), (25, 0.000534493534360081), (26, 0.000593696953728795), (27, 0.01526173111051321)], Temporal MSE: [(0, 8.60451052631106e-07), (1, 2.8209638003318105e-07), (2, 3.7582441336780903e-07), (3, 5.549189836528967e-07), (4, 3.847889615826716e-07), (5, 6.418499083338247e-07), (6, 7.118917437765049e-07), (7, 9.121629318542546e-07), (8, 4.0927693589765113e-07), (9, 5.486422764988674e-07), (10, 2.1809526060678763e-06), (11, 7.733705569989979e-05), (12, 4.7997345973271877e-05), (13, 4.239078043610789e-05), (14, 2.0290071915951557e-05), (15, 3.393336373846978e-05), (16, 2.2912943677511066e-05), (17, 1.9302980945212767e-05), (18, 2.9488983273040503e-05), (19, 2.418444273644127e-05), (20, 1.782272374839522e-05), (21, 4.71064486191608e-05), (22, 5.8131110563408583e-05), (23, 0.00011119029659312218), (24, 0.00013061676872894168), (25, 0.00016582115495111793), (26, 4.7876073949737474e-05), (27, 5.2001400035806e-05)]
+
+ 51%|█████▏    | 77/150 [11:28<09:46,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 432, Spatial MSE: [(0, 9.642255918151932e-07), (1, 1.4926383300917223e-05), (2, 3.9380240195896477e-05), (3, 1.5254911886586342e-05), (4, 2.414999653410632e-05), (5, 3.180208659614436e-05), (6, 6.556119478773326e-05), (7, 0.00013283737644087523), (8, 0.0001489911082899198), (9, 0.00021797379304189235), (10, 0.00019590664305724204), (11, 0.000233862767345272), (12, 0.0003052090760320425), (13, 0.002702143508940935), (14, 0.003034906694665551), (15, 0.0015173372812569141), (16, 0.039470452815294266), (17, 0.0006134694558568299), (18, 0.0006311754696071148), (19, 0.003118948545306921), (20, 0.0027016932144761086), (21, 0.00136309745721519), (22, 0.000934313633479178), (23, 0.0012880637077614665), (24, 0.0026231270749121904), (25, 0.0047379788011312485), (26, 0.00518804183229804), (27, 0.01532950159162283)], Temporal MSE: [(0, 9.689111948318896e-07), (1, 4.949075105287193e-07), (2, 1.616265763004776e-06), (3, 1.5381792763946578e-05), (4, 8.820345101412386e-06), (5, 1.733312092255801e-05), (6, 5.746767419623211e-05), (7, 3.351348277647048e-05), (8, 2.4061691874521784e-05), (9, 3.740977626875974e-05), (10, 6.327467417577282e-05), (11, 0.0001348999358015135), (12, 0.00011619163706200197), (13, 0.0001665229647187516), (14, 0.00022479338804259896), (15, 0.0002169730287278071), (16, 0.00025368976639583707), (17, 0.0002440144307911396), (18, 0.0002765742246992886), (19, 0.00040293423808179796), (20, 0.0006826003664173186), (21, 0.0005953523213975132), (22, 0.0004892022698186338), (23, 0.0007395786815322936), (24, 0.0022979816421866417), (25, 0.004022397566586733), (26, 0.0005256037693470716), (27, 7.71762352087535e-05)]
+
+ 52%|█████▏    | 78/150 [11:36<09:43,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 426, Spatial MSE: [(0, 5.168719212633732e-07), (1, 5.797097855975153e-06), (2, 1.355611857434269e-05), (3, 6.608593594137346e-06), (4, 1.4173132285577594e-06), (5, 1.0697713150875643e-06), (6, 2.0516058611974586e-06), (7, 2.794360398183926e-06), (8, 1.4039300140211708e-06), (9, 1.64864388807473e-06), (10, 2.1537416614592075e-06), (11, 4.585773694998352e-06), (12, 1.8363884009886533e-05), (13, 0.0007070882711559534), (14, 0.0007881855126470327), (15, 0.0003083315386902541), (16, 0.009069694206118584), (17, 2.1692940208595246e-05), (18, 3.400430432520807e-05), (19, 0.0004926691181026399), (20, 0.00017604204185772687), (21, 0.00021846745221409947), (22, 0.0001543721155030653), (23, 7.808036025380716e-05), (24, 0.00010832552652573213), (25, 0.00034875262645073235), (26, 0.0003692150057759136), (27, 0.008940786123275757)], Temporal MSE: [(0, 5.224963501859747e-07), (1, 2.3358272471796226e-07), (2, 3.27703816083158e-07), (3, 5.357900931812765e-07), (4, 3.9959306263881444e-07), (5, 6.333335704766796e-07), (6, 8.432659797108499e-07), (7, 8.20822947389388e-07), (8, 4.0319923755305354e-07), (9, 4.692256538874062e-07), (10, 9.218840091307356e-07), (11, 7.72251769376453e-06), (12, 1.1333274414937478e-05), (13, 2.7211364795221016e-05), (14, 1.2802215678675566e-05), (15, 1.4717279555043206e-05), (16, 1.0144228326680604e-05), (17, 6.020237833581632e-06), (18, 1.834379327192437e-05), (19, 1.3370254237088375e-05), (20, 1.0085629583045375e-05), (21, 1.9978988348157145e-05), (22, 2.7337431674823165e-05), (23, 7.195752550614998e-05), (24, 9.77938761934638e-05), (25, 9.138609311776236e-05), (26, 3.3173473639180884e-05), (27, 2.7529356884770095e-05)]
+
+ 53%|█████▎    | 79/150 [11:43<09:18,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 420, Spatial MSE: [(0, 8.849073083183612e-07), (1, 7.95676442066906e-06), (2, 2.0284018319216557e-05), (3, 7.981752787600271e-06), (4, 4.1297416828456335e-06), (5, 6.379411388479639e-06), (6, 1.865597914729733e-05), (7, 1.1325208106427453e-05), (8, 1.7336838936898857e-05), (9, 3.286968421889469e-05), (10, 3.952936458517797e-05), (11, 5.5983688071137294e-05), (12, 6.62915626890026e-05), (13, 0.000759845890570432), (14, 0.0008654164848849177), (15, 0.00032354987342841923), (16, 0.00899872649461031), (17, 4.000929766334593e-05), (18, 6.042030508979224e-05), (19, 0.0005013606860302389), (20, 0.0002037181257037446), (21, 0.0001325414632447064), (22, 0.0002106973552145064), (23, 0.00014389894204214215), (24, 0.0003707022697199136), (25, 0.0010164969135075808), (26, 0.0026821887586265802), (27, 0.00939112063497305)], Temporal MSE: [(0, 6.591490091523156e-07), (1, 3.81885826072903e-07), (2, 6.379543719958747e-07), (3, 1.5991953432603623e-06), (4, 2.0006648355774814e-06), (5, 2.2666947643301683e-06), (6, 3.9375072447001e-06), (7, 4.7881599130050745e-06), (8, 3.586369984986959e-06), (9, 4.589608579408377e-06), (10, 7.588147127535194e-06), (11, 1.3304273124958854e-05), (12, 1.8956048734253272e-05), (13, 3.82322832592763e-05), (14, 2.9692277166759595e-05), (15, 2.4957615096354857e-05), (16, 1.7885849956655875e-05), (17, 1.5203811017272528e-05), (18, 2.9058055588393472e-05), (19, 2.570097603893373e-05), (20, 3.271507011959329e-05), (21, 3.9800081140128896e-05), (22, 4.922886364511214e-05), (23, 9.940827294485644e-05), (24, 0.00019945357053074986), (25, 0.00042383206891827285), (26, 9.804854926187545e-05), (27, 3.364208168932237e-05)]
+
+ 53%|█████▎    | 80/150 [11:50<08:57,  7.68s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 414, Spatial MSE: [(0, 7.191986242105486e-07), (1, 6.373267297021812e-06), (2, 1.6712270735297352e-05), (3, 5.94185712543549e-06), (4, 1.4002188208905864e-06), (5, 1.0809458217408974e-06), (6, 2.2380786504072603e-06), (7, 2.2806070774095133e-06), (8, 1.4185175132297445e-06), (9, 2.2482431631942745e-06), (10, 1.7812892565416405e-06), (11, 2.806651536957361e-06), (12, 6.303182999545243e-06), (13, 0.00011369453568477184), (14, 0.00015739884111098945), (15, 0.00011768453987315297), (16, 0.005726948846131563), (17, 1.756716847012285e-05), (18, 3.0255867386586033e-05), (19, 0.00027320245862938464), (20, 6.933935219421983e-05), (21, 0.00011307505337754264), (22, 0.00012430903734639287), (23, 5.751515709562227e-05), (24, 9.11591705516912e-05), (25, 0.00024772374308668077), (26, 0.00041587359737604856), (27, 0.0033321205992251635)], Temporal MSE: [(0, 6.040932021278422e-07), (1, 2.5262011149607133e-07), (2, 3.5235203199590615e-07), (3, 5.346983584786358e-07), (4, 3.777113875003124e-07), (5, 6.077489729250374e-07), (6, 6.640354968112661e-07), (7, 7.388152312159946e-07), (8, 3.827780972187611e-07), (9, 4.5725664676865563e-07), (10, 7.74610100506834e-07), (11, 2.617432755869231e-06), (12, 3.1901781767373905e-06), (13, 6.109491096140118e-06), (14, 5.533699095394695e-06), (15, 9.70391101873247e-06), (16, 6.513413609354757e-06), (17, 5.545372459891951e-06), (18, 1.5144079952733591e-05), (19, 1.2301520655455533e-05), (20, 9.595241863280535e-06), (21, 1.3268286238599103e-05), (22, 2.150219188479241e-05), (23, 4.920076753478497e-05), (24, 5.540782512980513e-05), (25, 8.928646275307983e-05), (26, 2.555098762968555e-05), (27, 8.673446245666128e-06)]
+
+ 54%|█████▍    | 81/150 [11:58<08:40,  7.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 408, Spatial MSE: [(0, 7.3451724347251e-07), (1, 1.388388318446232e-05), (2, 2.9298802473931573e-05), (3, 1.3168898476578761e-05), (4, 2.2432346668210812e-05), (5, 2.968636908917688e-05), (6, 6.848845077911392e-05), (7, 0.00013840864994563162), (8, 0.0001402095367666334), (9, 0.0001952237798832357), (10, 0.00018820185505319387), (11, 0.00020093053171876818), (12, 0.00023487886937800795), (13, 0.001123772468417883), (14, 0.002615636447444558), (15, 0.002069689566269517), (16, 0.04811938479542732), (17, 0.00056550616864115), (18, 0.0006472487584687769), (19, 0.004561356268823147), (20, 0.004246654454618692), (21, 0.0009106002980843186), (22, 0.0008107688627205789), (23, 0.0011597657576203346), (24, 0.0023572188802063465), (25, 0.004284770227968693), (26, 0.0050660730339586735), (27, 0.0029672463424503803)], Temporal MSE: [(0, 9.021681535159587e-07), (1, 4.3692344320334087e-07), (2, 1.5870618881308474e-06), (3, 1.2931513992953114e-05), (4, 8.235952918766998e-06), (5, 1.642948882363271e-05), (6, 5.63890062039718e-05), (7, 3.293555346317589e-05), (8, 2.2446087314165197e-05), (9, 3.453678436926566e-05), (10, 5.6312837841687724e-05), (11, 6.421109719667584e-05), (12, 7.940246723592281e-05), (13, 0.00013472764112520963), (14, 0.00020614909590221941), (15, 0.00019384865299798548), (16, 0.0002297725441167131), (17, 0.0002166348567698151), (18, 0.00023565061565022916), (19, 0.00033287194673903286), (20, 0.000494047999382019), (21, 0.0004682693397626281), (22, 0.00044089541188441217), (23, 0.0006388447363860905), (24, 0.001954255625605583), (25, 0.0033626516815274954), (26, 0.000516745843924582), (27, 3.630152787081897e-05)]
+
+ 55%|█████▍    | 82/150 [12:06<08:55,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 402, Spatial MSE: [(0, 6.519288149320346e-07), (1, 5.979263278277358e-06), (2, 1.311358209932223e-05), (3, 6.442795438488247e-06), (4, 1.5519943872277508e-06), (5, 1.382547452521976e-06), (6, 2.3363236323348247e-06), (7, 3.4327215416851686e-06), (8, 1.683577238509315e-06), (9, 2.4446624138363404e-06), (10, 4.700958925241139e-06), (11, 1.9404777049203403e-05), (12, 5.6440880143782124e-05), (13, 0.001502315979450941), (14, 0.0012046084739267826), (15, 0.0003997801977675408), (16, 0.015259884297847748), (17, 3.9596048736711964e-05), (18, 5.194364712224342e-05), (19, 0.0009066982311196625), (20, 0.000322973559377715), (21, 0.00015294394688680768), (22, 0.00025671086041256785), (23, 0.00011313775030430406), (24, 0.00016231427434831858), (25, 0.00045291241258382797), (26, 0.0004925538087263703), (27, 0.01154081430286169)], Temporal MSE: [(0, 6.402298140528728e-07), (1, 2.945546100363572e-07), (2, 3.4834221196433646e-07), (3, 6.463025670200295e-07), (4, 4.512744737894536e-07), (5, 7.101778578544327e-07), (6, 8.19185345335427e-07), (7, 9.504902891421807e-07), (8, 4.658646446387138e-07), (9, 5.961949227639707e-07), (10, 1.9636793240351835e-06), (11, 4.738119605462998e-05), (12, 3.165463203913532e-05), (13, 3.2466938137076795e-05), (14, 1.4535632544721011e-05), (15, 3.0562230676878244e-05), (16, 1.7515932995593175e-05), (17, 1.649368459766265e-05), (18, 4.2240062612108886e-05), (19, 3.655583714134991e-05), (20, 2.69631727860542e-05), (21, 2.947678331111092e-05), (22, 4.376480501377955e-05), (23, 0.0001165363792097196), (24, 0.00013842785847373307), (25, 0.00020537858654279262), (26, 5.05579482705798e-05), (27, 3.4929034882225096e-05)]
+
+ 55%|█████▌    | 83/150 [12:16<09:29,  8.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 396, Spatial MSE: [(0, 8.710669021638751e-07), (1, 7.615888989676023e-06), (2, 1.652723403822165e-05), (3, 7.289066161320079e-06), (4, 4.056702437083004e-06), (5, 5.594224603555631e-06), (6, 1.7475056665716693e-05), (7, 9.856020369625185e-06), (8, 1.6561541997361928e-05), (9, 3.5720870073419064e-05), (10, 4.856118903262541e-05), (11, 8.24963572085835e-05), (12, 0.00012947323557455093), (13, 0.001962312962859869), (14, 0.0021630264818668365), (15, 0.0011424790136516094), (16, 0.03215382993221283), (17, 9.440290159545839e-05), (18, 0.00013251033669803292), (19, 0.0017793950391933322), (20, 0.0011894135968759656), (21, 0.00023060277453623712), (22, 0.00034369222703389823), (23, 0.0002293179859407246), (24, 0.0004885740345343947), (25, 0.0012434839736670256), (26, 0.002637665020301938), (27, 0.01145344041287899)], Temporal MSE: [(0, 6.111418429100013e-07), (1, 3.179364398420148e-07), (2, 6.183044547469763e-07), (3, 1.769995265021862e-06), (4, 2.100685833283933e-06), (5, 2.509197429390042e-06), (6, 4.704241746367188e-06), (7, 4.91193713969551e-06), (8, 3.922364157915581e-06), (9, 4.774723493028432e-06), (10, 9.019850949698593e-06), (11, 6.265714182518423e-05), (12, 5.103068906464614e-05), (13, 5.442246038001031e-05), (14, 4.267903932486661e-05), (15, 6.021578155923635e-05), (16, 3.2294577977154404e-05), (17, 3.0655010050395504e-05), (18, 5.8982783230021596e-05), (19, 5.225474524195306e-05), (20, 5.205679190112278e-05), (21, 7.405894575640559e-05), (22, 9.493279503658414e-05), (23, 0.00017232242680620402), (24, 0.00027303313254378736), (25, 0.0005672293482348323), (26, 0.00012946351489517838), (27, 4.1863793740049005e-05)]
+
+ 56%|█████▌    | 84/150 [12:25<09:23,  8.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 390, Spatial MSE: [(0, 6.557080496349954e-07), (1, 6.730226232321002e-06), (2, 1.421090928488411e-05), (3, 6.5022882154153194e-06), (4, 1.6945298284554156e-06), (5, 1.3254102668724954e-06), (6, 2.3520219656347763e-06), (7, 4.083273779542651e-06), (8, 1.9021362049898016e-06), (9, 2.2434026050177636e-06), (10, 6.360902261803858e-06), (11, 3.66552994819358e-05), (12, 0.00010427060624351725), (13, 0.002684358274564147), (14, 0.002235440304502845), (15, 0.0006633266457356513), (16, 0.020848087966442108), (17, 9.8623480880633e-05), (18, 9.059084550244734e-05), (19, 0.0013262017164379358), (20, 0.0006680170772597194), (21, 0.00011759073822759092), (22, 0.00026374805020168424), (23, 0.00012198152398923412), (24, 0.000217373832128942), (25, 0.000541514833457768), (26, 0.0005727339303120971), (27, 0.01934926211833954)], Temporal MSE: [(0, 6.442550670726632e-07), (1, 3.114877245025127e-07), (2, 3.8682702552250703e-07), (3, 6.553668754349928e-07), (4, 4.967838549418957e-07), (5, 7.582091825497628e-07), (6, 9.739959523358266e-07), (7, 1.0940348147414625e-06), (8, 5.038820631853014e-07), (9, 7.043869345579878e-07), (10, 3.5764160202234052e-06), (11, 9.683003736427054e-05), (12, 5.912076449021697e-05), (13, 5.21967449458316e-05), (14, 2.483285243215505e-05), (15, 5.18607885169331e-05), (16, 1.9944342056987807e-05), (17, 1.922299816214945e-05), (18, 4.662561696022749e-05), (19, 3.489774098852649e-05), (20, 2.0118930478929542e-05), (21, 3.7143981899134815e-05), (22, 5.630622763419524e-05), (23, 0.00013249812764115632), (24, 0.0001582498080097139), (25, 0.00027756011695601046), (26, 6.697011849610135e-05), (27, 6.296737410593778e-05)]
+
+ 57%|█████▋    | 85/150 [12:32<08:54,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 384, Spatial MSE: [(0, 8.300981448883249e-07), (1, 1.2191098903713282e-05), (2, 2.3533855710411444e-05), (3, 1.2121317013225053e-05), (4, 2.0677485736086965e-05), (5, 2.7948530259891413e-05), (6, 8.18429107312113e-05), (7, 0.00013585844135377556), (8, 0.00013027105887886137), (9, 0.0002112332877004519), (10, 0.0001750869705574587), (11, 0.00022566238476429135), (12, 0.0003287509025540203), (13, 0.0039573838002979755), (14, 0.005181388929486275), (15, 0.0028358178678900003), (16, 0.0752226710319519), (17, 0.0005669461679644883), (18, 0.0006091538816690445), (19, 0.004797680769115686), (20, 0.003885043552145362), (21, 0.0008395149488933384), (22, 0.0009188614203594625), (23, 0.0013198043452575803), (24, 0.002485285745933652), (25, 0.00480671739205718), (26, 0.00603362126275897), (27, 0.018951423466205597)], Temporal MSE: [(0, 7.256265917021665e-07), (1, 3.532032337716373e-07), (2, 1.4957701068851748e-06), (3, 9.571091140969656e-06), (4, 7.861487574700732e-06), (5, 1.683032860455569e-05), (6, 6.182827200973406e-05), (7, 3.311871114419773e-05), (8, 2.208908881584648e-05), (9, 3.353326610522345e-05), (10, 5.891627733944915e-05), (11, 0.00017729437968228012), (12, 0.00014534541696775705), (13, 0.00018756618374027312), (14, 0.00021585929789580405), (15, 0.00023095388314686716), (16, 0.00022856873692944646), (17, 0.00020963426504749805), (18, 0.0002484017750248313), (19, 0.0003303647390566766), (20, 0.00047439837362617254), (21, 0.0004763905017171055), (22, 0.0004989071749150753), (23, 0.0007468548137694597), (24, 0.001883611548691988), (25, 0.0031859104055911303), (26, 0.0005877178627997637), (27, 9.353907807962969e-05)]
+
+ 57%|█████▋    | 86/150 [12:42<09:19,  8.74s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 378, Spatial MSE: [(0, 8.056316005422559e-07), (1, 6.813038453401532e-06), (2, 1.5416373571497388e-05), (3, 6.249955731618684e-06), (4, 1.4219774584489642e-06), (5, 1.2470042065615417e-06), (6, 2.508098532416625e-06), (7, 3.522179440551554e-06), (8, 1.8551386347098742e-06), (9, 2.6597151645546546e-06), (10, 1.143632471212186e-05), (11, 6.0233858675928786e-05), (12, 0.00014604561147280037), (13, 0.0028571938164532185), (14, 0.002515374217182398), (15, 0.0007191617623902857), (16, 0.027798499912023544), (17, 0.00010286233737133443), (18, 0.00012048360804328695), (19, 0.0018363704439252615), (20, 0.0009968547383323312), (21, 0.00017511221813037992), (22, 0.000308483635308221), (23, 0.00014856382040306926), (24, 0.0002501057751942426), (25, 0.0006906336639076471), (26, 0.0005938690155744553), (27, 0.021563854068517685)], Temporal MSE: [(0, 6.164642059047765e-07), (1, 2.604514861559437e-07), (2, 3.5622377936306293e-07), (3, 6.151646516627807e-07), (4, 4.775470756612776e-07), (5, 7.661686822757474e-07), (6, 8.873016099641973e-07), (7, 9.294216738453542e-07), (8, 5.061780825599271e-07), (9, 7.463926294803969e-07), (10, 5.488143870024942e-06), (11, 0.00013821669563185424), (12, 6.634006422245875e-05), (13, 6.303127884166315e-05), (14, 2.642419894982595e-05), (15, 6.152461719466373e-05), (16, 3.170519266859628e-05), (17, 2.660160680534318e-05), (18, 5.7427638239460066e-05), (19, 4.49525578005705e-05), (20, 2.9876779080950655e-05), (21, 6.033034514985047e-05), (22, 7.884828664828092e-05), (23, 0.00017940954421646893), (24, 0.00018962744798045605), (25, 0.0003181299543939531), (26, 7.347396604018286e-05), (27, 7.605470455018803e-05)]
+
+ 58%|█████▊    | 87/150 [12:51<09:00,  8.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 372, Spatial MSE: [(0, 8.811172165223979e-07), (1, 8.956199053500313e-06), (2, 1.3229966498329304e-05), (3, 6.991235295572551e-06), (4, 4.5481410779757425e-06), (5, 6.045102963980753e-06), (6, 1.7472071704105474e-05), (7, 1.0314117389498278e-05), (8, 1.6871714251465164e-05), (9, 3.804176230914891e-05), (10, 6.026531627867371e-05), (11, 0.00012578535825014114), (12, 0.0001973368343897164), (13, 0.0029890367295593023), (14, 0.0028241847176104784), (15, 0.0010117790661752224), (16, 0.03827347978949547), (17, 0.0001471080322517082), (18, 0.00034212597529403865), (19, 0.003312349319458008), (20, 0.002808402292430401), (21, 0.0003062471223529428), (22, 0.00039866537554189563), (23, 0.00025081937201321125), (24, 0.000564799876883626), (25, 0.0014642507303506136), (26, 0.0034066264051944017), (27, 0.0214729942381382)], Temporal MSE: [(0, 6.664174634352094e-07), (1, 3.4252988712069055e-07), (2, 6.639250500484195e-07), (3, 1.8336941138841212e-06), (4, 2.0077579847566085e-06), (5, 2.6318068648834014e-06), (6, 4.815758529730374e-06), (7, 5.690836132998811e-06), (8, 4.3372974687372334e-06), (9, 5.748353942180984e-06), (10, 1.465249624743592e-05), (11, 0.00014575548993889242), (12, 7.72235362092033e-05), (13, 8.048954623518512e-05), (14, 5.0249997002538294e-05), (15, 8.015261846594512e-05), (16, 4.5168697397457436e-05), (17, 3.942616967833601e-05), (18, 8.033210906432942e-05), (19, 8.170022920239717e-05), (20, 8.805473771644756e-05), (21, 0.00010270900384057313), (22, 0.00011877962970174849), (23, 0.00021730153821408749), (24, 0.0003095922584179789), (25, 0.000637401535641402), (26, 0.00015100058226380497), (27, 8.432975300820544e-05)]
+
+ 59%|█████▊    | 88/150 [12:58<08:37,  8.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 366, Spatial MSE: [(0, 5.523367576643068e-07), (1, 6.539617970702238e-06), (2, 1.3715412933379412e-05), (3, 6.270958692766726e-06), (4, 1.3427987823888543e-06), (5, 1.156073039965122e-06), (6, 2.3238808353198692e-06), (7, 3.451891188888112e-06), (8, 1.9649971818580525e-06), (9, 2.449668954795925e-06), (10, 1.0392906006018165e-05), (11, 8.492128836223856e-05), (12, 0.00012643629452213645), (13, 0.0029206331819295883), (14, 0.002400551689788699), (15, 0.0008102910360321403), (16, 0.022135883569717407), (17, 6.195170863065869e-05), (18, 0.00011537098907865584), (19, 0.0014234796399250627), (20, 0.0005553068476729095), (21, 0.00017374995513819158), (22, 0.00032107034348882735), (23, 0.0001526877749711275), (24, 0.00023362952924799174), (25, 0.0005814545438624918), (26, 0.0005632815300486982), (27, 0.02094895951449871)], Temporal MSE: [(0, 4.598885539053299e-07), (1, 2.3554352424071112e-07), (2, 3.1775530828781484e-07), (3, 5.463252250592632e-07), (4, 4.5544183535639604e-07), (5, 7.238395483000204e-07), (6, 9.351171001981129e-07), (7, 8.93058654583001e-07), (8, 4.6131415842864953e-07), (9, 7.093931913004781e-07), (10, 4.778511538461316e-06), (11, 0.00014356472820509225), (12, 5.4660660680383444e-05), (13, 6.811420462327078e-05), (14, 2.7772806788561866e-05), (15, 5.0205628213007e-05), (16, 2.1857666069990955e-05), (17, 2.035935176536441e-05), (18, 5.898028030060232e-05), (19, 4.5490593038266525e-05), (20, 3.116505467914976e-05), (21, 3.526122964103706e-05), (22, 5.5002092267386615e-05), (23, 0.00015276283374987543), (24, 0.00018766973516903818), (25, 0.00029834129964001477), (26, 7.675206870771945e-05), (27, 6.397395918611437e-05)]
+
+ 59%|█████▉    | 89/150 [13:06<08:24,  8.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 360, Spatial MSE: [(0, 8.46881334837235e-07), (1, 1.2070024240529165e-05), (2, 2.0260418750694953e-05), (3, 1.1842904314107727e-05), (4, 2.014494748436846e-05), (5, 2.7507738195708953e-05), (6, 6.138355820439756e-05), (7, 0.0001235587988048792), (8, 0.00012260436778888106), (9, 0.00016686525486875325), (10, 0.00014214171096682549), (11, 0.00024129579833243042), (12, 0.00030534964753314853), (13, 0.0039354716427624226), (14, 0.004852422513067722), (15, 0.0023618529085069895), (16, 0.059099528938531876), (17, 0.00039338661008514464), (18, 0.0005134473321959376), (19, 0.0036320581566542387), (20, 0.0022846898064017296), (21, 0.0009220074280165136), (22, 0.0009549762471579015), (23, 0.0012769943568855524), (24, 0.0023181529249995947), (25, 0.00422641821205616), (26, 0.0055364822037518024), (27, 0.02078446000814438)], Temporal MSE: [(0, 7.408335136460664e-07), (1, 3.416191418637027e-07), (2, 1.6524844568266417e-06), (3, 1.070265716407448e-05), (4, 7.833105883037206e-06), (5, 1.5434388842550106e-05), (6, 5.829890142194927e-05), (7, 3.185004970873706e-05), (8, 2.0709905584226362e-05), (9, 3.2033000024966896e-05), (10, 5.483932909555733e-05), (11, 0.0001983704132726416), (12, 0.00012012341903755441), (13, 0.0001821794721763581), (14, 0.00019736707326956093), (15, 0.00019113953749183565), (16, 0.0001729947980493307), (17, 0.00016314195818267763), (18, 0.00021552873658947647), (19, 0.0002846660208888352), (20, 0.0004325328045524657), (21, 0.0004445157537702471), (22, 0.00047024613013491035), (23, 0.0007202866254374385), (24, 0.001955540617927909), (25, 0.00322628952562809), (26, 0.0005860825185663998), (27, 9.93626017589122e-05)]
+
+ 60%|██████    | 90/150 [13:15<08:29,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 354, Spatial MSE: [(0, 1.1344433232807205e-06), (1, 6.379064870998263e-06), (2, 1.3122911695973016e-05), (3, 6.5983472268271726e-06), (4, 1.6104347650980344e-06), (5, 1.7094901068048785e-06), (6, 2.939301111837267e-06), (7, 5.878488991584163e-06), (8, 2.415097014818457e-06), (9, 2.163835461033159e-06), (10, 5.130281351739541e-06), (11, 2.2390158846974373e-05), (12, 8.238494774559513e-05), (13, 0.0027922620065510273), (14, 0.002438762690871954), (15, 0.000717973685823381), (16, 0.0276336632668972), (17, 0.0001262568694073707), (18, 0.00010252080392092466), (19, 0.0020027654245495796), (20, 0.0012465778272598982), (21, 9.508885705145076e-05), (22, 0.00025255780201405287), (23, 0.00013104002573527396), (24, 0.0002420421369606629), (25, 0.0006316424114629626), (26, 0.0006190960993990302), (27, 0.02182425558567047)], Temporal MSE: [(0, 8.121970722640981e-07), (1, 3.270712056746561e-07), (2, 4.1231797354157607e-07), (3, 7.646777362424473e-07), (4, 5.887232532586495e-07), (5, 8.844316425893339e-07), (6, 1.200029601022834e-06), (7, 1.3297457144290092e-06), (8, 5.991256557535962e-07), (9, 7.998126534403127e-07), (10, 2.345271468584542e-06), (11, 5.3583447879645973e-05), (12, 5.215777855482884e-05), (13, 5.9826568758580834e-05), (14, 2.4729502911213785e-05), (15, 6.579102773685008e-05), (16, 3.5555538488551974e-05), (17, 3.512354669510387e-05), (18, 5.423908805823885e-05), (19, 4.3356616515666246e-05), (20, 2.6813653676072136e-05), (21, 5.909037645324133e-05), (22, 7.992061000550166e-05), (23, 0.0001644632575334981), (24, 0.00015495013212785125), (25, 0.00033535296097397804), (26, 7.332846871577203e-05), (27, 8.087977766990662e-05)]
+
+ 61%|██████    | 91/150 [13:25<08:35,  8.73s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 348, Spatial MSE: [(0, 7.648622499800695e-07), (1, 8.599206012149807e-06), (2, 1.0362946341047063e-05), (3, 5.969842732156394e-06), (4, 3.347274514453602e-06), (5, 5.532317118195351e-06), (6, 1.6101479559438303e-05), (7, 8.893131962395273e-06), (8, 1.6607602447038516e-05), (9, 3.428131094551645e-05), (10, 4.2706513340817764e-05), (11, 7.626307342434302e-05), (12, 0.00012800187687389553), (13, 0.0029359995387494564), (14, 0.0027860214468091726), (15, 0.0009287958964705467), (16, 0.029027175158262253), (17, 0.00016374644474126399), (18, 0.00012808392057195306), (19, 0.001954116625711322), (20, 0.001303854864090681), (21, 0.00012785715807694942), (22, 0.0003008144849445671), (23, 0.00019411309040151536), (24, 0.00048107083421200514), (25, 0.0012930624652653933), (26, 0.002705859486013651), (27, 0.020682603120803833)], Temporal MSE: [(0, 5.994799607833556e-07), (1, 2.685672484403767e-07), (2, 5.803199201181997e-07), (3, 1.7305111441601184e-06), (4, 1.8994454649146064e-06), (5, 2.248516921099508e-06), (6, 4.663105301005999e-06), (7, 5.097729626868386e-06), (8, 3.997112798970193e-06), (9, 4.533562787401024e-06), (10, 8.436937605438288e-06), (11, 6.255607877392322e-05), (12, 6.233533349586651e-05), (13, 7.41540570743382e-05), (14, 4.3210999137954786e-05), (15, 7.790723611833528e-05), (16, 4.283519592718221e-05), (17, 3.792070856434293e-05), (18, 6.317543011391535e-05), (19, 5.5937740398803726e-05), (20, 5.1961542339995503e-05), (21, 7.603348785778508e-05), (22, 9.470804070588201e-05), (23, 0.00018491913215257227), (24, 0.00024713200400583446), (25, 0.0005900372052565217), (26, 0.00014409757568500936), (27, 8.578447886975482e-05)]
+
+ 61%|██████▏   | 92/150 [13:32<08:04,  8.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 342, Spatial MSE: [(0, 6.07725439749629e-07), (1, 5.440194854600122e-06), (2, 1.0463532817084342e-05), (3, 6.002763257129118e-06), (4, 1.417468979525438e-06), (5, 1.4355987332237419e-06), (6, 1.7977181414607912e-06), (7, 3.92502079193946e-06), (8, 1.9438009530858835e-06), (9, 2.2282724785327446e-06), (10, 7.2879552135418635e-06), (11, 3.6422086850507185e-05), (12, 7.976077176863328e-05), (13, 0.002293824218213558), (14, 0.0020078395027667284), (15, 0.0006837526452727616), (16, 0.01837810128927231), (17, 4.735612674267031e-05), (18, 7.1982984081842e-05), (19, 0.0010598691878840327), (20, 0.00046087606460787356), (21, 0.00010089447459904477), (22, 0.00024085657787509263), (23, 0.00012352922931313515), (24, 0.00018761611136142164), (25, 0.00042856173240579665), (26, 0.00047943249228410423), (27, 0.0186091810464859)], Temporal MSE: [(0, 5.953638719802257e-07), (1, 3.1912207987261354e-07), (2, 3.634513632277958e-07), (3, 7.039751608317601e-07), (4, 5.59164789137867e-07), (5, 8.413458090217318e-07), (6, 1.2407012945914175e-06), (7, 1.1261620329605648e-06), (8, 5.609738309431123e-07), (9, 7.675301390008826e-07), (10, 4.162453933531651e-06), (11, 9.100293391384184e-05), (12, 4.163190533290617e-05), (13, 5.790148861706257e-05), (14, 2.286370909132529e-05), (15, 4.139547309023328e-05), (16, 1.787858309398871e-05), (17, 1.6622207112959586e-05), (18, 4.654498479794711e-05), (19, 3.602455399231985e-05), (20, 2.5110106435022317e-05), (21, 2.9298847948666662e-05), (22, 4.3829833884956315e-05), (23, 0.00012402671563904732), (24, 0.00014424428809434175), (25, 0.0002897994709201157), (26, 7.132402970455587e-05), (27, 6.2729770434089e-05)]
+
+ 62%|██████▏   | 93/150 [13:40<07:49,  8.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 336, Spatial MSE: [(0, 8.106229074655857e-07), (1, 1.0408499292680062e-05), (2, 1.5372255802503787e-05), (3, 1.1154165804327931e-05), (4, 1.9469413018669002e-05), (5, 2.6397276087664068e-05), (6, 7.287176413228735e-05), (7, 0.0001323711039731279), (8, 0.00012257654452696443), (9, 0.00018749704759102315), (10, 0.00017654657131060958), (11, 0.0002130913781002164), (12, 0.00027342085377313197), (13, 0.0025928826071321964), (14, 0.002491811290383339), (15, 0.001119326101616025), (16, 0.018878789618611336), (17, 0.00037259087548591197), (18, 0.0006066974019631743), (19, 0.0019956736359745264), (20, 0.002119973534718156), (21, 0.0009219135972671211), (22, 0.000866709859110415), (23, 0.0013816612772643566), (24, 0.0023776597809046507), (25, 0.0046129473485052586), (26, 0.0072816358879208565), (27, 0.01883760094642639)], Temporal MSE: [(0, 8.015176717890427e-07), (1, 3.885518822244194e-07), (2, 1.5840597598071327e-06), (3, 5.676476575899869e-06), (4, 7.975035259732977e-06), (5, 1.67718026204966e-05), (6, 7.025241939118132e-05), (7, 3.694082624861039e-05), (8, 2.1946096239844337e-05), (9, 3.4207780117867514e-05), (10, 5.617529677692801e-05), (11, 0.00014680165622849017), (12, 0.00010217542148893699), (13, 0.00016632028564345092), (14, 0.00018986998475156724), (15, 0.00018391263438388705), (16, 0.00019076337048318237), (17, 0.00017285067588090897), (18, 0.00023001722001936287), (19, 0.00035017222398892045), (20, 0.000589128234423697), (21, 0.0005288152606226504), (22, 0.00046254167682491243), (23, 0.0007490319549106061), (24, 0.0021583610214293003), (25, 0.003204859560355544), (26, 0.0006872661178931594), (27, 0.0001025101519189775)]
+
+ 63%|██████▎   | 94/150 [13:49<07:43,  8.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 330, Spatial MSE: [(0, 7.216983135549526e-07), (1, 6.618175120820524e-06), (2, 1.2965108908247203e-05), (3, 6.359903636621311e-06), (4, 1.6062615486589493e-06), (5, 1.4380815400727442e-06), (6, 2.348005637031747e-06), (7, 3.715605089382734e-06), (8, 1.912688958327635e-06), (9, 2.0814106846955838e-06), (10, 3.801734010266955e-06), (11, 1.3527309420169331e-05), (12, 4.9421960284234956e-05), (13, 0.001893359120003879), (14, 0.0017650448717176914), (15, 0.0006467782659456134), (16, 0.018089957535266876), (17, 4.865079608862288e-05), (18, 4.580731183523312e-05), (19, 0.001100095803849399), (20, 0.0004641529521904886), (21, 0.0001184110515168868), (22, 0.0002482314012013376), (23, 0.0001290667860303074), (24, 0.00017963456048164517), (25, 0.00042300845962017775), (26, 0.0005123049486428499), (27, 0.017994489520788193)], Temporal MSE: [(0, 6.077664806980465e-07), (1, 2.9442659865708265e-07), (2, 3.984895613484696e-07), (3, 7.121944918253575e-07), (4, 6.087588531045185e-07), (5, 8.875384196471714e-07), (6, 1.2022958344459767e-06), (7, 1.1367690149199916e-06), (8, 6.006509920553071e-07), (9, 7.38114692921954e-07), (10, 1.9256026462244336e-06), (11, 2.670737012522295e-05), (12, 3.2333628041669726e-05), (13, 5.1761922804871574e-05), (14, 2.1245314201223664e-05), (15, 3.775886580115184e-05), (16, 1.768107176758349e-05), (17, 1.654717925703153e-05), (18, 5.0078982894774526e-05), (19, 4.1043640521820635e-05), (20, 2.345168286410626e-05), (21, 2.8330847271718085e-05), (22, 4.263945811544545e-05), (23, 0.00011957991955569014), (24, 0.00013704139564651996), (25, 0.00033485121093690395), (26, 7.101993833202869e-05), (27, 6.189922714838758e-05)]
+
+ 63%|██████▎   | 95/150 [13:59<08:08,  8.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 324, Spatial MSE: [(0, 7.787976983308909e-07), (1, 7.056548383843619e-06), (2, 9.2746049631387e-06), (3, 5.809014510305133e-06), (4, 3.559588776624878e-06), (5, 5.8567147789290175e-06), (6, 1.6168616639333777e-05), (7, 9.859495548880659e-06), (8, 1.709288699203171e-05), (9, 3.303411358501762e-05), (10, 4.0627091948408633e-05), (11, 7.080523209879175e-05), (12, 9.857099939836189e-05), (13, 0.001961639616638422), (14, 0.0018417942337691784), (15, 0.0006862665177322924), (16, 0.018103618174791336), (17, 6.583967478945851e-05), (18, 7.670895865885541e-05), (19, 0.0011163478484377265), (20, 0.0004879482148680836), (21, 0.00013831273827236146), (22, 0.0002874788478948176), (23, 0.00019865146896336228), (24, 0.0004132495087105781), (25, 0.0011517633683979511), (26, 0.00275783590041101), (27, 0.01743229478597641)], Temporal MSE: [(0, 6.211978416104103e-07), (1, 3.20826075039804e-07), (2, 6.186585324030602e-07), (3, 1.8594033690533252e-06), (4, 2.036522573689581e-06), (5, 2.5207141334249172e-06), (6, 5.389006673794938e-06), (7, 5.598504685622174e-06), (8, 4.184165391052375e-06), (9, 5.095093001727946e-06), (10, 8.235746463469695e-06), (11, 3.462829772615805e-05), (12, 4.188179082120769e-05), (13, 6.502171891042963e-05), (14, 3.990151526522823e-05), (15, 4.827826705877669e-05), (16, 2.5513212676742114e-05), (17, 2.2430274839280173e-05), (18, 5.9192549088038504e-05), (19, 5.295369919622317e-05), (20, 4.2915638914564624e-05), (21, 4.462725337361917e-05), (22, 6.261898670345545e-05), (23, 0.00014855786866974086), (24, 0.0002296185411978513), (25, 0.0005768901901319623), (26, 0.00015128824452403933), (27, 6.918451254023239e-05)]
+
+ 64%|██████▍   | 96/150 [14:07<07:45,  8.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 318, Spatial MSE: [(0, 5.300726684254187e-07), (1, 5.2804894039581995e-06), (2, 8.994459676614497e-06), (3, 5.847501142852707e-06), (4, 1.50856169511826e-06), (5, 1.5275459190888796e-06), (6, 1.74503338712384e-06), (7, 3.4414956644468475e-06), (8, 1.7543263766128803e-06), (9, 1.828631866374053e-06), (10, 2.6857699140236946e-06), (11, 7.232872576423688e-06), (12, 2.368523200857453e-05), (13, 0.0007871100096963346), (14, 0.0006383751169778407), (15, 0.00020546882296912372), (16, 0.007838237099349499), (17, 2.3809743652236648e-05), (18, 3.42215716955252e-05), (19, 0.0005039623356424272), (20, 0.00019557269115466624), (21, 0.00010797945287777111), (22, 0.00014494404604192823), (23, 7.238190300995484e-05), (24, 0.00010373099939897656), (25, 0.0002205212804256007), (26, 0.00030141748720780015), (27, 0.006256545428186655)], Temporal MSE: [(0, 5.155039843884879e-07), (1, 2.876672056117968e-07), (2, 3.4118488656531554e-07), (3, 6.764651061530458e-07), (4, 5.774405735792243e-07), (5, 8.349188078682346e-07), (6, 1.1534430086612701e-06), (7, 1.0653720892150886e-06), (8, 5.637606363961822e-07), (9, 6.697204639749543e-07), (10, 1.478521994613402e-06), (11, 1.3694750123249833e-05), (12, 1.5280054867616855e-05), (13, 1.9663539205794223e-05), (14, 7.773261131660547e-06), (15, 1.5056303709570784e-05), (16, 1.0564993317530025e-05), (17, 1.043193151417654e-05), (18, 2.880866486520972e-05), (19, 2.4539051082683727e-05), (20, 1.724942194414325e-05), (21, 1.7958742319024168e-05), (22, 2.7115840566693805e-05), (23, 6.746525468770415e-05), (24, 6.780240073567256e-05), (25, 0.00015343405539169908), (26, 3.658274363260716e-05), (27, 1.9722298020496964e-05)]
+
+ 65%|██████▍   | 97/150 [14:14<07:14,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 312, Spatial MSE: [(0, 7.50325682474795e-07), (1, 9.223950655723456e-06), (2, 1.3206333278503735e-05), (3, 1.0186841791437473e-05), (4, 1.6460751794511452e-05), (5, 2.5724020815687254e-05), (6, 5.736279854318127e-05), (7, 0.00011663822806440294), (8, 0.00011255325807724148), (9, 0.00015631242422387004), (10, 0.00012520905875135213), (11, 0.00014848944556433707), (12, 0.00018713250756263733), (13, 0.0016270899213850498), (14, 0.0025664360728114843), (15, 0.0014323245268315077), (16, 0.0327288918197155), (17, 0.0003678583016153425), (18, 0.0004707665939349681), (19, 0.002428474137559533), (20, 0.002058125101029873), (21, 0.0008925902657210827), (22, 0.0008308975957334042), (23, 0.0014427744317799807), (24, 0.002526105148717761), (25, 0.0044438643380999565), (26, 0.006496344693005085), (27, 0.006536857690662146)], Temporal MSE: [(0, 6.83521420796751e-07), (1, 3.599687659061601e-07), (2, 1.6446692825411446e-06), (3, 6.0689808378810994e-06), (4, 7.606994131492684e-06), (5, 1.3642688827530947e-05), (6, 5.260403122520074e-05), (7, 3.162517168675549e-05), (8, 2.0008206774946302e-05), (9, 3.05199901049491e-05), (10, 4.7548488510074094e-05), (11, 6.541332550114021e-05), (12, 7.821430335752666e-05), (13, 0.00012746980064548552), (14, 0.0001838115422287956), (15, 0.00016608410805929452), (16, 0.00018541370809543878), (17, 0.00017533544450998306), (18, 0.00023016730847302824), (19, 0.00035959878005087376), (20, 0.000644112064037472), (21, 0.0005657743313349783), (22, 0.0004928290727548301), (23, 0.0007975568878464401), (24, 0.0024342555552721024), (25, 0.003305651480332017), (26, 0.0006707743741571903), (27, 6.517281872220337e-05)]
+
+ 65%|██████▌   | 98/150 [14:23<07:14,  8.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 306, Spatial MSE: [(0, 1.078431978385197e-06), (1, 6.765066245861817e-06), (2, 1.1277288649580441e-05), (3, 6.208545528352261e-06), (4, 1.5303430700441822e-06), (5, 1.8024703649643925e-06), (6, 2.757574293354992e-06), (7, 5.386894372350071e-06), (8, 2.2784602151659783e-06), (9, 2.0997488263674313e-06), (10, 3.1632882837584475e-06), (11, 1.1239732884860132e-05), (12, 3.825171006610617e-05), (13, 0.0013286274624988437), (14, 0.0010936594335362315), (15, 0.0003225746040698141), (16, 0.00922657921910286), (17, 2.945162123069167e-05), (18, 4.046777030453086e-05), (19, 0.0006511686951853335), (20, 0.0003049098595511168), (21, 7.743012974970043e-05), (22, 0.00013652768393512815), (23, 8.089157199719921e-05), (24, 0.00012990286631975323), (25, 0.0003154069709125906), (26, 0.00042231669067405164), (27, 0.011777441948652267)], Temporal MSE: [(0, 7.992778137122514e-07), (1, 3.33557153453512e-07), (2, 4.173863885625906e-07), (3, 8.214392437366769e-07), (4, 6.614993708353722e-07), (5, 9.647573051552172e-07), (6, 1.3355261216929648e-06), (7, 1.422287937202782e-06), (8, 6.685710332021699e-07), (9, 8.479908615299792e-07), (10, 1.8538917174737435e-06), (11, 2.4752003810135648e-05), (12, 2.5692657800391316e-05), (13, 3.470813317107968e-05), (14, 1.3567426321969833e-05), (15, 1.7958920579985715e-05), (16, 1.0181604011449963e-05), (17, 9.04752505448414e-06), (18, 2.7629148462438025e-05), (19, 2.2730309865437448e-05), (20, 1.3672025488631334e-05), (21, 1.7415701222489588e-05), (22, 2.871333526854869e-05), (23, 7.357171853072941e-05), (24, 8.964438893599436e-05), (25, 0.00028160231886431575), (26, 5.588910062215291e-05), (27, 3.899886360159144e-05)]
+
+ 66%|██████▌   | 99/150 [14:30<06:55,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 300, Spatial MSE: [(0, 9.919928061208338e-07), (1, 7.137787179090083e-06), (2, 7.851638656575233e-06), (3, 5.464335572469281e-06), (4, 4.3452118916320615e-06), (5, 6.850037152617006e-06), (6, 1.7441401723772287e-05), (7, 1.1483354683150537e-05), (8, 1.8182310668635182e-05), (9, 4.7808473027544096e-05), (10, 5.9536469052545726e-05), (11, 8.620946755399927e-05), (12, 9.759241947904229e-05), (13, 0.0014176653930917382), (14, 0.0011839031940326095), (15, 0.00038245879113674164), (16, 0.009269745089113712), (17, 5.2073482947889715e-05), (18, 0.0001466109970351681), (19, 0.0008195473928935826), (20, 0.0007577640935778618), (21, 0.00015470627113245428), (22, 0.00020448092254810035), (23, 0.00017572414071764797), (24, 0.0004136542265769094), (25, 0.0011761848581954837), (26, 0.0032562881242483854), (27, 0.011271888390183449)], Temporal MSE: [(0, 7.802178174642904e-07), (1, 3.9575394339408376e-07), (2, 7.000048753980082e-07), (3, 2.081741740767029e-06), (4, 2.2202143554750364e-06), (5, 2.9626912692037877e-06), (6, 6.105199190642452e-06), (7, 7.2601860665599816e-06), (8, 5.7791799008555245e-06), (9, 7.017322332103504e-06), (10, 1.177571903099306e-05), (11, 3.566980012692511e-05), (12, 3.8104484701761976e-05), (13, 5.222430991125293e-05), (14, 3.579429176170379e-05), (15, 3.1554969609715044e-05), (16, 2.0440824300749227e-05), (17, 1.727327071421314e-05), (18, 4.013571378891356e-05), (19, 4.395339055918157e-05), (20, 5.229784073890187e-05), (21, 4.1112576582236215e-05), (22, 5.463532215799205e-05), (23, 0.0001108561918954365), (24, 0.000204012991162017), (25, 0.0005546890897676349), (26, 0.00017019872029777616), (27, 4.875497688772157e-05)]
+
+ 67%|██████▋   | 100/150 [14:38<06:45,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 294, Spatial MSE: [(0, 5.974320629320573e-07), (1, 6.153592494229088e-06), (2, 9.958553164324258e-06), (3, 5.763017725257669e-06), (4, 1.5281833611879847e-06), (5, 1.5986880725904484e-06), (6, 1.7884029830383952e-06), (7, 3.9956635191629175e-06), (8, 1.92845050150936e-06), (9, 1.9648812212835765e-06), (10, 2.6887312287726672e-06), (11, 8.69460291141877e-06), (12, 3.048642429348547e-05), (13, 0.00116618606261909), (14, 0.0010201825061812997), (15, 0.00031932242563925683), (16, 0.008715290576219559), (17, 2.497804416634608e-05), (18, 2.4207001843024045e-05), (19, 0.00058994215214625), (20, 0.0002838453510776162), (21, 5.6262426369357854e-05), (22, 0.00011880683450726792), (23, 7.331428059842438e-05), (24, 0.00011049670138163492), (25, 0.00024438140098936856), (26, 0.0003366856253705919), (27, 0.010994705371558666)], Temporal MSE: [(0, 5.121048047840304e-07), (1, 2.764367650343047e-07), (2, 3.57826451136134e-07), (3, 7.090191047609551e-07), (4, 6.316709573184198e-07), (5, 9.062729304787354e-07), (6, 1.3840001429343829e-06), (7, 1.2460499192457064e-06), (8, 6.395731020347739e-07), (9, 7.845612230994448e-07), (10, 1.6047227973103873e-06), (11, 1.7690317690721713e-05), (12, 2.1388199456850998e-05), (13, 3.2728112273616716e-05), (14, 1.2760800927935634e-05), (15, 1.6487276297993958e-05), (16, 9.044249964063056e-06), (17, 7.311152330657933e-06), (18, 2.4053775632637553e-05), (19, 1.968859214684926e-05), (20, 1.0747554370027501e-05), (21, 1.47608297993429e-05), (22, 2.389912333455868e-05), (23, 6.546777876792476e-05), (24, 7.975875632837415e-05), (25, 0.00022923685901332647), (26, 4.917436672258191e-05), (27, 3.738829764188267e-05)]
+
+ 67%|██████▋   | 101/150 [14:46<06:35,  8.07s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 288, Spatial MSE: [(0, 7.569659032924392e-07), (1, 9.06798322830582e-06), (2, 1.2537264410639182e-05), (3, 9.649169442127459e-06), (4, 1.6716530808480456e-05), (5, 2.4726741685299203e-05), (6, 6.912717071827501e-05), (7, 0.00013410441169980913), (8, 0.00012116012658225372), (9, 0.00017062995175365359), (10, 0.00013445351214613765), (11, 0.00015716053894720972), (12, 0.00019852504192385823), (13, 0.0017127073369920254), (14, 0.002475420944392681), (15, 0.0014729324029758573), (16, 0.03210800513625145), (17, 0.0003666964767035097), (18, 0.00048286205856129527), (19, 0.0023640342988073826), (20, 0.0019233612110838294), (21, 0.000759222311899066), (22, 0.0007713494705967605), (23, 0.0013633599737659097), (24, 0.0022522611543536186), (25, 0.0042054783552885056), (26, 0.007251331117004156), (27, 0.010986502282321453)], Temporal MSE: [(0, 7.885392960815807e-07), (1, 3.959735863645619e-07), (2, 1.7590868992556352e-06), (3, 4.454371264728252e-06), (4, 8.811737643554807e-06), (5, 1.6264051737380214e-05), (6, 7.344053301494569e-05), (7, 3.974223363911733e-05), (8, 2.2348094717017375e-05), (9, 3.398688932065852e-05), (10, 5.199013321544044e-05), (11, 7.476918108295649e-05), (12, 8.311199053423479e-05), (13, 0.00014397619816008955), (14, 0.00020204823522362858), (15, 0.00018710742006078362), (16, 0.00019965849060099572), (17, 0.0001824904466047883), (18, 0.00022499611077364534), (19, 0.000350054120644927), (20, 0.0005730897537432611), (21, 0.0005026672151871026), (22, 0.0004537772329058498), (23, 0.0007103343959897757), (24, 0.002014043740928173), (25, 0.0028448239900171757), (26, 0.0006577668827958405), (27, 8.802900993032381e-05)]
+
+ 68%|██████▊   | 102/150 [14:55<06:38,  8.31s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 282, Spatial MSE: [(0, 7.045919119263999e-07), (1, 7.178207397373626e-06), (2, 1.1028450899175368e-05), (3, 6.137118361948524e-06), (4, 1.5849610690565896e-06), (5, 1.7506858966953587e-06), (6, 1.860447696344636e-06), (7, 3.943016963603441e-06), (8, 2.108203716488788e-06), (9, 2.1011185253883014e-06), (10, 3.126545607301523e-06), (11, 8.269350473710801e-06), (12, 2.883284651034046e-05), (13, 0.0010436458978801966), (14, 0.0010011306731030345), (15, 0.0003959721070714295), (16, 0.013088561594486237), (17, 3.389970515854657e-05), (18, 0.00012148589303251356), (19, 0.0006615548627451062), (20, 0.00029899022774770856), (21, 0.00012487558706197888), (22, 0.0001734251854941249), (23, 9.470986697124317e-05), (24, 0.00014899940288159996), (25, 0.0003074897394981235), (26, 0.00038253291859291494), (27, 0.011121170595288277)], Temporal MSE: [(0, 6.553328262270952e-07), (1, 3.254733655921882e-07), (2, 4.161592812579329e-07), (3, 7.963465122884372e-07), (4, 7.146078928599309e-07), (5, 1.0128903795703081e-06), (6, 1.3161787819626625e-06), (7, 1.3053730754108983e-06), (8, 7.214103447950038e-07), (9, 8.477661594952224e-07), (10, 1.7511512169221533e-06), (11, 1.5865010936977342e-05), (12, 1.8990598618984222e-05), (13, 3.209044734830968e-05), (14, 1.571308348502498e-05), (15, 2.549724194977898e-05), (16, 1.3365469385462347e-05), (17, 1.283534129470354e-05), (18, 3.375320738996379e-05), (19, 2.433533882140182e-05), (20, 1.662074282648973e-05), (21, 2.3328004317590967e-05), (22, 3.8347086956491694e-05), (23, 9.380675328429788e-05), (24, 0.00010299024870619178), (25, 0.00027546921046450734), (26, 5.7780103816185147e-05), (27, 4.043673834530637e-05)]
+
+ 69%|██████▊   | 103/150 [15:03<06:26,  8.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 276, Spatial MSE: [(0, 8.224175758186902e-07), (1, 5.718014108424541e-06), (2, 8.884406270226464e-06), (3, 5.707011041522492e-06), (4, 3.407076292205602e-06), (5, 6.795520675950684e-06), (6, 1.6027210222091526e-05), (7, 1.1514141988300253e-05), (8, 2.1052317606518045e-05), (9, 3.089087476837449e-05), (10, 4.5683318603551015e-05), (11, 6.517500150948763e-05), (12, 7.749995711492375e-05), (13, 0.0011782710207626224), (14, 0.0012516560964286327), (15, 0.0006343787536025047), (16, 0.0170830599963665), (17, 6.0362741351127625e-05), (18, 0.00011938222451135516), (19, 0.0011301496997475624), (20, 0.0008207160281017423), (21, 0.00014711753465235233), (22, 0.00022969467681832612), (23, 0.00020477312500588596), (24, 0.0004533040337264538), (25, 0.0011004380648955703), (26, 0.0028748256154358387), (27, 0.010975586250424385)], Temporal MSE: [(0, 6.619415557906905e-07), (1, 3.6856710039501195e-07), (2, 6.891209523018915e-07), (3, 2.0482000309129944e-06), (4, 2.309292767677107e-06), (5, 2.7998810310236877e-06), (6, 5.814523774461122e-06), (7, 6.477177976194071e-06), (8, 4.646544311981415e-06), (9, 5.645945293508703e-06), (10, 8.068608622124884e-06), (11, 2.4959823349490762e-05), (12, 3.021314478246495e-05), (13, 4.6926037612138316e-05), (14, 3.7533260183408856e-05), (15, 3.92091715184506e-05), (16, 2.3923901608213782e-05), (17, 2.4845039661158808e-05), (18, 4.457834438653663e-05), (19, 4.517640627454966e-05), (20, 5.607062848866917e-05), (21, 4.6752375055802986e-05), (22, 6.405983731383458e-05), (23, 0.00012632702419068664), (24, 0.00020578849944286048), (25, 0.0004908455302938819), (26, 0.000160073337610811), (27, 5.10172285430599e-05)]
+
+ 69%|██████▉   | 104/150 [15:12<06:17,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 270, Spatial MSE: [(0, 6.652739443779865e-07), (1, 6.977687917242292e-06), (2, 1.0290247701050248e-05), (3, 5.7361626204510685e-06), (4, 1.524894514659536e-06), (5, 1.7556445754962624e-06), (6, 1.843996415118454e-06), (7, 3.9235114854818676e-06), (8, 2.099394805554766e-06), (9, 2.0983611648262013e-06), (10, 3.4667280033318093e-06), (11, 8.221326424973086e-06), (12, 3.311922409920953e-05), (13, 0.0011577615514397621), (14, 0.0013714346569031477), (15, 0.0004886757815256715), (16, 0.02168232947587967), (17, 7.360302697634324e-05), (18, 0.00010734247189247981), (19, 0.0014788194093853235), (20, 0.0009558800957165658), (21, 0.00011303451174171641), (22, 0.00020806245447602123), (23, 0.0001166804286185652), (24, 0.00018421534332446754), (25, 0.00042216203291900456), (26, 0.0004109737928956747), (27, 0.014855459332466125)], Temporal MSE: [(0, 5.997442258376395e-07), (1, 3.3306372415609076e-07), (2, 4.042129830850172e-07), (3, 7.54683924242272e-07), (4, 7.080656700964028e-07), (5, 9.881117648546933e-07), (6, 1.3779116443402017e-06), (7, 1.3487374417309184e-06), (8, 7.146151119741262e-07), (9, 8.591673577029724e-07), (10, 1.816550366129377e-06), (11, 1.7004605979309417e-05), (12, 2.265314105898142e-05), (13, 3.567871681298129e-05), (14, 1.9728768165805377e-05), (15, 4.544555122265592e-05), (16, 2.23206116061192e-05), (17, 2.3961218175827526e-05), (18, 4.413224814925343e-05), (19, 3.4747168683679774e-05), (20, 2.2704798539052717e-05), (21, 5.012853944208473e-05), (22, 6.746155850123614e-05), (23, 0.00013092433800920844), (24, 0.00012060951121384278), (25, 0.0003066976205445826), (26, 6.383860454661772e-05), (27, 6.506702629849315e-05)]
+
+ 70%|███████   | 105/150 [15:19<06:03,  8.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 264, Spatial MSE: [(0, 8.479217399326444e-07), (1, 1.0588776603981387e-05), (2, 1.3351173038245179e-05), (3, 1.0045079761766829e-05), (4, 1.7031952666002326e-05), (5, 2.4709948775125667e-05), (6, 5.864693957846612e-05), (7, 0.0001278995187021792), (8, 0.00011168955825269222), (9, 0.00016992801101878285), (10, 0.0001481018989579752), (11, 0.00016471515118610114), (12, 0.00019749898638110608), (13, 0.001823292812332511), (14, 0.003086777636781335), (15, 0.0020742244087159634), (16, 0.04227719455957413), (17, 0.0004372451512608677), (18, 0.0005848316941410303), (19, 0.004171639680862427), (20, 0.003905264660716057), (21, 0.0007923016673885286), (22, 0.0008266180520877242), (23, 0.0012898582499474287), (24, 0.002027148613706231), (25, 0.00465115811675787), (26, 0.008266338147222996), (27, 0.014788356609642506)], Temporal MSE: [(0, 1.0530872032177285e-06), (1, 4.6889960003682063e-07), (2, 1.8251396340929205e-06), (3, 4.344516128185205e-06), (4, 8.92480784386862e-06), (5, 1.535349656478502e-05), (6, 6.74296315992251e-05), (7, 4.050830830237828e-05), (8, 2.26777556235902e-05), (9, 3.558311072993092e-05), (10, 5.431391764432192e-05), (11, 7.636498776264489e-05), (12, 8.522631105734035e-05), (13, 0.00014785156236030161), (14, 0.0002118941192748025), (15, 0.00020676612621173263), (16, 0.00020998652325943112), (17, 0.0001922699884744361), (18, 0.0002268731768708676), (19, 0.0003360334667377174), (20, 0.0005014744237996638), (21, 0.00043346217717044055), (22, 0.00041954239713959396), (23, 0.0006408228655345738), (24, 0.0015793214552104473), (25, 0.0023595041129738092), (26, 0.0006551636615768075), (27, 0.00011826984700746834)]
+
+ 71%|███████   | 106/150 [15:29<06:13,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 258, Spatial MSE: [(0, 6.515729182865471e-07), (1, 7.187735263869399e-06), (2, 1.1597966476983856e-05), (3, 6.24981521468726e-06), (4, 1.9153278572048293e-06), (5, 1.984643859032076e-06), (6, 1.9081828668277012e-06), (7, 3.439193505982985e-06), (8, 2.244364623038564e-06), (9, 2.7229705210629618e-06), (10, 3.729654281414696e-06), (11, 8.453576811007224e-06), (12, 2.6010782676166855e-05), (13, 0.0009271279559470713), (14, 0.0012518430594354868), (15, 0.0005123808514326811), (16, 0.01982790231704712), (17, 6.785022560507059e-05), (18, 5.327875987859443e-05), (19, 0.0013221033150330186), (20, 0.0007418628083541989), (21, 0.0001115894119720906), (22, 0.00021366927830968052), (23, 0.0001230446359841153), (24, 0.00017918131197802722), (25, 0.0003960898902732879), (26, 0.00038539012894034386), (27, 0.01565593294799328)], Temporal MSE: [(0, 7.229087373161747e-07), (1, 3.8976511973487504e-07), (2, 4.7641410105825344e-07), (3, 9.162266110251949e-07), (4, 7.878567203079001e-07), (5, 1.1035488114430336e-06), (6, 1.619213094272709e-06), (7, 1.421953470526205e-06), (8, 8.550616144020751e-07), (9, 9.862875458566123e-07), (10, 2.1514404124900466e-06), (11, 1.2565134966280311e-05), (12, 1.7381538782501593e-05), (13, 3.4355274692643434e-05), (14, 2.0415322069311514e-05), (15, 4.538252323982306e-05), (16, 2.6506850190344267e-05), (17, 2.225934986199718e-05), (18, 4.911711948807351e-05), (19, 4.076149343745783e-05), (20, 2.5852068574749865e-05), (21, 3.854938404401764e-05), (22, 5.455155769595876e-05), (23, 0.00012134372082073241), (24, 0.00012850324856117368), (25, 0.00031677610240876675), (26, 6.974153802730143e-05), (27, 7.027771789580584e-05)]
+
+ 71%|███████▏  | 107/150 [15:37<05:58,  8.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 252, Spatial MSE: [(0, 8.139978149301896e-07), (1, 7.718298547843006e-06), (2, 9.757314728631172e-06), (3, 5.675162356055807e-06), (4, 3.820470283244504e-06), (5, 7.166917384893168e-06), (6, 1.6168003639904782e-05), (7, 1.1787929906859063e-05), (8, 1.8660053683561273e-05), (9, 3.7066740333102643e-05), (10, 5.014269845560193e-05), (11, 6.94501941325143e-05), (12, 8.402128150919452e-05), (13, 0.001248653861694038), (14, 0.0019403939368203282), (15, 0.0009827952599152923), (16, 0.0315704382956028), (17, 0.0001081708978745155), (18, 0.00012237693590577692), (19, 0.0018939361907541752), (20, 0.0011605286272242665), (21, 0.000189150232472457), (22, 0.0002837887150235474), (23, 0.00024736765772104263), (24, 0.0005229460075497627), (25, 0.001500347862020135), (26, 0.003037941874936223), (27, 0.015628406777977943)], Temporal MSE: [(0, 6.824052434240002e-07), (1, 3.93085286987116e-07), (2, 7.086063078531879e-07), (3, 2.1637461031787097e-06), (4, 2.5135143459920073e-06), (5, 3.0802627861703513e-06), (6, 6.048222076060483e-06), (7, 6.911029231559951e-06), (8, 5.384156793297734e-06), (9, 6.127142569312127e-06), (10, 9.673522981756832e-06), (11, 2.4609154934296384e-05), (12, 3.1288815080188215e-05), (13, 5.479686660692096e-05), (14, 4.8291432904079556e-05), (15, 7.152122270781547e-05), (16, 4.095858093933202e-05), (17, 3.703603215399198e-05), (18, 6.630337884416804e-05), (19, 6.264405237743631e-05), (20, 5.659618545905687e-05), (21, 7.546792039647698e-05), (22, 0.00010032251884695143), (23, 0.00017635915719438344), (24, 0.00025752271176315844), (25, 0.0006812210776843131), (26, 0.00018721737433224916), (27, 8.348425762960687e-05)]
+
+ 72%|███████▏  | 108/150 [15:45<05:48,  8.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 246, Spatial MSE: [(0, 7.150627538976551e-07), (1, 8.880098903318867e-06), (2, 1.212216739077121e-05), (3, 5.903699275222607e-06), (4, 1.7648540051595774e-06), (5, 1.791216391211492e-06), (6, 1.9180097297066823e-06), (7, 4.57168607681524e-06), (8, 2.4022785964916693e-06), (9, 2.076593773381319e-06), (10, 6.411009962903336e-06), (11, 2.7471533030620776e-05), (12, 6.196300819283351e-05), (13, 0.0011711986735463142), (14, 0.00135809148196131), (15, 0.0005116345128044486), (16, 0.022740405052900314), (17, 0.00010489943815628067), (18, 5.5579032050445676e-05), (19, 0.0015283044194802642), (20, 0.000953948067035526), (21, 8.771914144745097e-05), (22, 0.0002066900342470035), (23, 0.00012698900536634028), (24, 0.00020206997578497976), (25, 0.0004131332680117339), (26, 0.0003989647957496345), (27, 0.015569441951811314)], Temporal MSE: [(0, 6.14559723999264e-07), (1, 3.2227555379904516e-07), (2, 4.2086429630217026e-07), (3, 7.627517675246054e-07), (4, 7.561484949292208e-07), (5, 1.0545562645347673e-06), (6, 1.530254166937084e-06), (7, 1.4325925121738692e-06), (8, 8.159789786077454e-07), (9, 1.0086707789014326e-06), (10, 3.257607431805809e-06), (11, 6.701828533550724e-05), (12, 2.7175530703971162e-05), (13, 2.6583491489873268e-05), (14, 2.2977757907938212e-05), (15, 5.1550330681493506e-05), (16, 2.7474132366478443e-05), (17, 2.2454365534940735e-05), (18, 4.831657133763656e-05), (19, 4.0153874579118565e-05), (20, 2.5101327992160805e-05), (21, 4.520948641584255e-05), (22, 6.536777800647542e-05), (23, 0.0001275445829378441), (24, 0.00012645023525692523), (25, 0.0003216837649233639), (26, 6.376059172907844e-05), (27, 7.26703365216963e-05)]
+
+ 73%|███████▎  | 109/150 [15:53<05:39,  8.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 240, Spatial MSE: [(0, 9.776526894711424e-07), (1, 9.553963536745869e-06), (2, 1.8813894712366164e-05), (3, 9.898541975417174e-06), (4, 1.525295647297753e-05), (5, 2.4148919692379422e-05), (6, 5.201085878070444e-05), (7, 0.00011686049401760101), (8, 0.0001172609263448976), (9, 0.0001623073039809242), (10, 0.00012626890384126455), (11, 0.0001552411267766729), (12, 0.00020419959037099034), (13, 0.001457035425119102), (14, 0.0021206908859312534), (15, 0.0012355615617707372), (16, 0.03544636070728302), (17, 0.00042440733523108065), (18, 0.00048010703176259995), (19, 0.0025670856703072786), (20, 0.0019540477078408003), (21, 0.0006968046654947102), (22, 0.0007810618262737989), (23, 0.0012858005939051509), (24, 0.00201903167180717), (25, 0.004182224161922932), (26, 0.0075378697365522385), (27, 0.01586172543466091)], Temporal MSE: [(0, 1.0341567531213514e-06), (1, 4.94313837862137e-07), (2, 1.9511278424033662e-06), (3, 4.7118787733779754e-06), (4, 9.346722436021082e-06), (5, 1.3905484593124129e-05), (6, 6.158501491881907e-05), (7, 3.964655479649082e-05), (8, 2.23937640839722e-05), (9, 3.4658092772588134e-05), (10, 5.294912261888385e-05), (11, 0.00011910973262274638), (12, 8.079261897364631e-05), (13, 0.00012716460332740098), (14, 0.00020760971528943628), (15, 0.00020867126295343041), (16, 0.00020399670756887645), (17, 0.00018192014249507338), (18, 0.00021235040912870318), (19, 0.0003032960812561214), (20, 0.00043868477223441005), (21, 0.00041278472053818405), (22, 0.00042202562326565385), (23, 0.000632364593911916), (24, 0.0015263285022228956), (25, 0.002522496273741126), (26, 0.0006778760580345988), (27, 0.00013592060713563114)]
+
+ 73%|███████▎  | 110/150 [16:02<05:34,  8.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 234, Spatial MSE: [(0, 6.787806228203408e-07), (1, 7.820572136552073e-06), (2, 1.21636630865396e-05), (3, 6.488662165793357e-06), (4, 1.9537380921974545e-06), (5, 2.3304360183828976e-06), (6, 2.0560478333209176e-06), (7, 4.7537419050058816e-06), (8, 2.712363539103535e-06), (9, 2.5232325242541265e-06), (10, 4.538953817245783e-06), (11, 6.774775101803243e-06), (12, 1.8475277101970278e-05), (13, 0.0007223148713819683), (14, 0.0009307471918873489), (15, 0.00046256801579147577), (16, 0.018349986523389816), (17, 6.849451892776415e-05), (18, 3.854169699479826e-05), (19, 0.0013725373428314924), (20, 0.0008991578943096101), (21, 7.172604091465473e-05), (22, 0.0001571343163959682), (23, 0.00011054339120164514), (24, 0.00017463171388953924), (25, 0.0003849353233817965), (26, 0.00037723398418165743), (27, 0.01356931310147047)], Temporal MSE: [(0, 7.648131372661737e-07), (1, 4.294819007100159e-07), (2, 5.114581540510699e-07), (3, 9.664504432294052e-07), (4, 9.217887964041438e-07), (5, 1.2620023426279658e-06), (6, 2.003455847443547e-06), (7, 1.7864111896415125e-06), (8, 1.036695380207675e-06), (9, 1.199904659188178e-06), (10, 2.413841684756335e-06), (11, 8.420424819632899e-06), (12, 1.1234585144848097e-05), (13, 2.7830292310682125e-05), (14, 1.881593925645575e-05), (15, 2.6531584808253683e-05), (16, 1.784953747119289e-05), (17, 2.5166242267005146e-05), (18, 3.806677705142647e-05), (19, 3.1834220862947404e-05), (20, 2.0378454792080447e-05), (21, 4.164372876402922e-05), (22, 5.6515971664339304e-05), (23, 0.00010747900523710996), (24, 0.00010036260209744796), (25, 0.0002751079446170479), (26, 5.877536750631407e-05), (27, 6.240943184820935e-05)]
+
+ 74%|███████▍  | 111/150 [16:09<05:11,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 228, Spatial MSE: [(0, 7.297620072677091e-07), (1, 7.582896159874508e-06), (2, 1.0361379281675909e-05), (3, 5.63993171454058e-06), (4, 4.105444986635121e-06), (5, 8.122615327010863e-06), (6, 2.0293642592150718e-05), (7, 1.5011310097179376e-05), (8, 2.0433162717381492e-05), (9, 4.1186543967342004e-05), (10, 6.314743950497359e-05), (11, 7.558413926744834e-05), (12, 7.586936408188194e-05), (13, 0.0008066291338764131), (14, 0.0010266833705827594), (15, 0.0005374508909881115), (16, 0.018454229459166527), (17, 9.713833424029872e-05), (18, 0.00016289176710415632), (19, 0.0017201160080730915), (20, 0.001487197820097208), (21, 0.00015589177201036364), (22, 0.00022163050016388297), (23, 0.00024057023983914405), (24, 0.0005690772668458521), (25, 0.0017671743407845497), (26, 0.004002634901553392), (27, 0.013230645097792149)], Temporal MSE: [(0, 7.793285590196319e-07), (1, 4.4785340946873475e-07), (2, 7.39744962174882e-07), (3, 2.028876451731776e-06), (4, 2.5523222575429827e-06), (5, 3.4514882827352267e-06), (6, 7.247610028571216e-06), (7, 8.747741048864555e-06), (8, 6.4200144151982386e-06), (9, 8.076207450358197e-06), (10, 1.2400786545185838e-05), (11, 2.0404042516020127e-05), (12, 2.434171074128244e-05), (13, 4.672122304327786e-05), (14, 4.549950244836509e-05), (15, 4.176180300419219e-05), (16, 2.8816557460231707e-05), (17, 3.358892354299314e-05), (18, 5.050894105806947e-05), (19, 5.6294324167538434e-05), (20, 6.0806883993791416e-05), (21, 6.920415762579069e-05), (22, 8.953124051913619e-05), (23, 0.00014917658700142056), (24, 0.00023120130936149508), (25, 0.0006537073059007525), (26, 0.00020715837308671325), (27, 7.854929572204128e-05)]
+
+ 75%|███████▍  | 112/150 [16:16<04:55,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 222, Spatial MSE: [(0, 7.033599445094296e-07), (1, 7.27837095837458e-06), (2, 1.2849353879573755e-05), (3, 6.112006758485222e-06), (4, 2.044016355284839e-06), (5, 2.8901051791763166e-06), (6, 2.2013630314177135e-06), (7, 5.725728442484979e-06), (8, 3.079368980252184e-06), (9, 2.5010979243234033e-06), (10, 3.760935442187474e-06), (11, 5.137279003974982e-06), (12, 9.239227438229136e-06), (13, 0.00013087283878121525), (14, 6.263305840548128e-05), (15, 3.9372862374875695e-05), (16, 0.0027132821269333363), (17, 1.0883794857363682e-05), (18, 3.0014469302841462e-05), (19, 0.00016174554184544832), (20, 6.361950363498181e-05), (21, 4.3421154259704053e-05), (22, 6.99544616509229e-05), (23, 5.352721564122476e-05), (24, 8.576861000619829e-05), (25, 0.0001712181547190994), (26, 0.00024125780328176916), (27, 0.0023186523467302322)], Temporal MSE: [(0, 8.931951356316858e-07), (1, 4.890191007689282e-07), (2, 5.740787401009584e-07), (3, 1.0220321655651787e-06), (4, 9.845717841017176e-07), (5, 1.324746790487552e-06), (6, 1.9846040686388733e-06), (7, 1.8613668544276152e-06), (8, 1.0432142971694702e-06), (9, 1.2688691413131892e-06), (10, 2.1828454919159412e-06), (11, 5.7678807934280485e-06), (12, 4.950019501848146e-06), (13, 4.083628937223693e-06), (14, 4.309544692659983e-06), (15, 6.4187488533207215e-06), (16, 5.931307896389626e-06), (17, 5.242487986834021e-06), (18, 1.1754638762795366e-05), (19, 1.0432800991111435e-05), (20, 8.397151759709232e-06), (21, 9.588701686880086e-06), (22, 1.537989010103047e-05), (23, 3.442572779022157e-05), (24, 4.1426705138292164e-05), (25, 8.286238880828023e-05), (26, 3.221068254788406e-05), (27, 1.2015669199172407e-05)]
+
+ 75%|███████▌  | 113/150 [16:23<04:38,  7.52s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 216, Spatial MSE: [(0, 7.582710850329022e-07), (1, 9.892295565805398e-06), (2, 2.090885209327098e-05), (3, 1.022964443109231e-05), (4, 1.4880717571941204e-05), (5, 2.5290335543104447e-05), (6, 5.288992906571366e-05), (7, 0.0001275269314646721), (8, 0.00012557368609122932), (9, 0.00017382482474204153), (10, 0.00012837047688663006), (11, 0.00014093720528762788), (12, 0.00016343023162335157), (13, 0.00037840084405615926), (14, 0.0004792426771018654), (15, 0.00041622333810664713), (16, 0.004780437331646681), (17, 0.0003311080508865416), (18, 0.00042028340976685286), (19, 0.0006338803214021027), (20, 0.0008495288202539086), (21, 0.000601409818045795), (22, 0.0006330499891191721), (23, 0.0012154122814536095), (24, 0.001917885267175734), (25, 0.00416905153542757), (26, 0.007114791311323643), (27, 0.002212312538176775)], Temporal MSE: [(0, 8.428328328591306e-07), (1, 4.747916193537094e-07), (2, 2.1836829091625987e-06), (3, 4.604589776135981e-06), (4, 1.0226103768218309e-05), (5, 1.430513020750368e-05), (6, 6.165663944557309e-05), (7, 4.33706991316285e-05), (8, 2.4518694772268645e-05), (9, 3.734304482350126e-05), (10, 5.5577071179868653e-05), (11, 6.522111652884632e-05), (12, 6.426834443118423e-05), (13, 0.0001132854085881263), (14, 0.00019996646733488888), (15, 0.0001675104722380638), (16, 0.00019384536426514387), (17, 0.00016889069229364395), (18, 0.00018366564472671598), (19, 0.0002688331587705761), (20, 0.0004198353854008019), (21, 0.0003460901789367199), (22, 0.0003535337746143341), (23, 0.0005255204741843045), (24, 0.0013377973809838295), (25, 0.00208294577896595), (26, 0.000680120603647083), (27, 8.21785070002079e-05)]
+
+ 76%|███████▌  | 114/150 [16:30<04:26,  7.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 210, Spatial MSE: [(0, 9.041258977049438e-07), (1, 9.343245437776204e-06), (2, 1.3223741007095668e-05), (3, 6.458057669078698e-06), (4, 2.085291498588049e-06), (5, 3.3133949273178587e-06), (6, 2.572174025772256e-06), (7, 6.686836059088819e-06), (8, 3.578833684514393e-06), (9, 3.0505027552862884e-06), (10, 6.053307060938096e-06), (11, 1.0992072020599153e-05), (12, 2.6768186216941103e-05), (13, 0.0007246497552841902), (14, 0.000600120285525918), (15, 0.00019216399232391268), (16, 0.00566426245495677), (17, 2.4326160200871527e-05), (18, 4.415411240188405e-05), (19, 0.00041558287921361625), (20, 0.0001970531011465937), (21, 7.155376806622371e-05), (22, 0.00010695397213567048), (23, 8.026307477848604e-05), (24, 0.00012468778004404157), (25, 0.0002522082068026066), (26, 0.00033825860009528697), (27, 0.007596517913043499)], Temporal MSE: [(0, 9.00571251349902e-07), (1, 4.962536195307621e-07), (2, 5.984555286886462e-07), (3, 1.0482621064511477e-06), (4, 1.0243869610349066e-06), (5, 1.403913984177052e-06), (6, 1.9808048818958923e-06), (7, 2.2888607418281026e-06), (8, 1.2141157412770553e-06), (9, 1.5366083516710205e-06), (10, 3.6797666780330474e-06), (11, 2.108190346916672e-05), (12, 1.5573494238196872e-05), (13, 1.9262961359345354e-05), (14, 9.581305675965268e-06), (15, 1.41482678372995e-05), (16, 1.0131910130439792e-05), (17, 9.46627369557973e-06), (18, 2.0858864445472136e-05), (19, 1.6954927559709176e-05), (20, 1.2367431736493018e-05), (21, 1.583781158842612e-05), (22, 2.5827479475992732e-05), (23, 6.175674934638664e-05), (24, 6.790257612010464e-05), (25, 0.00018552422989159822), (26, 5.312264693202451e-05), (27, 2.9185184757807292e-05)]
+
+ 77%|███████▋  | 115/150 [16:38<04:25,  7.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 204, Spatial MSE: [(0, 8.563160918129142e-07), (1, 8.049843927437905e-06), (2, 1.785448876034934e-05), (3, 6.197545644681668e-06), (4, 4.3407258090155665e-06), (5, 8.754492228035815e-06), (6, 2.199456503149122e-05), (7, 1.5201114365481772e-05), (8, 2.4052062144619413e-05), (9, 4.142843317822553e-05), (10, 5.851563037140295e-05), (11, 7.204680150607601e-05), (12, 8.208995859604329e-05), (13, 0.0008466269355267286), (14, 0.0007537285564467311), (15, 0.00036282173823565245), (16, 0.009499931707978249), (17, 5.067389429314062e-05), (18, 5.099425106891431e-05), (19, 0.0005810483708046377), (20, 0.00042586514609865844), (21, 9.09904483705759e-05), (22, 0.00017000366642605513), (23, 0.00018912716768682003), (24, 0.0004932520678266883), (25, 0.0014673311961814761), (26, 0.003831330919638276), (27, 0.0064313062466681)], Temporal MSE: [(0, 8.151704946612881e-07), (1, 5.027615657127171e-07), (2, 8.069350769801531e-07), (3, 2.3638078801013762e-06), (4, 2.765954604910803e-06), (5, 3.811348506133072e-06), (6, 7.403362360491883e-06), (7, 9.385100383951794e-06), (8, 7.0506162046513055e-06), (9, 7.924658348201774e-06), (10, 1.1123361218778882e-05), (11, 3.13202035613358e-05), (12, 2.8031236070091836e-05), (13, 3.764150824281387e-05), (14, 3.598109833546914e-05), (15, 3.376315362402238e-05), (16, 2.284906804561615e-05), (17, 2.1198155081947334e-05), (18, 3.401203139219433e-05), (19, 3.752583870664239e-05), (20, 4.9005488108377904e-05), (21, 3.8875728932907805e-05), (22, 5.21343172295019e-05), (23, 0.00010180604294873774), (24, 0.0002044549910351634), (25, 0.0005345458630472422), (26, 0.000210799858905375), (27, 4.347383219283074e-05)]
+
+ 77%|███████▋  | 116/150 [16:46<04:19,  7.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 198, Spatial MSE: [(0, 7.517673452639428e-07), (1, 9.67433697951492e-06), (2, 1.3512089026335161e-05), (3, 5.845869054610375e-06), (4, 2.1156731691007735e-06), (5, 2.7313117243465967e-06), (6, 2.456672291373252e-06), (7, 5.886156941414811e-06), (8, 3.403162736503873e-06), (9, 2.8846325221820734e-06), (10, 3.8693196984240785e-06), (11, 6.0079678405600134e-06), (12, 1.1954012734349817e-05), (13, 0.0002398784999968484), (14, 0.0002430107706459239), (15, 0.00011995377280982211), (16, 0.004164174664765596), (17, 1.7642676539253443e-05), (18, 2.622476858959999e-05), (19, 0.00025322887813672423), (20, 0.00011449305748101324), (21, 7.413819548673928e-05), (22, 9.752449841471389e-05), (23, 6.681906234007329e-05), (24, 9.52393384068273e-05), (25, 0.00022815406555309892), (26, 0.0002939840778708458), (27, 0.005552362184971571)], Temporal MSE: [(0, 8.877055392986222e-07), (1, 5.03383603245311e-07), (2, 5.837765115757065e-07), (3, 1.0000057955039665e-06), (4, 1.033387093229976e-06), (5, 1.344794668511895e-06), (6, 2.3620439151272876e-06), (7, 2.141180857506697e-06), (8, 1.2174880339443916e-06), (9, 1.4931048326616292e-06), (10, 2.339244247195893e-06), (11, 7.707318218308501e-06), (12, 7.189010375441285e-06), (13, 1.0617363841447514e-05), (14, 8.08009735919768e-06), (15, 1.4297869711299427e-05), (16, 9.700805094325915e-06), (17, 1.1320803423586767e-05), (18, 1.5526997231063433e-05), (19, 1.3465941265167203e-05), (20, 1.7723530618241057e-05), (21, 1.1478434316813946e-05), (22, 2.127600419044029e-05), (23, 4.3138348701177165e-05), (24, 5.412741666077636e-05), (25, 0.00012064875045325607), (26, 4.071301736985333e-05), (27, 3.066564750042744e-05)]
+
+ 78%|███████▊  | 117/150 [16:53<04:06,  7.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 192, Spatial MSE: [(0, 1.0415843689770554e-06), (1, 9.531018804409541e-06), (2, 2.7040421628043987e-05), (3, 9.907924322760664e-06), (4, 1.5552195691270754e-05), (5, 2.5800563889788464e-05), (6, 5.037945811636746e-05), (7, 0.00013214153295848519), (8, 0.0001276720140594989), (9, 0.00018816003284882754), (10, 0.00013803130423184484), (11, 0.00015617079043295234), (12, 0.00017076694348361343), (13, 0.0004202238342259079), (14, 0.0009903261670842767), (15, 0.0008593008969910443), (16, 0.015361473895609379), (17, 0.0003518429584801197), (18, 0.0004654867516364902), (19, 0.0016432186821475625), (20, 0.0017266158247366548), (21, 0.0006485879421234131), (22, 0.0007418034947477281), (23, 0.001297260052524507), (24, 0.0019991283770650625), (25, 0.004517871420830488), (26, 0.008639609441161156), (27, 0.004879047628492117)], Temporal MSE: [(0, 1.2918958418595139e-06), (1, 6.422294518415583e-07), (2, 2.2610390715271933e-06), (3, 5.371896804717835e-06), (4, 1.084088216884993e-05), (5, 1.547767169540748e-05), (6, 6.810094782849774e-05), (7, 4.840820110985078e-05), (8, 2.6841746148420498e-05), (9, 4.1021561628440395e-05), (10, 5.845208579557948e-05), (11, 7.008232933003455e-05), (12, 7.133487815735862e-05), (13, 0.00012378368410281837), (14, 0.0002149306528735906), (15, 0.0001848669198807329), (16, 0.00020419347856659442), (17, 0.00017765283701010048), (18, 0.00018745832494460046), (19, 0.00027575186686590314), (20, 0.00042549247154965997), (21, 0.0003845693136099726), (22, 0.0003674266627058387), (23, 0.0005247349618002772), (24, 0.0013878586469218135), (25, 0.0022391993552446365), (26, 0.0007567800348624587), (27, 0.00010463056241860613)]
+
+ 79%|███████▊  | 118/150 [17:00<03:58,  7.45s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 186, Spatial MSE: [(0, 7.823788905625406e-07), (1, 1.0418621968710795e-05), (2, 1.5112451364984736e-05), (3, 6.673534244328039e-06), (4, 2.2669862573820865e-06), (5, 3.0300807338790037e-06), (6, 2.58056616075919e-06), (7, 6.304896032816032e-06), (8, 3.5260391086922027e-06), (9, 3.0662511107948376e-06), (10, 3.084396212216234e-06), (11, 3.9706101233605295e-06), (12, 5.831063845107565e-06), (13, 5.142735244589858e-05), (14, 4.255789826856926e-05), (15, 1.8656206520972773e-05), (16, 2.7340985980117694e-05), (17, 7.222415661090054e-06), (18, 1.0083114830194972e-05), (19, 1.694002639851533e-05), (20, 9.785267138795462e-06), (21, 2.6637209884938784e-05), (22, 4.944303509546444e-05), (23, 3.9537353586638346e-05), (24, 6.979145109653473e-05), (25, 0.0001355632412014529), (26, 0.00020799484627787024), (27, 0.0015251656295731664)], Temporal MSE: [(0, 8.516813636560983e-07), (1, 5.051454650129017e-07), (2, 6.164764840832504e-07), (3, 1.0584667506918777e-06), (4, 1.1868141882587224e-06), (5, 1.537321622890886e-06), (6, 2.791168981275405e-06), (7, 2.3179604795586783e-06), (8, 1.456926838727668e-06), (9, 1.6549025758649805e-06), (10, 2.0397633306856733e-06), (11, 3.1690949526819168e-06), (12, 2.8459171517170034e-06), (13, 4.162450295552844e-06), (14, 4.675848231272539e-06), (15, 3.9961082620720845e-06), (16, 3.6044832540937932e-06), (17, 3.1801389468455454e-06), (18, 3.3741573588486062e-06), (19, 4.000252374680713e-06), (20, 4.838049790123478e-06), (21, 7.186885341070592e-06), (22, 1.3145692719263025e-05), (23, 1.8650422134669498e-05), (24, 2.5555704269208945e-05), (25, 7.22594340913929e-05), (26, 2.8376769478200004e-05), (27, 9.620193850423675e-06)]
+
+ 79%|███████▉  | 119/150 [17:08<03:51,  7.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 180, Spatial MSE: [(0, 1.040885422298743e-06), (1, 8.992289622256067e-06), (2, 2.107393265760038e-05), (3, 6.706612111884169e-06), (4, 5.229581347521162e-06), (5, 1.0261621355311945e-05), (6, 2.390546796959825e-05), (7, 1.775029522832483e-05), (8, 2.4205637600971386e-05), (9, 4.9722315452527255e-05), (10, 6.172121356939897e-05), (11, 7.277872646227479e-05), (12, 6.281847890932113e-05), (13, 0.0001335667329840362), (14, 0.00013101664080750197), (15, 9.845876775216311e-05), (16, 6.595224112970755e-05), (17, 4.0042166801868007e-05), (18, 5.668331868946552e-05), (19, 6.274729094002396e-05), (20, 5.8388217439642176e-05), (21, 7.459469634341076e-05), (22, 0.00011977191752521321), (23, 0.0001966685231309384), (24, 0.0005606378545053303), (25, 0.0016038187313824892), (26, 0.00411646906286478), (27, 0.0021976579446345568)], Temporal MSE: [(0, 1.136723540184903e-06), (1, 7.282696401489375e-07), (2, 9.70314204096212e-07), (3, 2.4601310997240944e-06), (4, 3.198243121005362e-06), (5, 4.232288119965233e-06), (6, 9.137391316471621e-06), (7, 1.120265824283706e-05), (8, 8.797655937087256e-06), (9, 9.723573384690098e-06), (10, 1.1852420357172377e-05), (11, 1.608817183296196e-05), (12, 1.723366221995093e-05), (13, 2.3695351046626456e-05), (14, 3.523972918628715e-05), (15, 2.3175000023911707e-05), (16, 1.8817188902175985e-05), (17, 1.519519901194144e-05), (18, 1.8063230527332053e-05), (19, 2.826573108904995e-05), (20, 3.9351034502033144e-05), (21, 3.563521386240609e-05), (22, 4.6490564272971824e-05), (23, 7.328040373977274e-05), (24, 0.00020235312695149332), (25, 0.0004730845394078642), (26, 0.00022550608264282346), (27, 3.738852683454752e-05)]
+
+ 80%|████████  | 120/150 [17:17<03:54,  7.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 174, Spatial MSE: [(0, 7.112373054951604e-07), (1, 1.0678835678845644e-05), (2, 1.5861323845456354e-05), (3, 6.183703135320684e-06), (4, 2.3045397483656416e-06), (5, 3.3759617963369237e-06), (6, 2.854086915249354e-06), (7, 7.539556918345625e-06), (8, 4.378148787509417e-06), (9, 2.5730341803864576e-06), (10, 3.740798547369195e-06), (11, 3.6637059110944392e-06), (12, 6.062036391085712e-06), (13, 5.8491568779572845e-05), (14, 2.4953365937108174e-05), (15, 7.738493877695873e-06), (16, 4.057067781104706e-05), (17, 1.087502641894389e-05), (18, 2.0676989151979797e-05), (19, 1.5276171325240284e-05), (20, 1.0278611625835765e-05), (21, 1.3109856809023768e-05), (22, 3.350236875121482e-05), (23, 4.148251537117176e-05), (24, 6.943559856154025e-05), (25, 0.00013653513451572508), (26, 0.00022863321646582335), (27, 0.0019760269206017256)], Temporal MSE: [(0, 9.656898782850476e-07), (1, 5.593523155766889e-07), (2, 6.5292101680825e-07), (3, 1.0736553122114856e-06), (4, 1.2229374988237396e-06), (5, 1.5999008837752626e-06), (6, 2.8549422950163716e-06), (7, 2.582091156000388e-06), (8, 1.4680067579320166e-06), (9, 1.7233983271580655e-06), (10, 2.0612424123100936e-06), (11, 3.639925125753507e-06), (12, 3.333055019538733e-06), (13, 3.8005805436114315e-06), (14, 3.9934207052283455e-06), (15, 4.179842562734848e-06), (16, 4.4407106543076225e-06), (17, 5.316171609592857e-06), (18, 3.706585175677901e-06), (19, 4.60513820144115e-06), (20, 5.568604592554038e-06), (21, 5.649157628795365e-06), (22, 9.588875400368124e-06), (23, 1.7559228581376374e-05), (24, 2.519916233723052e-05), (25, 5.4332384024746716e-05), (26, 3.0473136575892568e-05), (27, 1.1218121471756604e-05)]
+
+ 81%|████████  | 121/150 [17:24<03:43,  7.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 168, Spatial MSE: [(0, 1.0149271929549286e-06), (1, 1.1300380720058456e-05), (2, 3.506327266222797e-05), (3, 1.1110664672742132e-05), (4, 1.5589439499308355e-05), (5, 2.6621235519996844e-05), (6, 5.555746975005604e-05), (7, 0.00013958060299046338), (8, 0.000141383265145123), (9, 0.000191316386917606), (10, 0.00014189774810802191), (11, 0.00015369479660876095), (12, 0.00017019707593135536), (13, 0.0002960186975542456), (14, 0.00042422927799634635), (15, 0.0004224783042445779), (16, 0.00046948008821345866), (17, 0.0003732723998837173), (18, 0.0004262383154127747), (19, 0.000451590254670009), (20, 0.0005402536480687559), (21, 0.0006228885613381863), (22, 0.0006486046477220953), (23, 0.0013422572519630194), (24, 0.002173792337998748), (25, 0.0048717535100877285), (26, 0.009770467877388), (27, 0.0019422639161348343)], Temporal MSE: [(0, 1.427909296580765e-06), (1, 7.616381481057033e-07), (2, 2.5093888780247653e-06), (3, 6.32317778581637e-06), (4, 1.2234088899276685e-05), (5, 1.601999974809587e-05), (6, 6.647632835665718e-05), (7, 5.110346683068201e-05), (8, 2.9773094865959138e-05), (9, 4.4451629946706817e-05), (10, 6.163825310068205e-05), (11, 7.19958043191582e-05), (12, 7.516861660405993e-05), (13, 0.0001321349700447172), (14, 0.00024010504421312362), (15, 0.00019573832105379552), (16, 0.00021799010573886335), (17, 0.0001928883430082351), (18, 0.000192136358236894), (19, 0.0002770631108433008), (20, 0.00044645232264883816), (21, 0.00036644688225351274), (22, 0.00037622565287165344), (23, 0.0005448461160995066), (24, 0.0013968301936984062), (25, 0.002231478225439787), (26, 0.0008618649444542825), (27, 0.00011136173270642757)]
+
+ 81%|████████▏ | 122/150 [17:35<03:59,  8.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 162, Spatial MSE: [(0, 9.27424366636842e-07), (1, 1.246979536517756e-05), (2, 1.8443906810716726e-05), (3, 6.985865638853284e-06), (4, 2.4693983959878096e-06), (5, 4.991372406948358e-06), (6, 3.4519509881647537e-06), (7, 9.369034160044976e-06), (8, 5.4412389545177575e-06), (9, 3.447319386395975e-06), (10, 4.5241572479426395e-06), (11, 6.36207823845325e-06), (12, 1.0364733498136047e-05), (13, 0.0001456971513107419), (14, 5.772130316472612e-05), (15, 2.027674054261297e-05), (16, 0.0008194074616767466), (17, 1.3423350537777878e-05), (18, 4.394277129904367e-05), (19, 5.8344710851088166e-05), (20, 1.7149666746263392e-05), (21, 6.581553316209465e-05), (22, 7.02470206306316e-05), (23, 5.9772617532871664e-05), (24, 0.00010006057709688321), (25, 0.00019862846238538623), (26, 0.00027903937734663486), (27, 0.0022305597085505724)], Temporal MSE: [(0, 1.0491011153135332e-06), (1, 6.414633730855712e-07), (2, 7.449500003531284e-07), (3, 1.171817757494864e-06), (4, 1.3617126342069241e-06), (5, 1.801464350137394e-06), (6, 3.3883779906318523e-06), (7, 3.171358912368305e-06), (8, 1.8057762645184994e-06), (9, 2.1632831703755073e-06), (10, 2.854882040992379e-06), (11, 6.603605925192824e-06), (12, 6.21651179244509e-06), (13, 6.549732916028006e-06), (14, 6.572760412382195e-06), (15, 8.401645573030692e-06), (16, 7.117730092431884e-06), (17, 6.188155566633213e-06), (18, 7.712618753430434e-06), (19, 7.490908046747791e-06), (20, 8.003750735952053e-06), (21, 9.393512300448492e-06), (22, 1.761468047334347e-05), (23, 3.2462074159411713e-05), (24, 3.803460640483536e-05), (25, 8.454146154690534e-05), (26, 3.872128945658915e-05), (27, 1.5622250430169515e-05)]
+
+ 82%|████████▏ | 123/150 [17:42<03:45,  8.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 156, Spatial MSE: [(0, 1.0528458460612455e-06), (1, 1.1090563930338249e-05), (2, 3.4198546927655116e-05), (3, 7.90945159678813e-06), (4, 8.223218173952773e-06), (5, 1.2541387150122318e-05), (6, 2.39328874158673e-05), (7, 5.155176040716469e-05), (8, 6.660257349722087e-05), (9, 0.00010605984425637871), (10, 7.662814459763467e-05), (11, 9.083096665563062e-05), (12, 9.168163524009287e-05), (13, 0.0005984629387967288), (14, 0.0011664048070088029), (15, 0.0008542537689208984), (16, 0.019252706319093704), (17, 0.00016609930025879294), (18, 0.00019020495528820902), (19, 0.0017255559796467423), (20, 0.0016076245810836554), (21, 0.000292481534415856), (22, 0.0002699257747735828), (23, 0.0004903571680188179), (24, 0.0008246797369793057), (25, 0.002271170960739255), (26, 0.005390885751694441), (27, 0.001643107389099896)], Temporal MSE: [(0, 1.3440533166431123e-06), (1, 7.516260893680737e-07), (2, 1.4917720818630187e-06), (3, 2.5931044547178317e-06), (4, 3.95603137803846e-06), (5, 5.149843673279975e-06), (6, 2.5816401830525137e-05), (7, 2.249229873996228e-05), (8, 1.2153332136222161e-05), (9, 1.84445598279126e-05), (10, 1.9602644897531718e-05), (11, 2.8754207960446365e-05), (12, 3.2314892450813204e-05), (13, 4.861273555434309e-05), (14, 8.666784560773522e-05), (15, 7.49239043216221e-05), (16, 7.208294118754566e-05), (17, 6.090556416893378e-05), (18, 7.26513535482809e-05), (19, 9.24298947211355e-05), (20, 0.00013405423669610173), (21, 0.0001316483539994806), (22, 0.00012863446318078786), (23, 0.00019893918943125755), (24, 0.0004204471770208329), (25, 0.0006175320013426244), (26, 0.00027677897014655173), (27, 3.36666707880795e-05)]
+
+ 83%|████████▎ | 124/150 [17:51<03:40,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 150, Spatial MSE: [(0, 9.422712992090965e-07), (1, 1.3140382179699373e-05), (2, 2.032257179962471e-05), (3, 7.59550630391459e-06), (4, 2.982716978294775e-06), (5, 4.5993529056431726e-06), (6, 3.9551296140416525e-06), (7, 9.267815585189965e-06), (8, 5.7860611377691384e-06), (9, 3.7162008084123954e-06), (10, 4.5659126044483855e-06), (11, 4.640502083930187e-06), (12, 7.446027666446753e-06), (13, 6.380458944477141e-05), (14, 3.527108856360428e-05), (15, 1.7944705177797005e-05), (16, 0.0005534255178645253), (17, 1.3309742826095317e-05), (18, 6.398757250281051e-05), (19, 2.5642615582910366e-05), (20, 2.1716925402870402e-05), (21, 6.020140426699072e-05), (22, 5.268837048788555e-05), (23, 5.807670459034853e-05), (24, 0.00010001890041166916), (25, 0.00019052589777857065), (26, 0.00028138235211372375), (27, 0.0020997212268412113)], Temporal MSE: [(0, 1.2234080486450694e-06), (1, 7.600240792271507e-07), (2, 8.553276984457625e-07), (3, 1.3350793324207189e-06), (4, 1.6345940139217419e-06), (5, 2.122581918229116e-06), (6, 4.039340183226159e-06), (7, 3.6018984701513546e-06), (8, 2.2354076918418286e-06), (9, 2.519163217584719e-06), (10, 2.9953912417113315e-06), (11, 4.594511210598284e-06), (12, 4.179413735982962e-06), (13, 5.5087884902604856e-06), (14, 6.188289717101725e-06), (15, 6.930358722456731e-06), (16, 6.457960353145609e-06), (17, 6.3744464569026604e-06), (18, 5.97730968365795e-06), (19, 6.91025434207404e-06), (20, 8.05275521997828e-06), (21, 1.4679313608212397e-05), (22, 1.714970130706206e-05), (23, 2.63652818830451e-05), (24, 3.694164479384199e-05), (25, 8.044823334785178e-05), (26, 4.2852268961723894e-05), (27, 1.6526486433576792e-05)]
+
+ 83%|████████▎ | 125/150 [18:03<03:53,  9.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 144, Spatial MSE: [(0, 1.0891214969888097e-06), (1, 1.2041011359542608e-05), (2, 3.1796531402505934e-05), (3, 7.5466828093340155e-06), (4, 6.693174782412825e-06), (5, 1.2634714948944747e-05), (6, 2.5604334950912744e-05), (7, 2.1691206711693667e-05), (8, 2.9237678973004222e-05), (9, 5.595370748778805e-05), (10, 7.578893564641476e-05), (11, 9.199499618262053e-05), (12, 7.505880057578906e-05), (13, 0.00017828962882049382), (14, 0.00019821386376861483), (15, 0.00018578078015707433), (16, 0.0024660946801304817), (17, 5.475605939864181e-05), (18, 5.9954658354399726e-05), (19, 0.0001093749378924258), (20, 0.00011910744069609791), (21, 0.00017093461065087467), (22, 0.0001388133823638782), (23, 0.0002547356707509607), (24, 0.0007303539314307272), (25, 0.0018206279492005706), (26, 0.0046025225892663), (27, 0.0021492072846740484)], Temporal MSE: [(0, 1.3905161040383973e-06), (1, 9.163429695036029e-07), (2, 1.1380207070033066e-06), (3, 2.650511760293739e-06), (4, 3.6955693758500274e-06), (5, 5.07392405779683e-06), (6, 1.0549877515586559e-05), (7, 1.2705188964901026e-05), (8, 1.0756681149359792e-05), (9, 1.1829703908006195e-05), (10, 1.4083722817304078e-05), (11, 1.9107736079604365e-05), (12, 2.019722160184756e-05), (13, 2.8551203286042437e-05), (14, 4.405489016789943e-05), (15, 3.312479020678438e-05), (16, 2.7444892111816444e-05), (17, 2.479628710716497e-05), (18, 2.3964645151863806e-05), (19, 3.354409636813216e-05), (20, 4.993550828658044e-05), (21, 4.6640456275781617e-05), (22, 5.420215893536806e-05), (23, 9.115919237956405e-05), (24, 0.00026131144841201603), (25, 0.0005409355508163571), (26, 0.000303150387480855), (27, 5.533228613785468e-05)]
+
+ 84%|████████▍ | 126/150 [18:14<04:02, 10.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 138, Spatial MSE: [(0, 9.30217311179149e-07), (1, 1.3979296454635914e-05), (2, 2.215237691416405e-05), (3, 7.459052085323492e-06), (4, 3.0660182801511837e-06), (5, 4.412091584526934e-06), (6, 3.838059456029441e-06), (7, 8.417933713644743e-06), (8, 5.69469830224989e-06), (9, 4.726000952359755e-06), (10, 4.577789241011487e-06), (11, 5.338034952728776e-06), (12, 8.142299520841334e-06), (13, 0.00012146010703872889), (14, 0.0001429868134437129), (15, 9.954121196642518e-05), (16, 9.431412763660774e-05), (17, 1.8964798073284328e-05), (18, 1.6515043171239085e-05), (19, 3.4854558180086315e-05), (20, 1.820249599404633e-05), (21, 0.00010148347791982815), (22, 0.00013314088573679328), (23, 7.312683737836778e-05), (24, 0.00012054314720444381), (25, 0.00021174328867346048), (26, 0.0002598335559014231), (27, 0.002096158219501376)], Temporal MSE: [(0, 1.359434349978983e-06), (1, 8.019937354220019e-07), (2, 9.312220186075137e-07), (3, 1.371740495414997e-06), (4, 1.6907440567592857e-06), (5, 2.1611595002468675e-06), (6, 4.022600478492677e-06), (7, 3.623026259447215e-06), (8, 2.4328576273546787e-06), (9, 2.7260707611276302e-06), (10, 3.0940352644392988e-06), (11, 5.10365680383984e-06), (12, 5.227534074947471e-06), (13, 9.139917892753147e-06), (14, 1.1424368494772352e-05), (15, 1.0203196325164754e-05), (16, 8.438147233391646e-06), (17, 7.3248579610663e-06), (18, 5.669304300681688e-06), (19, 6.937892976566218e-06), (20, 8.58045768836746e-06), (21, 2.080563899653498e-05), (22, 3.2815551094245166e-05), (23, 4.066968540428206e-05), (24, 4.354360862635076e-05), (25, 8.035562495933846e-05), (26, 4.318910578149371e-05), (27, 1.8825319784809835e-05)]
+
+ 85%|████████▍ | 127/150 [18:26<04:03, 10.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 132, Spatial MSE: [(0, 1.1414093705752748e-06), (1, 1.4044097952137236e-05), (2, 4.952783274347894e-05), (3, 1.231754959007958e-05), (4, 1.626811172172893e-05), (5, 3.054528497159481e-05), (6, 6.655367178609595e-05), (7, 0.00015517749125137925), (8, 0.00017571626813150942), (9, 0.0002274677826790139), (10, 0.00015046725457068533), (11, 0.00017415077309124172), (12, 0.00018610765982884914), (13, 0.0007792407996021211), (14, 0.0021741054952144623), (15, 0.0015849549090489745), (16, 0.036797747015953064), (17, 0.0004852012498304248), (18, 0.0005193821853026748), (19, 0.002287960611283779), (20, 0.0018870416097342968), (21, 0.0008805067627690732), (22, 0.0008824167889542878), (23, 0.0015065517509356141), (24, 0.002530389931052923), (25, 0.005987769458442926), (26, 0.011792614124715328), (27, 0.0025133925955742598)], Temporal MSE: [(0, 1.736633748805616e-06), (1, 1.0015187399403658e-06), (2, 2.7653773031488527e-06), (3, 8.783664270595182e-06), (4, 1.4258010196499527e-05), (5, 1.802007682272233e-05), (6, 9.233132004737854e-05), (7, 6.6892069298774e-05), (8, 3.7960970075801015e-05), (9, 5.420711750048213e-05), (10, 6.983728962950408e-05), (11, 8.650071686133742e-05), (12, 9.693900210550055e-05), (13, 0.00016499546472914517), (14, 0.0002916950616054237), (15, 0.0002403311664238572), (16, 0.0002560371649451554), (17, 0.0002274166326969862), (18, 0.00022522131621371955), (19, 0.0003018289862666279), (20, 0.00046175398165360093), (21, 0.0004101531521882862), (22, 0.00043244464905001223), (23, 0.0006473223911598325), (24, 0.0014924510614946485), (25, 0.002426797989755869), (26, 0.0010484977392479777), (27, 0.00015549841918982565)]
+
+ 85%|████████▌ | 128/150 [18:36<03:48, 10.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 126, Spatial MSE: [(0, 9.816367310122587e-07), (1, 1.6256430171779357e-05), (2, 2.6550636903266422e-05), (3, 8.69062751007732e-06), (4, 3.612889031501254e-06), (5, 4.91510718347854e-06), (6, 4.357277248345781e-06), (7, 8.889322998584248e-06), (8, 6.341461357806111e-06), (9, 4.33842433267273e-06), (10, 6.457461495301686e-06), (11, 6.756606126145925e-06), (12, 1.1122770956717432e-05), (13, 0.00022432157129514962), (14, 0.0004911274882033467), (15, 0.0002845577255357057), (16, 0.016542639583349228), (17, 7.27818624000065e-05), (18, 0.00015940253797452897), (19, 0.0013146020937711), (20, 0.0010507326805964112), (21, 0.00010619048407534137), (22, 0.00019775563850998878), (23, 0.00012097431317670271), (24, 0.00023117107048165053), (25, 0.0005250420072115958), (26, 0.0003445202892180532), (27, 0.00772478524595499)], Temporal MSE: [(0, 1.4855721701678704e-06), (1, 8.741358215047512e-07), (2, 1.0117469173565041e-06), (3, 1.527742597318138e-06), (4, 2.0271884295652853e-06), (5, 2.6431275728100445e-06), (6, 4.331604941398837e-06), (7, 4.2056303755089175e-06), (8, 3.0607322969444795e-06), (9, 3.1422930533153703e-06), (10, 3.7857478218938923e-06), (11, 6.625536116189323e-06), (12, 7.757888852211181e-06), (13, 1.4942831512598787e-05), (14, 2.6330360924475826e-05), (15, 2.9816912501701154e-05), (16, 2.2621345124207437e-05), (17, 4.092710878467187e-05), (18, 3.6178247682983056e-05), (19, 2.890716859837994e-05), (20, 2.9152164643164724e-05), (21, 8.199199510272592e-05), (22, 9.372393105877563e-05), (23, 0.00013020150072406977), (24, 9.598559699952602e-05), (25, 0.00016869108367245644), (26, 7.157919026212767e-05), (27, 6.330900941975415e-05)]
+
+ 86%|████████▌ | 129/150 [18:45<03:26,  9.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 120, Spatial MSE: [(0, 1.1396597301427391e-06), (1, 1.4386438124347478e-05), (2, 4.121904567000456e-05), (3, 8.418641300522722e-06), (4, 8.561227332393173e-06), (5, 1.592768239788711e-05), (6, 3.315458889119327e-05), (7, 2.7052885343437083e-05), (8, 3.2829299016157165e-05), (9, 7.366787031060085e-05), (10, 9.180632332572713e-05), (11, 0.00010942813969450071), (12, 8.641352178528905e-05), (13, 0.00030923416488803923), (14, 0.0007149804732762277), (15, 0.0008112246287055314), (16, 0.0180174820125103), (17, 0.00014913907216396183), (18, 0.000413535803090781), (19, 0.001866331323981285), (20, 0.0023016645573079586), (21, 0.0002652455004863441), (22, 0.000315653916914016), (23, 0.00039156831917352974), (24, 0.0010087239788845181), (25, 0.0030646591912955046), (26, 0.005744023248553276), (27, 0.008946811780333519)], Temporal MSE: [(0, 1.66169172644004e-06), (1, 1.2317461823840858e-06), (2, 1.3011634791837423e-06), (3, 2.8153910989203723e-06), (4, 4.4021794565196615e-06), (5, 6.132547241577413e-06), (6, 1.459365103073651e-05), (7, 1.7186537661473267e-05), (8, 1.4299219401436858e-05), (9, 1.6272295397357084e-05), (10, 2.0208424757583998e-05), (11, 2.6728394004749134e-05), (12, 2.7435207812231965e-05), (13, 4.254762097843923e-05), (14, 7.575839845230803e-05), (15, 5.316666283761151e-05), (16, 4.617650847649202e-05), (17, 5.120491186971776e-05), (18, 6.495494744740427e-05), (19, 8.160567813320085e-05), (20, 0.00012241420336067677), (21, 0.00014067765732761472), (22, 0.00014755406300537288), (23, 0.00021542565082199872), (24, 0.000403427256969735), (25, 0.0008794101304374635), (26, 0.00041801147744990885), (27, 0.00015127005463000387)]
+
+ 87%|████████▋ | 130/150 [18:52<03:02,  9.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 114, Spatial MSE: [(0, 1.0136014907402568e-06), (1, 1.8430235286359675e-05), (2, 2.7647241950035095e-05), (3, 7.970505976118147e-06), (4, 3.7355478070821846e-06), (5, 6.109656624175841e-06), (6, 5.1241422625025734e-06), (7, 1.190298189612804e-05), (8, 8.04661431175191e-06), (9, 5.595822130999295e-06), (10, 6.120978468970861e-06), (11, 8.427943612332456e-06), (12, 1.742546919558663e-05), (13, 0.00035740085877478123), (14, 0.00027309314464218915), (15, 0.00022129714488983154), (16, 0.009199496358633041), (17, 5.287100429995917e-05), (18, 5.538644109037705e-05), (19, 0.0006552384584210813), (20, 0.00039346551056951284), (21, 8.949245238909498e-05), (22, 0.00016930066340137273), (23, 0.00013191011385060847), (24, 0.00020662444876506925), (25, 0.0004089224967174232), (26, 0.0004301544395275414), (27, 0.0071374233812093735)], Temporal MSE: [(0, 1.643911218707217e-06), (1, 1.0093266382682486e-06), (2, 1.0948218687190092e-06), (3, 1.5480466117878677e-06), (4, 2.1249156816338655e-06), (5, 2.642944764374988e-06), (6, 6.274788120208541e-06), (7, 4.55744020655402e-06), (8, 3.2982484299282078e-06), (9, 3.822486178250983e-06), (10, 4.338397502579028e-06), (11, 1.4212213500286452e-05), (12, 1.3589352420240175e-05), (13, 1.619119029783178e-05), (14, 1.759517181199044e-05), (15, 3.192673102603294e-05), (16, 2.1422434656415135e-05), (17, 2.4852733986335807e-05), (18, 3.2047209970187396e-05), (19, 3.3004311262629926e-05), (20, 4.477827314985916e-05), (21, 3.7536450690822676e-05), (22, 4.967921267962083e-05), (23, 8.72549062478356e-05), (24, 8.940271800383925e-05), (25, 0.00018993031699210405), (26, 7.816764991730452e-05), (27, 6.291424506343901e-05)]
+
+ 87%|████████▋ | 131/150 [18:59<02:43,  8.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 108, Spatial MSE: [(0, 1.2650532426050631e-06), (1, 1.766282002790831e-05), (2, 7.275700045283884e-05), (3, 1.4127313079370651e-05), (4, 1.934874126163777e-05), (5, 3.684268085635267e-05), (6, 8.034290658542886e-05), (7, 0.00019505491945892572), (8, 0.0002477675734553486), (9, 0.0002988192718476057), (10, 0.00023205217439681292), (11, 0.00027393485652282834), (12, 0.00030502595473080873), (13, 0.0033646977972239256), (14, 0.008988752029836178), (15, 0.00574347423389554), (16, 0.1445244550704956), (17, 0.000831421697512269), (18, 0.0008506578160449862), (19, 0.007989222183823586), (20, 0.006247663404792547), (21, 0.0012716222554445267), (22, 0.0011883273255079985), (23, 0.0020212011877447367), (24, 0.0034484360367059708), (25, 0.007631500717252493), (26, 0.014616690576076508), (27, 0.006337208207696676)], Temporal MSE: [(0, 2.1669445686711697e-06), (1, 1.281484514947806e-06), (2, 3.3693809200485703e-06), (3, 1.3984843462822028e-05), (4, 1.6960848370217718e-05), (5, 2.1515697881113738e-05), (6, 0.00011718468158505857), (7, 8.637206337880343e-05), (8, 5.322389188222587e-05), (9, 7.336046837735921e-05), (10, 0.00010495379683561623), (11, 0.00018069264478981495), (12, 0.00019993788737338036), (13, 0.0002840977977029979), (14, 0.0004216741363052279), (15, 0.0003825074527412653), (16, 0.00034544579102657735), (17, 0.000308839458739385), (18, 0.00033658271422609687), (19, 0.0004197983362246305), (20, 0.0006026508053764701), (21, 0.0006667275447398424), (22, 0.0006588111282326281), (23, 0.0009787866147235036), (24, 0.0019322020234540105), (25, 0.003124516224488616), (26, 0.0014436196070164442), (27, 0.0002637832658365369)]
+
+ 88%|████████▊ | 132/150 [19:07<02:30,  8.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 102, Spatial MSE: [(0, 1.1810176374638104e-06), (1, 2.216012217104435e-05), (2, 3.574147558538243e-05), (3, 1.018017155729467e-05), (4, 4.626830559573136e-06), (5, 7.121083399397321e-06), (6, 6.398799996532034e-06), (7, 1.2346578841970768e-05), (8, 8.837890163704287e-06), (9, 8.540174349036533e-06), (10, 1.3089422282064334e-05), (11, 4.208149402984418e-05), (12, 7.850300607969984e-05), (13, 0.0016848858213052154), (14, 0.0015520325396209955), (15, 0.0005080624250695109), (16, 0.010747057385742664), (17, 5.766444519395009e-05), (18, 7.621332042617723e-05), (19, 0.0006864935858175159), (20, 0.0002528577751945704), (21, 0.00035071256570518017), (22, 0.0003125391958747059), (23, 0.00020318345923442394), (24, 0.0003029745421372354), (25, 0.0005991677171550691), (26, 0.0005619449657388031), (27, 0.013649577274918556)], Temporal MSE: [(0, 1.7818118749346468e-06), (1, 1.1345996426825877e-06), (2, 1.3209327107688296e-06), (3, 1.7769971236702986e-06), (4, 2.680847956071375e-06), (5, 3.4719842005870305e-06), (6, 6.8340405050548725e-06), (7, 5.903163128095912e-06), (8, 4.662266746890964e-06), (9, 5.082013103674399e-06), (10, 8.405428161495365e-06), (11, 0.0001018271068460308), (12, 4.035675010527484e-05), (13, 4.896593964076601e-05), (14, 2.689984285098035e-05), (15, 4.658419493352994e-05), (16, 2.547968688304536e-05), (17, 2.117014810210094e-05), (18, 4.598976011038758e-05), (19, 3.3229523978661746e-05), (20, 2.6016881747636944e-05), (21, 5.292063360684551e-05), (22, 7.061713404254988e-05), (23, 0.0001457894832128659), (24, 0.00014816387556493282), (25, 0.00027014632360078394), (26, 0.00011958184040850028), (27, 8.11423160484992e-05)]
+
+ 89%|████████▊ | 133/150 [19:15<02:20,  8.26s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 96, Spatial MSE: [(0, 1.3436080052997568e-06), (1, 1.932543091243133e-05), (2, 6.535067950608209e-05), (3, 1.14532231236808e-05), (4, 1.345439250144409e-05), (5, 2.0650393707910553e-05), (6, 4.774794797413051e-05), (7, 7.951114821480587e-05), (8, 0.0001073590901796706), (9, 0.0001568793086335063), (10, 0.00014924549032002687), (11, 0.00019946516840718687), (12, 0.00020363455405458808), (13, 0.0022321429569274187), (14, 0.003051296342164278), (15, 0.001564054167829454), (16, 0.03455784171819687), (17, 0.0002963397419080138), (18, 0.0007532502640970051), (19, 0.003155290614813566), (20, 0.003661778289824724), (21, 0.0010114092146977782), (22, 0.0006714658229611814), (23, 0.0009099920280277729), (24, 0.0017008173745125532), (25, 0.00455467589199543), (26, 0.009933593682944775), (27, 0.014238040894269943)], Temporal MSE: [(0, 2.1403700429800665e-06), (1, 1.4080649179959437e-06), (2, 2.0479999420786044e-06), (3, 6.6606944528757595e-06), (4, 6.612652214244008e-06), (5, 8.112235263979528e-06), (6, 5.8267592976335436e-05), (7, 4.152082692598924e-05), (8, 2.7584743293118663e-05), (9, 3.542297054082155e-05), (10, 4.615166471921839e-05), (11, 0.00015184131916612387), (12, 8.813311433186755e-05), (13, 0.00012548403174150735), (14, 0.00015119757154025137), (15, 0.00015593887655995786), (16, 0.00011842466483358294), (17, 0.00010188325541093946), (18, 0.00013814907288178802), (19, 0.00018436308891978115), (20, 0.00025419227313250303), (21, 0.00028343938174657524), (22, 0.0002551624784246087), (23, 0.00041111340397037566), (24, 0.0008019928354769945), (25, 0.0012339113745838404), (26, 0.0005656535504385829), (27, 0.0001837102317949757)]
+
+ 89%|████████▉ | 134/150 [19:23<02:07,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 90, Spatial MSE: [(0, 1.2582967201524298e-06), (1, 2.269793549203314e-05), (2, 5.178634455660358e-05), (3, 7.777059181535151e-06), (4, 6.86576777297887e-06), (5, 1.0897335414483678e-05), (6, 2.2763884771848097e-05), (7, 4.069034184794873e-05), (8, 4.733471359941177e-05), (9, 6.728293374180794e-05), (10, 5.8546320360619575e-05), (11, 7.960163202369586e-05), (12, 7.801281753927469e-05), (13, 0.0013399043818935752), (14, 0.004278174601495266), (15, 0.0029664807952940464), (16, 0.0851292684674263), (17, 0.0001937297056429088), (18, 0.0002433560002828017), (19, 0.0069662160240113735), (20, 0.006402066443115473), (21, 0.0003096134460065514), (22, 0.0002589089854154736), (23, 0.0004310800868552178), (24, 0.0007614856003783643), (25, 0.0017409655265510082), (26, 0.003615106688812375), (27, 0.0031049924436956644)], Temporal MSE: [(0, 2.2458450530393748e-06), (1, 1.4255016367314965e-06), (2, 1.7852555629360722e-06), (3, 3.280500550317811e-06), (4, 3.673171931950492e-06), (5, 3.7032718864793424e-06), (6, 3.199308048351668e-05), (7, 1.802361111913342e-05), (8, 1.3150270206097048e-05), (9, 1.5752682884340174e-05), (10, 1.8156293663196266e-05), (11, 4.7877787437755615e-05), (12, 5.4944724979577586e-05), (13, 6.378697435138747e-05), (14, 8.833006722852588e-05), (15, 0.00010323206515749916), (16, 7.080624345690012e-05), (17, 6.057983773644082e-05), (18, 8.760143828112632e-05), (19, 0.00010884151561185718), (20, 0.00012795052316505462), (21, 0.00018391842604614794), (22, 0.00016482268983963877), (23, 0.00020919321104884148), (24, 0.00034025710192508996), (25, 0.0005010204040445387), (26, 0.0002111998328473419), (27, 9.191528079099953e-05)]
+
+ 90%|█████████ | 135/150 [19:30<01:57,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 84, Spatial MSE: [(0, 1.2925785313200322e-06), (1, 2.2804528271080926e-05), (2, 5.073045031167567e-05), (3, 7.946604455355555e-06), (4, 7.0213809522101656e-06), (5, 1.03431120805908e-05), (6, 2.351919283682946e-05), (7, 3.927005673176609e-05), (8, 4.466028622118756e-05), (9, 6.499870505649596e-05), (10, 5.0960119551746175e-05), (11, 6.643900997005403e-05), (12, 7.472702418453991e-05), (13, 0.001231638714671135), (14, 0.00410874979570508), (15, 0.0030100757721811533), (16, 0.08519445359706879), (17, 0.00019226723816245794), (18, 0.00023940432583913207), (19, 0.0069013419561088085), (20, 0.006375262979418039), (21, 0.0002771462604869157), (22, 0.0002516206877771765), (23, 0.00040561301284469664), (24, 0.0007744891918264329), (25, 0.0017258045263588428), (26, 0.0035328054800629616), (27, 0.003275552298873663)], Temporal MSE: [(0, 2.381169451837195e-06), (1, 1.5286130974345724e-06), (2, 1.7700497210171307e-06), (3, 3.794624035435845e-06), (4, 3.7003278521297034e-06), (5, 3.997962267021649e-06), (6, 3.30451330228243e-05), (7, 1.8537128198659047e-05), (8, 1.4271684449340682e-05), (9, 1.57830145326443e-05), (10, 1.6346397387678735e-05), (11, 4.2736344767035916e-05), (12, 5.312151915859431e-05), (13, 6.187401595525444e-05), (14, 8.752306894166395e-05), (15, 9.936614515027031e-05), (16, 6.801902782171965e-05), (17, 5.938939284533262e-05), (18, 8.554364467272535e-05), (19, 0.00010581949027255177), (20, 0.00011778219777625054), (21, 0.00018240536155644804), (22, 0.00016154843615368009), (23, 0.00020651501836255193), (24, 0.00032034210744313896), (25, 0.0005011974135413766), (26, 0.00020751352712977678), (27, 0.00010485677921678871)]
+
+ 91%|█████████ | 136/150 [19:38<01:50,  7.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 78, Spatial MSE: [(0, 1.304081933994894e-06), (1, 2.4916431357269175e-05), (2, 6.0267186199780554e-05), (3, 8.330555829161312e-06), (4, 8.262159099103883e-06), (5, 1.1082589480793104e-05), (6, 2.428850893920753e-05), (7, 4.365662607597187e-05), (8, 5.3139076044317335e-05), (9, 6.202140502864495e-05), (10, 5.5968212109291926e-05), (11, 6.545060023199767e-05), (12, 5.3595405915984884e-05), (13, 0.0004312307864893228), (14, 0.0016728807240724564), (15, 0.0011784478556364775), (16, 0.030430061742663383), (17, 0.00013353658141568303), (18, 0.00014673997065983713), (19, 0.0024461918510496616), (20, 0.00236117304302752), (21, 0.0001993055484490469), (22, 0.00019560751388780773), (23, 0.00038111984031274915), (24, 0.0007251657661981881), (25, 0.0015880519058555365), (26, 0.00356552773155272), (27, 0.0032418444752693176)], Temporal MSE: [(0, 2.429837195450091e-06), (1, 1.580796606504009e-06), (2, 1.9247618183726445e-06), (3, 3.734030542545952e-06), (4, 3.929929789592279e-06), (5, 3.892688255291432e-06), (6, 3.7994355807313696e-05), (7, 2.0087238226551563e-05), (8, 1.5190065823844634e-05), (9, 1.656719177844934e-05), (10, 1.5111291759239975e-05), (11, 2.7421894628787413e-05), (12, 2.7063439119956456e-05), (13, 3.847909829346463e-05), (14, 6.569972902070731e-05), (15, 7.496940088458359e-05), (16, 5.37518244527746e-05), (17, 4.689365596277639e-05), (18, 5.568449341808446e-05), (19, 7.222974090836942e-05), (20, 9.676584886619821e-05), (21, 0.0001113595426431857), (22, 0.00010452268907101825), (23, 0.0001476164470659569), (24, 0.000304525630781427), (25, 0.0004621901025529951), (26, 0.00022213769261725247), (27, 0.00013042862701695412)]
+
+ 91%|█████████▏| 137/150 [19:46<01:43,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 72, Spatial MSE: [(0, 1.3922116295361775e-06), (1, 2.642780964379199e-05), (2, 5.865828279638663e-05), (3, 8.649944902572315e-06), (4, 8.43405541672837e-06), (5, 1.2180942576378584e-05), (6, 2.646891152835451e-05), (7, 4.536862616077997e-05), (8, 5.4713422287022695e-05), (9, 7.256644312292337e-05), (10, 5.54258658667095e-05), (11, 6.002291775075719e-05), (12, 4.627520320354961e-05), (13, 0.0003146853414364159), (14, 0.001092920545488596), (15, 0.0006853346712887287), (16, 0.018080919981002808), (17, 0.00014145387103781104), (18, 0.00013035375741310418), (19, 0.0014788159169256687), (20, 0.0014282967895269394), (21, 0.0001673786755418405), (22, 0.00019313223310746253), (23, 0.00040302349952980876), (24, 0.0007796759600751102), (25, 0.0018914601532742381), (26, 0.004094525706022978), (27, 0.0037463088519871235)], Temporal MSE: [(0, 2.56637417805905e-06), (1, 1.6858128901731106e-06), (2, 1.9898516256944276e-06), (3, 4.130752586206654e-06), (4, 4.257019099895842e-06), (5, 4.339433417044347e-06), (6, 4.491879735724069e-05), (7, 2.0988105461583473e-05), (8, 1.7111113265855238e-05), (9, 1.8456485122442245e-05), (10, 1.714933205221314e-05), (11, 2.6914029149338603e-05), (12, 2.4241571736638434e-05), (13, 3.619015478761867e-05), (14, 6.0928108723601326e-05), (15, 6.661367660854012e-05), (16, 5.478927050717175e-05), (17, 4.504396201809868e-05), (18, 4.9443762691225857e-05), (19, 6.796250090701506e-05), (20, 9.400756971444935e-05), (21, 9.698605572339147e-05), (22, 9.992927516577765e-05), (23, 0.00014763367653358728), (24, 0.00032384044607169926), (25, 0.0005119770648889244), (26, 0.00024350668536499143), (27, 0.00016857165610417724)]
+
+ 92%|█████████▏| 138/150 [19:54<01:36,  8.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 66, Spatial MSE: [(0, 1.6366120689781383e-06), (1, 2.9859807909815572e-05), (2, 6.675162876490504e-05), (3, 9.8419213827583e-06), (4, 9.393796972290147e-06), (5, 1.47317941809888e-05), (6, 3.2765488867880777e-05), (7, 5.18516571901273e-05), (8, 6.971381662879139e-05), (9, 8.74808756634593e-05), (10, 7.274442759808153e-05), (11, 8.068582246778533e-05), (12, 5.9442416386445984e-05), (13, 0.00011113142682006583), (14, 0.00023469101870432496), (15, 0.0001864642254076898), (16, 0.00016046315431594849), (17, 0.0001613620697753504), (18, 0.00011450804595369846), (19, 0.0001417903695255518), (20, 0.00020386699179653078), (21, 0.00020696410501841456), (22, 0.0001954340550582856), (23, 0.0004749250947497785), (24, 0.0009954818524420261), (25, 0.0023733931593596935), (26, 0.004978969227522612), (27, 0.004110206384211779)], Temporal MSE: [(0, 3.2137681955646258e-06), (1, 2.080953663607943e-06), (2, 2.4919222596508916e-06), (3, 4.89174044560059e-06), (4, 5.05561547470279e-06), (5, 5.377340130507946e-06), (6, 5.107872129883617e-05), (7, 2.4343353288713843e-05), (8, 2.2386037016985938e-05), (9, 2.2574873582925647e-05), (10, 2.0743473214679398e-05), (11, 3.0389610401471145e-05), (12, 2.3055457859300077e-05), (13, 3.58383622369729e-05), (14, 6.832044164184481e-05), (15, 6.482714525191113e-05), (16, 5.979554771329276e-05), (17, 4.898217594018206e-05), (18, 4.862173955189064e-05), (19, 6.316522194538265e-05), (20, 9.476147533860058e-05), (21, 9.203187801176682e-05), (22, 9.727636643219739e-05), (23, 0.00016551624867133796), (24, 0.00038465546094812453), (25, 0.0006570116383954883), (26, 0.00032218656269833446), (27, 0.00024001466226764023)]
+
+ 93%|█████████▎| 139/150 [20:03<01:30,  8.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 60, Spatial MSE: [(0, 1.7649283563514473e-06), (1, 3.226807530154474e-05), (2, 7.363548502326012e-05), (3, 1.0779896001622546e-05), (4, 1.062257888406748e-05), (5, 1.5777173757669516e-05), (6, 3.528831803123467e-05), (7, 5.811801747768186e-05), (8, 7.436783198500052e-05), (9, 9.041701559908688e-05), (10, 8.336109021911398e-05), (11, 8.422119572060183e-05), (12, 6.637980550294742e-05), (13, 0.0011979324044659734), (14, 0.003959380090236664), (15, 0.0030989053193479776), (16, 0.08595240116119385), (17, 0.00025423828628845513), (18, 0.00029666416230611503), (19, 0.0069733429700136185), (20, 0.00646949140354991), (21, 0.0003627533442340791), (22, 0.00033462056308053434), (23, 0.0006146207451820374), (24, 0.001238279859535396), (25, 0.002822614274919033), (26, 0.005894451402127743), (27, 0.004387051332741976)], Temporal MSE: [(0, 3.38085646944819e-06), (1, 2.242181835754309e-06), (2, 2.795936779875774e-06), (3, 4.9538998609932605e-06), (4, 5.3009530347480904e-06), (5, 5.8536265896691475e-06), (6, 6.000666689942591e-05), (7, 2.7261936338618398e-05), (8, 2.578680869191885e-05), (9, 2.4883082005544566e-05), (10, 2.4583638150943443e-05), (11, 3.713659316417761e-05), (12, 5.0623315473785624e-05), (13, 6.835678505012766e-05), (14, 0.00011692463885992765), (15, 0.00012841443822253495), (16, 9.164439688902348e-05), (17, 7.39219321985729e-05), (18, 0.00010243233555229381), (19, 0.0001316246052738279), (20, 0.00015046050248201936), (21, 0.00022350819199346006), (22, 0.00019657511438708752), (23, 0.0002581436128821224), (24, 0.00048332501319237053), (25, 0.0008049191674217582), (26, 0.0003852692316286266), (27, 0.000333326926920563)]
+
+ 93%|█████████▎| 140/150 [20:12<01:23,  8.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 54, Spatial MSE: [(0, 1.8950493085867492e-06), (1, 3.5369830584386364e-05), (2, 8.18313128547743e-05), (3, 1.1229289157199673e-05), (4, 1.1318634278723039e-05), (5, 1.7679929442238063e-05), (6, 4.010802513221279e-05), (7, 6.558290624525398e-05), (8, 8.37383049656637e-05), (9, 0.00010311641381122172), (10, 8.848686411511153e-05), (11, 9.354381472803652e-05), (12, 5.8388130128150806e-05), (13, 0.0003725482092704624), (14, 0.0010261198040097952), (15, 0.0006400256534107029), (16, 0.012791471555829048), (17, 0.0001975347986444831), (18, 0.0001621523406356573), (19, 0.0010967161506414413), (20, 0.001136074773967266), (21, 0.0002767881378531456), (22, 0.00025639505474828184), (23, 0.0005972871440462768), (24, 0.0012572361156344414), (25, 0.002861477667465806), (26, 0.005775282625108957), (27, 0.00466321362182498)], Temporal MSE: [(0, 3.6180576898914296e-06), (1, 2.434080215607537e-06), (2, 3.059437858610181e-06), (3, 5.28912823938299e-06), (4, 5.864130343979923e-06), (5, 6.3589372985006776e-06), (6, 7.425130752380937e-05), (7, 3.07184636767488e-05), (8, 2.9239230570965447e-05), (9, 2.8696353183477186e-05), (10, 2.623778345878236e-05), (11, 3.372477658558637e-05), (12, 2.979880628117826e-05), (13, 4.701827492681332e-05), (14, 8.708078530617058e-05), (15, 9.308713197242469e-05), (16, 7.717271364526823e-05), (17, 5.9513793530641124e-05), (18, 6.243625830393285e-05), (19, 8.685720240464434e-05), (20, 0.00012997753219678998), (21, 0.00013174753985367715), (22, 0.00013229716569185257), (23, 0.00021785436547361314), (24, 0.0004911742289550602), (25, 0.0007908093393780291), (26, 0.00043032795656472445), (27, 0.000497798144351691)]
+
+ 94%|█████████▍| 141/150 [20:20<01:13,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 48, Spatial MSE: [(0, 2.102902044498478e-06), (1, 3.8851059798616916e-05), (2, 8.808782877167687e-05), (3, 1.2272938874957617e-05), (4, 1.286658789467765e-05), (5, 1.900805364130065e-05), (6, 4.2005507566500455e-05), (7, 6.573014979949221e-05), (8, 9.59277676884085e-05), (9, 0.00010886161180678755), (10, 0.0001061888542608358), (11, 9.33137271204032e-05), (12, 6.588103133253753e-05), (13, 0.00015608513785991818), (14, 0.00038439148920588195), (15, 0.0004910306888632476), (16, 0.006284370552748442), (17, 0.00021003563597332686), (18, 0.00015834842633921653), (19, 0.0006563516799360514), (20, 0.0007344577461481094), (21, 0.0002918574900832027), (22, 0.0002775102620944381), (23, 0.0006574223516508937), (24, 0.0013533529127016664), (25, 0.0028745969757437706), (26, 0.005952178966253996), (27, 0.004748532548546791)], Temporal MSE: [(0, 3.980660949309822e-06), (1, 2.6716872980614426e-06), (2, 3.4138470255129505e-06), (3, 5.434308150142897e-06), (4, 6.34266643828596e-06), (5, 7.047354301903397e-06), (6, 8.340372005477548e-05), (7, 3.210468639736064e-05), (8, 3.325272700749338e-05), (9, 2.9545552024501376e-05), (10, 2.8324539016466588e-05), (11, 3.8693306123605e-05), (12, 2.8898348318762146e-05), (13, 4.3258958612568676e-05), (14, 8.782804070506245e-05), (15, 9.377925016451627e-05), (16, 7.965052645886317e-05), (17, 6.245476834010333e-05), (18, 6.300143286352977e-05), (19, 9.230728028342128e-05), (20, 0.00014058832312002778), (21, 0.00014166912296786904), (22, 0.00013542796659749), (23, 0.0002232711121905595), (24, 0.0005274547729641199), (25, 0.0008313588914461434), (26, 0.0005279599572531879), (27, 0.0006725118728354573)]
+
+ 95%|█████████▍| 142/150 [20:27<01:03,  7.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 42, Spatial MSE: [(0, 2.5069168714253465e-06), (1, 4.6342043788172305e-05), (2, 0.00010725574247771874), (3, 1.4565674973709974e-05), (4, 1.4687971997773275e-05), (5, 2.3918553779367357e-05), (6, 5.5456570407841355e-05), (7, 8.336445171153173e-05), (8, 0.00012154615251347423), (9, 0.00013570403098128736), (10, 0.00013587650028057396), (11, 0.00012816455273423344), (12, 7.278306293301284e-05), (13, 0.0001990470482269302), (14, 0.0005771907162852585), (15, 0.0005157413543201983), (16, 0.00636311573907733), (17, 0.0002667741500772536), (18, 0.00020454292825888842), (19, 0.0007076510810293257), (20, 0.0007799679879099131), (21, 0.0003999526088591665), (22, 0.0003500969905871898), (23, 0.0008569614146836102), (24, 0.001915876753628254), (25, 0.003986624535173178), (26, 0.008287699893116951), (27, 0.005692408885806799)], Temporal MSE: [(0, 4.8378096835222095e-06), (1, 3.2524305879633175e-06), (2, 4.301959052099846e-06), (3, 6.636943908233661e-06), (4, 7.756119885016233e-06), (5, 8.734727089176886e-06), (6, 0.00010025562369264662), (7, 4.228188481647521e-05), (8, 4.53331449534744e-05), (9, 3.861661025439389e-05), (10, 3.843025479000062e-05), (11, 4.6311644837260246e-05), (12, 3.2488078431924805e-05), (13, 5.4924497817410156e-05), (14, 0.00010772548557724804), (15, 0.00012570945546030998), (16, 9.960771421901882e-05), (17, 7.646504673175514e-05), (18, 7.818699668860063e-05), (19, 0.00011044762504752725), (20, 0.00017198597197420895), (21, 0.00018137562437914312), (22, 0.00016915338346734643), (23, 0.0002927137247752398), (24, 0.000676208408549428), (25, 0.0011824689572677016), (26, 0.0007496765465475619), (27, 0.0012702611275017262)]
+
+ 95%|█████████▌| 143/150 [20:36<00:58,  8.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 36, Spatial MSE: [(0, 2.98915256280452e-06), (1, 5.334486559149809e-05), (2, 0.00012098043225705624), (3, 1.6601190509391017e-05), (4, 1.721791340969503e-05), (5, 2.874905476346612e-05), (6, 6.557953747687861e-05), (7, 9.901537850964814e-05), (8, 0.000152685766806826), (9, 0.00016045873053371906), (10, 0.00017181376460939646), (11, 0.00015745074779260904), (12, 9.285761916544288e-05), (13, 0.0006166049279272556), (14, 0.0024390823673456907), (15, 0.0018928031204268336), (16, 0.04372674971818924), (17, 0.00037270551547408104), (18, 0.00031662662513554096), (19, 0.0035581106785684824), (20, 0.0036449115723371506), (21, 0.0005487970192916691), (22, 0.0005059854593127966), (23, 0.0011219101725146174), (24, 0.0024434870574623346), (25, 0.005337947513908148), (26, 0.010400327853858471), (27, 0.006821850314736366)], Temporal MSE: [(0, 5.563744707615115e-06), (1, 3.892061613441911e-06), (2, 5.250638423603959e-06), (3, 7.6161595643498e-06), (4, 9.167865755443927e-06), (5, 1.0774473594210576e-05), (6, 0.00013737536210101098), (7, 4.795314816874452e-05), (8, 5.57736711925827e-05), (9, 4.7160745452856645e-05), (10, 4.724503378383815e-05), (11, 5.8446461480343714e-05), (12, 5.162664456292987e-05), (13, 7.96042222646065e-05), (14, 0.00014202851161826402), (15, 0.0001635843946132809), (16, 0.00012984918430447578), (17, 0.00010340379230910912), (18, 0.00011484185233712196), (19, 0.00015811616322025657), (20, 0.0002346689871046692), (21, 0.00026782340137287974), (22, 0.00024603470228612423), (23, 0.00040393011295236647), (24, 0.0008739062468521297), (25, 0.0015276450430974364), (26, 0.00103377818595618), (27, 0.002211523475125432)]
+
+ 96%|█████████▌| 144/150 [20:44<00:48,  8.06s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 30, Spatial MSE: [(0, 3.4777906421368243e-06), (1, 6.102793850004673e-05), (2, 0.00013219613174442202), (3, 1.9123755919281393e-05), (4, 1.9520924979588017e-05), (5, 3.21449879265856e-05), (6, 7.646358426427469e-05), (7, 0.00011623497266555205), (8, 0.00017577818653080612), (9, 0.00017775413289200515), (10, 0.00020306419173721224), (11, 0.00016976955521386117), (12, 9.168901306111366e-05), (13, 0.00024580731405876577), (14, 0.000713289191480726), (15, 0.0006535983411595225), (16, 0.006498354487121105), (17, 0.00038459335337392986), (18, 0.00029438972705975175), (19, 0.000851270102430135), (20, 0.0010000363690778613), (21, 0.0005589798092842102), (22, 0.0005273977876640856), (23, 0.00130668084602803), (24, 0.0029435891192406416), (25, 0.006069396622478962), (26, 0.012197582982480526), (27, 0.00812661275267601)], Temporal MSE: [(0, 6.745324299117783e-06), (1, 4.637929578166222e-06), (2, 6.317813131317962e-06), (3, 8.560396963730454e-06), (4, 1.0870800906559452e-05), (5, 1.2873529158241581e-05), (6, 0.00016031779523473233), (7, 5.680658796336502e-05), (8, 6.995534204179421e-05), (9, 5.494518336490728e-05), (10, 5.65547525184229e-05), (11, 6.453119567595422e-05), (12, 4.389483365230262e-05), (13, 7.526687841163948e-05), (14, 0.00014823385572526604), (15, 0.00017034915799740702), (16, 0.00013178910012356937), (17, 0.00010319844295736402), (18, 0.00010534463217481971), (19, 0.00015180278569459915), (20, 0.00023283045447897166), (21, 0.0002580700966063887), (22, 0.00024635979207232594), (23, 0.00043674203334376216), (24, 0.0010256641544401646), (25, 0.0018093438120558858), (26, 0.0014620078727602959), (27, 0.0041577257215976715)]
+
+ 97%|█████████▋| 145/150 [20:51<00:39,  7.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 24, Spatial MSE: [(0, 4.419955530465813e-06), (1, 7.155165076255798e-05), (2, 0.00015343826089520007), (3, 2.287578536197543e-05), (4, 2.392987698840443e-05), (5, 3.888769060722552e-05), (6, 9.597544703865424e-05), (7, 0.00012180021440144628), (8, 0.0001916393230203539), (9, 0.00022423452173825353), (10, 0.0002525346935726702), (11, 0.00022656412329524755), (12, 0.0001144598500104621), (13, 0.0005176979466341436), (14, 0.0019138094503432512), (15, 0.001823096303269267), (16, 0.03763323277235031), (17, 0.00048798328498378396), (18, 0.0004165932768955827), (19, 0.0033723050728440285), (20, 0.0034228030126541853), (21, 0.0006998308235779405), (22, 0.0006936552235856652), (23, 0.0016839336603879929), (24, 0.003916008397936821), (25, 0.007617761846631765), (26, 0.01493916753679514), (27, 0.009904079139232635)], Temporal MSE: [(0, 8.117956895148382e-06), (1, 5.799888640467543e-06), (2, 8.047519258980174e-06), (3, 9.398067959409673e-06), (4, 1.3728847989114001e-05), (5, 1.6047677490860224e-05), (6, 0.00019853080448228866), (7, 6.809610204072669e-05), (8, 9.277136996388435e-05), (9, 6.901157030370086e-05), (10, 7.514614117098972e-05), (11, 8.66220289026387e-05), (12, 6.330697215162218e-05), (13, 0.00010282406583428383), (14, 0.00019454193534329534), (15, 0.0002253586135338992), (16, 0.0001774472329998389), (17, 0.0001370633690385148), (18, 0.00014727328380104154), (19, 0.00021590491815004498), (20, 0.00031123007647693157), (21, 0.00034871749812737107), (22, 0.0003310552565380931), (23, 0.000581702683120966), (24, 0.001321714837104082), (25, 0.0023106976877897978), (26, 0.002295015612617135), (27, 0.008881238289177418)]
+
+ 97%|█████████▋| 146/150 [20:59<00:31,  7.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 18, Spatial MSE: [(0, 5.5675704970781226e-06), (1, 9.38562152441591e-05), (2, 0.00019234079809393734), (3, 2.88511291728355e-05), (4, 3.089194433414377e-05), (5, 4.959416401106864e-05), (6, 0.0001236213865922764), (7, 0.00018117911531589925), (8, 0.0002883311244659126), (9, 0.00027894513914361596), (10, 0.0003486868226900697), (11, 0.0002803521347232163), (12, 0.00016170187154784799), (13, 0.0015238673659041524), (14, 0.005327570252120495), (15, 0.0037195931654423475), (16, 0.08800286054611206), (17, 0.0007438730681315064), (18, 0.0006146437372080982), (19, 0.006988517474383116), (20, 0.007219883147627115), (21, 0.0010154533665627241), (22, 0.0010445347288623452), (23, 0.002342390129342675), (24, 0.005383642856031656), (25, 0.010190576314926147), (26, 0.02093653194606304), (27, 0.01516050472855568)], Temporal MSE: [(0, 1.0173576811212115e-05), (1, 7.195701073214877e-06), (2, 1.0837707122846041e-05), (3, 1.1318857104924973e-05), (4, 1.7529233446111903e-05), (5, 2.0720823158626445e-05), (6, 0.00027428794419392943), (7, 9.340470569441095e-05), (8, 0.00012663190136663616), (9, 9.097739530261606e-05), (10, 9.609761036699638e-05), (11, 0.00011822838132502511), (12, 0.00010990435112034902), (13, 0.00014634084072895348), (14, 0.00025525863748043776), (15, 0.00032215614919550717), (16, 0.00024291846784763038), (17, 0.00018198454927187413), (18, 0.00020689926168415695), (19, 0.00030863459687680006), (20, 0.0004600789397954941), (21, 0.0005153205129317939), (22, 0.00046100126928649843), (23, 0.0008037816151045263), (24, 0.0018213940784335136), (25, 0.003115598112344742), (26, 0.0038786346558481455), (27, 0.021970901638269424)]
+
+ 98%|█████████▊| 147/150 [21:07<00:23,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 12, Spatial MSE: [(0, 7.614643891429296e-06), (1, 0.0001221359270857647), (2, 0.000260494853137061), (3, 3.9238915633177385e-05), (4, 4.191553307464346e-05), (5, 6.599663174711168e-05), (6, 0.00016688827599864453), (7, 0.00021849825861863792), (8, 0.0003510703390929848), (9, 0.00035438453778624535), (10, 0.00048085025628097355), (11, 0.0003670192672871053), (12, 0.00019974139286205173), (13, 0.0017400988144800067), (14, 0.007333079352974892), (15, 0.005854529328644276), (16, 0.13875645399093628), (17, 0.0009344001300632954), (18, 0.0008301697089336812), (19, 0.01073949970304966), (20, 0.011042901314795017), (21, 0.001370636047795415), (22, 0.0014869431033730507), (23, 0.003272678004577756), (24, 0.007892822846770287), (25, 0.01428407896310091), (26, 0.030440162867307663), (27, 0.03083273395895958)], Temporal MSE: [(0, 1.3577652680396568e-05), (1, 9.858976227405947e-06), (2, 1.526423693576362e-05), (3, 1.4009684491611551e-05), (4, 2.3807191610103473e-05), (5, 2.8820428269682452e-05), (6, 0.0003559208125807345), (7, 0.00012568593956530094), (8, 0.0001835614093579352), (9, 0.00011788489064201713), (10, 0.00013750539801549166), (11, 0.00015239964704960585), (12, 0.00011953259672736749), (13, 0.0001952619495568797), (14, 0.00033580808667466044), (15, 0.00044288759818300605), (16, 0.00031317491084337234), (17, 0.0002410280576441437), (18, 0.0002804325777105987), (19, 0.000424947909777984), (20, 0.000587474147323519), (21, 0.0007099667564034462), (22, 0.000646642642095685), (23, 0.0011497564846649766), (24, 0.002625188324600458), (25, 0.004531959071755409), (26, 0.007987534627318382), (27, 0.06720331311225891)]
+
+ 99%|█████████▊| 148/150 [21:15<00:15,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 6, Spatial MSE: [(0, 1.3387289072852582e-05), (1, 0.00020335182489361614), (2, 0.0003882204182446003), (3, 6.588707765331492e-05), (4, 7.521614315919578e-05), (5, 0.00011071516200900078), (6, 0.0002774456806946546), (7, 0.0003588500840123743), (8, 0.0005884949932806194), (9, 0.0005885572754777968), (10, 0.0008341318462044001), (11, 0.0006603177171200514), (12, 0.00039426443981938064), (13, 0.0020856051705777645), (14, 0.008577955886721611), (15, 0.007011790759861469), (16, 0.15354226529598236), (17, 0.00143740966450423), (18, 0.0012596381129696965), (19, 0.012317970395088196), (20, 0.012718862853944302), (21, 0.002286101458594203), (22, 0.0026500571984797716), (23, 0.005717571824789047), (24, 0.015765981748700142), (25, 0.02759513445198536), (26, 0.0674700140953064), (27, 0.1555071622133255)], Temporal MSE: [(0, 2.2820799131295644e-05), (1, 1.6926956959650852e-05), (2, 2.807586861308664e-05), (3, 2.2463011191575788e-05), (4, 4.142538091400638e-05), (5, 5.121230788063258e-05), (6, 0.0006038323044776917), (7, 0.00021973354159854352), (8, 0.00033886273740790784), (9, 0.00020749260147567838), (10, 0.00025132112205028534), (11, 0.0003008819476235658), (12, 0.0002331254945602268), (13, 0.00031636017956770957), (14, 0.0005381587543524802), (15, 0.000717486604116857), (16, 0.0004795471322722733), (17, 0.0003701503446791321), (18, 0.00042740226490423083), (19, 0.0006540624308399856), (20, 0.0009146663942374289), (21, 0.0010386309586465359), (22, 0.0010745188919827342), (23, 0.0020026331767439842), (24, 0.004774864297360182), (25, 0.008568535558879375), (26, 0.023890944197773933), (27, 0.3465091586112976)]
+
+ 99%|█████████▉| 149/150 [21:23<00:07,  7.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 0, Spatial MSE: [(0, 8.490041363984346e-05), (1, 0.0007364902994595468), (2, 0.0023637141566723585), (3, 0.0004960780497640371), (4, 0.00041436380706727505), (5, 0.0005692444974556565), (6, 0.001063362113200128), (7, 0.0009141392074525356), (8, 0.001601437688805163), (9, 0.0017495169304311275), (10, 0.0020895313937216997), (11, 0.0018253251910209656), (12, 0.0017012456664815545), (13, 0.0029189663473516703), (14, 0.008647351525723934), (15, 0.007731038145720959), (16, 0.11743459850549698), (17, 0.0038152567576617002), (18, 0.005013167392462492), (19, 0.014924373477697372), (20, 0.015264485962688923), (21, 0.012870150618255138), (22, 0.013657000847160816), (23, 0.02432877942919731), (24, 0.08719600737094879), (25, 0.3943517804145813), (26, 0.6926858425140381), (27, 4.625109672546387)], Temporal MSE: [(0, 0.00014295615255832672), (1, 8.233504195231944e-05), (2, 0.00011641831952147186), (3, 9.485585906077176e-05), (4, 0.0002482253185007721), (5, 0.00021170552645344287), (6, 0.0014432822354137897), (7, 0.0007372249965555966), (8, 0.0011954351793974638), (9, 0.0006373667856678367), (10, 0.0010945203248411417), (11, 0.00112931861076504), (12, 0.0007712522055953741), (13, 0.0013190124882385135), (14, 0.0017963071586564183), (15, 0.0018370527541264892), (16, 0.0014943116111680865), (17, 0.0013278204714879394), (18, 0.001466164831072092), (19, 0.0024828240275382996), (20, 0.0035910196602344513), (21, 0.004417856689542532), (22, 0.004843643866479397), (23, 0.008293175138533115), (24, 0.024411119520664215), (25, 0.05637028068304062), (26, 0.16280336678028107), (27, 2.8372039794921875)]
+
+100%|██████████| 150/150 [21:31<00:00,  7.97s/it]
+100%|██████████| 150/150 [21:31<00:00,  8.61s/it]
+Saved spatial MLP MSE outputs to ./samples/opensora_plan/skip_mse/2_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora_plan/skip_mse/2_temporal_mlp_mse.pt
+
+  0%|          | 0/150 [00:00<?, ?it/s]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1726856428140309e-06), (1, 6.5867202465597074e-06), (2, 0.00016700020933058113), (3, 2.0781169951078482e-05), (4, 8.953959877544548e-06), (5, 7.04157946529449e-06), (6, 8.962761967268307e-06), (7, 5.973518454993609e-06), (8, 5.549535217141965e-06), (9, 5.869822416570969e-06), (10, 8.942232852859888e-06), (11, 1.0669928997231182e-05), (12, 1.4091850971453823e-05), (13, 2.6918862204183824e-05), (14, 0.00011822463420685381), (15, 0.00017095019575208426), (16, 0.0024068052880465984), (17, 7.036601891741157e-05), (18, 0.00011304930376354605), (19, 0.00017098004173021764), (20, 0.00012461676669772714), (21, 0.00012245144171174616), (22, 0.0001169012684840709), (23, 9.133012645179406e-05), (24, 0.00016771462105680257), (25, 0.00039875038783065975), (26, 0.0004109185538254678), (27, 0.0036916763056069613)], Temporal MSE: [(0, 1.6382546164095402e-06), (1, 1.2111614751120214e-06), (2, 1.6734885548430611e-06), (3, 1.5723920796517632e-06), (4, 1.0802729093484231e-06), (5, 1.4796827372265398e-06), (6, 7.537977921856509e-07), (7, 9.889578223010176e-07), (8, 6.590084922208916e-07), (9, 8.276557537101326e-07), (10, 1.8598520910018124e-06), (11, 1.4595224229196901e-06), (12, 1.5629578911102726e-06), (13, 4.422326583153335e-06), (14, 4.592443474393804e-06), (15, 6.105073680373607e-06), (16, 5.115095973451389e-06), (17, 9.469596989220008e-06), (18, 4.41744532508892e-06), (19, 4.865133178100223e-06), (20, 5.252077244222164e-06), (21, 5.081089966552099e-06), (22, 1.546749444969464e-05), (23, 1.0405863577034324e-05), (24, 1.8708065908867866e-05), (25, 1.6311390936607495e-05), (26, 3.2830951113282936e-06), (27, 1.1784347407228779e-05)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.1786537967850563e-08), (1, 1.4523391200782498e-07), (2, 1.4647574744230951e-06), (3, 2.6973327749146847e-07), (4, 1.1031524849158814e-07), (5, 8.65640146230362e-08), (6, 2.1272175843023433e-07), (7, 1.2648885672206234e-07), (8, 1.0968575026026883e-07), (9, 1.4119024172032368e-07), (10, 1.4884143695326202e-07), (11, 1.5339882963871787e-07), (12, 1.7906396010403114e-07), (13, 1.6331917152001552e-07), (14, 3.196241209479922e-07), (15, 5.808466312373639e-07), (16, 1.7152342479676008e-06), (17, 4.2410320588714967e-07), (18, 9.164456855614844e-07), (19, 6.000905727887584e-07), (20, 6.964609724491311e-07), (21, 9.756896588442032e-07), (22, 1.270145389753452e-06), (23, 1.0253853588437778e-06), (24, 2.9020197871432174e-06), (25, 4.834183982893592e-06), (26, 5.632310148939723e-06), (27, 2.857358231267426e-05)], Temporal MSE: [(0, 7.121972345203176e-09), (1, 9.567185266234901e-09), (2, 9.552175050941969e-09), (3, 1.1730410420796034e-08), (4, 1.520310632940891e-08), (5, 2.6591246538032465e-08), (6, 1.365119928209424e-08), (7, 1.776026969935174e-08), (8, 1.7099145210863753e-08), (9, 2.309562141533661e-08), (10, 3.108959134578981e-08), (11, 3.0531282391166314e-08), (12, 2.792228137593611e-08), (13, 4.642653195219282e-08), (14, 4.29719584360555e-08), (15, 7.645952848633897e-08), (16, 2.8920103645191375e-08), (17, 5.628963606341131e-08), (18, 5.6232600798011845e-08), (19, 4.966302924458432e-08), (20, 9.23349645631788e-08), (21, 7.261685652792949e-08), (22, 1.2299383911340556e-07), (23, 1.3364119411107822e-07), (24, 2.740542583978822e-07), (25, 3.5554251098801615e-07), (26, 1.349667115846387e-07), (27, 1.3468589088461158e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 9.197723329634755e-07), (1, 5.840467565576546e-06), (2, 0.00015693041495978832), (3, 1.528881875856314e-05), (4, 6.9066286414454225e-06), (5, 5.619117473543156e-06), (6, 5.661911472998327e-06), (7, 6.26833934802562e-06), (8, 5.702848739019828e-06), (9, 6.679465514025651e-06), (10, 1.0516221664147452e-05), (11, 1.295245692745084e-05), (12, 2.0101966583752073e-05), (13, 2.5105760869337246e-05), (14, 0.00011404001998016611), (15, 0.00021464837482199073), (16, 0.0011092490749433637), (17, 4.698574775829911e-05), (18, 7.645016739843413e-05), (19, 0.00013229947944637388), (20, 0.00014800879580434412), (21, 8.221601456170902e-05), (22, 7.497697515645996e-05), (23, 8.152532245730981e-05), (24, 0.00020954648789484054), (25, 0.0004701993602793664), (26, 0.00021855720842722803), (27, 0.0026602898724377155)], Temporal MSE: [(0, 1.314335122515331e-06), (1, 1.0266530807712115e-06), (2, 1.1462467455203296e-06), (3, 9.837425523073762e-07), (4, 9.045995739143109e-07), (5, 1.31942476855329e-06), (6, 7.063562748044205e-07), (7, 1.2841190937251667e-06), (8, 6.429328891499608e-07), (9, 9.910470453178277e-07), (10, 2.8802135147998342e-06), (11, 2.2424833332479466e-06), (12, 2.144361815226148e-06), (13, 5.76639922655886e-06), (14, 6.054753157513915e-06), (15, 8.334372068929952e-06), (16, 3.2420880415884312e-06), (17, 7.290621397260111e-06), (18, 3.6897586142004e-06), (19, 4.221176368446322e-06), (20, 5.408910965343239e-06), (21, 4.625491328624776e-06), (22, 1.1022992111975327e-05), (23, 8.379670362046454e-06), (24, 2.0590556232491508e-05), (25, 1.411509765603114e-05), (26, 3.4459731068636756e-06), (27, 7.680562703171745e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 1.0956839879838753e-08), (1, 1.228640371664369e-07), (2, 1.2451067732399679e-06), (3, 2.3158234796483157e-07), (4, 1.0269557293440812e-07), (5, 8.080341018512627e-08), (6, 1.9092694003575161e-07), (7, 1.1920693765432588e-07), (8, 1.0328022170824624e-07), (9, 1.322018619021037e-07), (10, 1.3929155784353497e-07), (11, 1.4299720874078048e-07), (12, 1.673979852512275e-07), (13, 1.6916459344429313e-07), (14, 5.381771188694984e-07), (15, 8.16239264622709e-07), (16, 4.506076038524043e-06), (17, 4.3709255237445177e-07), (18, 8.992854532152705e-07), (19, 8.252378052020504e-07), (20, 8.1235964444204e-07), (21, 9.818131729844026e-07), (22, 1.2890494645034778e-06), (23, 9.914922429743456e-07), (24, 2.7942037377215456e-06), (25, 4.711139808932785e-06), (26, 5.481291282194434e-06), (27, 2.7284080715617165e-05)], Temporal MSE: [(0, 6.6335759107971626e-09), (1, 8.538806994806691e-09), (2, 8.667064399503488e-09), (3, 1.0765141666979616e-08), (4, 1.4176682938682461e-08), (5, 2.4763139094829967e-08), (6, 1.2781912417381136e-08), (7, 1.6632148103212785e-08), (8, 1.5989579438269175e-08), (9, 2.125560882859645e-08), (10, 2.8990800871042666e-08), (11, 2.849207625388317e-08), (12, 2.6159968413708157e-08), (13, 4.6975245027169876e-08), (14, 4.18379286770687e-08), (15, 7.566617910015339e-08), (16, 2.8945834174010088e-08), (17, 5.782607814808216e-08), (18, 5.709995676284052e-08), (19, 5.139188274938533e-08), (20, 9.49798035776439e-08), (21, 7.53039586243176e-08), (22, 1.2969557872111181e-07), (23, 1.3868636017377867e-07), (24, 2.707583348637854e-07), (25, 3.489401478873333e-07), (26, 1.3085926298117556e-07), (27, 1.2901980994683981e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.3414946806733496e-06), (1, 3.665639724204084e-06), (2, 9.030422370415181e-05), (3, 1.5702364180469885e-05), (4, 6.701447091472801e-06), (5, 4.548498054646188e-06), (6, 5.424277787824394e-06), (7, 4.1493035496387165e-06), (8, 4.9696086534822825e-06), (9, 6.505686087621143e-06), (10, 8.16019564808812e-06), (11, 8.576959771744441e-06), (12, 1.2174939911346883e-05), (13, 1.4212769201549236e-05), (14, 6.373244832502678e-05), (15, 0.0001045569806592539), (16, 0.000838708714582026), (17, 2.4623353965580463e-05), (18, 3.415848550503142e-05), (19, 8.080220140982419e-05), (20, 5.066146331955679e-05), (21, 2.844816117431037e-05), (22, 4.368800364318304e-05), (23, 3.896791531587951e-05), (24, 9.72303532762453e-05), (25, 0.00018166666268371046), (26, 0.00019708792387973517), (27, 0.003975830972194672)], Temporal MSE: [(0, 1.6737906207708875e-06), (1, 9.957075235433877e-07), (2, 1.738444098009495e-06), (3, 1.4102820387051906e-06), (4, 1.138277411882882e-06), (5, 1.431374016647169e-06), (6, 7.836877671252296e-07), (7, 6.343211680359673e-07), (8, 1.1396127774787601e-06), (9, 1.0863880106626311e-06), (10, 1.4669404890810256e-06), (11, 1.2758744105667574e-06), (12, 1.7542936348036164e-06), (13, 2.961488917208044e-06), (14, 4.590126991388388e-06), (15, 3.163727569699404e-06), (16, 2.1663026927853934e-06), (17, 3.0860887818562333e-06), (18, 2.4388891688431613e-06), (19, 3.515498747219681e-06), (20, 4.49783101430512e-06), (21, 3.453509634709917e-06), (22, 4.1985208554251585e-06), (23, 5.780664650956169e-06), (24, 1.3870375369151589e-05), (25, 1.398095628246665e-05), (26, 4.1516068449709564e-06), (27, 3.840940280497307e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.1544432965138185e-08), (1, 1.416910180296327e-07), (2, 1.4312411167338723e-06), (3, 2.617039172037039e-07), (4, 1.1329448312835666e-07), (5, 8.704474652176941e-08), (6, 2.153186073883262e-07), (7, 1.2674323102146445e-07), (8, 1.1022419244000048e-07), (9, 1.4417334170957474e-07), (10, 1.5178160595041845e-07), (11, 1.581194197797231e-07), (12, 1.8664009360236378e-07), (13, 1.9417531404997135e-07), (14, 6.919073598510295e-07), (15, 1.5450063983735163e-06), (16, 1.2795775546692312e-05), (17, 5.01567910760059e-07), (18, 9.867537755781086e-07), (19, 1.394512651131663e-06), (20, 1.0964291732307174e-06), (21, 1.172887323264149e-06), (22, 1.5437124147865688e-06), (23, 1.115972850129765e-06), (24, 3.0965175028541125e-06), (25, 5.115934072819073e-06), (26, 5.896587026654743e-06), (27, 2.9368329705903307e-05)], Temporal MSE: [(0, 6.946108133121243e-09), (1, 9.262689282252268e-09), (2, 9.399648170926866e-09), (3, 1.1545292721848455e-08), (4, 1.5198825309425956e-08), (5, 2.649595032266916e-08), (6, 1.3769357209980626e-08), (7, 1.7868860169301115e-08), (8, 1.7274707886372198e-08), (9, 2.3216392364133753e-08), (10, 3.154645611402884e-08), (11, 3.0703244391361295e-08), (12, 2.9203588880477582e-08), (13, 5.3645500486254605e-08), (14, 5.145456682953409e-08), (15, 8.806096474245351e-08), (16, 3.702779949321666e-08), (17, 7.970206894469811e-08), (18, 6.698088128587187e-08), (19, 6.502430238697343e-08), (20, 1.0962236274281167e-07), (21, 8.748838808969595e-08), (22, 1.5230013161726674e-07), (23, 1.6453796547466482e-07), (24, 3.049639190066955e-07), (25, 3.7513166262215236e-07), (26, 1.4070337783778086e-07), (27, 1.3982624125219445e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 3.478064343198639e-07), (1, 9.264036293643585e-07), (2, 2.039624632743653e-05), (3, 4.07053039452876e-06), (4, 1.8141410009775427e-06), (5, 1.5083536482052295e-06), (6, 1.778067712621123e-06), (7, 1.5746174995001638e-06), (8, 2.245305495307548e-06), (9, 3.5125067370245233e-06), (10, 5.26192434335826e-06), (11, 5.665656317432877e-06), (12, 9.38134326133877e-06), (13, 9.607717402104754e-06), (14, 2.7376017897040583e-05), (15, 5.75255871808622e-05), (16, 8.004191477084532e-05), (17, 1.4307216588349547e-05), (18, 2.2835814888821915e-05), (19, 2.1612442651530728e-05), (20, 2.68286148639163e-05), (21, 1.2972206604899839e-05), (22, 1.9462400814518332e-05), (23, 2.096845128107816e-05), (24, 4.900375279248692e-05), (25, 9.366575977765024e-05), (26, 4.2462561395950615e-05), (27, 0.0006778538809157908)], Temporal MSE: [(0, 4.1253605331803556e-07), (1, 2.2400512023068586e-07), (2, 2.8396542006703385e-07), (3, 2.9218972485978156e-07), (4, 3.198056788278336e-07), (5, 4.693129653787764e-07), (6, 3.189320807450713e-07), (7, 4.344649653376109e-07), (8, 2.9301784820745524e-07), (9, 4.757607428018673e-07), (10, 1.1381526974219014e-06), (11, 8.69277414494718e-07), (12, 1.0476858278707368e-06), (13, 2.1560213099292014e-06), (14, 3.155551212330465e-06), (15, 2.889320967369713e-06), (16, 1.4307352103060111e-06), (17, 1.8720983234743471e-06), (18, 1.515500116511248e-06), (19, 1.4261261185311014e-06), (20, 1.884246103145415e-06), (21, 1.2936211533087771e-06), (22, 2.0966560896340525e-06), (23, 2.076608097922872e-06), (24, 6.62272850604495e-06), (25, 4.463756795303198e-06), (26, 1.4752333754586289e-06), (27, 2.1118062250025105e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 1.0804674488440469e-08), (1, 1.2106951885471062e-07), (2, 1.2204950508021284e-06), (3, 2.2917514286291407e-07), (4, 1.0197252464649864e-07), (5, 8.030250597812483e-08), (6, 1.9035404363876296e-07), (7, 1.1834120527964842e-07), (8, 1.0358699853441067e-07), (9, 1.326610430396613e-07), (10, 1.3929984277183394e-07), (11, 1.4376972501395358e-07), (12, 1.670403264597553e-07), (13, 1.5834649502721732e-07), (14, 4.577040897402185e-07), (15, 9.742365136844455e-07), (16, 7.645696314284578e-06), (17, 4.4315180502962903e-07), (18, 8.72470536705805e-07), (19, 9.772063549462473e-07), (20, 8.754119562581764e-07), (21, 1.0200750466538011e-06), (22, 1.3096195061734761e-06), (23, 9.892293064694968e-07), (24, 2.7941314328927547e-06), (25, 4.669079771701945e-06), (26, 5.416210569819668e-06), (27, 2.734820736804977e-05)], Temporal MSE: [(0, 6.544367270322482e-09), (1, 8.47371062207003e-09), (2, 8.678354923574716e-09), (3, 1.0749218404271232e-08), (4, 1.4182550245323e-08), (5, 2.463600701219093e-08), (6, 1.2745325683738429e-08), (7, 1.650143133247184e-08), (8, 1.604830046630923e-08), (9, 2.1180829534728218e-08), (10, 2.9063759399150513e-08), (11, 2.824532607803576e-08), (12, 2.6115611007071493e-08), (13, 4.414846088707236e-08), (14, 4.1991402355279206e-08), (15, 7.71875789951082e-08), (16, 3.1252248788860015e-08), (17, 6.540896180240452e-08), (18, 5.824417570465812e-08), (19, 5.3876284766829485e-08), (20, 9.560495328742036e-08), (21, 7.505091303983136e-08), (22, 1.3132303422480618e-07), (23, 1.3934226217315882e-07), (24, 2.6866214852816483e-07), (25, 3.473377319096471e-07), (26, 1.2962556183993001e-07), (27, 1.28876394001054e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 3.211089506294229e-06), (1, 3.058520087506622e-05), (2, 0.0007389670354314148), (3, 8.177091513061896e-05), (4, 3.100059984717518e-05), (5, 2.1689073037123308e-05), (6, 2.9079013984301127e-05), (7, 1.7220307199750096e-05), (8, 1.5428904589498416e-05), (9, 1.5860334315220825e-05), (10, 2.2104250092525035e-05), (11, 2.428807783871889e-05), (12, 3.570631088223308e-05), (13, 6.335874059004709e-05), (14, 0.0003392922808416188), (15, 0.0005360713694244623), (16, 0.0034834863618016243), (17, 0.00017423923418391496), (18, 0.000281542947050184), (19, 0.00043912066030316055), (20, 0.0005030081956647336), (21, 0.00021104518964421004), (22, 0.000273509620456025), (23, 0.00026263779727742076), (24, 0.0004898090846836567), (25, 0.0011213128454983234), (26, 0.0011929672909900546), (27, 0.015212167985737324)], Temporal MSE: [(0, 4.371160230220994e-06), (1, 4.091268692718586e-06), (2, 6.039466370566515e-06), (3, 5.637748927256325e-06), (4, 4.529747457127087e-06), (5, 4.705842911789659e-06), (6, 2.2205688310350524e-06), (7, 3.217344783479348e-06), (8, 2.0937998215231346e-06), (9, 2.3500804218201665e-06), (10, 4.774626631842693e-06), (11, 4.3156483116035815e-06), (12, 3.97719713873812e-06), (13, 1.4486254258372355e-05), (14, 1.0833839951374102e-05), (15, 1.7220783774973825e-05), (16, 7.121141152310884e-06), (17, 2.3265884010470472e-05), (18, 1.0853402272914536e-05), (19, 1.408110256306827e-05), (20, 2.1182189811952412e-05), (21, 1.3009457688895054e-05), (22, 2.6168690965278074e-05), (23, 2.340856553928461e-05), (24, 5.609758954960853e-05), (25, 5.039373354520649e-05), (26, 1.0477937394171022e-05), (27, 3.905668199877255e-05)]
+
+  1%|          | 1/150 [01:17<3:11:43, 77.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 1.2065861199062056e-08), (1, 1.5370997630270722e-07), (2, 1.5377544286820921e-06), (3, 2.7630341037365724e-07), (4, 1.1218304507565335e-07), (5, 8.78818013916316e-08), (6, 2.2004745403592096e-07), (7, 1.2910750513128733e-07), (8, 1.1201294114471239e-07), (9, 1.48158633805906e-07), (10, 1.5714559253865445e-07), (11, 1.642704177129417e-07), (12, 1.9612073742791836e-07), (13, 2.677774375570152e-07), (14, 1.6542561525056954e-06), (15, 2.540975629017339e-06), (16, 2.3609631170984358e-05), (17, 5.498453674590564e-07), (18, 1.0383025710325455e-06), (19, 2.3532077193522127e-06), (20, 1.807856733648805e-06), (21, 1.1927260175070842e-06), (22, 1.7181897646878497e-06), (23, 1.2230702850501984e-06), (24, 3.2805571663629962e-06), (25, 5.709907327400288e-06), (26, 5.994550519972108e-06), (27, 3.0749364668736234e-05)], Temporal MSE: [(0, 7.1000854084957155e-09), (1, 9.580326754132784e-09), (2, 9.726471184023922e-09), (3, 1.1834269564303668e-08), (4, 1.555405937381238e-08), (5, 2.7056671569880564e-08), (6, 1.4260026937051862e-08), (7, 1.827266338239042e-08), (8, 1.7686684117279583e-08), (9, 2.428073919702456e-08), (10, 3.270306336844442e-08), (11, 3.2642354597101075e-08), (12, 3.1006710088377076e-08), (13, 7.274876878682335e-08), (14, 6.425571541512909e-08), (15, 1.0008074013967416e-07), (16, 4.2523677024064455e-08), (17, 8.701277920408756e-08), (18, 7.789888911702292e-08), (19, 7.686603709089468e-08), (20, 1.3456474334816448e-07), (21, 1.1399617250162919e-07), (22, 1.8007769142514007e-07), (23, 2.052718173217727e-07), (24, 3.528579100020579e-07), (25, 4.050404811550834e-07), (26, 1.5203379177819443e-07), (27, 1.4747452325991617e-07)]
+
+  1%|▏         | 2/150 [01:27<1:33:46, 38.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 3.7679599245166173e-06), (1, 2.9669858122360893e-05), (2, 0.0008803302189335227), (3, 9.778434468898922e-05), (4, 3.949013989767991e-05), (5, 2.602523090899922e-05), (6, 4.009767508250661e-05), (7, 2.0809095076401718e-05), (8, 1.80143670149846e-05), (9, 1.8723372704698704e-05), (10, 2.622511783556547e-05), (11, 2.7542453608475626e-05), (12, 4.272283331374638e-05), (13, 6.173778820084408e-05), (14, 0.0002847700088750571), (15, 0.0005444914568215609), (16, 0.004433609079569578), (17, 0.00024351541651412845), (18, 0.0003353866341058165), (19, 0.000490166072268039), (20, 0.0006355985533446074), (21, 0.0003888006031047553), (22, 0.0003210170252714306), (23, 0.0003021106240339577), (24, 0.0006398645346052945), (25, 0.0016673569334670901), (26, 0.001437049824744463), (27, 0.02059020660817623)], Temporal MSE: [(0, 5.155536655365722e-06), (1, 4.805925982509507e-06), (2, 7.3681294452399015e-06), (3, 6.2802168940834235e-06), (4, 5.415930445451522e-06), (5, 6.135830972198164e-06), (6, 2.5879189706756733e-06), (7, 3.693099642987363e-06), (8, 2.6898517262452515e-06), (9, 2.755718924163375e-06), (10, 5.372377472667722e-06), (11, 5.0618514251254965e-06), (12, 4.924914264847757e-06), (13, 1.2188621440145653e-05), (14, 1.138844800152583e-05), (15, 1.7812173609854653e-05), (16, 1.0787030987557955e-05), (17, 2.7992662580800243e-05), (18, 1.3098113413434476e-05), (19, 1.57599297381239e-05), (20, 2.5120903956121765e-05), (21, 1.3467910321196541e-05), (22, 3.775611548917368e-05), (23, 2.5660789106041193e-05), (24, 6.464203761424869e-05), (25, 6.44325336907059e-05), (26, 1.354939831799129e-05), (27, 4.707784319180064e-05)]
+
+  2%|▏         | 3/150 [01:36<1:00:33, 24.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 1.1110223852028867e-08), (1, 1.2654578540605144e-07), (2, 1.2222742498124717e-06), (3, 2.2993856418906944e-07), (4, 1.0372760783639023e-07), (5, 7.990373518396154e-08), (6, 1.9585317545534053e-07), (7, 1.195170682422031e-07), (8, 1.0293376817571698e-07), (9, 1.3212205374202313e-07), (10, 1.3909895812957984e-07), (11, 1.4385258850779792e-07), (12, 1.696057836397813e-07), (13, 2.069756988021254e-07), (14, 6.993710144342913e-07), (15, 7.522771170442866e-07), (16, 5.702401267626556e-06), (17, 4.557055888199102e-07), (18, 9.724740266392473e-07), (19, 8.084762725957262e-07), (20, 8.013748242774454e-07), (21, 1.0613066478981636e-06), (22, 1.2853366797571653e-06), (23, 9.991465503844665e-07), (24, 2.8278241188672837e-06), (25, 4.600650299835252e-06), (26, 5.440258973976597e-06), (27, 2.769760976661928e-05)], Temporal MSE: [(0, 6.610850977750715e-09), (1, 8.394519745991147e-09), (2, 8.490956382445347e-09), (3, 1.0570233577311683e-08), (4, 1.3955304467572205e-08), (5, 2.457022141300058e-08), (6, 1.2873734078766574e-08), (7, 1.660390758218e-08), (8, 1.610693800557783e-08), (9, 2.1447796427764843e-08), (10, 2.9625907060903955e-08), (11, 2.9010946533958304e-08), (12, 2.7481990017008684e-08), (13, 5.600245600589915e-08), (14, 4.422431487682843e-08), (15, 8.2742410256742e-08), (16, 3.4847438712404255e-08), (17, 6.282696318749004e-08), (18, 5.905268807282482e-08), (19, 5.277523840163667e-08), (20, 9.523225230623211e-08), (21, 7.57152278652029e-08), (22, 1.3125344366926583e-07), (23, 1.370930249322555e-07), (24, 2.7247887146586436e-07), (25, 3.6429253214009805e-07), (26, 1.3353604799704044e-07), (27, 1.3402730303369026e-07)]
+
+  3%|▎         | 4/150 [01:44<43:58, 18.07s/it]  Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 870, Spatial MSE: [(0, 1.2112413060094696e-05), (1, 0.00010247105092275888), (2, 0.0025817034766077995), (3, 0.00028306193416938186), (4, 0.00011231815733481199), (5, 7.569404988316819e-05), (6, 9.423504525329918e-05), (7, 8.005806739674881e-05), (8, 5.66907983738929e-05), (9, 5.362523006624542e-05), (10, 7.061317592160776e-05), (11, 7.938451017253101e-05), (12, 0.0001162889675470069), (13, 0.00022692052880302072), (14, 0.0010858974419534206), (15, 0.0018292289460077882), (16, 0.015887929126620293), (17, 0.0006656214245595038), (18, 0.0009887375636026263), (19, 0.001586967846378684), (20, 0.0018013084772974253), (21, 0.0007905058446340263), (22, 0.0009753606864251196), (23, 0.0009250342845916748), (24, 0.0018158694729208946), (25, 0.005189336836338043), (26, 0.00451380480080843), (27, 0.05765639245510101)], Temporal MSE: [(0, 1.6858111848705448e-05), (1, 1.5469942809431814e-05), (2, 2.203991789428983e-05), (3, 1.8810869732988067e-05), (4, 1.5655301467631944e-05), (5, 1.7053996998583898e-05), (6, 8.80233073985437e-06), (7, 9.946176760422532e-06), (8, 8.201812306651846e-06), (9, 9.192115612677298e-06), (10, 1.726539085211698e-05), (11, 1.5732359315734357e-05), (12, 1.6273279470624402e-05), (13, 5.489822797244415e-05), (14, 4.2277330067008734e-05), (15, 6.372637290041894e-05), (16, 3.399813795113005e-05), (17, 0.00010039234621217474), (18, 4.410148903843947e-05), (19, 6.166889943415299e-05), (20, 8.998800331028178e-05), (21, 5.878062074771151e-05), (22, 9.571805276209489e-05), (23, 8.822858944768086e-05), (24, 0.00021380852558650076), (25, 0.0002521173155400902), (26, 4.098980571143329e-05), (27, 0.0001432641438441351)]
+
+  3%|▎         | 5/150 [01:52<35:00, 14.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 864, Spatial MSE: [(0, 1.5481258742511272e-05), (1, 0.0001388719683745876), (2, 0.0035933053586632013), (3, 0.00038559953100048006), (4, 0.00015984107449185103), (5, 0.00010487245162948966), (6, 0.00016422223416157067), (7, 8.244672790169716e-05), (8, 7.28923623682931e-05), (9, 8.390092989429832e-05), (10, 0.00011600772268138826), (11, 0.00011886144056916237), (12, 0.00019080244237557054), (13, 0.0003832096990663558), (14, 0.0014688202645629644), (15, 0.0021938281133770943), (16, 0.01362800132483244), (17, 0.0008982060826383531), (18, 0.0013077874900773168), (19, 0.0017900149105116725), (20, 0.002565489150583744), (21, 0.0010510663269087672), (22, 0.0011559322010725737), (23, 0.0011632508831098676), (24, 0.0022816506680101156), (25, 0.006451890338212252), (26, 0.005418206565082073), (27, 0.08841340988874435)], Temporal MSE: [(0, 2.099522862408776e-05), (1, 1.973073995031882e-05), (2, 2.7332869649399072e-05), (3, 2.5253439162042923e-05), (4, 2.1931617084192112e-05), (5, 2.525825948396232e-05), (6, 1.0860983820748515e-05), (7, 1.7827995179686695e-05), (8, 1.1657171853585169e-05), (9, 1.2596836313605309e-05), (10, 2.858239895431325e-05), (11, 2.6281180907972157e-05), (12, 2.8360991564113647e-05), (13, 8.871580212144181e-05), (14, 6.162947101984173e-05), (15, 9.174220031127334e-05), (16, 4.559151057037525e-05), (17, 0.00014019387890584767), (18, 6.784070137655362e-05), (19, 9.804023284232244e-05), (20, 0.0001849649997893721), (21, 7.434110739268363e-05), (22, 0.00011501569679239765), (23, 9.793133358471096e-05), (24, 0.0002364080719416961), (25, 0.0002969542401842773), (26, 5.190669253352098e-05), (27, 0.00015043145685922354)]
+
+  4%|▍         | 6/150 [02:03<31:53, 13.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 858, Spatial MSE: [(0, 3.9120142901083454e-05), (1, 0.0003487241338007152), (2, 0.008669581264257431), (3, 0.0009107751538977027), (4, 0.00037856961716897786), (5, 0.00025074544828385115), (6, 0.00029451987938955426), (7, 0.000253522302955389), (8, 0.0001816341682570055), (9, 0.00019109844288323075), (10, 0.00025624464615248144), (11, 0.00026388547848910093), (12, 0.00043720280518755317), (13, 0.0007298513664864004), (14, 0.002369567984715104), (15, 0.004585954826325178), (16, 0.022952895611524582), (17, 0.001986313844099641), (18, 0.002744511002674699), (19, 0.00351141020655632), (20, 0.005494615063071251), (21, 0.002205283846706152), (22, 0.0028204580303281546), (23, 0.0029952460899949074), (24, 0.006243542302399874), (25, 0.019132738932967186), (26, 0.014599683694541454), (27, 0.20687955617904663)], Temporal MSE: [(0, 5.5983913625823334e-05), (1, 4.976069976692088e-05), (2, 6.851510988781229e-05), (3, 6.0067213780712336e-05), (4, 5.219009472057223e-05), (5, 6.130037945695221e-05), (6, 2.828306060109753e-05), (7, 3.951511098421179e-05), (8, 2.8669888706644997e-05), (9, 3.226144690415822e-05), (10, 6.861861038487405e-05), (11, 6.361783016473055e-05), (12, 7.09909072611481e-05), (13, 0.00017785628733690828), (14, 0.00013720503193326294), (15, 0.00023083754058461636), (16, 0.00011377372720744461), (17, 0.00037341206916607916), (18, 0.0001685459428699687), (19, 0.00025554304011166096), (20, 0.00046594830928370357), (21, 0.00021133347763679922), (22, 0.00024342811957467347), (23, 0.0002317730977665633), (24, 0.0006017667474225163), (25, 0.0010147180873900652), (26, 0.00014553983055520803), (27, 0.00044619510299526155)]
+
+  5%|▍         | 7/150 [02:12<28:11, 11.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 852, Spatial MSE: [(0, 5.12951942255313e-07), (1, 3.1864021821093047e-06), (2, 5.300612247083336e-05), (3, 8.76625290402444e-06), (4, 2.941574166470673e-06), (5, 2.875221980502829e-06), (6, 6.585582013940439e-06), (7, 8.103873369691428e-06), (8, 8.003392395039555e-06), (9, 2.185452831326984e-05), (10, 2.557259358582087e-05), (11, 3.79126395273488e-05), (12, 4.4773438276024535e-05), (13, 5.955096276011318e-05), (14, 0.00012293251347728074), (15, 0.00025765440659597516), (16, 0.00036941873258911073), (17, 9.183524525724351e-05), (18, 0.00011005254782503471), (19, 0.00014447723515331745), (20, 0.00012781366240233183), (21, 9.472997044213116e-05), (22, 0.00014478567754849792), (23, 0.0001965057454071939), (24, 0.00041538261575624347), (25, 0.0005033051711507142), (26, 0.000237944012042135), (27, 0.0011352002620697021)], Temporal MSE: [(0, 6.053803645045264e-07), (1, 3.908729127033439e-07), (2, 5.601092993856582e-07), (3, 6.099911047385831e-07), (4, 5.642413611894881e-07), (5, 9.65473304859188e-07), (6, 7.438499665113341e-07), (7, 7.93746039562393e-07), (8, 8.949647849476605e-07), (9, 1.8101912928614183e-06), (10, 4.642695330403512e-06), (11, 5.000254077458521e-06), (12, 7.2423426900058985e-06), (13, 1.3921251593274064e-05), (14, 2.8336999093880877e-05), (15, 1.92870447790483e-05), (16, 1.5531983081018552e-05), (17, 2.2078809706727043e-05), (18, 1.8585091311251745e-05), (19, 2.5574099709047005e-05), (20, 4.807458390132524e-05), (21, 1.8133981939172372e-05), (22, 1.4706395631947089e-05), (23, 2.2914184228284284e-05), (24, 8.056018123170361e-05), (25, 5.2563475037459284e-05), (26, 9.792778655537404e-06), (27, 2.3506574962084414e-06)]
+
+  5%|▌         | 8/150 [02:21<25:54, 10.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 846, Spatial MSE: [(0, 2.200408516728203e-06), (1, 3.1583953386871144e-05), (2, 0.0006212922744452953), (3, 8.188653009710833e-05), (4, 2.5836177883320488e-05), (5, 1.926175900734961e-05), (6, 2.8093594664824195e-05), (7, 3.402500806259923e-05), (8, 2.2703141439706087e-05), (9, 2.9932458346593194e-05), (10, 4.007541065220721e-05), (11, 5.092098217573948e-05), (12, 6.266336276894435e-05), (13, 9.81063349172473e-05), (14, 0.0002652915718499571), (15, 0.0005533373332582414), (16, 0.0021122326143085957), (17, 0.0001641045673750341), (18, 0.0002303210349055007), (19, 0.0003332982596475631), (20, 0.000357578246621415), (21, 0.00017689826199784875), (22, 0.00026653733220882714), (23, 0.0003446554474066943), (24, 0.0007328616920858622), (25, 0.0016440199688076973), (26, 0.0011346300598233938), (27, 0.010068457573652267)], Temporal MSE: [(0, 3.3417938993807184e-06), (1, 3.3314070151391206e-06), (2, 4.5710953600064386e-06), (3, 4.033302047901088e-06), (4, 3.0896655971446307e-06), (5, 4.448812433111016e-06), (6, 2.4877551823010435e-06), (7, 2.9720774818997597e-06), (8, 2.1970774923829595e-06), (9, 3.900811407220317e-06), (10, 9.339796633867081e-06), (11, 8.88887279870687e-06), (12, 1.1796376384154428e-05), (13, 2.3762424461892806e-05), (14, 3.7397039704956114e-05), (15, 3.2434378226753324e-05), (16, 2.3907065042294562e-05), (17, 4.872567660640925e-05), (18, 3.514443596941419e-05), (19, 4.878270556218922e-05), (20, 0.00010775044211186469), (21, 4.200199691695161e-05), (22, 2.9909631848568097e-05), (23, 4.0827984776115045e-05), (24, 0.00011779673513956368), (25, 0.00016875876463018358), (26, 1.934108695422765e-05), (27, 2.1689980712835677e-05)]
+
+  6%|▌         | 9/150 [02:29<23:38, 10.06s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 840, Spatial MSE: [(0, 8.392804375034757e-06), (1, 6.50295551167801e-05), (2, 0.0019168367143720388), (3, 0.00029565009754151106), (4, 6.753339403076097e-05), (5, 3.468071736278944e-05), (6, 6.581347406608984e-05), (7, 3.544739956851117e-05), (8, 2.9350752811296843e-05), (9, 3.291671600891277e-05), (10, 2.9894410545239225e-05), (11, 4.8999569116858765e-05), (12, 5.588468047790229e-05), (13, 0.0005816972116008401), (14, 0.001392240752466023), (15, 0.0005700259353034198), (16, 0.02023281157016754), (17, 0.00018121747416444123), (18, 0.0006139671895653009), (19, 0.0011607742635533214), (20, 0.000327203975757584), (21, 0.0008068719180300832), (22, 0.0011866454733535647), (23, 0.0006372494390234351), (24, 0.001253671944141388), (25, 0.0030057879630476236), (26, 0.0045333183370530605), (27, 0.05382569506764412)], Temporal MSE: [(0, 1.100440294976579e-05), (1, 7.600228855153546e-06), (2, 1.0500711141503416e-05), (3, 9.769128155312501e-06), (4, 8.254777640104294e-06), (5, 9.278630386688747e-06), (6, 4.782629275723593e-06), (7, 5.330574367690133e-06), (8, 3.665923486551037e-06), (9, 4.947478373651393e-06), (10, 7.95998585090274e-06), (11, 1.5541676475550048e-05), (12, 1.8041018847725354e-05), (13, 6.814178777858615e-05), (14, 2.4521070372429676e-05), (15, 2.8885575375170447e-05), (16, 3.300976095488295e-05), (17, 5.2669674914795905e-05), (18, 2.6074183551827446e-05), (19, 2.9356662707868963e-05), (20, 3.978566383011639e-05), (21, 4.842037378693931e-05), (22, 0.0001545566919958219), (23, 0.0002038862876361236), (24, 0.0002840279776137322), (25, 0.00015121516480576247), (26, 7.680038834223524e-05), (27, 2.0891460735583678e-05)]
+
+  7%|▋         | 10/150 [02:37<21:40,  9.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 834, Spatial MSE: [(0, 9.524853339826223e-06), (1, 0.00019368236826267093), (2, 0.0030847699381411076), (3, 0.00023581927234772593), (4, 0.00015168012760113925), (5, 0.00010290453064953908), (6, 6.071828829590231e-05), (7, 0.00013514554302673787), (8, 8.414523472310975e-05), (9, 8.61209409777075e-05), (10, 0.00012113425327697769), (11, 0.00013784754264634103), (12, 0.00026201261789537966), (13, 0.0008649698575027287), (14, 0.0026836248580366373), (15, 0.002331146039068699), (16, 0.028973102569580078), (17, 0.0007267248583957553), (18, 0.0008938859100453556), (19, 0.0023248810321092606), (20, 0.0016322109149768949), (21, 0.0012590873520821333), (22, 0.0018725446425378323), (23, 0.0015779058448970318), (24, 0.003773484379053116), (25, 0.007478912826627493), (26, 0.00643846346065402), (27, 0.030319079756736755)], Temporal MSE: [(0, 1.0842407391464803e-05), (1, 1.9702203644555993e-05), (2, 1.841728953877464e-05), (3, 1.6582398529862985e-05), (4, 2.1454374291351996e-05), (5, 2.1379988538683392e-05), (6, 1.3550621588365175e-05), (7, 1.8535763956606388e-05), (8, 1.0529224709898699e-05), (9, 1.8109125448972918e-05), (10, 5.506998786586337e-05), (11, 5.846315252711065e-05), (12, 3.806960739893839e-05), (13, 0.0001433544239262119), (14, 8.59460633364506e-05), (15, 0.00014208653010427952), (16, 8.152426744345576e-05), (17, 0.0002200284943683073), (18, 0.00010261942952638492), (19, 0.00015448403428308666), (20, 0.00038543305709026754), (21, 0.00015562069893348962), (22, 0.00030946405604481697), (23, 0.0003566392988432199), (24, 0.0005766883841715753), (25, 0.0004205882432870567), (26, 0.0001463795779272914), (27, 0.00020628130005206913)]
+
+  7%|▋         | 11/150 [02:45<20:36,  8.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 828, Spatial MSE: [(0, 1.3124134056852199e-05), (1, 9.450729703530669e-05), (2, 0.002742923330515623), (3, 0.00042990551446564496), (4, 9.981677430914715e-05), (5, 5.2590945415431634e-05), (6, 8.547608013032004e-05), (7, 5.691907063010149e-05), (8, 4.396480289869942e-05), (9, 5.230035458225757e-05), (10, 4.786888894159347e-05), (11, 8.944346336647868e-05), (12, 0.00013093110464978963), (13, 0.0020645635668188334), (14, 0.0051138135604560375), (15, 0.0019347266061231494), (16, 0.06671717762947083), (17, 0.00039826155989430845), (18, 0.0014891016762703657), (19, 0.0037729463074356318), (20, 0.0011042022379115224), (21, 0.002107268664985895), (22, 0.00311927474103868), (23, 0.001487398985773325), (24, 0.0027208358515053988), (25, 0.00711022038012743), (26, 0.007552319671958685), (27, 0.1030774861574173)], Temporal MSE: [(0, 1.811315814848058e-05), (1, 1.13942805910483e-05), (2, 1.669392077019438e-05), (3, 1.5636771422578022e-05), (4, 1.3057049727649428e-05), (5, 1.526660707895644e-05), (6, 7.6145611274114344e-06), (7, 7.5465745794645045e-06), (8, 6.513040261779679e-06), (9, 8.20210152596701e-06), (10, 1.1949067811656278e-05), (11, 3.9241116610355675e-05), (12, 5.0566977733979e-05), (13, 0.0002525509044062346), (14, 7.754756370559335e-05), (15, 7.402966730296612e-05), (16, 8.261641778517514e-05), (17, 0.00012263415555935353), (18, 8.636901475256309e-05), (19, 8.035068458411843e-05), (20, 0.00011026510765077546), (21, 0.00014557140821125358), (22, 0.000397226627683267), (23, 0.0005914741195738316), (24, 0.000728258048184216), (25, 0.00028464835486374795), (26, 0.00016768764180596918), (27, 4.9082693294622004e-05)]
+
+  8%|▊         | 12/150 [02:53<19:46,  8.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 822, Spatial MSE: [(0, 1.2679667634074576e-05), (1, 0.0003172967699356377), (2, 0.003950370475649834), (3, 0.0003173617587890476), (4, 0.00019186885037925094), (5, 0.0001839296892285347), (6, 0.00017356926400680095), (7, 0.0004247870238032192), (8, 0.00030250096460804343), (9, 0.0006038728752173483), (10, 0.0008148656925186515), (11, 0.0010277983965352178), (12, 0.001602384028956294), (13, 0.0033716578036546707), (14, 0.00790623389184475), (15, 0.007583709433674812), (16, 0.07331635802984238), (17, 0.003200944745913148), (18, 0.004643091931939125), (19, 0.0070718503557145596), (20, 0.009885796345770359), (21, 0.003980645909905434), (22, 0.005740517750382423), (23, 0.004640615079551935), (24, 0.010133706033229828), (25, 0.04784911125898361), (26, 0.017671961337327957), (27, 0.06292302906513214)], Temporal MSE: [(0, 1.747809801599942e-05), (1, 3.155182275804691e-05), (2, 3.687536809593439e-05), (3, 2.9705672204727307e-05), (4, 2.932632742158603e-05), (5, 5.654880442307331e-05), (6, 3.2119223760673776e-05), (7, 4.992851972929202e-05), (8, 5.299061012919992e-05), (9, 7.819932216079906e-05), (10, 0.00019509118283167481), (11, 0.0001867602695710957), (12, 0.0002804220712278038), (13, 0.0006663479143753648), (14, 0.0007463719812221825), (15, 0.0006228679558262229), (16, 0.00048785394756123424), (17, 0.0009510417003184557), (18, 0.0006908021750859916), (19, 0.0007163943373598158), (20, 0.0012482554884627461), (21, 0.0009277751087211072), (22, 0.0007330423686653376), (23, 0.0010549452854320407), (24, 0.00234077381901443), (25, 0.0059213959611952305), (26, 0.0006994525319896638), (27, 0.00036283789086155593)]
+
+  9%|▊         | 13/150 [03:03<20:34,  9.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 816, Spatial MSE: [(0, 1.6584357581450604e-05), (1, 0.00011870748130604625), (2, 0.0031975749880075455), (3, 0.0005453406483866274), (4, 0.00011703898053383455), (5, 6.207494880072773e-05), (6, 0.00011605004692682996), (7, 6.891017983434722e-05), (8, 5.214842894929461e-05), (9, 6.3469291490037e-05), (10, 6.166170351207256e-05), (11, 0.00013115185720380396), (12, 0.00024607323575764894), (13, 0.005005598533898592), (14, 0.01036878116428852), (15, 0.0032526119612157345), (16, 0.11782123893499374), (17, 0.000580627762246877), (18, 0.00221285130828619), (19, 0.006897697225213051), (20, 0.002120311837643385), (21, 0.0030837086960673332), (22, 0.004768012091517448), (23, 0.002243798691779375), (24, 0.00394412549212575), (25, 0.010639549233019352), (26, 0.01005137525498867), (27, 0.14773207902908325)], Temporal MSE: [(0, 2.2402635295293294e-05), (1, 1.3744402167503722e-05), (2, 1.9162744138156995e-05), (3, 1.7562275388627313e-05), (4, 1.468363461754052e-05), (5, 1.839534888858907e-05), (6, 9.324527127319016e-06), (7, 8.843978321237955e-06), (8, 7.767589522700291e-06), (9, 1.0290371392329689e-05), (10, 1.543539110571146e-05), (11, 8.263526251539588e-05), (12, 0.00010951275908155367), (13, 0.0005450840690173209), (14, 0.00013577917707152665), (15, 0.00012551200052257627), (16, 0.0001538525684736669), (17, 0.00020099073299206793), (18, 0.0001371200050925836), (19, 0.00012652507575694472), (20, 0.000160550422151573), (21, 0.0002404936822131276), (22, 0.0006156821036711335), (23, 0.0010146584827452898), (24, 0.0012074742699041963), (25, 0.00044251783401705325), (26, 0.00027496955590322614), (27, 8.12433281680569e-05)]
+
+  9%|▉         | 14/150 [03:12<20:43,  9.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 810, Spatial MSE: [(0, 1.5883182641118765e-05), (1, 0.0003290547465439886), (2, 0.00447490531951189), (3, 0.0003749784082174301), (4, 0.00020951020997017622), (5, 0.000158681083121337), (6, 9.373346983920783e-05), (7, 0.00022038033057469875), (8, 0.00013057207979727536), (9, 0.00015422388969454914), (10, 0.00020708002557512373), (11, 0.0002528604818508029), (12, 0.0005580846918746829), (13, 0.005480751395225525), (14, 0.012625636532902718), (15, 0.0059118266217410564), (16, 0.1381085366010666), (17, 0.0013480469351634383), (18, 0.0024974190164357424), (19, 0.009389312006533146), (20, 0.005162461195141077), (21, 0.0032818359322845936), (22, 0.005379976239055395), (23, 0.0034766201861202717), (24, 0.00905284658074379), (25, 0.019247818738222122), (26, 0.011548902839422226), (27, 0.08788884431123734)], Temporal MSE: [(0, 1.8589202227303758e-05), (1, 3.2273099350277334e-05), (2, 2.725479680520948e-05), (3, 2.374547148065176e-05), (4, 3.127565287286416e-05), (5, 3.45028456649743e-05), (6, 2.4834604118950665e-05), (7, 2.9637280022143386e-05), (8, 1.9678627722896636e-05), (9, 3.4499866160331294e-05), (10, 8.475552749587223e-05), (11, 0.0001403826754540205), (12, 0.00015286605048459023), (13, 0.0006321277469396591), (14, 0.00022593516041524708), (15, 0.00030476070242002606), (16, 0.00024328268773388118), (17, 0.0004762429161928594), (18, 0.00028449404635466635), (19, 0.000372314389096573), (20, 0.0008645160705782473), (21, 0.0004772517131641507), (22, 0.0007921020733192563), (23, 0.001254883361980319), (24, 0.0016754092648625374), (25, 0.001459771767258644), (26, 0.00039782136445865035), (27, 0.00034818905987776816)]
+
+ 10%|█         | 15/150 [03:23<21:32,  9.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 804, Spatial MSE: [(0, 1.5170710867096204e-05), (1, 0.00011155725223943591), (2, 0.0026834311429411173), (3, 0.0005096901440992951), (4, 0.00010253396612824872), (5, 5.443413101602346e-05), (6, 0.0001209966721944511), (7, 5.2238225180190057e-05), (8, 4.43586177425459e-05), (9, 5.6574186601210386e-05), (10, 5.8289999287808314e-05), (11, 0.0001249228953383863), (12, 0.00025468977401033044), (13, 0.005310329608619213), (14, 0.010830608196556568), (15, 0.003291255561634898), (16, 0.1255698800086975), (17, 0.0006937198340892792), (18, 0.0025660258252173662), (19, 0.007481277454644442), (20, 0.0022991138976067305), (21, 0.003441130043938756), (22, 0.0051763616502285), (23, 0.0021502284798771143), (24, 0.003723572241142392), (25, 0.009233585558831692), (26, 0.009305577725172043), (27, 0.13306964933872223)], Temporal MSE: [(0, 1.9593177057686262e-05), (1, 1.2255520232429262e-05), (2, 1.5257414816005621e-05), (3, 1.5192080354609061e-05), (4, 1.2875193533545826e-05), (5, 1.6745725588407367e-05), (6, 7.762334462313447e-06), (7, 8.298758984892629e-06), (8, 6.555883828696096e-06), (9, 9.148980097961612e-06), (10, 1.498318033554824e-05), (11, 8.910053293220699e-05), (12, 0.00011321488273097202), (13, 0.0005344036035239697), (14, 0.0001420393818989396), (15, 0.0001655682863201946), (16, 0.00014558847760781646), (17, 0.00017403681704308838), (18, 0.00020141353888902813), (19, 0.00016149721341207623), (20, 0.00024564468185417354), (21, 0.00024787953589111567), (22, 0.000648773624561727), (23, 0.0009778946405276656), (24, 0.0011182576417922974), (25, 0.00046659744111821055), (26, 0.00025511649437248707), (27, 8.032265759538859e-05)]
+
+ 11%|█         | 16/150 [03:30<19:57,  8.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 798, Spatial MSE: [(0, 1.4925360119377729e-05), (1, 0.000365714164217934), (2, 0.004575961735099554), (3, 0.00035702280001714826), (4, 0.000177135385456495), (5, 0.0002096321841236204), (6, 0.00021162707707844675), (7, 0.0005643110489472747), (8, 0.00043206335976719856), (9, 0.0008190986118279397), (10, 0.0010030672419816256), (11, 0.0011219355510547757), (12, 0.001985548995435238), (13, 0.007401955779641867), (14, 0.015343687497079372), (15, 0.007743998430669308), (16, 0.1536039412021637), (17, 0.0036893077194690704), (18, 0.004999316763132811), (19, 0.012047775089740753), (20, 0.013965842314064503), (21, 0.005281132645905018), (22, 0.00880955345928669), (23, 0.006252564024180174), (24, 0.012540340423583984), (25, 0.07063303887844086), (26, 0.030260775238275528), (27, 0.09930489212274551)], Temporal MSE: [(0, 2.127932930306997e-05), (1, 4.072270530741662e-05), (2, 4.190110121271573e-05), (3, 3.3373802580172196e-05), (4, 3.2682281016604975e-05), (5, 6.669789581792429e-05), (6, 4.9105685320682824e-05), (7, 6.684406253043562e-05), (8, 7.003439532127231e-05), (9, 0.0001106309427996166), (10, 0.0002015878853853792), (11, 0.00029851379804313183), (12, 0.00035598015529103577), (13, 0.0009192420984618366), (14, 0.0007754496182315052), (15, 0.000677063362672925), (16, 0.0005054243956692517), (17, 0.0006695042247883976), (18, 0.0007532254094257951), (19, 0.0006120429025031626), (20, 0.0008920507389120758), (21, 0.0010584513656795025), (22, 0.0011175089748576283), (23, 0.00162752706091851), (24, 0.0033229172695428133), (25, 0.009860358200967312), (26, 0.001303007360547781), (27, 0.0003650951257441193)]
+
+ 11%|█▏        | 17/150 [03:38<19:10,  8.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 792, Spatial MSE: [(0, 1.8781996914185584e-05), (1, 0.0001424717193003744), (2, 0.0029746550135314465), (3, 0.0005925410659983754), (4, 0.00011683905177051201), (5, 6.271882011787966e-05), (6, 0.00013382939505390823), (7, 6.497121648862958e-05), (8, 5.211933239479549e-05), (9, 6.858384585939348e-05), (10, 7.084320532158017e-05), (11, 0.00014011544408276677), (12, 0.00027775720809586346), (13, 0.005861781537532806), (14, 0.010340654291212559), (15, 0.0033221303019672632), (16, 0.12167902290821075), (17, 0.0005707701202481985), (18, 0.001956085441634059), (19, 0.0073419311083853245), (20, 0.002461070427671075), (21, 0.002895444631576538), (22, 0.004627736750990152), (23, 0.002270543249323964), (24, 0.00403011217713356), (25, 0.009943339042365551), (26, 0.011088772676885128), (27, 0.15492793917655945)], Temporal MSE: [(0, 2.443729499645997e-05), (1, 1.4096277482167352e-05), (2, 1.8656983229448088e-05), (3, 1.7485283024143428e-05), (4, 1.4635887964686844e-05), (5, 1.9863735360559076e-05), (6, 9.717688044474926e-06), (7, 9.209469681081828e-06), (8, 8.15731800685171e-06), (9, 1.1369233106961474e-05), (10, 1.7833177480497397e-05), (11, 0.00010460826888447627), (12, 0.00012504671758506447), (13, 0.0005417306092567742), (14, 0.00014183323946781456), (15, 0.00014038628432899714), (16, 0.0001508138666395098), (17, 0.0001657411048654467), (18, 0.0001710642100078985), (19, 0.00013932767615187913), (20, 0.0001777433935785666), (21, 0.00023662802414037287), (22, 0.0006012824014760554), (23, 0.0010300104040652514), (24, 0.0012246744008734822), (25, 0.0005376499611884356), (26, 0.00030636778683401644), (27, 9.132338891504332e-05)]
+
+ 12%|█▏        | 18/150 [03:46<18:42,  8.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 786, Spatial MSE: [(0, 1.734770012262743e-05), (1, 0.00038466238765977323), (2, 0.005004189442843199), (3, 0.00046452038805000484), (4, 0.0001864117366494611), (5, 0.00016374474216718227), (6, 0.00012139359751017764), (7, 0.0002518115215934813), (8, 0.00018867020844481885), (9, 0.00026866677217185497), (10, 0.00033152883406728506), (11, 0.00047934966278262436), (12, 0.0007354308618232608), (13, 0.006822942290455103), (14, 0.013244164176285267), (15, 0.006177991162985563), (16, 0.14997652173042297), (17, 0.001186746172606945), (18, 0.002449875697493553), (19, 0.010525251738727093), (20, 0.006269667763262987), (21, 0.003358502872288227), (22, 0.005177538376301527), (23, 0.003321656957268715), (24, 0.007615550886839628), (25, 0.019605666399002075), (26, 0.011511620134115219), (27, 0.10039517283439636)], Temporal MSE: [(0, 2.0468771253945306e-05), (1, 3.2917250791797414e-05), (2, 2.8062246201443486e-05), (3, 2.521823080314789e-05), (4, 3.212257433915511e-05), (5, 3.9437873056158423e-05), (6, 2.7904743546969257e-05), (7, 3.0026134481886402e-05), (8, 2.8635004127863795e-05), (9, 3.726227441802621e-05), (10, 7.567658030893654e-05), (11, 0.00017914753698278219), (12, 0.00020240175945218652), (13, 0.0006463344907388091), (14, 0.00024829775793477893), (15, 0.00028195104096084833), (16, 0.0002429773157928139), (17, 0.00043387943878769875), (18, 0.0003331736952532083), (19, 0.0004090415022801608), (20, 0.0009179810876958072), (21, 0.0005755347665399313), (22, 0.0007521964143961668), (23, 0.0012981124455109239), (24, 0.0018102469621226192), (25, 0.0023852079175412655), (26, 0.000406362785724923), (27, 0.0002845668641384691)]
+
+ 13%|█▎        | 19/150 [03:54<18:18,  8.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 780, Spatial MSE: [(0, 2.923673855548259e-05), (1, 0.00020258536096662283), (2, 0.003970166202634573), (3, 0.0008485738071613014), (4, 0.00015169262769632041), (5, 8.552763756597415e-05), (6, 0.0002109520573867485), (7, 8.523250289727002e-05), (8, 7.111173908924684e-05), (9, 9.429128840565681e-05), (10, 0.00010167065192945302), (11, 0.00022695015650242567), (12, 0.0005184047040529549), (13, 0.011442635208368301), (14, 0.01863761432468891), (15, 0.005437201354652643), (16, 0.18822506070137024), (17, 0.0008766168029978871), (18, 0.002589907729998231), (19, 0.011974208056926727), (20, 0.00427952129393816), (21, 0.0035256147384643555), (22, 0.006549412850290537), (23, 0.0031321148853749037), (24, 0.005507310386747122), (25, 0.01324058324098587), (26, 0.014820756390690804), (27, 0.24741359055042267)], Temporal MSE: [(0, 3.6862715205643326e-05), (1, 1.984411574085243e-05), (2, 2.3963315470609814e-05), (3, 2.3093942218110897e-05), (4, 1.872240136435721e-05), (5, 2.7286163458484225e-05), (6, 1.3593568837677594e-05), (7, 1.3346388186619151e-05), (8, 1.0987531823047902e-05), (9, 1.5485034964513034e-05), (10, 2.624834633024875e-05), (11, 0.00023087688896339387), (12, 0.0002502290008123964), (13, 0.0008807191043160856), (14, 0.0002311634598299861), (15, 0.00024175792350433767), (16, 0.00020657699496950954), (17, 0.0002108404296450317), (18, 0.00029826592071913183), (19, 0.0002200958551838994), (20, 0.0002674724964890629), (21, 0.0003495010023470968), (22, 0.0008439482771791518), (23, 0.0014964903239160776), (24, 0.0018239113269373775), (25, 0.0008679864113219082), (26, 0.000468186306534335), (27, 0.00017411581939086318)]
+
+ 13%|█▎        | 20/150 [04:02<17:53,  8.25s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 774, Spatial MSE: [(0, 2.6927311409963295e-05), (1, 0.000624928972683847), (2, 0.007407237309962511), (3, 0.0005975948297418654), (4, 0.0002305478265043348), (5, 0.0003228521381970495), (6, 0.0004513692110776901), (7, 0.0007840069010853767), (8, 0.0007671995554119349), (9, 0.0012048210483044386), (10, 0.0012288119178265333), (11, 0.001430849777534604), (12, 0.00261492352001369), (13, 0.014626621268689632), (14, 0.025879206135869026), (15, 0.011869494803249836), (16, 0.2389836311340332), (17, 0.00560365617275238), (18, 0.006243977230042219), (19, 0.019746283069252968), (20, 0.02236039936542511), (21, 0.005725427996367216), (22, 0.011420419439673424), (23, 0.009027189575135708), (24, 0.014904716052114964), (25, 0.10706740617752075), (26, 0.05676936358213425), (27, 0.17905168235301971)], Temporal MSE: [(0, 3.514909258228727e-05), (1, 5.934987711952999e-05), (2, 6.131640839157626e-05), (3, 5.1452472689561546e-05), (4, 4.996768257115036e-05), (5, 9.186608804157004e-05), (6, 9.190067066811025e-05), (7, 0.00010566166019998491), (8, 0.00012705940753221512), (9, 0.00019248129683546722), (10, 0.00031580316135659814), (11, 0.0005296448362059891), (12, 0.0005805216496810317), (13, 0.001362510840408504), (14, 0.000981697696261108), (15, 0.0009866011096164584), (16, 0.0007241193670779467), (17, 0.0008769312989898026), (18, 0.0009980256436392665), (19, 0.0008148911292664707), (20, 0.001822938909754157), (21, 0.0014789608540013433), (22, 0.0015428307233378291), (23, 0.0024779848754405975), (24, 0.004757686983793974), (25, 0.018915100023150444), (26, 0.0025305242743343115), (27, 0.0005626494530588388)]
+
+ 14%|█▍        | 21/150 [04:11<18:11,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 768, Spatial MSE: [(0, 3.788244066527113e-05), (1, 0.00030728225829079747), (2, 0.004693757276982069), (3, 0.001176767866127193), (4, 0.00019316465477459133), (5, 0.00010788558574859053), (6, 0.0003319946117699146), (7, 9.933752880897373e-05), (8, 9.212685836246237e-05), (9, 0.0001248792395927012), (10, 0.00013783681788481772), (11, 0.00027825741562992334), (12, 0.0005843497347086668), (13, 0.013020003214478493), (14, 0.020437944680452347), (15, 0.006367188412696123), (16, 0.20853638648986816), (17, 0.0007986060227267444), (18, 0.0022038589231669903), (19, 0.01326209120452404), (20, 0.005258593242615461), (21, 0.002980718854814768), (22, 0.006628895178437233), (23, 0.003516786964610219), (24, 0.0063840546645224094), (25, 0.01472399476915598), (26, 0.018722690641880035), (27, 0.30271008610725403)], Temporal MSE: [(0, 4.689341949415393e-05), (1, 2.4462902729283087e-05), (2, 2.9111168259987608e-05), (3, 2.731499080255162e-05), (4, 2.33607497648336e-05), (5, 3.6541885492624715e-05), (6, 1.9221090042265132e-05), (7, 1.67751368280733e-05), (8, 1.4473132978309877e-05), (9, 2.223508636234328e-05), (10, 3.6015291698277e-05), (11, 0.0003094662679359317), (12, 0.0002713552676141262), (13, 0.0009194956510327756), (14, 0.0002691034460440278), (15, 0.00023937072546686977), (16, 0.00023082188272383064), (17, 0.00021867136820219457), (18, 0.00033477789838798344), (19, 0.00024197007587645203), (20, 0.0002716259623412043), (21, 0.00038228448829613626), (22, 0.0008751833811402321), (23, 0.0016729102935642004), (24, 0.0021421725396066904), (25, 0.001259244978427887), (26, 0.0006006069015711546), (27, 0.00022728081967215985)]
+
+ 15%|█▍        | 22/150 [04:19<17:32,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 762, Spatial MSE: [(0, 3.2699768780730665e-05), (1, 0.0006497430149465799), (2, 0.007539269048720598), (3, 0.0007218198734335601), (4, 0.0002016135404119268), (5, 0.00022362262825481594), (6, 0.00039464200381189585), (7, 0.0002676432195585221), (8, 0.00023316682199947536), (9, 0.0003193864249624312), (10, 0.00037151205469854176), (11, 0.0005208621732890606), (12, 0.0009842792060226202), (13, 0.014580092392861843), (14, 0.024071700870990753), (15, 0.009163448587059975), (16, 0.2574641704559326), (17, 0.0014033964835107327), (18, 0.003162417560815811), (19, 0.01741524413228035), (20, 0.00943700596690178), (21, 0.004344659857451916), (22, 0.007676065433770418), (23, 0.004416326526552439), (24, 0.00979592278599739), (25, 0.029739273712038994), (26, 0.018774522468447685), (27, 0.19476059079170227)], Temporal MSE: [(0, 3.633787855505943e-05), (1, 5.01965478179045e-05), (2, 4.1132039768854156e-05), (3, 3.6475263186730444e-05), (4, 4.052947770105675e-05), (5, 5.228216832620092e-05), (6, 4.299369174987078e-05), (7, 3.835271127172746e-05), (8, 3.789319089264609e-05), (9, 5.27082484040875e-05), (10, 8.138133125612512e-05), (11, 0.0004250300698913634), (12, 0.00038014608435332775), (13, 0.0010355415288358927), (14, 0.0003781429841183126), (15, 0.0004015193844679743), (16, 0.00035465109976939857), (17, 0.0005871166940778494), (18, 0.0005764997913502157), (19, 0.0006244365940801799), (20, 0.0009365042787976563), (21, 0.0008461799588985741), (22, 0.0011575536336749792), (23, 0.0022047623060643673), (24, 0.0031166914850473404), (25, 0.005859370809048414), (26, 0.0006861299043521285), (27, 0.00044115327182225883)]
+
+ 15%|█▌        | 23/150 [04:27<17:34,  8.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 756, Spatial MSE: [(0, 4.717382398666814e-05), (1, 0.00037023707409389317), (2, 0.004558984190225601), (3, 0.0012566004879772663), (4, 0.00019529569544829428), (5, 0.00011187206109752879), (6, 0.0004603760608006269), (7, 9.708062134450302e-05), (8, 0.00010012267011916265), (9, 0.00013361718447413296), (10, 0.00015990639803931117), (11, 0.0003787873429246247), (12, 0.000828928139526397), (13, 0.018769321963191032), (14, 0.027027295902371407), (15, 0.008446071296930313), (16, 0.2539714276790619), (17, 0.0008529343758709729), (18, 0.0020736358128488064), (19, 0.016251562163233757), (20, 0.006645684130489826), (21, 0.0031490663532167673), (22, 0.007133478298783302), (23, 0.003777586854994297), (24, 0.006788970436900854), (25, 0.015896104276180267), (26, 0.02015376277267933), (27, 0.3673993647098541)], Temporal MSE: [(0, 5.585501276073046e-05), (1, 2.7075559046352282e-05), (2, 3.0818813684163615e-05), (3, 2.795374530251138e-05), (4, 2.288140785822179e-05), (5, 3.93856389564462e-05), (6, 2.173935172322672e-05), (7, 1.9025443179998547e-05), (8, 1.587788574397564e-05), (9, 2.4834014766383916e-05), (10, 4.195659494143911e-05), (11, 0.0005253534181974828), (12, 0.00038631929783150554), (13, 0.001128189847804606), (14, 0.00033892609644681215), (15, 0.00030559522565454245), (16, 0.0002730948617681861), (17, 0.00022444562637247145), (18, 0.00038969196612015367), (19, 0.00027700455393642187), (20, 0.00027747932472266257), (21, 0.0004405076033435762), (22, 0.0009563697967678308), (23, 0.002006090246140957), (24, 0.002534876111894846), (25, 0.0017724763602018356), (26, 0.0007393971318379045), (27, 0.000325940694892779)]
+
+ 16%|█▌        | 24/150 [04:35<16:47,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 750, Spatial MSE: [(0, 4.070579234394245e-05), (1, 0.0009287100983783603), (2, 0.009634711779654026), (3, 0.0007863845094107091), (4, 0.00028042428311891854), (5, 0.000428814411861822), (6, 0.0006365852896124125), (7, 0.0010375648271292448), (8, 0.0011495936196297407), (9, 0.0016869055107235909), (10, 0.00209194072522223), (11, 0.002414831891655922), (12, 0.003244953928515315), (13, 0.023173952475190163), (14, 0.03451543673872948), (15, 0.015593255870044231), (16, 0.30616796016693115), (17, 0.005885526072233915), (18, 0.005861037876456976), (19, 0.02483181841671467), (20, 0.01813548430800438), (21, 0.009373703971505165), (22, 0.010848983190953732), (23, 0.00837773084640503), (24, 0.01603659801185131), (25, 0.0976020097732544), (26, 0.0689450204372406), (27, 0.2635301351547241)], Temporal MSE: [(0, 4.60651281173341e-05), (1, 7.717427070019767e-05), (2, 6.87113861204125e-05), (3, 7.06165301380679e-05), (4, 6.606346141779795e-05), (5, 0.00012469355715438724), (6, 0.00014969371841289103), (7, 0.00017290658433921635), (8, 0.00022674973297398537), (9, 0.00033629938843660057), (10, 0.0006345297442749143), (11, 0.001124439644627273), (12, 0.0009071349631994963), (13, 0.001954876584932208), (14, 0.0015237497864291072), (15, 0.0017311651026830077), (16, 0.0011982200667262077), (17, 0.0011855978518724442), (18, 0.0013585335109382868), (19, 0.0013832752592861652), (20, 0.003975295461714268), (21, 0.0020733585115522146), (22, 0.0018577848095446825), (23, 0.0032507148571312428), (24, 0.006345637142658234), (25, 0.02650895155966282), (26, 0.004341800231486559), (27, 0.000529397395439446)]
+
+ 17%|█▋        | 25/150 [04:43<16:39,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 744, Spatial MSE: [(0, 4.066161272930913e-05), (1, 0.00033985773916356266), (2, 0.003188150702044368), (3, 0.0009634245652705431), (4, 0.00014627884957008064), (5, 8.467295992886648e-05), (6, 0.00041996693471446633), (7, 7.467728573828936e-05), (8, 7.990413723746315e-05), (9, 0.00010502132499823347), (10, 0.00013408297672867775), (11, 0.000288585462840274), (12, 0.0007209475734271109), (13, 0.01784450188279152), (14, 0.025496728718280792), (15, 0.008338842540979385), (16, 0.27726417779922485), (17, 0.000951852067373693), (18, 0.0020781613420695066), (19, 0.018031174317002296), (20, 0.00788204651325941), (21, 0.002946317195892334), (22, 0.00656083831563592), (23, 0.003291649743914604), (24, 0.0059896232560276985), (25, 0.012777423486113548), (26, 0.015937434509396553), (27, 0.3260517716407776)], Temporal MSE: [(0, 4.690425703302026e-05), (1, 2.0631270672311075e-05), (2, 2.3388802219415084e-05), (3, 2.0523215425782837e-05), (4, 1.668621007411275e-05), (5, 3.113608545390889e-05), (6, 1.96899982256582e-05), (7, 1.5576712030451745e-05), (8, 1.3751924598182086e-05), (9, 2.233784289273899e-05), (10, 3.5439043131191283e-05), (11, 0.0003980336769018322), (12, 0.0003626182733569294), (13, 0.0010220423573628068), (14, 0.00035482263774611056), (15, 0.0003375899977982044), (16, 0.00027994863921776414), (17, 0.00022498442558571696), (18, 0.00047523051034659147), (19, 0.0003200671053491533), (20, 0.00032105378340929747), (21, 0.0004797808069270104), (22, 0.0009500464075244963), (23, 0.0019785412587225437), (24, 0.002462739823386073), (25, 0.0021603975910693407), (26, 0.0007495956378988922), (27, 0.0003799654368776828)]
+
+ 17%|█▋        | 26/150 [04:51<16:45,  8.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 738, Spatial MSE: [(0, 3.4263506677234545e-05), (1, 0.0005466695292852819), (2, 0.005135110579431057), (3, 0.0005174439284019172), (4, 0.00013299507554620504), (5, 0.0001436595484847203), (6, 0.0003904135664924979), (7, 0.0001744611217873171), (8, 0.00020403170492500067), (9, 0.0002868634765036404), (10, 0.0003803891595453024), (11, 0.0005167426425032318), (12, 0.0009625308448448777), (13, 0.018062856048345566), (14, 0.026185818016529083), (15, 0.00926944613456726), (16, 0.27931439876556396), (17, 0.0012480246368795633), (18, 0.003194136079400778), (19, 0.01844947040081024), (20, 0.008795836009085178), (21, 0.005358734633773565), (22, 0.007637652102857828), (23, 0.003314164001494646), (24, 0.00882016308605671), (25, 0.0210579801350832), (26, 0.013021140359342098), (27, 0.23048964142799377)], Temporal MSE: [(0, 3.4571505239000544e-05), (1, 3.535374708008021e-05), (2, 3.2556097721681e-05), (3, 2.7373131160857156e-05), (4, 2.6725834686658345e-05), (5, 3.800464401138015e-05), (6, 3.399243360036053e-05), (7, 2.7943293389398605e-05), (8, 3.147288225591183e-05), (9, 4.4624692236538976e-05), (10, 6.772678170818835e-05), (11, 0.0004281381261534989), (12, 0.00040100878686644137), (13, 0.001054608728736639), (14, 0.0004202215641271323), (15, 0.0004212153726257384), (16, 0.0003570224507711828), (17, 0.000430523999966681), (18, 0.0006293879123404622), (19, 0.0005676901200786233), (20, 0.0006045433110557497), (21, 0.0007244563312269747), (22, 0.001121650100685656), (23, 0.002356938784942031), (24, 0.0033821179531514645), (25, 0.007035395130515099), (26, 0.0009128302335739136), (27, 0.000422589568188414)]
+
+ 18%|█▊        | 27/150 [05:00<16:56,  8.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 732, Spatial MSE: [(0, 3.567364547052421e-05), (1, 0.00029514345806092024), (2, 0.002242811257019639), (3, 0.0007423120550811291), (4, 0.00010604543058434501), (5, 6.238614150788635e-05), (6, 0.00040574726881459355), (7, 5.603629324468784e-05), (8, 5.871665052836761e-05), (9, 7.746666960883886e-05), (10, 0.0001057908302755095), (11, 0.0002611542004160583), (12, 0.0006262885872274637), (13, 0.01496555283665657), (14, 0.019628213718533516), (15, 0.0063744899816811085), (16, 0.19816988706588745), (17, 0.0006699086516164243), (18, 0.0012600935297086835), (19, 0.01312257256358862), (20, 0.005881940480321646), (21, 0.001947848591953516), (22, 0.004420953802764416), (23, 0.0021253563463687897), (24, 0.0039703622460365295), (25, 0.008035143837332726), (26, 0.010598835535347462), (27, 0.24971653521060944)], Temporal MSE: [(0, 3.974851642851718e-05), (1, 1.5796229490661062e-05), (2, 1.7364816812914796e-05), (3, 1.4585083590645809e-05), (4, 1.1078497664129827e-05), (5, 2.277277053508442e-05), (6, 1.6022066120058298e-05), (7, 1.3107771337672602e-05), (8, 1.014400004351046e-05), (9, 1.6702175344107673e-05), (10, 2.7752066671382636e-05), (11, 0.00043542124330997467), (12, 0.00029100620304234326), (13, 0.0007258160039782524), (14, 0.0002681360929273069), (15, 0.0002569892385508865), (16, 0.00019628682639449835), (17, 0.0001443408109480515), (18, 0.0003485435445327312), (19, 0.00023833011800888926), (20, 0.00022509080008603632), (21, 0.00035091882455162704), (22, 0.0006600033957511187), (23, 0.0013956010807305574), (24, 0.0017280031461268663), (25, 0.0021567184012383223), (26, 0.0005672408733516932), (27, 0.00033565633930265903)]
+
+ 19%|█▊        | 28/150 [05:07<16:07,  7.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 726, Spatial MSE: [(0, 2.8891683541587554e-05), (1, 0.0005861807148903608), (2, 0.005073870066553354), (3, 0.00045491589116863906), (4, 0.00019106942636426538), (5, 0.0002925183216575533), (6, 0.0007179591921158135), (7, 0.0008535736706107855), (8, 0.0008336564642377198), (9, 0.0011650744127109647), (10, 0.001339697395451367), (11, 0.0017614651005715132), (12, 0.002243374241515994), (13, 0.0182796623557806), (14, 0.026842422783374786), (15, 0.013469450175762177), (16, 0.2704123556613922), (17, 0.003769548609852791), (18, 0.004000822547823191), (19, 0.020728562027215958), (20, 0.013617328368127346), (21, 0.007404915057122707), (22, 0.007513157557696104), (23, 0.0056954133324325085), (24, 0.013251114636659622), (25, 0.0453469380736351), (26, 0.03000110201537609), (27, 0.19052016735076904)], Temporal MSE: [(0, 2.8406702767824754e-05), (1, 4.154491034569219e-05), (2, 3.682549868244678e-05), (3, 4.010605698567815e-05), (4, 6.0469807067420334e-05), (5, 0.0001605686265975237), (6, 0.00018721951346378773), (7, 0.00017284388013649732), (8, 0.00013964850222691894), (9, 0.0002211428218288347), (10, 0.00045398404472507536), (11, 0.000881555606611073), (12, 0.0006621842621825635), (13, 0.001308137783780694), (14, 0.0011967177269980311), (15, 0.0013520222855731845), (16, 0.001115518738515675), (17, 0.0010466971434652805), (18, 0.0012269129510968924), (19, 0.0012574577704071999), (20, 0.0029340495821088552), (21, 0.0016507587861269712), (22, 0.001503339153714478), (23, 0.0028262955602258444), (24, 0.006263652350753546), (25, 0.021786658093333244), (26, 0.0041082995012402534), (27, 0.00037183501990512013)]
+
+ 19%|█▉        | 29/150 [05:15<16:12,  8.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 720, Spatial MSE: [(0, 2.2557374904863536e-05), (1, 0.00018157417071051896), (2, 0.0011735674925148487), (3, 0.0003974422870669514), (4, 5.9795584093080834e-05), (5, 3.521996404742822e-05), (6, 0.0002429286832921207), (7, 3.6112487578066066e-05), (8, 3.457610364421271e-05), (9, 4.387352601042949e-05), (10, 5.75401536480058e-05), (11, 0.00011767098476411775), (12, 0.0003110752732027322), (13, 0.008413699455559254), (14, 0.009793887846171856), (15, 0.0032946080900728703), (16, 0.09670020639896393), (17, 0.0003059314622078091), (18, 0.0006786217563785613), (19, 0.0061285425908863544), (20, 0.0027357046492397785), (21, 0.0010909721022471786), (22, 0.0020722467452287674), (23, 0.0010864888317883015), (24, 0.00202730018645525), (25, 0.00405754242092371), (26, 0.006276736035943031), (27, 0.1414984166622162)], Temporal MSE: [(0, 2.447194310661871e-05), (1, 8.874095328792464e-06), (2, 9.921498531184625e-06), (3, 8.48508807393955e-06), (4, 6.1299933804548346e-06), (5, 1.38692012114916e-05), (6, 1.105522005673265e-05), (7, 8.160562174452934e-06), (8, 6.354202923830599e-06), (9, 1.0364115951233543e-05), (10, 1.5045932741486467e-05), (11, 0.00016128328570630401), (12, 0.00016472699644509703), (13, 0.0003833638911601156), (14, 0.00015022058505564928), (15, 0.00012065676128258929), (16, 0.00010542341624386609), (17, 7.47038793633692e-05), (18, 0.00017293162818532437), (19, 0.00011796303442679346), (20, 0.00011431170423747972), (21, 0.00018615685985423625), (22, 0.00032103827106766403), (23, 0.0007023217622190714), (24, 0.0009018665878102183), (25, 0.0014238207368180156), (26, 0.00033831255859695375), (27, 0.0001788237423170358)]
+
+ 20%|██        | 30/150 [05:22<15:37,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 714, Spatial MSE: [(0, 1.7577172911842354e-05), (1, 0.00023600210261065513), (2, 0.0018442162545397878), (3, 0.00020213100651744753), (4, 5.7295794249512255e-05), (5, 5.473816054291092e-05), (6, 0.00014168318011797965), (7, 9.749837045092136e-05), (8, 0.00014197952987160534), (9, 0.00023902938119135797), (10, 0.00034783108276315033), (11, 0.0004341672465670854), (12, 0.0005795930046588182), (13, 0.008924010209739208), (14, 0.010960198938846588), (15, 0.004122563172131777), (16, 0.10899899154901505), (17, 0.00047978246584534645), (18, 0.0015622482169419527), (19, 0.007415099069476128), (20, 0.003865817328915), (21, 0.0023136897943913937), (22, 0.0028534429147839546), (23, 0.0012427264591678977), (24, 0.0036021743435412645), (25, 0.006488168146461248), (26, 0.00486740143969655), (27, 0.10161779075860977)], Temporal MSE: [(0, 1.5750954844406806e-05), (1, 1.148601859313203e-05), (2, 1.4338133951241616e-05), (3, 1.2496278031903785e-05), (4, 1.1048112355638295e-05), (5, 1.763863838277757e-05), (6, 2.072440656775143e-05), (7, 1.522887123428518e-05), (8, 2.052527997875586e-05), (9, 3.0523176974384114e-05), (10, 4.6988607209641486e-05), (11, 0.00019736503600142896), (12, 0.00020236143609508872), (13, 0.00043398721027188003), (14, 0.00022070572595112026), (15, 0.0001843430509325117), (16, 0.00015929350047372282), (17, 0.00015703645476605743), (18, 0.00025511247804388404), (19, 0.00023249829246196896), (20, 0.00025929140974767506), (21, 0.00031655305065214634), (22, 0.0004460575873963535), (23, 0.0009573249844834208), (24, 0.00148260488640517), (25, 0.003349762409925461), (26, 0.0004631796036846936), (27, 0.00016322912415489554)]
+
+ 21%|██        | 31/150 [05:30<15:32,  7.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 708, Spatial MSE: [(0, 1.2225268619658891e-05), (1, 0.00010604141425574198), (2, 0.0006105919019319117), (3, 0.00021900300635024905), (4, 3.327750164316967e-05), (5, 1.9061557395616546e-05), (6, 0.00014694871788378805), (7, 2.0597597540472634e-05), (8, 1.844118924054783e-05), (9, 2.3518496163887903e-05), (10, 2.927482819359284e-05), (11, 5.386722841649316e-05), (12, 0.00014837505295872688), (13, 0.004765587393194437), (14, 0.00554911233484745), (15, 0.0019946384709328413), (16, 0.06271573156118393), (17, 0.00020337353635113686), (18, 0.0004648951580747962), (19, 0.004199662245810032), (20, 0.001959987683221698), (21, 0.00074237328954041), (22, 0.0014997203834354877), (23, 0.00064251379808411), (24, 0.0011898137163370848), (25, 0.002375231822952628), (26, 0.0035332893021404743), (27, 0.07780101895332336)], Temporal MSE: [(0, 1.3581895473180339e-05), (1, 4.885915132035734e-06), (2, 5.683511062670732e-06), (3, 4.659300429921132e-06), (4, 3.1577305890095886e-06), (5, 7.508384442189708e-06), (6, 6.573513473995263e-06), (7, 5.0044391173287295e-06), (8, 3.5027821922994917e-06), (9, 5.789116130472394e-06), (10, 7.475371603504755e-06), (11, 6.31559087196365e-05), (12, 8.47024712129496e-05), (13, 0.00020944816060364246), (14, 8.974457159638405e-05), (15, 7.324838952627033e-05), (16, 6.62407765048556e-05), (17, 4.742794408230111e-05), (18, 0.00012952106771990657), (19, 8.717583841644228e-05), (20, 8.517975948052481e-05), (21, 0.00013228264288045466), (22, 0.00023609254276379943), (23, 0.00046662139357067645), (24, 0.0005457531660795212), (25, 0.0008733553113415837), (26, 0.00019306769536342472), (27, 0.00011063805141020566)]
+
+ 21%|██▏       | 32/150 [05:39<15:48,  8.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 702, Spatial MSE: [(0, 9.9455446616048e-06), (1, 0.00016587766003794968), (2, 0.0013249682961031795), (3, 0.0001561650278745219), (4, 7.581021054647863e-05), (5, 0.00011898776574525982), (6, 0.00024364874116145074), (7, 0.0002769989368971437), (8, 0.0003868533531203866), (9, 0.000523226975928992), (10, 0.000895234290510416), (11, 0.0011940211988985538), (12, 0.0011506021255627275), (13, 0.008211943320930004), (14, 0.01461450383067131), (15, 0.009384759701788425), (16, 0.19591574370861053), (17, 0.0024926536716520786), (18, 0.0023150888737291098), (19, 0.013604192063212395), (20, 0.011022062972187996), (21, 0.0022305052261799574), (22, 0.003780192695558071), (23, 0.0024944364558905363), (24, 0.005760597065091133), (25, 0.017703134566545486), (26, 0.009148161858320236), (27, 0.06003652885556221)], Temporal MSE: [(0, 9.87027942755958e-06), (1, 1.4375669707078487e-05), (2, 1.1874132724187803e-05), (3, 2.0581966964527965e-05), (4, 3.169755291310139e-05), (5, 6.0417569329729304e-05), (6, 7.856494630686939e-05), (7, 6.68587745167315e-05), (8, 5.7427201682003215e-05), (9, 0.00010280017886543646), (10, 0.00017085089348256588), (11, 0.00027003276045434177), (12, 0.00033050004276447), (13, 0.0005813338211737573), (14, 0.0006999077158980072), (15, 0.0007724105380475521), (16, 0.0007509507122449577), (17, 0.0007430317346006632), (18, 0.0006484571495093405), (19, 0.0006263378891162574), (20, 0.0010924007510766387), (21, 0.0008031860343180597), (22, 0.0007906951941549778), (23, 0.0012740049278363585), (24, 0.0027994525153189898), (25, 0.010662172921001911), (26, 0.0017617602134123445), (27, 0.0001271036162506789)]
+
+ 22%|██▏       | 33/150 [05:48<16:06,  8.26s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 696, Spatial MSE: [(0, 8.354394594789483e-06), (1, 5.9104273532284424e-05), (2, 0.0003445774200372398), (3, 0.0001256752002518624), (4, 1.8547054423834197e-05), (5, 1.1253848242631648e-05), (6, 8.593204984208569e-05), (7, 1.5916506526991725e-05), (8, 1.1109240404039156e-05), (9, 1.4683857443742454e-05), (10, 2.035367833741475e-05), (11, 6.805268640164286e-05), (12, 0.00017748265236150473), (13, 0.004532852675765753), (14, 0.004572258796542883), (15, 0.001384918112307787), (16, 0.04285174235701561), (17, 0.00016478791076224297), (18, 0.0005576849216595292), (19, 0.002912795403972268), (20, 0.0013850918039679527), (21, 0.0005272547714412212), (22, 0.0010042455978691578), (23, 0.0004091547743882984), (24, 0.0008175121038220823), (25, 0.0016338643617928028), (26, 0.0022243151906877756), (27, 0.054304707795381546)], Temporal MSE: [(0, 8.510171937814448e-06), (1, 2.7360013064026134e-06), (2, 3.3193364288308658e-06), (3, 2.736269379965961e-06), (4, 1.7507064740129863e-06), (5, 4.3855411604454275e-06), (6, 3.6994711081206333e-06), (7, 3.967235898016952e-06), (8, 2.0429824871825986e-06), (9, 3.454488933130051e-06), (10, 6.725637376803206e-06), (11, 0.0001402213965775445), (12, 9.672390297055244e-05), (13, 0.00017020439554471523), (14, 6.827888864791021e-05), (15, 5.1502447604434565e-05), (16, 5.138718552188948e-05), (17, 3.9515602111350745e-05), (18, 9.19503509066999e-05), (19, 6.230882718227804e-05), (20, 7.163939153542742e-05), (21, 0.00011738340981537476), (22, 0.0001780809834599495), (23, 0.0003370109770912677), (24, 0.00039549972279928625), (25, 0.0007033802103251219), (26, 0.00014674743579234928), (27, 8.673030242789537e-05)]
+
+ 23%|██▎       | 34/150 [05:55<15:39,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 690, Spatial MSE: [(0, 6.2267658904602285e-06), (1, 5.3614920034306124e-05), (2, 0.0004427208623383194), (3, 3.940985698136501e-05), (4, 1.5672298104618676e-05), (5, 1.5886500477790833e-05), (6, 3.3059488487197086e-05), (7, 2.4177539671654813e-05), (8, 3.196748730260879e-05), (9, 5.787478221463971e-05), (10, 7.840376201784238e-05), (11, 0.00013163016410544515), (12, 0.0002279901527799666), (13, 0.004672704264521599), (14, 0.005072165280580521), (15, 0.0018535241251811385), (16, 0.05383416637778282), (17, 0.00022853190603200346), (18, 0.0008759128395467997), (19, 0.003583404468372464), (20, 0.002603578381240368), (21, 0.0006100442842580378), (22, 0.001201078062877059), (23, 0.0005132285295985639), (24, 0.0013302593724802136), (25, 0.002680310048162937), (26, 0.0018842414719983935), (27, 0.03984907269477844)], Temporal MSE: [(0, 4.666685981646879e-06), (1, 2.65849189418077e-06), (2, 4.809535312233493e-06), (3, 3.991390713053988e-06), (4, 4.705483661382459e-06), (5, 5.666996457875939e-06), (6, 7.655460649402812e-06), (7, 4.480120878724847e-06), (8, 6.51467735224287e-06), (9, 7.712335900578182e-06), (10, 1.2648697520489804e-05), (11, 0.00013991106243338436), (12, 0.00010293033119523898), (13, 0.00018626198288984597), (14, 8.84261098690331e-05), (15, 7.249126065289602e-05), (16, 6.998354365350679e-05), (17, 6.600860069738701e-05), (18, 0.00011778822226915509), (19, 0.00010241870040772483), (20, 0.00012132588744862005), (21, 0.00015574153803754598), (22, 0.00020661769667640328), (23, 0.00039245240623131394), (24, 0.0006317489314824343), (25, 0.001253665192052722), (26, 0.00020879764633718878), (27, 8.13184815342538e-05)]
+
+ 23%|██▎       | 35/150 [06:03<15:21,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 684, Spatial MSE: [(0, 3.952736733481288e-06), (1, 3.3380445529473945e-05), (2, 0.00019305720343254507), (3, 6.804202712373808e-05), (4, 1.047696696332423e-05), (5, 6.648761427641148e-06), (6, 4.590267053572461e-05), (7, 9.342528755951207e-06), (8, 6.5954918682109565e-06), (9, 9.10956623556558e-06), (10, 1.569217783981003e-05), (11, 5.838039214722812e-05), (12, 0.00016971654258668423), (13, 0.004253150895237923), (14, 0.004305490292608738), (15, 0.0011865230044350028), (16, 0.04107316955924034), (17, 0.00018674206512514502), (18, 0.000330635579302907), (19, 0.002716032788157463), (20, 0.0013320372672751546), (21, 0.0004031878779642284), (22, 0.0008509235340170562), (23, 0.00034005066845566034), (24, 0.0006976844160817564), (25, 0.0014833087334409356), (26, 0.0018719391664490104), (27, 0.043634846806526184)], Temporal MSE: [(0, 5.027830866310978e-06), (1, 1.7304785160376923e-06), (2, 2.1625385215884307e-06), (3, 1.8175477407567087e-06), (4, 1.093020955522661e-06), (5, 2.650457417985308e-06), (6, 2.0747031612700084e-06), (7, 2.262348516524071e-06), (8, 1.2157890978414798e-06), (9, 2.0509282876446377e-06), (10, 4.621911557478597e-06), (11, 0.00012122630141675472), (12, 9.809713810682297e-05), (13, 0.00014046933210920542), (14, 6.294986815191805e-05), (15, 5.697721280739643e-05), (16, 4.410172186908312e-05), (17, 3.044224467885215e-05), (18, 7.281739090103656e-05), (19, 4.9236678023589775e-05), (20, 4.88722616864834e-05), (21, 8.48287163535133e-05), (22, 0.0001437435275875032), (23, 0.00030112278182059526), (24, 0.0003530093526933342), (25, 0.00046174656017683446), (26, 0.0001285848265979439), (27, 7.75928347138688e-05)]
+
+ 24%|██▍       | 36/150 [06:11<15:13,  8.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 678, Spatial MSE: [(0, 3.76264665646886e-06), (1, 6.132199632702395e-05), (2, 0.0005069755134172738), (3, 6.327511073322967e-05), (4, 4.005799928563647e-05), (5, 6.383769505191594e-05), (6, 0.0002442618424538523), (7, 0.00022214885393623263), (8, 0.00025612246827222407), (9, 0.00043868354987353086), (10, 0.0012468069326132536), (11, 0.0014625932089984417), (12, 0.001159873092547059), (13, 0.006545366253703833), (14, 0.009604699909687042), (15, 0.008298260159790516), (16, 0.10993433743715286), (17, 0.004284556023776531), (18, 0.0034522456116974354), (19, 0.011330530978739262), (20, 0.01143069937825203), (21, 0.00508579146116972), (22, 0.0023308356758207083), (23, 0.0015316031640395522), (24, 0.003878316842019558), (25, 0.008426036685705185), (26, 0.004084187559783459), (27, 0.03849013149738312)], Temporal MSE: [(0, 4.286416697141249e-06), (1, 5.5261680245166644e-06), (2, 4.8141746447072364e-06), (3, 1.1706154509738553e-05), (4, 1.811248694139067e-05), (5, 5.596359187620692e-05), (6, 9.040431905305013e-05), (7, 6.285959534579888e-05), (8, 4.861910565523431e-05), (9, 9.986216173274443e-05), (10, 0.00019141522352583706), (11, 0.00031429744558408856), (12, 0.0003553119604475796), (13, 0.0005566212930716574), (14, 0.0009331898181699216), (15, 0.0015009120106697083), (16, 0.0018679398344829679), (17, 0.001801628852263093), (18, 0.0010326590854674578), (19, 0.0013436974259093404), (20, 0.0015751826576888561), (21, 0.0011916781077161431), (22, 0.0006015651160851121), (23, 0.0009141465998254716), (24, 0.0021145823411643505), (25, 0.00542239798232913), (26, 0.0006640480714850128), (27, 8.678292215336114e-05)]
+
+ 25%|██▍       | 37/150 [06:19<15:01,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 672, Spatial MSE: [(0, 3.7266427170834504e-06), (1, 2.5503435608698055e-05), (2, 0.00016650759789627045), (3, 4.982745667803101e-05), (4, 8.123822226480115e-06), (5, 5.534356205316726e-06), (6, 3.2322634069714695e-05), (7, 7.306206498469692e-06), (8, 5.614015663013561e-06), (9, 8.145501851686276e-06), (10, 1.7543936337460764e-05), (11, 0.00010353778634453192), (12, 0.00024689920246601105), (13, 0.0057207453064620495), (14, 0.005757956765592098), (15, 0.001507893088273704), (16, 0.049784041941165924), (17, 0.0002097163669532165), (18, 0.00028363498859107494), (19, 0.0038297835271805525), (20, 0.002036214107647538), (21, 0.0004564377595670521), (22, 0.0009010966750793159), (23, 0.00036049491609446704), (24, 0.0007757152779959142), (25, 0.0014910193858668208), (26, 0.0017078759847208858), (27, 0.05378958582878113)], Temporal MSE: [(0, 3.9846090658102185e-06), (1, 1.3778505945083452e-06), (2, 1.7403972378815524e-06), (3, 1.505113345956488e-06), (4, 8.652853011881234e-07), (5, 2.1074874894111417e-06), (6, 1.7119714357249904e-06), (7, 2.0466529804252787e-06), (8, 1.0227354323433246e-06), (9, 1.740978973430174e-06), (10, 6.647588179475861e-06), (11, 0.0002439735981170088), (12, 0.00012794428039342165), (13, 0.00019669438188429922), (14, 7.687966717639938e-05), (15, 7.208268652902916e-05), (16, 5.5772885389160365e-05), (17, 3.729964737431146e-05), (18, 9.114302520174533e-05), (19, 6.151816342025995e-05), (20, 5.740641427109949e-05), (21, 0.00010705789463827386), (22, 0.0001707785268081352), (23, 0.00034055495052598417), (24, 0.0003997020539827645), (25, 0.0005736693274229765), (26, 0.00015036117110867053), (27, 0.00010731341171776876)]
+
+ 25%|██▌       | 38/150 [06:26<14:35,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 666, Spatial MSE: [(0, 3.089194706262788e-06), (1, 2.8983939046156593e-05), (2, 0.0002445463615003973), (3, 2.4856539312168024e-05), (4, 1.063684521795949e-05), (5, 1.1132704457850195e-05), (6, 1.468088521505706e-05), (7, 1.7020323866745457e-05), (8, 2.1858742911717854e-05), (9, 4.498434645938687e-05), (10, 7.249973714351654e-05), (11, 0.00016492804570589215), (12, 0.0002982070145662874), (13, 0.005874126683920622), (14, 0.006364135537296534), (15, 0.0019019744358956814), (16, 0.06428724527359009), (17, 0.0003086883807554841), (18, 0.0004475559981074184), (19, 0.004641817416995764), (20, 0.0029957201331853867), (21, 0.0005424366681836545), (22, 0.0010674070799723268), (23, 0.00047783090849407017), (24, 0.0012306575663387775), (25, 0.0023097991943359375), (26, 0.0019996461924165487), (27, 0.04749002307653427)], Temporal MSE: [(0, 2.5656158868514467e-06), (1, 1.351686933048768e-06), (2, 2.7037738163926406e-06), (3, 2.847761152224848e-06), (4, 3.7481320305232657e-06), (5, 3.6052022096555447e-06), (6, 5.044190402259119e-06), (7, 3.2057942007668316e-06), (8, 4.786815225088503e-06), (9, 5.6658736866666e-06), (10, 1.1741561138478573e-05), (11, 0.0002470078179612756), (12, 0.0001291645603487268), (13, 0.00020784609660040587), (14, 9.954470442607999e-05), (15, 9.429498459212482e-05), (16, 7.619941607117653e-05), (17, 6.290563032962382e-05), (18, 0.00012161736231064424), (19, 0.00010199725511483848), (20, 0.00011887818982359022), (21, 0.0001602663251105696), (22, 0.00022709900804329664), (23, 0.000418471492594108), (24, 0.0006444135797210038), (25, 0.0010867015225812793), (26, 0.00021833795472048223), (27, 0.00010607807780615985)]
+
+ 26%|██▌       | 39/150 [06:34<14:29,  7.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 660, Spatial MSE: [(0, 2.0044039956701454e-06), (1, 1.6202919141505845e-05), (2, 9.408428013557568e-05), (3, 3.2530995667912066e-05), (4, 5.146608600625768e-06), (5, 3.337749831189285e-06), (6, 2.147788654838223e-05), (7, 4.590398930304218e-06), (8, 3.4624542877281783e-06), (9, 4.527987130131805e-06), (10, 9.102238436753396e-06), (11, 4.8072215577121824e-05), (12, 0.00017225566261913627), (13, 0.005089588463306427), (14, 0.004573695361614227), (15, 0.001302288263104856), (16, 0.04288611561059952), (17, 0.00016750978829804808), (18, 0.00021493453823495656), (19, 0.0031667056027799845), (20, 0.0016873626736924052), (21, 0.00024519074941053987), (22, 0.0006745954160578549), (23, 0.00027583158225752413), (24, 0.0006139397737570107), (25, 0.0012276103952899575), (26, 0.0012274160981178284), (27, 0.04343778267502785)], Temporal MSE: [(0, 2.3352897642325843e-06), (1, 8.499091563862748e-07), (2, 1.0999613095918903e-06), (3, 9.475414231019386e-07), (4, 5.608382025457104e-07), (5, 1.3782567975795246e-06), (6, 1.061122361534217e-06), (7, 1.2828633089156938e-06), (8, 6.411943900275219e-07), (9, 1.114635438170808e-06), (10, 2.916682433351525e-06), (11, 0.00011861341045005247), (12, 0.00010218258103122935), (13, 0.00016977309132926166), (14, 6.70723820803687e-05), (15, 4.392391929286532e-05), (16, 4.245417585480027e-05), (17, 2.9059756343485788e-05), (18, 7.804889173712581e-05), (19, 5.6076805776683614e-05), (20, 5.302008503349498e-05), (21, 8.376452751690522e-05), (22, 0.00012764814891852438), (23, 0.00027424562722444534), (24, 0.0003438497078604996), (25, 0.0004662846913561225), (26, 0.00012526821228675544), (27, 9.098245209315792e-05)]
+
+ 27%|██▋       | 40/150 [06:42<14:14,  7.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 654, Spatial MSE: [(0, 2.253043476230232e-06), (1, 4.0306433220393956e-05), (2, 0.0002928141038864851), (3, 4.4407126551959664e-05), (4, 3.0195173167157918e-05), (5, 4.764981713378802e-05), (6, 0.0002859269443433732), (7, 0.00020997053070459515), (8, 0.00024710356956347823), (9, 0.00046633175224997103), (10, 0.0010218177922070026), (11, 0.001361711067147553), (12, 0.0010976912453770638), (13, 0.007312575820833445), (14, 0.010736488737165928), (15, 0.009022155776619911), (16, 0.11289770156145096), (17, 0.004944497253745794), (18, 0.004276962019503117), (19, 0.009978611022233963), (20, 0.012467660009860992), (21, 0.010974109172821045), (22, 0.0030154187697917223), (23, 0.0023908510338515043), (24, 0.006000457797199488), (25, 0.010574348270893097), (26, 0.005877424497157335), (27, 0.04254944622516632)], Temporal MSE: [(0, 3.0334392704389757e-06), (1, 3.1469044188270345e-06), (2, 3.7105614865140524e-06), (3, 7.151557383622276e-06), (4, 1.3179134839447215e-05), (5, 5.271447662380524e-05), (6, 9.62043777690269e-05), (7, 5.914115536143072e-05), (8, 4.060390347149223e-05), (9, 7.767377246636897e-05), (10, 0.0001700391003396362), (11, 0.0002859322412405163), (12, 0.0003329938044771552), (13, 0.0005295102600939572), (14, 0.000957805837970227), (15, 0.0015872796066105366), (16, 0.002360501792281866), (17, 0.0023720681201666594), (18, 0.001677580876275897), (19, 0.003057162743061781), (20, 0.0034439112059772015), (21, 0.0026270942762494087), (22, 0.0009452472440898418), (23, 0.0013168224832043052), (24, 0.003034608205780387), (25, 0.005194548517465591), (26, 0.0007501898799091578), (27, 0.00010361873137298971)]
+
+ 27%|██▋       | 41/150 [06:50<14:09,  7.80s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 648, Spatial MSE: [(0, 2.87454031422385e-06), (1, 1.7992373614106327e-05), (2, 9.903904719976708e-05), (3, 3.215397009626031e-05), (4, 5.579623575613368e-06), (5, 3.2910668323893333e-06), (6, 2.6320265533286147e-05), (7, 6.252764251257759e-06), (8, 4.2192286855424754e-06), (9, 6.223587661224883e-06), (10, 1.0559570000623353e-05), (11, 3.652221130323596e-05), (12, 0.00012437488476280123), (13, 0.0035153550561517477), (14, 0.003951186779886484), (15, 0.0012519715819507837), (16, 0.04803629592061043), (17, 0.0002408749860478565), (18, 0.0004561815585475415), (19, 0.0031871586106717587), (20, 0.0019655872602015734), (21, 0.0003504023770801723), (22, 0.0007295729010365903), (23, 0.00027575745480135083), (24, 0.000617878686171025), (25, 0.001279171439819038), (26, 0.0013070945860818028), (27, 0.03848778456449509)], Temporal MSE: [(0, 2.8739850677084178e-06), (1, 9.030384262587177e-07), (2, 1.0859060921575292e-06), (3, 9.916949466060032e-07), (4, 5.898213544242026e-07), (5, 1.3274436696519842e-06), (6, 1.0750557066785404e-06), (7, 1.6216965832427377e-06), (8, 7.346873189817416e-07), (9, 1.2302066352276597e-06), (10, 3.5970526823803084e-06), (11, 6.706854765070602e-05), (12, 7.700355490669608e-05), (13, 0.00013857122394256294), (14, 6.873279926367104e-05), (15, 8.154485112754628e-05), (16, 6.48607310722582e-05), (17, 5.043723649578169e-05), (18, 8.52310040500015e-05), (19, 6.837036198703572e-05), (20, 6.997859600232914e-05), (21, 0.00011678065493470058), (22, 0.00015122520562727004), (23, 0.0002940092817880213), (24, 0.0003313334018457681), (25, 0.0005318280309438705), (26, 0.0001189923714264296), (27, 9.405195305589586e-05)]
+
+ 28%|██▊       | 42/150 [06:57<13:48,  7.67s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 642, Spatial MSE: [(0, 2.6604539016261697e-06), (1, 2.2758677005185746e-05), (2, 0.0001934168831212446), (3, 2.6238616555929184e-05), (4, 9.879940080281813e-06), (5, 9.033673450176138e-06), (6, 1.354799223918235e-05), (7, 2.2322104996419512e-05), (8, 2.8459202439989895e-05), (9, 6.48954592179507e-05), (10, 0.0001064141106326133), (11, 0.0001362558250548318), (12, 0.0002186219790019095), (13, 0.003776310011744499), (14, 0.004702761769294739), (15, 0.001980356639251113), (16, 0.05600907281041145), (17, 0.0003312155313324183), (18, 0.0006039072177372873), (19, 0.00480876537039876), (20, 0.0034658717922866344), (21, 0.0005483458517119288), (22, 0.0009085666388273239), (23, 0.0004505170800257474), (24, 0.0011271174298599362), (25, 0.0021514459513127804), (26, 0.0025200091768056154), (27, 0.03592304512858391)], Temporal MSE: [(0, 2.335034650968737e-06), (1, 1.2283129535717308e-06), (2, 2.3876759769336786e-06), (3, 2.6949230687023373e-06), (4, 3.4673755635594716e-06), (5, 3.479163297015475e-06), (6, 5.247765329841059e-06), (7, 3.750959876924753e-06), (8, 4.210911811242113e-06), (9, 6.035087153577479e-06), (10, 1.2307789802434854e-05), (11, 7.815387652954087e-05), (12, 8.560078276786953e-05), (13, 0.00016357902495656163), (14, 9.862002480076626e-05), (15, 0.00010762484453152865), (16, 8.692050323588774e-05), (17, 6.643650704063475e-05), (18, 0.00011756228195736185), (19, 0.00010809057857841253), (20, 0.0001252178626600653), (21, 0.00016929749108385295), (22, 0.00020152406068518758), (23, 0.00034909442183561623), (24, 0.0005593100213445723), (25, 0.0009864290477707982), (26, 0.00018354188068769872), (27, 9.563099592924118e-05)]
+
+ 29%|██▊       | 43/150 [07:05<13:42,  7.68s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 636, Spatial MSE: [(0, 1.7517412516099284e-06), (1, 1.3834396668244153e-05), (2, 7.249345071613789e-05), (3, 2.2049103790777735e-05), (4, 4.134451501158765e-06), (5, 2.6470631837582914e-06), (6, 1.5858251572353765e-05), (7, 4.390339199744631e-06), (8, 3.0771627734793583e-06), (9, 4.412547696119873e-06), (10, 1.0189703971263953e-05), (11, 5.906348451389931e-05), (12, 0.00020978668180759996), (13, 0.005862995982170105), (14, 0.005812473129481077), (15, 0.0019175283377990127), (16, 0.07148496806621552), (17, 0.0003136122541036457), (18, 0.00035473480238579214), (19, 0.00553229171782732), (20, 0.0032600874546915293), (21, 0.00034227283322252333), (22, 0.001038572983816266), (23, 0.00037863399484194815), (24, 0.000836429069750011), (25, 0.0018470474751666188), (26, 0.0014909675810486078), (27, 0.05407389998435974)], Temporal MSE: [(0, 2.054017613772885e-06), (1, 7.79936613071186e-07), (2, 1.0009573543356964e-06), (3, 8.282175940621528e-07), (4, 4.710304608579463e-07), (5, 1.0242898724754923e-06), (6, 7.943704076751601e-07), (7, 1.1410387514843023e-06), (8, 5.790500949842681e-07), (9, 9.045068054547301e-07), (10, 3.95502047467744e-06), (11, 0.00014937273226678371), (12, 0.00012721661187242717), (13, 0.00018971871759276837), (14, 9.808385948417708e-05), (15, 0.00010661059786798432), (16, 7.501101936213672e-05), (17, 5.9342844906495884e-05), (18, 0.00012410350609570742), (19, 9.40631507546641e-05), (20, 8.836564666125923e-05), (21, 0.00015727461141068488), (22, 0.00022516355966217816), (23, 0.00042748666601255536), (24, 0.00047189302858896554), (25, 0.0006494134431704879), (26, 0.0001657087414059788), (27, 0.00015085504855960608)]
+
+ 29%|██▉       | 44/150 [07:12<13:21,  7.56s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 630, Spatial MSE: [(0, 2.208460500696674e-06), (1, 3.643002128228545e-05), (2, 0.0002459694806020707), (3, 4.0771206840872765e-05), (4, 2.958119694085326e-05), (5, 5.391251761466265e-05), (6, 0.0005535469972528517), (7, 0.0003474464174360037), (8, 0.0004087684501428157), (9, 0.0009120613103732467), (10, 0.0018113310215994716), (11, 0.002538334345445037), (12, 0.001896530739031732), (13, 0.010516575537621975), (14, 0.017237314954400063), (15, 0.014087410643696785), (16, 0.17475178837776184), (17, 0.007268307264894247), (18, 0.009152163751423359), (19, 0.02243318036198616), (20, 0.030024126172065735), (21, 0.03417670354247093), (22, 0.008175985887646675), (23, 0.00864094216376543), (24, 0.0238639023154974), (25, 0.04319730028510094), (26, 0.01866268552839756), (27, 0.05285995453596115)], Temporal MSE: [(0, 2.7166684049007017e-06), (1, 2.9786131108267e-06), (2, 3.2343532438972034e-06), (3, 1.214797066495521e-05), (4, 2.018865052377805e-05), (5, 9.364014840684831e-05), (6, 0.00021800334798172116), (7, 0.00010612310143187642), (8, 6.729623419232666e-05), (9, 0.00014023798576090485), (10, 0.0003428003692533821), (11, 0.0005206741625443101), (12, 0.0005688740056939423), (13, 0.0008466331637464464), (14, 0.0017312340205535293), (15, 0.0024894827511161566), (16, 0.0034230726305395365), (17, 0.0036202664487063885), (18, 0.0036589456722140312), (19, 0.007800656370818615), (20, 0.011259072460234165), (21, 0.010997534729540348), (22, 0.0034605697728693485), (23, 0.0048594288527965546), (24, 0.012305773794651031), (25, 0.021976495161652565), (26, 0.0027790626045316458), (27, 0.0001807170920073986)]
+
+ 30%|███       | 45/150 [07:21<13:51,  7.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 624, Spatial MSE: [(0, 2.0023244360345416e-06), (1, 1.6024838259909302e-05), (2, 8.782800432527438e-05), (3, 2.1927606212557293e-05), (4, 4.2392830437165685e-06), (5, 2.8397757887432817e-06), (6, 1.464349861635128e-05), (7, 4.266700216248864e-06), (8, 3.2664070204191376e-06), (9, 4.732147772301687e-06), (10, 1.3012449926463887e-05), (11, 6.912205572007224e-05), (12, 0.00017333027790300548), (13, 0.004652602132409811), (14, 0.004859662614762783), (15, 0.0015942254103720188), (16, 0.04886053875088692), (17, 0.00018351111793890595), (18, 0.0003481286112219095), (19, 0.0032130125910043716), (20, 0.0016570227453485131), (21, 0.00020031683379784226), (22, 0.0007324274047277868), (23, 0.0003039719886146486), (24, 0.0006706197746098042), (25, 0.0012648145202547312), (26, 0.0012723914114758372), (27, 0.04363945499062538)], Temporal MSE: [(0, 1.956788992174552e-06), (1, 7.348411941165978e-07), (2, 1.0046196621260606e-06), (3, 8.455199917989376e-07), (4, 4.886979922957835e-07), (5, 1.0791121667352854e-06), (6, 8.844005492392171e-07), (7, 1.1581718126763008e-06), (8, 6.549431645908044e-07), (9, 1.0234587080049096e-06), (10, 5.465323738462757e-06), (11, 0.00015188161341939121), (12, 9.605450031813234e-05), (13, 0.00016286640311591327), (14, 8.02518188720569e-05), (15, 7.193411147454754e-05), (16, 4.985927080269903e-05), (17, 3.7594545574393123e-05), (18, 9.867526387097314e-05), (19, 6.577712338184938e-05), (20, 5.5587090173503384e-05), (21, 9.750431490829214e-05), (22, 0.00015185076335910708), (23, 0.0003323492710478604), (24, 0.0003635866742115468), (25, 0.0006654175813309848), (26, 0.0001476890902267769), (27, 0.00010578623914625496)]
+
+ 31%|███       | 46/150 [07:29<13:47,  7.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 618, Spatial MSE: [(0, 1.8592332935440936e-06), (1, 1.9496488675940782e-05), (2, 0.0001259812415810302), (3, 1.6249525288003497e-05), (4, 6.9788488872291055e-06), (5, 6.683336778223747e-06), (6, 8.580896064813714e-06), (7, 1.089166289602872e-05), (8, 1.4375259524967987e-05), (9, 2.987524203490466e-05), (10, 4.620845356839709e-05), (11, 0.00011188683129148558), (12, 0.00022856879513710737), (13, 0.004933890886604786), (14, 0.005534288939088583), (15, 0.0021545104682445526), (16, 0.06428015232086182), (17, 0.00024716753978282213), (18, 0.0005088080652058125), (19, 0.0041791461408138275), (20, 0.002667716471478343), (21, 0.0003650765575002879), (22, 0.0009217736660502851), (23, 0.00044337118742987514), (24, 0.0011322835925966501), (25, 0.002021662425249815), (26, 0.0022453302517533302), (27, 0.040604881942272186)], Temporal MSE: [(0, 1.5493427554247319e-06), (1, 7.676840141357388e-07), (2, 1.933678959176177e-06), (3, 2.0546412997646257e-06), (4, 3.0274618438852485e-06), (5, 2.4309824766532984e-06), (6, 3.713907517521875e-06), (7, 2.403608050371986e-06), (8, 3.1819595278648194e-06), (9, 3.4884292290371377e-06), (10, 9.244326065527275e-06), (11, 0.00016479473561048508), (12, 0.0001115333434427157), (13, 0.0001786012144293636), (14, 9.988883539335802e-05), (15, 9.400479029864073e-05), (16, 6.772482447559014e-05), (17, 5.342630538507365e-05), (18, 0.00012465089093893766), (19, 0.00011177642591064796), (20, 0.00012615327432285994), (21, 0.00017894171469379216), (22, 0.0002242261980427429), (23, 0.0004068411944899708), (24, 0.0005953055806457996), (25, 0.0011776952305808663), (26, 0.00022990205616224557), (27, 0.000107595682493411)]
+
+ 31%|███▏      | 47/150 [07:37<13:41,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 612, Spatial MSE: [(0, 1.1188325288458145e-06), (1, 1.0193499292654451e-05), (2, 5.1027116569457576e-05), (3, 1.662115755607374e-05), (4, 3.2043938062997768e-06), (5, 1.9130641248921165e-06), (6, 1.298955157835735e-05), (7, 3.6917563193128444e-06), (8, 2.359034851906472e-06), (9, 3.5503408071235754e-06), (10, 7.928518243716098e-06), (11, 5.473335841088556e-05), (12, 0.00013332389062270522), (13, 0.0033930176869034767), (14, 0.003294080263003707), (15, 0.0010085884714499116), (16, 0.03371405601501465), (17, 0.000129713153000921), (18, 0.00022727255418431014), (19, 0.0022636910434812307), (20, 0.0012506564380601048), (21, 0.00011151475337101147), (22, 0.00040919616003520787), (23, 0.00018901083967648447), (24, 0.00044021240319125354), (25, 0.000857932202052325), (26, 0.0009456215775571764), (27, 0.029085177928209305)], Temporal MSE: [(0, 1.2461757705750642e-06), (1, 5.212901896811672e-07), (2, 6.318981604636065e-07), (3, 6.089908310968895e-07), (4, 3.6242545320419595e-07), (5, 7.611504884152964e-07), (6, 6.181172125252488e-07), (7, 9.240815188604756e-07), (8, 4.6865432068443624e-07), (9, 7.703241067247291e-07), (10, 3.6037881727679633e-06), (11, 0.00014275321154855192), (12, 6.925029447302222e-05), (13, 0.00010727092740125954), (14, 4.9566951929591596e-05), (15, 5.330838757799938e-05), (16, 3.112117701675743e-05), (17, 2.727402898017317e-05), (18, 6.508683145511895e-05), (19, 4.530635487753898e-05), (20, 3.730175012606196e-05), (21, 6.338686216622591e-05), (22, 9.678454807726666e-05), (23, 0.00021055778779555112), (24, 0.00022813670511823148), (25, 0.00043972692219540477), (26, 9.664969547884539e-05), (27, 7.422945782309398e-05)]
+
+ 32%|███▏      | 48/150 [07:45<13:30,  7.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 606, Spatial MSE: [(0, 1.51859319430514e-06), (1, 3.061093229916878e-05), (2, 0.00017724078497849405), (3, 3.1095776648726314e-05), (4, 2.460839823470451e-05), (5, 4.395833457238041e-05), (6, 0.00022763364540878683), (7, 0.00023985707957763225), (8, 0.00045915611553937197), (9, 0.0010043698130175471), (10, 0.0018724616384133697), (11, 0.003117889165878296), (12, 0.0024394069332629442), (13, 0.007954287342727184), (14, 0.014725538901984692), (15, 0.013517620973289013), (16, 0.12513820827007294), (17, 0.013755648396909237), (18, 0.01909947767853737), (19, 0.04528243467211723), (20, 0.05368524417281151), (21, 0.05930311605334282), (22, 0.02336551435291767), (23, 0.029650405049324036), (24, 0.0781421884894371), (25, 0.17825336754322052), (26, 0.03871006891131401), (27, 0.03195373713970184)], Temporal MSE: [(0, 2.010376647376688e-06), (1, 1.7136563883468625e-06), (2, 2.6614691250870237e-06), (3, 6.965320608287584e-06), (4, 1.2828731087211054e-05), (5, 5.26063704455737e-05), (6, 0.00011957322567468509), (7, 9.599140321370214e-05), (8, 0.00011126910976599902), (9, 0.0002513637882657349), (10, 0.0005708259413950145), (11, 0.0008649015799164772), (12, 0.0008353966986760497), (13, 0.001199756865389645), (14, 0.0029263533651828766), (15, 0.004315362311899662), (16, 0.006596988998353481), (17, 0.008335360325872898), (18, 0.010395859368145466), (19, 0.019315345212817192), (20, 0.031142670661211014), (21, 0.04142387583851814), (22, 0.011728846468031406), (23, 0.019859518855810165), (24, 0.05844780057668686), (25, 0.13616447150707245), (26, 0.006577948108315468), (27, 0.00018133086268790066)]
+
+ 33%|███▎      | 49/150 [07:54<13:45,  8.18s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 600, Spatial MSE: [(0, 1.69224676938029e-06), (1, 1.4509770153381396e-05), (2, 7.089992141118273e-05), (3, 2.1043442757218145e-05), (4, 4.291544883017195e-06), (5, 2.5295853447460104e-06), (6, 1.4351347999763675e-05), (7, 4.703792910731863e-06), (8, 3.031150527021964e-06), (9, 4.414378508954542e-06), (10, 6.15870294495835e-06), (11, 1.118785803555511e-05), (12, 2.6808229449670762e-05), (13, 0.0007200039690360427), (14, 0.001371090766042471), (15, 0.0004573567130137235), (16, 0.024348335340619087), (17, 0.0001344933989457786), (18, 0.00010904349619522691), (19, 0.0019309341441839933), (20, 0.001222600694745779), (21, 0.00011090315820183605), (22, 0.0003391237696632743), (23, 0.00014803760859649628), (24, 0.00030395700014196336), (25, 0.000716489099431783), (26, 0.001014662440866232), (27, 0.015594209544360638)], Temporal MSE: [(0, 1.703210614323325e-06), (1, 6.807550221310521e-07), (2, 8.59495855820569e-07), (3, 8.182537385437172e-07), (4, 4.899699774796318e-07), (5, 9.331861861028301e-07), (6, 8.264981943284511e-07), (7, 1.4379155572896707e-06), (8, 6.503002509816724e-07), (9, 9.235976108357136e-07), (10, 2.281818069604924e-06), (11, 9.70394739852054e-06), (12, 1.9804458133876324e-05), (13, 2.993597081513144e-05), (14, 2.5823444957495667e-05), (15, 5.334938759915531e-05), (16, 3.001456570927985e-05), (17, 2.302259781572502e-05), (18, 5.325298843672499e-05), (19, 4.061164509039372e-05), (20, 3.132841447950341e-05), (21, 7.387486402876675e-05), (22, 8.653499389765784e-05), (23, 0.00014451249444391578), (24, 0.00013170274905860424), (25, 0.0003572306304704398), (26, 6.0188627685420215e-05), (27, 4.8173536924878135e-05)]
+
+ 33%|███▎      | 50/150 [08:01<13:19,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 594, Spatial MSE: [(0, 1.6660354731357074e-06), (1, 1.2676918231591117e-05), (2, 9.695456537883729e-05), (3, 1.4601990187657066e-05), (4, 6.665174169029342e-06), (5, 6.4840519371500704e-06), (6, 8.41239670990035e-06), (7, 1.1838605132652447e-05), (8, 1.6301130017382093e-05), (9, 3.669576472020708e-05), (10, 5.109918492962606e-05), (11, 5.9603480622172356e-05), (12, 8.527017780579627e-05), (13, 0.000846075068693608), (14, 0.002103544771671295), (15, 0.0016654912615194917), (16, 0.04507740959525108), (17, 0.0002451422333251685), (18, 0.0002041065163211897), (19, 0.0027659728657454252), (20, 0.0025077215395867825), (21, 0.00047389150131493807), (22, 0.000554922386072576), (23, 0.00038120962562970817), (24, 0.0010177252115681767), (25, 0.002078377641737461), (26, 0.0023751906119287014), (27, 0.012685240246355534)], Temporal MSE: [(0, 1.07158382434136e-06), (1, 5.630710688819818e-07), (2, 1.5864897022765945e-06), (3, 1.8881768255596398e-06), (4, 3.1079680411494337e-06), (5, 2.337323621759424e-06), (6, 3.135531187581364e-06), (7, 2.664604608071386e-06), (8, 3.3997591799561633e-06), (9, 3.6456697216635803e-06), (10, 6.538555680890568e-06), (11, 1.5574321878375486e-05), (12, 2.5076682504732162e-05), (13, 4.193162385490723e-05), (14, 5.765089008491486e-05), (15, 6.770154141122475e-05), (16, 5.130728459334932e-05), (17, 4.159924719715491e-05), (18, 8.019018423510715e-05), (19, 9.126980876317248e-05), (20, 0.00011559980339370668), (21, 0.00016357317508663982), (22, 0.0001727298367768526), (23, 0.00023739882453810424), (24, 0.0004568457370623946), (25, 0.0012878937413915992), (26, 0.00017143646255135536), (27, 4.783174881595187e-05)]
+
+ 34%|███▍      | 51/150 [08:09<12:56,  7.85s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 588, Spatial MSE: [(0, 2.19560797631857e-06), (1, 1.0872981874854304e-05), (2, 5.474659337778576e-05), (3, 1.859035728557501e-05), (4, 3.476002802926814e-06), (5, 2.2064575659896946e-06), (6, 1.4459245903708506e-05), (7, 4.846851425099885e-06), (8, 2.810033720379579e-06), (9, 3.675141897474532e-06), (10, 6.42635041003814e-06), (11, 2.8878594093839638e-05), (12, 8.208615327021107e-05), (13, 0.001943035633303225), (14, 0.0017923533450812101), (15, 0.0006780829862691462), (16, 0.01996741257607937), (17, 9.132541163126007e-05), (18, 0.00017344603838864714), (19, 0.0013724808814004064), (20, 0.0008219632436521351), (21, 0.00027176860021427274), (22, 0.0003633280284702778), (23, 0.00014583527809008956), (24, 0.00031681975815445185), (25, 0.0006660689832642674), (26, 0.0008503554854542017), (27, 0.01701895333826542)], Temporal MSE: [(0, 1.9632864223240176e-06), (1, 6.058755843696417e-07), (2, 7.390578957711114e-07), (3, 6.986137464082276e-07), (4, 4.370594979263842e-07), (5, 8.100702189040021e-07), (6, 7.273275173247384e-07), (7, 1.3451583527057664e-06), (8, 5.657940391756711e-07), (9, 8.465814289593254e-07), (10, 2.8444526378734736e-06), (11, 7.508220733143389e-05), (12, 4.6593500883318484e-05), (13, 4.6832545194774866e-05), (14, 3.072758772759698e-05), (15, 3.401321009732783e-05), (16, 1.7033626136253588e-05), (17, 1.7504435163573362e-05), (18, 4.320959124015644e-05), (19, 3.1740175472805277e-05), (20, 2.530139681766741e-05), (21, 6.926758214831352e-05), (22, 9.00794766494073e-05), (23, 0.00015639167395420372), (24, 0.00012783003330696374), (25, 0.00037536965101026), (26, 6.127735832706094e-05), (27, 4.901807551505044e-05)]
+
+ 35%|███▍      | 52/150 [08:17<13:13,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 582, Spatial MSE: [(0, 1.889274926725193e-06), (1, 2.690519977477379e-05), (2, 0.00017718055460136384), (3, 2.853150544979144e-05), (4, 2.6415884349262342e-05), (5, 5.5546221119584516e-05), (6, 0.00036188994999974966), (7, 0.00044697299017570913), (8, 0.0007839358877390623), (9, 0.0020646261982619762), (10, 0.0029719150625169277), (11, 0.00353927630931139), (12, 0.0028082295320928097), (13, 0.01024776604026556), (14, 0.023220261558890343), (15, 0.025077341124415398), (16, 0.24468712508678436), (17, 0.016131462529301643), (18, 0.013621103018522263), (19, 0.032769575715065), (20, 0.039459291845560074), (21, 0.051076993346214294), (22, 0.013438595458865166), (23, 0.019648103043437004), (24, 0.04552464932203293), (25, 0.07505351305007935), (26, 0.037636976689100266), (27, 0.019644083455204964)], Temporal MSE: [(0, 1.8400438648313866e-06), (1, 1.3766152733296622e-06), (2, 3.1167683118837886e-06), (3, 8.848843208397739e-06), (4, 1.6823929399834014e-05), (5, 6.039638537913561e-05), (6, 0.00019568114657886326), (7, 0.00012619595509022474), (8, 0.00014837704657111317), (9, 0.00028801895678043365), (10, 0.00075735297286883), (11, 0.0010637604864314198), (12, 0.0011601522564888), (13, 0.0020291574764996767), (14, 0.005251897498965263), (15, 0.008322070352733135), (16, 0.010147708468139172), (17, 0.009945094585418701), (18, 0.008200827054679394), (19, 0.015019666403532028), (20, 0.027706418186426163), (21, 0.027364859357476234), (22, 0.007136629894375801), (23, 0.010546679608523846), (24, 0.023779943585395813), (25, 0.03782965615391731), (26, 0.005745608359575272), (27, 0.00017580816347617656)]
+
+ 35%|███▌      | 53/150 [08:25<12:44,  7.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 576, Spatial MSE: [(0, 1.2960133517481154e-06), (1, 1.0791476597660221e-05), (2, 4.696126052294858e-05), (3, 1.4179593563312665e-05), (4, 2.696033106985851e-06), (5, 1.677888917583914e-06), (6, 8.263764357252512e-06), (7, 4.5456176849256735e-06), (8, 2.288649056936265e-06), (9, 2.879887688322924e-06), (10, 6.070067229302367e-06), (11, 1.5013941265351605e-05), (12, 3.740834290510975e-05), (13, 0.0007838095771148801), (14, 0.0011855160119011998), (15, 0.0004152285400778055), (16, 0.017092468217015266), (17, 8.48977651912719e-05), (18, 0.00010432238195789978), (19, 0.001138511928729713), (20, 0.0008147775661200285), (21, 0.00040376579272560775), (22, 0.0003652064478956163), (23, 0.00012544529454316944), (24, 0.00028011391987092793), (25, 0.0006271882448345423), (26, 0.000724039797205478), (27, 0.011241361498832703)], Temporal MSE: [(0, 1.1765117733375519e-06), (1, 4.5052630071040767e-07), (2, 5.900085966459301e-07), (3, 5.720575018131058e-07), (4, 3.7593298429783317e-07), (5, 6.479335183939838e-07), (6, 6.542775281559443e-07), (7, 1.03655941074976e-06), (8, 4.990732804799336e-07), (9, 7.739268994555459e-07), (10, 3.5418452171143144e-06), (11, 2.372776361880824e-05), (12, 2.014242454606574e-05), (13, 2.709713407966774e-05), (14, 2.1163681594771333e-05), (15, 3.309675958007574e-05), (16, 2.075536576739978e-05), (17, 2.581319677119609e-05), (18, 3.69400622730609e-05), (19, 2.9669739888049662e-05), (20, 2.9502061806851998e-05), (21, 7.787167851347476e-05), (22, 9.212976146955043e-05), (23, 0.00014516491501126438), (24, 0.00010815718997037038), (25, 0.00026703960611484945), (26, 4.448637628229335e-05), (27, 3.989541073678993e-05)]
+
+ 36%|███▌      | 54/150 [08:32<12:18,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 570, Spatial MSE: [(0, 1.5060275018186076e-06), (1, 1.3409206985670608e-05), (2, 7.967997953528538e-05), (3, 1.4139061931928154e-05), (4, 6.443542588385753e-06), (5, 6.04978004048462e-06), (6, 8.644318768347148e-06), (7, 1.6215364667004906e-05), (8, 2.3952907213242725e-05), (9, 5.6751352531136945e-05), (10, 0.00010863792704185471), (11, 0.00013454620784614235), (12, 0.00014309473044704646), (13, 0.0009564859792590141), (14, 0.0016559903742745519), (15, 0.0009307902073487639), (16, 0.030881043523550034), (17, 0.0001639976690057665), (18, 0.0009988034144043922), (19, 0.004109679721295834), (20, 0.00395498750731349), (21, 0.0008870561723597348), (22, 0.0006055687554180622), (23, 0.000346346409060061), (24, 0.0011154765961691737), (25, 0.0016759237041696906), (26, 0.002414802322164178), (27, 0.0102834552526474)], Temporal MSE: [(0, 1.0829753591679037e-06), (1, 5.579183266490872e-07), (2, 1.4573076896340353e-06), (3, 1.835480475165241e-06), (4, 2.762179065030068e-06), (5, 2.195775323343696e-06), (6, 3.3890105441969354e-06), (7, 3.03952560898324e-06), (8, 3.4647289339773124e-06), (9, 4.990761681256117e-06), (10, 1.3788422620564234e-05), (11, 3.9205242501338944e-05), (12, 3.704451592057012e-05), (13, 5.245515058049932e-05), (14, 6.258049688767642e-05), (15, 6.955490971449763e-05), (16, 4.8788806452648714e-05), (17, 4.5521865104092285e-05), (18, 8.221480675274506e-05), (19, 0.00011257743608439341), (20, 0.00017516415391582996), (21, 0.00018660820205695927), (22, 0.00018077272397931665), (23, 0.00023116472584661096), (24, 0.0004331254167482257), (25, 0.0008829791331663728), (26, 0.0001642883726162836), (27, 4.356797580840066e-05)]
+
+ 37%|███▋      | 55/150 [08:40<12:19,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 564, Spatial MSE: [(0, 1.3760994761469192e-06), (1, 8.35775335872313e-06), (2, 3.950445170630701e-05), (3, 1.2713730939140078e-05), (4, 2.5018618998728925e-06), (5, 1.7571211401445908e-06), (6, 9.234036042471416e-06), (7, 3.4886677440226777e-06), (8, 2.149611873392132e-06), (9, 2.5707611257530516e-06), (10, 3.426453986321576e-06), (11, 6.241984465304995e-06), (12, 2.1058731363154948e-05), (13, 0.000626422232016921), (14, 0.0009657691698521376), (15, 0.0002802105445880443), (16, 0.02325914427638054), (17, 0.00011112309584859759), (18, 0.00011730096593964845), (19, 0.0019357736455276608), (20, 0.0014642898458987474), (21, 0.00023029523435980082), (22, 0.0002489581529516727), (23, 0.00010210311302216724), (24, 0.0002482210984453559), (25, 0.0006650079158134758), (26, 0.0006545434589497745), (27, 0.01202639751136303)], Temporal MSE: [(0, 1.3275862329464871e-06), (1, 4.5456891939466004e-07), (2, 6.148092666080629e-07), (3, 5.556252062888234e-07), (4, 3.5468008263705997e-07), (5, 6.375087764354248e-07), (6, 6.149937803456851e-07), (7, 1.059186615748331e-06), (8, 4.6728868596801476e-07), (9, 6.452411298596417e-07), (10, 1.4625785524913226e-06), (11, 1.0560405826254282e-05), (12, 1.559004158480093e-05), (13, 2.0029225197504275e-05), (14, 1.819265708036255e-05), (15, 3.9971157093532383e-05), (16, 2.2405774871003814e-05), (17, 3.267716965638101e-05), (18, 4.451393033377826e-05), (19, 3.8108388253021985e-05), (20, 2.951488204416819e-05), (21, 8.489564788760617e-05), (22, 8.099971455521882e-05), (23, 0.00012350734323263168), (24, 9.148914250545204e-05), (25, 0.0002618562721181661), (26, 3.873049354297109e-05), (27, 5.867809522897005e-05)]
+
+ 37%|███▋      | 56/150 [08:47<11:38,  7.44s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 558, Spatial MSE: [(0, 1.4484073744824855e-06), (1, 2.4764540285104886e-05), (2, 0.0001339315640507266), (3, 2.4488432245561853e-05), (4, 3.0272080039139837e-05), (5, 5.078477624920197e-05), (6, 0.0001810376561479643), (7, 0.0002937988901976496), (8, 0.000648022280074656), (9, 0.0013894460862502456), (10, 0.0015267011476680636), (11, 0.0018517164280638099), (12, 0.0018641835777089), (13, 0.004273183178156614), (14, 0.010244724340736866), (15, 0.01004011183977127), (16, 0.12155360728502274), (17, 0.0056572104804217815), (18, 0.006128151901066303), (19, 0.013786072842776775), (20, 0.01762961782515049), (21, 0.01717594638466835), (22, 0.009072527289390564), (23, 0.012122852727770805), (24, 0.025218892842531204), (25, 0.038508087396621704), (26, 0.022707978263497353), (27, 0.013971238397061825)], Temporal MSE: [(0, 1.5169978269113926e-06), (1, 1.0972006521114963e-06), (2, 2.536639385652961e-06), (3, 1.072690611181315e-05), (4, 1.781815262802411e-05), (5, 2.8465050490922295e-05), (6, 8.908096060622483e-05), (7, 9.886381303658709e-05), (8, 0.0001334070839220658), (9, 0.0002795909240376204), (10, 0.0007283592130988836), (11, 0.0007171824108809233), (12, 0.000657995231449604), (13, 0.0010644609574228525), (14, 0.0020625763572752476), (15, 0.002142605371773243), (16, 0.00261069112457335), (17, 0.0030478376429528), (18, 0.0035413966979831457), (19, 0.005029656924307346), (20, 0.010442741215229034), (21, 0.008598222397267818), (22, 0.0036753329914063215), (23, 0.0055681574158370495), (24, 0.01342672761529684), (25, 0.023375997319817543), (26, 0.003719835076481104), (27, 0.00010383210610598326)]
+
+ 38%|███▊      | 57/150 [08:55<11:48,  7.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 552, Spatial MSE: [(0, 1.2209826536491164e-06), (1, 9.549660717311781e-06), (2, 4.216561865177937e-05), (3, 1.3723030861001462e-05), (4, 2.7836515528179007e-06), (5, 1.7737810367179918e-06), (6, 7.98190558271017e-06), (7, 4.350928975327406e-06), (8, 2.3702511953160865e-06), (9, 2.971549747599056e-06), (10, 5.0442718020349275e-06), (11, 8.531668754585553e-06), (12, 1.9668041204568e-05), (13, 0.0004276941472198814), (14, 0.001054822001606226), (15, 0.0003683477989397943), (16, 0.020813899114727974), (17, 0.00010494941670913249), (18, 9.555578435538337e-05), (19, 0.0017491382313892245), (20, 0.001196890021674335), (21, 0.00028996155015192926), (22, 0.0002659452729858458), (23, 0.00010664734145393595), (24, 0.0002424774138489738), (25, 0.0007358206203207374), (26, 0.0007309477077797055), (27, 0.011569011025130749)], Temporal MSE: [(0, 1.2254271268830053e-06), (1, 4.5056444264446327e-07), (2, 6.090144211157167e-07), (3, 6.558095151376619e-07), (4, 4.4447997993302124e-07), (5, 7.170441449488862e-07), (6, 7.902548873062187e-07), (7, 1.2119390930820373e-06), (8, 5.598162715614308e-07), (9, 7.780109285704384e-07), (10, 2.7932946977671236e-06), (11, 7.670676495763473e-06), (12, 1.1309014553262386e-05), (13, 1.8302660464541987e-05), (14, 2.232516635558568e-05), (15, 4.536624328466132e-05), (16, 2.1276602637954056e-05), (17, 2.2370446458808146e-05), (18, 3.4131902793888e-05), (19, 2.762794611044228e-05), (20, 2.3226770281326026e-05), (21, 7.238279067678377e-05), (22, 8.762629295233637e-05), (23, 0.00014310635742731392), (24, 8.999096462503076e-05), (25, 0.000278171559330076), (26, 4.447375249583274e-05), (27, 4.72220235678833e-05)]
+
+ 39%|███▊      | 58/150 [09:01<11:16,  7.35s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 546, Spatial MSE: [(0, 1.4302963791124057e-06), (1, 1.484045606048312e-05), (2, 8.421638631261885e-05), (3, 1.3242710338090546e-05), (4, 6.199636573001044e-06), (5, 5.938150479778415e-06), (6, 1.0825031495187432e-05), (7, 1.1764892406063154e-05), (8, 1.6529031199752353e-05), (9, 3.870215732604265e-05), (10, 5.91400676057674e-05), (11, 6.221812509465963e-05), (12, 0.00010010580444941297), (13, 0.0004914072924293578), (14, 0.0013656484661623836), (15, 0.000764557917136699), (16, 0.03170914202928543), (17, 0.00017397734336555004), (18, 0.00015984912170097232), (19, 0.0024760160595178604), (20, 0.0020631516817957163), (21, 0.0004292628727853298), (22, 0.00040842569433152676), (23, 0.0002537265245337039), (24, 0.0006726174033246934), (25, 0.0014700901228934526), (26, 0.0026612505316734314), (27, 0.01004176214337349)], Temporal MSE: [(0, 1.0896234243773506e-06), (1, 5.713094992643164e-07), (2, 1.3855230918125017e-06), (3, 1.7998436305788346e-06), (4, 2.872415507226833e-06), (5, 2.247366637675441e-06), (6, 2.9883283332310384e-06), (7, 3.0916876312403474e-06), (8, 3.714747890626313e-06), (9, 4.262884431227576e-06), (10, 9.187644536723383e-06), (11, 2.218013469246216e-05), (12, 2.7960595616605133e-05), (13, 3.1825515179662034e-05), (14, 4.857009116676636e-05), (15, 6.566179945366457e-05), (16, 4.1364546632394195e-05), (17, 3.842472506221384e-05), (18, 5.6440741900587454e-05), (19, 5.955660526524298e-05), (20, 8.25918250484392e-05), (21, 0.00013227970339357853), (22, 0.00014885330165270716), (23, 0.0002034307981375605), (24, 0.0003160447522532195), (25, 0.0006962674669921398), (26, 0.00014554039807990193), (27, 5.225011773291044e-05)]
+
+ 39%|███▉      | 59/150 [09:09<11:15,  7.42s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 540, Spatial MSE: [(0, 1.1681805744956364e-06), (1, 8.031129254959524e-06), (2, 3.706372808665037e-05), (3, 9.774624231795315e-06), (4, 2.0814748040720588e-06), (5, 1.4060275361771346e-06), (6, 5.398650955612538e-06), (7, 3.1116983336687554e-06), (8, 1.9489641545078484e-06), (9, 2.333089923922671e-06), (10, 5.974105079076253e-06), (11, 1.2368006537144538e-05), (12, 2.7489257263368927e-05), (13, 0.0004087571578565985), (14, 0.0013104439713060856), (15, 0.0005121846916154027), (16, 0.03164524585008621), (17, 0.00014572804502677172), (18, 0.0006072984542697668), (19, 0.002373451367020607), (20, 0.0016270497580990195), (21, 0.00044469974818639457), (22, 0.00040016157436184585), (23, 0.00014021758397575468), (24, 0.0003177930193487555), (25, 0.0008973796502687037), (26, 0.0006408147164620459), (27, 0.01170277874916792)], Temporal MSE: [(0, 9.707169965622597e-07), (1, 3.519273832353065e-07), (2, 5.345460749595077e-07), (3, 5.046570095146308e-07), (4, 3.640259933490597e-07), (5, 5.545476255974791e-07), (6, 6.029252972439281e-07), (7, 9.10639471385366e-07), (8, 4.5445392515830463e-07), (9, 6.545905648636108e-07), (10, 3.326349542476237e-06), (11, 1.182810592581518e-05), (12, 2.0524090359685943e-05), (13, 1.7224438124685548e-05), (14, 2.7078294806415215e-05), (15, 4.0172551962314174e-05), (16, 2.4056102120084688e-05), (17, 2.6846943001146428e-05), (18, 4.9834994570119306e-05), (19, 3.8262125599430874e-05), (20, 3.2243551686406136e-05), (21, 0.00010551774175837636), (22, 0.00013601618411485106), (23, 0.00020986649906262755), (24, 0.0001251079811481759), (25, 0.00026289484230801463), (26, 5.027138104196638e-05), (27, 6.44825558993034e-05)]
+
+ 40%|████      | 60/150 [09:17<11:14,  7.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 534, Spatial MSE: [(0, 1.1518288829392986e-06), (1, 1.998790139623452e-05), (2, 9.997528832172975e-05), (3, 2.4260409190901555e-05), (4, 3.270204979344271e-05), (5, 5.306612001731992e-05), (6, 0.0002625852939672768), (7, 0.00042727647814899683), (8, 0.0007885069353505969), (9, 0.0014493827475234866), (10, 0.001710138632915914), (11, 0.002542148344218731), (12, 0.0025569975841790438), (13, 0.005794080439954996), (14, 0.01609322987496853), (15, 0.01914498582482338), (16, 0.15113651752471924), (17, 0.01624172553420067), (18, 0.012319395318627357), (19, 0.021978430449962616), (20, 0.023644153028726578), (21, 0.018224377185106277), (22, 0.011152918450534344), (23, 0.01563376560807228), (24, 0.03057846426963806), (25, 0.04657872021198273), (26, 0.026234501972794533), (27, 0.012828968465328217)], Temporal MSE: [(0, 1.132147190219257e-06), (1, 8.013496426428901e-07), (2, 2.5297563297499437e-06), (3, 7.521263796661515e-06), (4, 1.6042626157286577e-05), (5, 4.198470924166031e-05), (6, 0.00019791176600847393), (7, 0.00014540781558025628), (8, 0.0001555249182274565), (9, 0.0003293441841378808), (10, 0.0009624894591979682), (11, 0.001046172110363841), (12, 0.0012113659176975489), (13, 0.0028846992645412683), (14, 0.007214476820081472), (15, 0.009457575157284737), (16, 0.011002943851053715), (17, 0.011404615826904774), (18, 0.006205584388226271), (19, 0.008546686731278896), (20, 0.016712915152311325), (21, 0.014461042359471321), (22, 0.004940547980368137), (23, 0.007349657826125622), (24, 0.01702965423464775), (25, 0.026900138705968857), (26, 0.004321834072470665), (27, 0.0001448354305466637)]
+
+ 41%|████      | 61/150 [09:25<11:37,  7.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 528, Spatial MSE: [(0, 1.292582737733028e-06), (1, 1.2653797966777347e-05), (2, 3.911467138095759e-05), (3, 1.3290070455695968e-05), (4, 3.216256345695001e-06), (5, 1.7452239262638614e-06), (6, 8.143676495819818e-06), (7, 3.5574560115492204e-06), (8, 2.401846586508327e-06), (9, 3.409143573662732e-06), (10, 1.053155938279815e-05), (11, 4.76713030366227e-05), (12, 8.337924373336136e-05), (13, 0.001436332706362009), (14, 0.001871389918960631), (15, 0.00045855605276301503), (16, 0.027455583214759827), (17, 0.00011994548549409956), (18, 0.00012669673014897853), (19, 0.0024006874300539494), (20, 0.001679284148849547), (21, 0.0001849044783739373), (22, 0.0002716044255066663), (23, 0.00011793066369136795), (24, 0.0002960326091852039), (25, 0.0007633630884811282), (26, 0.0007494472083635628), (27, 0.017293008044362068)], Temporal MSE: [(0, 1.107474417949561e-06), (1, 4.240120858867158e-07), (2, 5.434458785202878e-07), (3, 6.039817321834562e-07), (4, 4.376681204121269e-07), (5, 6.855883611933677e-07), (6, 7.225416993605904e-07), (7, 1.2008650855932501e-06), (8, 5.818599220219767e-07), (9, 8.295965017168783e-07), (10, 6.039517757017165e-06), (11, 8.950974734034389e-05), (12, 4.1550403693690896e-05), (13, 4.168154191575013e-05), (14, 2.8989448765059933e-05), (15, 4.221700146445073e-05), (16, 2.776311885099858e-05), (17, 2.581033913884312e-05), (18, 4.449595508049242e-05), (19, 3.590190681279637e-05), (20, 3.053954060305841e-05), (21, 8.75972182257101e-05), (22, 0.00010366958304075524), (23, 0.000160863550263457), (24, 0.00013135600602254272), (25, 0.0002762002986855805), (26, 6.725026469212025e-05), (27, 6.690830923616886e-05)]
+
+ 41%|████▏     | 62/150 [09:33<11:24,  7.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 522, Spatial MSE: [(0, 1.3206708899815567e-06), (1, 1.2659445019380655e-05), (2, 6.839634443167597e-05), (3, 1.128635722125182e-05), (4, 5.5658642850175966e-06), (5, 6.024126378179062e-06), (6, 1.075183718057815e-05), (7, 1.0918791303993203e-05), (8, 1.7180695067509077e-05), (9, 3.909998122253455e-05), (10, 6.611699791392311e-05), (11, 9.93210487649776e-05), (12, 0.00016382787725888193), (13, 0.0014970464399084449), (14, 0.0020287407096475363), (15, 0.0006689233123324811), (16, 0.03615338355302811), (17, 0.00017771526472643018), (18, 0.00020840720389969647), (19, 0.0028789627831429243), (20, 0.0023250372614711523), (21, 0.00032945245038717985), (22, 0.0003885440528392792), (23, 0.00024566648062318563), (24, 0.0006531557301059365), (25, 0.0014293473213911057), (26, 0.003373044077306986), (27, 0.016171708703041077)], Temporal MSE: [(0, 9.346323963654868e-07), (1, 5.31168211637123e-07), (2, 1.2214081834827084e-06), (3, 1.6806343410280533e-06), (4, 2.7487142233439954e-06), (5, 2.1623686734528746e-06), (6, 2.9921266104793176e-06), (7, 3.300580601717229e-06), (8, 4.090027687198017e-06), (9, 4.382271981739905e-06), (10, 1.2866858924098779e-05), (11, 0.00010066531103802845), (12, 5.500836778082885e-05), (13, 6.162019417388365e-05), (14, 5.8492420066613704e-05), (15, 6.579295586561784e-05), (16, 4.9819635023595765e-05), (17, 4.519679714576341e-05), (18, 7.014961738605052e-05), (19, 6.628742266912013e-05), (20, 7.619180541951209e-05), (21, 0.00014713878044858575), (22, 0.00015491392696276307), (23, 0.00021645904053002596), (24, 0.0003278798540122807), (25, 0.0006466286722570658), (26, 0.00016562048404011875), (27, 7.111939339665696e-05)]
+
+ 42%|████▏     | 63/150 [09:40<10:56,  7.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 516, Spatial MSE: [(0, 8.303335334858275e-07), (1, 8.27036546979798e-06), (2, 3.053407999686897e-05), (3, 9.232284355675802e-06), (4, 2.1235737222013995e-06), (5, 1.3507503808796173e-06), (6, 4.874471414950676e-06), (7, 2.876501412174548e-06), (8, 1.7184622720378684e-06), (9, 2.0007619241368957e-06), (10, 4.169384283159161e-06), (11, 8.126904504024424e-06), (12, 1.7245816707145423e-05), (13, 0.0004730226646643132), (14, 0.0008438591030426323), (15, 0.0002496365341357887), (16, 0.017446447163820267), (17, 6.727897562086582e-05), (18, 6.437269621528685e-05), (19, 0.0014860164374113083), (20, 0.0010271263308823109), (21, 6.72599344397895e-05), (22, 0.00017768045654520392), (23, 8.465039718430489e-05), (24, 0.0001837357267504558), (25, 0.0004883062792941928), (26, 0.0005349629209376872), (27, 0.009853491559624672)], Temporal MSE: [(0, 8.833765150484396e-07), (1, 3.5043717616645154e-07), (2, 5.30202385107259e-07), (3, 4.993192987967632e-07), (4, 3.64894304993868e-07), (5, 5.34703588073171e-07), (6, 5.52448170765274e-07), (7, 8.532117021786689e-07), (8, 4.2406992406540667e-07), (9, 5.797749054181622e-07), (10, 2.3331933789449977e-06), (11, 7.991225174919236e-06), (12, 1.0242511052638292e-05), (13, 1.9942626749980263e-05), (14, 1.8911590814241208e-05), (15, 2.24099221668439e-05), (16, 1.4849400031380355e-05), (17, 2.0015986592625268e-05), (18, 2.7693338779499754e-05), (19, 2.4185917936847545e-05), (20, 1.985255403269548e-05), (21, 5.5298307415796444e-05), (22, 6.0634662077063695e-05), (23, 9.915989357978106e-05), (24, 8.41916844365187e-05), (25, 0.00018500603619031608), (26, 4.302619709051214e-05), (27, 4.005451773991808e-05)]
+
+ 43%|████▎     | 64/150 [09:48<11:01,  7.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 510, Spatial MSE: [(0, 1.3189435321692144e-06), (1, 1.868249091785401e-05), (2, 9.83338977675885e-05), (3, 2.1272724552545696e-05), (4, 3.1641695386497304e-05), (5, 4.361314859124832e-05), (6, 0.00015520220040343702), (7, 0.0002637701400090009), (8, 0.000608332920819521), (9, 0.0008931058691814542), (10, 0.0009813718497753143), (11, 0.0012354077771306038), (12, 0.0012768470915034413), (13, 0.0037422499153763056), (14, 0.009330936707556248), (15, 0.008206577971577644), (16, 0.10619127005338669), (17, 0.005076277535408735), (18, 0.00458489777520299), (19, 0.009800718165934086), (20, 0.009565792046487331), (21, 0.006958857178688049), (22, 0.004685965832322836), (23, 0.006746518891304731), (24, 0.013126889243721962), (25, 0.021698150783777237), (26, 0.01278010569512844), (27, 0.010924751870334148)], Temporal MSE: [(0, 1.260612179976306e-06), (1, 7.736905445199227e-07), (2, 2.6065497422678163e-06), (3, 7.562422979390249e-06), (4, 1.405626335326815e-05), (5, 2.2719052140018903e-05), (6, 9.355146175948903e-05), (7, 9.969437815016136e-05), (8, 0.0001156871221610345), (9, 0.00019411163520999253), (10, 0.0004477217444218695), (11, 0.0005326896207407117), (12, 0.0006248002755455673), (13, 0.001205936772748828), (14, 0.0026310342364013195), (15, 0.0028579693753272295), (16, 0.003123019589111209), (17, 0.0032634204253554344), (18, 0.002147321356460452), (19, 0.0029554632492363453), (20, 0.0051656789146363735), (21, 0.004266686271876097), (22, 0.0021217598114162683), (23, 0.003041396150365472), (24, 0.006776726339012384), (25, 0.011074273847043514), (26, 0.001796627533622086), (27, 8.046068251132965e-05)]
+
+ 43%|████▎     | 65/150 [09:57<11:23,  8.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 504, Spatial MSE: [(0, 1.4670988548459718e-06), (1, 8.25769984658109e-06), (2, 3.165002271998674e-05), (3, 9.722031791170593e-06), (4, 2.1470577848958783e-06), (5, 1.5296760693672695e-06), (6, 4.997162704967195e-06), (7, 3.952478891733335e-06), (8, 2.1059320260974346e-06), (9, 2.2927677036932437e-06), (10, 4.8597271415928844e-06), (11, 1.097897893487243e-05), (12, 1.2977926417079289e-05), (13, 0.00029365308000706136), (14, 0.00041808991227298975), (15, 0.00021068092610221356), (16, 0.009837746620178223), (17, 6.037815910531208e-05), (18, 4.3860331061296165e-05), (19, 0.0008865944691933692), (20, 0.000619180325884372), (21, 0.00015084611368365586), (22, 0.00017644031322561204), (23, 7.159812230383977e-05), (24, 0.0001530062290839851), (25, 0.00041708466596901417), (26, 0.0006717463838867843), (27, 0.0062464019283652306)], Temporal MSE: [(0, 1.0595769026622293e-06), (1, 3.449609664585296e-07), (2, 4.660436445647065e-07), (3, 5.365616289054742e-07), (4, 4.2410545120219467e-07), (5, 5.857087330696231e-07), (6, 6.518782242892485e-07), (7, 1.0946605470962822e-06), (8, 5.25378993643244e-07), (9, 6.744622851329041e-07), (10, 3.0959095056459773e-06), (11, 5.960209819022566e-06), (12, 8.160373909049667e-06), (13, 9.30628721107496e-06), (14, 1.6691010387148708e-05), (15, 1.089679244614672e-05), (16, 7.29798739484977e-06), (17, 1.424848051101435e-05), (18, 1.4404226021724753e-05), (19, 1.3631822184834164e-05), (20, 1.262431305804057e-05), (21, 4.110267764190212e-05), (22, 4.674524825531989e-05), (23, 8.458593947580084e-05), (24, 5.875994611415081e-05), (25, 0.00017638182907830924), (26, 3.9585989725310355e-05), (27, 2.411362947896123e-05)]
+
+ 44%|████▍     | 66/150 [10:06<11:51,  8.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 498, Spatial MSE: [(0, 1.2469895409594756e-06), (1, 1.1499049833219033e-05), (2, 4.738030111184344e-05), (3, 1.0309408025932498e-05), (4, 5.5479927141277585e-06), (5, 5.5010464166116435e-06), (6, 8.805105608189479e-06), (7, 1.288201383431442e-05), (8, 1.8881051801145077e-05), (9, 4.310919757699594e-05), (10, 7.640642434125766e-05), (11, 7.908699626568705e-05), (12, 9.563669300405309e-05), (13, 0.00036457093665376306), (14, 0.0005614953115582466), (15, 0.0002964521409012377), (16, 0.009813372045755386), (17, 0.00011334795999573544), (18, 0.0005918062524870038), (19, 0.0016853840788826346), (20, 0.0015357071533799171), (21, 0.0006184541271068156), (22, 0.00029694102704524994), (23, 0.00021768215810880065), (24, 0.0005872969049960375), (25, 0.0011135514359921217), (26, 0.002927094465121627), (27, 0.004620277788490057)], Temporal MSE: [(0, 7.568624482701125e-07), (1, 4.0353427266381914e-07), (2, 1.1170482139277738e-06), (3, 1.564493686601054e-06), (4, 2.5436784198973328e-06), (5, 2.162431883334648e-06), (6, 2.879855401261011e-06), (7, 3.2964703677862417e-06), (8, 4.080879534740234e-06), (9, 4.908790742774727e-06), (10, 1.1231411917833611e-05), (11, 1.9214381609344855e-05), (12, 2.6354060537414625e-05), (13, 3.392303187865764e-05), (14, 5.161494482308626e-05), (15, 3.68467990483623e-05), (16, 3.10997202177532e-05), (17, 3.198434569640085e-05), (18, 4.264103336026892e-05), (19, 6.0402118833735585e-05), (20, 8.982441067928448e-05), (21, 9.695887274574488e-05), (22, 9.643652447266504e-05), (23, 0.00013446994125843048), (24, 0.00024808853049762547), (25, 0.0004267288895789534), (26, 0.00015415043162647635), (27, 2.945622691186145e-05)]
+
+ 45%|████▍     | 67/150 [10:19<13:26,  9.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 492, Spatial MSE: [(0, 7.406830491163419e-07), (1, 7.852189810364507e-06), (2, 2.466823389113415e-05), (3, 7.650491170352325e-06), (4, 2.1004964310122887e-06), (5, 1.1223123692616355e-06), (6, 3.4320000850129873e-06), (7, 2.035740408246056e-06), (8, 1.585785412316909e-06), (9, 2.2477738639281597e-06), (10, 4.760002866532886e-06), (11, 1.060520935425302e-05), (12, 1.9965491446782835e-05), (13, 0.0002509615442249924), (14, 0.000768916099332273), (15, 0.0003428942000027746), (16, 0.022362112998962402), (17, 0.00010080636275233701), (18, 0.0001020037816488184), (19, 0.002030003583058715), (20, 0.001512218383140862), (21, 7.764940528431907e-05), (22, 0.00019239609537180513), (23, 7.694995292695239e-05), (24, 0.00017948400636669248), (25, 0.00048366081318818033), (26, 0.0003887782513629645), (27, 0.0095948725938797)], Temporal MSE: [(0, 7.003663995419629e-07), (1, 2.8863811962764885e-07), (2, 4.311226575737237e-07), (3, 4.46716171609296e-07), (4, 3.500994694149995e-07), (5, 4.7637701072744676e-07), (6, 5.439757728709083e-07), (7, 7.582498255942483e-07), (8, 4.254287091498554e-07), (9, 5.831449243487441e-07), (10, 2.96013786282856e-06), (11, 8.343121407961007e-06), (12, 1.1381771400920115e-05), (13, 1.1686203833960462e-05), (14, 2.2695079678669572e-05), (15, 3.847378684440628e-05), (16, 1.9505530872265808e-05), (17, 2.2343709133565426e-05), (18, 3.687129355967045e-05), (19, 2.947275061160326e-05), (20, 2.4271883376059122e-05), (21, 7.561829261248931e-05), (22, 8.059071842581034e-05), (23, 0.00011065515718655661), (24, 7.47728772694245e-05), (25, 0.00014717520389240235), (26, 3.705753988469951e-05), (27, 5.226840221439488e-05)]
+
+ 45%|████▌     | 68/150 [10:28<12:57,  9.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 486, Spatial MSE: [(0, 1.1735004363799817e-06), (1, 1.8818398530129343e-05), (2, 7.875241863075644e-05), (3, 2.128639789589215e-05), (4, 2.919710641435813e-05), (5, 4.481730502448045e-05), (6, 0.00013999550719745457), (7, 0.0002909870818257332), (8, 0.000564204587135464), (9, 0.0007428042008541524), (10, 0.0010118433274328709), (11, 0.0011586603941395879), (12, 0.0011624894104897976), (13, 0.00254162959754467), (14, 0.0067385598085820675), (15, 0.006803139578551054), (16, 0.06331629306077957), (17, 0.0059198844246566296), (18, 0.005400612484663725), (19, 0.00825202651321888), (20, 0.008518155664205551), (21, 0.007850096561014652), (22, 0.005533477757126093), (23, 0.009038438089191914), (24, 0.01676284335553646), (25, 0.02735617198050022), (26, 0.01895345002412796), (27, 0.010507134720683098)], Temporal MSE: [(0, 1.1655860134851537e-06), (1, 7.499314733649953e-07), (2, 2.4420189674856374e-06), (3, 7.125459887902252e-06), (4, 1.4149167327559553e-05), (5, 2.5745895982254297e-05), (6, 0.0001311756786890328), (7, 0.00011728922982001677), (8, 0.00011538444960024208), (9, 0.0001840795885073021), (10, 0.00044254414387978613), (11, 0.0005029152380302548), (12, 0.0005902437842451036), (13, 0.0012253066524863243), (14, 0.0028745944146066904), (15, 0.0032009982969611883), (16, 0.0037071050610393286), (17, 0.004150633234530687), (18, 0.0024887111503630877), (19, 0.0037933068815618753), (20, 0.007312153000384569), (21, 0.006583233363926411), (22, 0.0025776904076337814), (23, 0.0038620405830442905), (24, 0.008681679144501686), (25, 0.013255209662020206), (26, 0.0020594687666743994), (27, 0.00010615536302793771)]
+
+ 46%|████▌     | 69/150 [10:39<13:20,  9.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 480, Spatial MSE: [(0, 8.423578492511297e-07), (1, 9.631057764636353e-06), (2, 2.6289228117093444e-05), (3, 8.413840987486765e-06), (4, 2.4142600523191504e-06), (5, 1.2358365211184719e-06), (6, 3.586384309528512e-06), (7, 2.484437345628976e-06), (8, 1.6907791859921417e-06), (9, 2.7652617973217275e-06), (10, 1.0082798326038755e-05), (11, 4.478649498196319e-05), (12, 9.515330020803958e-05), (13, 0.001967706484720111), (14, 0.002377438126131892), (15, 0.0006744155543856323), (16, 0.03034951724112034), (17, 0.0001237234682776034), (18, 0.00037787450128234923), (19, 0.002123201498761773), (20, 0.0013874716823920608), (21, 0.0001878153852885589), (22, 0.00036658684257417917), (23, 0.0001424988586222753), (24, 0.00032587433815933764), (25, 0.000725753081496805), (26, 0.0005973080405965447), (27, 0.01794155314564705)], Temporal MSE: [(0, 7.161077064665733e-07), (1, 2.85064515992417e-07), (2, 4.1529602867740323e-07), (3, 4.950535981151916e-07), (4, 4.038432734887465e-07), (5, 5.51564028228313e-07), (6, 5.775927434115147e-07), (7, 8.295603493024828e-07), (8, 5.105993636789208e-07), (9, 7.802468644513283e-07), (10, 6.904116617079126e-06), (11, 0.00010137813660549), (12, 4.9331134505337104e-05), (13, 5.266589505481534e-05), (14, 3.0949759093346074e-05), (15, 6.473198300227523e-05), (16, 3.3606596844037995e-05), (17, 3.673250103020109e-05), (18, 5.126302130520344e-05), (19, 4.213987631374039e-05), (20, 3.710484816110693e-05), (21, 7.600871322210878e-05), (22, 0.00010638112871674821), (23, 0.00020073528867214918), (24, 0.00014989732881076634), (25, 0.00031921963091008365), (26, 8.091409836197272e-05), (27, 7.619717507623136e-05)]
+
+ 47%|████▋     | 70/150 [10:47<12:32,  9.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 474, Spatial MSE: [(0, 1.0004557680076687e-06), (1, 1.0494176422071178e-05), (2, 4.275087849237025e-05), (3, 9.013612725539133e-06), (4, 4.410183009895263e-06), (5, 5.118643912283005e-06), (6, 8.495108886563685e-06), (7, 8.757886462262832e-06), (8, 1.4829609426669776e-05), (9, 2.6337796953157522e-05), (10, 4.9018020945368335e-05), (11, 8.226418867707253e-05), (12, 0.00015156930021476), (13, 0.0020319868344813585), (14, 0.002496167318895459), (15, 0.0007917991606518626), (16, 0.03005058318376541), (17, 0.00015858293045312166), (18, 0.0003207020345143974), (19, 0.00217193434946239), (20, 0.001431626151315868), (21, 0.0002226240758318454), (22, 0.0004187143931630999), (23, 0.0002354872558498755), (24, 0.0005718106986023486), (25, 0.0012255534529685974), (26, 0.002349603921175003), (27, 0.017411228269338608)], Temporal MSE: [(0, 8.04709031854145e-07), (1, 4.065583425472141e-07), (2, 1.0565886441327166e-06), (3, 1.4987059557824978e-06), (4, 2.601467485874309e-06), (5, 1.9343751773703843e-06), (6, 2.7972421321464935e-06), (7, 3.1752251743455417e-06), (8, 3.537311158652301e-06), (9, 3.832621587207541e-06), (10, 1.1676595022436231e-05), (11, 0.00010860506881726906), (12, 5.9947455156361684e-05), (13, 6.816055974923074e-05), (14, 5.576272087637335e-05), (15, 7.945876131998375e-05), (16, 4.61895797343459e-05), (17, 4.362690378911793e-05), (18, 6.543486233567819e-05), (19, 5.965377204120159e-05), (20, 6.650985596934333e-05), (21, 0.00010342367750126868), (22, 0.00013378412404563278), (23, 0.0002375217154622078), (24, 0.0002753608569037169), (25, 0.00047418635222129524), (26, 0.0001869785919552669), (27, 8.240544411819428e-05)]
+
+ 47%|████▋     | 71/150 [10:55<11:47,  8.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 468, Spatial MSE: [(0, 8.719338211449212e-07), (1, 6.311974630079931e-06), (2, 2.395098636043258e-05), (3, 6.925358775333734e-06), (4, 1.787920382412267e-06), (5, 1.212184201904165e-06), (6, 2.6322825306124287e-06), (7, 2.995591557919397e-06), (8, 1.5906082353467355e-06), (9, 1.8711617713051965e-06), (10, 2.479484919604147e-06), (11, 6.052173830539687e-06), (12, 1.8129043382941745e-05), (13, 0.000702144461683929), (14, 0.0010003098286688328), (15, 0.0003085751086473465), (16, 0.02224523015320301), (17, 7.981149246916175e-05), (18, 0.00019159424118697643), (19, 0.0016637282678857446), (20, 0.0011381314834579825), (21, 7.959674985613674e-05), (22, 0.00021982743055559695), (23, 9.775100625120103e-05), (24, 0.00022612142493017018), (25, 0.0005786671536043286), (26, 0.0005501831183210015), (27, 0.009569872170686722)], Temporal MSE: [(0, 7.251477995851019e-07), (1, 2.983490787755727e-07), (2, 4.1186675048265897e-07), (3, 4.6096258188299544e-07), (4, 3.955936733746057e-07), (5, 5.147199431121408e-07), (6, 5.580302513408242e-07), (7, 9.073087881006359e-07), (8, 4.767447023823479e-07), (9, 5.766331696577254e-07), (10, 1.4570927078239038e-06), (11, 9.900259101414122e-06), (12, 1.5108601473912131e-05), (13, 2.2028552848496474e-05), (14, 2.1195277440710925e-05), (15, 3.2678741263225675e-05), (16, 1.5734989574411884e-05), (17, 2.7108864742331207e-05), (18, 3.2154977816389874e-05), (19, 2.5630191885284148e-05), (20, 2.146271799574606e-05), (21, 5.3275547543307766e-05), (22, 7.490102871088311e-05), (23, 0.00014075981744099408), (24, 9.376123489346355e-05), (25, 0.00020885822596028447), (26, 5.551130379899405e-05), (27, 5.161769513506442e-05)]
+
+ 48%|████▊     | 72/150 [11:02<11:07,  8.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 462, Spatial MSE: [(0, 9.378576351082302e-07), (1, 1.4160908904159442e-05), (2, 5.821835657116026e-05), (3, 1.85049902938772e-05), (4, 3.6880755942547694e-05), (5, 4.2650786781450734e-05), (6, 0.00010262196883559227), (7, 0.00024018078693188727), (8, 0.0003761306870728731), (9, 0.0004996901261620224), (10, 0.0005999899585731328), (11, 0.0007398735615424812), (12, 0.0008123227744363248), (13, 0.00246925069950521), (14, 0.005262311082333326), (15, 0.0046660457737743855), (16, 0.0578668937087059), (17, 0.003900379640981555), (18, 0.0036343915853649378), (19, 0.008698935620486736), (20, 0.00856745708733797), (21, 0.005285430699586868), (22, 0.003804737003520131), (23, 0.006429435685276985), (24, 0.011692875064909458), (25, 0.019471118226647377), (26, 0.014634697698056698), (27, 0.01035923883318901)], Temporal MSE: [(0, 9.792192940949462e-07), (1, 5.350559035832703e-07), (2, 2.35359107136901e-06), (3, 7.080869181663729e-06), (4, 1.3495350685843732e-05), (5, 2.0880537704215385e-05), (6, 9.459344437345862e-05), (7, 9.351039625471458e-05), (8, 7.666277087992057e-05), (9, 0.00011887401342391968), (10, 0.00027310452423989773), (11, 0.0003303350240457803), (12, 0.0003830987843684852), (13, 0.000752598571125418), (14, 0.001757728517986834), (15, 0.0020300119649618864), (16, 0.0023697023279964924), (17, 0.0023556703235954046), (18, 0.0015677824849262834), (19, 0.002357645658776164), (20, 0.004209654871374369), (21, 0.0037728899624198675), (22, 0.0017927787266671658), (23, 0.0027148406952619553), (24, 0.00619675824418664), (25, 0.009973937645554543), (26, 0.0017010809388011694), (27, 0.00010169500455958769)]
+
+ 49%|████▊     | 73/150 [11:12<11:10,  8.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 456, Spatial MSE: [(0, 6.591492933694099e-07), (1, 6.9169827838777564e-06), (2, 1.891800457087811e-05), (3, 7.733968232059851e-06), (4, 1.835947614381439e-06), (5, 1.2787376135747763e-06), (6, 2.720642214626423e-06), (7, 3.028996388820815e-06), (8, 1.7711811324261362e-06), (9, 1.9433355191722512e-06), (10, 5.341295945981983e-06), (11, 2.8205009584780782e-05), (12, 6.725376442773268e-05), (13, 0.0016900984337553382), (14, 0.001413705525919795), (15, 0.00035566044971346855), (16, 0.008975531905889511), (17, 3.6528374039335176e-05), (18, 3.399043271201663e-05), (19, 0.0007602971745654941), (20, 0.0004599628737196326), (21, 3.684878902276978e-05), (22, 0.0001321877643931657), (23, 8.053096098592505e-05), (24, 0.00017715638387016952), (25, 0.0003267854335717857), (26, 0.0005159308784641325), (27, 0.013088020496070385)], Temporal MSE: [(0, 6.815828896833409e-07), (1, 2.9596520789709757e-07), (2, 3.878418226577196e-07), (3, 5.149801154402667e-07), (4, 4.5794516267960717e-07), (5, 5.660981514665764e-07), (6, 7.280219733729609e-07), (7, 9.554514690535143e-07), (8, 5.396375968302891e-07), (9, 6.882965521981532e-07), (10, 3.5987416140415007e-06), (11, 7.204762368928641e-05), (12, 3.6292141885496676e-05), (13, 4.0667669964022934e-05), (14, 1.510174206487136e-05), (15, 1.8537342839408666e-05), (16, 1.0769969776447397e-05), (17, 9.138032510236371e-06), (18, 2.500700247765053e-05), (19, 2.626243804115802e-05), (20, 1.8734828699962236e-05), (21, 2.340128230571281e-05), (22, 3.043269316549413e-05), (23, 7.431292760884389e-05), (24, 7.919507334008813e-05), (25, 0.000232393154874444), (26, 6.0868005675729364e-05), (27, 3.334253415232524e-05)]
+
+ 49%|████▉     | 74/150 [11:20<10:48,  8.53s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 450, Spatial MSE: [(0, 1.0685462257242762e-06), (1, 9.343542842543684e-06), (2, 3.124419163214043e-05), (3, 8.056073056650348e-06), (4, 4.366263965493999e-06), (5, 5.45645025340491e-06), (6, 9.801115083973855e-06), (7, 8.931947377277538e-06), (8, 1.4199446013662964e-05), (9, 3.141935303574428e-05), (10, 5.025584323448129e-05), (11, 6.735417264280841e-05), (12, 0.00012352009071037173), (13, 0.0017425903351977468), (14, 0.0016173496842384338), (15, 0.0006021024310030043), (16, 0.015053680166602135), (17, 8.523335418431088e-05), (18, 0.00012321319081820548), (19, 0.0010158229852095246), (20, 0.0006616506143473089), (21, 0.00011648709187284112), (22, 0.00021462026052176952), (23, 0.00018991599790751934), (24, 0.0004649359034374356), (25, 0.0008958179969340563), (26, 0.002500110538676381), (27, 0.012287796474993229)], Temporal MSE: [(0, 6.232231726244208e-07), (1, 3.7006904562986165e-07), (2, 8.977146990218898e-07), (3, 1.5246488374032197e-06), (4, 2.672987420737627e-06), (5, 2.0676270651165396e-06), (6, 3.0056176001380663e-06), (7, 3.5581122119765496e-06), (8, 4.0931172406999394e-06), (9, 4.226964847475756e-06), (10, 9.701842827780638e-06), (11, 7.753576937830076e-05), (12, 4.665709275286645e-05), (13, 5.669506572303362e-05), (14, 4.337180143920705e-05), (15, 4.141827957937494e-05), (16, 3.0693459848407656e-05), (17, 2.7033884180127643e-05), (18, 4.484952296479605e-05), (19, 4.6100070903776214e-05), (20, 5.242529732640833e-05), (21, 6.112481059972197e-05), (22, 7.06187856849283e-05), (23, 0.00011905830615432933), (24, 0.00022841985628474504), (25, 0.0004910890711471438), (26, 0.0001525511033833027), (27, 4.056676334585063e-05)]
+
+ 50%|█████     | 75/150 [11:28<10:29,  8.39s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 444, Spatial MSE: [(0, 6.936849104022258e-07), (1, 6.0858028518850915e-06), (2, 2.0322979253251106e-05), (3, 6.031056727806572e-06), (4, 1.7044737887772499e-06), (5, 1.1225300795558724e-06), (6, 1.8401851775706746e-06), (7, 2.5136048407148337e-06), (8, 1.4817136388955987e-06), (9, 1.8657019609236158e-06), (10, 2.3827485620131483e-06), (11, 8.067832823144272e-06), (12, 2.5427569198654965e-05), (13, 0.0008369176066480577), (14, 0.0008151177316904068), (15, 0.00024467476760037243), (16, 0.00638794619590044), (17, 2.6741174224298447e-05), (18, 2.2024823920219205e-05), (19, 0.0005136181716807187), (20, 0.00029597553657367826), (21, 3.284591366536915e-05), (22, 0.00012489179789554328), (23, 6.692460738122463e-05), (24, 0.00013673107605427504), (25, 0.00027341130771674216), (26, 0.00044424034422263503), (27, 0.00855762604624033)], Temporal MSE: [(0, 6.305704687292746e-07), (1, 2.8354861569823697e-07), (2, 4.1695187746881857e-07), (3, 4.664689754463325e-07), (4, 4.3870124954992207e-07), (5, 5.362687716115033e-07), (6, 6.059809152247908e-07), (7, 8.527329100616043e-07), (8, 4.89314857077261e-07), (9, 5.581335926763131e-07), (10, 1.3888967487218906e-06), (11, 1.5777619410073385e-05), (12, 1.7920738173415884e-05), (13, 2.474261782481335e-05), (14, 1.2004466043435968e-05), (15, 1.4618652130593546e-05), (16, 8.24502876639599e-06), (17, 7.266800366778625e-06), (18, 2.050551847787574e-05), (19, 2.0528928871499375e-05), (20, 1.7336175005766563e-05), (21, 2.103450424328912e-05), (22, 2.6348299797973596e-05), (23, 5.992188016534783e-05), (24, 6.194353045430034e-05), (25, 0.0002208915539085865), (26, 4.706937033915892e-05), (27, 2.525333002267871e-05)]
+
+ 51%|█████     | 76/150 [11:35<09:59,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 438, Spatial MSE: [(0, 8.7762674638725e-07), (1, 1.5706658814451657e-05), (2, 5.0972903409274295e-05), (3, 1.7180422219098546e-05), (4, 3.0397106456803158e-05), (5, 4.305617039790377e-05), (6, 9.54587958403863e-05), (7, 0.00022787308262195438), (8, 0.00029267818899825215), (9, 0.0003500937600620091), (10, 0.00042470876360312104), (11, 0.0005179390427656472), (12, 0.0005973360384814441), (13, 0.0026898132637143135), (14, 0.007573961280286312), (15, 0.006081890780478716), (16, 0.10491884499788284), (17, 0.0031460318714380264), (18, 0.0029948786832392216), (19, 0.008156766183674335), (20, 0.00809093751013279), (21, 0.004974287934601307), (22, 0.003806078340858221), (23, 0.006661978550255299), (24, 0.011744033545255661), (25, 0.019896237179636955), (26, 0.015546421520411968), (27, 0.009450133889913559)], Temporal MSE: [(0, 1.0533094609854743e-06), (1, 5.320922014107055e-07), (2, 2.264503109472571e-06), (3, 7.462664143531583e-06), (4, 1.3860740182281006e-05), (5, 2.05886535695754e-05), (6, 8.821293158689514e-05), (7, 8.170269575202838e-05), (8, 5.6913493608590215e-05), (9, 9.000638965517282e-05), (10, 0.00020468368893489242), (11, 0.00025993381859734654), (12, 0.0003011646622326225), (13, 0.0006252164021134377), (14, 0.0013697901740670204), (15, 0.0016354724066331983), (16, 0.001953862840309739), (17, 0.002020273357629776), (18, 0.0013751836959272623), (19, 0.002154753776267171), (20, 0.003928558900952339), (21, 0.003607013262808323), (22, 0.001765238936059177), (23, 0.002656362485140562), (24, 0.005999565590173006), (25, 0.009737994521856308), (26, 0.0016958701889961958), (27, 7.927270780783147e-05)]
+
+ 51%|█████▏    | 77/150 [11:44<10:00,  8.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 432, Spatial MSE: [(0, 7.520889084844384e-07), (1, 7.855564035708085e-06), (2, 2.1831770936842076e-05), (3, 6.690981081192149e-06), (4, 1.794021159184922e-06), (5, 1.2307151564527885e-06), (6, 2.1131897938175825e-06), (7, 2.142381163139362e-06), (8, 1.66497750342387e-06), (9, 2.673980134204612e-06), (10, 7.065244517434621e-06), (11, 4.3543139327084646e-05), (12, 7.131204620236531e-05), (13, 0.0014349214034155011), (14, 0.0011470266617834568), (15, 0.0002894959761761129), (16, 0.006760265212506056), (17, 2.9357683160924353e-05), (18, 2.7060672437073663e-05), (19, 0.0005849303561262786), (20, 0.00037407438503578305), (21, 3.841898433165625e-05), (22, 0.0001213138602906838), (23, 7.703201845288277e-05), (24, 0.00016172384493984282), (25, 0.0003063083568122238), (26, 0.0004827488155569881), (27, 0.010508016683161259)], Temporal MSE: [(0, 6.933692588972917e-07), (1, 3.118236975296895e-07), (2, 4.10287725571834e-07), (3, 5.134303933118645e-07), (4, 4.821551442546479e-07), (5, 5.697384040104225e-07), (6, 6.653606874351681e-07), (7, 8.552551662432961e-07), (8, 5.597789254352392e-07), (9, 6.614848189201439e-07), (10, 4.210226961731678e-06), (11, 9.9200799013488e-05), (12, 3.1032941478770226e-05), (13, 3.385301897651516e-05), (14, 1.5058260942169e-05), (15, 1.4209846085577738e-05), (16, 9.078982657229062e-06), (17, 7.345102403633064e-06), (18, 1.834063368733041e-05), (19, 2.392099122516811e-05), (20, 2.0262046746211126e-05), (21, 2.1539177396334708e-05), (22, 2.6757448722491972e-05), (23, 6.498629227280617e-05), (24, 6.759721873095259e-05), (25, 0.00023553792561870068), (26, 5.538782352232374e-05), (27, 2.6241939849569462e-05)]
+
+ 52%|█████▏    | 78/150 [11:51<09:44,  8.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 426, Spatial MSE: [(0, 1.0109406503033824e-06), (1, 8.721603990125004e-06), (2, 2.3360620616585948e-05), (3, 7.963853022374678e-06), (4, 4.983310191164492e-06), (5, 5.411308848124463e-06), (6, 8.411509043071419e-06), (7, 1.0516832844587043e-05), (8, 1.5395447917398997e-05), (9, 3.1420095183420926e-05), (10, 5.8284193073632196e-05), (11, 9.4610259111505e-05), (12, 0.00013901529018767178), (13, 0.001887548598460853), (14, 0.0021005875896662474), (15, 0.0007939308416098356), (16, 0.018492912873625755), (17, 0.00010488776752026752), (18, 0.0004875201266258955), (19, 0.0018618055619299412), (20, 0.0015637472970411181), (21, 0.00027092150412499905), (22, 0.00024578889133408666), (23, 0.00024183774075936526), (24, 0.0005628929357044399), (25, 0.0010370665695518255), (26, 0.0024563006591051817), (27, 0.009282084181904793)], Temporal MSE: [(0, 5.53128927549551e-07), (1, 2.926875311004551e-07), (2, 8.323663109877089e-07), (3, 1.4561441048499546e-06), (4, 2.7267676614428638e-06), (5, 2.110804189214832e-06), (6, 3.1456572742172284e-06), (7, 3.6218721106706653e-06), (8, 4.087322849954944e-06), (9, 4.4903822526976e-06), (10, 1.3708177903026808e-05), (11, 0.00011578316480154172), (12, 5.0444385124137625e-05), (13, 5.8336645452072844e-05), (14, 5.042222255724482e-05), (15, 4.7336248826468363e-05), (16, 3.182888758601621e-05), (17, 2.838603722921107e-05), (18, 5.061840056441724e-05), (19, 7.060566713334993e-05), (20, 9.158583270618692e-05), (21, 8.058849925873801e-05), (22, 8.337226609000936e-05), (23, 0.00013329826469998807), (24, 0.00023073960619512945), (25, 0.0004626894660759717), (26, 0.00017336203018203378), (27, 3.643796662800014e-05)]
+
+ 53%|█████▎    | 79/150 [11:59<09:33,  8.08s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 420, Spatial MSE: [(0, 1.10410735487676e-06), (1, 7.233169071696466e-06), (2, 1.830875226005446e-05), (3, 7.012726655375445e-06), (4, 1.8059113244817127e-06), (5, 1.5961495591909625e-06), (6, 3.0510916531056864e-06), (7, 4.01028000851511e-06), (8, 2.051973979178001e-06), (9, 2.3496388621424558e-06), (10, 7.793183613102883e-06), (11, 2.4957045752671547e-05), (12, 5.443048939923756e-05), (13, 0.0014122870052233338), (14, 0.0013214710634201765), (15, 0.00039923228905536234), (16, 0.009142166934907436), (17, 3.9580474549438804e-05), (18, 6.489975203294307e-05), (19, 0.000715436355676502), (20, 0.00043605457176454365), (21, 6.835321255493909e-05), (22, 0.0001380553439958021), (23, 8.329181582666934e-05), (24, 0.00018325202108826488), (25, 0.0003380470734555274), (26, 0.0005330173880793154), (27, 0.012166746892035007)], Temporal MSE: [(0, 7.95487380855775e-07), (1, 3.174388609750167e-07), (2, 3.88796252082102e-07), (3, 5.251760057944921e-07), (4, 5.072579369880259e-07), (5, 5.907979812036501e-07), (6, 7.03757166320429e-07), (7, 1.1055769846279873e-06), (8, 5.947964041297382e-07), (9, 8.014312129489554e-07), (10, 6.131077043391997e-06), (11, 5.652281834045425e-05), (12, 2.7125883207190782e-05), (13, 3.355677108629607e-05), (14, 1.5858198821661063e-05), (15, 2.5999672288889997e-05), (16, 1.0926165487035178e-05), (17, 1.0893324542848859e-05), (18, 2.5074847144423984e-05), (19, 2.9579918191302568e-05), (20, 2.859654887288343e-05), (21, 3.5986737202620134e-05), (22, 3.6601930332835764e-05), (23, 7.838142482796684e-05), (24, 8.220050222007558e-05), (25, 0.00030449070618487895), (26, 6.256021879380569e-05), (27, 3.5389442928135395e-05)]
+
+ 53%|█████▎    | 80/150 [12:07<09:07,  7.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 414, Spatial MSE: [(0, 1.0235419267701218e-06), (1, 1.528308894194197e-05), (2, 4.407013693707995e-05), (3, 1.3505413335224148e-05), (4, 2.5120190912275575e-05), (5, 4.0875595004763454e-05), (6, 6.224615208338946e-05), (7, 0.00018222711514681578), (8, 0.00021687203843612224), (9, 0.00027923143352381885), (10, 0.00034305150620639324), (11, 0.00047213188372552395), (12, 0.0005444606649689376), (13, 0.0027607264928519726), (14, 0.005356306210160255), (15, 0.004366264212876558), (16, 0.06700041890144348), (17, 0.0028911863919347525), (18, 0.002805245341733098), (19, 0.0059981453232467175), (20, 0.006231039296835661), (21, 0.004828343167901039), (22, 0.003366408869624138), (23, 0.005963031202554703), (24, 0.010159255936741829), (25, 0.017576348036527634), (26, 0.014081524685025215), (27, 0.012455729767680168)], Temporal MSE: [(0, 8.583497788094974e-07), (1, 4.504022683704534e-07), (2, 1.93844607565552e-06), (3, 8.135672942444216e-06), (4, 1.4275886314862873e-05), (5, 1.9036209778278135e-05), (6, 7.230946357594803e-05), (7, 6.580245826626197e-05), (8, 4.728971907752566e-05), (9, 7.748833013465628e-05), (10, 0.00018777897639665753), (11, 0.0002761577779892832), (12, 0.0002791954029817134), (13, 0.0005776790203526616), (14, 0.0012440058635547757), (15, 0.0015193871222436428), (16, 0.0018478527199476957), (17, 0.001828306820243597), (18, 0.001279456657357514), (19, 0.0020188845228403807), (20, 0.003716530743986368), (21, 0.0033229179680347443), (22, 0.0015699333744123578), (23, 0.0023572761565446854), (24, 0.0051291570998728275), (25, 0.008607381954789162), (26, 0.0016255495138466358), (27, 9.037399286171421e-05)]
+
+ 54%|█████▍    | 81/150 [12:16<09:21,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 408, Spatial MSE: [(0, 8.23300183583342e-07), (1, 6.364672117342707e-06), (2, 1.6893269275897183e-05), (3, 6.07658876106143e-06), (4, 1.5441360119439196e-06), (5, 1.1214244750590296e-06), (6, 1.8360968851993675e-06), (7, 2.6254410840920173e-06), (8, 1.4897939308866626e-06), (9, 1.5696396076236852e-06), (10, 2.9597445063700434e-06), (11, 5.656391749653267e-06), (12, 1.7637295968597755e-05), (13, 0.0007649844628758729), (14, 0.0012425187742337584), (15, 0.0004384474304970354), (16, 0.019057463854551315), (17, 8.132908988045529e-05), (18, 4.9625607061898336e-05), (19, 0.0014700882602483034), (20, 0.0009993609273806214), (21, 0.0002110797504428774), (22, 0.00023720698663964868), (23, 0.00010105086403200403), (24, 0.000222388916881755), (25, 0.0005077986279502511), (26, 0.0004258991975802928), (27, 0.012203150428831577)], Temporal MSE: [(0, 7.074083328006964e-07), (1, 2.722526915022172e-07), (2, 3.9527307649223076e-07), (3, 4.954955556968343e-07), (4, 4.680574932081072e-07), (5, 5.755165375376237e-07), (6, 6.551110800501192e-07), (7, 8.425677719969826e-07), (8, 5.450414732877107e-07), (9, 5.999345376039855e-07), (10, 1.9271978999313433e-06), (11, 5.914541816309793e-06), (12, 1.241305290022865e-05), (13, 2.8454354833229445e-05), (14, 2.1881085558561608e-05), (15, 4.1154133214149624e-05), (16, 2.314776611456182e-05), (17, 2.5424840714549646e-05), (18, 3.634275708463974e-05), (19, 3.605632809922099e-05), (20, 2.862378823920153e-05), (21, 8.191059896489605e-05), (22, 7.206799637060612e-05), (23, 0.0001292182132601738), (24, 0.00010026575910160318), (25, 0.0003081571194343269), (26, 6.459748692577705e-05), (27, 5.531253191293217e-05)]
+
+ 55%|█████▍    | 82/150 [12:24<09:11,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 402, Spatial MSE: [(0, 1.016425358102424e-06), (1, 8.268768397101667e-06), (2, 2.0300083633628674e-05), (3, 6.951041086722398e-06), (4, 4.007733423350146e-06), (5, 5.910615072934888e-06), (6, 9.10944800125435e-06), (7, 7.763999747112393e-06), (8, 1.2868036719737574e-05), (9, 2.875869358831551e-05), (10, 4.012730278191157e-05), (11, 4.6044042392168194e-05), (12, 7.455150625901297e-05), (13, 0.0009365816949866712), (14, 0.0018314304761588573), (15, 0.0010033062426373363), (16, 0.03275464102625847), (17, 0.00015971709217410535), (18, 0.00015566653746645898), (19, 0.0020843639504164457), (20, 0.0016571150626987219), (21, 0.00038020810461603105), (22, 0.00035369835677556694), (23, 0.0002489567268639803), (24, 0.0005888055893592536), (25, 0.0012018666602671146), (26, 0.002557293511927128), (27, 0.012005163356661797)], Temporal MSE: [(0, 6.36729510006262e-07), (1, 3.627461353516992e-07), (2, 8.346282243110181e-07), (3, 1.5203115708573023e-06), (4, 2.749110990407644e-06), (5, 2.0369116100482643e-06), (6, 3.0470534966298146e-06), (7, 3.867773102683714e-06), (8, 4.097829787497176e-06), (9, 4.2106207729375456e-06), (10, 8.63252080307575e-06), (11, 1.6746620531193912e-05), (12, 2.619020960992202e-05), (13, 4.752790118800476e-05), (14, 5.0244565500179306e-05), (15, 6.11364230280742e-05), (16, 4.1291041270596907e-05), (17, 4.313777753850445e-05), (18, 5.928489554207772e-05), (19, 6.378522084560245e-05), (20, 8.674403215991333e-05), (21, 0.00011358144547557458), (22, 0.0001130135206039995), (23, 0.00017238127475138754), (24, 0.00025672977790236473), (25, 0.000555497535970062), (26, 0.0001911547442432493), (27, 6.529590609716251e-05)]
+
+ 55%|█████▌    | 83/150 [12:32<09:09,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 396, Spatial MSE: [(0, 7.894028613009141e-07), (1, 7.242681022034958e-06), (2, 1.644342773943208e-05), (3, 6.645513167313766e-06), (4, 1.8354969597567106e-06), (5, 1.4007056279297103e-06), (6, 1.8993258663613233e-06), (7, 2.519423333069426e-06), (8, 1.8648562445378047e-06), (9, 2.4226633286161814e-06), (10, 8.656357749714516e-06), (11, 3.771772753680125e-05), (12, 7.307162013603374e-05), (13, 0.0013070333516225219), (14, 0.001783365267328918), (15, 0.0004447127284947783), (16, 0.026368141174316406), (17, 0.00011342890502419323), (18, 7.491761061828583e-05), (19, 0.002179825911298394), (20, 0.0015343418344855309), (21, 7.912655564723536e-05), (22, 0.00032224360620602965), (23, 0.00011674108100123703), (24, 0.0002553032827563584), (25, 0.0006685463013127446), (26, 0.00048310135025531054), (27, 0.015398863703012466)], Temporal MSE: [(0, 6.370747769324225e-07), (1, 2.848594249371672e-07), (2, 3.806234474268422e-07), (3, 5.507587275133119e-07), (4, 5.71795624182414e-07), (5, 6.630824600506458e-07), (6, 7.964331416587811e-07), (7, 1.0084114592245896e-06), (8, 6.563566898876161e-07), (9, 8.416927812504582e-07), (10, 5.7353013289684895e-06), (11, 7.446613744832575e-05), (12, 3.325149373267777e-05), (13, 3.334686698508449e-05), (14, 2.710770786507055e-05), (15, 4.5748904085485265e-05), (16, 3.2567058951826766e-05), (17, 3.0342127502080984e-05), (18, 4.1024351958185434e-05), (19, 3.886527701979503e-05), (20, 3.239844227209687e-05), (21, 9.539397433400154e-05), (22, 0.00010480351920705289), (23, 0.00016540147771593183), (24, 0.00012161560880485922), (25, 0.0003033917455468327), (26, 7.233618089230731e-05), (27, 7.024215301498771e-05)]
+
+ 56%|█████▌    | 84/150 [12:41<09:20,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 390, Spatial MSE: [(0, 1.0295955235051224e-06), (1, 1.4207735148374923e-05), (2, 3.9108996134018525e-05), (3, 1.7098547687055543e-05), (4, 3.4500564652262256e-05), (5, 4.4470234570326284e-05), (6, 8.51584118208848e-05), (7, 0.00021193010616116226), (8, 0.00019658142991829664), (9, 0.00029682315653190017), (10, 0.0003977443848270923), (11, 0.000557497376576066), (12, 0.0006116664153523743), (13, 0.0022624351549893618), (14, 0.004254535771906376), (15, 0.0035859926138073206), (16, 0.040930986404418945), (17, 0.002921679290011525), (18, 0.0029681604355573654), (19, 0.006129591725766659), (20, 0.006185883656144142), (21, 0.004498316906392574), (22, 0.003275014692917466), (23, 0.0055200811475515366), (24, 0.008974147029221058), (25, 0.017764998599886894), (26, 0.016354355961084366), (27, 0.016338037326931953)], Temporal MSE: [(0, 9.511668963568809e-07), (1, 4.80427218008117e-07), (2, 2.444950041535776e-06), (3, 7.1627487159275915e-06), (4, 1.4848500541120302e-05), (5, 2.158706593036186e-05), (6, 7.697097316849977e-05), (7, 6.680672231595963e-05), (8, 4.9981314077740535e-05), (9, 8.524874283466488e-05), (10, 0.00021038559498265386), (11, 0.00032705359626561403), (12, 0.00032818131148815155), (13, 0.0006314170313999057), (14, 0.001439498970285058), (15, 0.0016557845519855618), (16, 0.0018866379978135228), (17, 0.001794170355424285), (18, 0.0012472181115299463), (19, 0.0019468322861939669), (20, 0.0033494506496936083), (21, 0.002847114345058799), (22, 0.0014283732743933797), (23, 0.002110221656039357), (24, 0.00426336657255888), (25, 0.007090803701430559), (26, 0.0015416747191920877), (27, 0.00012401081039570272)]
+
+ 57%|█████▋    | 85/150 [12:49<09:03,  8.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 384, Spatial MSE: [(0, 5.79078289320023e-07), (1, 6.68791926727863e-06), (2, 1.584843084856402e-05), (3, 6.2125909607857466e-06), (4, 1.5706460771980346e-06), (5, 1.2002859648418962e-06), (6, 1.449795263397391e-06), (7, 1.9647047793114325e-06), (8, 1.4369635437105899e-06), (9, 1.8416799321130384e-06), (10, 3.6565227219398366e-06), (11, 1.659718509472441e-05), (12, 4.64053446194157e-05), (13, 0.0011856678174808621), (14, 0.0016271783970296383), (15, 0.000492120801936835), (16, 0.02720082364976406), (17, 9.26680804695934e-05), (18, 7.508130511268973e-05), (19, 0.002264101058244705), (20, 0.0017039107624441385), (21, 8.22915171738714e-05), (22, 0.00026372450520284474), (23, 0.00012646774121094495), (24, 0.0002606587950140238), (25, 0.0005612504901364446), (26, 0.0004749734071083367), (27, 0.016448257490992546)], Temporal MSE: [(0, 5.183437679079361e-07), (1, 2.392123406025348e-07), (2, 3.4227664968966565e-07), (3, 4.995279709874012e-07), (4, 5.29399528659269e-07), (5, 6.044180054232129e-07), (6, 7.124223202481517e-07), (7, 8.302117748826277e-07), (8, 5.684928510163445e-07), (9, 6.138926096355135e-07), (10, 2.0933405266987393e-06), (11, 4.042469663545489e-05), (12, 2.8393769753165543e-05), (13, 3.3291846193606034e-05), (14, 2.6692958272178657e-05), (15, 4.3938551243627444e-05), (16, 2.7862106435350142e-05), (17, 2.9473623726516962e-05), (18, 4.179597817710601e-05), (19, 6.106242653913796e-05), (20, 6.802167627029121e-05), (21, 0.00011366547551006079), (22, 9.723437688080594e-05), (23, 0.000163196527864784), (24, 0.00011649267980828881), (25, 0.0003134647267870605), (26, 7.838641613489017e-05), (27, 7.182364061009139e-05)]
+
+ 57%|█████▋    | 86/150 [12:57<08:41,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 378, Spatial MSE: [(0, 8.749724429435446e-07), (1, 7.903113328211475e-06), (2, 1.9843359041260555e-05), (3, 6.248657427931903e-06), (4, 3.6909402751916787e-06), (5, 5.348006197891664e-06), (6, 7.691693099332042e-06), (7, 7.426325737469597e-06), (8, 1.1640614502539393e-05), (9, 2.2749136405764148e-05), (10, 3.394264058442786e-05), (11, 5.4529351473320276e-05), (12, 0.00010180630488321185), (13, 0.0013845398789271712), (14, 0.0024577644653618336), (15, 0.0012922071618959308), (16, 0.04018595442175865), (17, 0.0001843487552832812), (18, 0.00017705252685118467), (19, 0.002756105037406087), (20, 0.002202806295827031), (21, 0.00022108860139269382), (22, 0.0003590578562580049), (23, 0.00028728009783662856), (24, 0.0006680826772935688), (25, 0.0015469787176698446), (26, 0.0030100210569798946), (27, 0.015635276213288307)], Temporal MSE: [(0, 5.804305942547217e-07), (1, 3.043223841814324e-07), (2, 8.112177170005452e-07), (3, 1.3667761322722072e-06), (4, 2.652237526490353e-06), (5, 2.109630941049545e-06), (6, 2.9896841624577064e-06), (7, 3.6634178286476526e-06), (8, 3.7800245991093107e-06), (9, 3.974727405875456e-06), (10, 9.159224646282382e-06), (11, 5.352902371669188e-05), (12, 4.395323776407167e-05), (13, 5.1943541620858014e-05), (14, 5.914967186981812e-05), (15, 7.205922884168103e-05), (16, 4.7338024160126224e-05), (17, 4.85955060867127e-05), (18, 6.314528582151979e-05), (19, 8.935830555856228e-05), (20, 0.00011068066669395193), (21, 0.00015992019325494766), (22, 0.00014412100426852703), (23, 0.0002186842611990869), (24, 0.0002808185527101159), (25, 0.000617378915194422), (26, 0.0002193191321566701), (27, 8.207510836655274e-05)]
+
+ 58%|█████▊    | 87/150 [13:06<08:52,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 372, Spatial MSE: [(0, 8.073722028711927e-07), (1, 6.883158675918821e-06), (2, 1.3139331713318825e-05), (3, 6.33022636975511e-06), (4, 1.8310990981262876e-06), (5, 1.5876275938353501e-06), (6, 1.9017205659110914e-06), (7, 2.5969548005377874e-06), (8, 1.9941951450164197e-06), (9, 2.276987743243808e-06), (10, 9.034087270265445e-06), (11, 3.9644270145799965e-05), (12, 6.642490188824013e-05), (13, 0.0011874836636707187), (14, 0.0012935842387378216), (15, 0.00035850328276865184), (16, 0.013495850376784801), (17, 5.8855868701357394e-05), (18, 0.00010872167331399396), (19, 0.001015092246234417), (20, 0.000673266127705574), (21, 9.106645302381366e-05), (22, 0.0002528218610677868), (23, 9.008740744320676e-05), (24, 0.00019837240688502789), (25, 0.0004143819969613105), (26, 0.00045120326103642583), (27, 0.0097332913428545)], Temporal MSE: [(0, 6.855428296148602e-07), (1, 3.0129896799735434e-07), (2, 3.615797652400943e-07), (3, 5.536485332413577e-07), (4, 5.827035920447088e-07), (5, 6.837500450274092e-07), (6, 7.386367428807716e-07), (7, 1.0036945923275198e-06), (8, 6.778103056603868e-07), (9, 8.384859029320069e-07), (10, 7.948219717945904e-06), (11, 8.215808338718489e-05), (12, 2.859900450857822e-05), (13, 2.6130950573133305e-05), (14, 1.9113567759632133e-05), (15, 3.620645293267444e-05), (16, 1.785317544999998e-05), (17, 1.7495356587460265e-05), (18, 2.512938044674229e-05), (19, 2.153283821826335e-05), (20, 1.5663714293623343e-05), (21, 6.82324098306708e-05), (22, 7.444671064149588e-05), (23, 0.0001217290191561915), (24, 9.913435496855527e-05), (25, 0.00023755420988891274), (26, 5.5427710321964696e-05), (27, 4.148607331444509e-05)]
+
+ 59%|█████▊    | 88/150 [13:14<08:41,  8.42s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 366, Spatial MSE: [(0, 8.389466188418737e-07), (1, 1.1524276487762108e-05), (2, 3.154157820972614e-05), (3, 1.3766699339612387e-05), (4, 2.9287562938407063e-05), (5, 3.880431177094579e-05), (6, 5.542826693272218e-05), (7, 0.0001537894131615758), (8, 0.00016962325025815517), (9, 0.00023032407625578344), (10, 0.00031458548619411886), (11, 0.0004384620988275856), (12, 0.0004995370400138199), (13, 0.002723101992160082), (14, 0.00642459187656641), (15, 0.004870749544352293), (16, 0.09412146359682083), (17, 0.002309839241206646), (18, 0.0023861140944063663), (19, 0.007048895116895437), (20, 0.006691904738545418), (21, 0.00426066480576992), (22, 0.0029343809001147747), (23, 0.004839416593313217), (24, 0.007787754293531179), (25, 0.014095217920839787), (26, 0.012767736800014973), (27, 0.010277565568685532)], Temporal MSE: [(0, 7.539470630035794e-07), (1, 3.6130595049144176e-07), (2, 2.06478785003128e-06), (3, 8.153797352861147e-06), (4, 1.4215214832802303e-05), (5, 1.899466587929055e-05), (6, 6.131349073257297e-05), (7, 5.537047400139272e-05), (8, 4.2145282350247726e-05), (9, 7.178364467108622e-05), (10, 0.0001802616607164964), (11, 0.00029215627000667155), (12, 0.00028797975392080843), (13, 0.0005356163601391017), (14, 0.0011200301814824343), (15, 0.001287305261939764), (16, 0.0014579863054677844), (17, 0.0014313015853986144), (18, 0.0010526891564950347), (19, 0.001581404940225184), (20, 0.002703513950109482), (21, 0.0023981069680303335), (22, 0.0012967000948265195), (23, 0.0018884259043261409), (24, 0.0037283271085470915), (25, 0.006336635909974575), (26, 0.0013597235083580017), (27, 0.00010468360414961353)]
+
+ 59%|█████▉    | 89/150 [13:23<08:27,  8.32s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 360, Spatial MSE: [(0, 5.340472739590041e-07), (1, 6.403361567208776e-06), (2, 1.4134011507849209e-05), (3, 5.971422524453374e-06), (4, 1.4314365444079158e-06), (5, 1.218294073623838e-06), (6, 1.2969005638296949e-06), (7, 1.6997171314869775e-06), (8, 1.4486969348581624e-06), (9, 1.8877714182963246e-06), (10, 1.6093838439701358e-06), (11, 3.408967131690588e-06), (12, 1.0294790627085604e-05), (13, 0.0002763553347904235), (14, 0.0007552682072855532), (15, 0.00023067272559273988), (16, 0.014172112569212914), (17, 5.2627030527219176e-05), (18, 6.812200444983318e-05), (19, 0.0011537921382114291), (20, 0.0008403385290876031), (21, 8.743517537368461e-05), (22, 0.0001949552824953571), (23, 7.160483073676005e-05), (24, 0.00016082421643659472), (25, 0.0004277828265912831), (26, 0.0003074216074310243), (27, 0.006321359891444445)], Temporal MSE: [(0, 4.847453851652972e-07), (1, 2.528905156395922e-07), (2, 3.622879205522622e-07), (3, 5.280193704493286e-07), (4, 5.802139071420243e-07), (5, 6.39835661786492e-07), (6, 7.242383617267478e-07), (7, 8.61438934407488e-07), (8, 6.136409069767979e-07), (9, 6.195218134052993e-07), (10, 1.0298981578671373e-06), (11, 4.211338819004595e-06), (12, 9.030517503560986e-06), (13, 1.0329960787203163e-05), (14, 1.6721362044336274e-05), (15, 2.6434030587552115e-05), (16, 1.3060122000752017e-05), (17, 1.7134543668362312e-05), (18, 2.1734344045398757e-05), (19, 1.9739718482014723e-05), (20, 1.605283068784047e-05), (21, 7.034930604277179e-05), (22, 7.325033220695332e-05), (23, 0.00011517734674271196), (24, 7.616531365783885e-05), (25, 0.00015499713481403887), (26, 4.330908632255159e-05), (27, 3.38826903316658e-05)]
+
+ 60%|██████    | 90/150 [13:34<09:10,  9.18s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 354, Spatial MSE: [(0, 8.832858497953566e-07), (1, 6.90507295075804e-06), (2, 1.4859297152725048e-05), (3, 6.84190354149905e-06), (4, 4.217075002088677e-06), (5, 6.531029612233397e-06), (6, 9.828566362557467e-06), (7, 1.049447564582806e-05), (8, 1.4719632417836692e-05), (9, 3.0009126930963248e-05), (10, 4.249390258337371e-05), (11, 5.4807238484499976e-05), (12, 7.136509520933032e-05), (13, 0.00046736604417674243), (14, 0.0015847102040424943), (15, 0.0010407447116449475), (16, 0.027812838554382324), (17, 0.00013338816643226892), (18, 0.000567877315916121), (19, 0.0027661388739943504), (20, 0.0024543371982872486), (21, 0.0004031992284581065), (22, 0.0003667726705316454), (23, 0.00028257572557777166), (24, 0.0006730490713380277), (25, 0.001742664841003716), (26, 0.0034392070956528187), (27, 0.0063111488707363605)], Temporal MSE: [(0, 7.122500278455846e-07), (1, 3.6232646039024985e-07), (2, 8.116261369650601e-07), (3, 1.4413643611987936e-06), (4, 2.959682660730323e-06), (5, 2.297797209394048e-06), (6, 3.680240524772671e-06), (7, 5.242176030151313e-06), (8, 4.741997599921888e-06), (9, 5.344385044736555e-06), (10, 9.136766493611503e-06), (11, 1.5782750779180788e-05), (12, 2.334445161977783e-05), (13, 3.5830558772431687e-05), (14, 5.8460056607145816e-05), (15, 5.653249900205992e-05), (16, 3.738817758858204e-05), (17, 3.585012746043503e-05), (18, 4.954476025886834e-05), (19, 7.547158020315692e-05), (20, 0.00012406556925270706), (21, 0.00014821179502177984), (22, 0.0001383426715619862), (23, 0.00017404356913175434), (24, 0.0002538142434787005), (25, 0.0005059729446657002), (26, 0.00019388151122257113), (27, 4.745576006826013e-05)]
+
+ 61%|██████    | 91/150 [13:44<09:15,  9.42s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 348, Spatial MSE: [(0, 7.810418765075156e-07), (1, 6.538721663673641e-06), (2, 1.15284783532843e-05), (3, 5.7887646107701585e-06), (4, 1.4039976576896152e-06), (5, 1.2986163255845895e-06), (6, 1.5593875559716253e-06), (7, 2.1944586023892043e-06), (8, 1.5135079820538522e-06), (9, 1.699237031971279e-06), (10, 3.5006978578167036e-06), (11, 8.977900506579317e-06), (12, 1.2552409316413105e-05), (13, 0.0003729112504515797), (14, 0.0005663694464601576), (15, 0.00015275787154678255), (16, 0.01134539395570755), (17, 5.111181235406548e-05), (18, 3.590273990994319e-05), (19, 0.0009647750994190574), (20, 0.0006535054999403656), (21, 6.040683729224838e-05), (22, 0.0001365526404697448), (23, 6.386274617398158e-05), (24, 0.00013759596913587302), (25, 0.00031878502340987325), (26, 0.0002957240503747016), (27, 0.005397094879299402)], Temporal MSE: [(0, 6.817496114308597e-07), (1, 2.3819521288714895e-07), (2, 3.177500218498608e-07), (3, 5.160341061127838e-07), (4, 5.600081749435049e-07), (5, 6.015453664076631e-07), (6, 6.80750588344381e-07), (7, 8.532299489161232e-07), (8, 5.793447712676425e-07), (9, 6.537326271427446e-07), (10, 2.2850333607493667e-06), (11, 5.366943241824629e-06), (12, 1.0156732059840579e-05), (13, 1.1359656127751805e-05), (14, 1.0190365173912141e-05), (15, 2.579392821644433e-05), (16, 1.2563488780870102e-05), (17, 1.4818666386418045e-05), (18, 2.1729325453634374e-05), (19, 2.1753148757852614e-05), (20, 2.7351799872121774e-05), (21, 4.5432851038640365e-05), (22, 5.3182669944362715e-05), (23, 8.095729572232813e-05), (24, 7.326753257075325e-05), (25, 0.00020028906874358654), (26, 4.364691994851455e-05), (27, 2.7130661692353897e-05)]
+
+ 61%|██████▏   | 92/150 [13:51<08:33,  8.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 342, Spatial MSE: [(0, 1.1330927236485877e-06), (1, 8.587007869209629e-06), (2, 1.8853204892366193e-05), (3, 1.2011954822810367e-05), (4, 2.5531602659611963e-05), (5, 3.867564009851776e-05), (6, 5.4261610785033554e-05), (7, 0.0001414826838299632), (8, 0.00016157192294485867), (9, 0.00023176051035989076), (10, 0.000312142918119207), (11, 0.0004389586974866688), (12, 0.0004655220254790038), (13, 0.0017802438233047724), (14, 0.004677847493439913), (15, 0.004019525367766619), (16, 0.05759485438466072), (17, 0.002546879695728421), (18, 0.0024873968213796616), (19, 0.005507707595825195), (20, 0.005296977236866951), (21, 0.00429141242057085), (22, 0.0028871505055576563), (23, 0.005070704035460949), (24, 0.008125186897814274), (25, 0.014561685733497143), (26, 0.013184575363993645), (27, 0.00612192926928401)], Temporal MSE: [(0, 9.53917719925812e-07), (1, 4.092796928034659e-07), (2, 2.09049380828219e-06), (3, 6.199119525263086e-06), (4, 1.430930115020601e-05), (5, 1.818905002437532e-05), (6, 5.751793287345208e-05), (7, 5.406935088103637e-05), (8, 4.208666359772906e-05), (9, 7.22514305380173e-05), (10, 0.0001893169101094827), (11, 0.00023863569367676973), (12, 0.0002779824426397681), (13, 0.000524503062479198), (14, 0.0011559546692296863), (15, 0.0014095378573983908), (16, 0.0016244995640590787), (17, 0.0015843132277950644), (18, 0.0011270686518400908), (19, 0.0016994449542835355), (20, 0.002958305412903428), (21, 0.002524625975638628), (22, 0.0013197502121329308), (23, 0.0019287124741822481), (24, 0.003931277431547642), (25, 0.0068366918712854385), (26, 0.0014931245241314173), (27, 0.000103900303656701)]
+
+ 62%|██████▏   | 93/150 [14:01<08:40,  9.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 336, Spatial MSE: [(0, 5.796849791295244e-07), (1, 6.378582838806324e-06), (2, 1.2747575965477154e-05), (3, 6.082419986341847e-06), (4, 1.3873426496502361e-06), (5, 1.546188741485821e-06), (6, 1.4518909665639512e-06), (7, 2.4049015792115824e-06), (8, 1.6616523907941882e-06), (9, 1.7295820953222574e-06), (10, 1.9495239484967897e-06), (11, 5.627361133520026e-06), (12, 2.0689074517576955e-05), (13, 0.0007090797880664468), (14, 0.001622157753445208), (15, 0.0004928823327645659), (16, 0.027445122599601746), (17, 0.00010872675193240866), (18, 9.441610745852813e-05), (19, 0.002239180263131857), (20, 0.0016123788664117455), (21, 0.00019090647401753813), (22, 0.00027821282856166363), (23, 0.00011212201206944883), (24, 0.00025341397849842906), (25, 0.0006437926786020398), (26, 0.0004483188677113503), (27, 0.014208941720426083)], Temporal MSE: [(0, 5.432318630482769e-07), (1, 2.771270146695315e-07), (2, 3.7013430187471386e-07), (3, 5.773833890998503e-07), (4, 6.704758561681956e-07), (5, 7.218451401058701e-07), (6, 8.378625580007792e-07), (7, 1.0127514542546123e-06), (8, 6.857995913378545e-07), (9, 7.279164719875553e-07), (10, 1.277245132769167e-06), (11, 7.482537967007374e-06), (12, 1.709704338281881e-05), (13, 2.531850441300776e-05), (14, 2.9211796572781168e-05), (15, 6.605200906051323e-05), (16, 3.513763294904493e-05), (17, 3.427968476898968e-05), (18, 4.164345227763988e-05), (19, 3.3827567676780745e-05), (20, 2.6435947802383453e-05), (21, 0.00012192620488349348), (22, 0.00012103361950721592), (23, 0.00018064852338284254), (24, 0.0001354518171865493), (25, 0.00030421148403547704), (26, 7.570274465251714e-05), (27, 7.219031977001578e-05)]
+
+ 63%|██████▎   | 94/150 [14:09<08:07,  8.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 330, Spatial MSE: [(0, 8.220584390983277e-07), (1, 7.120748705347069e-06), (2, 1.3781347661279142e-05), (3, 6.560036126757041e-06), (4, 3.6476808418228757e-06), (5, 5.991651050862856e-06), (6, 9.050009794009384e-06), (7, 8.132640687108506e-06), (8, 1.2613261787919328e-05), (9, 2.5964414817281067e-05), (10, 3.5600798582891e-05), (11, 5.224560300121084e-05), (12, 8.342176442965865e-05), (13, 0.0009064921760000288), (14, 0.002319823484867811), (15, 0.001152098411694169), (16, 0.04014052450656891), (17, 0.00017519209359306842), (18, 0.00021763649419881403), (19, 0.0026885634288191795), (20, 0.002060616621747613), (21, 0.00032765622017905116), (22, 0.0003971767728216946), (23, 0.00029702074243687093), (24, 0.0007406918448396027), (25, 0.0018852123757824302), (26, 0.0035233243834227324), (27, 0.013643148355185986)], Temporal MSE: [(0, 5.832629312862991e-07), (1, 3.1399562772094214e-07), (2, 7.905657639639685e-07), (3, 1.4957894336475874e-06), (4, 2.883815568566206e-06), (5, 2.3512320694862865e-06), (6, 3.8051234696467873e-06), (7, 4.656730652641272e-06), (8, 4.51851474281284e-06), (9, 4.838422228203854e-06), (10, 8.344387424585875e-06), (11, 2.2026184524293058e-05), (12, 3.431231380091049e-05), (13, 4.7628684114897624e-05), (14, 6.338751700241119e-05), (15, 9.363796561956406e-05), (16, 5.65582413400989e-05), (17, 5.3579529776470736e-05), (18, 6.36792610748671e-05), (19, 6.581696652574465e-05), (20, 7.96695239841938e-05), (21, 0.00015876810357440263), (22, 0.00016749442147556692), (23, 0.00022944573720451444), (24, 0.0003080155292991549), (25, 0.0006320576649159193), (26, 0.0002111341163981706), (27, 8.294167491840199e-05)]
+
+ 63%|██████▎   | 95/150 [14:18<08:12,  8.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 324, Spatial MSE: [(0, 6.830864549556281e-07), (1, 6.292048055911437e-06), (2, 1.1426990567997564e-05), (3, 5.655580935126636e-06), (4, 1.442363895876042e-06), (5, 1.621999103917915e-06), (6, 1.6033955034799874e-06), (7, 2.5158799417113187e-06), (8, 1.7144195680884877e-06), (9, 1.9256124232924776e-06), (10, 3.950317477574572e-06), (11, 1.0489657142898068e-05), (12, 2.923718602687586e-05), (13, 0.00047293491661548615), (14, 0.0006374834338203073), (15, 0.0002115502575179562), (16, 0.011261392384767532), (17, 6.101024700910784e-05), (18, 0.00012138298188801855), (19, 0.0008659629384055734), (20, 0.000602366344537586), (21, 0.00020477428915910423), (22, 0.0002178058202844113), (23, 7.947355334181339e-05), (24, 0.0001664832525420934), (25, 0.00035819440381601453), (26, 0.000303665961837396), (27, 0.004678817931562662)], Temporal MSE: [(0, 6.405189765246178e-07), (1, 2.973527557514899e-07), (2, 3.76227490050951e-07), (3, 5.758789143328613e-07), (4, 6.666898570983903e-07), (5, 7.226579441521608e-07), (6, 8.429363447248761e-07), (7, 1.0440418236612459e-06), (8, 6.879800480419362e-07), (9, 7.792290261932067e-07), (10, 2.48345713771414e-06), (11, 1.71125811903039e-05), (12, 1.4283468772191554e-05), (13, 1.2719791811832692e-05), (14, 1.5189277291938197e-05), (15, 3.285155617049895e-05), (16, 1.6299060007440858e-05), (17, 1.850927947089076e-05), (18, 1.925943979586009e-05), (19, 1.4462319995800499e-05), (20, 1.3900940757594071e-05), (21, 5.410073572420515e-05), (22, 7.075192843331024e-05), (23, 0.00010478379408596084), (24, 8.529731712769717e-05), (25, 0.00016576987400185317), (26, 4.2684197978815064e-05), (27, 2.4934124667197466e-05)]
+
+ 64%|██████▍   | 96/150 [14:26<07:38,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 318, Spatial MSE: [(0, 8.629401690996019e-07), (1, 9.127460543822963e-06), (2, 1.6421749023720622e-05), (3, 1.274025362363318e-05), (4, 2.7630509066511877e-05), (5, 3.6179699236527085e-05), (6, 5.459908425109461e-05), (7, 0.00014468411973211914), (8, 0.00017637830751482397), (9, 0.00026156578678637743), (10, 0.0003930639068130404), (11, 0.0005421761889010668), (12, 0.0005468101589940488), (13, 0.002034526551142335), (14, 0.005046351812779903), (15, 0.004390955902636051), (16, 0.06447480618953705), (17, 0.0029029659926891327), (18, 0.002844051690772176), (19, 0.007766388822346926), (20, 0.007436502259224653), (21, 0.004281234927475452), (22, 0.0028891523834317923), (23, 0.004925483837723732), (24, 0.007566745858639479), (25, 0.014967318624258041), (26, 0.014820773154497147), (27, 0.0053144837729632854)], Temporal MSE: [(0, 8.110895350910141e-07), (1, 3.691020253882016e-07), (2, 2.2127010197436903e-06), (3, 5.203047294344287e-06), (4, 1.3374598893278744e-05), (5, 1.8791195543599315e-05), (6, 5.9786874771816656e-05), (7, 6.0429752920754254e-05), (8, 4.8044570576166734e-05), (9, 8.621705637779087e-05), (10, 0.0002441002579871565), (11, 0.0002942389983218163), (12, 0.0003389621269889176), (13, 0.0006479044095613062), (14, 0.0014254817506298423), (15, 0.0017283110646530986), (16, 0.0019435248104855418), (17, 0.0017557902028784156), (18, 0.0011930698528885841), (19, 0.001774486736394465), (20, 0.002824939088895917), (21, 0.002409011824056506), (22, 0.0012600685004144907), (23, 0.0017861834494397044), (24, 0.0033801053650677204), (25, 0.005689695477485657), (26, 0.0014404322719201446), (27, 9.691505692899227e-05)]
+
+ 65%|██████▍   | 97/150 [14:35<07:36,  8.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 312, Spatial MSE: [(0, 5.035799404140562e-07), (1, 6.408457920770161e-06), (2, 1.1363810699549504e-05), (3, 6.0602446865232196e-06), (4, 1.3537201084545814e-06), (5, 1.6320400391123258e-06), (6, 1.4821963532085647e-06), (7, 2.24539121518319e-06), (8, 1.5846364931348944e-06), (9, 1.916556357173249e-06), (10, 1.6717704056645744e-06), (11, 3.3371181871189037e-06), (12, 9.95930349745322e-06), (13, 0.00020527788728941232), (14, 0.0005232334951870143), (15, 0.0001390387915307656), (16, 0.008958234451711178), (17, 3.884835678036325e-05), (18, 2.8347592888167128e-05), (19, 0.0008177765994332731), (20, 0.0006109656533226371), (21, 6.284272240009159e-05), (22, 0.00014327798271551728), (23, 6.001019573886879e-05), (24, 0.0001241435675183311), (25, 0.0002935683005489409), (26, 0.00028646341525018215), (27, 0.004977749660611153)], Temporal MSE: [(0, 5.298982159729348e-07), (1, 2.7529381441127043e-07), (2, 3.65138305369328e-07), (3, 5.838217020937009e-07), (4, 6.945332415853045e-07), (5, 7.506091606046539e-07), (6, 8.419597179454286e-07), (7, 1.0071654514831607e-06), (8, 7.190438395809906e-07), (9, 7.285133847290126e-07), (10, 1.0804246812767815e-06), (11, 4.067328973178519e-06), (12, 7.006051419011783e-06), (13, 7.83619088906562e-06), (14, 1.1509075193316676e-05), (15, 2.130466054950375e-05), (16, 1.2596760825545061e-05), (17, 1.2144430002081208e-05), (18, 1.4531977285514586e-05), (19, 1.3279217455419712e-05), (20, 1.1204220754734706e-05), (21, 4.0973365685204044e-05), (22, 5.025810969527811e-05), (23, 7.266940519912168e-05), (24, 5.3279763960745186e-05), (25, 0.00012587029777932912), (26, 3.715350612765178e-05), (27, 2.4414050130872056e-05)]
+
+ 65%|██████▌   | 98/150 [14:42<07:15,  8.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 306, Spatial MSE: [(0, 8.472720196550654e-07), (1, 7.149208158807596e-06), (2, 9.462793968850747e-06), (3, 6.044280326022999e-06), (4, 3.4603424410306616e-06), (5, 6.660533017566195e-06), (6, 1.031873034662567e-05), (7, 8.655408237245865e-06), (8, 1.3207017218519468e-05), (9, 2.4976867280201986e-05), (10, 3.2373420253861696e-05), (11, 4.460020500118844e-05), (12, 5.934934233664535e-05), (13, 0.00025778336566872895), (14, 0.0006319659878499806), (15, 0.00018722651293501258), (16, 0.009037552401423454), (17, 7.093486783560365e-05), (18, 7.284109597094357e-05), (19, 0.0008738897740840912), (20, 0.0006664140964858234), (21, 0.00010282246512360871), (22, 0.00022308912593871355), (23, 0.00019926749519072473), (24, 0.0005500382976606488), (25, 0.001658531022258103), (26, 0.0036804573610424995), (27, 0.004537020809948444)], Temporal MSE: [(0, 7.01168005434738e-07), (1, 3.4355051070633635e-07), (2, 7.428129720210563e-07), (3, 1.4349483308251365e-06), (4, 2.861159828171367e-06), (5, 2.4295036382682156e-06), (6, 3.880094936903333e-06), (7, 5.359649549063761e-06), (8, 4.585021542879986e-06), (9, 4.955811618856387e-06), (10, 7.980436748766806e-06), (11, 1.2578734640555922e-05), (12, 1.8077646018355154e-05), (13, 2.3268641598406248e-05), (14, 3.655461114249192e-05), (15, 3.930594903067686e-05), (16, 2.9194123271736316e-05), (17, 2.527386459405534e-05), (18, 2.919798316725064e-05), (19, 3.060691960854456e-05), (20, 3.885785554302856e-05), (21, 6.766367732780054e-05), (22, 8.185320621123537e-05), (23, 0.00011684285709634423), (24, 0.00021955555712338537), (25, 0.0004633637727238238), (26, 0.00019896398589480668), (27, 4.028958937851712e-05)]
+
+ 66%|██████▌   | 99/150 [14:50<06:57,  8.20s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 300, Spatial MSE: [(0, 6.925664592927205e-07), (1, 6.149290584289702e-06), (2, 1.0729669156717137e-05), (3, 5.655300356011139e-06), (4, 1.5347709449997637e-06), (5, 1.8895863149737124e-06), (6, 1.689332975729485e-06), (7, 2.77128037851071e-06), (8, 1.9159281237079995e-06), (9, 2.596555532363709e-06), (10, 1.991880253626732e-06), (11, 2.824886450980557e-06), (12, 6.796964953537099e-06), (13, 0.000197020432096906), (14, 0.00035318618756718934), (15, 0.0002197968278778717), (16, 0.005185949616134167), (17, 3.501353785395622e-05), (18, 3.2367704989155754e-05), (19, 0.0004530360456556082), (20, 0.00034032738767564297), (21, 8.866293501341715e-05), (22, 0.00016587604477535933), (23, 6.510337698273361e-05), (24, 0.00012392326607368886), (25, 0.0002669747918844223), (26, 0.00027065316680818796), (27, 0.003334563923999667)], Temporal MSE: [(0, 7.34084380837885e-07), (1, 3.66706700560826e-07), (2, 4.4646066044151667e-07), (3, 6.558536256306979e-07), (4, 7.849825465200411e-07), (5, 8.575005381317169e-07), (6, 1.0004773685068358e-06), (7, 1.173259761344525e-06), (8, 8.275230243270926e-07), (9, 9.038566304298001e-07), (10, 1.3237040548119694e-06), (11, 2.4796513571345713e-06), (12, 4.8000274546211585e-06), (13, 7.802958862157539e-06), (14, 1.3528542694984935e-05), (15, 9.273013347410597e-06), (16, 8.452265319647267e-06), (17, 1.2318490007601213e-05), (18, 1.0426088010717649e-05), (19, 9.6302101155743e-06), (20, 1.0124831533175893e-05), (21, 4.5502438297262415e-05), (22, 5.014220005250536e-05), (23, 7.299306889763102e-05), (24, 5.9423226048238575e-05), (25, 0.00011914830975001678), (26, 3.780370388994925e-05), (27, 1.5578230886603706e-05)]
+
+ 67%|██████▋   | 100/150 [14:59<07:06,  8.53s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 294, Spatial MSE: [(0, 7.879473287175642e-07), (1, 9.048451829585247e-06), (2, 1.6907204553717747e-05), (3, 1.3494565791916102e-05), (4, 2.530147321522236e-05), (5, 3.625298268161714e-05), (6, 5.4996264225337654e-05), (7, 0.00013612314069177955), (8, 0.0001757179998094216), (9, 0.0002479034592397511), (10, 0.00033027637982741), (11, 0.0004053930751979351), (12, 0.0004333541146479547), (13, 0.002038288628682494), (14, 0.006465573329478502), (15, 0.004852235317230225), (16, 0.09545484930276871), (17, 0.0019766909535974264), (18, 0.002086496911942959), (19, 0.006542328745126724), (20, 0.006014951504766941), (21, 0.003177823033183813), (22, 0.0023764935322105885), (23, 0.0041072736494243145), (24, 0.006177159491926432), (25, 0.012867958284914494), (26, 0.014684941619634628), (27, 0.004169150721281767)], Temporal MSE: [(0, 7.50598474041908e-07), (1, 3.5660556818584155e-07), (2, 2.4261123598989798e-06), (3, 4.836093012272613e-06), (4, 1.4316387932922225e-05), (5, 1.8696220649871975e-05), (6, 5.795072502223775e-05), (7, 5.977793625788763e-05), (8, 4.500366412685253e-05), (9, 7.89677505963482e-05), (10, 0.00019880410400219262), (11, 0.0002258451422676444), (12, 0.00026956512010656297), (13, 0.0004914627061225474), (14, 0.0010855053551495075), (15, 0.0011318853357806802), (16, 0.0012915945844724774), (17, 0.0012418528785929084), (18, 0.0008713842835277319), (19, 0.001244257902726531), (20, 0.0020054299384355545), (21, 0.0017730735708028078), (22, 0.001048324746079743), (23, 0.0014841912779957056), (24, 0.0027283066883683205), (25, 0.004545281175523996), (26, 0.0012944552581757307), (27, 9.190630953526124e-05)]
+
+ 67%|██████▋   | 101/150 [15:08<07:00,  8.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 288, Spatial MSE: [(0, 6.20462458300608e-07), (1, 6.746308372385101e-06), (2, 1.1271692528680433e-05), (3, 6.133363058324903e-06), (4, 1.4162530987960054e-06), (5, 2.064124146272661e-06), (6, 1.7204240521095926e-06), (7, 2.7928831514145713e-06), (8, 1.843323843786493e-06), (9, 1.921377361213672e-06), (10, 3.358623416716e-06), (11, 7.826891305739991e-06), (12, 1.0832468433363829e-05), (13, 0.0002830138837452978), (14, 0.0009547612280584872), (15, 0.0003202633233740926), (16, 0.02531041018664837), (17, 9.402921568835154e-05), (18, 6.286131247179583e-05), (19, 0.0020578072872012854), (20, 0.0015811652410775423), (21, 0.0001416391460224986), (22, 0.0002533838851377368), (23, 0.00010040940105682239), (24, 0.00023086744477041066), (25, 0.0006419260753318667), (26, 0.000380413985112682), (27, 0.010843966156244278)], Temporal MSE: [(0, 5.786990868728026e-07), (1, 3.052566057704098e-07), (2, 3.840735303128895e-07), (3, 6.232857572285866e-07), (4, 7.564296424789063e-07), (5, 8.428983733210771e-07), (6, 9.076155151888088e-07), (7, 1.129748397943331e-06), (8, 7.939327133499319e-07), (9, 8.557259434383013e-07), (10, 2.3791972125764005e-06), (11, 4.196707777737174e-06), (12, 7.664264558115974e-06), (13, 1.1739095498342067e-05), (14, 1.968337528523989e-05), (15, 3.545592335285619e-05), (16, 2.5882249246933497e-05), (17, 4.511885344982147e-05), (18, 3.723326881299727e-05), (19, 3.501322862575762e-05), (20, 2.501451308489777e-05), (21, 0.000112774723675102), (22, 0.00010760126315290108), (23, 0.00015416288806591183), (24, 0.0001057203407981433), (25, 0.00022822011669632047), (26, 6.735552597092465e-05), (27, 6.570355617441237e-05)]
+
+ 68%|██████▊   | 102/150 [15:19<07:20,  9.18s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 282, Spatial MSE: [(0, 8.120063625938201e-07), (1, 7.589948836539406e-06), (2, 9.433904779143631e-06), (3, 6.252675575524336e-06), (4, 3.850756002066191e-06), (5, 6.735700935678324e-06), (6, 9.09132904780563e-06), (7, 9.49028162722243e-06), (8, 1.391351815982489e-05), (9, 2.9853648811695166e-05), (10, 4.186977457720786e-05), (11, 5.539597987080924e-05), (12, 6.47386914351955e-05), (13, 0.0003977683372795582), (14, 0.0013835671124979854), (15, 0.00057280930923298), (16, 0.02432238683104515), (17, 0.0001422640198143199), (18, 0.0004898609477095306), (19, 0.0026677914429455996), (20, 0.0022673914209008217), (21, 0.00032580099650658667), (22, 0.00039281125646084547), (23, 0.0003103540511801839), (24, 0.000806348689366132), (25, 0.00208231876604259), (26, 0.003567011561244726), (27, 0.010164261795580387)], Temporal MSE: [(0, 6.847376425866969e-07), (1, 3.383200635198591e-07), (2, 7.35948219698912e-07), (3, 1.4027204997546505e-06), (4, 2.9681202704523457e-06), (5, 2.5893943984556245e-06), (6, 4.018551408080384e-06), (7, 5.7270744946436025e-06), (8, 4.8560382310824934e-06), (9, 5.604959369520657e-06), (10, 1.0408072739664931e-05), (11, 1.7739057511789724e-05), (12, 2.2056656234781258e-05), (13, 3.079791713389568e-05), (14, 4.7072993766050786e-05), (15, 5.63960311410483e-05), (16, 4.219032052787952e-05), (17, 5.231053364695981e-05), (18, 5.345213139662519e-05), (19, 6.625128298765048e-05), (20, 9.678216156316921e-05), (21, 0.0001542483369121328), (22, 0.0001525646075606346), (23, 0.00020128116011619568), (24, 0.0002873404882848263), (25, 0.0005718164029531181), (26, 0.000244012990151532), (27, 8.238802547566593e-05)]
+
+ 69%|██████▊   | 103/150 [15:28<07:15,  9.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 276, Spatial MSE: [(0, 4.967384938936448e-07), (1, 6.365318768075667e-06), (2, 1.1001953680533916e-05), (3, 5.759016403317219e-06), (4, 1.594776790625474e-06), (5, 2.1650032522302354e-06), (6, 1.6801458286863635e-06), (7, 3.548510221662582e-06), (8, 2.2805031676398357e-06), (9, 1.7920929167303257e-06), (10, 4.585327587847132e-06), (11, 2.7025009330827743e-05), (12, 7.154805643949658e-05), (13, 0.0019435514695942402), (14, 0.001810858491808176), (15, 0.0005159928114153445), (16, 0.0179738812148571), (17, 6.46090556983836e-05), (18, 6.624167872359976e-05), (19, 0.0014760532649233937), (20, 0.0010014508152380586), (21, 0.0002085183805320412), (22, 0.0002614374679978937), (23, 0.00012678760685957968), (24, 0.0002510550257284194), (25, 0.0004958392819389701), (26, 0.0004053973243571818), (27, 0.01614794321358204)], Temporal MSE: [(0, 6.311332754194154e-07), (1, 3.4625554690137506e-07), (2, 4.2962327029272274e-07), (3, 6.507344210149313e-07), (4, 8.490633831570449e-07), (5, 8.771808666097058e-07), (6, 1.2110651823604712e-06), (7, 1.2329172704994562e-06), (8, 8.421590678153734e-07), (9, 9.450221227780276e-07), (10, 3.3977935345319565e-06), (11, 7.326775084948167e-05), (12, 4.0601484215585515e-05), (13, 3.555411967681721e-05), (14, 2.6372163119958714e-05), (15, 4.734219692181796e-05), (16, 2.081043021462392e-05), (17, 2.4093775209621526e-05), (18, 3.5532535548554733e-05), (19, 3.1756502721691504e-05), (20, 3.459783329162747e-05), (21, 8.092215284705162e-05), (22, 9.562270133756101e-05), (23, 0.0001437751780031249), (24, 0.0001386768853990361), (25, 0.000305160996504128), (26, 9.113699343288317e-05), (27, 6.29029527772218e-05)]
+
+ 69%|██████▉   | 104/150 [15:38<07:14,  9.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 270, Spatial MSE: [(0, 8.320578785969701e-07), (1, 9.351179869554471e-06), (2, 1.4538100913341623e-05), (3, 1.1827019989141263e-05), (4, 2.1518149878829718e-05), (5, 3.5273409594083205e-05), (6, 4.858072134084068e-05), (7, 0.00012163445353507996), (8, 0.00017478455265518278), (9, 0.0002186942583648488), (10, 0.00028743009897880256), (11, 0.0003527618246152997), (12, 0.0004345080815255642), (13, 0.0027391212061047554), (14, 0.00428438326343894), (15, 0.003455640049651265), (16, 0.047107044607400894), (17, 0.0017373624723404646), (18, 0.0018214641604572535), (19, 0.004087395034730434), (20, 0.003749074414372444), (21, 0.0028478973545134068), (22, 0.0021846166346222162), (23, 0.0036722873337566853), (24, 0.005497221369296312), (25, 0.011773129925131798), (26, 0.013375597074627876), (27, 0.01610490307211876)], Temporal MSE: [(0, 8.079244935288443e-07), (1, 3.989592016750976e-07), (2, 2.132411282218527e-06), (3, 4.634923698176863e-06), (4, 1.3999270777276251e-05), (5, 1.7532423953525722e-05), (6, 5.163850073586218e-05), (7, 5.697043889085762e-05), (8, 4.4425727537600324e-05), (9, 7.142138201743364e-05), (10, 0.00017306898371316493), (11, 0.00025454876595176756), (12, 0.00025921696214936674), (13, 0.0004536804626695812), (14, 0.0010321229929104447), (15, 0.0010153410257771611), (16, 0.0011522714048624039), (17, 0.0010901052737608552), (18, 0.0007839250611141324), (19, 0.001101848902180791), (20, 0.0017417817143723369), (21, 0.0014508655294775963), (22, 0.0009064329788088799), (23, 0.0013116106856614351), (24, 0.0024683272931724787), (25, 0.004315442871302366), (26, 0.0012706745183095336), (27, 0.00013916369061917067)]
+
+ 70%|███████   | 105/150 [15:46<06:48,  9.07s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 264, Spatial MSE: [(0, 1.0343352414565743e-06), (1, 7.620978522027144e-06), (2, 1.1916741641471162e-05), (3, 6.437326646846486e-06), (4, 1.6435471934528323e-06), (5, 2.3502825570176356e-06), (6, 2.0764903183589922e-06), (7, 3.230913989682449e-06), (8, 2.2809381334809586e-06), (9, 2.3981085632840404e-06), (10, 2.4829566882544896e-06), (11, 5.592081834038254e-06), (12, 1.549092667119112e-05), (13, 0.000656162213999778), (14, 0.0015458962880074978), (15, 0.0008119458798319101), (16, 0.030606048181653023), (17, 9.63286220212467e-05), (18, 0.0001876687747426331), (19, 0.002276405692100525), (20, 0.0016428128583356738), (21, 0.000391379464417696), (22, 0.00041314642294310033), (23, 0.00016942928778007627), (24, 0.0003363773867022246), (25, 0.0008017722284421325), (26, 0.0004688313929364085), (27, 0.01697702892124653)], Temporal MSE: [(0, 9.0453175971561e-07), (1, 3.980680673976167e-07), (2, 5.163565788279811e-07), (3, 7.511098942813987e-07), (4, 9.361851880385075e-07), (5, 1.023861159410444e-06), (6, 1.1532416692716652e-06), (7, 1.424155811946548e-06), (8, 1.0101028919962118e-06), (9, 1.0577388138699462e-06), (10, 1.6929926687225816e-06), (11, 8.757484465604648e-06), (12, 1.1461752364994027e-05), (13, 2.4955163098638877e-05), (14, 4.25158541474957e-05), (15, 5.0284095777897164e-05), (16, 3.0198874810594134e-05), (17, 4.101539161638357e-05), (18, 5.2420233259908855e-05), (19, 4.2368668800918385e-05), (20, 3.285482671344653e-05), (21, 0.00012834295921493322), (22, 0.00015699661162216216), (23, 0.0002383617975283414), (24, 0.0001820322941057384), (25, 0.0005212605465203524), (26, 0.00010430248221382499), (27, 9.078649600269273e-05)]
+
+ 71%|███████   | 106/150 [15:55<06:32,  8.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 258, Spatial MSE: [(0, 1.1611019772317377e-06), (1, 7.676200766582042e-06), (2, 8.851156053424347e-06), (3, 6.390973794623278e-06), (4, 3.7464376418938627e-06), (5, 7.216680842248024e-06), (6, 1.0318374734197278e-05), (7, 9.361973752675112e-06), (8, 1.2878848792752251e-05), (9, 2.825331648637075e-05), (10, 3.2419466151623055e-05), (11, 4.948247442371212e-05), (12, 5.675967986462638e-05), (13, 0.0006577902822755277), (14, 0.0017088778549805284), (15, 0.0010310836369171739), (16, 0.03513177111744881), (17, 0.00013711203064303845), (18, 0.00023444586258847266), (19, 0.0026182932779192924), (20, 0.0020888312719762325), (21, 0.0004695439129136503), (22, 0.0005498689133673906), (23, 0.0003654141037259251), (24, 0.0008339342311955988), (25, 0.002108530839905143), (26, 0.0037931164260953665), (27, 0.0162209365516901)], Temporal MSE: [(0, 8.213999080908252e-07), (1, 4.1662460148472746e-07), (2, 8.334630479112093e-07), (3, 1.6358751508960268e-06), (4, 3.3349147088301834e-06), (5, 2.800459924401366e-06), (6, 4.45594514530967e-06), (7, 5.962276645732345e-06), (8, 5.646391855407273e-06), (9, 5.773559678345919e-06), (10, 8.913731107895728e-06), (11, 1.7346495951642282e-05), (12, 2.2221929611987434e-05), (13, 3.769802788156085e-05), (14, 6.343182030832395e-05), (15, 7.108069985406473e-05), (16, 4.647216701414436e-05), (17, 5.890902684768662e-05), (18, 7.179840031312779e-05), (19, 6.818655674578622e-05), (20, 8.40849315864034e-05), (21, 0.00016688316827639937), (22, 0.00019657236407510936), (23, 0.00028254828066565096), (24, 0.0003644044045358896), (25, 0.0007327907369472086), (26, 0.0003097747976426035), (27, 0.00011000080121448264)]
+
+ 71%|███████▏  | 107/150 [16:05<06:33,  9.16s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 252, Spatial MSE: [(0, 6.194213710841723e-07), (1, 7.446657491527731e-06), (2, 1.1540279956534505e-05), (3, 5.979292382107815e-06), (4, 1.5572597931168275e-06), (5, 2.3157799660111777e-06), (6, 1.8404174397801398e-06), (7, 2.962687403851305e-06), (8, 2.0953018520231126e-06), (9, 2.125775836248067e-06), (10, 4.039496616314864e-06), (11, 1.0709127309382893e-05), (12, 3.1076036975719035e-05), (13, 0.0009013660601340234), (14, 0.0020803092047572136), (15, 0.0006853844388388097), (16, 0.04439491033554077), (17, 0.00013607318396680057), (18, 0.00010373166878707707), (19, 0.003937189467251301), (20, 0.0029252860695123672), (21, 0.00027648129616864026), (22, 0.000397791329305619), (23, 0.0001806871296139434), (24, 0.0003848527849186212), (25, 0.0009327190928161144), (26, 0.0005073045613244176), (27, 0.021473318338394165)], Temporal MSE: [(0, 6.601022164431924e-07), (1, 3.6699634620163124e-07), (2, 4.5738985932075593e-07), (3, 6.917130122019444e-07), (4, 9.454532232666679e-07), (5, 1.025934579956811e-06), (6, 1.2528636261777137e-06), (7, 1.3452100802169298e-06), (8, 9.72505176832783e-07), (9, 1.0789823363666073e-06), (10, 3.1037020562507678e-06), (11, 1.2520275049610063e-05), (12, 2.4199069230235182e-05), (13, 2.7831631086883135e-05), (14, 3.677859785966575e-05), (15, 7.512121374020353e-05), (16, 4.054991222801618e-05), (17, 6.030372605891898e-05), (18, 6.791840132791549e-05), (19, 5.8483550674282014e-05), (20, 4.267778786015697e-05), (21, 0.00015709438594058156), (22, 0.00019466110097710043), (23, 0.0002713886497076601), (24, 0.00018236761388834566), (25, 0.0005106060416437685), (26, 0.0001161519976449199), (27, 0.00012000230344710872)]
+
+ 72%|███████▏  | 108/150 [16:13<06:21,  9.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 246, Spatial MSE: [(0, 6.901971687511832e-07), (1, 1.011353378999047e-05), (2, 1.388213877362432e-05), (3, 1.1902611731784418e-05), (4, 2.222159855591599e-05), (5, 3.4454758861102164e-05), (6, 4.7040110075613484e-05), (7, 0.00012015507672913373), (8, 0.00016364175826311111), (9, 0.00020245657651685178), (10, 0.0002607509377412498), (11, 0.0003269725712016225), (12, 0.00037412880919873714), (13, 0.0018447240581735969), (14, 0.005235474556684494), (15, 0.0036503789015114307), (16, 0.09395531564950943), (17, 0.0016696733655408025), (18, 0.0018734341720119119), (19, 0.009098506532609463), (20, 0.008072960190474987), (21, 0.002753147389739752), (22, 0.0021218012552708387), (23, 0.003408849937841296), (24, 0.0051026190631091595), (25, 0.011328763328492641), (26, 0.012999008409678936), (27, 0.021396739408373833)], Temporal MSE: [(0, 8.075198252299742e-07), (1, 3.7573187228190363e-07), (2, 2.283462208652054e-06), (3, 4.185292254987871e-06), (4, 1.4401984117284883e-05), (5, 1.715276994218584e-05), (6, 5.153761230758391e-05), (7, 5.610286461887881e-05), (8, 4.302566958358511e-05), (9, 6.914370169397444e-05), (10, 0.00016113099991343915), (11, 0.00019535419414751232), (12, 0.00023485766723752022), (13, 0.0004068775160703808), (14, 0.0009786817245185375), (15, 0.0009734220802783966), (16, 0.0010963332606479526), (17, 0.0010419588070362806), (18, 0.0007344158366322517), (19, 0.0010018872562795877), (20, 0.001522593665868044), (21, 0.0013962206430733204), (22, 0.0009706589044071734), (23, 0.0013549410505220294), (24, 0.0023107388988137245), (25, 0.004150936845690012), (26, 0.0012656506150960922), (27, 0.00020082166884094477)]
+
+ 73%|███████▎  | 109/150 [16:22<06:01,  8.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 240, Spatial MSE: [(0, 6.595067816306255e-07), (1, 7.822015504643787e-06), (2, 1.1747598364308942e-05), (3, 6.544970801769523e-06), (4, 1.741819005474099e-06), (5, 2.8633769488806138e-06), (6, 2.094525370921474e-06), (7, 3.3569363040442113e-06), (8, 2.4426376512565184e-06), (9, 2.552438445491134e-06), (10, 4.057114438182907e-06), (11, 1.1007503417204134e-05), (12, 2.5561541406204924e-05), (13, 0.001005204045213759), (14, 0.0020038397051393986), (15, 0.00064020452555269), (16, 0.04581332951784134), (17, 0.000146813879837282), (18, 9.126972145168111e-05), (19, 0.004055733326822519), (20, 0.002991696121171117), (21, 0.00016252561181318015), (22, 0.00040288770105689764), (23, 0.00017639614816289395), (24, 0.00037104461807757616), (25, 0.0009130614344030619), (26, 0.00048512511420994997), (27, 0.02230817638337612)], Temporal MSE: [(0, 7.528763603659172e-07), (1, 4.2418832890689373e-07), (2, 5.405815954873106e-07), (3, 8.056425144786772e-07), (4, 1.1313150025671348e-06), (5, 1.1866978866237332e-06), (6, 1.4794753724345355e-06), (7, 1.6147937458299566e-06), (8, 1.1537247246451443e-06), (9, 1.2079929092578823e-06), (10, 2.872639015549794e-06), (11, 1.2585610420501325e-05), (12, 2.133988164132461e-05), (13, 2.837234569597058e-05), (14, 4.149281812715344e-05), (15, 6.338109960779548e-05), (16, 3.812490831478499e-05), (17, 6.742899131495506e-05), (18, 6.808352918596938e-05), (19, 6.149789987830445e-05), (20, 4.5108841732144356e-05), (21, 0.00016305583994835615), (22, 0.00018375653598923236), (23, 0.0002662627084646374), (24, 0.00018434246885590255), (25, 0.00047615740913897753), (26, 0.00011581830040086061), (27, 0.00012105101632187143)]
+
+ 73%|███████▎  | 110/150 [16:29<05:39,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 234, Spatial MSE: [(0, 9.803319471757277e-07), (1, 7.510952855227515e-06), (2, 8.553809493605513e-06), (3, 6.775228030164726e-06), (4, 3.872289198625367e-06), (5, 8.399861144425813e-06), (6, 1.1283581443422008e-05), (7, 1.0518569069972727e-05), (8, 1.415889710187912e-05), (9, 2.6631001674104482e-05), (10, 3.820916754193604e-05), (11, 5.660668102791533e-05), (12, 7.30638494133018e-05), (13, 0.0010690612252801657), (14, 0.0021588997915387154), (15, 0.0007364119519479573), (16, 0.046336278319358826), (17, 0.00018715360783971846), (18, 0.0001871901040431112), (19, 0.004123974125832319), (20, 0.0030624503269791603), (21, 0.0002153671666746959), (22, 0.0005159647553227842), (23, 0.00039316393667832017), (24, 0.001036158879287541), (25, 0.0027475696988403797), (26, 0.004635870922356844), (27, 0.02202831394970417)], Temporal MSE: [(0, 8.44985663661646e-07), (1, 4.5139145754546917e-07), (2, 8.307615075864305e-07), (3, 1.6905067923289607e-06), (4, 3.405287088753539e-06), (5, 3.0569633509003324e-06), (6, 4.8540409807174e-06), (7, 6.834539817646146e-06), (8, 6.148795364424586e-06), (9, 6.539658897963818e-06), (10, 1.081584832718363e-05), (11, 2.2990041543380357e-05), (12, 3.454764373600483e-05), (13, 4.6814435336273164e-05), (14, 7.23625926184468e-05), (15, 8.507203165208921e-05), (16, 5.877561488887295e-05), (17, 8.490415348205715e-05), (18, 8.658373553771526e-05), (19, 8.258357411250472e-05), (20, 7.608239684486762e-05), (21, 0.00019782184972427785), (22, 0.0002247192314825952), (23, 0.00031949696131050587), (24, 0.0004140507662668824), (25, 0.0008938240353018045), (26, 0.00033688105759210885), (27, 0.00014466993161477149)]
+
+ 74%|███████▍  | 111/150 [16:37<05:22,  8.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 228, Spatial MSE: [(0, 8.381281872971158e-07), (1, 8.203663128369953e-06), (2, 1.146336489910027e-05), (3, 5.926688572799321e-06), (4, 1.6314254480676027e-06), (5, 2.6923330551653635e-06), (6, 2.1596517854050035e-06), (7, 4.337638074503047e-06), (8, 2.7889163902727887e-06), (9, 2.050712964773993e-06), (10, 2.483303205735865e-06), (11, 4.910232746624388e-06), (12, 1.674691702646669e-05), (13, 0.0006061738822609186), (14, 0.0013077399926260114), (15, 0.00046870147343724966), (16, 0.028295064345002174), (17, 9.102230978896841e-05), (18, 9.080216113943607e-05), (19, 0.0023227110505104065), (20, 0.0017241784371435642), (21, 0.0001414889411535114), (22, 0.00025941600324586034), (23, 0.00013817254512105137), (24, 0.0002826211857609451), (25, 0.0006742656696587801), (26, 0.0004370627866592258), (27, 0.015389453619718552)], Temporal MSE: [(0, 7.90916374171502e-07), (1, 4.079093116615695e-07), (2, 5.303500643094594e-07), (3, 7.298729087779066e-07), (4, 1.0122964795300504e-06), (5, 1.1251411251578247e-06), (6, 1.452169158255856e-06), (7, 1.5424936918861931e-06), (8, 1.0970557013934013e-06), (9, 1.1383360742911464e-06), (10, 1.7124060605056002e-06), (11, 7.112988441804191e-06), (12, 1.3072000911051873e-05), (13, 1.9338380297995172e-05), (14, 2.7893349397345446e-05), (15, 4.468002953217365e-05), (16, 2.8814085453632288e-05), (17, 5.3514548199018463e-05), (18, 4.3151259887963533e-05), (19, 3.778939208132215e-05), (20, 2.9785178412566893e-05), (21, 0.00010903781367233023), (22, 0.00012111817341065034), (23, 0.00017780085909180343), (24, 0.0001433121506124735), (25, 0.00036885254667140543), (26, 9.611478162696585e-05), (27, 7.960182119859383e-05)]
+
+ 75%|███████▍  | 112/150 [16:45<05:04,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 222, Spatial MSE: [(0, 1.0783110155898612e-06), (1, 9.034357390191872e-06), (2, 1.5118043847905938e-05), (3, 1.1746738891815767e-05), (4, 1.8274075046065263e-05), (5, 3.579007534426637e-05), (6, 5.086359306005761e-05), (7, 0.00012720191443804651), (8, 0.00017158158880192786), (9, 0.00020533207862172276), (10, 0.00021818710956722498), (11, 0.00027674957527779043), (12, 0.0003200867213308811), (13, 0.001476621488109231), (14, 0.003311299718916416), (15, 0.0024396765511482954), (16, 0.04704073816537857), (17, 0.0014226686907932162), (18, 0.0014689817326143384), (19, 0.004288158845156431), (20, 0.003820874262601137), (21, 0.0022231026086956263), (22, 0.001848716172389686), (23, 0.0032384584192186594), (24, 0.004927753005176783), (25, 0.011489217169582844), (26, 0.015314937569200993), (27, 0.015267457813024521)], Temporal MSE: [(0, 1.1454377499831025e-06), (1, 5.63548780974088e-07), (2, 2.6083882858074503e-06), (3, 4.693289156421088e-06), (4, 1.3323421626409981e-05), (5, 1.7410391592420638e-05), (6, 5.3066687542013824e-05), (7, 5.7642220781417564e-05), (8, 4.288188210921362e-05), (9, 6.500017479993403e-05), (10, 0.0001424365909770131), (11, 0.00016929060802794993), (12, 0.00019904357031919062), (13, 0.00036199315218254924), (14, 0.0008631327073089778), (15, 0.0008375172037631273), (16, 0.0009679396753199399), (17, 0.0009637610637582839), (18, 0.0006575586157850921), (19, 0.0008865337586030364), (20, 0.0013500928180292249), (21, 0.0011614126851782203), (22, 0.0007968247518874705), (23, 0.0011475058272480965), (24, 0.0020624264143407345), (25, 0.003779179882258177), (26, 0.0012783242855221033), (27, 0.0001625053700990975)]
+
+ 75%|███████▌  | 113/150 [16:53<05:01,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 216, Spatial MSE: [(0, 6.520208444271702e-07), (1, 9.076872629520949e-06), (2, 1.2471906302380376e-05), (3, 6.372948973876191e-06), (4, 1.6879851045814576e-06), (5, 2.765289309536456e-06), (6, 2.238897195638856e-06), (7, 3.244440904381918e-06), (8, 2.3908708044473315e-06), (9, 2.8060533168172697e-06), (10, 2.488448899384821e-06), (11, 5.41830422662315e-06), (12, 1.9492223145789467e-05), (13, 0.0008498159586451948), (14, 0.0015936261042952538), (15, 0.0008062014821916819), (16, 0.04978087544441223), (17, 0.00016746518667787313), (18, 0.0001814227580325678), (19, 0.004256159532815218), (20, 0.0031758130062371492), (21, 0.00029132768395356834), (22, 0.00047376620932482183), (23, 0.0002029072493314743), (24, 0.00042935795499943197), (25, 0.0010760726872831583), (26, 0.000468467827886343), (27, 0.022439494729042053)], Temporal MSE: [(0, 7.297446700249566e-07), (1, 4.0351972074859077e-07), (2, 5.293956064633676e-07), (3, 7.716389518463984e-07), (4, 1.114137603508425e-06), (5, 1.20944605441764e-06), (6, 1.6780568330432288e-06), (7, 1.551935497445811e-06), (8, 1.2200874834888964e-06), (9, 1.24360224162956e-06), (10, 1.860283532550966e-06), (11, 8.577801963838283e-06), (12, 1.8360489775659516e-05), (13, 2.6666359190130606e-05), (14, 4.777081267093308e-05), (15, 5.997248081257567e-05), (16, 3.9364491385640576e-05), (17, 7.92857535998337e-05), (18, 8.655852434458211e-05), (19, 7.40323739591986e-05), (20, 5.5019194405758753e-05), (21, 0.00018607389938551933), (22, 0.0002152671222575009), (23, 0.00030712917214259505), (24, 0.00021915206161793321), (25, 0.0005553230294026434), (26, 0.00012467046326491982), (27, 0.00014145192108117044)]
+
+ 76%|███████▌  | 114/150 [17:01<04:46,  7.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 210, Spatial MSE: [(0, 8.77884872352297e-07), (1, 9.411021892447025e-06), (2, 1.1416964298405219e-05), (3, 8.58481780596776e-06), (4, 7.474767699022777e-06), (5, 1.542241261631716e-05), (6, 2.006683098443318e-05), (7, 3.989046308561228e-05), (8, 5.449149466585368e-05), (9, 7.749588985461742e-05), (10, 8.22145157144405e-05), (11, 0.0001104380062315613), (12, 0.0001272763911401853), (13, 0.0012182262726128101), (14, 0.002896653488278389), (15, 0.0019708583131432533), (16, 0.061556603759527206), (17, 0.0005645622150041163), (18, 0.0007968902355059981), (19, 0.006078007631003857), (20, 0.005066183861345053), (21, 0.000893151038326323), (22, 0.0010273752268403769), (23, 0.0012163015780970454), (24, 0.002068900503218174), (25, 0.005857447162270546), (26, 0.008254731073975563), (27, 0.022666070610284805)], Temporal MSE: [(0, 9.944199064193526e-07), (1, 5.097572852719168e-07), (2, 1.371022563034785e-06), (3, 1.8765299500955734e-06), (4, 4.370085662230849e-06), (5, 4.768337930727284e-06), (6, 1.6888336176634766e-05), (7, 2.009250601986423e-05), (8, 1.2636554856726434e-05), (9, 2.0885940102743916e-05), (10, 4.3249416194157675e-05), (11, 5.620162846753374e-05), (12, 7.779153384035453e-05), (13, 0.00012881950533483177), (14, 0.00030564205371774733), (15, 0.00028807565104216337), (16, 0.0003005425096489489), (17, 0.0003181970096193254), (18, 0.0002559653657954186), (19, 0.0003084916970692575), (20, 0.0004183719283901155), (21, 0.00046317256055772305), (22, 0.0004132202884647995), (23, 0.0005896074580959976), (24, 0.0007737638079561293), (25, 0.0015184857184067369), (26, 0.00047034979797899723), (27, 0.00016445023356936872)]
+
+ 77%|███████▋  | 115/150 [17:09<04:42,  8.07s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 204, Spatial MSE: [(0, 6.803672363275837e-07), (1, 9.558040801493917e-06), (2, 1.3376789866015315e-05), (3, 6.907238912390312e-06), (4, 1.8420323613099754e-06), (5, 3.390243591638864e-06), (6, 2.5715198717080057e-06), (7, 4.63188553112559e-06), (8, 3.0364369649760192e-06), (9, 2.359775635341066e-06), (10, 5.0228477448399644e-06), (11, 1.0037727406597696e-05), (12, 2.657769073266536e-05), (13, 0.0006911155069246888), (14, 0.001798652345314622), (15, 0.0005022230907343328), (16, 0.04003049060702324), (17, 0.00013043372018728405), (18, 9.355505608255044e-05), (19, 0.003531161928549409), (20, 0.002635939745232463), (21, 0.00013402357581071556), (22, 0.00035975102218799293), (23, 0.0001579344243509695), (24, 0.000334993441356346), (25, 0.0008146401960402727), (26, 0.00041813336429186165), (27, 0.01836448162794113)], Temporal MSE: [(0, 7.417389724651002e-07), (1, 4.288443733457825e-07), (2, 5.813937491438992e-07), (3, 8.164833502632973e-07), (4, 1.229111262546212e-06), (5, 1.350201614513935e-06), (6, 1.6515366496605566e-06), (7, 1.7585738305569976e-06), (8, 1.3514983265849878e-06), (9, 1.4677932540507754e-06), (10, 3.750431460503023e-06), (11, 8.335216989507899e-06), (12, 2.0252338799764402e-05), (13, 2.3185797545011155e-05), (14, 3.5717828723136336e-05), (15, 8.136560063576326e-05), (16, 4.737660492537543e-05), (17, 6.542803748743609e-05), (18, 6.0086193116148934e-05), (19, 5.290405169944279e-05), (20, 3.884959369315766e-05), (21, 0.00014787768304813653), (22, 0.00016792892711237073), (23, 0.00023258528381120414), (24, 0.00015763840929139405), (25, 0.00047168423770926893), (26, 0.00011177436681464314), (27, 0.00011058786913054064)]
+
+ 77%|███████▋  | 116/150 [17:17<04:39,  8.23s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 198, Spatial MSE: [(0, 9.398413567396346e-07), (1, 9.2286927610985e-06), (2, 1.1652929060801398e-05), (3, 6.968834895815235e-06), (4, 4.186005298834061e-06), (5, 9.473072168475483e-06), (6, 1.2197891919640824e-05), (7, 1.1388648999854922e-05), (8, 1.561159297125414e-05), (9, 2.6989240723196417e-05), (10, 3.9641181501792744e-05), (11, 5.89429255342111e-05), (12, 7.32745393179357e-05), (13, 0.0007250108174048364), (14, 0.0018588316161185503), (15, 0.0005737669998779893), (16, 0.04012651741504669), (17, 0.0001683055452303961), (18, 0.00015709928993601352), (19, 0.003586175851523876), (20, 0.002686185296624899), (21, 0.00019324489403516054), (22, 0.0004846448719035834), (23, 0.00040883259498514235), (24, 0.0010125106200575829), (25, 0.002854572609066963), (26, 0.0049637784250080585), (27, 0.018780380487442017)], Temporal MSE: [(0, 9.479360301156703e-07), (1, 5.640488893732254e-07), (2, 8.759444654060644e-07), (3, 1.5859960740272072e-06), (4, 3.4867186968767783e-06), (5, 3.3171115774166537e-06), (6, 5.576702278631274e-06), (7, 7.621046279382426e-06), (8, 6.554274477821309e-06), (9, 7.249318969115848e-06), (10, 1.3091567780065816e-05), (11, 2.0915578716085292e-05), (12, 3.383163129910827e-05), (13, 3.9623853808734566e-05), (14, 6.512576510431245e-05), (15, 0.0001015988236758858), (16, 6.754980859113857e-05), (17, 8.087839523795992e-05), (18, 7.894406007835642e-05), (19, 7.533469033660367e-05), (20, 7.270010246429592e-05), (21, 0.000181155075551942), (22, 0.00021173465938773006), (23, 0.00029113393975421786), (24, 0.0003829165652859956), (25, 0.0008280647452920675), (26, 0.0003834819362964481), (27, 0.0001429100229870528)]
+
+ 78%|███████▊  | 117/150 [17:26<04:35,  8.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 192, Spatial MSE: [(0, 6.687059794785455e-07), (1, 9.356077498523518e-06), (2, 1.3624679013446439e-05), (3, 6.50190395390382e-06), (4, 1.9821704881906044e-06), (5, 3.661263463072828e-06), (6, 2.6136331143788993e-06), (7, 3.7790482565469574e-06), (8, 2.9197442472650437e-06), (9, 3.065017608605558e-06), (10, 3.4718468668870628e-06), (11, 8.66654227138497e-06), (12, 3.0719304049853235e-05), (13, 0.001468032831326127), (14, 0.0027345330454409122), (15, 0.0008742495556361973), (16, 0.059285443276166916), (17, 0.00020364519150462002), (18, 0.00013507399125956), (19, 0.005250116344541311), (20, 0.0039253016002476215), (21, 0.0005972837097942829), (22, 0.0006514732958748937), (23, 0.00026224946486763656), (24, 0.0005330170388333499), (25, 0.0013054478913545609), (26, 0.0005835036863572896), (27, 0.026642784476280212)], Temporal MSE: [(0, 9.631316970626358e-07), (1, 5.646484737553692e-07), (2, 7.08545428551588e-07), (3, 9.008265351440059e-07), (4, 1.3311847624208895e-06), (5, 1.4622195294577978e-06), (6, 1.8797916254698066e-06), (7, 1.8907250023403321e-06), (8, 1.4459100157182547e-06), (9, 1.474123337175115e-06), (10, 2.46709009843471e-06), (11, 1.416854684066493e-05), (12, 2.6043322577606887e-05), (13, 4.2578958527883515e-05), (14, 5.708644675905816e-05), (15, 9.074050467461348e-05), (16, 5.605678234132938e-05), (17, 0.00012890923244412988), (18, 9.278541256207973e-05), (19, 7.953812746563926e-05), (20, 6.144589133327827e-05), (21, 0.0002577837440185249), (22, 0.00030532319215126336), (23, 0.000386516738217324), (24, 0.0002624513872433454), (25, 0.0006846718606539071), (26, 0.000153269647853449), (27, 0.00014747735986020416)]
+
+ 79%|███████▊  | 118/150 [17:36<04:41,  8.80s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 186, Spatial MSE: [(0, 8.292818733934837e-07), (1, 1.0429531357658561e-05), (2, 1.7868722352432087e-05), (3, 1.2106132999178953e-05), (4, 1.6217225493164733e-05), (5, 3.6215387808624655e-05), (6, 4.7863024519756436e-05), (7, 0.00011608306522248313), (8, 0.00015735354099888355), (9, 0.00017595765530131757), (10, 0.00018128703231923282), (11, 0.00023814717133063823), (12, 0.0002887711743824184), (13, 0.002005174057558179), (14, 0.004223425406962633), (15, 0.0025416561402380466), (16, 0.07606194913387299), (17, 0.001278570620343089), (18, 0.0013206067960709333), (19, 0.006910808850079775), (20, 0.0057802931405603886), (21, 0.002489709295332432), (22, 0.002098212717100978), (23, 0.0031731638591736555), (24, 0.0048946719616651535), (25, 0.012120380997657776), (26, 0.014752618968486786), (27, 0.02705240249633789)], Temporal MSE: [(0, 1.0656347058102256e-06), (1, 5.644515681524354e-07), (2, 2.332989879505476e-06), (3, 4.293786787457066e-06), (4, 1.3626522559206933e-05), (5, 1.667871947574895e-05), (6, 5.3936237236484885e-05), (7, 5.556974792852998e-05), (8, 3.988685057265684e-05), (9, 5.9643796703312546e-05), (10, 0.00012361427070572972), (11, 0.00014833841123618186), (12, 0.00018458106205798686), (13, 0.0003237759228795767), (14, 0.0007706882315687835), (15, 0.0007185964495874941), (16, 0.0008073173812590539), (17, 0.0008637254359200597), (18, 0.0005832270835526288), (19, 0.0007596145733259618), (20, 0.0011104149743914604), (21, 0.001091598067432642), (22, 0.0009087935904972255), (23, 0.001266365870833397), (24, 0.002067344030365348), (25, 0.003851481480523944), (26, 0.001396249863319099), (27, 0.00025518008624203503)]
+
+ 79%|███████▉  | 119/150 [17:44<04:22,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 180, Spatial MSE: [(0, 7.795365490892436e-07), (1, 1.1126820027129725e-05), (2, 1.4789339729759376e-05), (3, 7.336950147873722e-06), (4, 2.238158458567341e-06), (5, 4.075960077898344e-06), (6, 3.136280611215625e-06), (7, 4.718333002529107e-06), (8, 3.5005418794753496e-06), (9, 3.104240704487893e-06), (10, 3.696381327245035e-06), (11, 6.5650715441734064e-06), (12, 2.1613050193991512e-05), (13, 0.0008897699299268425), (14, 0.001983985072001815), (15, 0.0009686201810836792), (16, 0.06293657422065735), (17, 0.00023045607667881995), (18, 0.0001903150259749964), (19, 0.005388872232288122), (20, 0.004031803924590349), (21, 0.0005166656919755042), (22, 0.0005998153355903924), (23, 0.00027935398975387216), (24, 0.0005631096428260207), (25, 0.0013850992545485497), (26, 0.000629079935606569), (27, 0.02631761133670807)], Temporal MSE: [(0, 1.0284211384714581e-06), (1, 6.10812378454284e-07), (2, 7.542234925494995e-07), (3, 9.92466425486782e-07), (4, 1.6062092527135974e-06), (5, 1.7112944306063582e-06), (6, 2.2770163923269138e-06), (7, 2.261891950183781e-06), (8, 1.7685015336610377e-06), (9, 1.7606099618205917e-06), (10, 2.736287342486321e-06), (11, 9.365361620439216e-06), (12, 1.8700076907407492e-05), (13, 2.882215085264761e-05), (14, 6.533244595630094e-05), (15, 8.0740952398628e-05), (16, 5.257279553916305e-05), (17, 0.00011234801058890298), (18, 9.513449185760692e-05), (19, 8.90207156771794e-05), (20, 8.159728167811409e-05), (21, 0.00025173870380967855), (22, 0.00028774046222679317), (23, 0.0003913430846296251), (24, 0.00028050632681697607), (25, 0.0006841148715466261), (26, 0.00016198394587263465), (27, 0.00017248489893972874)]
+
+ 80%|████████  | 120/150 [17:50<03:56,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 174, Spatial MSE: [(0, 9.416880857315846e-07), (1, 1.0666800335457083e-05), (2, 1.6172185496543534e-05), (3, 7.400309186778031e-06), (4, 4.8503306970815174e-06), (5, 1.1529362382134423e-05), (6, 1.4526125596603379e-05), (7, 1.4025883501744829e-05), (8, 1.876992973848246e-05), (9, 3.05104476865381e-05), (10, 4.754690598929301e-05), (11, 6.485977792181075e-05), (12, 7.33344058971852e-05), (13, 0.0010406684596091509), (14, 0.0025067671667784452), (15, 0.001649453304708004), (16, 0.07384002953767776), (17, 0.00031422858592122793), (18, 0.00039413769263774157), (19, 0.006145081482827663), (20, 0.004936144687235355), (21, 0.0006570428377017379), (22, 0.0007864762446843088), (23, 0.0006173255387693644), (24, 0.0014035049825906754), (25, 0.003954355604946613), (26, 0.00551957730203867), (27, 0.025661908090114594)], Temporal MSE: [(0, 1.0763563977889135e-06), (1, 6.301073653958156e-07), (2, 9.315618854088825e-07), (3, 1.654606649026391e-06), (4, 3.7879328829149017e-06), (5, 3.8050004604883725e-06), (6, 6.392916020558914e-06), (7, 9.530742318020202e-06), (8, 7.6142764555697795e-06), (9, 8.883307600626722e-06), (10, 1.3549903087550774e-05), (11, 2.5724395527504385e-05), (12, 3.6114026443101466e-05), (13, 4.9705904530128464e-05), (14, 9.571071132086217e-05), (15, 0.00011953632201766595), (16, 7.966445991769433e-05), (17, 0.00013301314902491868), (18, 0.00011758712935261428), (19, 0.00012014457024633884), (20, 0.0001379658788209781), (21, 0.000294671073788777), (22, 0.0003474193508736789), (23, 0.00045424967538565397), (24, 0.0005247367080301046), (25, 0.0010917517356574535), (26, 0.0004647672176361084), (27, 0.0002167161146644503)]
+
+ 81%|████████  | 121/150 [17:57<03:41,  7.64s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 168, Spatial MSE: [(0, 7.695591079937003e-07), (1, 1.2616537787835114e-05), (2, 1.7394375390722416e-05), (3, 7.3263058766315226e-06), (4, 2.541125923016807e-06), (5, 4.488641025091056e-06), (6, 3.5508553537511034e-06), (7, 5.163080913916929e-06), (8, 3.961831680499017e-06), (9, 3.690460744110169e-06), (10, 8.116703611449338e-06), (11, 2.8185664632474072e-05), (12, 8.469188469462097e-05), (13, 0.0023704662453383207), (14, 0.0034618338104337454), (15, 0.0011800715001299977), (16, 0.06619764864444733), (17, 0.0002406254643574357), (18, 0.0007060733041726053), (19, 0.005531232338398695), (20, 0.00406653992831707), (21, 0.00031426086206920445), (22, 0.0006637756014242768), (23, 0.00031230179592967033), (24, 0.00061478913994506), (25, 0.0013956902548670769), (26, 0.0006761325057595968), (27, 0.0332980677485466)], Temporal MSE: [(0, 1.185632527267444e-06), (1, 6.691500402666861e-07), (2, 8.638448889541905e-07), (3, 1.050435798788385e-06), (4, 1.6809406133688753e-06), (5, 1.8145543663194985e-06), (6, 2.5279382498410996e-06), (7, 2.392233000136912e-06), (8, 1.964873263204936e-06), (9, 2.1699033823097125e-06), (10, 6.252084403968183e-06), (11, 5.794769458589144e-05), (12, 6.264290277613327e-05), (13, 5.495249206433073e-05), (14, 6.779720570193604e-05), (15, 0.00014540126721840352), (16, 7.249040208989754e-05), (17, 0.00010749824286904186), (18, 0.00012189171684440225), (19, 0.00010295257379766554), (20, 0.00011253920820308849), (21, 0.00030705775134265423), (22, 0.00032033486058935523), (23, 0.00042179840966127813), (24, 0.0003129894321318716), (25, 0.0008504875004291534), (26, 0.00020370447600726038), (27, 0.00019916039309464395)]
+
+ 81%|████████▏ | 122/150 [18:06<03:44,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 162, Spatial MSE: [(0, 8.91041281647631e-07), (1, 1.27819184854161e-05), (2, 2.4379365640925243e-05), (3, 1.2051466001139488e-05), (4, 1.6270592823275365e-05), (5, 4.207055826555006e-05), (6, 5.37017040187493e-05), (7, 0.00012791396875400096), (8, 0.00017513574857730418), (9, 0.00018199991609435529), (10, 0.0001894793676910922), (11, 0.0002631161187309772), (12, 0.00034413018147461116), (13, 0.0030131542589515448), (14, 0.00520825432613492), (15, 0.002901247702538967), (16, 0.07974214851856232), (17, 0.0013881373452022672), (18, 0.0019029135582968593), (19, 0.007204707246273756), (20, 0.005731203127652407), (21, 0.0021208978723734617), (22, 0.002188753103837371), (23, 0.003465131623670459), (24, 0.005451049190014601), (25, 0.013816144317388535), (26, 0.01737084425985813), (27, 0.03384494036436081)], Temporal MSE: [(0, 1.1356053164490731e-06), (1, 7.184697210504964e-07), (2, 2.6971645183948567e-06), (3, 5.507233709067805e-06), (4, 1.3962074262963142e-05), (5, 1.8415485101286322e-05), (6, 5.746858732891269e-05), (7, 6.028057759976946e-05), (8, 4.2854029743466526e-05), (9, 6.026896517141722e-05), (10, 0.00012744493142236024), (11, 0.0001989651209441945), (12, 0.00022713861835654825), (13, 0.0003541914338711649), (14, 0.0008169353241100907), (15, 0.0007920217467471957), (16, 0.0008752241847105324), (17, 0.0009137731976807117), (18, 0.0006372308707796037), (19, 0.0008112068753689528), (20, 0.0011841882951557636), (21, 0.0011681144824251533), (22, 0.0009660106152296066), (23, 0.001373973092995584), (24, 0.002217780100181699), (25, 0.004317793995141983), (26, 0.0016149686416611075), (27, 0.0003210820141248405)]
+
+ 82%|████████▏ | 123/150 [18:14<03:37,  8.06s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 156, Spatial MSE: [(0, 7.750224426672503e-07), (1, 1.4054846360522788e-05), (2, 1.8008218830800615e-05), (3, 7.293100225069793e-06), (4, 2.5149586235784227e-06), (5, 4.676882326748455e-06), (6, 3.5462328469293425e-06), (7, 5.566073923546355e-06), (8, 4.312455530453008e-06), (9, 4.219331913191127e-06), (10, 7.143410584831145e-06), (11, 2.90507850877475e-05), (12, 6.774458597647026e-05), (13, 0.0018515364499762654), (14, 0.0036868636962026358), (15, 0.0011877418728545308), (16, 0.08698098361492157), (17, 0.00030983443139120936), (18, 0.00018181008636020124), (19, 0.007499832659959793), (20, 0.005594859831035137), (21, 0.0005140597932040691), (22, 0.000626367807853967), (23, 0.00034093696740455925), (24, 0.0006890408112667501), (25, 0.0016793680842965841), (26, 0.0007397158187814057), (27, 0.039413969963788986)], Temporal MSE: [(0, 1.1704602229656302e-06), (1, 6.716721827615402e-07), (2, 8.701278488842945e-07), (3, 1.0505536920391023e-06), (4, 1.722050456010038e-06), (5, 1.8976511455548462e-06), (6, 2.7857854547619354e-06), (7, 2.5293459202657687e-06), (8, 2.0926054276060313e-06), (9, 2.1609998839267064e-06), (10, 5.155172857485013e-06), (11, 6.115635187597945e-05), (12, 4.4828939280705526e-05), (13, 5.435353159555234e-05), (14, 7.745811308268458e-05), (15, 0.00014906880096532404), (16, 9.30113164940849e-05), (17, 0.00016329670324921608), (18, 0.00013298779958859086), (19, 0.00011435700434958562), (20, 8.732337300898507e-05), (21, 0.0003322108823340386), (22, 0.0003567576641216874), (23, 0.00046538605238310993), (24, 0.00033260657801292837), (25, 0.0010341444285586476), (26, 0.00023226863413583487), (27, 0.00022600218653678894)]
+
+ 83%|████████▎ | 124/150 [18:22<03:23,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 150, Spatial MSE: [(0, 1.0462287036716589e-06), (1, 1.3105690413794946e-05), (2, 2.20909460040275e-05), (3, 8.057180821197107e-06), (4, 5.683194558514515e-06), (5, 1.3443513125821482e-05), (6, 1.709784919512458e-05), (7, 1.564631747896783e-05), (8, 2.116234099958092e-05), (9, 3.1964576919563115e-05), (10, 5.2359264373080805e-05), (11, 8.967941539594904e-05), (12, 0.00012271823652554303), (13, 0.00188248825725168), (14, 0.003850421169772744), (15, 0.0014020432718098164), (16, 0.08339496701955795), (17, 0.0003680068184621632), (18, 0.0002464482677169144), (19, 0.007278147153556347), (20, 0.0054456996731460094), (21, 0.0006031897501088679), (22, 0.0008033652557060122), (23, 0.0006776948575861752), (24, 0.0016320969443768263), (25, 0.0044031208381056786), (26, 0.006299296859651804), (27, 0.03944459185004234)], Temporal MSE: [(0, 1.30218700178375e-06), (1, 8.84782423327124e-07), (2, 1.0921611419689725e-06), (3, 1.8957638303618296e-06), (4, 4.424434337124694e-06), (5, 4.349073606135789e-06), (6, 7.741489753243513e-06), (7, 1.0782291610667016e-05), (8, 9.293205039284658e-06), (9, 9.832586329139303e-06), (10, 1.7712409317027777e-05), (11, 7.667465251870453e-05), (12, 6.0296770243439823e-05), (13, 7.501873915316537e-05), (14, 0.00011674410779960454), (15, 0.00017056602519005537), (16, 0.00011425230331951752), (17, 0.0001761565508786589), (18, 0.00015334594354499131), (19, 0.0001439804764231667), (20, 0.0001309574581682682), (21, 0.0003718792286235839), (22, 0.00040672742761671543), (23, 0.0005440550157800317), (24, 0.0006469744257628918), (25, 0.0015591103583574295), (26, 0.0006220453651621938), (27, 0.0002855791535694152)]
+
+ 83%|████████▎ | 125/150 [18:29<03:13,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 144, Spatial MSE: [(0, 7.697911428294901e-07), (1, 1.5005814930191264e-05), (2, 1.7634172763791867e-05), (3, 6.947518613742432e-06), (4, 2.59568128058163e-06), (5, 4.8474144023202825e-06), (6, 3.826640750048682e-06), (7, 6.515984750876669e-06), (8, 4.528152203420177e-06), (9, 3.711428234964842e-06), (10, 6.057696282368852e-06), (11, 1.4364926755661145e-05), (12, 4.146946957916953e-05), (13, 0.0013048008549958467), (14, 0.0024443797301501036), (15, 0.0009622614597901702), (16, 0.05274514481425285), (17, 0.00017842343368101865), (18, 0.00014998619735706598), (19, 0.004408473148941994), (20, 0.003286664839833975), (21, 0.0004453787114471197), (22, 0.0005339059280231595), (23, 0.0002842872345354408), (24, 0.0005409500445239246), (25, 0.0011495016515254974), (26, 0.0006318906089290977), (27, 0.02660905010998249)], Temporal MSE: [(0, 1.2371453976811608e-06), (1, 7.2738731660138e-07), (2, 9.091219794754579e-07), (3, 1.0708057516239933e-06), (4, 1.8209262862001196e-06), (5, 1.909722186610452e-06), (6, 2.9673353765247157e-06), (7, 2.7176615731150378e-06), (8, 2.192435658798786e-06), (9, 2.1964576717437012e-06), (10, 4.409995653986698e-06), (11, 1.9879427782143466e-05), (12, 3.266879866714589e-05), (13, 3.70879570255056e-05), (14, 5.200352461542934e-05), (15, 9.82633137027733e-05), (16, 5.76893326069694e-05), (17, 0.00010205350554315373), (18, 9.472295641899109e-05), (19, 7.99639310571365e-05), (20, 6.331336044240743e-05), (21, 0.0002298198960488662), (22, 0.00025767451734282076), (23, 0.00032997384550981224), (24, 0.00024797130026854575), (25, 0.0007244147709570825), (26, 0.00019189342856407166), (27, 0.00016295842942781746)]
+
+ 84%|████████▍ | 126/150 [18:36<03:01,  7.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 138, Spatial MSE: [(0, 1.0618441592669114e-06), (1, 1.444586359866662e-05), (2, 2.8598880817298777e-05), (3, 1.2383919965941459e-05), (4, 1.609733772056643e-05), (5, 4.3822892621392384e-05), (6, 5.574799433816224e-05), (7, 0.00012472402886487544), (8, 0.00016441576008219272), (9, 0.00018685351824387908), (10, 0.00020957372908014804), (11, 0.00029174843803048134), (12, 0.00037600460927933455), (13, 0.004565815441310406), (14, 0.013960535638034344), (15, 0.008280600421130657), (16, 0.2264607697725296), (17, 0.0014669379452243447), (18, 0.0020269949454814196), (19, 0.02120651863515377), (20, 0.0196567140519619), (21, 0.0025183053221553564), (22, 0.0025240422692149878), (23, 0.003674059174954891), (24, 0.005557042546570301), (25, 0.014606730081140995), (26, 0.017691131681203842), (27, 0.026616835966706276)], Temporal MSE: [(0, 1.6033541214710567e-06), (1, 9.020126299219555e-07), (2, 2.862626615751651e-06), (3, 6.56479642202612e-06), (4, 1.45905678436975e-05), (5, 1.8236252799397334e-05), (6, 6.309882155619562e-05), (7, 6.383469008142129e-05), (8, 4.371015529613942e-05), (9, 6.243658572202548e-05), (10, 0.00014289487444330007), (11, 0.0001986714341910556), (12, 0.00029936706414446235), (13, 0.000441962358308956), (14, 0.0009216623147949576), (15, 0.0008864395204000175), (16, 0.000911061477381736), (17, 0.0009432099759578705), (18, 0.0006920263986103237), (19, 0.0008772655273787677), (20, 0.0012295384658500552), (21, 0.0013002104824408889), (22, 0.000986107625067234), (23, 0.0013430127874016762), (24, 0.002297213301062584), (25, 0.004450137726962566), (26, 0.0018393282080069184), (27, 0.00031537903123535216)]
+
+ 85%|████████▍ | 127/150 [18:44<02:57,  7.74s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 132, Spatial MSE: [(0, 8.986357329376915e-07), (1, 1.685393726802431e-05), (2, 2.073583527817391e-05), (3, 7.921121323306579e-06), (4, 3.022656983375782e-06), (5, 5.921336196479388e-06), (6, 4.7463886403420474e-06), (7, 7.434788585669594e-06), (8, 5.275941475701984e-06), (9, 5.113702627568273e-06), (10, 5.861837053089403e-06), (11, 1.0500448297534604e-05), (12, 3.0910119676264e-05), (13, 0.0015364912105724216), (14, 0.0043889982625842094), (15, 0.0014951551565900445), (16, 0.13158053159713745), (17, 0.0004744812613353133), (18, 0.0002560877474024892), (19, 0.011352629400789738), (20, 0.008486578240990639), (21, 0.00041588780004531145), (22, 0.0008919732645153999), (23, 0.000448074220912531), (24, 0.0009037969284690917), (25, 0.002217088360339403), (26, 0.001007441314868629), (27, 0.05325419455766678)], Temporal MSE: [(0, 1.4635078287028591e-06), (1, 8.998679845717561e-07), (2, 1.0801800272020046e-06), (3, 1.226287281497207e-06), (4, 2.2498243197333068e-06), (5, 2.4007813408388756e-06), (6, 4.305801667214837e-06), (7, 3.4570382467791205e-06), (8, 2.8082381504646037e-06), (9, 2.750127805484226e-06), (10, 4.454610007087467e-06), (11, 1.2025897376588546e-05), (12, 2.9525936042773537e-05), (13, 6.844518793513998e-05), (14, 0.00010277470573782921), (15, 0.00021435361122712493), (16, 0.000138847652124241), (17, 0.0002693907299544662), (18, 0.00019630067981779575), (19, 0.00017458591901231557), (20, 0.00012634170707315207), (21, 0.00045433815103024244), (22, 0.0004769425722770393), (23, 0.0006393745425157249), (24, 0.00044970374437980354), (25, 0.0012710078153759241), (26, 0.0003116614534519613), (27, 0.00033159300801344216)]
+
+ 85%|████████▌ | 128/150 [18:52<02:47,  7.59s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 126, Spatial MSE: [(0, 1.1403659527786658e-06), (1, 1.5138042726903223e-05), (2, 2.959752600872889e-05), (3, 9.018569471663795e-06), (4, 6.465225396823371e-06), (5, 1.502605573477922e-05), (6, 1.9010069081559777e-05), (7, 1.8035350876743905e-05), (8, 2.305442831129767e-05), (9, 2.9990445909788832e-05), (10, 5.176727790967561e-05), (11, 8.152225200319663e-05), (12, 9.25096683204174e-05), (13, 0.0015405964804813266), (14, 0.0047273291274905205), (15, 0.002248993143439293), (16, 0.13390624523162842), (17, 0.0005589565262198448), (18, 0.0003582062781788409), (19, 0.01119389757514), (20, 0.008393917232751846), (21, 0.0006108286324888468), (22, 0.001068935147486627), (23, 0.0008759540505707264), (24, 0.0021026853937655687), (25, 0.005311430897563696), (26, 0.00637341383844614), (27, 0.052313998341560364)], Temporal MSE: [(0, 1.6414267065556487e-06), (1, 1.055046823239536e-06), (2, 1.2368747093205457e-06), (3, 2.04600723918702e-06), (4, 4.770713985635666e-06), (5, 4.968271696270676e-06), (6, 9.060900993063115e-06), (7, 1.1970625564572401e-05), (8, 1.038480786519358e-05), (9, 1.0458035831106827e-05), (10, 1.753515789459925e-05), (11, 2.8915039365529083e-05), (12, 4.793987318407744e-05), (13, 9.394741937285289e-05), (14, 0.00015140908362809569), (15, 0.00022896705195307732), (16, 0.00016169992159120739), (17, 0.0002763677039183676), (18, 0.0002262301422888413), (19, 0.00021988677326589823), (20, 0.00018980579625349492), (21, 0.0005045335274189711), (22, 0.0005297992029227316), (23, 0.0007213227218016982), (24, 0.0008492936030961573), (25, 0.001905890996567905), (26, 0.0008090161136351526), (27, 0.00041895548929460347)]
+
+ 86%|████████▌ | 129/150 [18:59<02:38,  7.53s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 120, Spatial MSE: [(0, 1.0142175597138703e-06), (1, 1.740482730383519e-05), (2, 2.161456723115407e-05), (3, 8.058568710112013e-06), (4, 3.1744718853587983e-06), (5, 6.6699785747914575e-06), (6, 5.365111974242609e-06), (7, 9.02129613677971e-06), (8, 5.996529125695815e-06), (9, 5.7692154769029e-06), (10, 6.033608315192396e-06), (11, 1.0396274774393532e-05), (12, 2.897377999033779e-05), (13, 0.0008993721567094326), (14, 0.001888759434223175), (15, 0.0006811983766965568), (16, 0.05377684161067009), (17, 0.0002062431740341708), (18, 0.0003777123347390443), (19, 0.0044625247828662395), (20, 0.003249740693718195), (21, 0.00017304920766036958), (22, 0.0005123158334754407), (23, 0.00028042306075803936), (24, 0.0005348810227587819), (25, 0.001252451678737998), (26, 0.0006737438379786909), (27, 0.020649094134569168)], Temporal MSE: [(0, 1.5921192471068935e-06), (1, 9.726735470394487e-07), (2, 1.2115532399548101e-06), (3, 1.304299530602293e-06), (4, 2.3518539364886237e-06), (5, 2.5019271561177447e-06), (6, 4.716523108072579e-06), (7, 3.831359663308831e-06), (8, 3.0820779102214146e-06), (9, 3.105241603407194e-06), (10, 4.653111318475567e-06), (11, 1.3635349205287639e-05), (12, 2.6622559744282626e-05), (13, 3.354774889885448e-05), (14, 5.138968481332995e-05), (15, 9.377812966704369e-05), (16, 6.205089448485523e-05), (17, 0.00010284014570061117), (18, 8.376475307159126e-05), (19, 7.440350600518286e-05), (20, 5.675429929397069e-05), (21, 0.00020859361393377185), (22, 0.0002296863531228155), (23, 0.00032099217060022056), (24, 0.00025049096439033747), (25, 0.000613523880019784), (26, 0.00018295667541678995), (27, 0.00015283374523278326)]
+
+ 87%|████████▋ | 130/150 [19:06<02:26,  7.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 114, Spatial MSE: [(0, 1.1891991107404465e-06), (1, 1.7658405340625905e-05), (2, 3.8727710489183664e-05), (3, 1.3785663213639054e-05), (4, 1.704585156403482e-05), (5, 4.7444485971936956e-05), (6, 6.577937892870978e-05), (7, 0.0001402945490553975), (8, 0.00018127723888028413), (9, 0.00018873029330279678), (10, 0.00017800532805267721), (11, 0.0002379744837526232), (12, 0.0002921548730228096), (13, 0.002687811851501465), (14, 0.008189970627427101), (15, 0.005232945084571838), (16, 0.12343736737966537), (17, 0.0012666371185332537), (18, 0.0016062838258221745), (19, 0.008879403583705425), (20, 0.008161285892128944), (21, 0.0020147908944636583), (22, 0.00215234630741179), (23, 0.003624978009611368), (24, 0.005915995687246323), (25, 0.015794208273291588), (26, 0.021117405965924263), (27, 0.022036191076040268)], Temporal MSE: [(0, 1.8954963252326706e-06), (1, 1.1255270919718896e-06), (2, 3.4023348689515842e-06), (3, 8.656618774693925e-06), (4, 1.5412591892527416e-05), (5, 2.0298464733059518e-05), (6, 8.575260289944708e-05), (7, 7.106851990101859e-05), (8, 4.88458426843863e-05), (9, 6.247327837627381e-05), (10, 0.00012239423813298345), (11, 0.00015799984976183623), (12, 0.00021871732315048575), (13, 0.00034549509291537106), (14, 0.0007819388993084431), (15, 0.0007119398796930909), (16, 0.0007353530381806195), (17, 0.0008128246408887208), (18, 0.0005426847492344677), (19, 0.0006974599091336131), (20, 0.0010363172041252255), (21, 0.001045353477820754), (22, 0.0009318475495092571), (23, 0.0013489306438714266), (24, 0.0023093693889677525), (25, 0.004406362771987915), (26, 0.0019583364482969046), (27, 0.0003614283341448754)]
+
+ 87%|████████▋ | 131/150 [19:16<02:34,  8.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 108, Spatial MSE: [(0, 1.071505039362819e-06), (1, 2.267261515953578e-05), (2, 3.081259274040349e-05), (3, 1.0149989066121634e-05), (4, 4.1410244193684775e-06), (5, 9.857521945377812e-06), (6, 7.040528544166591e-06), (7, 1.301663269259734e-05), (8, 9.130252692557406e-06), (9, 6.143665814306587e-06), (10, 1.0820075658557471e-05), (11, 1.779889134922996e-05), (12, 4.048680784762837e-05), (13, 0.0010086498223245144), (14, 0.00308910571038723), (15, 0.0011333866277709603), (16, 0.08743292838335037), (17, 0.00032536222715862095), (18, 0.00018506102787796408), (19, 0.007310299202799797), (20, 0.005427285563200712), (21, 0.00029214195092208683), (22, 0.0006870016804896295), (23, 0.00040042196633294225), (24, 0.0007671184721402824), (25, 0.001961057772859931), (26, 0.000950017012655735), (27, 0.03335520625114441)], Temporal MSE: [(0, 1.838879825299955e-06), (1, 1.1640127013379242e-06), (2, 1.4696289554194664e-06), (3, 1.5838924127820064e-06), (4, 3.0538894861820154e-06), (5, 3.3119831641670316e-06), (6, 5.492475793289486e-06), (7, 5.007196250517154e-06), (8, 4.238263954903232e-06), (9, 4.219484253553674e-06), (10, 7.932911103125662e-06), (11, 1.7175047105411068e-05), (12, 2.751573447312694e-05), (13, 4.2865452996920794e-05), (14, 6.823522562626749e-05), (15, 0.00017776536697056144), (16, 0.00010860290058190003), (17, 0.00020614636014215648), (18, 0.00013849248352926224), (19, 0.0001219026162289083), (20, 8.765343955019489e-05), (21, 0.0003311543841846287), (22, 0.0003528700617607683), (23, 0.0004983191029168665), (24, 0.00035598702379502356), (25, 0.000979852513410151), (26, 0.00026568264001980424), (27, 0.00022738480765838176)]
+
+ 88%|████████▊ | 132/150 [19:26<02:37,  8.77s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 102, Spatial MSE: [(0, 1.2757994909406989e-06), (1, 2.0015635527670383e-05), (2, 4.122555037611164e-05), (3, 9.858763405645732e-06), (4, 8.095546945696697e-06), (5, 1.9047711248276755e-05), (6, 2.526549724279903e-05), (7, 2.4274275347124785e-05), (8, 3.1767485779710114e-05), (9, 5.0282207666896284e-05), (10, 7.014197035459802e-05), (11, 0.00010612670303089544), (12, 0.00010848282545339316), (13, 0.0010611132020130754), (14, 0.0035622003488242626), (15, 0.0015974750276654959), (16, 0.08888451009988785), (17, 0.0004265494935680181), (18, 0.0008294092258438468), (19, 0.009381688199937344), (20, 0.007270073518157005), (21, 0.000658783596009016), (22, 0.0009987677913159132), (23, 0.0009650769061408937), (24, 0.0023512786719948053), (25, 0.005655569490045309), (26, 0.007654031738638878), (27, 0.034403640776872635)], Temporal MSE: [(0, 1.889649752229161e-06), (1, 1.36307255615975e-06), (2, 1.5139340803216328e-06), (3, 2.1913185719313333e-06), (4, 5.537741799344076e-06), (5, 5.945947577856714e-06), (6, 1.219697151100263e-05), (7, 1.6522542864549905e-05), (8, 1.4112333701632451e-05), (9, 1.4418586943065748e-05), (10, 2.5108129193540663e-05), (11, 3.788927642744966e-05), (12, 4.976746640750207e-05), (13, 7.617688970640302e-05), (14, 0.00013861308980267495), (15, 0.00020878341456409544), (16, 0.00013936914911027998), (17, 0.0002233834529761225), (18, 0.00017547127208672464), (19, 0.00020133623911533505), (20, 0.0002185646299039945), (21, 0.00043396660475991666), (22, 0.00045904936268925667), (23, 0.0006173887522891164), (24, 0.0008711764239706099), (25, 0.0018783429404720664), (26, 0.0009655021131038666), (27, 0.00047443812945857644)]
+
+ 89%|████████▊ | 133/150 [19:35<02:31,  8.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 96, Spatial MSE: [(0, 1.0992376928697922e-06), (1, 2.3744678401271813e-05), (2, 2.982278965646401e-05), (3, 7.880197699705604e-06), (4, 1.145285841630539e-05), (5, 3.4821398003259674e-05), (6, 4.732954403152689e-05), (7, 0.00010846611257875338), (8, 0.00012167826935183257), (9, 0.00011829878349089995), (10, 0.00010983342508552596), (11, 0.0001456499594496563), (12, 0.00017469831800553948), (13, 0.0008933224598877132), (14, 0.00379172177053988), (15, 0.002936799079179764), (16, 0.060816023498773575), (17, 0.0007227353635244071), (18, 0.0008102392894215882), (19, 0.006004365161061287), (20, 0.005735656246542931), (21, 0.0012031749356538057), (22, 0.0011539589613676071), (23, 0.0023181450087577105), (24, 0.003591262036934495), (25, 0.008341866545379162), (26, 0.010099510662257671), (27, 0.0028714651707559824)], Temporal MSE: [(0, 2.0746558675455162e-06), (1, 1.3924479844718007e-06), (2, 3.0899152534402674e-06), (3, 7.772567187203094e-06), (4, 1.0204448699369095e-05), (5, 1.3360528100747615e-05), (6, 6.870841752970591e-05), (7, 4.882532812189311e-05), (8, 3.240998921683058e-05), (9, 3.993876453023404e-05), (10, 7.914481830084696e-05), (11, 9.738535300130025e-05), (12, 0.00012693769531324506), (13, 0.0002255288854939863), (14, 0.000541938585229218), (15, 0.00043053695117123425), (16, 0.00048153617535717785), (17, 0.00048051102203316987), (18, 0.00033592464751563966), (19, 0.0004670853086281568), (20, 0.0007238696562126279), (21, 0.0005920723779127002), (22, 0.0005301545606926084), (23, 0.000730138854123652), (24, 0.0015756691573187709), (25, 0.0027584468480199575), (26, 0.0014600707218050957), (27, 0.00028791240765713155)]
+
+ 89%|████████▉ | 134/150 [19:44<02:19,  8.74s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 90, Spatial MSE: [(0, 1.1845804692711681e-06), (1, 2.3184940801002085e-05), (2, 3.629318962339312e-05), (3, 6.3970160226745065e-06), (4, 5.115660314913839e-06), (5, 1.2703088032139931e-05), (6, 1.5830606571398675e-05), (7, 2.3990587578737177e-05), (8, 2.7323007088853046e-05), (9, 3.090697282459587e-05), (10, 2.7642177883535624e-05), (11, 3.964695133618079e-05), (12, 3.415373794268817e-05), (13, 5.5068201618269086e-05), (14, 9.56199801294133e-05), (15, 0.00012437884288374335), (16, 0.0001387959928251803), (17, 0.0001227937318617478), (18, 0.00011979727423749864), (19, 0.00014106358867138624), (20, 0.00013658417447004467), (21, 0.00018602552881930023), (22, 0.00022782334417570382), (23, 0.0005014152848161757), (24, 0.0010159221710637212), (25, 0.002585826674476266), (26, 0.003963230177760124), (27, 0.003150910371914506)], Temporal MSE: [(0, 2.2317090042633936e-06), (1, 1.4012304063726333e-06), (2, 1.8554599137132755e-06), (3, 2.287262532263412e-06), (4, 3.082813691435149e-06), (5, 3.0118617360130884e-06), (6, 2.0318324459367432e-05), (7, 1.2649400559894275e-05), (8, 9.075383786694147e-06), (9, 9.836925528361462e-06), (10, 1.5094598893483635e-05), (11, 1.6310801584040746e-05), (12, 1.8646926037035882e-05), (13, 3.224590182071552e-05), (14, 7.985206320881844e-05), (15, 6.615417805733159e-05), (16, 7.518468919442967e-05), (17, 6.665581167908385e-05), (18, 5.117922046338208e-05), (19, 6.731404573656619e-05), (20, 0.00010167489381274208), (21, 8.678147423779592e-05), (22, 8.312007412314415e-05), (23, 0.00014057867520023137), (24, 0.00028922714409418404), (25, 0.0005661636241711676), (26, 0.0002402417012490332), (27, 0.00016238910029642284)]
+
+ 90%|█████████ | 135/150 [19:54<02:19,  9.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 84, Spatial MSE: [(0, 1.2610996691364562e-06), (1, 2.5152488888124935e-05), (2, 3.4504668292356655e-05), (3, 7.102906693035038e-06), (4, 5.765512469224632e-06), (5, 1.4066346011532005e-05), (6, 1.9182167307008058e-05), (7, 2.6039280783152208e-05), (8, 3.0220584449125454e-05), (9, 3.569583350326866e-05), (10, 2.897976628446486e-05), (11, 3.908407597918995e-05), (12, 3.613014268921688e-05), (13, 5.60443768335972e-05), (14, 9.136235166806728e-05), (15, 0.00013771085650660098), (16, 0.00014194156392477453), (17, 0.00013042998034507036), (18, 0.00012444081949070096), (19, 0.00014680642925668508), (20, 0.0001487281551817432), (21, 0.00019313881057314575), (22, 0.00023876357590779662), (23, 0.000531892292201519), (24, 0.0010459801414981484), (25, 0.002413672860711813), (26, 0.003848432330414653), (27, 0.00317461509257555)], Temporal MSE: [(0, 2.3720908757240977e-06), (1, 1.5413156688737217e-06), (2, 2.0801414848392596e-06), (3, 2.8519432362372754e-06), (4, 3.486048171907896e-06), (5, 3.4767956549330847e-06), (6, 2.287884126417339e-05), (7, 1.4117367754806764e-05), (8, 1.0484725862625055e-05), (9, 1.0802790711750276e-05), (10, 1.6001862604753114e-05), (11, 1.8143146007787436e-05), (12, 1.8828664906322956e-05), (13, 3.3492051443317905e-05), (14, 8.431263267993927e-05), (15, 7.188029121607542e-05), (16, 7.792982796672732e-05), (17, 7.03069381415844e-05), (18, 5.216867430135608e-05), (19, 6.8371431552805e-05), (20, 0.00010316403495380655), (21, 8.808900747681037e-05), (22, 8.464351412840188e-05), (23, 0.00014678649313282222), (24, 0.00029826388345099986), (25, 0.0005357348709367216), (26, 0.00025282305432483554), (27, 0.00021604541689157486)]
+
+ 91%|█████████ | 136/150 [20:03<02:07,  9.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 78, Spatial MSE: [(0, 1.2659464800890419e-06), (1, 2.678727287275251e-05), (2, 4.004261427326128e-05), (3, 7.09058031134191e-06), (4, 5.950997092440957e-06), (5, 1.4089572687225882e-05), (6, 2.0295337890274823e-05), (7, 3.0945655453251675e-05), (8, 3.3274471206823364e-05), (9, 3.3006028388626873e-05), (10, 3.0888524634065107e-05), (11, 4.078297934029251e-05), (12, 3.575057053240016e-05), (13, 5.36407642357517e-05), (14, 0.00010015279985964298), (15, 0.00013530996511690319), (16, 0.00013859289174433798), (17, 0.00012334967323113233), (18, 0.00012407064787112176), (19, 0.0001426235685357824), (20, 0.0001470271236030385), (21, 0.00019762120791710913), (22, 0.00024314406618941575), (23, 0.0005424318951554596), (24, 0.001062279217876494), (25, 0.0024571481626480818), (26, 0.004053487442433834), (27, 0.0033100158907473087)], Temporal MSE: [(0, 2.4164055503206328e-06), (1, 1.583034190844046e-06), (2, 2.1660991933458718e-06), (3, 2.801300070132129e-06), (4, 3.5705900245375233e-06), (5, 3.5013893011637265e-06), (6, 2.7909210984944366e-05), (7, 1.570758831803687e-05), (8, 1.0840242794074584e-05), (9, 1.0844662028830498e-05), (10, 1.5483816241612658e-05), (11, 1.7298651073360816e-05), (12, 1.7092244888772257e-05), (13, 3.0266062822192907e-05), (14, 7.258315599756315e-05), (15, 7.117664790712297e-05), (16, 8.980900747701526e-05), (17, 7.898906915215775e-05), (18, 5.410553785623051e-05), (19, 7.250939233927056e-05), (20, 0.00011116168752778322), (21, 9.48410524870269e-05), (22, 8.578286360716447e-05), (23, 0.00014677234867122024), (24, 0.0002988736960105598), (25, 0.0005463679553940892), (26, 0.0002717248280532658), (27, 0.0002906803856603801)]
+
+ 91%|█████████▏| 137/150 [20:12<01:57,  9.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 72, Spatial MSE: [(0, 1.3575531738752034e-06), (1, 2.818197572196368e-05), (2, 3.9700920751784e-05), (3, 7.5795824159285985e-06), (4, 6.314196980383713e-06), (5, 1.5239847925840877e-05), (6, 2.1578067389782518e-05), (7, 3.048718281206675e-05), (8, 3.4985838283319026e-05), (9, 3.7971163692418486e-05), (10, 3.0243085348047316e-05), (11, 4.0090504626277834e-05), (12, 3.547810410964303e-05), (13, 0.0001509358553448692), (14, 0.0005665739881806076), (15, 0.0005949843325652182), (16, 0.012323069386184216), (17, 0.00014519710384774953), (18, 0.0001566240971442312), (19, 0.0012751579051837325), (20, 0.0012098986189812422), (21, 0.0002351783332414925), (22, 0.00027828075690194964), (23, 0.0005904637509956956), (24, 0.0011554525699466467), (25, 0.0026465067639946938), (26, 0.0042789694853127), (27, 0.0036729604471474886)], Temporal MSE: [(0, 2.556364734118688e-06), (1, 1.6984317880996969e-06), (2, 2.2754115889256354e-06), (3, 2.948704150185222e-06), (4, 3.905763605871471e-06), (5, 3.868008661811473e-06), (6, 3.149619806208648e-05), (7, 1.627059100428596e-05), (8, 1.223487106472021e-05), (9, 1.2106698704883456e-05), (10, 1.724350659060292e-05), (11, 1.919335227285046e-05), (12, 1.9868011804646812e-05), (13, 3.673390529002063e-05), (14, 8.425510168308392e-05), (15, 7.611236651428044e-05), (16, 7.773676770739257e-05), (17, 6.878478598082438e-05), (18, 5.844699262524955e-05), (19, 8.397763303946704e-05), (20, 0.0001261531433556229), (21, 0.00011726601223926991), (22, 0.00010512702283449471), (23, 0.00016846091602928936), (24, 0.0003366494202055037), (25, 0.0006019459688104689), (26, 0.00030493794474750757), (27, 0.00041158657404594123)]
+
+ 92%|█████████▏| 138/150 [20:20<01:45,  8.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 66, Spatial MSE: [(0, 1.582597178639844e-06), (1, 3.2824846130097285e-05), (2, 4.528940553427674e-05), (3, 8.398352292715572e-06), (4, 7.738054591754917e-06), (5, 1.7638161807553843e-05), (6, 2.7177244191989303e-05), (7, 3.523874329403043e-05), (8, 4.057677506352775e-05), (9, 4.32541492045857e-05), (10, 3.739075691555627e-05), (11, 4.916279431199655e-05), (12, 4.3458239815663546e-05), (13, 7.63577118050307e-05), (14, 0.00012507192150224), (15, 0.00015674163296353072), (16, 0.00015002515283413231), (17, 0.00014624479808844626), (18, 0.0001322566531598568), (19, 0.00016816389688756317), (20, 0.0001790413516573608), (21, 0.00022344077297020704), (22, 0.00028642540564760566), (23, 0.0006517900619655848), (24, 0.001337406225502491), (25, 0.0029386626556515694), (26, 0.004864086862653494), (27, 0.0039801462553441525)], Temporal MSE: [(0, 3.1971851512935245e-06), (1, 2.1139260297786677e-06), (2, 2.9201812594692456e-06), (3, 3.6491021546680713e-06), (4, 4.549325240077451e-06), (5, 4.715861450677039e-06), (6, 3.551576446625404e-05), (7, 1.8680713765206747e-05), (8, 1.532760143163614e-05), (9, 1.4201445992512163e-05), (10, 1.990961391129531e-05), (11, 2.1725898477598093e-05), (12, 2.2307467588689178e-05), (13, 3.684537296066992e-05), (14, 8.29557902761735e-05), (15, 7.169893069658428e-05), (16, 7.91078491602093e-05), (17, 7.173157791839913e-05), (18, 5.759276245953515e-05), (19, 7.729447679594159e-05), (20, 0.00011737724707927555), (21, 0.000101563120551873), (22, 0.00010243558790534735), (23, 0.00018108916992787272), (24, 0.0003719050728250295), (25, 0.000681333418469876), (26, 0.0003791061171796173), (27, 0.0006296004285104573)]
+
+ 93%|█████████▎| 139/150 [20:29<01:36,  8.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 60, Spatial MSE: [(0, 1.7205156837007962e-06), (1, 3.5615143133327365e-05), (2, 4.8747424443718046e-05), (3, 9.241754014510661e-06), (4, 8.492396773362998e-06), (5, 1.8761897081276402e-05), (6, 3.125833973172121e-05), (7, 4.127646388951689e-05), (8, 4.4961034291191027e-05), (9, 4.395235373522155e-05), (10, 4.147432264289819e-05), (11, 5.389158468460664e-05), (12, 4.7943547542672604e-05), (13, 0.00048146367771551013), (14, 0.002235325053334236), (15, 0.002073985757306218), (16, 0.05042855441570282), (17, 0.00020708389638457447), (18, 0.00026798530598171055), (19, 0.004581187851727009), (20, 0.004438038915395737), (21, 0.00040573879959993064), (22, 0.00041320276795886457), (23, 0.0007967152050696313), (24, 0.001574600231833756), (25, 0.003257985692471266), (26, 0.005517828278243542), (27, 0.00434797303751111)], Temporal MSE: [(0, 3.3623427952989005e-06), (1, 2.3033428533381084e-06), (2, 3.2850896332092816e-06), (3, 3.7235288345982553e-06), (4, 4.91200353280874e-06), (5, 5.280840923660435e-06), (6, 4.176092261332087e-05), (7, 2.1220450435066596e-05), (8, 1.775768760126084e-05), (9, 1.5707240891060792e-05), (10, 2.2753640223527327e-05), (11, 2.623146792757325e-05), (12, 2.846381175913848e-05), (13, 5.6984732509590685e-05), (14, 0.00012175332085462287), (15, 0.0001102295282180421), (16, 0.00010003119678003713), (17, 9.037938434630632e-05), (18, 8.91579911694862e-05), (19, 0.00012542410695459694), (20, 0.00017263839254155755), (21, 0.00019157685164827853), (22, 0.00016641488764435053), (23, 0.00024479080457240343), (24, 0.0004644149448722601), (25, 0.0008611962548457086), (26, 0.0004684925952460617), (27, 0.0009779444662854075)]
+
+ 93%|█████████▎| 140/150 [20:37<01:25,  8.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 54, Spatial MSE: [(0, 1.8618537751535769e-06), (1, 3.916567948181182e-05), (2, 5.451841207104735e-05), (3, 9.633989975554869e-06), (4, 9.143157512880862e-06), (5, 2.0809820853173733e-05), (6, 3.6248235119273886e-05), (7, 4.640115366782993e-05), (8, 5.1647064537974074e-05), (9, 5.380826405598782e-05), (10, 4.4031352445017546e-05), (11, 6.0891296016052365e-05), (12, 5.264380160951987e-05), (13, 0.0006856235559098423), (14, 0.003696296364068985), (15, 0.003005716949701309), (16, 0.07321570068597794), (17, 0.0002545457100495696), (18, 0.00032650743378326297), (19, 0.006634692661464214), (20, 0.006320930551737547), (21, 0.0005163577734492719), (22, 0.00047047363477759063), (23, 0.0008842346142046154), (24, 0.0017474782653152943), (25, 0.0034985425882041454), (26, 0.00557353999465704), (27, 0.004694711416959763)], Temporal MSE: [(0, 3.5986947750643594e-06), (1, 2.5135027499345597e-06), (2, 3.5794253108178964e-06), (3, 3.978239874413703e-06), (4, 5.506550678546773e-06), (5, 5.889812200621236e-06), (6, 5.234478885540739e-05), (7, 2.4471293727401644e-05), (8, 2.0227880668244325e-05), (9, 1.7729884348227642e-05), (10, 2.3908129151095636e-05), (11, 2.7191617846256122e-05), (12, 3.523828127072193e-05), (13, 6.868234777357429e-05), (14, 0.00013984508404973894), (15, 0.0001268804189749062), (16, 0.00011079306568717584), (17, 9.299880912294611e-05), (18, 0.00010548850696068257), (19, 0.00015664842794649303), (20, 0.00020273432892281562), (21, 0.00023146480089053512), (22, 0.00019825746130663902), (23, 0.0002818908542394638), (24, 0.0005257370066829026), (25, 0.0009342036792077124), (26, 0.0005562685546465218), (27, 0.001618395559489727)]
+
+ 94%|█████████▍| 141/150 [20:44<01:13,  8.17s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 48, Spatial MSE: [(0, 2.068931735266233e-06), (1, 4.34463654528372e-05), (2, 6.0269561799941584e-05), (3, 1.0381333595432807e-05), (4, 1.0355036465625744e-05), (5, 2.177512760681566e-05), (6, 4.141280805924907e-05), (7, 5.058460737927817e-05), (8, 5.772451913799159e-05), (9, 5.680250978912227e-05), (10, 5.1619204896269366e-05), (11, 6.326408765744418e-05), (12, 6.251544982660562e-05), (13, 0.000680482538882643), (14, 0.0033534590620547533), (15, 0.002538201166316867), (16, 0.060919344425201416), (17, 0.00026259757578372955), (18, 0.0003092220867983997), (19, 0.005425109062343836), (20, 0.005344973877072334), (21, 0.00041326842620037496), (22, 0.00048495244118385017), (23, 0.0009258636855520308), (24, 0.0018127685179933906), (25, 0.0033506755717098713), (26, 0.005690569523721933), (27, 0.004909411538392305)], Temporal MSE: [(0, 3.963126800954342e-06), (1, 2.7682524432748323e-06), (2, 4.0209411054092925e-06), (3, 4.19316620536847e-06), (4, 5.881645392946666e-06), (5, 6.5879294197657146e-06), (6, 6.075035344110802e-05), (7, 2.6187379262410104e-05), (8, 2.3121381673263386e-05), (9, 1.860900920291897e-05), (10, 2.555664104875177e-05), (11, 3.1299670808948576e-05), (12, 4.1000144847203046e-05), (13, 6.498275615740567e-05), (14, 0.00012685659748967737), (15, 0.0001237689284607768), (16, 0.00010818416194524616), (17, 9.555090946378186e-05), (18, 0.00010000205656979233), (19, 0.00014298252062872052), (20, 0.00020181010768283159), (21, 0.00022136527695693076), (22, 0.00018985186761710793), (23, 0.0002833474427461624), (24, 0.0005484296125359833), (25, 0.0009581954800523818), (26, 0.000672778463922441), (27, 0.002354815835133195)]
+
+ 95%|█████████▍| 142/150 [20:55<01:11,  8.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 42, Spatial MSE: [(0, 2.4560938527429244e-06), (1, 5.300204429659061e-05), (2, 7.202869892353192e-05), (3, 1.2340082321316004e-05), (4, 1.2653160410991404e-05), (5, 2.6072520995512605e-05), (6, 5.4914966312935576e-05), (7, 6.248809222597629e-05), (8, 7.078344060573727e-05), (9, 6.787550955777988e-05), (10, 6.301075336523354e-05), (11, 8.260730101028457e-05), (12, 7.033304427750409e-05), (13, 0.0009625169332139194), (14, 0.0058551752008497715), (15, 0.0044390796683728695), (16, 0.10721160471439362), (17, 0.000321040628477931), (18, 0.0004471709835343063), (19, 0.009321136400103569), (20, 0.009065463207662106), (21, 0.0006141364574432373), (22, 0.0006528635858558118), (23, 0.0011870774906128645), (24, 0.0024535399861633778), (25, 0.004250957164913416), (26, 0.007616946939378977), (27, 0.0064480360597372055)], Temporal MSE: [(0, 4.806992365047336e-06), (1, 3.4027864330710145e-06), (2, 4.972306669515092e-06), (3, 5.161528861208353e-06), (4, 7.1881986514199525e-06), (5, 8.220402378356084e-06), (6, 7.100213406374678e-05), (7, 3.4075910662068054e-05), (8, 3.061424649786204e-05), (9, 2.3325639631366357e-05), (10, 3.2532090699533e-05), (11, 3.7159330531721935e-05), (12, 4.6238055801950395e-05), (13, 9.296743519371375e-05), (14, 0.00017273931007366627), (15, 0.0001749256916809827), (16, 0.0001383983762934804), (17, 0.00011686416110023856), (18, 0.00013814851990900934), (19, 0.00020337778551038355), (20, 0.0002731992572080344), (21, 0.00032442965311929584), (22, 0.00025789072969928384), (23, 0.0003793007927015424), (24, 0.0007122282986529171), (25, 0.0012770871398970485), (26, 0.0009788095485419035), (27, 0.004844993352890015)]
+
+ 95%|█████████▌| 143/150 [21:04<01:02,  8.88s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 36, Spatial MSE: [(0, 2.947525899799075e-06), (1, 6.155580922495574e-05), (2, 8.253300620708615e-05), (3, 1.4151652067084797e-05), (4, 1.5757828805362806e-05), (5, 3.0408902603085153e-05), (6, 6.783680873923004e-05), (7, 7.630285108461976e-05), (8, 9.019147546496242e-05), (9, 8.372360753128305e-05), (10, 7.995755731826648e-05), (11, 9.841719293035567e-05), (12, 8.051557233557105e-05), (13, 0.0004889370175078511), (14, 0.002768576378002763), (15, 0.0029423581436276436), (16, 0.06133463978767395), (17, 0.0003503614279907197), (18, 0.0003685455594677478), (19, 0.00574068445712328), (20, 0.00552780507132411), (21, 0.0006661011138930917), (22, 0.0006432774243876338), (23, 0.0012970194220542908), (24, 0.002753687324002385), (25, 0.004872075282037258), (26, 0.00893514882773161), (27, 0.009026026353240013)], Temporal MSE: [(0, 5.535462605621433e-06), (1, 4.117852768104058e-06), (2, 6.125731033534976e-06), (3, 5.84635381528642e-06), (4, 8.405332664551679e-06), (5, 1.0063154149975162e-05), (6, 9.740906534716487e-05), (7, 3.946842116420157e-05), (8, 3.7805446481797844e-05), (9, 2.868659248633776e-05), (10, 3.855735849356279e-05), (11, 4.172700209892355e-05), (12, 4.5455879444489256e-05), (13, 7.587762229377404e-05), (14, 0.00016661178960930556), (15, 0.00015431460633408278), (16, 0.00013886933447793126), (17, 0.00011739147885236889), (18, 0.00012218016490805894), (19, 0.00017735030269250274), (20, 0.0002545235329307616), (21, 0.00027769076405093074), (22, 0.0002487741585355252), (23, 0.0003976358275394887), (24, 0.0007869566325098276), (25, 0.0014413029421120882), (26, 0.0012903481256216764), (27, 0.008845760487020016)]
+
+ 96%|█████████▌| 144/150 [21:11<00:51,  8.55s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 30, Spatial MSE: [(0, 3.4269071420567343e-06), (1, 7.047753751976416e-05), (2, 8.9212269813288e-05), (3, 1.663361217651982e-05), (4, 1.891756801342126e-05), (5, 3.3151194656966254e-05), (6, 8.029564196476713e-05), (7, 8.735286246519536e-05), (8, 0.00010075301543110982), (9, 9.198969200951979e-05), (10, 9.121684706769884e-05), (11, 0.00010444423242006451), (12, 8.76560079632327e-05), (13, 0.0004955052281729877), (14, 0.0026397714391350746), (15, 0.002365885302424431), (16, 0.04899285361170769), (17, 0.00033323877141810954), (18, 0.0003564446233212948), (19, 0.004511591047048569), (20, 0.004349738359451294), (21, 0.0005182242020964622), (22, 0.0007008135435171425), (23, 0.0014548336621373892), (24, 0.0032747851219028234), (25, 0.00555224297568202), (26, 0.010695383884012699), (27, 0.013725196942687035)], Temporal MSE: [(0, 6.715272320434451e-06), (1, 4.948932200932177e-06), (2, 7.398163234029198e-06), (3, 6.809411388530862e-06), (4, 9.815965313464403e-06), (5, 1.2158982826804277e-05), (6, 0.00011115096276625991), (7, 4.654045551433228e-05), (8, 4.693647861131467e-05), (9, 3.328027742099948e-05), (10, 4.502173396758735e-05), (11, 4.8894406063482165e-05), (12, 4.77913417853415e-05), (13, 8.370418072445318e-05), (14, 0.00017231950187124312), (15, 0.0001619351387489587), (16, 0.0001489822898292914), (17, 0.00012199389311717823), (18, 0.0001227762404596433), (19, 0.00018956292478833348), (20, 0.00026160565903410316), (21, 0.00026504939887672663), (22, 0.00026146331219933927), (23, 0.000445548037532717), (24, 0.0009159148903563619), (25, 0.0016544866375625134), (26, 0.001828608918003738), (27, 0.017423348501324654)]
+
+ 97%|█████████▋| 145/150 [21:20<00:42,  8.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 24, Spatial MSE: [(0, 4.306911705498351e-06), (1, 8.375949983019382e-05), (2, 0.00010578470391919836), (3, 2.0201126972096972e-05), (4, 2.4468257834087126e-05), (5, 3.879324140143581e-05), (6, 0.00010726074106059968), (7, 8.891893230611458e-05), (8, 0.00011128270125482231), (9, 0.00012151806731708348), (10, 0.00011367147817509249), (11, 0.0001371737744193524), (12, 0.00010238050890620798), (13, 0.0002850457967724651), (14, 0.001163027249276638), (15, 0.0014585963217541575), (16, 0.024806175380945206), (17, 0.0003454584802966565), (18, 0.00035344756906852126), (19, 0.002535153180360794), (20, 0.002487235702574253), (21, 0.0005627268110401928), (22, 0.0007896882016211748), (23, 0.0016734946984797716), (24, 0.003995528444647789), (25, 0.006672159302979708), (26, 0.013305461034178734), (27, 0.02602134831249714)], Temporal MSE: [(0, 8.073450771917123e-06), (1, 6.209697403392056e-06), (2, 9.340487849840429e-06), (3, 7.97408301878022e-06), (4, 1.2477160453272518e-05), (5, 1.535645242256578e-05), (6, 0.0001383728376822546), (7, 5.6946442782646045e-05), (8, 6.21499857516028e-05), (9, 4.1995037463493645e-05), (10, 5.58733481739182e-05), (11, 5.9293819504091516e-05), (12, 5.238196536083706e-05), (13, 8.526328019797802e-05), (14, 0.00018006202299147844), (15, 0.00016984541434794664), (16, 0.00016298236732836813), (17, 0.00012997111480217427), (18, 0.00012445537140592933), (19, 0.000195981003344059), (20, 0.00027644954388961196), (21, 0.0002616860147099942), (22, 0.00027322035748511553), (23, 0.0005079468828625977), (24, 0.0010989380534738302), (25, 0.0019655469805002213), (26, 0.0027828658930957317), (27, 0.03683947026729584)]
+
+ 97%|█████████▋| 146/150 [21:30<00:35,  8.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 18, Spatial MSE: [(0, 5.483702807396185e-06), (1, 0.00011146267934236676), (2, 0.00013362847676035017), (3, 2.5622999601182528e-05), (4, 3.2859359635040164e-05), (5, 4.782498945132829e-05), (6, 0.00013850393588654697), (7, 0.00013263810251373798), (8, 0.00016556975606363267), (9, 0.0001553344918647781), (10, 0.00015409992192871869), (11, 0.0001609789178473875), (12, 0.0001320414012297988), (13, 0.0007969197467900813), (14, 0.005490932613611221), (15, 0.005278268828988075), (16, 0.11250246316194534), (17, 0.0005719168693758547), (18, 0.0006563282804563642), (19, 0.010407613590359688), (20, 0.009977676905691624), (21, 0.0008655189303681254), (22, 0.0011931261979043484), (23, 0.0023113221395760775), (24, 0.00556617695838213), (25, 0.009363715536892414), (26, 0.018847662955522537), (27, 0.06414850801229477)], Temporal MSE: [(0, 1.0114369615621399e-05), (1, 7.909919986559544e-06), (2, 1.254983726539649e-05), (3, 1.0017636668635532e-05), (4, 1.5480824004043825e-05), (5, 1.9579121726565063e-05), (6, 0.0001876568130683154), (7, 7.649035251233727e-05), (8, 8.192313543986529e-05), (9, 5.418249202193692e-05), (10, 6.990620022406802e-05), (11, 7.475960592273623e-05), (12, 7.247459143400192e-05), (13, 0.00012508548388723284), (14, 0.00026133834035135806), (15, 0.0002616094134282321), (16, 0.0002168090722989291), (17, 0.00017404169193468988), (18, 0.00019664027786348015), (19, 0.00031547373509965837), (20, 0.00042724705417640507), (21, 0.0004470683343242854), (22, 0.00042964352178387344), (23, 0.0007231743657030165), (24, 0.0015370800392702222), (25, 0.002664815867319703), (26, 0.004580889828503132), (27, 0.08501047641038895)]
+
+ 98%|█████████▊| 147/150 [21:39<00:26,  8.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 12, Spatial MSE: [(0, 7.467268460459309e-06), (1, 0.0001473679149057716), (2, 0.00019094588060397655), (3, 3.479991210042499e-05), (4, 4.649799666367471e-05), (5, 6.162372301332653e-05), (6, 0.0001965659175766632), (7, 0.00015593148418702185), (8, 0.00020500944810919464), (9, 0.00020444106485228986), (10, 0.00021682385704480112), (11, 0.0002088767651002854), (12, 0.00017771411512512714), (13, 0.0004949317662976682), (14, 0.0021788233425468206), (15, 0.0017534592188894749), (16, 0.02996917814016342), (17, 0.0005547875771299005), (18, 0.0005957253160886467), (19, 0.003295322647318244), (20, 0.003250954207032919), (21, 0.0009519265731796622), (22, 0.0013537928462028503), (23, 0.00282988790422678), (24, 0.007494449615478516), (25, 0.014264630153775215), (26, 0.028443850576877594), (27, 0.21125422418117523)], Temporal MSE: [(0, 1.351121136394795e-05), (1, 1.0807048965943977e-05), (2, 1.7121745258918963e-05), (3, 1.2820997653761879e-05), (4, 2.0776309611392207e-05), (5, 2.7465339371701702e-05), (6, 0.00024113063409458846), (7, 0.00010449782712385058), (8, 0.00011747112148441374), (9, 7.104435644578189e-05), (10, 9.546514775138348e-05), (11, 0.00010033662692876533), (12, 8.622367022326216e-05), (13, 0.00013686744205188006), (14, 0.00027111833333037794), (15, 0.0002622512402012944), (16, 0.00023305871582124382), (17, 0.00018536798597779125), (18, 0.00018511053349357098), (19, 0.00030349628650583327), (20, 0.00041474762838333845), (21, 0.0003893448447342962), (22, 0.0004510203725658357), (23, 0.0008636820130050182), (24, 0.0020178649574518204), (25, 0.003484654938802123), (26, 0.008430872112512589), (27, 0.2097354531288147)]
+
+ 99%|█████████▊| 148/150 [21:47<00:17,  8.70s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 6, Spatial MSE: [(0, 1.315005556534743e-05), (1, 0.0002506249293219298), (2, 0.00029337420710362494), (3, 6.170567212393507e-05), (4, 8.522700954927132e-05), (5, 9.75478978944011e-05), (6, 0.00032083920086733997), (7, 0.0002521649585105479), (8, 0.00035065971314907074), (9, 0.0003527882508933544), (10, 0.00037249960587359965), (11, 0.00036727526457980275), (12, 0.0003225551045034081), (13, 0.0007406593067571521), (14, 0.0026303266640752554), (15, 0.0030375546775758266), (16, 0.05073803663253784), (17, 0.0008619822328910232), (18, 0.0009433929226361215), (19, 0.005501650273799896), (20, 0.005289016291499138), (21, 0.0018861073767766356), (22, 0.0024257798213511705), (23, 0.004796900320798159), (24, 0.01452408917248249), (25, 0.03424948826432228), (26, 0.0671435073018074), (27, 1.2486319541931152)], Temporal MSE: [(0, 2.266947558382526e-05), (1, 1.9057231838814914e-05), (2, 3.161161293974146e-05), (3, 2.2272695787250996e-05), (4, 3.5190289054298773e-05), (5, 4.747688217321411e-05), (6, 0.0003971390542574227), (7, 0.0001791590912034735), (8, 0.00020847306586802006), (9, 0.0001237882097484544), (10, 0.00016590632731094956), (11, 0.00018225653911940753), (12, 0.00015354322385974228), (13, 0.00022237258963286877), (14, 0.00045801408123224974), (15, 0.0004483221855480224), (16, 0.00036395591450855136), (17, 0.00029066792922094464), (18, 0.0003071255050599575), (19, 0.0005003769765608013), (20, 0.0006520497845485806), (21, 0.000632291252259165), (22, 0.0007742177695035934), (23, 0.0014891293831169605), (24, 0.0035668043419718742), (25, 0.006154029630124569), (26, 0.020874369889497757), (27, 0.6316867470741272)]
+
+ 99%|█████████▉| 149/150 [21:55<00:08,  8.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 0, Spatial MSE: [(0, 8.45270260469988e-05), (1, 0.0009714089101180434), (2, 0.003108629724010825), (3, 0.0005313648725859821), (4, 0.0004440204065758735), (5, 0.0006400909624062479), (6, 0.0011990884086117148), (7, 0.0007813806878402829), (8, 0.0011703759664669633), (9, 0.0010842266492545605), (10, 0.0011568452464416623), (11, 0.001280507305637002), (12, 0.0015574678545817733), (13, 0.003266512183472514), (14, 0.011039283126592636), (15, 0.012885688804090023), (16, 0.24228917062282562), (17, 0.003476375015452504), (18, 0.005088859237730503), (19, 0.026744846254587173), (20, 0.026179373264312744), (21, 0.02078912779688835), (22, 0.04258172586560249), (23, 0.07496494054794312), (24, 0.16075965762138367), (25, 0.9769148230552673), (26, 1.3812240362167358), (27, 4.260166645050049)], Temporal MSE: [(0, 0.00014276568253990263), (1, 8.953348879003897e-05), (2, 0.00011616459232755005), (3, 9.529375529382378e-05), (4, 0.00025030181859619915), (5, 0.00023128156317397952), (6, 0.0009938973234966397), (7, 0.0005868455045856535), (8, 0.0007086030673235655), (9, 0.0004335435514803976), (10, 0.0008339708438143134), (11, 0.000873102864716202), (12, 0.0007247676840052009), (13, 0.0011414136970415711), (14, 0.001934544532559812), (15, 0.0018334250198677182), (16, 0.0013928881380707026), (17, 0.0014865751145407557), (18, 0.001574088935740292), (19, 0.0029234886169433594), (20, 0.0035840074997395277), (21, 0.004546000622212887), (22, 0.008674174547195435), (23, 0.018308069556951523), (24, 0.042935896664857864), (25, 0.07698742300271988), (26, 0.11950463056564331), (27, 0.7631118297576904)]
+
+100%|██████████| 150/150 [22:03<00:00,  8.32s/it]
+100%|██████████| 150/150 [22:03<00:00,  8.82s/it]
+Saved spatial MLP MSE outputs to ./samples/opensora_plan/skip_mse/3_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora_plan/skip_mse/3_temporal_mlp_mse.pt
+
+  0%|          | 0/150 [00:00<?, ?it/s]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.2005166354356334e-06), (1, 6.511560968647245e-06), (2, 0.00016623677220195532), (3, 2.118652264471166e-05), (4, 9.129171303356998e-06), (5, 7.024330898275366e-06), (6, 8.274473657365888e-06), (7, 6.678691534034442e-06), (8, 7.135971827665344e-06), (9, 9.147499440587126e-06), (10, 1.379213517793687e-05), (11, 1.7949674656847492e-05), (12, 2.087555367324967e-05), (13, 3.4828277421183884e-05), (14, 0.00011571389768505469), (15, 0.0002095347736030817), (16, 0.0011920725228264928), (17, 7.320849545067176e-05), (18, 8.83237153175287e-05), (19, 0.00013034742732997984), (20, 0.00013126213161740452), (21, 8.930325566325337e-05), (22, 0.00010242143616778776), (23, 9.395390952704474e-05), (24, 0.00018030204228125513), (25, 0.00042166313505731523), (26, 0.0004164402198512107), (27, 0.0036458761896938086)], Temporal MSE: [(0, 1.6457676110803732e-06), (1, 1.2035425243084319e-06), (2, 1.701735982351238e-06), (3, 1.5977523162291618e-06), (4, 1.1118949032606906e-06), (5, 1.5746113604109269e-06), (6, 8.653072995912225e-07), (7, 1.082295284504653e-06), (8, 8.054317959249602e-07), (9, 1.2656264516408555e-06), (10, 2.942038008768577e-06), (11, 2.036196974586346e-06), (12, 2.588442612250219e-06), (13, 7.2521434049122036e-06), (14, 6.889561518619303e-06), (15, 8.050782525970135e-06), (16, 5.228082045505289e-06), (17, 1.0120978913619183e-05), (18, 5.8172586250293534e-06), (19, 6.149897217255784e-06), (20, 1.0087011105497368e-05), (21, 6.0648872022284195e-06), (22, 1.2788643289241008e-05), (23, 9.561612387187779e-06), (24, 2.164954275940545e-05), (25, 2.255587969557382e-05), (26, 3.847672815027181e-06), (27, 1.198564041260397e-05)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 891, Spatial MSE: [(0, 1.183411502125864e-08), (1, 1.4693505079321767e-07), (2, 1.4941464314688346e-06), (3, 2.794026841002051e-07), (4, 1.1201140637240314e-07), (5, 8.940934748125073e-08), (6, 2.2095952090239734e-07), (7, 1.305433272591472e-07), (8, 1.1965714463713084e-07), (9, 1.5860487678764912e-07), (10, 1.750476457118566e-07), (11, 1.8515009969632956e-07), (12, 2.1515657522286347e-07), (13, 2.5538423642501584e-07), (14, 8.147644052769465e-07), (15, 1.4730964039699757e-06), (16, 8.99168117030058e-06), (17, 5.693707976206497e-07), (18, 1.1285361551927053e-06), (19, 1.2307081078688498e-06), (20, 1.079307139661978e-06), (21, 1.1438737601565663e-06), (22, 1.4806120134380762e-06), (23, 1.1751385500247125e-06), (24, 3.2442808333144058e-06), (25, 5.402177521318663e-06), (26, 5.898384642932797e-06), (27, 3.0574901757063344e-05)], Temporal MSE: [(0, 7.094066223345408e-09), (1, 9.628079666867961e-09), (2, 9.692401548022644e-09), (3, 1.1827375523409955e-08), (4, 1.5407016107360505e-08), (5, 2.7104226418828148e-08), (6, 1.423702045144637e-08), (7, 1.8463188311557133e-08), (8, 1.819716644035907e-08), (9, 2.477538707523763e-08), (10, 3.4311380403551084e-08), (11, 3.3604329985337245e-08), (12, 3.2670431693304636e-08), (13, 6.279424979993564e-08), (14, 5.989355145175068e-08), (15, 9.526161903750108e-08), (16, 4.108786200163195e-08), (17, 7.799287971010926e-08), (18, 7.600363716164793e-08), (19, 6.952746645083607e-08), (20, 1.2581985231463477e-07), (21, 9.49467562350037e-08), (22, 1.495135393270175e-07), (23, 1.6651650014409825e-07), (24, 3.301822175671987e-07), (25, 3.975645768150571e-07), (26, 1.4624751543124148e-07), (27, 1.4448025353885896e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 9.316811429016525e-07), (1, 5.775046702183317e-06), (2, 0.00015734508633613586), (3, 1.5882096704444848e-05), (4, 7.079116130626062e-06), (5, 5.604172656603623e-06), (6, 5.644654265779536e-06), (7, 7.223905413411558e-06), (8, 7.872903552197386e-06), (9, 1.1662603355944157e-05), (10, 1.7079792087315582e-05), (11, 2.2462685592472553e-05), (12, 2.8597642085514963e-05), (13, 3.526945147314109e-05), (14, 0.00012019287532893941), (15, 0.00023903716646600515), (16, 0.0007409299141727388), (17, 6.282372487476096e-05), (18, 8.720460027689114e-05), (19, 0.00011015927884727716), (20, 0.00014093483332544565), (21, 7.635670772287995e-05), (22, 7.637491216883063e-05), (23, 8.934989455156028e-05), (24, 0.0002292902790941298), (25, 0.0004923928063362837), (26, 0.00023960988619364798), (27, 0.002598891733214259)], Temporal MSE: [(0, 1.3205631148593966e-06), (1, 1.0196733910561306e-06), (2, 1.1659185474854894e-06), (3, 1.0096377991430927e-06), (4, 8.997441796054773e-07), (5, 1.3927485724707367e-06), (6, 8.273523803836724e-07), (7, 1.4078394769967417e-06), (8, 8.558535569136438e-07), (9, 1.6104910400827066e-06), (10, 4.394243660499342e-06), (11, 2.890367113650427e-06), (12, 3.5282159842608962e-06), (13, 9.699190741230268e-06), (14, 9.494700861978345e-06), (15, 1.1047645784856286e-05), (16, 5.80026653551613e-06), (17, 1.2359735592326615e-05), (18, 6.445554845413426e-06), (19, 7.706328688072972e-06), (20, 1.221794809680432e-05), (21, 6.430990652006585e-06), (22, 1.1644254300335888e-05), (23, 9.349120773549657e-06), (24, 2.3703527403995395e-05), (25, 2.1686608306481503e-05), (26, 4.397003522171872e-06), (27, 7.92549781181151e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 888, Spatial MSE: [(0, 1.1052409654155326e-08), (1, 1.2527746662271966e-07), (2, 1.2474702089093626e-06), (3, 2.353889811956833e-07), (4, 1.0408975725795244e-07), (5, 8.209679691617566e-08), (6, 1.9887410473984346e-07), (7, 1.1982977810021112e-07), (8, 1.0857633014893509e-07), (9, 1.3959068212443526e-07), (10, 1.497778043813014e-07), (11, 1.534122304747143e-07), (12, 1.7947506592008722e-07), (13, 1.8966987624935427e-07), (14, 4.6109227014312637e-07), (15, 8.234131883000373e-07), (16, 6.2527965383196715e-06), (17, 4.5407816173792526e-07), (18, 9.177073252431001e-07), (19, 8.878467951944913e-07), (20, 8.461938705295324e-07), (21, 1.0391693194833351e-06), (22, 1.3151878874850809e-06), (23, 9.990797025238862e-07), (24, 2.8398710583132925e-06), (25, 4.723484380519949e-06), (26, 5.531449005502509e-06), (27, 2.7970225346507505e-05)], Temporal MSE: [(0, 6.693923193523688e-09), (1, 8.663159078992066e-09), (2, 8.782551574881836e-09), (3, 1.0702009056728912e-08), (4, 1.4121172675629623e-08), (5, 2.478644312020606e-08), (6, 1.2889914913216671e-08), (7, 1.6788714418680684e-08), (8, 1.6428991500561096e-08), (9, 2.1725579557596575e-08), (10, 2.994734416006395e-08), (11, 2.9424507275166434e-08), (12, 2.7506162680879243e-08), (13, 4.9249965883291225e-08), (14, 4.328470026848663e-08), (15, 7.803510015946813e-08), (16, 3.381927626833203e-08), (17, 6.728304668968121e-08), (18, 5.931768498612655e-08), (19, 5.465298613671621e-08), (20, 9.737220807437552e-08), (21, 7.582996630617345e-08), (22, 1.3450355140776082e-07), (23, 1.4044609031316213e-07), (24, 2.7373840794098214e-07), (25, 3.5763122241405654e-07), (26, 1.3269027476781048e-07), (27, 1.3278102528602176e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.3298019894136814e-06), (1, 3.726682052729302e-06), (2, 9.232187585439533e-05), (3, 1.6909789337660186e-05), (4, 6.972832125029527e-06), (5, 5.055524525232613e-06), (6, 7.334667316172272e-06), (7, 4.877906121691922e-06), (8, 7.329796062549576e-06), (9, 1.2754262570524588e-05), (10, 1.591049658600241e-05), (11, 1.83990068762796e-05), (12, 2.1421848941827193e-05), (13, 2.770073842839338e-05), (14, 7.598627416882664e-05), (15, 0.00010524697427172214), (16, 0.000666500476654619), (17, 3.699062654050067e-05), (18, 5.420293746283278e-05), (19, 6.74536349833943e-05), (20, 4.714640454039909e-05), (21, 3.9969239878701046e-05), (22, 5.436451829154976e-05), (23, 5.683051494997926e-05), (24, 0.00012497929856181145), (25, 0.00020895757188554853), (26, 0.00020725579815916717), (27, 0.004073753487318754)], Temporal MSE: [(0, 1.671874656494765e-06), (1, 1.000023985398002e-06), (2, 1.7367544842272764e-06), (3, 1.411041921528522e-06), (4, 1.1039520586564322e-06), (5, 1.4796073628531303e-06), (6, 8.830647288959881e-07), (7, 7.469162142115238e-07), (8, 1.4168742836773163e-06), (9, 1.7155753084807657e-06), (10, 2.7433513878349913e-06), (11, 2.2433016511058668e-06), (12, 3.375044116182835e-06), (13, 6.676032171526458e-06), (14, 8.17472391645424e-06), (15, 5.620257525151828e-06), (16, 4.705116225522943e-06), (17, 4.763925062434282e-06), (18, 4.743574208987411e-06), (19, 5.196609436097788e-06), (20, 8.412663191847969e-06), (21, 4.353587428340688e-06), (22, 5.385651093092747e-06), (23, 6.8016252043889835e-06), (24, 1.91333001566818e-05), (25, 1.3553512872022111e-05), (26, 4.79890468341182e-06), (27, 3.720547738339519e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 885, Spatial MSE: [(0, 1.178616404473587e-08), (1, 1.4535048364905379e-07), (2, 1.48104993513698e-06), (3, 2.791090594200796e-07), (4, 1.1737267868738854e-07), (5, 9.100429565478407e-08), (6, 2.2368875818301603e-07), (7, 1.3195833048484928e-07), (8, 1.2232209201101796e-07), (9, 1.6532577262751147e-07), (10, 1.8547690672221506e-07), (11, 1.9506606463437492e-07), (12, 2.3142793281749618e-07), (13, 2.670601588761201e-07), (14, 6.623972126362787e-07), (15, 1.0457898724780534e-06), (16, 4.709797849500319e-06), (17, 5.633094133372651e-07), (18, 1.0954881872748956e-06), (19, 9.280070116801653e-07), (20, 9.410674124410434e-07), (21, 1.1799297681136522e-06), (22, 1.58220268531295e-06), (23, 1.226119024977379e-06), (24, 3.3910580441443017e-06), (25, 5.429204975371249e-06), (26, 6.187184681039071e-06), (27, 3.1880128517514095e-05)], Temporal MSE: [(0, 6.9882921671649e-09), (1, 9.563978942139784e-09), (2, 9.625089170128831e-09), (3, 1.1752174344792365e-08), (4, 1.5549989740293313e-08), (5, 2.7169248184577555e-08), (6, 1.450610298547872e-08), (7, 1.8669387813474714e-08), (8, 1.859792675418248e-08), (9, 2.5315904039757697e-08), (10, 3.533221715201762e-08), (11, 3.4432481754720357e-08), (12, 3.4780416768853684e-08), (13, 6.5581460262365e-08), (14, 6.577425182285879e-08), (15, 9.849873805478637e-08), (16, 4.433482914123488e-08), (17, 7.851885897025568e-08), (18, 7.643381394473181e-08), (19, 7.13892944759209e-08), (20, 1.349211942169859e-07), (21, 9.458215544100312e-08), (22, 1.5469125003164663e-07), (23, 1.653265400136661e-07), (24, 3.417488869672525e-07), (25, 4.106864253117237e-07), (26, 1.522318484603602e-07), (27, 1.5454521928859322e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 3.680488021018391e-07), (1, 9.466759820497828e-07), (2, 2.1592681150650606e-05), (3, 5.118159606354311e-06), (4, 2.087096845571068e-06), (5, 1.828868334996514e-06), (6, 3.112650801995187e-06), (7, 2.685638719412964e-06), (8, 5.138855613040505e-06), (9, 1.0663638022379018e-05), (10, 1.4303305761131924e-05), (11, 1.7784350347938016e-05), (12, 2.107696491293609e-05), (13, 2.6018878998002037e-05), (14, 6.428030610550195e-05), (15, 0.0001358082954538986), (16, 0.00017356059106532484), (17, 3.775836739805527e-05), (18, 4.747042476083152e-05), (19, 4.737499330076389e-05), (20, 4.729703869088553e-05), (21, 2.979771488753613e-05), (22, 4.13241250498686e-05), (23, 4.7427362005691975e-05), (24, 9.962077456293628e-05), (25, 0.00016006614896468818), (26, 6.606683746213093e-05), (27, 0.0006666838889941573)], Temporal MSE: [(0, 4.1829966335171775e-07), (1, 2.261637064293609e-07), (2, 2.9639508625223243e-07), (3, 3.185894854595972e-07), (4, 3.4609738008839486e-07), (5, 5.635857291963475e-07), (6, 4.844520731239754e-07), (7, 6.030294912306999e-07), (8, 6.130467227194458e-07), (9, 1.2449758060029126e-06), (10, 2.731110726017505e-06), (11, 1.9561332464945735e-06), (12, 2.8966314857825637e-06), (13, 6.6411762418283615e-06), (14, 8.56446240504738e-06), (15, 7.1197996476257686e-06), (16, 5.02757029607892e-06), (17, 7.2968109634530265e-06), (18, 6.303947884589434e-06), (19, 6.679048510704888e-06), (20, 1.2933273865201045e-05), (21, 5.413471171777928e-06), (22, 5.076582510810113e-06), (23, 5.835178853885736e-06), (24, 1.6876556401257403e-05), (25, 1.1226498827454634e-05), (26, 2.9818825169058982e-06), (27, 2.3203617729450343e-06)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 882, Spatial MSE: [(0, 1.1013877809773476e-08), (1, 1.2289686424082902e-07), (2, 1.2324413773967535e-06), (3, 2.3430325768458715e-07), (4, 1.0439755726565636e-07), (5, 8.22865970917519e-08), (6, 1.969318219607885e-07), (7, 1.2140884564360022e-07), (8, 1.097662973847946e-07), (9, 1.425438398428014e-07), (10, 1.517227588010428e-07), (11, 1.557529714091288e-07), (12, 1.816085415384805e-07), (13, 1.7972055843529233e-07), (14, 3.8719991835023393e-07), (15, 7.131625352485571e-07), (16, 3.4639363093447173e-06), (17, 4.4628882278630044e-07), (18, 8.687780450600258e-07), (19, 7.261196515173651e-07), (20, 7.646663107152563e-07), (21, 9.765655022420106e-07), (22, 1.2651752285819384e-06), (23, 9.932911098076147e-07), (24, 2.831603524100501e-06), (25, 4.703622380475281e-06), (26, 5.523401796381222e-06), (27, 2.801473056024406e-05)], Temporal MSE: [(0, 6.674174546361655e-09), (1, 8.632594195034926e-09), (2, 8.763397119082583e-09), (3, 1.0717578824426255e-08), (4, 1.4177110152502337e-08), (5, 2.4739236437198997e-08), (6, 1.2892877876424791e-08), (7, 1.681837602518499e-08), (8, 1.6658662005397673e-08), (9, 2.20450839805153e-08), (10, 3.016290150981149e-08), (11, 2.935609622056745e-08), (12, 2.756531536363127e-08), (13, 4.653809781984819e-08), (14, 4.412492415895031e-08), (15, 7.821728331691702e-08), (16, 3.189617459042893e-08), (17, 6.200207280926406e-08), (18, 5.807168079741132e-08), (19, 5.256711688161886e-08), (20, 9.702763748009602e-08), (21, 7.421321868150699e-08), (22, 1.2696355611296894e-07), (23, 1.3453080782710458e-07), (24, 2.7309684469400963e-07), (25, 3.583005252494331e-07), (26, 1.3288534717048606e-07), (27, 1.3245113450466306e-07)]
+Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 3.257347771068453e-06), (1, 3.0482970032608137e-05), (2, 0.0007385446806438267), (3, 8.26596369734034e-05), (4, 3.151630153297447e-05), (5, 2.1679585188394412e-05), (6, 2.6245510525768623e-05), (7, 1.8901777366409078e-05), (8, 1.9078661352978088e-05), (9, 2.4107419449137524e-05), (10, 3.3859705581562594e-05), (11, 4.0361159335589036e-05), (12, 5.285551378619857e-05), (13, 8.012286707526073e-05), (14, 0.000273342797299847), (15, 0.000550911296159029), (16, 0.0015478153945878148), (17, 0.0001896924659376964), (18, 0.0002960480924230069), (19, 0.00031365727772936225), (20, 0.00042464700527489185), (21, 0.0002002318942686543), (22, 0.0002542953297961503), (23, 0.0002688754757400602), (24, 0.0005304950173012912), (25, 0.0011711011175066233), (26, 0.0012303260155022144), (27, 0.014994676224887371)], Temporal MSE: [(0, 4.381097824079916e-06), (1, 4.077522135048639e-06), (2, 6.113509698479902e-06), (3, 5.6392177611996885e-06), (4, 4.587338480632752e-06), (5, 4.887584509560838e-06), (6, 2.4391333681705873e-06), (7, 3.5099214983347338e-06), (8, 2.4646956262586173e-06), (9, 3.442097067818395e-06), (10, 7.261785412993049e-06), (11, 5.3320459301176015e-06), (12, 6.539133664773544e-06), (13, 1.972706741071306e-05), (14, 1.7547596144140698e-05), (15, 2.261367808387149e-05), (16, 1.2841134775953833e-05), (17, 3.542933700373396e-05), (18, 1.7337843019049615e-05), (19, 2.387316999374889e-05), (20, 4.1638057155068964e-05), (21, 1.7186352124554105e-05), (22, 2.5159080905723386e-05), (23, 2.1954861949780025e-05), (24, 6.115438736742362e-05), (25, 7.299065327970311e-05), (26, 1.1787786206696182e-05), (27, 3.9278333133552223e-05)]
+
+  1%|          | 1/150 [01:19<3:18:06, 79.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 879, Spatial MSE: [(0, 1.2334981036588033e-08), (1, 1.5749162685096962e-07), (2, 1.577516968609416e-06), (3, 2.960605058888177e-07), (4, 1.1771125230097823e-07), (5, 9.289598068562555e-08), (6, 2.3581996799748595e-07), (7, 1.3586267755272274e-07), (8, 1.2865112353210861e-07), (9, 1.7803398577598273e-07), (10, 2.0327814809206757e-07), (11, 2.212557177472263e-07), (12, 2.686291225018067e-07), (13, 2.964452505693771e-07), (14, 5.570466896642756e-07), (15, 8.69850339313416e-07), (16, 1.5706775684520835e-06), (17, 6.114497637099703e-07), (18, 1.1524758747327724e-06), (19, 7.354135505011072e-07), (20, 9.071802651305916e-07), (21, 1.2031208598273224e-06), (22, 1.605473471499863e-06), (23, 1.2842282330893795e-06), (24, 3.5248010590294143e-06), (25, 5.6379681154794525e-06), (26, 6.172846951812971e-06), (27, 3.158484469167888e-05)], Temporal MSE: [(0, 7.27496862751309e-09), (1, 9.761973451816175e-09), (2, 9.983108562039433e-09), (3, 1.2120662695735973e-08), (4, 1.6016628023862722e-08), (5, 2.8283540842721777e-08), (6, 1.540999683413702e-08), (7, 1.9575553622530606e-08), (8, 1.9718411792268853e-08), (9, 2.704006440978901e-08), (10, 3.887889121756416e-08), (11, 3.832731465536199e-08), (12, 3.907215173626355e-08), (13, 6.878415348410272e-08), (14, 7.690636749657642e-08), (15, 1.0850060050415777e-07), (16, 5.224427823691258e-08), (17, 8.464969880606077e-08), (18, 8.76013146466903e-08), (19, 7.710525551374303e-08), (20, 1.490808188009396e-07), (21, 1.0313053877553102e-07), (22, 1.5791782459473325e-07), (23, 1.6775349820363772e-07), (24, 3.624681994551793e-07), (25, 4.3551261796892504e-07), (26, 1.5786379492510605e-07), (27, 1.5424546973008546e-07)]
+
+  1%|▏         | 2/150 [01:28<1:33:08, 37.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 3.802310175160528e-06), (1, 2.962243161164224e-05), (2, 0.0008802111260592937), (3, 9.918017167365178e-05), (4, 4.0193714085035026e-05), (5, 2.6135392545256764e-05), (6, 3.63517137884628e-05), (7, 2.2956168322707526e-05), (8, 2.23295883188257e-05), (9, 2.8974976885365322e-05), (10, 4.017719402327202e-05), (11, 4.691139838541858e-05), (12, 6.476396083598956e-05), (13, 9.343348938273266e-05), (14, 0.0003145175287500024), (15, 0.0006932622054591775), (16, 0.002782968105748296), (17, 0.0002686331863515079), (18, 0.0003318553790450096), (19, 0.0004153391346335411), (20, 0.0005826078122481704), (21, 0.00033576079295016825), (22, 0.0003185936075169593), (23, 0.0003260638623032719), (24, 0.0007107650162652135), (25, 0.001815709052607417), (26, 0.0015340772224590182), (27, 0.02031617984175682)], Temporal MSE: [(0, 5.162603883945849e-06), (1, 4.7972080210456625e-06), (2, 7.456076673406642e-06), (3, 6.2621493270853534e-06), (4, 5.4629758778901305e-06), (5, 6.373147698468529e-06), (6, 2.83271879197855e-06), (7, 4.110823283554055e-06), (8, 3.1394690722663654e-06), (9, 4.072055162396282e-06), (10, 8.296218766190577e-06), (11, 6.342289452732075e-06), (12, 7.816508514224552e-06), (13, 2.2046026060706936e-05), (14, 2.095674426527694e-05), (15, 2.601903361210134e-05), (16, 1.704575515759643e-05), (17, 4.335389894549735e-05), (18, 2.357323501200881e-05), (19, 3.0765062547288835e-05), (20, 5.665666685672477e-05), (21, 2.210639650002122e-05), (22, 3.7254012568155304e-05), (23, 2.86374688585056e-05), (24, 7.7415032137651e-05), (25, 9.685235272627324e-05), (26, 1.605613397259731e-05), (27, 4.7279256250476465e-05)]
+
+  2%|▏         | 3/150 [01:37<1:00:39, 24.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 876, Spatial MSE: [(0, 1.109057468084984e-08), (1, 1.284055315409205e-07), (2, 1.2312955277593574e-06), (3, 2.3471886834158795e-07), (4, 1.0482184364946079e-07), (5, 8.103534554493308e-08), (6, 1.995530993781358e-07), (7, 1.194502345924775e-07), (8, 1.0967241337311862e-07), (9, 1.4337670961594995e-07), (10, 1.5290365240616666e-07), (11, 1.6014631398775236e-07), (12, 1.8754374764284876e-07), (13, 2.2279165534655476e-07), (14, 4.996055054107273e-07), (15, 7.807843189766572e-07), (16, 2.902985443142825e-06), (17, 4.6754277605032257e-07), (18, 9.442275654691912e-07), (19, 7.227553169286693e-07), (20, 7.739245688753726e-07), (21, 1.180844606096798e-06), (22, 1.3063114465694525e-06), (23, 1.0426550716147176e-06), (24, 2.971025423903484e-06), (25, 4.731100489152595e-06), (26, 5.477113973029191e-06), (27, 2.7841097107739188e-05)], Temporal MSE: [(0, 6.544488950765981e-09), (1, 8.45955128170317e-09), (2, 8.544481566730155e-09), (3, 1.0533133476542389e-08), (4, 1.3901392037496407e-08), (5, 2.4515507845990214e-08), (6, 1.3008972565842214e-08), (7, 1.6947206304962492e-08), (8, 1.6839594607631625e-08), (9, 2.251356434612717e-08), (10, 3.1204212547208954e-08), (11, 3.049999008908344e-08), (12, 2.9491596720276902e-08), (13, 5.6305228923747563e-08), (14, 4.789322360920778e-08), (15, 8.48632808470029e-08), (16, 3.556142758043279e-08), (17, 6.616357239863646e-08), (18, 6.411627850866353e-08), (19, 5.7789186769241496e-08), (20, 1.0719473664266843e-07), (21, 8.049359934148015e-08), (22, 1.2970681950719154e-07), (23, 1.3957479438886367e-07), (24, 2.8898043069602863e-07), (25, 3.859306332287815e-07), (26, 1.3739492032982525e-07), (27, 1.3427172973479173e-07)]
+
+  3%|▎         | 4/150 [01:45<43:55, 18.05s/it]  Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 870, Spatial MSE: [(0, 1.2235654139658436e-05), (1, 0.00010254892549710348), (2, 0.0025922467466443777), (3, 0.00029033931787125766), (4, 0.00011505620932439342), (5, 7.681795977987349e-05), (6, 8.667763904668391e-05), (7, 9.311093162978068e-05), (8, 7.893981819506735e-05), (9, 0.00010642356937751174), (10, 0.00014128127077128738), (11, 0.00016863438941072673), (12, 0.00021549071243498474), (13, 0.0003109475946985185), (14, 0.0008212795364670455), (15, 0.001893280423246324), (16, 0.005097310058772564), (17, 0.000734063214622438), (18, 0.0010371492244303226), (19, 0.0010313316015526652), (20, 0.0014433249598369002), (21, 0.0006690426380373538), (22, 0.0009589653927832842), (23, 0.001016260706819594), (24, 0.0021452801302075386), (25, 0.005742092151194811), (26, 0.004705057945102453), (27, 0.056978799402713776)], Temporal MSE: [(0, 1.689921373326797e-05), (1, 1.5465377146028914e-05), (2, 2.2390164303942584e-05), (3, 1.8784043277264573e-05), (4, 1.593881825101562e-05), (5, 1.7988304534810595e-05), (6, 9.828804650169332e-06), (7, 1.1689523489621934e-05), (8, 1.0445198313391302e-05), (9, 1.4905596799508203e-05), (10, 3.014100730069913e-05), (11, 2.3071806936059147e-05), (12, 3.002178345923312e-05), (13, 8.164660539478064e-05), (14, 8.185992919607088e-05), (15, 9.234010940417647e-05), (16, 5.972634971840307e-05), (17, 0.00015481543960049748), (18, 8.373661694349721e-05), (19, 0.00012658609193749726), (20, 0.0002284538932144642), (21, 0.00010184983693761751), (22, 9.808440518099815e-05), (23, 0.00010069413838209584), (24, 0.00027520558796823025), (25, 0.00039757101330906153), (26, 5.317063187249005e-05), (27, 0.00014429654402192682)]
+
+  3%|▎         | 5/150 [01:52<34:38, 14.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 864, Spatial MSE: [(0, 1.5615787560818717e-05), (1, 0.0001389457320328802), (2, 0.003605743171647191), (3, 0.00039373248000629246), (4, 0.00016279729607049376), (5, 0.00010544196993578225), (6, 0.00014446717977989465), (7, 9.464172035222873e-05), (8, 9.835972014116123e-05), (9, 0.0001412084384355694), (10, 0.00018871386419050395), (11, 0.00021535893029067665), (12, 0.0002914068172685802), (13, 0.0003978861204814166), (14, 0.0011919393436983228), (15, 0.0023749282117933035), (16, 0.010358491912484169), (17, 0.0009675921755842865), (18, 0.0013845316134393215), (19, 0.0015578827587887645), (20, 0.0020615861285477877), (21, 0.0009684772812761366), (22, 0.0012173941358923912), (23, 0.0012602419592440128), (24, 0.002698064548894763), (25, 0.007132950238883495), (26, 0.005740720313042402), (27, 0.08712255954742432)], Temporal MSE: [(0, 2.1033412849646993e-05), (1, 1.9721421267604455e-05), (2, 2.766617944871541e-05), (3, 2.5044786525540985e-05), (4, 2.2030450054444373e-05), (5, 2.637777288327925e-05), (6, 1.1829532923002262e-05), (7, 1.98727248061914e-05), (8, 1.4063531125430018e-05), (9, 1.7797670807340182e-05), (10, 4.4573032937478274e-05), (11, 3.293590998509899e-05), (12, 3.856079274555668e-05), (13, 0.00010989174188580364), (14, 0.0001107874559238553), (15, 0.00012821305426768959), (16, 8.091918425634503e-05), (17, 0.00021759406081400812), (18, 0.00011757844185922295), (19, 0.00017930445028468966), (20, 0.0003782566636800766), (21, 0.00014024700794834644), (22, 0.00013646905426867306), (23, 0.00013101505464874208), (24, 0.00032197259133681655), (25, 0.00045451175537891686), (26, 6.845275493105873e-05), (27, 0.0001527371641714126)]
+
+  4%|▍         | 6/150 [02:01<29:29, 12.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 858, Spatial MSE: [(0, 3.9214210119098425e-05), (1, 0.00034890466486103833), (2, 0.008680407889187336), (3, 0.0009203909430652857), (4, 0.0003831347275990993), (5, 0.0002513610234018415), (6, 0.00025870726676657796), (7, 0.0002776952169369906), (8, 0.00023342712665908039), (9, 0.00030825770227238536), (10, 0.00040962122147902846), (11, 0.0004554053011815995), (12, 0.0006704293191432953), (13, 0.0009725241689011455), (14, 0.0032070360612124205), (15, 0.005937651265412569), (16, 0.03344918042421341), (17, 0.0022189170122146606), (18, 0.002801966853439808), (19, 0.0043152449652552605), (20, 0.005762515589594841), (21, 0.0022842418402433395), (22, 0.0030168513767421246), (23, 0.003275139955803752), (24, 0.007294515147805214), (25, 0.021855903789401054), (26, 0.015491226688027382), (27, 0.20415866374969482)], Temporal MSE: [(0, 5.599997166427784e-05), (1, 4.9737536755856127e-05), (2, 6.904584006406367e-05), (3, 5.931303894612938e-05), (4, 5.174484977032989e-05), (5, 6.353775097522885e-05), (6, 2.9795650334563106e-05), (7, 4.441284545464441e-05), (8, 3.3519183489261195e-05), (9, 4.234751031617634e-05), (10, 9.798754763323814e-05), (11, 7.439307228196412e-05), (12, 8.75341720529832e-05), (13, 0.0002620623854454607), (14, 0.0002502530114725232), (15, 0.0003142347268294543), (16, 0.00019462518685031682), (17, 0.0005787467816844583), (18, 0.00027964229229837656), (19, 0.0004357095167506486), (20, 0.0009123036288656294), (21, 0.0004022458742838353), (22, 0.0002989503263961524), (23, 0.00031766871688887477), (24, 0.0008045525173656642), (25, 0.0015447462210431695), (26, 0.00019512948347255588), (27, 0.00045849388698115945)]
+
+  5%|▍         | 7/150 [02:10<27:07, 11.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 852, Spatial MSE: [(0, 5.256838448985945e-07), (1, 3.3611229355301475e-06), (2, 6.112511618994176e-05), (3, 1.4769062545383349e-05), (4, 3.7521419926633826e-06), (5, 4.13584439229453e-06), (6, 1.5527266441495158e-05), (7, 1.6038171452237293e-05), (8, 2.3531229089712724e-05), (9, 5.409176446846686e-05), (10, 8.35675309645012e-05), (11, 0.00010823611955856904), (12, 0.00011797044862760231), (13, 0.00016613572370260954), (14, 0.0002855683851521462), (15, 0.0005138962296769023), (16, 0.0008511985652148724), (17, 0.00023430405417457223), (18, 0.0002605595509521663), (19, 0.0002610324590932578), (20, 0.00026173488004133105), (21, 0.00019587847054935992), (22, 0.00024910480715334415), (23, 0.0002794300380628556), (24, 0.000631504924967885), (25, 0.0007722933660261333), (26, 0.00033480196725577116), (27, 0.0011042325058951974)], Temporal MSE: [(0, 6.079049512663914e-07), (1, 4.0425362612950266e-07), (2, 6.078864771552617e-07), (3, 6.40968096377037e-07), (4, 6.198738446983043e-07), (5, 1.3349455230127205e-06), (6, 1.3450231790557154e-06), (7, 1.7584197848918848e-06), (8, 2.570995775386109e-06), (9, 4.180164069111925e-06), (10, 1.2463128769013565e-05), (11, 1.1446794815128669e-05), (12, 1.6306998077197932e-05), (13, 3.4926048101624474e-05), (14, 5.788505950476974e-05), (15, 4.28135426773224e-05), (16, 3.233152528991923e-05), (17, 4.871287092100829e-05), (18, 3.359249603818171e-05), (19, 4.526733391685411e-05), (20, 0.00010065191600006074), (21, 4.235910455463454e-05), (22, 2.7503867386258207e-05), (23, 3.781354826060124e-05), (24, 0.00011008561705239117), (25, 7.810504030203447e-05), (26, 1.4388278032129165e-05), (27, 2.877152837754693e-06)]
+
+  5%|▌         | 8/150 [02:18<24:21, 10.29s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 846, Spatial MSE: [(0, 2.8831757390435087e-06), (1, 2.0983246940886602e-05), (2, 0.0006068468210287392), (3, 0.00010785409540403634), (4, 2.4463990484946407e-05), (5, 1.254360449820524e-05), (6, 2.9043027097941376e-05), (7, 1.1530511073942762e-05), (8, 1.1634542715910356e-05), (9, 1.3234005564299878e-05), (10, 1.270489883609116e-05), (11, 2.3179718482424505e-05), (12, 2.7297408450976945e-05), (13, 0.0002970613131765276), (14, 0.0006944898632355034), (15, 0.00029956235084682703), (16, 0.008375185541808605), (17, 7.335266127483919e-05), (18, 0.00026153825456276536), (19, 0.00047220283886417747), (20, 0.00013646057050209492), (21, 0.00036246285890229046), (22, 0.0005340426578186452), (23, 0.00025892359553836286), (24, 0.00052662193775177), (25, 0.001233777147717774), (26, 0.0013082902878522873), (27, 0.0161211546510458)], Temporal MSE: [(0, 3.978522272518603e-06), (1, 2.6208074359601596e-06), (2, 3.9606288737559225e-06), (3, 3.738896339200437e-06), (4, 3.3336793876515003e-06), (5, 3.969330009567784e-06), (6, 1.878708417279995e-06), (7, 1.9496972072374774e-06), (8, 1.6194027239180286e-06), (9, 2.2069018541515106e-06), (10, 3.3327526125503937e-06), (11, 6.809673777752323e-06), (12, 9.187437171931379e-06), (13, 3.4288063034182414e-05), (14, 1.1444681149441749e-05), (15, 1.4778710465179756e-05), (16, 1.3784939255856443e-05), (17, 1.889284249045886e-05), (18, 1.1126629942737054e-05), (19, 1.1738857210730202e-05), (20, 1.729631912894547e-05), (21, 2.189274709962774e-05), (22, 6.687499990221113e-05), (23, 9.288860746892169e-05), (24, 0.00011518139945110306), (25, 4.810235623153858e-05), (26, 2.7422787752584554e-05), (27, 8.596926818427164e-06)]
+
+  6%|▌         | 9/150 [02:26<22:04,  9.39s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 840, Spatial MSE: [(0, 6.074689281376777e-06), (1, 0.00013827046495862305), (2, 0.0024189308751374483), (3, 0.00021652538271155208), (4, 0.00010291900980519131), (5, 7.007435488048941e-05), (6, 5.526623863261193e-05), (7, 9.497122664470226e-05), (8, 6.84036931488663e-05), (9, 7.454797014361247e-05), (10, 9.64935461524874e-05), (11, 0.00010058916814159602), (12, 0.0001616141089471057), (13, 0.0003642973897513002), (14, 0.0010569667210802436), (15, 0.0013249616604298353), (16, 0.014441112987697124), (17, 0.00044684286694973707), (18, 0.0005254463176243007), (19, 0.0012226346880197525), (20, 0.0008564768941141665), (21, 0.0007202906999737024), (22, 0.0011359304189682007), (23, 0.0009662972297519445), (24, 0.0023482509423047304), (25, 0.0047179837711155415), (26, 0.0040634977631270885), (27, 0.026582028716802597)], Temporal MSE: [(0, 7.299372100533219e-06), (1, 1.3919148841523565e-05), (2, 1.3985430996399373e-05), (3, 1.1783837180701084e-05), (4, 1.3105916877975687e-05), (5, 1.5523719412158243e-05), (6, 9.071068234334234e-06), (7, 1.5654064554837532e-05), (8, 7.910033673397265e-06), (9, 1.1480526154628024e-05), (10, 4.060500214109197e-05), (11, 3.168200055370107e-05), (12, 2.242515438410919e-05), (13, 7.175574137363583e-05), (14, 5.3523228416452184e-05), (15, 8.529274055035785e-05), (16, 4.775448906002566e-05), (17, 0.00013178089284338057), (18, 6.155925075290725e-05), (19, 9.726969437906519e-05), (20, 0.00024796530487947166), (21, 9.382556891068816e-05), (22, 0.00019044865621253848), (23, 0.00020384351955726743), (24, 0.0003487001813482493), (25, 0.00025142438244074583), (26, 8.413477189606056e-05), (27, 9.162379137706012e-05)]
+
+  7%|▋         | 10/150 [02:33<20:28,  8.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 834, Spatial MSE: [(0, 1.0664638466550969e-05), (1, 7.546638516942039e-05), (2, 0.002322417451068759), (3, 0.00035427403054200113), (4, 8.080556290224195e-05), (5, 4.394069765112363e-05), (6, 6.176254828460515e-05), (7, 4.672396607929841e-05), (8, 3.93242844438646e-05), (9, 4.351795723778196e-05), (10, 4.024192821816541e-05), (11, 6.578506872756407e-05), (12, 7.521463703596964e-05), (13, 0.0007976294727995992), (14, 0.0018226891988888383), (15, 0.0007990035228431225), (16, 0.03200085461139679), (17, 0.00026005684048868716), (18, 0.0009303997503593564), (19, 0.0017898129299283028), (20, 0.0005176261765882373), (21, 0.0013633405324071646), (22, 0.0019248205935582519), (23, 0.0009731448953971267), (24, 0.00188416987657547), (25, 0.004188563674688339), (26, 0.006033243611454964), (27, 0.06912803649902344)], Temporal MSE: [(0, 1.4483411177934613e-05), (1, 9.342842531623319e-06), (2, 1.3864162610843778e-05), (3, 1.3210049473855179e-05), (4, 1.0439432116982061e-05), (5, 1.2084196896466892e-05), (6, 6.027060408086982e-06), (7, 5.8677292145148385e-06), (8, 5.199018232815433e-06), (9, 6.778609986213269e-06), (10, 1.0231574378849473e-05), (11, 1.946369775396306e-05), (12, 2.2731775970896706e-05), (13, 9.546727233100682e-05), (14, 3.326709702378139e-05), (15, 4.710741268354468e-05), (16, 4.829145109397359e-05), (17, 6.605103408219293e-05), (18, 4.048214395879768e-05), (19, 4.2940591811202466e-05), (20, 5.531097485800274e-05), (21, 7.176119834184647e-05), (22, 0.00023798886104486883), (23, 0.000306718866340816), (24, 0.00040741966222412884), (25, 0.00019696344679687172), (26, 0.00010913537698797882), (27, 3.134663711534813e-05)]
+
+  7%|▋         | 11/150 [02:41<19:29,  8.41s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 828, Spatial MSE: [(0, 1.160472402261803e-05), (1, 0.0002743518562056124), (2, 0.003676046384498477), (3, 0.00031221937388181686), (4, 0.00018130772514268756), (5, 0.00019230473844800144), (6, 0.00031991986907087266), (7, 0.0006433613016270101), (8, 0.0007041501812636852), (9, 0.0013992342865094543), (10, 0.0019250467885285616), (11, 0.002268790965899825), (12, 0.002628512214869261), (13, 0.004453469533473253), (14, 0.008705360814929008), (15, 0.00999602023512125), (16, 0.05668875575065613), (17, 0.004500762093812227), (18, 0.0053613740019500256), (19, 0.0067930580116808414), (20, 0.007514006923884153), (21, 0.004367128014564514), (22, 0.006114020943641663), (23, 0.0054654949344694614), (24, 0.012882447801530361), (25, 0.03830965608358383), (26, 0.014958074316382408), (27, 0.042612332850694656)], Temporal MSE: [(0, 1.616871486476157e-05), (1, 2.53343259828398e-05), (2, 3.4217973734484985e-05), (3, 2.7979240257991478e-05), (4, 2.82354394585127e-05), (5, 6.080131061025895e-05), (6, 4.3204097892157733e-05), (7, 7.56671724957414e-05), (8, 8.679202437633649e-05), (9, 0.00013414706336334348), (10, 0.0003752933698706329), (11, 0.0003372906649019569), (12, 0.00042854531784541905), (13, 0.0010139963123947382), (14, 0.001219226629473269), (15, 0.0009732123580761254), (16, 0.000675373652484268), (17, 0.0012110433308407664), (18, 0.0007896526367403567), (19, 0.0010176391806453466), (20, 0.0023585602175444365), (21, 0.001392959151417017), (22, 0.0008479153038933873), (23, 0.001116566825658083), (24, 0.0028667994774878025), (25, 0.005562976934015751), (26, 0.0007114137406460941), (27, 0.0003149755939375609)]
+
+  8%|▊         | 12/150 [02:48<18:44,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 822, Spatial MSE: [(0, 1.4243230907595716e-05), (1, 0.00010601908434182405), (2, 0.0029261913150548935), (3, 0.0005179193685762584), (4, 0.00011231125972699374), (5, 5.803917520097457e-05), (6, 0.00010914006998063996), (7, 5.4470696341013536e-05), (8, 5.06780379510019e-05), (9, 6.0150337958475575e-05), (10, 5.951177445240319e-05), (11, 0.00010838124580914155), (12, 0.00015356155927293003), (13, 0.0022991467267274857), (14, 0.005078435409814119), (15, 0.0017296081641688943), (16, 0.07039675861597061), (17, 0.00043655955232679844), (18, 0.0016579177463427186), (19, 0.003927035257220268), (20, 0.0011226424248889089), (21, 0.0024157122243195772), (22, 0.003239853773266077), (23, 0.001548010390251875), (24, 0.00290055014193058), (25, 0.0067818788811564445), (26, 0.008603171445429325), (27, 0.1090104877948761)], Temporal MSE: [(0, 1.869260086095892e-05), (1, 1.2457666343834717e-05), (2, 1.6705213056411594e-05), (3, 1.646824603085406e-05), (4, 1.4113691577222198e-05), (5, 1.6855990907060914e-05), (6, 7.90911781223258e-06), (7, 8.726163287064992e-06), (8, 7.136031854315661e-06), (9, 9.562704690324608e-06), (10, 1.3974888133816421e-05), (11, 4.198900569463149e-05), (12, 5.118480839882977e-05), (13, 0.00023424353275913745), (14, 7.160776294767857e-05), (15, 9.673093154560775e-05), (16, 8.864630945026875e-05), (17, 0.00010620531247695908), (18, 9.533758566249162e-05), (19, 9.10677044885233e-05), (20, 0.00012237245391588658), (21, 0.0001328917423961684), (22, 0.0003974914434365928), (23, 0.0005760600906796753), (24, 0.0007332191453315318), (25, 0.0003314526111353189), (26, 0.00018326296412851661), (27, 5.009636879549362e-05)]
+
+  9%|▊         | 13/150 [02:55<17:53,  7.84s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 816, Spatial MSE: [(0, 1.4164957974571735e-05), (1, 0.0002792988088913262), (2, 0.004249297548085451), (3, 0.00034514337312430143), (4, 0.0001930917496792972), (5, 0.00014184671454131603), (6, 0.00010935278987744823), (7, 0.00020515217329375446), (8, 0.00014586615725420415), (9, 0.00017982468125410378), (10, 0.00025209615705534816), (11, 0.00026436435291543603), (12, 0.0004653184732887894), (13, 0.0027872079517692327), (14, 0.006132718175649643), (15, 0.0028875130228698254), (16, 0.07387316972017288), (17, 0.0011039396049454808), (18, 0.0020732504781335592), (19, 0.004731315653771162), (20, 0.001946930424310267), (21, 0.00289517710916698), (22, 0.004035287536680698), (23, 0.002758031478151679), (24, 0.0074881636537611485), (25, 0.014769373461604118), (26, 0.010724629275500774), (27, 0.05903772637248039)], Temporal MSE: [(0, 1.6454596334369853e-05), (1, 2.9255446861498058e-05), (2, 2.5639536033850163e-05), (3, 2.0865967599092983e-05), (4, 2.7523728931555524e-05), (5, 3.243738683522679e-05), (6, 2.38460906984983e-05), (7, 2.970851528516505e-05), (8, 2.1896954422118142e-05), (9, 3.518806988722645e-05), (10, 8.968064503278583e-05), (11, 9.662093361839652e-05), (12, 9.103028423851356e-05), (13, 0.00033385344431735575), (14, 0.00015209005505312234), (15, 0.0002587472554296255), (16, 0.0001696707186056301), (17, 0.00033933637314476073), (18, 0.00022822046594228595), (19, 0.0003088825033046305), (20, 0.0008310592966154218), (21, 0.00035715202102437615), (22, 0.0005736630992032588), (23, 0.0008149325149133801), (24, 0.0011931746266782284), (25, 0.0012385289883241057), (26, 0.0002873616467695683), (27, 0.0003067083016503602)]
+
+  9%|▉         | 14/150 [03:05<18:52,  8.32s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 810, Spatial MSE: [(0, 1.9166867787134834e-05), (1, 0.00013279322593007237), (2, 0.0033592593390494585), (3, 0.000618529855273664), (4, 0.00013020339247304946), (5, 6.977772136451676e-05), (6, 0.00014078678213991225), (7, 7.048480620142072e-05), (8, 6.202235090313479e-05), (9, 7.45803554309532e-05), (10, 7.595621718792245e-05), (11, 0.00014888582518324256), (12, 0.00024391677288804203), (13, 0.004076800309121609), (14, 0.007561549078673124), (15, 0.002377535682171583), (16, 0.08318544179201126), (17, 0.00047685415484011173), (18, 0.0014875233173370361), (19, 0.004925737157464027), (20, 0.0015298642683774233), (21, 0.0021209963597357273), (22, 0.0036382991820573807), (23, 0.0018460240680724382), (24, 0.0033650044351816177), (25, 0.008272713050246239), (26, 0.00970461405813694), (27, 0.14211708307266235)], Temporal MSE: [(0, 2.5575889594620094e-05), (1, 1.5566631191177294e-05), (2, 2.0766294255736284e-05), (3, 1.9308561604702845e-05), (4, 1.6916626918828115e-05), (5, 2.041166226263158e-05), (6, 9.919985132000875e-06), (7, 1.0305296200385783e-05), (8, 8.881294888851698e-06), (9, 1.1771690878958907e-05), (10, 1.7997870600083843e-05), (11, 9.365686128148809e-05), (12, 9.471045632380992e-05), (13, 0.00036814139457419515), (14, 9.659361967351288e-05), (15, 0.00011930479377042502), (16, 0.00010699126141844317), (17, 0.00011850116425193846), (18, 0.00010603963164612651), (19, 9.465248149354011e-05), (20, 0.0001087663258658722), (21, 0.00015673559391871095), (22, 0.0004491304571274668), (23, 0.0007238629623316228), (24, 0.0009231321746483445), (25, 0.0003849654458463192), (26, 0.00023152842186391354), (27, 6.59168217680417e-05)]
+
+ 10%|█         | 15/150 [03:13<18:53,  8.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 804, Spatial MSE: [(0, 1.4045413990970701e-05), (1, 0.0003671205777209252), (2, 0.004138078540563583), (3, 0.0003342367126606405), (4, 0.00020399819186422974), (5, 0.000253488018643111), (6, 0.0004945097025483847), (7, 0.0009069867664948106), (8, 0.0010535740293562412), (9, 0.0016631437465548515), (10, 0.0023947583977133036), (11, 0.0026715153362601995), (12, 0.002814966021105647), (13, 0.007404746487736702), (14, 0.015842663124203682), (15, 0.011517760343849659), (16, 0.130922332406044), (17, 0.005054767709225416), (18, 0.005190087482333183), (19, 0.011766648851335049), (20, 0.012707401067018509), (21, 0.005360471084713936), (22, 0.007280008401721716), (23, 0.005835985764861107), (24, 0.01322583481669426), (25, 0.06330669671297073), (26, 0.027986248955130577), (27, 0.07358258962631226)], Temporal MSE: [(0, 1.8146454749512486e-05), (1, 3.697124702739529e-05), (2, 3.919126174878329e-05), (3, 3.2592211937299e-05), (4, 3.484779881546274e-05), (5, 7.316285336855799e-05), (6, 6.517836300190538e-05), (7, 9.213681914843619e-05), (8, 0.00013144385593477637), (9, 0.00017358978220727295), (10, 0.0003719579544849694), (11, 0.0003918302827514708), (12, 0.0004592999175656587), (13, 0.0010060104541480541), (14, 0.0010079446947202086), (15, 0.0008506394806317985), (16, 0.000672923750244081), (17, 0.0010978695936501026), (18, 0.000870697433128953), (19, 0.0009522892069071531), (20, 0.001872192951850593), (21, 0.001399246510118246), (22, 0.0010799044976010919), (23, 0.0014668956864625216), (24, 0.0035176612436771393), (25, 0.01011371985077858), (26, 0.001090681180357933), (27, 0.0003527667431626469)]
+
+ 11%|█         | 16/150 [03:21<18:21,  8.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 798, Spatial MSE: [(0, 1.7958520402316935e-05), (1, 0.00013315324031282216), (2, 0.003068957244977355), (3, 0.0005908770253881812), (4, 0.00011346556129865348), (5, 6.283532275119796e-05), (6, 0.00012670503929257393), (7, 6.147204840090126e-05), (8, 5.55925362277776e-05), (9, 6.961470353417099e-05), (10, 7.260494749061763e-05), (11, 0.00013233469508122653), (12, 0.0002256843727082014), (13, 0.00415537366643548), (14, 0.008943851105868816), (15, 0.0031688124872744083), (16, 0.12059769034385681), (17, 0.0005649419035762548), (18, 0.0019916752353310585), (19, 0.006966042332351208), (20, 0.0021623698994517326), (21, 0.003015117719769478), (22, 0.004922401160001755), (23, 0.0022467097733169794), (24, 0.003780841128900647), (25, 0.009210878983139992), (26, 0.010401586070656776), (27, 0.14127680659294128)], Temporal MSE: [(0, 2.3167805920820683e-05), (1, 1.3637790289067198e-05), (2, 1.7567288523423485e-05), (3, 1.6676609448040836e-05), (4, 1.4091455341258552e-05), (5, 1.8589540559332818e-05), (6, 8.999562851386145e-06), (7, 9.080683412321378e-06), (8, 7.816786819603294e-06), (9, 1.1120890121674165e-05), (10, 1.7140182535513304e-05), (11, 7.518124039052054e-05), (12, 8.902978152036667e-05), (13, 0.0003998781321570277), (14, 0.00012210464046802372), (15, 0.00015376362716779113), (16, 0.00013134547043591738), (17, 0.00013461170601658523), (18, 0.00018464642926119268), (19, 0.00014155021926853806), (20, 0.0001965846458915621), (21, 0.00021071785886306316), (22, 0.0005884345155209303), (23, 0.0009628909174352884), (24, 0.0011478461092337966), (25, 0.0005064986762590706), (26, 0.0002783288946375251), (27, 7.834332791389897e-05)]
+
+ 11%|█▏        | 17/150 [03:28<17:35,  7.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 792, Spatial MSE: [(0, 1.694710590527393e-05), (1, 0.00032875340548343956), (2, 0.004328201990574598), (3, 0.0003570504195522517), (4, 0.00015815584629308432), (5, 0.00014950026525184512), (6, 0.00019406256615184247), (7, 0.0002154925896320492), (8, 0.0001715764810796827), (9, 0.00021595836733467877), (10, 0.00029881965019740164), (11, 0.0003279964148532599), (12, 0.0005260900943540037), (13, 0.004522734321653843), (14, 0.009754654951393604), (15, 0.004678281489759684), (16, 0.12042772024869919), (17, 0.0011391955194994807), (18, 0.002379506127908826), (19, 0.007671114057302475), (20, 0.004306596238166094), (21, 0.003020238596946001), (22, 0.00512625090777874), (23, 0.003138978499919176), (24, 0.0074027287773787975), (25, 0.01867181807756424), (26, 0.011977843940258026), (27, 0.07919000089168549)], Temporal MSE: [(0, 1.8947486751130782e-05), (1, 2.9606906537082978e-05), (2, 2.5192081011482514e-05), (3, 2.1592271878034808e-05), (4, 2.7835021683131345e-05), (5, 3.347202073200606e-05), (6, 2.8726322852890007e-05), (7, 2.9288938094396144e-05), (8, 2.6104347853106447e-05), (9, 3.610167550505139e-05), (10, 7.820563041605055e-05), (11, 0.00013674031652044505), (12, 0.00014088931493461132), (13, 0.0004626903682947159), (14, 0.0001917880872497335), (15, 0.0002893123892135918), (16, 0.00021356894285418093), (17, 0.00040265373536385596), (18, 0.00033145889756269753), (19, 0.00041639857226982713), (20, 0.0010437089949846268), (21, 0.0005076766246929765), (22, 0.00067228201078251), (23, 0.001167279202491045), (24, 0.0016538449563086033), (25, 0.0023488057777285576), (26, 0.00036935918615199625), (27, 0.000330164359183982)]
+
+ 12%|█▏        | 18/150 [03:35<16:54,  7.69s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 786, Spatial MSE: [(0, 2.26471292990027e-05), (1, 0.0001696479885140434), (2, 0.0034312589559704065), (3, 0.0007317958516068757), (4, 0.0001298719726037234), (5, 7.275721873156726e-05), (6, 0.00017216110427398235), (7, 6.958708399906754e-05), (8, 6.619330815738067e-05), (9, 8.44732130644843e-05), (10, 9.174107253784314e-05), (11, 0.00017625678447075188), (12, 0.00032827502582222223), (13, 0.006325327325612307), (14, 0.010615160688757896), (15, 0.0031647311989217997), (16, 0.12142333388328552), (17, 0.0006629836861975491), (18, 0.0019231257028877735), (19, 0.007481152657419443), (20, 0.0024861448910087347), (21, 0.0027977919671684504), (22, 0.00513622909784317), (23, 0.002329619601368904), (24, 0.0038805047515779734), (25, 0.009286941960453987), (26, 0.010698641650378704), (27, 0.1650926023721695)], Temporal MSE: [(0, 2.8972237487323582e-05), (1, 1.635180160519667e-05), (2, 2.0411860532476567e-05), (3, 1.896554749691859e-05), (4, 1.6071866411948577e-05), (5, 2.248268538096454e-05), (6, 1.08498443296412e-05), (7, 1.1257649020990357e-05), (8, 9.398117072123569e-06), (9, 1.3318543096829671e-05), (10, 2.192133615608327e-05), (11, 0.00012770835019182414), (12, 0.00013727258192375302), (13, 0.0004815617867279798), (14, 0.0001320877781836316), (15, 0.00018076693231705576), (16, 0.00013719050912186503), (17, 0.00012852228246629238), (18, 0.00019181861716788262), (19, 0.00015034805983304977), (20, 0.00019521279318723828), (21, 0.00022214818454813212), (22, 0.0006156411836855114), (23, 0.0010103491367772222), (24, 0.0011798577615991235), (25, 0.0005767256370745599), (26, 0.0002959031262435019), (27, 9.719237277749926e-05)]
+
+ 13%|█▎        | 19/150 [03:43<16:38,  7.62s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 780, Spatial MSE: [(0, 2.1987125364830717e-05), (1, 0.0005554055678658187), (2, 0.006524677854031324), (3, 0.0005524722510017455), (4, 0.00023693012190051377), (5, 0.00035642163129523396), (6, 0.0007717614062130451), (7, 0.001064418232999742), (8, 0.0011370951542630792), (9, 0.001775776850990951), (10, 0.002215869491919875), (11, 0.0022417122963815928), (12, 0.0030151696410030127), (13, 0.010468430817127228), (14, 0.019225675612688065), (15, 0.012399760074913502), (16, 0.1661858707666397), (17, 0.006294352002441883), (18, 0.007056113798171282), (19, 0.014826277270913124), (20, 0.01706664450466633), (21, 0.006202444434165955), (22, 0.01025459636002779), (23, 0.007884960621595383), (24, 0.015841340646147728), (25, 0.0970923975110054), (26, 0.04268631339073181), (27, 0.12696047127246857)], Temporal MSE: [(0, 3.0299974241643213e-05), (1, 5.1377443014644086e-05), (2, 5.395343396230601e-05), (3, 4.5143657189328223e-05), (4, 5.332312866812572e-05), (5, 9.292161121265963e-05), (6, 0.00010762692545540631), (7, 0.00013237923849374056), (8, 0.00014717798330821097), (9, 0.00021361230756156147), (10, 0.0003917992580682039), (11, 0.00044547911966219544), (12, 0.0005326397949829698), (13, 0.001219617435708642), (14, 0.0012512627290561795), (15, 0.0009896716801449656), (16, 0.0009317767107859254), (17, 0.001475549302995205), (18, 0.001277624978683889), (19, 0.0014336103340610862), (20, 0.002383341547101736), (21, 0.002093580085784197), (22, 0.0015700001968070865), (23, 0.002297037746757269), (24, 0.005935289431363344), (25, 0.020304664969444275), (26, 0.0020695661660283804), (27, 0.000356672186171636)]
+
+ 13%|█▎        | 20/150 [03:51<16:47,  7.75s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 774, Spatial MSE: [(0, 3.334029679535888e-05), (1, 0.00024380566901527345), (2, 0.004292767029255629), (3, 0.0009979605674743652), (4, 0.00017184777243528515), (5, 9.950403909897432e-05), (6, 0.00024499796563759446), (7, 8.827281271805987e-05), (8, 9.012361988425255e-05), (9, 0.00011731456470442936), (10, 0.00012776243966072798), (11, 0.00023008709831628948), (12, 0.00044581215479411185), (13, 0.009295353665947914), (14, 0.014615008607506752), (15, 0.0044907270930707455), (16, 0.1561070680618286), (17, 0.0006633601733483374), (18, 0.0018223992083221674), (19, 0.00980408862233162), (20, 0.0036186412908136845), (21, 0.0027877655811607838), (22, 0.005854269023984671), (23, 0.003144756658002734), (24, 0.00532641913741827), (25, 0.01258943323045969), (26, 0.017099667340517044), (27, 0.2437625527381897)], Temporal MSE: [(0, 4.183524288237095e-05), (1, 2.2237571101868525e-05), (2, 2.7169518943992443e-05), (3, 2.561431028880179e-05), (4, 2.1563015252468176e-05), (5, 3.209973147022538e-05), (6, 1.5342306141974404e-05), (7, 1.5525598428212106e-05), (8, 1.327433255937649e-05), (9, 1.916675319080241e-05), (10, 3.065332930418663e-05), (11, 0.00018430856289342046), (12, 0.00020349895930849016), (13, 0.000660509685985744), (14, 0.00018682605877984315), (15, 0.00021712582383770496), (16, 0.0001699379354249686), (17, 0.00014963504509069026), (18, 0.0002505273441784084), (19, 0.00018495052063371986), (20, 0.000201630566152744), (21, 0.00027127700741402805), (22, 0.0007056005415506661), (23, 0.001311769476160407), (24, 0.0016882187919691205), (25, 0.0009649876155890524), (26, 0.0004833903512917459), (27, 0.00015291395538952202)]
+
+ 14%|█▍        | 21/150 [04:00<17:16,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 768, Spatial MSE: [(0, 3.0719165806658566e-05), (1, 0.0006121469778008759), (2, 0.007384935859590769), (3, 0.0007747559575363994), (4, 0.00021865764574613422), (5, 0.00025026267394423485), (6, 0.00046606405521743), (7, 0.0003848541236948222), (8, 0.0004241114074829966), (9, 0.0005938961403444409), (10, 0.0007292913505807519), (11, 0.0008791242144070566), (12, 0.0011062815319746733), (13, 0.010653642006218433), (14, 0.01804926246404648), (15, 0.007106546312570572), (16, 0.19059240818023682), (17, 0.0014424455584958196), (18, 0.0028879521414637566), (19, 0.013945195823907852), (20, 0.007878263480961323), (21, 0.004504004027694464), (22, 0.006931780371814966), (23, 0.004093051888048649), (24, 0.009337671101093292), (25, 0.028894701972603798), (26, 0.017985766753554344), (27, 0.15298834443092346)], Temporal MSE: [(0, 3.4550350392237306e-05), (1, 4.925631947116926e-05), (2, 3.98875490645878e-05), (3, 3.649002974270843e-05), (4, 4.114039256819524e-05), (5, 5.6274784583365545e-05), (6, 4.8692836571717635e-05), (7, 4.621805783244781e-05), (8, 5.1233728299848735e-05), (9, 6.152230344014242e-05), (10, 0.00011764336522901431), (11, 0.0003052633546758443), (12, 0.0003201414074283093), (13, 0.0007912012515589595), (14, 0.00032427976839244366), (15, 0.0003890999942086637), (16, 0.00031392849632538855), (17, 0.0005150521174073219), (18, 0.0005078011890873313), (19, 0.0006856156978756189), (20, 0.0013872853014618158), (21, 0.0009047297644428909), (22, 0.0009444312308914959), (23, 0.0017998215043917298), (24, 0.0027355551719665527), (25, 0.006467138417065144), (26, 0.0005907793529331684), (27, 0.00040389859350398183)]
+
+ 15%|█▍        | 22/150 [04:08<17:09,  8.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 762, Spatial MSE: [(0, 4.191254993202165e-05), (1, 0.00031454823329113424), (2, 0.004445540253072977), (3, 0.0011436959030106664), (4, 0.0001809529640013352), (5, 0.00010615576320560649), (6, 0.00032751463004387915), (7, 9.21821192605421e-05), (8, 9.873673116089776e-05), (9, 0.00012635708844754845), (10, 0.0001490709837526083), (11, 0.0002942605933640152), (12, 0.0006322520202957094), (13, 0.013618103228509426), (14, 0.018467918038368225), (15, 0.005068397149443626), (16, 0.1626240313053131), (17, 0.0008210165542550385), (18, 0.0018526390194892883), (19, 0.011171186342835426), (20, 0.004546306561678648), (21, 0.002826359821483493), (22, 0.0056801349855959415), (23, 0.0031642597168684006), (24, 0.005395473912358284), (25, 0.012073123827576637), (26, 0.018155913800001144), (27, 0.28759297728538513)], Temporal MSE: [(0, 5.009321466786787e-05), (1, 2.4835138901835307e-05), (2, 2.8821179512306117e-05), (3, 2.663185659912415e-05), (4, 2.1899524654145353e-05), (5, 3.570753688109107e-05), (6, 1.804206294764299e-05), (7, 1.7129706975538284e-05), (8, 1.4376537365023978e-05), (9, 2.158538882213179e-05), (10, 3.528086745063774e-05), (11, 0.0003085573553107679), (12, 0.0003018830029759556), (13, 0.0007654897053726017), (14, 0.00020896090427413583), (15, 0.00026841062936000526), (16, 0.000178700836841017), (17, 0.00015638871991541237), (18, 0.0002914211945608258), (19, 0.00021594158897642046), (20, 0.00023853633319959044), (21, 0.00029221767908893526), (22, 0.0007122529204934835), (23, 0.0013028092216700315), (24, 0.0017281530890613794), (25, 0.0012159812031313777), (26, 0.000526359595824033), (27, 0.00021178550377953798)]
+
+ 15%|█▌        | 23/150 [04:16<17:18,  8.18s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 756, Spatial MSE: [(0, 3.859918797388673e-05), (1, 0.0009363397839479148), (2, 0.009618236683309078), (3, 0.0008091551135294139), (4, 0.00033051116042770445), (5, 0.0004946296685375273), (6, 0.001118831685744226), (7, 0.0014969350304454565), (8, 0.0016960357315838337), (9, 0.002172111300751567), (10, 0.0024435061495751143), (11, 0.0027187687810510397), (12, 0.003241186961531639), (13, 0.016155051067471504), (14, 0.023354044184088707), (15, 0.011492655612528324), (16, 0.18770213425159454), (17, 0.006382557097822428), (18, 0.006028654985129833), (19, 0.016392115503549576), (20, 0.017715007066726685), (21, 0.006505685858428478), (22, 0.008914435282349586), (23, 0.00678727263584733), (24, 0.01585620827972889), (25, 0.09773015975952148), (26, 0.057929158210754395), (27, 0.19866910576820374)], Temporal MSE: [(0, 4.526830525719561e-05), (1, 7.761372398817912e-05), (2, 7.077685586409643e-05), (3, 6.487643258878961e-05), (4, 8.056139631662518e-05), (5, 0.00016920659982133657), (6, 0.00018601541523821652), (7, 0.00022128499404061586), (8, 0.00021675352763850242), (9, 0.0003038848517462611), (10, 0.0005044018034823239), (11, 0.0006563419010490179), (12, 0.0006924266926944256), (13, 0.001330555067397654), (14, 0.0011945253936573863), (15, 0.001165666151791811), (16, 0.0010559851070865989), (17, 0.0015921128215268254), (18, 0.0013846629299223423), (19, 0.0018299707444384694), (20, 0.0025932639837265015), (21, 0.0029715029522776604), (22, 0.0016291976207867265), (23, 0.002446381375193596), (24, 0.006278074346482754), (25, 0.029572926461696625), (26, 0.0027434751391410828), (27, 0.0004253289080224931)]
+
+ 16%|█▌        | 24/150 [04:24<17:04,  8.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 750, Spatial MSE: [(0, 4.8395610065199435e-05), (1, 0.00038430659333243966), (2, 0.004182504490017891), (3, 0.0012579199392348528), (4, 0.00018581350741442293), (5, 0.00011123248987132683), (6, 0.0004505842225626111), (7, 9.767136361915618e-05), (8, 0.00010638796084094793), (9, 0.0001278812123928219), (10, 0.00015339393576141447), (11, 0.00027671680436469615), (12, 0.0005537447868846357), (13, 0.012693638913333416), (14, 0.01767294853925705), (15, 0.005712416023015976), (16, 0.1818702518939972), (17, 0.0007038613548502326), (18, 0.001607924117706716), (19, 0.011686551384627819), (20, 0.0047418526373803616), (21, 0.0024760859087109566), (22, 0.005610388703644276), (23, 0.0031963770743459463), (24, 0.005225805100053549), (25, 0.011730779893696308), (26, 0.018283510580658913), (27, 0.298194944858551)], Temporal MSE: [(0, 5.608781793853268e-05), (1, 2.52935951721156e-05), (2, 2.8760421628248878e-05), (3, 2.6535079086897895e-05), (4, 2.206085446232464e-05), (5, 3.9003745769150555e-05), (6, 2.1935964468866587e-05), (7, 1.9045281078433618e-05), (8, 1.566049104440026e-05), (9, 2.5022671252372675e-05), (10, 3.692339305416681e-05), (11, 0.0002742293872870505), (12, 0.0002597840502858162), (13, 0.0006969352252781391), (14, 0.00023025856353342533), (15, 0.00025735952658578753), (16, 0.00018874782836064696), (17, 0.00014992129581514746), (18, 0.00032421547803096473), (19, 0.00023390870774164796), (20, 0.00022273752256296575), (21, 0.000310149131109938), (22, 0.0007097367779351771), (23, 0.0014483790146186948), (24, 0.001874107401818037), (25, 0.0017161770956590772), (26, 0.000606807938311249), (27, 0.0002607890055514872)]
+
+ 17%|█▋        | 25/150 [04:32<16:28,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 744, Spatial MSE: [(0, 3.7866113416384906e-05), (1, 0.0005997339612804353), (2, 0.005965390242636204), (3, 0.0005882360273972154), (4, 0.00016801153833512217), (5, 0.00017086256411857903), (6, 0.0004144305712543428), (7, 0.00024115016276482493), (8, 0.0002788337878882885), (9, 0.0003760125837288797), (10, 0.0004811605322174728), (11, 0.0006744834245182574), (12, 0.001038254238665104), (13, 0.013826167210936546), (14, 0.020029541105031967), (15, 0.007861255668103695), (16, 0.21837471425533295), (17, 0.0011905741412192583), (18, 0.0030460089910775423), (19, 0.015341256745159626), (20, 0.008973171934485435), (21, 0.005013476125895977), (22, 0.0066855535842478275), (23, 0.0033622118644416332), (24, 0.008691125549376011), (25, 0.024230562150478363), (26, 0.01561719924211502), (27, 0.17342765629291534)], Temporal MSE: [(0, 3.786153320106678e-05), (1, 4.132780304644257e-05), (2, 3.612469299696386e-05), (3, 3.112526974291541e-05), (4, 3.339959948789328e-05), (5, 4.9520582251716405e-05), (6, 4.1130177123704925e-05), (7, 3.632246807683259e-05), (8, 3.467755959718488e-05), (9, 4.492578955250792e-05), (10, 8.495774818584323e-05), (11, 0.0003817635297309607), (12, 0.0003607198013924062), (13, 0.0007751129451207817), (14, 0.0003198068297933787), (15, 0.00039722659857943654), (16, 0.00030663915094919503), (17, 0.00043237514910288155), (18, 0.0005737939500249922), (19, 0.0007063536322675645), (20, 0.0008633836987428367), (21, 0.0007829696987755597), (22, 0.0009613566799089313), (23, 0.001981033245101571), (24, 0.0032327971421182156), (25, 0.010165066458284855), (26, 0.000782302871812135), (27, 0.000356734381057322)]
+
+ 17%|█▋        | 26/150 [04:40<16:34,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 738, Spatial MSE: [(0, 3.8007852708688006e-05), (1, 0.0003091817779932171), (2, 0.002674351679161191), (3, 0.0008087157038971782), (4, 0.00012438050180207938), (5, 7.645828009117395e-05), (6, 0.0003497185534797609), (7, 6.977946759434417e-05), (8, 7.556530181318521e-05), (9, 8.927757880883291e-05), (10, 0.00012629557750187814), (11, 0.00021223282965365797), (12, 0.00043863418977707624), (13, 0.009405272081494331), (14, 0.012653043493628502), (15, 0.0037380780559033155), (16, 0.1345871239900589), (17, 0.0006742124096490443), (18, 0.0013959184288978577), (19, 0.009687782265245914), (20, 0.004426896572113037), (21, 0.00189037318341434), (22, 0.003905976191163063), (23, 0.002006252994760871), (24, 0.0035393666476011276), (25, 0.007496139034628868), (26, 0.011908357962965965), (27, 0.20641446113586426)], Temporal MSE: [(0, 4.3218020437052473e-05), (1, 1.8618431568029337e-05), (2, 2.0554742150125094e-05), (3, 1.870125015557278e-05), (4, 1.4496743460767902e-05), (5, 2.769784077827353e-05), (6, 1.7585765817784704e-05), (7, 1.3987682905280963e-05), (8, 1.1790823009505402e-05), (9, 1.8932143575511873e-05), (10, 2.927107925643213e-05), (11, 0.00022204422566574067), (12, 0.00020157641847617924), (13, 0.00045534715172834694), (14, 0.00016594929911661893), (15, 0.00022200378589332104), (16, 0.00012998704914934933), (17, 0.00011212719255127013), (18, 0.0002816140477079898), (19, 0.00020925480930600315), (20, 0.00020448291616048664), (21, 0.0002732020220719278), (22, 0.0005737471510656178), (23, 0.001001187483780086), (24, 0.0012128330999985337), (25, 0.0014388344716280699), (26, 0.00039887530147098005), (27, 0.00021707548876293004)]
+
+ 18%|█▊        | 27/150 [04:47<15:54,  7.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 732, Spatial MSE: [(0, 3.27785819536075e-05), (1, 0.000693125301040709), (2, 0.006081850733608007), (3, 0.0005603302270174026), (4, 0.0003041225136257708), (5, 0.0003644198877736926), (6, 0.000835872720927), (7, 0.0014139771228656173), (8, 0.0014845944242551923), (9, 0.0019502205541357398), (10, 0.0022627604193985462), (11, 0.003099407535046339), (12, 0.003132475772872567), (13, 0.012392939068377018), (14, 0.020809117704629898), (15, 0.015480141155421734), (16, 0.16975726187229156), (17, 0.012301081791520119), (18, 0.00887482613325119), (19, 0.020471451804041862), (20, 0.023936664685606956), (21, 0.019874844700098038), (22, 0.007902652956545353), (23, 0.006704286672174931), (24, 0.016923587769269943), (25, 0.047042299062013626), (26, 0.0303265992552042), (27, 0.1391458660364151)], Temporal MSE: [(0, 3.287277286290191e-05), (1, 5.026812868891284e-05), (2, 4.3458690925035626e-05), (3, 5.02196817251388e-05), (4, 7.235920202219859e-05), (5, 0.00019393989350646734), (6, 0.00027592695550993085), (7, 0.00021563003247138113), (8, 0.00019432867702562362), (9, 0.0003208062844350934), (10, 0.0007117827772162855), (11, 0.0008947007008828223), (12, 0.0009054585243575275), (13, 0.002261520130559802), (14, 0.0028573230374604464), (15, 0.00482142111286521), (16, 0.003689285134896636), (17, 0.005436904262751341), (18, 0.0038261667359620333), (19, 0.007959066890180111), (20, 0.014163329266011715), (21, 0.012498019263148308), (22, 0.0022108375560492277), (23, 0.0031488758977502584), (24, 0.009820635430514812), (25, 0.025300823152065277), (26, 0.0026776317972689867), (27, 0.00027738959761336446)]
+
+ 19%|█▊        | 28/150 [04:55<15:41,  7.72s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 726, Spatial MSE: [(0, 3.041105082957074e-05), (1, 0.00023860343208070844), (2, 0.001696956460364163), (3, 0.0005372629966586828), (4, 8.256299770437181e-05), (5, 5.1298749895067886e-05), (6, 0.0002844634582288563), (7, 5.014195266994648e-05), (8, 5.091686034575105e-05), (9, 5.7468507293378934e-05), (10, 7.622435805387795e-05), (11, 0.00011057007213821635), (12, 0.00023094101925380528), (13, 0.005383280571550131), (14, 0.007710553240031004), (15, 0.0026386920362710953), (16, 0.10415774583816528), (17, 0.00045057522947899997), (18, 0.0008938889368437231), (19, 0.007669459562748671), (20, 0.0038163536228239536), (21, 0.0012777714291587472), (22, 0.0025652507320046425), (23, 0.0013446180382743478), (24, 0.002280422719195485), (25, 0.0048489817418158054), (26, 0.008237877860665321), (27, 0.14235623180866241)], Temporal MSE: [(0, 3.266122439526953e-05), (1, 1.2115517165511847e-05), (2, 1.375764168187743e-05), (3, 1.235589115822222e-05), (4, 9.111871804634575e-06), (5, 1.900948518596124e-05), (6, 1.450174386263825e-05), (7, 1.036225239658961e-05), (8, 8.04695810074918e-06), (9, 1.286207316297805e-05), (10, 1.673347105679568e-05), (11, 8.691724360687658e-05), (12, 0.0001122693283832632), (13, 0.0002504082804080099), (14, 0.00011603244638536125), (15, 0.0001821611076593399), (16, 9.8046162747778e-05), (17, 9.653069719206542e-05), (18, 0.000225995754590258), (19, 0.00016886100638657808), (20, 0.0001561535318614915), (21, 0.00023030280135571957), (22, 0.00040208117570728064), (23, 0.0007132170721888542), (24, 0.0008406491251662374), (25, 0.0013736998662352562), (26, 0.000303724518744275), (27, 0.0001973756734514609)]
+
+ 19%|█▉        | 29/150 [05:02<15:27,  7.66s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 720, Spatial MSE: [(0, 2.270291224704124e-05), (1, 0.00027934598620049655), (2, 0.0023496211506426334), (3, 0.0002277837775181979), (4, 7.512317824875936e-05), (5, 6.317218503681943e-05), (6, 0.0001642952556721866), (7, 0.00010445575026096776), (8, 0.00012931700621265918), (9, 0.00017723416385706514), (10, 0.0002244912029709667), (11, 0.00026479471125639975), (12, 0.0003898311697412282), (13, 0.005513097159564495), (14, 0.008132882416248322), (15, 0.003637688932940364), (16, 0.10368818044662476), (17, 0.0006646753172390163), (18, 0.0014855493791401386), (19, 0.007508717477321625), (20, 0.0044238208793103695), (21, 0.0027174700517207384), (22, 0.003068376099690795), (23, 0.0014243866316974163), (24, 0.004047324415296316), (25, 0.008811132982373238), (26, 0.0056098448112607), (27, 0.08314988762140274)], Temporal MSE: [(0, 1.9100179997622035e-05), (1, 1.4961423403292429e-05), (2, 1.7297568774665706e-05), (3, 1.4104728506936226e-05), (4, 1.440346022718586e-05), (5, 2.1277724954416044e-05), (6, 2.1722980818594806e-05), (7, 1.6310246792272665e-05), (8, 1.646384953346569e-05), (9, 2.3663764295633882e-05), (10, 3.644150274340063e-05), (11, 9.711972961667925e-05), (12, 0.0001193460775539279), (13, 0.0002782910887617618), (14, 0.00016153223987203091), (15, 0.00021431308414321393), (16, 0.00015831321070436388), (17, 0.00020257288997527212), (18, 0.0003451821394264698), (19, 0.00040977541357278824), (20, 0.00045671878615394235), (21, 0.00046773350914008915), (22, 0.0004784476477652788), (23, 0.0009416438988409936), (24, 0.0017093472415581346), (25, 0.005267092026770115), (26, 0.0004275985120330006), (27, 0.00015676923794671893)]
+
+ 20%|██        | 30/150 [05:10<15:15,  7.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 714, Spatial MSE: [(0, 1.8612141502671875e-05), (1, 0.0001452401775168255), (2, 0.0009204715024679899), (3, 0.000312848569592461), (4, 4.6556397137464955e-05), (5, 2.915712320827879e-05), (6, 0.00018336530774831772), (7, 3.0686282116221264e-05), (8, 2.9294888008735143e-05), (9, 3.308070154162124e-05), (10, 4.622069900506176e-05), (11, 7.877344614826143e-05), (12, 0.00017914827913045883), (13, 0.004419195931404829), (14, 0.0052414825186133385), (15, 0.001738165970891714), (16, 0.06066630408167839), (17, 0.0002648176159709692), (18, 0.0005716510349884629), (19, 0.004334317985922098), (20, 0.002156995004042983), (21, 0.0007694246014580131), (22, 0.0016334143001586199), (23, 0.0007874798029661179), (24, 0.0013378658331930637), (25, 0.002866062568500638), (26, 0.004728778265416622), (27, 0.0877489298582077)], Temporal MSE: [(0, 2.081587808788754e-05), (1, 7.312647539947648e-06), (2, 8.37696461530868e-06), (3, 7.471662229363574e-06), (4, 5.180726930120727e-06), (5, 1.0992785064445343e-05), (6, 9.07853518583579e-06), (7, 6.627198672504164e-06), (8, 4.7210664888552856e-06), (9, 7.444868515449343e-06), (10, 1.043327029037755e-05), (11, 0.00010508544801268727), (12, 9.360609692521393e-05), (13, 0.0001623715361347422), (14, 7.713416562182829e-05), (15, 0.00010217212548013777), (16, 5.6786426284816116e-05), (17, 5.2420746214920655e-05), (18, 0.00013262966240290552), (19, 9.798000246519223e-05), (20, 9.338771633338183e-05), (21, 0.0001547538849990815), (22, 0.0002552263322286308), (23, 0.00045328328269533813), (24, 0.0005087147583253682), (25, 0.0008650989038869739), (26, 0.00018371643091086298), (27, 0.0001203935025841929)]
+
+ 21%|██        | 31/150 [05:17<15:10,  7.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 708, Spatial MSE: [(0, 1.4330751582747325e-05), (1, 0.00024422997375950217), (2, 0.0018825080478563905), (3, 0.00020653771935030818), (4, 0.00012886036711279303), (5, 0.00015106049249880016), (6, 0.00025663996348157525), (7, 0.0005263350903987885), (8, 0.0005965945310890675), (9, 0.0007986538112163544), (10, 0.0011407327838242054), (11, 0.0017009851289913058), (12, 0.0016902979696169496), (13, 0.007759788539260626), (14, 0.01087976898998022), (15, 0.006047562230378389), (16, 0.11025086045265198), (17, 0.002378256991505623), (18, 0.002671904396265745), (19, 0.009440328925848007), (20, 0.007625175639986992), (21, 0.0029613038059324026), (22, 0.0036563491448760033), (23, 0.00268605281598866), (24, 0.0063669150695204735), (25, 0.016100823879241943), (26, 0.011008289642632008), (27, 0.05488127842545509)], Temporal MSE: [(0, 1.2895354302600026e-05), (1, 1.8634917068993673e-05), (2, 1.609036735317204e-05), (3, 2.6419567802804522e-05), (4, 4.575425191433169e-05), (5, 9.280901576858014e-05), (6, 0.0001791661634342745), (7, 0.00011499678657855839), (8, 9.613914880901575e-05), (9, 0.00017467838188167661), (10, 0.00033208023523911834), (11, 0.00046904321061447263), (12, 0.0004033452132716775), (13, 0.0007667909376323223), (14, 0.0007364382036030293), (15, 0.000835085695143789), (16, 0.0007643686840310693), (17, 0.0007578275981359184), (18, 0.0007120274822227657), (19, 0.0008558375993743539), (20, 0.0017715038266032934), (21, 0.0011094720102846622), (22, 0.0008540176786482334), (23, 0.0013436888111755252), (24, 0.003586950246244669), (25, 0.012879460118710995), (26, 0.0017393800662830472), (27, 0.00012784828140866011)]
+
+ 21%|██▏       | 32/150 [05:25<15:02,  7.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 702, Spatial MSE: [(0, 8.985723070509266e-06), (1, 6.961721373954788e-05), (2, 0.0004120524681638926), (3, 0.0001420006447006017), (4, 2.125224273186177e-05), (5, 1.3399554518400691e-05), (6, 8.861524111125618e-05), (7, 1.619984868739266e-05), (8, 1.4115068552200682e-05), (9, 1.6263047655229457e-05), (10, 2.5916029699146748e-05), (11, 5.982323636999354e-05), (12, 0.00012392680218908936), (13, 0.0027932862285524607), (14, 0.0031978876795619726), (15, 0.0010934456950053573), (16, 0.03840478137135506), (17, 0.00013943150406703353), (18, 0.0003599039337132126), (19, 0.0023473913315683603), (20, 0.0009887373307719827), (21, 0.0004655862576328218), (22, 0.0009552000556141138), (23, 0.00043006191845051944), (24, 0.0007377798901870847), (25, 0.0015815627994015813), (26, 0.0025083606597036123), (27, 0.04829642176628113)], Temporal MSE: [(0, 9.980008144339081e-06), (1, 3.3657438507361803e-06), (2, 4.066887413500808e-06), (3, 3.708976237248862e-06), (4, 2.3855054678278975e-06), (5, 5.242551651463145e-06), (6, 4.526952125161188e-06), (7, 3.879756604874274e-06), (8, 2.277321300425683e-06), (9, 3.619462404458318e-06), (10, 6.238400146685308e-06), (11, 0.0001071952428901568), (12, 5.151347795617767e-05), (13, 0.00010485464736120775), (14, 4.496875044424087e-05), (15, 5.957615212537348e-05), (16, 3.848037886200473e-05), (17, 3.1050305551616475e-05), (18, 8.07835313025862e-05), (19, 5.7824388932203874e-05), (20, 5.779356797575019e-05), (21, 8.22667425381951e-05), (22, 0.00014143333828542382), (23, 0.0002839060907717794), (24, 0.0003117854066658765), (25, 0.0005545627209357917), (26, 0.00011719176836777478), (27, 6.793975626351312e-05)]
+
+ 22%|██▏       | 33/150 [05:33<14:46,  7.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 696, Spatial MSE: [(0, 7.454946626239689e-06), (1, 8.482715202262625e-05), (2, 0.0006215258035808802), (3, 7.834928692318499e-05), (4, 2.604195469757542e-05), (5, 2.5235876819351688e-05), (6, 4.402084960020147e-05), (7, 5.709499237127602e-05), (8, 8.154386887326837e-05), (9, 0.00012895878171548247), (10, 0.00019193445041310042), (11, 0.00020208570640534163), (12, 0.00024372701591346413), (13, 0.0029788375832140446), (14, 0.0034851906821131706), (15, 0.0012501559685915709), (16, 0.03901822119951248), (17, 0.00021813444618601352), (18, 0.0006393041112460196), (19, 0.003268943866714835), (20, 0.0018030786886811256), (21, 0.0013563277898356318), (22, 0.0013007934903725982), (23, 0.0005673292907886207), (24, 0.0015791789628565311), (25, 0.002821402158588171), (26, 0.002746796468272805), (27, 0.034461356699466705)], Temporal MSE: [(0, 5.881046035938198e-06), (1, 3.5396942621446215e-06), (2, 5.797625362902181e-06), (3, 5.140385837876238e-06), (4, 5.520338618225651e-06), (5, 8.073459866864141e-06), (6, 1.0688316251616925e-05), (7, 8.681889994477388e-06), (8, 8.571728358219843e-06), (9, 1.3469906662066933e-05), (10, 2.1125280909473076e-05), (11, 0.00012061209417879581), (12, 6.410023343050852e-05), (13, 0.00012175055599072948), (14, 7.291054498637095e-05), (15, 8.374626486329362e-05), (16, 6.340051913866773e-05), (17, 6.472356471931562e-05), (18, 0.0001305610639974475), (19, 0.00013281685824040323), (20, 0.00020439839863684028), (21, 0.0001594238419784233), (22, 0.00021008674229960889), (23, 0.0003934091655537486), (24, 0.0007656041416339576), (25, 0.0015639913035556674), (26, 0.0001967532152775675), (27, 5.827880522701889e-05)]
+
+ 23%|██▎       | 34/150 [05:40<14:42,  7.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 690, Spatial MSE: [(0, 4.897582130070077e-06), (1, 3.662950257421471e-05), (2, 0.00022994774917606264), (3, 6.984064384596422e-05), (4, 1.1039740456908476e-05), (5, 7.524422017013421e-06), (6, 4.3744072172557935e-05), (7, 8.249537131632678e-06), (8, 8.322637768287677e-06), (9, 9.558557394484524e-06), (10, 2.3281383619178087e-05), (11, 6.821396527811885e-05), (12, 0.0001562432007631287), (13, 0.003428491996601224), (14, 0.0031391745433211327), (15, 0.0009147359523922205), (16, 0.03131677210330963), (17, 0.00010982732783304527), (18, 0.00017648767970968038), (19, 0.0022731309290975332), (20, 0.0010750272776931524), (21, 0.0003299145901110023), (22, 0.0006643449887633324), (23, 0.0003105803916696459), (24, 0.0005562659935094416), (25, 0.0011304595973342657), (26, 0.0015306505374610424), (27, 0.035758040845394135)], Temporal MSE: [(0, 5.5083087318053e-06), (1, 1.9626208995759953e-06), (2, 2.4225018933066167e-06), (3, 2.15562567973393e-06), (4, 1.2905549056085874e-06), (5, 2.986509116453817e-06), (6, 2.5942815682356013e-06), (7, 2.0653969841077924e-06), (8, 1.2873103969468502e-06), (9, 2.1320677205949323e-06), (10, 6.12036228631041e-06), (11, 0.0001337179564870894), (12, 6.685294647468254e-05), (13, 8.854894986143336e-05), (14, 3.435088729020208e-05), (15, 4.850549885304645e-05), (16, 2.845558810804505e-05), (17, 2.3955099095473997e-05), (18, 6.752557237632573e-05), (19, 5.048288585385308e-05), (20, 4.156762224738486e-05), (21, 5.1961415010737255e-05), (22, 9.653903543949127e-05), (23, 0.0002134388341801241), (24, 0.0002394232142250985), (25, 0.00041011086432263255), (26, 8.974822412710637e-05), (27, 5.605652404483408e-05)]
+
+ 23%|██▎       | 35/150 [05:48<14:25,  7.52s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 684, Spatial MSE: [(0, 4.757389433507342e-06), (1, 7.277743134181947e-05), (2, 0.0005419941735453904), (3, 7.740916043985635e-05), (4, 5.661157410941087e-05), (5, 7.329367508646101e-05), (6, 0.00020732988195959479), (7, 0.00025540648493915796), (8, 0.00030441669514402747), (9, 0.0004293405218049884), (10, 0.0007359599112533033), (11, 0.0008083066786639392), (12, 0.0009569665999151766), (13, 0.005871683359146118), (14, 0.00848305318504572), (15, 0.004787439946085215), (16, 0.10760870575904846), (17, 0.0012923646718263626), (18, 0.0015430022031068802), (19, 0.008206814527511597), (20, 0.006489059887826443), (21, 0.0015640659257769585), (22, 0.0023352974094450474), (23, 0.0017985565355047584), (24, 0.0040023671463131905), (25, 0.010119589976966381), (26, 0.0053470744751393795), (27, 0.032360468059778214)], Temporal MSE: [(0, 5.061751835455652e-06), (1, 5.835105184814893e-06), (2, 5.931321084062802e-06), (3, 1.4663415640825406e-05), (4, 2.2581858502235264e-05), (5, 5.8125253417529166e-05), (6, 0.00013987862621434033), (7, 7.249669579323381e-05), (8, 4.220593473291956e-05), (9, 8.634221012471244e-05), (10, 0.00014928374730516225), (11, 0.0003087842487730086), (12, 0.000233476996072568), (13, 0.0003847865154966712), (14, 0.000419597519794479), (15, 0.0004427100357133895), (16, 0.0004560629022307694), (17, 0.0004940179642289877), (18, 0.0004565645649563521), (19, 0.0005805585533380508), (20, 0.0009923840407282114), (21, 0.000690148794092238), (22, 0.0005876616924069822), (23, 0.000911964219994843), (24, 0.0024405703879892826), (25, 0.008257929235696793), (26, 0.0010628503514453769), (27, 6.72944079269655e-05)]
+
+ 24%|██▍       | 36/150 [05:56<14:48,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 678, Spatial MSE: [(0, 4.317234925110824e-06), (1, 2.8765476599801332e-05), (2, 0.0001719052525004372), (3, 6.390181806636974e-05), (4, 9.571141163178254e-06), (5, 6.077734724385664e-06), (6, 4.781784082297236e-05), (7, 8.860713933245279e-06), (8, 6.871223376947455e-06), (9, 7.53878521209117e-06), (10, 1.1656497008516453e-05), (11, 1.5173611245700158e-05), (12, 3.8772253901697695e-05), (13, 0.0009599755867384374), (14, 0.0015646452084183693), (15, 0.0006147008389234543), (16, 0.032408006489276886), (17, 0.00015089449880179018), (18, 0.0002299555781064555), (19, 0.002269522286951542), (20, 0.0012097152648493648), (21, 0.0003345957084093243), (22, 0.0005756895407103002), (23, 0.00025011159596033394), (24, 0.000416635099099949), (25, 0.0009370963089168072), (26, 0.0013108215061947703), (27, 0.022311033681035042)], Temporal MSE: [(0, 4.727100531454198e-06), (1, 1.531711632196675e-06), (2, 1.8125610949937254e-06), (3, 1.8202089222540963e-06), (4, 1.0962871783704031e-06), (5, 2.4114422103593824e-06), (6, 1.914276026582229e-06), (7, 2.789960035443073e-06), (8, 1.107366642827401e-06), (9, 1.7023966165652382e-06), (10, 2.8081765321985586e-06), (11, 1.4089562682784162e-05), (12, 2.401803430984728e-05), (13, 3.612480213632807e-05), (14, 2.489322105247993e-05), (15, 5.7455159549135715e-05), (16, 2.8909120374009944e-05), (17, 2.7988087822450325e-05), (18, 6.287242285907269e-05), (19, 5.033635534346104e-05), (20, 4.317866114433855e-05), (21, 6.774815847165883e-05), (22, 0.00010143226245418191), (23, 0.00019198853988200426), (24, 0.00018349674064666033), (25, 0.0003989122051279992), (26, 6.910101365065202e-05), (27, 4.7789184463908896e-05)]
+
+ 25%|██▍       | 37/150 [06:04<14:54,  7.91s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 672, Spatial MSE: [(0, 3.6486744647845626e-06), (1, 3.495283453958109e-05), (2, 0.0003183335647918284), (3, 3.7827798223588616e-05), (4, 1.523446826467989e-05), (5, 1.2459040590329096e-05), (6, 2.1994377675582655e-05), (7, 2.5592029487597756e-05), (8, 2.8021981052006595e-05), (9, 5.237617369857617e-05), (10, 8.518829417880625e-05), (11, 8.497582894051448e-05), (12, 0.00010167368600377813), (13, 0.001257838448509574), (14, 0.002149424981325865), (15, 0.0010755625553429127), (16, 0.04359271377325058), (17, 0.00022457454178947955), (18, 0.00037295278161764145), (19, 0.0030929476488381624), (20, 0.002063013846054673), (21, 0.000477593916002661), (22, 0.0007290581124834716), (23, 0.0004031981516163796), (24, 0.0010459384648129344), (25, 0.0021095070987939835), (26, 0.0017773224972188473), (27, 0.017844269052147865)], Temporal MSE: [(0, 2.9951136184536153e-06), (1, 1.7228162505489308e-06), (2, 3.300988510090974e-06), (3, 3.4044383028231096e-06), (4, 3.7236586649669334e-06), (5, 4.451714175957022e-06), (6, 5.315077487466624e-06), (7, 4.547153366729617e-06), (8, 4.307486960897222e-06), (9, 6.200158168212511e-06), (10, 9.434469575353432e-06), (11, 2.4667024263180792e-05), (12, 3.6870173062197864e-05), (13, 5.2771185437450185e-05), (14, 4.733862078865059e-05), (15, 8.268358942586929e-05), (16, 4.89031444885768e-05), (17, 5.497590609593317e-05), (18, 9.395217784913257e-05), (19, 9.106034121941775e-05), (20, 0.00012680693180300295), (21, 0.00012987082300242037), (22, 0.00016504593077115715), (23, 0.00028458007727749646), (24, 0.0005006985738873482), (25, 0.0010901266941800714), (26, 0.0001317014975938946), (27, 4.806692595593631e-05)]
+
+ 25%|██▌       | 38/150 [06:11<14:24,  7.72s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 666, Spatial MSE: [(0, 3.1811141525395215e-06), (1, 2.3689919544267468e-05), (2, 0.0001321759045822546), (3, 4.3523163185454905e-05), (4, 7.541818376921583e-06), (5, 4.686511601903476e-06), (6, 3.111678597633727e-05), (7, 6.469435902545229e-06), (8, 5.407246590038994e-06), (9, 6.313689027592773e-06), (10, 1.2265709301573224e-05), (11, 3.388533878023736e-05), (12, 9.495271660853177e-05), (13, 0.0019000080646947026), (14, 0.0017071919282898307), (15, 0.0003853302914649248), (16, 0.018356841057538986), (17, 0.0001307582570007071), (18, 0.00013944182137493044), (19, 0.0016150663141161203), (20, 0.0009753058548085392), (21, 0.00014730797556694597), (22, 0.00046708001173101366), (23, 0.00018098931468557566), (24, 0.00037026897189207375), (25, 0.0008478800882585347), (26, 0.0011526625603437424), (27, 0.01749153435230255)], Temporal MSE: [(0, 3.4539284570200834e-06), (1, 1.1790489224949852e-06), (2, 1.4413840290217195e-06), (3, 1.3632158015752793e-06), (4, 7.964334258758754e-07), (5, 1.7525668454254628e-06), (6, 1.4222905520000495e-06), (7, 1.6727245792935719e-06), (8, 8.513442821822537e-07), (9, 1.2638801081266138e-06), (10, 3.362488769198535e-06), (11, 6.546191434608772e-05), (12, 4.801704562851228e-05), (13, 3.726010982063599e-05), (14, 1.9496894310577773e-05), (15, 3.743672641576268e-05), (16, 1.807213448046241e-05), (17, 1.8689181160880253e-05), (18, 4.0379673009738326e-05), (19, 3.3012507628882304e-05), (20, 2.812963975884486e-05), (21, 5.3567848226521164e-05), (22, 8.684033673489466e-05), (23, 0.00014360426575876772), (24, 0.000132864952320233), (25, 0.00025425414787605405), (26, 5.137624611961655e-05), (27, 3.734806159627624e-05)]
+
+ 26%|██▌       | 39/150 [06:19<14:07,  7.64s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 660, Spatial MSE: [(0, 2.672126584002399e-06), (1, 4.166728467680514e-05), (2, 0.0002729212574195117), (3, 4.4955995690543205e-05), (4, 3.988302341895178e-05), (5, 5.7886922149918973e-05), (6, 0.00019436466391198337), (7, 0.000257369305472821), (8, 0.0002663256600499153), (9, 0.00040312952478416264), (10, 0.0006045026821084321), (11, 0.0006817432004027069), (12, 0.0008731404668651521), (13, 0.0040795947425067425), (14, 0.007057745475322008), (15, 0.005127930548042059), (16, 0.09613582491874695), (17, 0.0017405592370778322), (18, 0.0019598721992224455), (19, 0.011023570783436298), (20, 0.009938931092619896), (21, 0.0021952094975858927), (22, 0.0019788083154708147), (23, 0.0018325569108128548), (24, 0.004078373778611422), (25, 0.008632762357592583), (26, 0.004232678562402725), (27, 0.014942680485546589)], Temporal MSE: [(0, 2.6473669549886836e-06), (1, 3.184952674928354e-06), (2, 3.5845239381160354e-06), (3, 1.272103236260591e-05), (4, 1.8584107237984426e-05), (5, 4.9157872126670554e-05), (6, 0.00014919895329512656), (7, 5.9841713664354756e-05), (8, 3.917827780242078e-05), (9, 7.184017886174843e-05), (10, 0.00012845035234931856), (11, 0.00020078373199794441), (12, 0.0002077220124192536), (13, 0.000344080530339852), (14, 0.0004524976829998195), (15, 0.000524356379173696), (16, 0.0006397730903699994), (17, 0.0006761990953236818), (18, 0.0004899344057776034), (19, 0.0007459570770151913), (20, 0.0011956858215853572), (21, 0.0008497278322465718), (22, 0.0005865610437467694), (23, 0.0008259399910457432), (24, 0.002647405955940485), (25, 0.006757589057087898), (26, 0.0007725055911578238), (27, 4.5362514356384054e-05)]
+
+ 27%|██▋       | 40/150 [06:27<14:03,  7.67s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 654, Spatial MSE: [(0, 2.596288140921388e-06), (1, 1.7755781300365925e-05), (2, 0.00010664940054994076), (3, 3.8888538256287575e-05), (4, 6.0743554968212266e-06), (5, 4.1751900425879285e-06), (6, 2.634840893733781e-05), (7, 6.5038179855037015e-06), (8, 4.376013293949654e-06), (9, 5.128464181325398e-06), (10, 6.661826319032116e-06), (11, 8.710256224730983e-06), (12, 2.1069810827611946e-05), (13, 0.0004311604716349393), (14, 0.0006350212497636676), (15, 0.00017482096154708415), (16, 0.011628003790974617), (17, 9.291696915170178e-05), (18, 8.1655103713274e-05), (19, 0.0010385215282440186), (20, 0.0006635732715949416), (21, 0.00020552358182612807), (22, 0.00025927636306732893), (23, 0.0001325910707237199), (24, 0.0002218861336586997), (25, 0.0005763833760283887), (26, 0.0010543869575485587), (27, 0.012113668955862522)], Temporal MSE: [(0, 2.9491211535059847e-06), (1, 1.0793029332489823e-06), (2, 1.3450238611767418e-06), (3, 1.348614659946179e-06), (4, 7.606043936903006e-07), (5, 1.603746454748034e-06), (6, 1.2762900496454677e-06), (7, 1.745332724567561e-06), (8, 7.437520821440557e-07), (9, 1.1108813851024024e-06), (10, 1.761026396707166e-06), (11, 1.0430817383166868e-05), (12, 1.3836320249538403e-05), (13, 1.496794902777765e-05), (14, 1.2799439900845755e-05), (15, 2.3530341422883794e-05), (16, 1.697732113825623e-05), (17, 1.7168162230518647e-05), (18, 2.3165779566625133e-05), (19, 1.9964189050369896e-05), (20, 1.889316445158329e-05), (21, 3.468830982455984e-05), (22, 5.179941945243627e-05), (23, 8.154092211043462e-05), (24, 8.315140439663082e-05), (25, 0.00018619414186105132), (26, 3.763203858397901e-05), (27, 2.4419359760941006e-05)]
+
+ 27%|██▋       | 41/150 [06:34<13:49,  7.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 648, Spatial MSE: [(0, 2.5866647774819285e-06), (1, 2.2377849745680578e-05), (2, 0.00017726176884025335), (3, 2.3753209461574443e-05), (4, 8.548819096176885e-06), (5, 9.41836060519563e-06), (6, 1.1745976735255681e-05), (7, 1.7443637261749245e-05), (8, 1.927622543007601e-05), (9, 4.148761217948049e-05), (10, 6.640925130341202e-05), (11, 6.72661917633377e-05), (12, 6.910142110427842e-05), (13, 0.000642894534394145), (14, 0.001090061734430492), (15, 0.0005836655618622899), (16, 0.023538250476121902), (17, 0.00014865554112475365), (18, 0.0002117761323461309), (19, 0.0016567256534472108), (20, 0.0014159036800265312), (21, 0.0003399547131266445), (22, 0.0003815300588030368), (23, 0.0002426922001177445), (24, 0.0007164744893088937), (25, 0.0014157721307128668), (26, 0.0017186718760058284), (27, 0.008745016530156136)], Temporal MSE: [(0, 1.946932570717763e-06), (1, 9.84234702627873e-07), (2, 2.1123678379808553e-06), (3, 2.349889427932794e-06), (4, 2.7160654099134263e-06), (5, 3.080518126807874e-06), (6, 3.841937541437801e-06), (7, 3.612383125073393e-06), (8, 3.084189074797905e-06), (9, 4.1404796320421156e-06), (10, 7.10162521500024e-06), (11, 1.6895639419090003e-05), (12, 2.2322654331219383e-05), (13, 2.692774432944134e-05), (14, 2.8204933187225834e-05), (15, 4.4760115997632965e-05), (16, 3.109111639787443e-05), (17, 3.48896901414264e-05), (18, 4.3205509427934885e-05), (19, 4.477130642044358e-05), (20, 7.512056617997587e-05), (21, 8.469720341963693e-05), (22, 0.00010972683958243579), (23, 0.00015249205171130598), (24, 0.00032047537388280034), (25, 0.0005873372429050505), (26, 9.950375533662736e-05), (27, 2.501318340364378e-05)]
+
+ 28%|██▊       | 42/150 [06:43<14:12,  7.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 642, Spatial MSE: [(0, 2.9126181289029773e-06), (1, 1.723324749036692e-05), (2, 0.00010947006376227364), (3, 3.51601374859456e-05), (4, 6.310912795015611e-06), (5, 3.694175347845885e-06), (6, 2.5714403818710707e-05), (7, 7.1434915298596025e-06), (8, 4.587546754919458e-06), (9, 5.473249075294007e-06), (10, 9.200433851219714e-06), (11, 1.5930490917526186e-05), (12, 3.585834929253906e-05), (13, 0.0004200274997856468), (14, 0.0005751855787821114), (15, 0.00017050110909622163), (16, 0.012951653450727463), (17, 0.00010161423415411264), (18, 0.00013135385233908892), (19, 0.0011690607061609626), (20, 0.0007735367398709059), (21, 0.00010314212704543024), (22, 0.00022413024271372706), (23, 0.0001353958941763267), (24, 0.0002618522266857326), (25, 0.0006526429788209498), (26, 0.0013177117798477411), (27, 0.013219529762864113)], Temporal MSE: [(0, 3.0603043796872953e-06), (1, 1.0488977295608493e-06), (2, 1.2684193961831625e-06), (3, 1.4453615904130857e-06), (4, 7.944683488858573e-07), (5, 1.632355633773841e-06), (6, 1.2750094811053714e-06), (7, 1.6981065300569753e-06), (8, 7.45961301618081e-07), (9, 1.1692244470395963e-06), (10, 3.1337917789642233e-06), (11, 2.477768612152431e-05), (12, 1.620389593881555e-05), (13, 1.0715408279793337e-05), (14, 1.2716272976831533e-05), (15, 1.954397521330975e-05), (16, 1.1909914064744953e-05), (17, 1.2888554920209572e-05), (18, 2.616173333080951e-05), (19, 2.1681977159460075e-05), (20, 1.9109065760858357e-05), (21, 3.78572876797989e-05), (22, 5.847805005032569e-05), (23, 8.671660907566547e-05), (24, 8.186233026208356e-05), (25, 0.0001952130551217124), (26, 4.2554864194244146e-05), (27, 2.5950024792109616e-05)]
+
+ 29%|██▊       | 43/150 [06:54<15:43,  8.82s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 636, Spatial MSE: [(0, 2.4718531221878948e-06), (1, 3.2729738450143486e-05), (2, 0.00020938758098054677), (3, 3.4028576919808984e-05), (4, 3.5071599995717406e-05), (5, 5.1966417231597006e-05), (6, 0.0002839614462573081), (7, 0.0002807988494168967), (8, 0.00030717658228240907), (9, 0.0005406767595559359), (10, 0.0009260885417461395), (11, 0.0009811240015551448), (12, 0.0010984918335452676), (13, 0.0017867657588794827), (14, 0.0033092463854700327), (15, 0.0046636383049190044), (16, 0.01917681284248829), (17, 0.0030191296245902777), (18, 0.002827898832038045), (19, 0.0033358687069267035), (20, 0.004286105278879404), (21, 0.00332080852240324), (22, 0.00270185898989439), (23, 0.003115001367405057), (24, 0.006489860359579325), (25, 0.01235081348568201), (26, 0.006514661479741335), (27, 0.00719012226909399)], Temporal MSE: [(0, 2.4877617761376314e-06), (1, 2.4463004137942335e-06), (2, 3.370493004695163e-06), (3, 1.511330992798321e-05), (4, 2.0371579012135044e-05), (5, 4.9503883928991854e-05), (6, 0.00016998958017211407), (7, 6.347025191644207e-05), (8, 4.5681568735744804e-05), (9, 8.980868733488023e-05), (10, 0.00016860541654750705), (11, 0.00019706759485416114), (12, 0.00023058037913870066), (13, 0.0004271524085197598), (14, 0.0007001832709647715), (15, 0.0009124232456088066), (16, 0.0011832024902105331), (17, 0.0012333061313256621), (18, 0.0007485271198675036), (19, 0.0013182599795982242), (20, 0.002352179726585746), (21, 0.0015964722260832787), (22, 0.0007935285684652627), (23, 0.0011750967241823673), (24, 0.003713678801432252), (25, 0.0077354274690151215), (26, 0.0008249919046647847), (27, 3.98009542550426e-05)]
+
+ 29%|██▉       | 44/150 [07:01<14:58,  8.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 630, Spatial MSE: [(0, 2.9701809580728877e-06), (1, 1.544088991067838e-05), (2, 8.455464558210224e-05), (3, 3.250870213378221e-05), (4, 5.404593593993923e-06), (5, 3.4135141504521016e-06), (6, 2.728351864789147e-05), (7, 7.106106750143226e-06), (8, 4.294681275496259e-06), (9, 5.058016995462822e-06), (10, 5.966621301922714e-06), (11, 8.174730282917153e-06), (12, 1.9120441720588133e-05), (13, 0.0004909838899038732), (14, 0.0006262775277718902), (15, 0.0002576955303084105), (16, 0.015311905182898045), (17, 8.258165325969458e-05), (18, 0.0001281888980884105), (19, 0.0014615284744650126), (20, 0.0009722516988404095), (21, 0.00011941192497033626), (22, 0.00023924368724692613), (23, 0.00010872379061765969), (24, 0.00021605247457046062), (25, 0.0005656231660395861), (26, 0.0010207424638792872), (27, 0.012119977734982967)], Temporal MSE: [(0, 2.849609018085175e-06), (1, 8.441139129899966e-07), (2, 9.987519433707348e-07), (3, 1.1881606951646972e-06), (4, 6.88120621816779e-07), (5, 1.382832579110982e-06), (6, 1.1615354651439702e-06), (7, 2.2507829271489754e-06), (8, 7.3565092861827e-07), (9, 1.0449983847138355e-06), (10, 1.8292506638317718e-06), (11, 1.095181141863577e-05), (12, 1.1197452295164112e-05), (13, 1.548587897559628e-05), (14, 1.491597868152894e-05), (15, 1.8338912923354656e-05), (16, 1.3313064300746191e-05), (17, 1.595907633600291e-05), (18, 3.330662366352044e-05), (19, 2.7707796107279137e-05), (20, 2.3295404389500618e-05), (21, 5.0806484068743885e-05), (22, 6.459908763645217e-05), (23, 9.554230200592428e-05), (24, 8.599369175499305e-05), (25, 0.00023034409969113767), (26, 4.1139584936900064e-05), (27, 3.511213435558602e-05)]
+
+ 30%|███       | 45/150 [07:08<14:05,  8.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 624, Spatial MSE: [(0, 2.3660081751586404e-06), (1, 1.9358471035957336e-05), (2, 0.00015710869047325104), (3, 1.9726478058146313e-05), (4, 9.604207662050612e-06), (5, 1.0064509297080804e-05), (6, 1.592780972714536e-05), (7, 2.3349724870058708e-05), (8, 2.8997954359510913e-05), (9, 5.6021639466052875e-05), (10, 9.872671216726303e-05), (11, 0.00011620266013778746), (12, 0.00010361871682107449), (13, 0.0005338195478543639), (14, 0.0007849584799259901), (15, 0.00048221374163404107), (16, 0.01995285600423813), (17, 0.0001339167356491089), (18, 0.00031605997355654836), (19, 0.0020969181787222624), (20, 0.0018264121608808637), (21, 0.00022152828751131892), (22, 0.0003642732626758516), (23, 0.00024168743402697146), (24, 0.0007270628120750189), (25, 0.0013662442797794938), (26, 0.0019992850720882416), (27, 0.009916199371218681)], Temporal MSE: [(0, 1.6961074607024784e-06), (1, 8.31375928100897e-07), (2, 2.108662329192157e-06), (3, 2.3472980501537677e-06), (4, 2.7565301934373565e-06), (5, 3.3443009215261554e-06), (6, 4.62915613752557e-06), (7, 3.9964525058167055e-06), (8, 3.703765287355054e-06), (9, 5.57419707547524e-06), (10, 9.936364222085103e-06), (11, 1.9969394998042844e-05), (12, 2.2949709091335535e-05), (13, 2.707461317186244e-05), (14, 3.388729601283558e-05), (15, 3.621085488703102e-05), (16, 2.6628593332134187e-05), (17, 3.2760166504886e-05), (18, 5.167574272491038e-05), (19, 5.892071931157261e-05), (20, 9.166603558696806e-05), (21, 9.805267472984269e-05), (22, 0.00011771167191909626), (23, 0.0001578678929945454), (24, 0.00033579953014850616), (25, 0.0005023680860176682), (26, 9.721310198074207e-05), (27, 3.466092312010005e-05)]
+
+ 31%|███       | 46/150 [07:17<14:03,  8.11s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 618, Spatial MSE: [(0, 1.712598873382376e-06), (1, 1.3763620700046886e-05), (2, 6.336474325507879e-05), (3, 2.054015567409806e-05), (4, 3.454416173553909e-06), (5, 2.429039568596636e-06), (6, 1.2377832717902493e-05), (7, 3.742662102013128e-06), (8, 2.850679948096513e-06), (9, 3.6632291084970348e-06), (10, 1.0033391845354345e-05), (11, 1.6631656762911007e-05), (12, 3.694381666718982e-05), (13, 0.0004057918849866837), (14, 0.0009039514698088169), (15, 0.0001965347328223288), (16, 0.01743992418050766), (17, 0.00013452353596221656), (18, 0.00013245511217974126), (19, 0.0016449299873784184), (20, 0.0011177953565493226), (21, 0.0001664915180299431), (22, 0.0003095623105764389), (23, 0.0001135251804953441), (24, 0.00023851219157222658), (25, 0.0005670246318913996), (26, 0.0007727697375230491), (27, 0.009537985548377037)], Temporal MSE: [(0, 1.80012727923895e-06), (1, 6.608669309571269e-07), (2, 8.700457669874595e-07), (3, 8.389084769078181e-07), (4, 4.5557942485174863e-07), (5, 9.295898735217634e-07), (6, 8.227695502682764e-07), (7, 1.028416704684787e-06), (8, 5.062686341261724e-07), (9, 8.165141593963199e-07), (10, 3.820609435933875e-06), (11, 1.5744320990052074e-05), (12, 1.9902363419532776e-05), (13, 1.1786490176746156e-05), (14, 1.4705285138916224e-05), (15, 3.799043770413846e-05), (16, 1.98779580387054e-05), (17, 1.958188477146905e-05), (18, 3.242417369619943e-05), (19, 2.621597559482325e-05), (20, 2.3392756702378392e-05), (21, 5.483095446834341e-05), (22, 7.903472578618675e-05), (23, 0.00010496998584130779), (24, 7.962805102579296e-05), (25, 0.00015082780737429857), (26, 3.1793890229891986e-05), (27, 3.6781071685254574e-05)]
+
+ 31%|███▏      | 47/150 [07:26<14:46,  8.60s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 612, Spatial MSE: [(0, 1.5604981626893277e-06), (1, 2.6008612621808425e-05), (2, 0.00012859106936957687), (3, 2.7252577638137154e-05), (4, 2.9445929612847976e-05), (5, 4.788774094777182e-05), (6, 0.00018780905520543456), (7, 0.0002801612135954201), (8, 0.0005325018428266048), (9, 0.0008669123635627329), (10, 0.0012681513326242566), (11, 0.0012099614832550287), (12, 0.0014468592125922441), (13, 0.002917459700256586), (14, 0.006020859349519014), (15, 0.006141762249171734), (16, 0.04992543160915375), (17, 0.0038914282340556383), (18, 0.0038015905302017927), (19, 0.006639250088483095), (20, 0.008943228051066399), (21, 0.007621970027685165), (22, 0.004231828264892101), (23, 0.005485875997692347), (24, 0.010971348732709885), (25, 0.018262065947055817), (26, 0.009957158006727695), (27, 0.008270889520645142)], Temporal MSE: [(0, 1.4991508123785025e-06), (1, 1.598729113538866e-06), (2, 2.4180935724871233e-06), (3, 1.3588921319751535e-05), (4, 1.637482455407735e-05), (5, 3.6755238397745416e-05), (6, 0.00015207471733447164), (7, 7.496814941987395e-05), (8, 6.388106703525409e-05), (9, 0.00012589333346113563), (10, 0.00024384249991271645), (11, 0.0002624907065182924), (12, 0.0003495481505524367), (13, 0.0006543524214066565), (14, 0.0010633005294948816), (15, 0.0012154782889410853), (16, 0.001591595821082592), (17, 0.0017030040035024285), (18, 0.0012634375598281622), (19, 0.0027492705266922712), (20, 0.005611353553831577), (21, 0.004171191249042749), (22, 0.001389785436913371), (23, 0.0020124621223658323), (24, 0.00612057326361537), (25, 0.008124326355755329), (26, 0.0011535969097167253), (27, 6.0730242694262415e-05)]
+
+ 32%|███▏      | 48/150 [07:35<14:53,  8.76s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 606, Spatial MSE: [(0, 2.2051776795706246e-06), (1, 1.2491962479543872e-05), (2, 6.73118483973667e-05), (3, 2.387330459896475e-05), (4, 3.857186584355077e-06), (5, 2.538474745961139e-06), (6, 1.6056004824349657e-05), (7, 5.162570687389234e-06), (8, 3.177603048243327e-06), (9, 3.57759586222528e-06), (10, 4.4000207708450034e-06), (11, 4.785879355040379e-06), (12, 8.534169865015429e-06), (13, 0.00014243778423406184), (14, 0.00012039153079967946), (15, 6.72142268740572e-05), (16, 0.0030875548254698515), (17, 2.296117781952489e-05), (18, 0.000151427200762555), (19, 0.00011153279046993703), (20, 1.800925747375004e-05), (21, 5.0311587983742356e-05), (22, 0.00013370368105825037), (23, 6.974819552851841e-05), (24, 0.00010766389459604397), (25, 0.0002756855101324618), (26, 0.000784873147495091), (27, 0.005245745647698641)], Temporal MSE: [(0, 2.0123127342230873e-06), (1, 6.517464044009103e-07), (2, 8.83721440914087e-07), (3, 9.418256468052277e-07), (4, 5.240859195509984e-07), (5, 1.0671529935279978e-06), (6, 9.272085890188464e-07), (7, 1.4707143236591946e-06), (8, 5.65858158552146e-07), (9, 8.09594496331556e-07), (10, 1.2902737580589019e-06), (11, 3.0026844797248486e-06), (12, 3.978537279181182e-06), (13, 6.006161584082292e-06), (14, 5.4972651923890226e-06), (15, 7.651034138689283e-06), (16, 5.410456651588902e-06), (17, 5.2627783588832244e-06), (18, 8.71508200361859e-06), (19, 4.607457412930671e-06), (20, 6.11914629189414e-06), (21, 1.466249432269251e-05), (22, 2.2065625671530142e-05), (23, 4.005958544439636e-05), (24, 3.763767017517239e-05), (25, 0.00013861812476534396), (26, 2.6431784135638736e-05), (27, 4.899100531474687e-06)]
+
+ 33%|███▎      | 49/150 [07:46<15:23,  9.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 600, Spatial MSE: [(0, 2.1836883661308093e-06), (1, 1.7958833268494345e-05), (2, 0.00012531428365036845), (3, 1.919421811180655e-05), (4, 7.606322924402775e-06), (5, 8.114030606520828e-06), (6, 1.3287298315844964e-05), (7, 1.5502791939070448e-05), (8, 1.915496432047803e-05), (9, 4.189835090073757e-05), (10, 6.225541437743232e-05), (11, 6.819237023591995e-05), (12, 6.117510929470882e-05), (13, 0.0002840544912032783), (14, 0.00029129619360901415), (15, 0.00010921243665507063), (16, 0.0031344888266175985), (17, 5.0551676395116374e-05), (18, 0.00023667361529078335), (19, 0.00015415780944749713), (20, 7.552804163424298e-05), (21, 0.00010589714656816795), (22, 0.00023865005641710013), (23, 0.00019043375505134463), (24, 0.0005616105045191944), (25, 0.0010362302418798208), (26, 0.0021072481758892536), (27, 0.003233174793422222)], Temporal MSE: [(0, 1.4895484810040216e-06), (1, 7.813201250428392e-07), (2, 1.7012280295602977e-06), (3, 2.1591788481600815e-06), (4, 2.5256965727749048e-06), (5, 2.673841663636267e-06), (6, 3.2184789233724587e-06), (7, 3.599014689825708e-06), (8, 3.156579168717144e-06), (9, 4.039964096591575e-06), (10, 6.707739885314368e-06), (11, 1.0388130249339156e-05), (12, 1.3589286936621647e-05), (13, 1.6579044313402846e-05), (14, 2.0138411855441518e-05), (15, 1.911558683787007e-05), (16, 1.6933428923948668e-05), (17, 1.7848551578936167e-05), (18, 2.5282530259573832e-05), (19, 2.5833929612417705e-05), (20, 5.115543171996251e-05), (21, 4.9896665586857125e-05), (22, 6.78516662446782e-05), (23, 0.00010570656013442203), (24, 0.00026755634462460876), (25, 0.0004900859203189611), (26, 9.175056766252965e-05), (27, 5.940745268162573e-06)]
+
+ 33%|███▎      | 50/150 [07:55<15:29,  9.30s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 594, Spatial MSE: [(0, 1.022961896524066e-06), (1, 8.658732440380845e-06), (2, 5.014148700865917e-05), (3, 1.5356219591922127e-05), (4, 2.6283048555342248e-06), (5, 1.7448458038415993e-06), (6, 7.763430403429084e-06), (7, 3.7046959278086433e-06), (8, 2.202849827881437e-06), (9, 2.5336287308164174e-06), (10, 3.1499014312430518e-06), (11, 3.815734999079723e-06), (12, 7.89193290984258e-06), (13, 0.00011844625259982422), (14, 0.0001374200510326773), (15, 6.277272041188553e-05), (16, 0.0020820347126573324), (17, 1.5768555385875516e-05), (18, 9.455961844651029e-05), (19, 6.814023072365671e-05), (20, 1.5226649338728748e-05), (21, 3.440431100898422e-05), (22, 0.00012040561705362052), (23, 5.873265763511881e-05), (24, 0.00010597197251627222), (25, 0.00026651282678358257), (26, 0.0005935634835623205), (27, 0.0033903613220900297)], Temporal MSE: [(0, 1.1123573813165422e-06), (1, 4.5663026071451895e-07), (2, 6.988308882682759e-07), (3, 7.158153039199533e-07), (4, 3.99770271997113e-07), (5, 7.936563974908495e-07), (6, 7.825472039257875e-07), (7, 9.407103789271787e-07), (8, 4.0794245137476537e-07), (9, 5.809116601085407e-07), (10, 9.78298089648888e-07), (11, 2.700176992220804e-06), (12, 4.610156338458182e-06), (13, 4.886819169769296e-06), (14, 4.5414922169584315e-06), (15, 5.5610703384445515e-06), (16, 4.3642207856464665e-06), (17, 4.207119673083071e-06), (18, 5.724284164898563e-06), (19, 3.4917943594336975e-06), (20, 5.18935166837764e-06), (21, 1.0400292921985965e-05), (22, 1.7100446711992845e-05), (23, 3.116034349659458e-05), (24, 3.239473153371364e-05), (25, 9.477974526816979e-05), (26, 1.8995709979208186e-05), (27, 1.6692530380169046e-06)]
+
+ 34%|███▍      | 51/150 [08:03<14:37,  8.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 588, Spatial MSE: [(0, 1.531542579868983e-06), (1, 2.5830368031165563e-05), (2, 0.00013673592184204608), (3, 2.963352744700387e-05), (4, 2.906724694184959e-05), (5, 4.794830238097347e-05), (6, 0.00025819017901085317), (7, 0.00027200314798392355), (8, 0.00031332619255408645), (9, 0.0005559484707191586), (10, 0.0007943871314637363), (11, 0.000746360863558948), (12, 0.0008868025615811348), (13, 0.001813533017411828), (14, 0.0037593008019030094), (15, 0.0034499105531722307), (16, 0.03914527967572212), (17, 0.002100637648254633), (18, 0.0023975777439773083), (19, 0.005742425564676523), (20, 0.006447799038141966), (21, 0.003220811951905489), (22, 0.002622509142383933), (23, 0.0033172050025314093), (24, 0.006930407136678696), (25, 0.011573944240808487), (26, 0.007475798483937979), (27, 0.005235351622104645)], Temporal MSE: [(0, 1.8523805920267478e-06), (1, 1.50039488744369e-06), (2, 2.5240487957489677e-06), (3, 1.4147680303722154e-05), (4, 1.6522784790140577e-05), (5, 3.768601891351864e-05), (6, 0.00014083039422985166), (7, 6.430748908314854e-05), (8, 4.289414209779352e-05), (9, 8.052960765780881e-05), (10, 0.00015757248911540955), (11, 0.00017026114801410586), (12, 0.00021496953559108078), (13, 0.0004144282720517367), (14, 0.0006466139457188547), (15, 0.0007320193690247834), (16, 0.0009421103168278933), (17, 0.0008973218500614166), (18, 0.0006903751636855304), (19, 0.0011274374555796385), (20, 0.0019909546244889498), (21, 0.0015114574925974011), (22, 0.0008777568000368774), (23, 0.0013394319685176015), (24, 0.004004314541816711), (25, 0.006538585294038057), (26, 0.0008343601366505027), (27, 3.3463929867139086e-05)]
+
+ 35%|███▍      | 52/150 [08:12<14:28,  8.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 582, Spatial MSE: [(0, 1.5705365967733087e-06), (1, 1.1246048416069243e-05), (2, 6.0398891946533695e-05), (3, 1.6049159967224114e-05), (4, 3.3197518405359006e-06), (5, 2.0397751541167963e-06), (6, 8.929009709390812e-06), (7, 3.591946779124555e-06), (8, 2.6968791644321755e-06), (9, 3.391601467228611e-06), (10, 3.6635080959968036e-06), (11, 4.434549282450462e-06), (12, 6.717440101056127e-06), (13, 8.703999628778547e-05), (14, 9.574864088790491e-05), (15, 4.007327515864745e-05), (16, 0.0009534250712022185), (17, 1.6602851246716455e-05), (18, 2.4948707505245693e-05), (19, 9.465299808653072e-05), (20, 3.220810685888864e-05), (21, 4.681031350628473e-05), (22, 0.00014012286555953324), (23, 6.692115857731551e-05), (24, 0.00010486898827366531), (25, 0.000266988790826872), (26, 0.0006619383930228651), (27, 0.003910537343472242)], Temporal MSE: [(0, 1.5988305221981136e-06), (1, 6.004065085107868e-07), (2, 8.596400107308e-07), (3, 8.949789389589569e-07), (4, 4.770399755216204e-07), (5, 9.067613291335874e-07), (6, 8.293305882034474e-07), (7, 1.072309487426537e-06), (8, 5.025820541959547e-07), (9, 6.894835564708046e-07), (10, 1.24959649383527e-06), (11, 1.9758504095079843e-06), (12, 2.7010714802599978e-06), (13, 3.482125293885474e-06), (14, 4.346094101492781e-06), (15, 4.039944087708136e-06), (16, 4.06885328629869e-06), (17, 3.940047463402152e-06), (18, 8.69178711582208e-06), (19, 8.606279152445495e-06), (20, 1.0967869457090273e-05), (21, 1.1764954251702875e-05), (22, 1.7686585124465637e-05), (23, 2.8064041543984786e-05), (24, 3.118932727375068e-05), (25, 0.0001232866634381935), (26, 2.3574279111926444e-05), (27, 2.4140124423865927e-06)]
+
+ 35%|███▌      | 53/150 [08:20<14:11,  8.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 576, Spatial MSE: [(0, 1.601557528374542e-06), (1, 1.321882700722199e-05), (2, 8.173004607670009e-05), (3, 1.4150985407468397e-05), (4, 6.237002253328683e-06), (5, 7.154679678933462e-06), (6, 1.1612647540459875e-05), (7, 1.3592790310212877e-05), (8, 1.6977257473627105e-05), (9, 3.13604723487515e-05), (10, 4.964398613083176e-05), (11, 5.5327178415609524e-05), (12, 5.378897185437381e-05), (13, 0.00013594959455076605), (14, 0.00017193796520587057), (15, 7.89106052252464e-05), (16, 0.0009842731524258852), (17, 3.915686102118343e-05), (18, 7.012100832071155e-05), (19, 0.00011724336218321696), (20, 7.75313819758594e-05), (21, 8.597708074375987e-05), (22, 0.0002123611920978874), (23, 0.0001511109876446426), (24, 0.0004603546694852412), (25, 0.000860497064422816), (26, 0.001998667139559984), (27, 0.002547692274674773)], Temporal MSE: [(0, 1.0143654662897461e-06), (1, 5.197902055442682e-07), (2, 1.3130832030583406e-06), (3, 1.741488176776329e-06), (4, 2.279183945574914e-06), (5, 2.3293141566682607e-06), (6, 2.816167580022011e-06), (7, 2.966731699416414e-06), (8, 2.658875700944918e-06), (9, 3.3572539450688055e-06), (10, 5.5857826737337746e-06), (11, 7.300544893951155e-06), (12, 8.83718348632101e-06), (13, 1.2648671145143453e-05), (14, 1.7542146451887675e-05), (15, 1.3870771908841562e-05), (16, 1.3722230505663902e-05), (17, 1.421490014763549e-05), (18, 2.2594738766201772e-05), (19, 2.9951479518786073e-05), (20, 5.088444231660105e-05), (21, 4.052886652061716e-05), (22, 5.512573625310324e-05), (23, 7.991820893948898e-05), (24, 0.00021499361901078373), (25, 0.0003703054098878056), (26, 8.854373299982399e-05), (27, 4.477866696106503e-06)]
+
+ 36%|███▌      | 54/150 [08:28<13:32,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 570, Spatial MSE: [(0, 1.0609633136482444e-06), (1, 9.401661372976378e-06), (2, 4.559412263915874e-05), (3, 1.4737856872670818e-05), (4, 2.992583404193283e-06), (5, 1.6611229511909187e-06), (6, 8.13308452052297e-06), (7, 3.004525751748588e-06), (8, 2.167888169424259e-06), (9, 2.8534893772302894e-06), (10, 3.000716787937563e-06), (11, 3.2068724067357834e-06), (12, 4.6729146561119705e-06), (13, 3.814929004875012e-05), (14, 2.6683914256864227e-05), (15, 1.0324951290385798e-05), (16, 1.2623809197975788e-05), (17, 5.505515218828805e-06), (18, 7.850983820389956e-06), (19, 1.5331605027313344e-05), (20, 9.16808767215116e-06), (21, 2.171334926970303e-05), (22, 3.2482810638612136e-05), (23, 3.861657751258463e-05), (24, 8.162266021827236e-05), (25, 0.00022076893947087228), (26, 0.0005621674936264753), (27, 0.003613066393882036)], Temporal MSE: [(0, 1.155335098701471e-06), (1, 4.7352418164336996e-07), (2, 6.539524406434793e-07), (3, 7.294213446584763e-07), (4, 4.0998608596964914e-07), (5, 7.650893962818373e-07), (6, 6.838097306172131e-07), (7, 8.554190458198718e-07), (8, 4.209912844999053e-07), (9, 6.067504045859096e-07), (10, 8.99712858881685e-07), (11, 1.3985272744321264e-06), (12, 1.6496949228894664e-06), (13, 1.6238611806329573e-06), (14, 2.152894467144506e-06), (15, 1.4080225128054735e-06), (16, 2.311889602424344e-06), (17, 1.5950599845382385e-06), (18, 2.297086211910937e-06), (19, 2.777203462756006e-06), (20, 4.0450772758049425e-06), (21, 3.3091980640165275e-06), (22, 6.147646672616247e-06), (23, 1.0021421985584311e-05), (24, 1.9506347598508e-05), (25, 6.064029366825707e-05), (26, 1.8224203813588247e-05), (27, 1.1879999419761589e-06)]
+
+ 37%|███▋      | 55/150 [08:38<14:04,  8.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 564, Spatial MSE: [(0, 1.3441172086459119e-06), (1, 2.0630373910535127e-05), (2, 0.00010325213224859908), (3, 2.4752947865636088e-05), (4, 2.8081518394174054e-05), (5, 4.39883915532846e-05), (6, 0.00017552723875269294), (7, 0.00023459845397155732), (8, 0.0003685243136715144), (9, 0.000595957797486335), (10, 0.0006641884101554751), (11, 0.0007155020139180124), (12, 0.0008429790614172816), (13, 0.002383599756285548), (14, 0.005307316314429045), (15, 0.004039995837956667), (16, 0.06146226450800896), (17, 0.0018394037615507841), (18, 0.0019826339557766914), (19, 0.005170132033526897), (20, 0.004712298978120089), (21, 0.0032308497466146946), (22, 0.0023431614972651005), (23, 0.0031490111723542213), (24, 0.006717870477586985), (25, 0.011194069869816303), (26, 0.007626314647495747), (27, 0.003027543891221285)], Temporal MSE: [(0, 1.4968688901717542e-06), (1, 1.1124760703751235e-06), (2, 2.4898306492104894e-06), (3, 1.3527835108106956e-05), (4, 1.5074445400387049e-05), (5, 2.9763383281533606e-05), (6, 0.00011012565664714202), (7, 6.50597212370485e-05), (8, 4.388985689729452e-05), (9, 8.76606281963177e-05), (10, 0.00018288090359419584), (11, 0.00020517419034149498), (12, 0.0002330940478714183), (13, 0.0004369816742837429), (14, 0.0006669384893029928), (15, 0.0007948031416162848), (16, 0.0009303994011133909), (17, 0.0009670041617937386), (18, 0.0007408849778585136), (19, 0.0012447363696992397), (20, 0.002422909252345562), (21, 0.001868545776233077), (22, 0.000870550749823451), (23, 0.0013229161268100142), (24, 0.0035683803725987673), (25, 0.005556277930736542), (26, 0.0009015801479108632), (27, 2.5790592189878225e-05)]
+
+ 37%|███▋      | 56/150 [08:54<17:07, 10.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 558, Spatial MSE: [(0, 1.372124756926496e-06), (1, 1.0640395885275211e-05), (2, 5.05003408761695e-05), (3, 1.4479098354058806e-05), (4, 2.9296973025338957e-06), (5, 1.8021725054495619e-06), (6, 7.458435902663041e-06), (7, 3.4627273635123856e-06), (8, 2.327868287466117e-06), (9, 2.733658220677171e-06), (10, 3.0779319786233827e-06), (11, 3.9949504753167275e-06), (12, 7.906191058282275e-06), (13, 0.00018490645743440837), (14, 0.00030934333335608244), (15, 0.0001320649462286383), (16, 0.005064271856099367), (17, 3.9869657484814525e-05), (18, 3.5714576370082796e-05), (19, 0.00042047337046824396), (20, 0.0002957599936053157), (21, 0.00012295471969991922), (22, 0.00016866304213181138), (23, 7.128144352464005e-05), (24, 0.0001405487855663523), (25, 0.00033607083605602384), (26, 0.0006458425195887685), (27, 0.005095626227557659)], Temporal MSE: [(0, 1.3083036947136861e-06), (1, 4.960884893989714e-07), (2, 7.225563081192377e-07), (3, 7.873683784964669e-07), (4, 4.3668441662703117e-07), (5, 8.059520268943743e-07), (6, 7.452072736668924e-07), (7, 9.914710972225294e-07), (8, 4.5264735604177986e-07), (9, 6.20883099600178e-07), (10, 9.752603773449664e-07), (11, 2.3169247924670344e-06), (12, 4.446307684702333e-06), (13, 7.993686267582234e-06), (14, 1.0456769814481959e-05), (15, 8.60986710904399e-06), (16, 7.343859579123091e-06), (17, 9.54605911829276e-06), (18, 1.0165182175114751e-05), (19, 9.409677659277804e-06), (20, 1.0499295058252756e-05), (21, 2.401343226665631e-05), (22, 3.50786249327939e-05), (23, 5.607787170447409e-05), (24, 4.6399309212574735e-05), (25, 9.818370745051652e-05), (26, 2.8165166440885514e-05), (27, 1.1699132301146165e-05)]
+
+ 38%|███▊      | 57/150 [09:07<17:56, 11.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 552, Spatial MSE: [(0, 1.4821704326095642e-06), (1, 1.2594170584634412e-05), (2, 7.247754547279328e-05), (3, 1.4274844033934642e-05), (4, 7.54283792048227e-06), (5, 8.342524779436644e-06), (6, 1.8861865100916475e-05), (7, 1.910330684040673e-05), (8, 2.3554981453344226e-05), (9, 4.520453148870729e-05), (10, 7.50291146687232e-05), (11, 8.767034887569025e-05), (12, 8.15659441286698e-05), (13, 0.0002974466187879443), (14, 0.0007136266212910414), (15, 0.0005832531023770571), (16, 0.011393437162041664), (17, 0.00010499953350517899), (18, 0.0003966134099755436), (19, 0.0012810768093913794), (20, 0.001443810062482953), (21, 0.0005552889197133482), (22, 0.00028419503360055387), (23, 0.0002143829333363101), (24, 0.0006252781604416668), (25, 0.0010354475816711783), (26, 0.0021029901690781116), (27, 0.0038445221725851297)], Temporal MSE: [(0, 1.041794007505814e-06), (1, 5.485609335664776e-07), (2, 1.234675437444821e-06), (3, 1.917968575071427e-06), (4, 2.525096988392761e-06), (5, 2.8351378205115907e-06), (6, 3.7231548049021512e-06), (7, 4.056239504279802e-06), (8, 3.162416987834149e-06), (9, 4.555655777949141e-06), (10, 7.914182788226753e-06), (11, 1.0322293746867217e-05), (12, 1.358524059469346e-05), (13, 1.9724653611774556e-05), (14, 2.9145392545615323e-05), (15, 2.188252801715862e-05), (16, 1.913281812448986e-05), (17, 1.9264389266027138e-05), (18, 2.8786109396605752e-05), (19, 4.7089131840039045e-05), (20, 0.00010739901335909963), (21, 8.265456563094631e-05), (22, 9.025198960443959e-05), (23, 0.00011872963659698144), (24, 0.000257805862929672), (25, 0.00042633674456737936), (26, 9.682171366875991e-05), (27, 1.4651201126980595e-05)]
+
+ 39%|███▊      | 58/150 [09:17<17:11, 11.22s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 546, Spatial MSE: [(0, 1.417373823642265e-06), (1, 9.97017377812881e-06), (2, 4.452438952284865e-05), (3, 1.3292943549458869e-05), (4, 2.7040000531997066e-06), (5, 1.7854758880275767e-06), (6, 7.314260528801242e-06), (7, 3.6526475923892576e-06), (8, 2.4621938337077154e-06), (9, 3.2710483992559602e-06), (10, 1.0389149792899843e-05), (11, 6.58190474496223e-05), (12, 8.289107790915295e-05), (13, 0.0012570180697366595), (14, 0.0010008608223870397), (15, 0.00022758811246603727), (16, 0.005152958910912275), (17, 2.8581431251950562e-05), (18, 3.056303467019461e-05), (19, 0.0004701847501564771), (20, 0.00024508379283361137), (21, 2.6934591005556285e-05), (22, 0.00012888247147202492), (23, 6.85141421854496e-05), (24, 0.00015551579417660832), (25, 0.00033057009568437934), (26, 0.0006429178174585104), (27, 0.008983103558421135)], Temporal MSE: [(0, 1.3009768053962034e-06), (1, 4.7477112730121007e-07), (2, 6.443430606850598e-07), (3, 7.096721788002469e-07), (4, 4.176577874659415e-07), (5, 7.732585345365806e-07), (6, 7.276714200088463e-07), (7, 1.1041536254197126e-06), (8, 4.644662681130285e-07), (9, 6.803401220167871e-07), (10, 4.7259331950044725e-06), (11, 0.00011740813351934776), (12, 2.9112799893482588e-05), (13, 2.8181069865240715e-05), (14, 1.1852162970171776e-05), (15, 1.4546772035828326e-05), (16, 8.069235263974406e-06), (17, 7.48187903809594e-06), (18, 1.7014857803587802e-05), (19, 1.7625983673497103e-05), (20, 2.325806417502463e-05), (21, 2.3705350031377748e-05), (22, 2.485256663931068e-05), (23, 4.706410982180387e-05), (24, 5.550381320063025e-05), (25, 0.00016681416309438646), (26, 3.8333724660333246e-05), (27, 1.717976738291327e-05)]
+
+ 39%|███▉      | 59/150 [09:26<16:02, 10.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 540, Spatial MSE: [(0, 1.258638462786621e-06), (1, 2.1095038391649723e-05), (2, 9.860014688456431e-05), (3, 2.4344035409740172e-05), (4, 2.8530494091683067e-05), (5, 4.480764255276881e-05), (6, 0.0001588124578120187), (7, 0.0002675122523214668), (8, 0.0003503759508021176), (9, 0.0005150596844032407), (10, 0.000665715429931879), (11, 0.0006611421704292297), (12, 0.0007879201439209282), (13, 0.0042139748111367226), (14, 0.007746986113488674), (15, 0.004770252387970686), (16, 0.09788915514945984), (17, 0.0014203853206709027), (18, 0.0015876885736361146), (19, 0.007219144608825445), (20, 0.005692943464964628), (21, 0.00193983584176749), (22, 0.0017748145619407296), (23, 0.0020479310769587755), (24, 0.004512128420174122), (25, 0.00709904870018363), (26, 0.006264737341552973), (27, 0.009951828047633171)], Temporal MSE: [(0, 1.2622919030036428e-06), (1, 8.024998692235386e-07), (2, 2.570551032476942e-06), (3, 1.2967659131390974e-05), (4, 1.4317341083369683e-05), (5, 3.1330433557741344e-05), (6, 0.00012402342690620571), (7, 6.531432882184163e-05), (8, 4.0856917621567845e-05), (9, 7.792588439770043e-05), (10, 0.00015336062642745674), (11, 0.00028877341537736356), (12, 0.00022579115466214716), (13, 0.00039652668056078255), (14, 0.0005414995248429477), (15, 0.0005843418766744435), (16, 0.0006440727738663554), (17, 0.00058899907162413), (18, 0.000520759669598192), (19, 0.0007361408788710833), (20, 0.0011950463522225618), (21, 0.0009055752889253199), (22, 0.0006748426822014153), (23, 0.000975143862888217), (24, 0.002656355034559965), (25, 0.004500867333263159), (26, 0.0006843347800895572), (27, 4.209324106341228e-05)]
+
+ 40%|████      | 60/150 [09:36<15:21, 10.24s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 534, Spatial MSE: [(0, 7.995583359843295e-07), (1, 6.809611477365252e-06), (2, 3.120529800071381e-05), (3, 1.0898904292844236e-05), (4, 2.0757436232088367e-06), (5, 1.3265033658171888e-06), (6, 4.425568477017805e-06), (7, 2.9618904591188766e-06), (8, 1.7628802879698924e-06), (9, 2.0757518086611526e-06), (10, 3.160916548949899e-06), (11, 7.247214398375945e-06), (12, 2.7994296033284627e-05), (13, 0.0009175650775432587), (14, 0.0009114425629377365), (15, 0.00030526757473126054), (16, 0.009340493939816952), (17, 3.650850703706965e-05), (18, 5.101271017338149e-05), (19, 0.0008595766266807914), (20, 0.0005138331325724721), (21, 3.888843275490217e-05), (22, 0.00012368911120574921), (23, 6.767872400814667e-05), (24, 0.00016012009291443974), (25, 0.00037588796112686396), (26, 0.0005349613493308425), (27, 0.009019081480801105)], Temporal MSE: [(0, 8.487238005727704e-07), (1, 3.4987866115443467e-07), (2, 5.277898935673875e-07), (3, 6.257450309021806e-07), (4, 3.707304472300166e-07), (5, 6.805429393352824e-07), (6, 7.17523278126464e-07), (7, 8.568878229198162e-07), (8, 3.819672258487117e-07), (9, 4.882049893240037e-07), (10, 1.1343861388013465e-06), (11, 1.0094793651660439e-05), (12, 1.6961868823273107e-05), (13, 2.133088855771348e-05), (14, 1.3239447071100585e-05), (15, 1.3186626347305719e-05), (16, 7.804699635016732e-06), (17, 9.164390576188453e-06), (18, 2.6487108698347583e-05), (19, 2.1114819901413284e-05), (20, 1.9987801351817325e-05), (21, 2.4654096705489792e-05), (22, 3.183042645105161e-05), (23, 6.619667692575604e-05), (24, 6.256602500798181e-05), (25, 0.00014632796228397638), (26, 3.673457831609994e-05), (27, 2.429486994515173e-05)]
+
+ 41%|████      | 61/150 [09:43<14:03,  9.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 528, Spatial MSE: [(0, 1.1874922165588941e-06), (1, 1.3807065442961175e-05), (2, 5.6160413805628195e-05), (3, 1.3836833204550203e-05), (4, 5.746675924456213e-06), (5, 6.813814707129495e-06), (6, 1.717308441584464e-05), (7, 1.2888001037936192e-05), (8, 1.7153448425233364e-05), (9, 3.404303788556717e-05), (10, 4.958168574376032e-05), (11, 5.857853466295637e-05), (12, 8.596720726927742e-05), (13, 0.0012287377612665296), (14, 0.0017333102878183126), (15, 0.0009448518394492567), (16, 0.028145786374807358), (17, 0.00010376125283073634), (18, 0.0002240209432784468), (19, 0.0016968495910987258), (20, 0.0008566956385038793), (21, 0.00016003783093765378), (22, 0.0002236714499304071), (23, 0.00020571259665302932), (24, 0.0005312004941515625), (25, 0.0009791036136448383), (26, 0.002042425097897649), (27, 0.009217475540935993)], Temporal MSE: [(0, 9.343198144051712e-07), (1, 4.928623411615263e-07), (2, 1.0484671975063975e-06), (3, 1.7011655017995508e-06), (4, 2.2892274955665926e-06), (5, 2.447509359626565e-06), (6, 3.2211933103099e-06), (7, 3.7004022033215733e-06), (8, 2.9478437681973446e-06), (9, 3.6089807053940604e-06), (10, 6.458967618527822e-06), (11, 1.9044186046812683e-05), (12, 2.8566308174049482e-05), (13, 3.911445674020797e-05), (14, 3.591558925108984e-05), (15, 4.0379338315688074e-05), (16, 2.5218849259545095e-05), (17, 3.0695337045472115e-05), (18, 4.861025445279665e-05), (19, 4.388656816445291e-05), (20, 5.474078352563083e-05), (21, 7.257590186782181e-05), (22, 9.683051030151546e-05), (23, 0.00013242843851912767), (24, 0.0002251437836093828), (25, 0.0004003500798717141), (26, 0.00010999096411978826), (27, 2.9304255804163404e-05)]
+
+ 41%|████▏     | 62/150 [09:51<13:07,  8.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 522, Spatial MSE: [(0, 1.5014295513537945e-06), (1, 8.15608655102551e-06), (2, 3.5954926715930924e-05), (3, 1.0805857527884655e-05), (4, 2.304172767253476e-06), (5, 1.6368157957913354e-06), (6, 5.723321010009386e-06), (7, 3.56360396835953e-06), (8, 2.3527952635049587e-06), (9, 3.2899336019909242e-06), (10, 5.7729835134523455e-06), (11, 1.018350576487137e-05), (12, 1.4629352335759904e-05), (13, 0.00021546956850215793), (14, 0.0004839523753616959), (15, 0.00013318820856511593), (16, 0.009702298790216446), (17, 6.971281982259825e-05), (18, 6.476838461821899e-05), (19, 0.0009022494195960462), (20, 0.0006194431916810572), (21, 0.00021772108448203653), (22, 0.00013787952775601298), (23, 6.410977221094072e-05), (24, 0.0001441705389879644), (25, 0.0003419439890421927), (26, 0.0007291804067790508), (27, 0.006538779009133577)], Temporal MSE: [(0, 1.1688109680108028e-06), (1, 4.229526666676975e-07), (2, 5.607088269243832e-07), (3, 6.85382019582903e-07), (4, 4.304370975205529e-07), (5, 7.451382657563954e-07), (6, 7.081245030349237e-07), (7, 1.097839117392141e-06), (8, 4.6022952915336646e-07), (9, 7.046452878967102e-07), (10, 2.617536210891558e-06), (11, 5.509213679033564e-06), (12, 7.421040663757594e-06), (13, 7.29472503735451e-06), (14, 9.150139703706373e-06), (15, 1.6403686458943412e-05), (16, 1.0957523045362905e-05), (17, 9.997705092246179e-06), (18, 1.9176681234966964e-05), (19, 1.813294511521235e-05), (20, 1.9100692952633835e-05), (21, 3.361278140801005e-05), (22, 5.161517401575111e-05), (23, 6.753839261364192e-05), (24, 5.4817122872918844e-05), (25, 0.0002054268989013508), (26, 3.1939143809722736e-05), (27, 2.3231556042446755e-05)]
+
+ 42%|████▏     | 63/150 [10:00<12:57,  8.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 516, Spatial MSE: [(0, 1.430514316780318e-06), (1, 2.1215320884948596e-05), (2, 7.842414925107732e-05), (3, 2.1294408725225367e-05), (4, 2.548414886405226e-05), (5, 4.402991180540994e-05), (6, 0.00012326121213845909), (7, 0.0002328999398741871), (8, 0.00032310382812283933), (9, 0.00046536463196389377), (10, 0.0006754227797500789), (11, 0.000554929138161242), (12, 0.000638955389149487), (13, 0.0023376960307359695), (14, 0.005579766351729631), (15, 0.003617118811234832), (16, 0.08541730046272278), (17, 0.0012457316042855382), (18, 0.0018232217989861965), (19, 0.009501188062131405), (20, 0.009630382061004639), (21, 0.0021596301812678576), (22, 0.00156431517098099), (23, 0.001867476967163384), (24, 0.0039821648970246315), (25, 0.00620146282017231), (26, 0.006031418219208717), (27, 0.005431595724076033)], Temporal MSE: [(0, 1.2823448969356832e-06), (1, 6.996181127760792e-07), (2, 2.5901119897753233e-06), (3, 1.1907135558431037e-05), (4, 1.297361268370878e-05), (5, 2.491007762728259e-05), (6, 8.363067172467709e-05), (7, 5.395089101511985e-05), (8, 3.859875869238749e-05), (9, 7.501777145080268e-05), (10, 0.00013730823411606252), (11, 0.00015563861234113574), (12, 0.00017430860316380858), (13, 0.00031900135218165815), (14, 0.00046542889322154224), (15, 0.00047691745567135513), (16, 0.000527415017131716), (17, 0.0005043830024078488), (18, 0.00046625777031295), (19, 0.0007089772843755782), (20, 0.0011408051941543818), (21, 0.0007967790006659925), (22, 0.0006515841232612729), (23, 0.0009029844659380615), (24, 0.0025418540462851524), (25, 0.004577811807394028), (26, 0.000665519735775888), (27, 4.384627754916437e-05)]
+
+ 43%|████▎     | 64/150 [10:09<12:42,  8.86s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 510, Spatial MSE: [(0, 9.718621640786296e-07), (1, 7.430659024976194e-06), (2, 3.443852983764373e-05), (3, 1.0297561857441906e-05), (4, 2.284256879647728e-06), (5, 1.4865925095364219e-06), (6, 3.6859446481685154e-06), (7, 2.989017048093956e-06), (8, 1.9678786884469446e-06), (9, 2.739357569225831e-06), (10, 2.3125035113480408e-06), (11, 3.4290376333956374e-06), (12, 6.056198799342383e-06), (13, 0.00015173744759522378), (14, 0.0002802122326102108), (15, 0.00010836339788511395), (16, 0.009282580576837063), (17, 4.276198160368949e-05), (18, 3.805035157711245e-05), (19, 0.0008793537272140384), (20, 0.0006487848586402833), (21, 6.0562786529771984e-05), (22, 9.464249887969345e-05), (23, 5.1347735279705375e-05), (24, 0.00011394472676329315), (25, 0.00029859819915145636), (26, 0.00046433479292318225), (27, 0.005283868405967951)], Temporal MSE: [(0, 9.87294697551988e-07), (1, 4.163393327871745e-07), (2, 6.136418733149185e-07), (3, 6.931488769623684e-07), (4, 4.3737102828345087e-07), (5, 7.672222182009136e-07), (6, 8.962039714788261e-07), (7, 9.444809165870538e-07), (8, 4.412156329181016e-07), (9, 5.806896297144704e-07), (10, 9.777938885235926e-07), (11, 1.8953361404783209e-06), (12, 3.3759713460312923e-06), (13, 6.461958037107252e-06), (14, 1.0270523489452899e-05), (15, 8.81543837749632e-06), (16, 7.754805665172171e-06), (17, 1.3978354218124878e-05), (18, 1.7213982573593967e-05), (19, 1.536678973934613e-05), (20, 1.3043780199950561e-05), (21, 2.8871419999632053e-05), (22, 3.772975105675869e-05), (23, 5.1147078920621425e-05), (24, 4.07321349484846e-05), (25, 0.0001132563702412881), (26, 2.56084676948376e-05), (27, 2.123506237694528e-05)]
+
+ 43%|████▎     | 65/150 [10:18<12:40,  8.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 504, Spatial MSE: [(0, 1.1704689768521348e-06), (1, 1.1538428225321695e-05), (2, 5.243288978817873e-05), (3, 1.1673437256831676e-05), (4, 5.354248969524633e-06), (5, 6.867173851787811e-06), (6, 1.835783405113034e-05), (7, 1.4120405467110686e-05), (8, 1.6608941223239526e-05), (9, 3.2326075597666204e-05), (10, 4.785568671650253e-05), (11, 6.428470078390092e-05), (12, 6.400170968845487e-05), (13, 0.0003093498235102743), (14, 0.0006106259534135461), (15, 0.00036353131872601807), (16, 0.010759557597339153), (17, 9.068888903129846e-05), (18, 7.433342398144305e-05), (19, 0.000890751660335809), (20, 0.000761771691031754), (21, 0.00010868666868191212), (22, 0.0001543916150694713), (23, 0.00015339370293077081), (24, 0.0004693996743299067), (25, 0.0010437044547870755), (26, 0.002920589642599225), (27, 0.0054476819932460785)], Temporal MSE: [(0, 8.095799444163276e-07), (1, 4.2806124156413716e-07), (2, 9.625504162613652e-07), (3, 1.5944465303618927e-06), (4, 2.1720429685956333e-06), (5, 2.4750170268816873e-06), (6, 3.5276491416880162e-06), (7, 4.182604698144132e-06), (8, 3.250849886171636e-06), (9, 3.762836058740504e-06), (10, 6.508292244689073e-06), (11, 8.828618774714414e-06), (12, 1.2119829989387654e-05), (13, 2.0576140741468407e-05), (14, 2.560454413469415e-05), (15, 2.1299045329215005e-05), (16, 1.7118150935857557e-05), (17, 1.978927684831433e-05), (18, 3.098538945778273e-05), (19, 3.635773100540973e-05), (20, 4.874770456808619e-05), (21, 4.906997855869122e-05), (22, 6.50694637442939e-05), (23, 8.730008266866207e-05), (24, 0.00019038435129914433), (25, 0.0003880893927998841), (26, 0.00010018960165325552), (27, 2.4747410861891694e-05)]
+
+ 44%|████▍     | 66/150 [10:27<12:40,  9.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 498, Spatial MSE: [(0, 8.612960868958908e-07), (1, 6.952298008400248e-06), (2, 2.5821749659371562e-05), (3, 9.293923540099058e-06), (4, 1.9094638901151484e-06), (5, 1.2760471008732566e-06), (6, 3.563708105502883e-06), (7, 2.749615305219777e-06), (8, 1.766219384080614e-06), (9, 2.0289546682761284e-06), (10, 2.0425190996320453e-06), (11, 3.963072686019586e-06), (12, 1.1510886906762607e-05), (13, 0.0003465248446445912), (14, 0.0002720138872973621), (15, 0.0001438660838175565), (16, 0.006651872303336859), (17, 4.271225770935416e-05), (18, 0.00010163740807911381), (19, 0.00040978440665639937), (20, 0.00025158459902741015), (21, 3.7904930650256574e-05), (22, 0.00013143867545295507), (23, 5.848557339049876e-05), (24, 0.0001298243587370962), (25, 0.0002873617922887206), (26, 0.000500763300806284), (27, 0.003406187752261758)], Temporal MSE: [(0, 7.694483201703406e-07), (1, 3.20303740863892e-07), (2, 4.689362356202764e-07), (3, 5.829629117215518e-07), (4, 3.8136838043101307e-07), (5, 6.419152214220958e-07), (6, 6.893834552101907e-07), (7, 8.402091111747723e-07), (8, 3.8173200778146565e-07), (9, 4.989275907973933e-07), (10, 8.686068895258359e-07), (11, 5.575459908868652e-06), (12, 8.48196850711247e-06), (13, 8.378480742976535e-06), (14, 7.368515980488155e-06), (15, 1.2769520253641531e-05), (16, 5.956894710834604e-06), (17, 1.0095967809320427e-05), (18, 1.1851999261125457e-05), (19, 9.009839232021477e-06), (20, 8.6294685388566e-06), (21, 1.744887777022086e-05), (22, 3.0419272661674768e-05), (23, 5.4916654335102066e-05), (24, 5.07445729454048e-05), (25, 0.0001216678720084019), (26, 2.8613960239454173e-05), (27, 1.1063374586228747e-05)]
+
+ 45%|████▍     | 67/150 [10:36<12:34,  9.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 492, Spatial MSE: [(0, 1.0992710031132447e-06), (1, 1.8274420654051937e-05), (2, 6.168113031890243e-05), (3, 1.9129276552121155e-05), (4, 2.283913090650458e-05), (5, 4.1275619878433645e-05), (6, 9.271843009628356e-05), (7, 0.00019719678675755858), (8, 0.00023218858405016363), (9, 0.0003567304229363799), (10, 0.0004492776934057474), (11, 0.000404312857426703), (12, 0.0004944480606354773), (13, 0.001889878185465932), (14, 0.0033411914482712746), (15, 0.0022528928238898516), (16, 0.032991956919431686), (17, 0.0009973123669624329), (18, 0.001015502493828535), (19, 0.0028248135931789875), (20, 0.0026971078477799892), (21, 0.0013533169403672218), (22, 0.0011927739251405), (23, 0.0014946793671697378), (24, 0.003219767240807414), (25, 0.0052344719879329205), (26, 0.005283128470182419), (27, 0.0033872618805617094)], Temporal MSE: [(0, 1.0494931075299974e-06), (1, 5.987697022646898e-07), (2, 2.181905529141659e-06), (3, 1.2787142622983083e-05), (4, 1.1879656995006371e-05), (5, 2.2791806259192526e-05), (6, 7.28016093489714e-05), (7, 4.637258825823665e-05), (8, 3.3027055906131864e-05), (9, 6.245182157726958e-05), (10, 0.00012151334522059187), (11, 0.0001416645827703178), (12, 0.0001496607146691531), (13, 0.00025001249741762877), (14, 0.00037443972541950643), (15, 0.000387595355277881), (16, 0.00046518840827047825), (17, 0.00043554112198762596), (18, 0.000366751104593277), (19, 0.0005600980366580188), (20, 0.0009153560386039317), (21, 0.0006399449775926769), (22, 0.0004964553518220782), (23, 0.0007207434973679483), (24, 0.002052662428468466), (25, 0.003530723275616765), (26, 0.0006138497847132385), (27, 3.1278108508558944e-05)]
+
+ 45%|████▌     | 68/150 [10:45<12:03,  8.83s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 486, Spatial MSE: [(0, 9.571053851686884e-07), (1, 7.41222538636066e-06), (2, 2.804153882607352e-05), (3, 1.0216191185463686e-05), (4, 1.9604126464400906e-06), (5, 1.5029955875434098e-06), (6, 3.6054877909919014e-06), (7, 3.977374944952317e-06), (8, 2.1581124656222528e-06), (9, 2.389703013250255e-06), (10, 6.102138740970986e-06), (11, 2.889771349146031e-05), (12, 6.183057121234015e-05), (13, 0.001271908637136221), (14, 0.0011263983324170113), (15, 0.0003656463813968003), (16, 0.009581553749740124), (17, 3.442511660978198e-05), (18, 0.00013798181316815317), (19, 0.0005413988255895674), (20, 0.00024280790239572525), (21, 5.377936759032309e-05), (22, 0.00014426439884118736), (23, 8.33305311971344e-05), (24, 0.00015655611059628427), (25, 0.00029990565963089466), (26, 0.0006710263551212847), (27, 0.009017557837069035)], Temporal MSE: [(0, 8.609505357526359e-07), (1, 3.710964620040613e-07), (2, 5.390375008573756e-07), (3, 6.967655394873873e-07), (4, 4.664646837682085e-07), (5, 7.78404796619725e-07), (6, 8.881002599991916e-07), (7, 1.09924678781681e-06), (8, 4.7049917384356377e-07), (9, 6.444480504796957e-07), (10, 2.940883632618352e-06), (11, 6.668540299870074e-05), (12, 2.5385190383531153e-05), (13, 2.7777921786764637e-05), (14, 1.2064720976923127e-05), (15, 2.0997273168177344e-05), (16, 9.462160960538313e-06), (17, 1.0268557161907665e-05), (18, 2.5375356926815584e-05), (19, 1.5884850654401816e-05), (20, 1.5241867004078813e-05), (21, 2.3742350094835274e-05), (22, 2.8800417567254044e-05), (23, 7.311708031920716e-05), (24, 7.535263284808025e-05), (25, 0.00023050773597788066), (26, 5.172863166080788e-05), (27, 2.1495525288628414e-05)]
+
+ 46%|████▌     | 69/150 [10:52<11:27,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 480, Spatial MSE: [(0, 1.1412662388465833e-06), (1, 1.032277214108035e-05), (2, 3.8635156670352444e-05), (3, 8.638065992272459e-06), (4, 5.469269126479048e-06), (5, 6.456872142734937e-06), (6, 2.1410431145341136e-05), (7, 1.4537000424752478e-05), (8, 1.8654956875252537e-05), (9, 4.199432805762626e-05), (10, 5.92343567404896e-05), (11, 9.60796169238165e-05), (12, 0.00011597530829021707), (13, 0.001329556223936379), (14, 0.0012008878402411938), (15, 0.00040537616587243974), (16, 0.009607241488993168), (17, 5.819321086164564e-05), (18, 0.00010957127233268693), (19, 0.0009814932709559798), (20, 0.0006900148000568151), (21, 0.00015924764738883823), (22, 0.00019581960805226117), (23, 0.0001564257690915838), (24, 0.0004153118352405727), (25, 0.0008557586115784943), (26, 0.0022033737041056156), (27, 0.008752929046750069)], Temporal MSE: [(0, 6.564711156897829e-07), (1, 3.694797783282411e-07), (2, 9.466024835091957e-07), (3, 1.6061892438301584e-06), (4, 2.1828727767569944e-06), (5, 2.63526635535527e-06), (6, 3.7471679661393864e-06), (7, 4.497713234741241e-06), (8, 3.194668579453719e-06), (9, 4.1426264942856506e-06), (10, 8.471968612866476e-06), (11, 7.29808525647968e-05), (12, 3.2767940865596756e-05), (13, 3.810414273175411e-05), (14, 2.564008900662884e-05), (15, 2.9525195714086294e-05), (16, 1.8311569874640554e-05), (17, 1.7789187040762044e-05), (18, 4.1392602724954486e-05), (19, 4.3938212911598384e-05), (20, 5.798042911919765e-05), (21, 4.23124511144124e-05), (22, 5.301819328451529e-05), (23, 9.766189032234251e-05), (24, 0.00019161344971507788), (25, 0.00048016884829849005), (26, 0.00013367454812396318), (27, 2.665573265403509e-05)]
+
+ 47%|████▋     | 70/150 [11:01<11:18,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 474, Spatial MSE: [(0, 1.2795350130545557e-06), (1, 7.08339848642936e-06), (2, 2.5583778551663272e-05), (3, 9.164503353531472e-06), (4, 2.05003948394733e-06), (5, 1.4024230949871708e-06), (6, 3.957829903811216e-06), (7, 3.4807023894245503e-06), (8, 1.9787526071013417e-06), (9, 2.0117843178013572e-06), (10, 2.7605151444731746e-06), (11, 4.651652943721274e-06), (12, 1.3178980225347914e-05), (13, 0.00040248219738714397), (14, 0.000438929331721738), (15, 0.000193346495507285), (16, 0.006091471761465073), (17, 1.9766526747844182e-05), (18, 2.64996342593804e-05), (19, 0.0003995677689090371), (20, 0.0001821486366679892), (21, 3.250703593948856e-05), (22, 9.904285980155692e-05), (23, 5.770795905846171e-05), (24, 0.00010314701648894697), (25, 0.000235280254855752), (26, 0.00048477918608114123), (27, 0.006161464378237724)], Temporal MSE: [(0, 1.028131237035268e-06), (1, 3.4391425174362666e-07), (2, 5.044532258580148e-07), (3, 6.615514962504676e-07), (4, 4.357484328920691e-07), (5, 6.865054160698492e-07), (6, 8.017062782528228e-07), (7, 9.821480944083305e-07), (8, 4.161209687936207e-07), (9, 5.594103527073457e-07), (10, 1.1999011348962085e-06), (11, 4.6516065594914835e-06), (12, 6.713547463732539e-06), (13, 1.0733056114986539e-05), (14, 7.206682766991435e-06), (15, 1.2393134966259822e-05), (16, 6.321452474367106e-06), (17, 8.276428161479998e-06), (18, 2.223372575826943e-05), (19, 1.9323601009091362e-05), (20, 2.441779179207515e-05), (21, 1.8156477381126024e-05), (22, 1.9224011339247227e-05), (23, 4.3507701775524765e-05), (24, 4.9135818699141964e-05), (25, 0.00019099298515357077), (26, 3.826914689852856e-05), (27, 1.6700063497410156e-05)]
+
+ 47%|████▋     | 71/150 [11:08<10:43,  8.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 468, Spatial MSE: [(0, 1.2377849998301826e-06), (1, 1.6894618966034614e-05), (2, 6.0957867390243337e-05), (3, 1.9494382286211476e-05), (4, 2.54346668953076e-05), (5, 4.123766484553926e-05), (6, 8.426238491665572e-05), (7, 0.00021026648755650967), (8, 0.0002016523212660104), (9, 0.00031892023980617523), (10, 0.00043176463805139065), (11, 0.00039187175570987165), (12, 0.00044996285578235984), (13, 0.0008955653174780309), (14, 0.0012713561300188303), (15, 0.0011957826791331172), (16, 0.007253627292811871), (17, 0.0008361224899999797), (18, 0.0008707426604814827), (19, 0.0011719315079972148), (20, 0.0013379516312852502), (21, 0.0011781810317188501), (22, 0.0010320032015442848), (23, 0.0014047030126675963), (24, 0.0029836527537554502), (25, 0.004992845002561808), (26, 0.006562232971191406), (27, 0.00573035329580307)], Temporal MSE: [(0, 1.0606047453620704e-06), (1, 5.834093030898657e-07), (2, 2.2919762159290258e-06), (3, 1.3120085895934608e-05), (4, 1.1403200005588587e-05), (5, 2.355861579417251e-05), (6, 6.866125477245077e-05), (7, 4.583769623422995e-05), (8, 3.08599992422387e-05), (9, 5.796774348709732e-05), (10, 0.00011268674279563129), (11, 0.00012090601376257837), (12, 0.00011788932897616178), (13, 0.00021445805032271892), (14, 0.0003444274188950658), (15, 0.00036924067535437644), (16, 0.0004505989491008222), (17, 0.0003993498394265771), (18, 0.0003306573489680886), (19, 0.0005255102878436446), (20, 0.0008995013195089996), (21, 0.0006102502811700106), (22, 0.0004514635365922004), (23, 0.0006426284089684486), (24, 0.0021644115913659334), (25, 0.0036117881536483765), (26, 0.0006174911977723241), (27, 3.8078567740740255e-05)]
+
+ 48%|████▊     | 72/150 [11:17<10:45,  8.28s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 462, Spatial MSE: [(0, 8.204731898331374e-07), (1, 6.83984126226278e-06), (2, 2.16349490074208e-05), (3, 8.998847988550551e-06), (4, 1.6526973922736943e-06), (5, 1.3472515547618968e-06), (6, 2.9948685096314875e-06), (7, 2.534944769649883e-06), (8, 1.971574420167599e-06), (9, 2.8473771180870244e-06), (10, 8.276014341390692e-06), (11, 5.927021993556991e-05), (12, 0.00010936300532193854), (13, 0.00203112605959177), (14, 0.0016233240021392703), (15, 0.0004133176407776773), (16, 0.009556199423968792), (17, 4.1516104829497635e-05), (18, 2.7141515602124855e-05), (19, 0.0008085800800472498), (20, 0.0004299802239984274), (21, 4.204233118798584e-05), (22, 0.00012784788850694895), (23, 7.830774848116562e-05), (24, 0.00016471408889628947), (25, 0.00029278258443810046), (26, 0.0004985978011973202), (27, 0.01285475306212902)], Temporal MSE: [(0, 6.551293836309924e-07), (1, 2.777862277980603e-07), (2, 4.388505772112694e-07), (3, 6.165551553749538e-07), (4, 4.5240972212923225e-07), (5, 7.406063105008798e-07), (6, 8.701832143742649e-07), (7, 8.69116092871991e-07), (8, 4.670121995786758e-07), (9, 6.817467124164978e-07), (10, 3.8057078199926764e-06), (11, 0.0001335731940343976), (12, 4.2559542634990066e-05), (13, 3.810640919255093e-05), (14, 1.2570723811222706e-05), (15, 2.3952066840138286e-05), (16, 1.0525319339649286e-05), (17, 1.1227457434870303e-05), (18, 3.4342865546932444e-05), (19, 2.935656266345177e-05), (20, 2.389430665061809e-05), (21, 2.2318978153634816e-05), (22, 2.5038425519596785e-05), (23, 6.745155405951664e-05), (24, 7.561413804069161e-05), (25, 0.0002705323277041316), (26, 5.797177436761558e-05), (27, 3.166529495501891e-05)]
+
+ 49%|████▊     | 73/150 [11:24<10:16,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 456, Spatial MSE: [(0, 1.104029706766596e-06), (1, 8.390809853153769e-06), (2, 2.8445483621908352e-05), (3, 9.96365906758001e-06), (4, 4.760444880957948e-06), (5, 6.530367500090506e-06), (6, 1.8968803487950936e-05), (7, 1.189138311019633e-05), (8, 1.6986907212412916e-05), (9, 3.0415192668442614e-05), (10, 4.8015961510827765e-05), (11, 0.00010923772060777992), (12, 0.00016006302030291408), (13, 0.0020809448324143887), (14, 0.0016948375850915909), (15, 0.00044881878420710564), (16, 0.009579207748174667), (17, 6.568647222593427e-05), (18, 5.645257988362573e-05), (19, 0.0008315053419210017), (20, 0.0004591920005623251), (21, 6.754724745405838e-05), (22, 0.0001798962475731969), (23, 0.00016065625823102891), (24, 0.0004204152792226523), (25, 0.0008913655765354633), (26, 0.0024590925313532352), (27, 0.013002322055399418)], Temporal MSE: [(0, 6.680477895315562e-07), (1, 3.7995698676240863e-07), (2, 8.161872528944514e-07), (3, 1.7208626559295226e-06), (4, 2.4523624233552255e-06), (5, 2.749278792180121e-06), (6, 4.28477596869925e-06), (7, 4.738012194138719e-06), (8, 3.211104285583133e-06), (9, 3.856838702631649e-06), (10, 8.70207713887794e-06), (11, 0.00014155247481539845), (12, 4.9794492952059954e-05), (13, 4.880041888100095e-05), (14, 2.700102595554199e-05), (15, 3.253201793995686e-05), (16, 1.8627470126375556e-05), (17, 1.8552233086666092e-05), (18, 4.44301767856814e-05), (19, 4.188741513644345e-05), (20, 4.386082582641393e-05), (21, 3.8613765354966745e-05), (22, 5.2390572818694636e-05), (23, 9.824164590099826e-05), (24, 0.0001896565081551671), (25, 0.000507872027810663), (26, 0.00013908658002037555), (27, 3.7996022001607344e-05)]
+
+ 49%|████▉     | 74/150 [11:31<09:53,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 450, Spatial MSE: [(0, 7.499022558477009e-07), (1, 6.981813385209534e-06), (2, 1.9910048649762757e-05), (3, 8.265159522125032e-06), (4, 1.8458274553267984e-06), (5, 1.2749559346048045e-06), (6, 2.503662699382403e-06), (7, 2.4330283849849366e-06), (8, 1.6399426385760307e-06), (9, 2.180132696594228e-06), (10, 2.801888513204176e-06), (11, 5.419597528089071e-06), (12, 1.7228499928023666e-05), (13, 0.0006706746062263846), (14, 0.0007263941806741059), (15, 0.00030235867598094046), (16, 0.00750711327418685), (17, 2.798486821120605e-05), (18, 1.938787681865506e-05), (19, 0.0004980487865395844), (20, 0.00025519111659377813), (21, 2.9304877898539416e-05), (22, 0.00010356185521231964), (23, 6.048072100384161e-05), (24, 0.00010556814231676981), (25, 0.0002123517042491585), (26, 0.00040712335612624884), (27, 0.007905958220362663)], Temporal MSE: [(0, 6.761899840057595e-07), (1, 2.8539631102830754e-07), (2, 4.397257384880504e-07), (3, 6.169860853333375e-07), (4, 4.3622205225801736e-07), (5, 6.690838176837133e-07), (6, 7.267099135788158e-07), (7, 8.606253913967521e-07), (8, 4.2015878420897934e-07), (9, 5.464137871058483e-07), (10, 1.2276606184968841e-06), (11, 5.399798283178825e-06), (12, 9.241549378202762e-06), (13, 1.813524795579724e-05), (14, 9.022859558172058e-06), (15, 1.6602478353888728e-05), (16, 7.462292160198558e-06), (17, 9.661560397944413e-06), (18, 2.5132132577709854e-05), (19, 2.1881360225961544e-05), (20, 2.0775758457602933e-05), (21, 2.015853715420235e-05), (22, 1.9592389435274526e-05), (23, 4.849068864132278e-05), (24, 5.386911652749404e-05), (25, 0.00020248662622179836), (26, 4.0537674067309126e-05), (27, 2.2719063053955324e-05)]
+
+ 50%|█████     | 75/150 [11:39<09:45,  7.81s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 444, Spatial MSE: [(0, 1.1718515224856674e-06), (1, 1.4641796951764263e-05), (2, 4.756106500281021e-05), (3, 1.77329402504256e-05), (4, 2.6492572942515835e-05), (5, 4.3423264287412167e-05), (6, 9.596248128218576e-05), (7, 0.00022466368682216853), (8, 0.0001950179721461609), (9, 0.00031349947676062584), (10, 0.0003824258456006646), (11, 0.00038739299634471536), (12, 0.0004454180598258972), (13, 0.0013271638890728354), (14, 0.0020947305019944906), (15, 0.0016220358666032553), (16, 0.019455038011074066), (17, 0.0008387856069020927), (18, 0.0008501469856128097), (19, 0.0022742440924048424), (20, 0.0022182553075253963), (21, 0.0011355356546118855), (22, 0.0009851917857304215), (23, 0.0013637493830174208), (24, 0.0027507543563842773), (25, 0.00473405746743083), (26, 0.0070186033844947815), (27, 0.008581351488828659)], Temporal MSE: [(0, 9.884429346129764e-07), (1, 5.119676984577382e-07), (2, 2.520539055694826e-06), (3, 1.2072647223249078e-05), (4, 1.2295221495151054e-05), (5, 2.4289998691529036e-05), (6, 6.757473602192476e-05), (7, 4.861916750087403e-05), (8, 3.0473109291051514e-05), (9, 5.4389878641813993e-05), (10, 9.969012899091467e-05), (11, 0.00011419223301345482), (12, 0.00011663616169244051), (13, 0.00021992319670971483), (14, 0.0003447599592618644), (15, 0.0003780716215260327), (16, 0.0004489567072596401), (17, 0.0003968826204072684), (18, 0.00033569999504834414), (19, 0.0005237848963588476), (20, 0.0008992175571620464), (21, 0.0006012184894643724), (22, 0.0004358798614703119), (23, 0.0006224978715181351), (24, 0.001742842490784824), (25, 0.002929678652435541), (26, 0.0006243335083127022), (27, 4.740370422950946e-05)]
+
+ 51%|█████     | 76/150 [11:47<09:44,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 438, Spatial MSE: [(0, 1.245820953954535e-06), (1, 7.123301202227594e-06), (2, 2.003651206905488e-05), (3, 9.076886271941476e-06), (4, 1.7529971501062391e-06), (5, 1.3456136684908415e-06), (6, 2.756082039923058e-06), (7, 3.143554295093054e-06), (8, 1.9642093320726417e-06), (9, 2.057617848549853e-06), (10, 2.8944803034391953e-06), (11, 1.581663491379004e-05), (12, 4.267429540050216e-05), (13, 0.0009038630523718894), (14, 0.0007844485226087272), (15, 0.0002023990236921236), (16, 0.004648599773645401), (17, 2.3752902052365243e-05), (18, 2.790986945910845e-05), (19, 0.00039473874494433403), (20, 0.00019784066535066813), (21, 3.551216286723502e-05), (22, 8.797520422376692e-05), (23, 5.134175444254652e-05), (24, 0.00010332951933378354), (25, 0.0002160102449124679), (26, 0.0004976625787094235), (27, 0.007072066888213158)], Temporal MSE: [(0, 8.891323659554473e-07), (1, 2.9758913910882256e-07), (2, 4.607441610460228e-07), (3, 6.967175067984499e-07), (4, 4.897365215583704e-07), (5, 7.597677722515073e-07), (6, 8.535444635526801e-07), (7, 9.816969850362511e-07), (8, 4.858728743784013e-07), (9, 6.431857855204726e-07), (10, 1.3101988542985055e-06), (11, 3.6317647754913196e-05), (12, 2.1771646061097272e-05), (13, 1.9176311980118044e-05), (14, 8.34148158901371e-06), (15, 1.2658460036618635e-05), (16, 6.798698905186029e-06), (17, 6.594165370188421e-06), (18, 1.8363320123171434e-05), (19, 1.8623873984324746e-05), (20, 1.4854083929094486e-05), (21, 1.3827459952153731e-05), (22, 1.7446098354412243e-05), (23, 4.247383549227379e-05), (24, 4.6799359552096575e-05), (25, 0.00018580157484393567), (26, 4.256911779521033e-05), (27, 1.8603332136990502e-05)]
+
+ 51%|█████▏    | 77/150 [11:56<09:42,  7.97s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 432, Spatial MSE: [(0, 1.2516576362031628e-06), (1, 9.414649866812397e-06), (2, 3.015365291503258e-05), (3, 8.148408596753143e-06), (4, 4.105196239834186e-06), (5, 5.948151738266461e-06), (6, 1.5876237739576027e-05), (7, 9.771648365131114e-06), (8, 1.5319970771088265e-05), (9, 3.50398950104136e-05), (10, 3.6677596654044464e-05), (11, 6.396355456672609e-05), (12, 8.764257654547691e-05), (13, 0.0009610499255359173), (14, 0.0008616133709438145), (15, 0.00025359317078255117), (16, 0.004832210950553417), (17, 4.552190875983797e-05), (18, 7.546848064521328e-05), (19, 0.0004256614192854613), (20, 0.00022943814110476524), (21, 5.7896926591638476e-05), (22, 0.00014331532292999327), (23, 0.00013097688497509807), (24, 0.00035332379047758877), (25, 0.0008985059102997184), (26, 0.0025925133377313614), (27, 0.006933656986802816)], Temporal MSE: [(0, 6.967690637793567e-07), (1, 3.76479675878727e-07), (2, 8.984641794995696e-07), (3, 1.5392192835861351e-06), (4, 2.2743533918401226e-06), (5, 2.547143367337412e-06), (6, 4.508082838583505e-06), (7, 4.604253717843676e-06), (8, 3.3306948807876324e-06), (9, 3.534109055181034e-06), (10, 6.004989700159058e-06), (11, 4.238397741573863e-05), (12, 2.8597729397006333e-05), (13, 2.9664604880963452e-05), (14, 2.3251161110238172e-05), (15, 2.2769911083742045e-05), (16, 1.5711199012002908e-05), (17, 1.5900184735073708e-05), (18, 2.8889340683235787e-05), (19, 3.2432650186819956e-05), (20, 3.5012784792343155e-05), (21, 3.004550490004476e-05), (22, 4.38453171227593e-05), (23, 7.118636858649552e-05), (24, 0.00014827679842710495), (25, 0.0004171506443526596), (26, 0.0001282783196074888), (27, 2.4949391445261426e-05)]
+
+ 52%|█████▏    | 78/150 [12:04<09:36,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 426, Spatial MSE: [(0, 5.699412213289179e-07), (1, 5.767944458057173e-06), (2, 1.4751769413123839e-05), (3, 7.980855116329622e-06), (4, 1.542581117064401e-06), (5, 1.2085571370334947e-06), (6, 1.8194530184700852e-06), (7, 2.4784260403976077e-06), (8, 1.5010780316515593e-06), (9, 1.6606712733846507e-06), (10, 1.7284287423535716e-06), (11, 2.1776866105938097e-06), (12, 4.020775122626219e-06), (13, 5.0532336899777874e-05), (14, 3.629796628956683e-05), (15, 1.0463143553351983e-05), (16, 1.865744161477778e-05), (17, 5.343241355149075e-06), (18, 5.6856915762182325e-06), (19, 1.3017461242270656e-05), (20, 6.411274625861552e-06), (21, 9.195574421028141e-06), (22, 3.5527657018974423e-05), (23, 2.3001735826255754e-05), (24, 4.084024840267375e-05), (25, 9.291259630117565e-05), (26, 0.000265202164882794), (27, 0.0013063179794698954)], Temporal MSE: [(0, 5.343168822946609e-07), (1, 2.4026596179282933e-07), (2, 3.926325007341802e-07), (3, 5.91876982980466e-07), (4, 4.6173707346497395e-07), (5, 6.851550438113918e-07), (6, 9.030310934576846e-07), (7, 8.576623713452136e-07), (8, 4.253330700976221e-07), (9, 5.000770215701777e-07), (10, 8.16371596101817e-07), (11, 1.3083231351629365e-06), (12, 1.6479469877594966e-06), (13, 1.8384630493528675e-06), (14, 1.9606525256676832e-06), (15, 1.4075719718675828e-06), (16, 1.5180391983449226e-06), (17, 1.3790286175208166e-06), (18, 2.196514515162562e-06), (19, 2.643183734107879e-06), (20, 3.1457748264074326e-06), (21, 5.32486319571035e-06), (22, 6.40156304143602e-06), (23, 1.0958374332403764e-05), (24, 1.5802350390003994e-05), (25, 4.3930944229941815e-05), (26, 1.7481106624472886e-05), (27, 1.3369694897846784e-06)]
+
+ 53%|█████▎    | 79/150 [12:11<09:25,  7.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 420, Spatial MSE: [(0, 1.1326092135277577e-06), (1, 1.4415257282962557e-05), (2, 3.427750925766304e-05), (3, 1.6288689948851243e-05), (4, 2.0263358237571083e-05), (5, 3.973721686634235e-05), (6, 6.323094567051157e-05), (7, 0.00016310258070006967), (8, 0.00015213860024232417), (9, 0.0002238075976492837), (10, 0.0002250104589620605), (11, 0.000247299816692248), (12, 0.0003216673794668168), (13, 0.0012681807857006788), (14, 0.0031456872820854187), (15, 0.002163136610761285), (16, 0.05125665292143822), (17, 0.0006485026096925139), (18, 0.0009084360208362341), (19, 0.0034769882913678885), (20, 0.0030140497256070375), (21, 0.001024073688313365), (22, 0.0008517314563505352), (23, 0.0012380880070850253), (24, 0.002452652668580413), (25, 0.0042680976912379265), (26, 0.006583838723599911), (27, 0.002388397930189967)], Temporal MSE: [(0, 9.394067888024438e-07), (1, 5.715833708563878e-07), (2, 2.0709753698611166e-06), (3, 1.417671865056036e-05), (4, 1.0374577868788037e-05), (5, 1.9702674762811512e-05), (6, 4.586705472320318e-05), (7, 3.690801531774923e-05), (8, 2.4448179829050787e-05), (9, 4.063055166625418e-05), (10, 7.669122715014964e-05), (11, 8.739057375350967e-05), (12, 9.833029616856948e-05), (13, 0.00017264085181523114), (14, 0.00026969611644744873), (15, 0.00028983221272937953), (16, 0.00032853533048182726), (17, 0.00028919658507220447), (18, 0.0002560209250077605), (19, 0.0003888616047333926), (20, 0.0006404782179743052), (21, 0.0004705312894657254), (22, 0.0004152262699790299), (23, 0.000556229380890727), (24, 0.0014696706784889102), (25, 0.0025089425034821033), (26, 0.0005713541759178042), (27, 2.9356924642343074e-05)]
+
+ 53%|█████▎    | 80/150 [12:20<09:20,  8.01s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 414, Spatial MSE: [(0, 8.353566727237194e-07), (1, 6.659558493993245e-06), (2, 1.941810842254199e-05), (3, 8.516420166415628e-06), (4, 1.6001213225536048e-06), (5, 1.3121498341206461e-06), (6, 1.924383923324058e-06), (7, 2.1730543267040048e-06), (8, 1.6096065564852324e-06), (9, 2.233409759355709e-06), (10, 1.6713779587007593e-06), (11, 2.5411175101908157e-06), (12, 4.053339125675848e-06), (13, 7.246132008731365e-05), (14, 9.506777132628486e-05), (15, 0.00010624681453919038), (16, 0.004533478058874607), (17, 2.4050581487244926e-05), (18, 7.058567280182615e-05), (19, 0.00041365480865351856), (20, 0.0003050916420761496), (21, 3.0201692425180227e-05), (22, 5.503333159140311e-05), (23, 3.743551133084111e-05), (24, 8.931376942200586e-05), (25, 0.000198545545572415), (26, 0.0003872205561492592), (27, 0.0032719008158892393)], Temporal MSE: [(0, 6.27623819582368e-07), (1, 2.682332080894412e-07), (2, 4.4416711375561135e-07), (3, 6.78517949381785e-07), (4, 5.028317673350102e-07), (5, 7.457022661583323e-07), (6, 8.322382996084343e-07), (7, 8.457220133095689e-07), (8, 4.6320667479449185e-07), (9, 5.366322284316993e-07), (10, 8.346835898009886e-07), (11, 1.4105536365605076e-06), (12, 1.7731616708260844e-06), (13, 3.1406443667947315e-06), (14, 4.457338036445435e-06), (15, 6.400551683327649e-06), (16, 7.520724921050714e-06), (17, 1.4890048987581395e-05), (18, 1.5401721611851826e-05), (19, 9.087047146749683e-06), (20, 8.512762178725097e-06), (21, 1.8104434275301173e-05), (22, 2.186262463510502e-05), (23, 3.237364944652654e-05), (24, 3.128728349111043e-05), (25, 9.027609485201538e-05), (26, 2.8053027563146316e-05), (27, 1.269498716283124e-05)]
+
+ 54%|█████▍    | 81/150 [12:27<09:02,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 408, Spatial MSE: [(0, 9.779535048437538e-07), (1, 7.90533704275731e-06), (2, 2.2514694137498736e-05), (3, 6.8411941356316675e-06), (4, 4.49542494607158e-06), (5, 5.927623988100095e-06), (6, 1.7543639842187986e-05), (7, 1.1318705219309777e-05), (8, 1.4863333490211517e-05), (9, 2.6160652851103805e-05), (10, 3.9626371290069073e-05), (11, 5.439096275949851e-05), (12, 5.562963997363113e-05), (13, 0.00019710320339072496), (14, 0.0004135640920139849), (15, 0.0004135559138376266), (16, 0.010290663689374924), (17, 5.5138854804681614e-05), (18, 0.00035224316525273025), (19, 0.0013110035797581077), (20, 0.0015294213080778718), (21, 0.0002648387453518808), (22, 0.0001397186133544892), (23, 0.0001360760215902701), (24, 0.00039749665302224457), (25, 0.0009816591627895832), (26, 0.0022960645146667957), (27, 0.002696525538340211)], Temporal MSE: [(0, 6.693247769362642e-07), (1, 3.399315460228536e-07), (2, 8.689156061336689e-07), (3, 1.582576373948541e-06), (4, 2.3808613605069695e-06), (5, 2.639739477672265e-06), (6, 4.054516466567293e-06), (7, 4.580854692903813e-06), (8, 2.931538574557635e-06), (9, 3.5269424643047387e-06), (10, 5.799786322313594e-06), (11, 8.26272662379779e-06), (12, 1.0217118870059494e-05), (13, 1.454775883757975e-05), (14, 2.0764458895428106e-05), (15, 2.4106937416945584e-05), (16, 1.640010486880783e-05), (17, 2.172398853872437e-05), (18, 3.064208431169391e-05), (19, 4.676693424698897e-05), (20, 9.498743020230904e-05), (21, 5.6224675063276663e-05), (22, 6.0673326515825465e-05), (23, 7.862618076615036e-05), (24, 0.00015079398872330785), (25, 0.0003759745159186423), (26, 0.0001277281844522804), (27, 2.1970381567371078e-05)]
+
+ 55%|█████▍    | 82/150 [12:35<09:03,  7.99s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 402, Spatial MSE: [(0, 6.990552492425195e-07), (1, 5.799643531645415e-06), (2, 1.432036606274778e-05), (3, 7.878344149503391e-06), (4, 1.7428931187168928e-06), (5, 1.4980605556047522e-06), (6, 2.0023337583552347e-06), (7, 2.9569764592451975e-06), (8, 1.7314532669843175e-06), (9, 2.189761971749249e-06), (10, 2.703718109842157e-06), (11, 4.363108473626198e-06), (12, 1.2686381523963064e-05), (13, 0.00025258417008444667), (14, 0.00040982820792123675), (15, 0.00010877341264858842), (16, 0.009720108471810818), (17, 4.3972460844088346e-05), (18, 3.031247433682438e-05), (19, 0.0009578083408996463), (20, 0.000733802211470902), (21, 5.92612377658952e-05), (22, 7.42171541787684e-05), (23, 4.5301407226361334e-05), (24, 0.00010674589429982007), (25, 0.00023836802574805915), (26, 0.0003388763580005616), (27, 0.005003665573894978)], Temporal MSE: [(0, 6.546544568664103e-07), (1, 3.0018543384358054e-07), (2, 4.215515900796163e-07), (3, 7.007508884271374e-07), (4, 5.176212312107964e-07), (5, 7.558656420769694e-07), (6, 8.535856750313542e-07), (7, 9.87137127594906e-07), (8, 4.894113203590678e-07), (9, 5.958639235359442e-07), (10, 1.3806104561808752e-06), (11, 4.6007589844521135e-06), (12, 7.572366030217381e-06), (13, 5.277191576169571e-06), (14, 7.588419521198375e-06), (15, 3.68571090803016e-05), (16, 2.449830026307609e-05), (17, 2.6578272809274495e-05), (18, 1.6924072042456828e-05), (19, 1.5454657841473818e-05), (20, 1.4911584912624676e-05), (21, 3.608053157222457e-05), (22, 4.0182643715525046e-05), (23, 5.2863200835417956e-05), (24, 4.3344498408259824e-05), (25, 0.00010061720240628347), (26, 2.53125235758489e-05), (27, 2.2451984477811493e-05)]
+
+ 55%|█████▌    | 83/150 [12:43<08:41,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 396, Spatial MSE: [(0, 1.112535869651765e-06), (1, 1.2379936379147694e-05), (2, 2.8143040253780782e-05), (3, 1.628089194127824e-05), (4, 2.3747274099150673e-05), (5, 3.971154728787951e-05), (6, 6.897723505971953e-05), (7, 0.00015215357416309416), (8, 0.00014485660358332098), (9, 0.00021809640747960657), (10, 0.0002266523806611076), (11, 0.00024246149405371398), (12, 0.0003202900988981128), (13, 0.0008939781109802425), (14, 0.0018394256476312876), (15, 0.0013596494682133198), (16, 0.02069465070962906), (17, 0.0005893452907912433), (18, 0.0006262864335440099), (19, 0.0018269454594701529), (20, 0.0019230055622756481), (21, 0.000929858535528183), (22, 0.0007913546287454665), (23, 0.0012352040503174067), (24, 0.0023667917121201754), (25, 0.0045573837123811245), (26, 0.0071982513181865215), (27, 0.005365925841033459)], Temporal MSE: [(0, 9.121397397393594e-07), (1, 4.2411551248733304e-07), (2, 2.1659204776369734e-06), (3, 1.4170681424729992e-05), (4, 1.121968398365425e-05), (5, 2.08344790735282e-05), (6, 5.181978485779837e-05), (7, 3.8392128772102296e-05), (8, 2.4435626983176917e-05), (9, 4.022607390652411e-05), (10, 7.147271389840171e-05), (11, 8.219583833124489e-05), (12, 9.248025162378326e-05), (13, 0.0001577747316332534), (14, 0.00026298221200704575), (15, 0.00028354773530736566), (16, 0.00029258313588798046), (17, 0.0002575881080701947), (18, 0.0002342640800634399), (19, 0.00034643925027921796), (20, 0.0005694430437870324), (21, 0.00042815165943466127), (22, 0.00038314051926136017), (23, 0.0005172078963369131), (24, 0.0013709469931200147), (25, 0.002349844668060541), (26, 0.0006125543732196093), (27, 4.9910089728655294e-05)]
+
+ 56%|█████▌    | 84/150 [12:51<08:47,  8.00s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 390, Spatial MSE: [(0, 7.483411650355265e-07), (1, 6.621781722060405e-06), (2, 1.602296470082365e-05), (3, 8.808386155578773e-06), (4, 1.8234134131489554e-06), (5, 1.5904747669992503e-06), (6, 2.0377437977003865e-06), (7, 3.5143848435836844e-06), (8, 2.0045590645167977e-06), (9, 1.9861422515532468e-06), (10, 1.9453841559879947e-06), (11, 2.6715174499258865e-06), (12, 4.786803856404731e-06), (13, 6.462087912950665e-05), (14, 5.082412826595828e-05), (15, 1.5348216038546525e-05), (16, 2.578605381131638e-05), (17, 8.464864549750928e-06), (18, 1.7200491129187867e-05), (19, 1.421362776454771e-05), (20, 8.129578418447636e-06), (21, 1.5301948224077933e-05), (22, 4.3594551243586466e-05), (23, 2.8842907340731472e-05), (24, 6.228819984244183e-05), (25, 0.0001431432319805026), (26, 0.00033762006205506623), (27, 0.0017465234268456697)], Temporal MSE: [(0, 6.680685942228592e-07), (1, 3.2384471637669776e-07), (2, 4.922984544464271e-07), (3, 7.962522090565471e-07), (4, 6.275441251091252e-07), (5, 8.951263339440629e-07), (6, 1.1119325336039765e-06), (7, 1.1703706377375056e-06), (8, 5.805822524962423e-07), (9, 6.9625843934773e-07), (10, 1.0147118700842839e-06), (11, 1.7365082385367714e-06), (12, 2.0811926333408337e-06), (13, 2.588297320471611e-06), (14, 2.67261702902033e-06), (15, 2.123678541465779e-06), (16, 2.0797951947315596e-06), (17, 2.0684262835857226e-06), (18, 2.0671134279837133e-06), (19, 2.5263746010750765e-06), (20, 3.3021715353243053e-06), (21, 7.134998668334447e-06), (22, 7.924761121103074e-06), (23, 1.4075819308345672e-05), (24, 2.043553104158491e-05), (25, 8.088966569630429e-05), (26, 2.448786653985735e-05), (27, 2.19215303332021e-06)]
+
+ 57%|█████▋    | 85/150 [12:59<08:31,  7.87s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 384, Spatial MSE: [(0, 1.0733829185483046e-06), (1, 7.081787316565169e-06), (2, 1.721169610391371e-05), (3, 6.620087333431002e-06), (4, 3.78811660084466e-06), (5, 6.303357622527983e-06), (6, 1.513313691248186e-05), (7, 1.0035404557129368e-05), (8, 1.4364582057169173e-05), (9, 3.252874012105167e-05), (10, 3.6444489524001256e-05), (11, 5.50460463273339e-05), (12, 5.2920317102689296e-05), (13, 0.0001300797302974388), (14, 0.00029122555861249566), (15, 0.00019976822659373283), (16, 0.005667818710207939), (17, 3.821001519099809e-05), (18, 6.887794006615877e-05), (19, 0.000337760808179155), (20, 0.00038693001260980964), (21, 5.3892643336439505e-05), (22, 0.00011182408343302086), (23, 0.00012508629879448563), (24, 0.0003635816101450473), (25, 0.0009012411464937031), (26, 0.0023860780056566), (27, 0.00117316038813442)], Temporal MSE: [(0, 5.096437689644517e-07), (1, 2.8459325562835147e-07), (2, 7.071215009091247e-07), (3, 1.5966182900228887e-06), (4, 2.5296037620137213e-06), (5, 2.7032212983613135e-06), (6, 4.636480298358947e-06), (7, 4.912839358439669e-06), (8, 3.4483948638808215e-06), (9, 3.7253598748065997e-06), (10, 5.730271368520334e-06), (11, 8.211759450205136e-06), (12, 9.247229172615334e-06), (13, 1.4384600035555195e-05), (14, 1.919086389534641e-05), (15, 1.5543446352239698e-05), (16, 1.2264374163351022e-05), (17, 1.3240737644082401e-05), (18, 1.668978075031191e-05), (19, 2.0216708435327746e-05), (20, 3.24141037708614e-05), (21, 2.5575584004400298e-05), (22, 3.938933878089301e-05), (23, 4.765196717926301e-05), (24, 0.00013223370478954166), (25, 0.00035653405939228833), (26, 0.0001187199231935665), (27, 1.0619869499350898e-05)]
+
+ 57%|█████▋    | 86/150 [13:06<08:17,  7.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 378, Spatial MSE: [(0, 8.913552846934181e-07), (1, 5.846046406077221e-06), (2, 1.5172410712693818e-05), (3, 7.570607067464152e-06), (4, 1.434241653441859e-06), (5, 1.3640126326208701e-06), (6, 1.863079660324729e-06), (7, 2.8661947908403818e-06), (8, 1.7368435010212124e-06), (9, 1.8515311239752918e-06), (10, 1.693879084996297e-06), (11, 2.4257362838397967e-06), (12, 4.766461188410176e-06), (13, 0.00011365387035766616), (14, 0.0002593130338937044), (15, 8.709566463949159e-05), (16, 0.009636457078158855), (17, 5.9350350056774914e-05), (18, 2.9057166102575138e-05), (19, 0.0006859716959297657), (20, 0.0005652965628542006), (21, 2.811539343383629e-05), (22, 7.838835881557316e-05), (23, 5.8646528486860916e-05), (24, 0.00013535137986764312), (25, 0.0003148988471366465), (26, 0.00038615739322267473), (27, 0.00466474425047636)], Temporal MSE: [(0, 6.339249125630886e-07), (1, 2.610069316233421e-07), (2, 4.2504595398895617e-07), (3, 6.898401920807373e-07), (4, 5.535974878512206e-07), (5, 8.097756563074654e-07), (6, 9.034613412950421e-07), (7, 9.503580145064916e-07), (8, 5.047185140938382e-07), (9, 5.896327479604224e-07), (10, 9.750891649673576e-07), (11, 1.6572388403801597e-06), (12, 2.358199708396569e-06), (13, 4.96929942528368e-06), (14, 5.518731541087618e-06), (15, 1.268529194931034e-05), (16, 1.5582820196868852e-05), (17, 2.336478246434126e-05), (18, 1.3869122085452545e-05), (19, 1.3719875823881011e-05), (20, 1.1631783308985177e-05), (21, 2.9197668482083827e-05), (22, 3.833868686342612e-05), (23, 5.519136902876198e-05), (24, 4.945036562276073e-05), (25, 0.00014870474115014076), (26, 3.276758798165247e-05), (27, 2.1446925529744476e-05)]
+
+ 58%|█████▊    | 87/150 [13:14<08:10,  7.79s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 372, Spatial MSE: [(0, 1.1276586064923322e-06), (1, 1.3126716112310532e-05), (2, 2.3339427571045235e-05), (3, 1.5917235941742547e-05), (4, 2.372682502027601e-05), (5, 3.9819507946958765e-05), (6, 7.564113184344023e-05), (7, 0.00014257777365855873), (8, 0.000140920135891065), (9, 0.0002184188342653215), (10, 0.00022237708617467433), (11, 0.0002443422272335738), (12, 0.0003013349778484553), (13, 0.001916556153446436), (14, 0.005805297289043665), (15, 0.0038158672396093607), (16, 0.10617536306381226), (17, 0.0006478662835434079), (18, 0.0008996643009595573), (19, 0.010960213840007782), (20, 0.009839045815169811), (21, 0.0011275605065748096), (22, 0.0008903223788365722), (23, 0.0012812098721042275), (24, 0.002297868486493826), (25, 0.0048103416338562965), (26, 0.008665834553539753), (27, 0.005257401615381241)], Temporal MSE: [(0, 9.810341907723341e-07), (1, 4.44126214915741e-07), (2, 2.4841826871124795e-06), (3, 1.192152649309719e-05), (4, 1.1195589650014881e-05), (5, 2.0989336917409673e-05), (6, 4.464882658794522e-05), (7, 3.9984752220334485e-05), (8, 2.4891869543353096e-05), (9, 4.1514453187119216e-05), (10, 6.9786052335985e-05), (11, 8.240777970058843e-05), (12, 0.00010462139471201226), (13, 0.00018535059643909335), (14, 0.000304763438180089), (15, 0.0003175684832967818), (16, 0.0002932266506832093), (17, 0.0002662280749063939), (18, 0.00026851004804484546), (19, 0.00039538717828691006), (20, 0.0005894905189052224), (21, 0.0005387318087741733), (22, 0.0004637040838133544), (23, 0.0005751266144216061), (24, 0.0012886676704511046), (25, 0.002155178226530552), (26, 0.000607210211455822), (27, 5.550753849092871e-05)]
+
+ 59%|█████▊    | 88/150 [13:23<08:18,  8.04s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 366, Spatial MSE: [(0, 6.62161028230912e-07), (1, 5.7830447985907085e-06), (2, 1.5166744560701773e-05), (3, 8.49297794047743e-06), (4, 1.4698072163810139e-06), (5, 1.3622850474348525e-06), (6, 1.466518028792052e-06), (7, 3.2430054943688447e-06), (8, 1.6713230479581398e-06), (9, 1.60900890477933e-06), (10, 1.799457095330581e-06), (11, 3.336296458655852e-06), (12, 9.987146768253297e-06), (13, 0.0002872926997952163), (14, 0.00010960816871374846), (15, 4.1664374293759465e-05), (16, 0.0004729724023491144), (17, 9.756377039593644e-06), (18, 2.22627968469169e-05), (19, 1.8693266611080617e-05), (20, 1.1017054930562153e-05), (21, 9.682671952759847e-05), (22, 8.041675755521283e-05), (23, 4.220798655296676e-05), (24, 8.602793968748301e-05), (25, 0.00015991592954378575), (26, 0.00031285197474062443), (27, 0.001410511787980795)], Temporal MSE: [(0, 4.801373734153458e-07), (1, 2.4166726575458597e-07), (2, 4.199947056804376e-07), (3, 6.962416136957472e-07), (4, 6.007182946632383e-07), (5, 8.520505048181803e-07), (6, 1.0727588914960506e-06), (7, 9.8011378213414e-07), (8, 5.173360477783717e-07), (9, 5.820436967951537e-07), (10, 9.577760238244082e-07), (11, 5.470811174745904e-06), (12, 6.999121069384273e-06), (13, 3.919971732102567e-06), (14, 4.076220648130402e-06), (15, 2.6874872673943173e-06), (16, 2.7951266474701697e-06), (17, 2.158022198273102e-06), (18, 2.5347383143525803e-06), (19, 2.6017523850896396e-06), (20, 3.7467739275598433e-06), (21, 1.3097144801577087e-05), (22, 1.8865295714931563e-05), (23, 2.7723333914764225e-05), (24, 3.1701136322226375e-05), (25, 6.991878035478294e-05), (26, 2.9481469027814455e-05), (27, 2.3964823867572704e-06)]
+
+ 59%|█████▉    | 89/150 [13:31<08:12,  8.07s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 360, Spatial MSE: [(0, 1.1105953490186948e-06), (1, 6.860161647637142e-06), (2, 1.383111157338135e-05), (3, 6.608929197682301e-06), (4, 3.6711542179546086e-06), (5, 5.869430879101856e-06), (6, 1.1769555385399144e-05), (7, 8.688294656167272e-06), (8, 1.3350078006624244e-05), (9, 2.7964315449935384e-05), (10, 2.8148149795015343e-05), (11, 4.3794720113510266e-05), (12, 5.1320530474185944e-05), (13, 0.0003210288123227656), (14, 0.00014171723159961402), (15, 6.630743155255914e-05), (16, 0.0002548721677158028), (17, 2.778261296043638e-05), (18, 6.902348104631528e-05), (19, 5.940209302934818e-05), (20, 4.763490505865775e-05), (21, 8.721469930605963e-05), (22, 0.00012285982666071504), (23, 0.00011103802535217255), (24, 0.00031359787681140006), (25, 0.0008177268318831921), (26, 0.0021783870179206133), (27, 0.0010003619827330112)], Temporal MSE: [(0, 5.243666691967519e-07), (1, 2.961467941986484e-07), (2, 7.383970910268545e-07), (3, 1.622599370421085e-06), (4, 2.757084530458087e-06), (5, 2.7403366402722895e-06), (6, 4.5820220293535385e-06), (7, 4.653123596654041e-06), (8, 3.1141273666435154e-06), (9, 3.3382732453901554e-06), (10, 5.026321105106035e-06), (11, 1.0830982319021132e-05), (12, 1.2844595403294079e-05), (13, 1.278595118492376e-05), (14, 1.6427971786470152e-05), (15, 1.0208010280621238e-05), (16, 9.916081580740865e-06), (17, 8.153661838150583e-06), (18, 1.052380321198143e-05), (19, 1.267972584173549e-05), (20, 1.8906484910985455e-05), (21, 2.728344770730473e-05), (22, 3.738637315109372e-05), (23, 5.089654223411344e-05), (24, 0.00011584838648559526), (25, 0.00035041614319197834), (26, 0.00013516347098629922), (27, 1.3394672350841574e-05)]
+
+ 60%|██████    | 90/150 [13:39<08:07,  8.12s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 354, Spatial MSE: [(0, 1.2298428373469505e-06), (1, 5.6633498388691805e-06), (2, 1.3300247701408807e-05), (3, 8.014705599634908e-06), (4, 1.7259507103517535e-06), (5, 1.872126517810102e-06), (6, 2.443860921630403e-06), (7, 4.733430614578538e-06), (8, 2.468053480697563e-06), (9, 1.9761459952860605e-06), (10, 2.0843408492510207e-06), (11, 3.0201435947674327e-06), (12, 5.661553132085828e-06), (13, 9.685486293165013e-05), (14, 9.289084118790925e-05), (15, 4.131977402721532e-05), (16, 0.0034127647522836924), (17, 2.8060572731192224e-05), (18, 2.7828727979795076e-05), (19, 0.00013991608284413815), (20, 0.00016660439723636955), (21, 2.262002271891106e-05), (22, 7.596965588163584e-05), (23, 4.148781590629369e-05), (24, 9.082799078896642e-05), (25, 0.00021714522154070437), (26, 0.00036700902273878455), (27, 0.003009021980687976)], Temporal MSE: [(0, 8.383272529499664e-07), (1, 3.2950569561762677e-07), (2, 4.909541644337878e-07), (3, 8.335567258654919e-07), (4, 6.801276981605042e-07), (5, 9.450968718738295e-07), (6, 1.1636577710305573e-06), (7, 1.3326969110494247e-06), (8, 6.256358915379678e-07), (9, 7.726304716015875e-07), (10, 1.2196483112347778e-06), (11, 2.1710100099880947e-06), (12, 2.711334445848479e-06), (13, 3.6165499750495655e-06), (14, 4.920588708046125e-06), (15, 5.074364253232488e-06), (16, 5.092701485409634e-06), (17, 9.679702088760678e-06), (18, 7.274063136719633e-06), (19, 8.97643985808827e-06), (20, 8.246662218880374e-06), (21, 3.0855502700433135e-05), (22, 2.5263110728701577e-05), (23, 3.164104055031203e-05), (24, 3.088837183895521e-05), (25, 0.0001045695535140112), (26, 2.9328961318242364e-05), (27, 1.4532701243297197e-05)]
+
+ 61%|██████    | 91/150 [13:47<07:53,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 348, Spatial MSE: [(0, 9.857569693849655e-07), (1, 1.2188053005957045e-05), (2, 1.5790876204846427e-05), (3, 1.3199351997172926e-05), (4, 1.8375114450464025e-05), (5, 3.6635276046581566e-05), (6, 4.886886017629877e-05), (7, 0.00010778283467516303), (8, 0.00011932699999306351), (9, 0.00016574535402469337), (10, 0.00015282085223589092), (11, 0.00017794015002436936), (12, 0.00022637189249508083), (13, 0.0009047892526723444), (14, 0.002270015422254801), (15, 0.001857173629105091), (16, 0.042012397199869156), (17, 0.00045316925388760865), (18, 0.0006283845868892968), (19, 0.0024893314111977816), (20, 0.001997210318222642), (21, 0.0008377511985599995), (22, 0.0007437010644935071), (23, 0.0011365097016096115), (24, 0.0020586936734616756), (25, 0.003945628646761179), (26, 0.006260920315980911), (27, 0.0024823262356221676)], Temporal MSE: [(0, 8.875859975887579e-07), (1, 3.2690550710867683e-07), (2, 1.9250705918238964e-06), (3, 1.6082807633210905e-05), (4, 1.0779963304230478e-05), (5, 1.8076780179399066e-05), (6, 3.5872843000106514e-05), (7, 2.975429015350528e-05), (8, 2.0519131794571877e-05), (9, 3.356693559908308e-05), (10, 5.828064968227409e-05), (11, 6.232497980818152e-05), (12, 7.123057002900168e-05), (13, 0.00012674450408667326), (14, 0.00022471988631878048), (15, 0.0002090259367832914), (16, 0.00020395404135342687), (17, 0.000191624290891923), (18, 0.00019414894632063806), (19, 0.00027264084201306105), (20, 0.0004263909358996898), (21, 0.000386616971809417), (22, 0.00037801190046593547), (23, 0.0005021533579565585), (24, 0.0012177267344668508), (25, 0.002145938342437148), (26, 0.0005971629871055484), (27, 4.6061290049692616e-05)]
+
+ 61%|██████▏   | 92/150 [13:55<07:51,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 342, Spatial MSE: [(0, 7.059050517455034e-07), (1, 5.498227892530849e-06), (2, 1.314923520112643e-05), (3, 8.902849003789015e-06), (4, 1.6704507288523018e-06), (5, 1.74026263266569e-06), (6, 1.701684936961101e-06), (7, 3.614344905145117e-06), (8, 2.0656873402913334e-06), (9, 1.9875201360264327e-06), (10, 2.8248221042304067e-06), (11, 5.069877261121292e-06), (12, 1.517935743322596e-05), (13, 0.0002799950889311731), (14, 0.0004860262561123818), (15, 0.00014717716840095818), (16, 0.010335657745599747), (17, 7.852118142182007e-05), (18, 3.6585843190550804e-05), (19, 0.0008488031453453004), (20, 0.0007396786822937429), (21, 0.00011308799003018066), (22, 0.0001376342843286693), (23, 5.609371874015778e-05), (24, 0.00012927850184496492), (25, 0.0002842870890162885), (26, 0.000307874521240592), (27, 0.005249198526144028)], Temporal MSE: [(0, 6.122281206444313e-07), (1, 3.329778337501921e-07), (2, 4.930242312184419e-07), (3, 8.807632525531517e-07), (4, 7.333704274969932e-07), (5, 1.0220383046544157e-06), (6, 1.3802653029415524e-06), (7, 1.2682127135121846e-06), (8, 6.731127655257296e-07), (9, 7.79269726081111e-07), (10, 1.6157539448613534e-06), (11, 5.736087587138172e-06), (12, 9.638912160880864e-06), (13, 9.112403859035112e-06), (14, 1.2524857083917595e-05), (15, 2.9526423531933688e-05), (16, 1.6863892597029917e-05), (17, 1.5994768546079285e-05), (18, 1.5763718693051487e-05), (19, 1.561414683237672e-05), (20, 1.3448871868604328e-05), (21, 4.00907956645824e-05), (22, 4.486552279558964e-05), (23, 7.220295810839161e-05), (24, 5.380662696552463e-05), (25, 9.771828626981005e-05), (26, 3.27644229400903e-05), (27, 3.095316787948832e-05)]
+
+ 62%|██████▏   | 93/150 [14:03<07:37,  8.03s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 336, Spatial MSE: [(0, 1.043459860738949e-06), (1, 6.7989217313879635e-06), (2, 1.138960215030238e-05), (3, 6.92319190420676e-06), (4, 4.57682608612231e-06), (5, 7.601973720738897e-06), (6, 1.7665330233285204e-05), (7, 1.1841106243082322e-05), (8, 1.784948108252138e-05), (9, 3.704708433360793e-05), (10, 5.1120197895215824e-05), (11, 6.870159995742142e-05), (12, 7.619781536050141e-05), (13, 0.0005837911739945412), (14, 0.001392215141095221), (15, 0.0008199499570764601), (16, 0.034867942333221436), (17, 0.00013677371316589415), (18, 0.00019329065980855376), (19, 0.0019344070460647345), (20, 0.001379704219289124), (21, 0.0002816856140270829), (22, 0.0002356589975534007), (23, 0.00022611889289692044), (24, 0.0006283880793489516), (25, 0.0013989355647936463), (26, 0.003100478323176503), (27, 0.005317092407494783)], Temporal MSE: [(0, 5.649544050356781e-07), (1, 3.2920564763117e-07), (2, 7.127225103431556e-07), (3, 1.9668989352794597e-06), (4, 2.98314284918888e-06), (5, 3.3934688872250263e-06), (6, 5.661975592374802e-06), (7, 6.697329354210524e-06), (8, 4.126323801756371e-06), (9, 4.941684437653748e-06), (10, 7.622933935635956e-06), (11, 1.4096580343903042e-05), (12, 1.9857039660564624e-05), (13, 2.526298885641154e-05), (14, 4.0341747080674395e-05), (15, 6.56478077871725e-05), (16, 3.34092510456685e-05), (17, 3.737772931344807e-05), (18, 3.9589078369317576e-05), (19, 4.146893843426369e-05), (20, 5.9074351156596094e-05), (21, 0.00011867215653182939), (22, 0.00011822600936284289), (23, 0.0001316012640018016), (24, 0.0002135544636985287), (25, 0.0005573858506977558), (26, 0.00017141930584330112), (27, 4.639900726033375e-05)]
+
+ 63%|██████▎   | 94/150 [14:11<07:29,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 330, Spatial MSE: [(0, 8.111097713481286e-07), (1, 6.23678033662145e-06), (2, 1.3730007594858762e-05), (3, 7.866900887165684e-06), (4, 1.6650716361255036e-06), (5, 1.6057399534474825e-06), (6, 1.8220013089376152e-06), (7, 3.1373745059681823e-06), (8, 1.9719070678547723e-06), (9, 1.9483989035506966e-06), (10, 2.3502122985519236e-06), (11, 4.928416728944285e-06), (12, 1.5465479009435512e-05), (13, 0.00024232555006165057), (14, 0.00043689162703230977), (15, 0.00017313164426013827), (16, 0.01331488136202097), (17, 0.00010644445137586445), (18, 5.727243478759192e-05), (19, 0.0007855595904402435), (20, 0.000685356673784554), (21, 0.00012759637320414186), (22, 0.0001327759091509506), (23, 7.361205643974245e-05), (24, 0.00015316455392166972), (25, 0.00027533003594726324), (26, 0.00034693392808549106), (27, 0.005949371960014105)], Temporal MSE: [(0, 6.290029546107689e-07), (1, 3.0115165827737655e-07), (2, 4.912302529191948e-07), (3, 8.094296504168597e-07), (4, 7.205938459264871e-07), (5, 9.773676765689743e-07), (6, 1.2170232821517857e-06), (7, 1.1655470188998152e-06), (8, 6.296131687122397e-07), (9, 7.657984042452881e-07), (10, 1.4747140539839165e-06), (11, 6.147507065179525e-06), (12, 8.661399988341145e-06), (13, 7.392466613964643e-06), (14, 1.1712864761648234e-05), (15, 2.2471089323516935e-05), (16, 1.5378316675196402e-05), (17, 1.7506226868135855e-05), (18, 1.916939436341636e-05), (19, 1.7741238480084576e-05), (20, 1.5076260751811787e-05), (21, 6.09663657087367e-05), (22, 6.771340849809349e-05), (23, 7.35776120563969e-05), (24, 6.209096318343654e-05), (25, 0.00011405810073483735), (26, 3.4429434890626e-05), (27, 3.489894515951164e-05)]
+
+ 63%|██████▎   | 95/150 [14:19<07:19,  7.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 324, Spatial MSE: [(0, 9.5635982688691e-07), (1, 1.0754872164397966e-05), (2, 1.597673872311134e-05), (3, 1.4243260920920875e-05), (4, 2.1106587155372836e-05), (5, 3.490946255624294e-05), (6, 7.016784365987405e-05), (7, 0.00013205548748373985), (8, 0.0001375526626361534), (9, 0.0002057331003015861), (10, 0.00019659507961478084), (11, 0.0002259795874124393), (12, 0.0002831113524734974), (13, 0.0010045465314760804), (14, 0.0023848863784223795), (15, 0.001721216132864356), (16, 0.03527941554784775), (17, 0.000640067970380187), (18, 0.0007024191436357796), (19, 0.002315141260623932), (20, 0.002238017274066806), (21, 0.0008980329730547965), (22, 0.0008380346116609871), (23, 0.001320205512456596), (24, 0.0023889250587671995), (25, 0.005112192593514919), (26, 0.008438545279204845), (27, 0.005670175887644291)], Temporal MSE: [(0, 8.918419780457043e-07), (1, 3.9306058852162096e-07), (2, 2.216535904153716e-06), (3, 1.0340151675336529e-05), (4, 1.1221037311770488e-05), (5, 2.0707449948531576e-05), (6, 4.893920413451269e-05), (7, 4.09796011808794e-05), (8, 2.466722798999399e-05), (9, 4.17921255575493e-05), (10, 6.861107249278575e-05), (11, 8.1460784713272e-05), (12, 9.228324779542163e-05), (13, 0.0001584799902047962), (14, 0.0002768964332062751), (15, 0.00027651761774905026), (16, 0.0002555700484663248), (17, 0.0002304094232385978), (18, 0.00022843509213998914), (19, 0.00031969568226486444), (20, 0.0004909205017611384), (21, 0.0004364273918326944), (22, 0.00040621214429847896), (23, 0.0005431593162938952), (24, 0.0012415339006111026), (25, 0.002215493470430374), (26, 0.0007141061360016465), (27, 7.344316691160202e-05)]
+
+ 64%|██████▍   | 96/150 [14:27<07:17,  8.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 318, Spatial MSE: [(0, 6.471407232311321e-07), (1, 5.5330410759779625e-06), (2, 1.1554124284884892e-05), (3, 8.926478585635778e-06), (4, 1.7284296518482734e-06), (5, 1.847332896431908e-06), (6, 1.528538518869027e-06), (7, 3.21701236316585e-06), (8, 1.949646275534178e-06), (9, 1.8677252455745474e-06), (10, 2.4230307644756977e-06), (11, 4.463740424398566e-06), (12, 1.5382249330286868e-05), (13, 0.00035170780029147863), (14, 0.0006737920339219272), (15, 0.00022266291489358991), (16, 0.023569317534565926), (17, 0.00011469604214653373), (18, 5.3046245739096776e-05), (19, 0.0018222740618512034), (20, 0.0014622731832787395), (21, 6.555609434144571e-05), (22, 0.00011366636317688972), (23, 8.312627323903143e-05), (24, 0.0001948144636116922), (25, 0.0004031022544950247), (26, 0.0003217254125047475), (27, 0.009275104850530624)], Temporal MSE: [(0, 5.313093538461544e-07), (1, 3.0239877446547325e-07), (2, 4.7631306188122835e-07), (3, 8.61427906784229e-07), (4, 7.887694550845481e-07), (5, 1.0494864000065718e-06), (6, 1.279663138120668e-06), (7, 1.21678954201343e-06), (8, 6.970313393139804e-07), (9, 7.941747526274412e-07), (10, 1.4768497749173548e-06), (11, 5.57924977329094e-06), (12, 9.619568118068855e-06), (13, 1.4302537238108926e-05), (14, 1.6291825886582956e-05), (15, 4.369883754407056e-05), (16, 3.4639379009604454e-05), (17, 6.0588205087697133e-05), (18, 3.1825427868170664e-05), (19, 2.8699270842480473e-05), (20, 2.2027681552572176e-05), (21, 7.86473901825957e-05), (22, 8.143018203554675e-05), (23, 0.00011043096310459077), (24, 8.293503196910024e-05), (25, 0.0001461277570342645), (26, 3.713085243362002e-05), (27, 5.387135752243921e-05)]
+
+ 65%|██████▍   | 97/150 [14:35<06:58,  7.89s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 312, Spatial MSE: [(0, 1.0320861747459276e-06), (1, 6.982776540098712e-06), (2, 9.897614290821366e-06), (3, 7.2022135100269224e-06), (4, 4.1695184336276725e-06), (5, 7.130887297535082e-06), (6, 1.669804441917222e-05), (7, 9.66264633461833e-06), (8, 1.7583904991624877e-05), (9, 3.207248664693907e-05), (10, 4.049580456921831e-05), (11, 5.5380129197146744e-05), (12, 6.641218351433054e-05), (13, 0.0006033148965798318), (14, 0.0013641020050272346), (15, 0.000949673994909972), (16, 0.031123235821723938), (17, 0.00019170176528859884), (18, 0.00021061829465907067), (19, 0.0018885969184339046), (20, 0.0015077657299116254), (21, 0.00016384442278649658), (22, 0.00020764273358508945), (23, 0.0002145905455108732), (24, 0.0005620432202704251), (25, 0.0013497663894668221), (26, 0.002694421447813511), (27, 0.0088321752846241)], Temporal MSE: [(0, 5.445878628052014e-07), (1, 3.2183262987928174e-07), (2, 6.915443009347655e-07), (3, 2.031050144069013e-06), (4, 3.188845084878267e-06), (5, 3.180971361871343e-06), (6, 5.1554397941799834e-06), (7, 6.166310413391329e-06), (8, 3.7391771456896095e-06), (9, 4.282218924345216e-06), (10, 6.865790055599064e-06), (11, 1.542838799650781e-05), (12, 2.0626217519748025e-05), (13, 2.6223971872241236e-05), (14, 3.925458076992072e-05), (15, 5.935017907177098e-05), (16, 3.6541259760269895e-05), (17, 5.574425813392736e-05), (18, 4.383737905300222e-05), (19, 4.629799150279723e-05), (20, 6.288639997364953e-05), (21, 0.00012295682972762734), (22, 0.00011906796135008335), (23, 0.00014511379413306713), (24, 0.0002135825779987499), (25, 0.0005392848397605121), (26, 0.00017822232621256262), (27, 6.527386722154915e-05)]
+
+ 65%|██████▌   | 98/150 [14:43<06:57,  8.02s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 306, Spatial MSE: [(0, 1.2417852985890931e-06), (1, 7.502405878767604e-06), (2, 1.3736591427004896e-05), (3, 8.803488526609726e-06), (4, 2.0712441255454905e-06), (5, 2.253783804917475e-06), (6, 4.4080270527047105e-06), (7, 4.60142928204732e-06), (8, 3.121210511380923e-06), (9, 2.9429293135763146e-06), (10, 1.114443875849247e-05), (11, 4.125649866182357e-05), (12, 8.135039388434961e-05), (13, 0.0013523150701075792), (14, 0.0013885051012039185), (15, 0.0004204126016702503), (16, 0.023145033046603203), (17, 0.0001354856212856248), (18, 7.626163278473541e-05), (19, 0.0016659792745485902), (20, 0.0012747609289363027), (21, 7.605463906656951e-05), (22, 0.00016527272236999124), (23, 0.00011011439346475527), (24, 0.00022232170158531517), (25, 0.00041550275636836886), (26, 0.0004286959010642022), (27, 0.013512189500033855)], Temporal MSE: [(0, 8.330384844157379e-07), (1, 3.52254488689141e-07), (2, 5.369409450395324e-07), (3, 9.501855515736679e-07), (4, 8.225068768297206e-07), (5, 1.1194272246939363e-06), (6, 1.373155896544631e-06), (7, 1.4882994037179742e-06), (8, 7.662600864932756e-07), (9, 1.2036499583700788e-06), (10, 6.977905286476016e-06), (11, 8.376533514820039e-05), (12, 3.5211276554036885e-05), (13, 2.163000863220077e-05), (14, 1.9785451513598673e-05), (15, 6.298852531472221e-05), (16, 3.2739146263338625e-05), (17, 3.97323165088892e-05), (18, 4.228494071867317e-05), (19, 3.6724341043736786e-05), (20, 3.855646718875505e-05), (21, 8.945897570811212e-05), (22, 8.09419434517622e-05), (23, 0.00011417558562243357), (24, 9.898550342768431e-05), (25, 0.0002554337843321264), (26, 5.896730363019742e-05), (27, 6.404911255231127e-05)]
+
+ 66%|██████▌   | 99/150 [14:51<06:43,  7.92s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 300, Spatial MSE: [(0, 1.2154890782767325e-06), (1, 1.090196928998921e-05), (2, 1.3502025467460044e-05), (3, 1.3639742974191904e-05), (4, 2.0493516785791144e-05), (5, 3.4159507777076215e-05), (6, 6.56081538181752e-05), (7, 0.00012117547885281965), (8, 0.00012792191409971565), (9, 0.00019865593640133739), (10, 0.00017721168114803731), (11, 0.00023643241729587317), (12, 0.0003196153847966343), (13, 0.002303104382008314), (14, 0.0036414628848433495), (15, 0.0017347880639135838), (16, 0.05375555530190468), (17, 0.000577007420361042), (18, 0.000826892617624253), (19, 0.005141910631209612), (20, 0.0049603902734816074), (21, 0.0008594365208409727), (22, 0.0008280297624878585), (23, 0.0012651925208047032), (24, 0.002219763118773699), (25, 0.004872349090874195), (26, 0.009092994034290314), (27, 0.013360313139855862)], Temporal MSE: [(0, 1.1639158401521854e-06), (1, 4.7224787635968823e-07), (2, 2.2571100544155343e-06), (3, 8.430878551735077e-06), (4, 1.0942550034087617e-05), (5, 1.9595152480178513e-05), (6, 4.492200605454855e-05), (7, 3.925035343854688e-05), (8, 2.3211559891933575e-05), (9, 3.955577631131746e-05), (10, 6.507783109555021e-05), (11, 0.00014971043856348842), (12, 0.0001142878900282085), (13, 0.00015559453458990902), (14, 0.0002443410048726946), (15, 0.00026924311532638967), (16, 0.0002361479273531586), (17, 0.00021713081514462829), (18, 0.0002344034000998363), (19, 0.0003076239372603595), (20, 0.00044703122694045305), (21, 0.00046834134263917804), (22, 0.00043030083179473877), (23, 0.0005705053918063641), (24, 0.0011594370007514954), (25, 0.0021767206490039825), (26, 0.000760899216402322), (27, 0.0001063603995135054)]
+
+ 67%|██████▋   | 100/150 [14:59<06:38,  7.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 294, Spatial MSE: [(0, 7.477008807654784e-07), (1, 6.346761892928043e-06), (2, 1.2181565580249298e-05), (3, 8.841899216349702e-06), (4, 1.8029080592896207e-06), (5, 1.9930937469325727e-06), (6, 2.0472571122809313e-06), (7, 3.642299816419836e-06), (8, 2.270194727316266e-06), (9, 2.093794819302275e-06), (10, 2.8030108296661638e-06), (11, 4.547334810922621e-06), (12, 1.4415790246857796e-05), (13, 0.0005459208623506129), (14, 0.0008258494199253619), (15, 0.0002809339785017073), (16, 0.014611531049013138), (17, 6.46822081762366e-05), (18, 3.9542959711980075e-05), (19, 0.0011342850048094988), (20, 0.0008169086067937315), (21, 5.204219633014873e-05), (22, 0.00011018093937309459), (23, 7.842286140657961e-05), (24, 0.00014172036026138812), (25, 0.00023912136384751648), (26, 0.00032067601568996906), (27, 0.009233133867383003)], Temporal MSE: [(0, 5.40306984930794e-07), (1, 2.9450271199493727e-07), (2, 5.089015644443862e-07), (3, 9.165711389869102e-07), (4, 8.634202686153003e-07), (5, 1.133426280830463e-06), (6, 1.5133963415792095e-06), (7, 1.4056355439606705e-06), (8, 7.929526759653527e-07), (9, 9.201493753607792e-07), (10, 1.7558122635819018e-06), (11, 5.882640380150406e-06), (12, 1.0774387192213908e-05), (13, 1.79225935426075e-05), (14, 1.2202570360386744e-05), (15, 3.1785148166818544e-05), (16, 1.509531921328744e-05), (17, 1.9789016732829623e-05), (18, 2.9975441066198982e-05), (19, 2.4226670575444587e-05), (20, 2.326548201381229e-05), (21, 6.119760655565187e-05), (22, 4.6501056203851476e-05), (23, 7.394651038339362e-05), (24, 6.368743197526783e-05), (25, 0.00019066256936639547), (26, 4.5486052840715274e-05), (27, 4.238184919813648e-05)]
+
+ 67%|██████▋   | 101/150 [15:07<06:27,  7.90s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 288, Spatial MSE: [(0, 9.932894045050489e-07), (1, 6.5974982135230675e-06), (2, 7.963054486026522e-06), (3, 6.118043074820889e-06), (4, 3.6716235172207234e-06), (5, 7.219389772217255e-06), (6, 1.3543733075493947e-05), (7, 1.0194752576353494e-05), (8, 1.701539986243006e-05), (9, 3.074689448112622e-05), (10, 3.485356137389317e-05), (11, 5.375985347200185e-05), (12, 5.788776252302341e-05), (13, 0.000584377208724618), (14, 0.0009079421870410442), (15, 0.0003197657351847738), (16, 0.014761528931558132), (17, 8.400481601711363e-05), (18, 6.254065374378115e-05), (19, 0.0011542841093614697), (20, 0.0008446946158073843), (21, 8.390595758100972e-05), (22, 0.00016620851238258183), (23, 0.0001695549872238189), (24, 0.00044349240488372743), (25, 0.001199745456688106), (26, 0.0030676620081067085), (27, 0.00901953224092722)], Temporal MSE: [(0, 5.900707265027449e-07), (1, 3.4434509643688216e-07), (2, 6.693420573355979e-07), (3, 1.965010824278579e-06), (4, 3.1944755392032675e-06), (5, 3.194425062247319e-06), (6, 5.657383098878199e-06), (7, 6.846625637990655e-06), (8, 4.076746336068027e-06), (9, 4.574779723043321e-06), (10, 6.7646883508132305e-06), (11, 1.2611916645255405e-05), (12, 1.816765870898962e-05), (13, 2.9523684133891948e-05), (14, 2.997756382683292e-05), (15, 4.363108746474609e-05), (16, 2.3495098503190093e-05), (17, 2.6798110411618836e-05), (18, 3.9233324059750885e-05), (19, 3.7081594200572e-05), (20, 4.3597603507805616e-05), (21, 8.052023622440174e-05), (22, 7.505920802941546e-05), (23, 0.00010649523756001145), (24, 0.00017964163271244615), (25, 0.0005382536328397691), (26, 0.00020351480634417385), (27, 5.817170313093811e-05)]
+
+ 68%|██████▊   | 102/150 [15:15<06:28,  8.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 282, Spatial MSE: [(0, 8.136523774737725e-07), (1, 6.910078809596598e-06), (2, 1.1448909390310291e-05), (3, 7.72230760048842e-06), (4, 1.6968048157650628e-06), (5, 1.9846656869049184e-06), (6, 1.8901024532169686e-06), (7, 3.4145036806876305e-06), (8, 2.3248658180818893e-06), (9, 2.0870304524578387e-06), (10, 2.433750978525495e-06), (11, 3.3086205348809017e-06), (12, 7.3768569563981146e-06), (13, 0.00012320242240093648), (14, 8.492245979141444e-05), (15, 2.15152685996145e-05), (16, 6.220395880518481e-05), (17, 8.472137778881006e-06), (18, 9.794373909244314e-06), (19, 1.86408815352479e-05), (20, 9.326429790235125e-06), (21, 1.7670059605734423e-05), (22, 6.993870192673057e-05), (23, 3.6160163290333e-05), (24, 7.63794087106362e-05), (25, 0.00014229888620320708), (26, 0.0002519658301025629), (27, 0.001630429527722299)], Temporal MSE: [(0, 6.747930569872551e-07), (1, 3.33674165631237e-07), (2, 5.271327836453565e-07), (3, 9.153415021501132e-07), (4, 8.562058724237431e-07), (5, 1.122713570111955e-06), (6, 1.2911473277199548e-06), (7, 1.375721353724657e-06), (8, 7.675453161937185e-07), (9, 8.790688070803299e-07), (10, 1.4696586276841117e-06), (11, 2.965227622553357e-06), (12, 3.87388627132168e-06), (13, 4.1755624806683045e-06), (14, 4.182623342785519e-06), (15, 3.2283437576552387e-06), (16, 2.88307387563691e-06), (17, 2.706814939301694e-06), (18, 2.143334768334171e-06), (19, 2.8896902222186327e-06), (20, 4.115799583814805e-06), (21, 1.7665130144450814e-05), (22, 1.4547023965860717e-05), (23, 2.099602534144651e-05), (24, 2.6787618480739184e-05), (25, 8.411188173340634e-05), (26, 2.893476266763173e-05), (27, 5.057670477981446e-06)]
+
+ 69%|██████▊   | 103/150 [15:24<06:31,  8.33s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 276, Spatial MSE: [(0, 1.0312171525583835e-06), (1, 8.099107617454138e-06), (2, 1.2095872989448253e-05), (3, 1.3371559361985419e-05), (4, 1.617481939319987e-05), (5, 3.48120738635771e-05), (6, 4.958796489518136e-05), (7, 0.00010589631710899994), (8, 0.00012614075967576355), (9, 0.0001458293991163373), (10, 0.00012324033014010638), (11, 0.00015588720270898193), (12, 0.00020133020007051528), (13, 0.0011343026999384165), (14, 0.0025745611637830734), (15, 0.001954365521669388), (16, 0.04695789888501167), (17, 0.0004054393502883613), (18, 0.0005581425502896309), (19, 0.0028005558997392654), (20, 0.002181040355935693), (21, 0.0007285354658961296), (22, 0.0007271163049153984), (23, 0.001184440916404128), (24, 0.002021681983023882), (25, 0.00411978829652071), (26, 0.007219281978905201), (27, 0.001853889087215066)], Temporal MSE: [(0, 9.251442634194973e-07), (1, 4.165940481470898e-07), (2, 2.100072379107587e-06), (3, 9.387327736476436e-06), (4, 1.13770956886583e-05), (5, 1.7515214494778775e-05), (6, 3.703611218952574e-05), (7, 3.3148891816381365e-05), (8, 2.0593955923686735e-05), (9, 3.389333505765535e-05), (10, 5.2644183597294614e-05), (11, 5.966026219539344e-05), (12, 7.302551239263266e-05), (13, 0.0001226685562869534), (14, 0.00021421756537165493), (15, 0.0001935426116688177), (16, 0.00018016694230027497), (17, 0.00016841944307088852), (18, 0.00017874770855996758), (19, 0.00024111410311888903), (20, 0.00037899016751907766), (21, 0.0003782098356168717), (22, 0.0003829321067314595), (23, 0.0005037094815634191), (24, 0.0011124806478619576), (25, 0.0020737312734127045), (26, 0.0007329056388698518), (27, 5.426469215308316e-05)]
+
+ 69%|██████▉   | 104/150 [15:34<06:40,  8.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 270, Spatial MSE: [(0, 8.435206950707652e-07), (1, 7.093891781551065e-06), (2, 1.2672527191170957e-05), (3, 8.739983968553133e-06), (4, 1.8221824120701058e-06), (5, 2.1460086827573832e-06), (6, 1.9019438468603767e-06), (7, 3.660494485302479e-06), (8, 2.456390120642027e-06), (9, 2.1626440229738364e-06), (10, 2.5569484023435507e-06), (11, 2.836888370438828e-06), (12, 7.108871159289265e-06), (13, 0.00018826773157343268), (14, 0.0002552037767600268), (15, 0.00015142677875701338), (16, 0.005005297716706991), (17, 3.404054223210551e-05), (18, 1.783555853762664e-05), (19, 0.00046828194172121584), (20, 0.0003655028704088181), (21, 5.621076707029715e-05), (22, 0.000137019669637084), (23, 6.19413549429737e-05), (24, 0.00012436472752597183), (25, 0.00023617320402991027), (26, 0.00028197301435284317), (27, 0.003223818028345704)], Temporal MSE: [(0, 6.265041747610667e-07), (1, 3.5001551168534206e-07), (2, 5.603882868854271e-07), (3, 9.802520253288094e-07), (4, 9.73078954302764e-07), (5, 1.2455639080144465e-06), (6, 1.5544108009635238e-06), (7, 1.5331696658904548e-06), (8, 8.760894729675783e-07), (9, 9.891876970868907e-07), (10, 1.4943440191927948e-06), (11, 2.7087098715128377e-06), (12, 4.197954694973305e-06), (13, 6.810673767176922e-06), (14, 9.661980584496632e-06), (15, 9.026177394844126e-06), (16, 8.765910024521872e-06), (17, 1.5248057934513781e-05), (18, 8.669355338497553e-06), (19, 8.761783647059929e-06), (20, 8.365318535652477e-06), (21, 5.587580017163418e-05), (22, 4.1434122977079824e-05), (23, 5.6695673265494406e-05), (24, 4.621441621566191e-05), (25, 0.00010851329716388136), (26, 3.5354431020095944e-05), (27, 1.4050498066353612e-05)]
+
+ 70%|███████   | 105/150 [15:42<06:29,  8.65s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 264, Spatial MSE: [(0, 1.0287726581736933e-06), (1, 9.178344953397755e-06), (2, 9.717803550302051e-06), (3, 8.830666956782807e-06), (4, 8.462881851301063e-06), (5, 1.1376104339433368e-05), (6, 3.221068254788406e-05), (7, 3.742712578969076e-05), (8, 4.418442040332593e-05), (9, 7.360248127952218e-05), (10, 7.146982534322888e-05), (11, 0.00010153456241823733), (12, 9.528041846351698e-05), (13, 0.000565341382753104), (14, 0.0012183961225673556), (15, 0.0007493485463783145), (16, 0.017336349934339523), (17, 0.0001645963202463463), (18, 0.0004903181688860059), (19, 0.002231562277302146), (20, 0.0023795170709490776), (21, 0.0005006642895750701), (22, 0.0003415312385186553), (23, 0.0004652360803447664), (24, 0.0008104895241558552), (25, 0.002676911884918809), (26, 0.00606207549571991), (27, 0.00332024903036654)], Temporal MSE: [(0, 9.786324426386273e-07), (1, 4.2619276996447297e-07), (2, 1.006615661935939e-06), (3, 1.8882864196712035e-06), (4, 3.5011082673008787e-06), (5, 6.718278200423811e-06), (6, 1.9843133486574516e-05), (7, 1.7883940017782152e-05), (8, 8.107425855996553e-06), (9, 1.381978472636547e-05), (10, 1.868699473561719e-05), (11, 2.4731185476412065e-05), (12, 3.0082384910201654e-05), (13, 4.7275676479330286e-05), (14, 7.931031723273918e-05), (15, 7.738688873359933e-05), (16, 6.850291538285092e-05), (17, 6.409785419236869e-05), (18, 6.218059570528567e-05), (19, 9.951804531738162e-05), (20, 0.0001433616562280804), (21, 0.0001761746680131182), (22, 0.00014219516015145928), (23, 0.00018402647401671857), (24, 0.00035483905230648816), (25, 0.0005630056257359684), (26, 0.00020502074039541185), (27, 2.3957421944942325e-05)]
+
+ 71%|███████   | 106/150 [15:51<06:19,  8.63s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 258, Spatial MSE: [(0, 7.646319204468455e-07), (1, 7.089107839419739e-06), (2, 1.2312238141021226e-05), (3, 8.710952897672541e-06), (4, 2.064883801722317e-06), (5, 2.2972044462221675e-06), (6, 1.934504325618036e-06), (7, 3.2805983209982514e-06), (8, 2.587631570349913e-06), (9, 2.8813053631893126e-06), (10, 6.243308689590776e-06), (11, 9.166897143586539e-06), (12, 1.5777675798744895e-05), (13, 0.00022577504569198936), (14, 0.00045712763676419854), (15, 0.0001291809167014435), (16, 0.010057380422949791), (17, 6.992572889430448e-05), (18, 2.3203163436846808e-05), (19, 0.0008956859819591045), (20, 0.0007318559801205993), (21, 4.5280892663868144e-05), (22, 6.792201020289212e-05), (23, 5.465832509798929e-05), (24, 0.00012041293666698039), (25, 0.0002440175012452528), (26, 0.00025639377417974174), (27, 0.004983810242265463)], Temporal MSE: [(0, 7.388853759948688e-07), (1, 4.0261488720716443e-07), (2, 6.285360996116651e-07), (3, 1.0993306887030485e-06), (4, 1.0186255394728505e-06), (5, 1.3191729522077367e-06), (6, 1.6993910776363919e-06), (7, 1.5787113625265192e-06), (8, 9.77306967797631e-07), (9, 1.1758196478695027e-06), (10, 3.543078264556243e-06), (11, 8.351838005182799e-06), (12, 6.770106210751692e-06), (13, 6.518846475955797e-06), (14, 9.229001989297103e-06), (15, 4.0777689719107e-05), (16, 2.5751322027645074e-05), (17, 3.0821716791251674e-05), (18, 1.3437595043797046e-05), (19, 1.2711563613265753e-05), (20, 1.0153607036045287e-05), (21, 5.7272791309515014e-05), (22, 4.549896766548045e-05), (23, 5.5567197705386207e-05), (24, 4.6517609007423744e-05), (25, 0.00011051571345888078), (26, 3.600266791181639e-05), (27, 2.428291190881282e-05)]
+
+ 71%|███████▏  | 107/150 [15:59<06:01,  8.40s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 252, Spatial MSE: [(0, 9.764695505509735e-07), (1, 6.7611917984322645e-06), (2, 8.331056051247288e-06), (3, 6.445945928135188e-06), (4, 4.146502760704607e-06), (5, 7.957401976455003e-06), (6, 1.4362671208800748e-05), (7, 1.1507511771924328e-05), (8, 1.8968245058204047e-05), (9, 3.5435914469417185e-05), (10, 4.664076186600141e-05), (11, 6.819031841587275e-05), (12, 6.632402073591948e-05), (13, 0.0002665017091203481), (14, 0.0005359331262297928), (15, 0.00018775329226627946), (16, 0.010191698558628559), (17, 9.717775537865236e-05), (18, 6.439433491323143e-05), (19, 0.0009303424158133566), (20, 0.0007727212505415082), (21, 7.741259469185024e-05), (22, 0.00013680438860319555), (23, 0.00017855456098914146), (24, 0.0005073227221146226), (25, 0.0015433430671691895), (26, 0.0032188170589506626), (27, 0.005131551530212164)], Temporal MSE: [(0, 7.041474532343273e-07), (1, 4.0503257991986175e-07), (2, 7.553259706583049e-07), (3, 2.304311237821821e-06), (4, 3.621142013798817e-06), (5, 3.5833952551911352e-06), (6, 6.574556209670845e-06), (7, 7.864392500778195e-06), (8, 5.167432391317561e-06), (9, 5.843068265676266e-06), (10, 1.0101453881361522e-05), (11, 1.645539487071801e-05), (12, 1.505570617155172e-05), (13, 1.9257649910287e-05), (14, 2.9125478249625303e-05), (15, 5.2766343287657946e-05), (16, 3.537001975928433e-05), (17, 3.986806405009702e-05), (18, 2.4537845092709176e-05), (19, 2.8255626602913253e-05), (20, 3.369632759131491e-05), (21, 7.94400621089153e-05), (22, 7.657466630917042e-05), (23, 9.638394840294495e-05), (24, 0.0002016351791098714), (25, 0.0006083960761316121), (26, 0.00022399793670047075), (27, 4.6255194320110604e-05)]
+
+ 72%|███████▏  | 108/150 [16:07<05:51,  8.38s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 246, Spatial MSE: [(0, 8.611126531832269e-07), (1, 7.666389137739316e-06), (2, 1.1432250175857916e-05), (3, 7.644460310984869e-06), (4, 1.749794705574459e-06), (5, 2.070086566163809e-06), (6, 1.8791741922541405e-06), (7, 3.959723017032957e-06), (8, 2.5918604933394818e-06), (9, 1.9287056147732073e-06), (10, 2.381376589255524e-06), (11, 2.9739317142229993e-06), (12, 6.490021860372508e-06), (13, 0.00017287473019678146), (14, 0.00020858908828813583), (15, 8.992622315417975e-05), (16, 6.826293974881992e-05), (17, 1.3524870155379176e-05), (18, 1.585300560691394e-05), (19, 2.9500035452656448e-05), (20, 1.2933453035657294e-05), (21, 6.264679541345686e-05), (22, 0.00013630175089929253), (23, 5.065795994596556e-05), (24, 9.985920769395307e-05), (25, 0.0001624532596906647), (26, 0.0002318685146747157), (27, 0.0013961605727672577)], Temporal MSE: [(0, 6.379998467309633e-07), (1, 3.237011583223648e-07), (2, 5.405801175584202e-07), (3, 9.172247246169718e-07), (4, 9.343256692773139e-07), (5, 1.1936888313357485e-06), (6, 1.5257428458426148e-06), (7, 1.5133771285036346e-06), (8, 8.733463232601935e-07), (9, 1.001647319753829e-06), (10, 1.3597169754575589e-06), (11, 2.527642891436699e-06), (12, 3.747565870071412e-06), (13, 6.103668965806719e-06), (14, 8.533585059922189e-06), (15, 5.410787252912996e-06), (16, 4.0547902244725265e-06), (17, 2.6869170142163057e-06), (18, 2.9452535272866953e-06), (19, 3.354442014824599e-06), (20, 4.9935288188862614e-06), (21, 5.7466804719297215e-05), (22, 3.7023957702331245e-05), (23, 3.528849265421741e-05), (24, 3.361361814313568e-05), (25, 8.774420712143183e-05), (26, 3.123080750810914e-05), (27, 5.829496330989059e-06)]
+
+ 73%|███████▎  | 109/150 [16:15<05:35,  8.17s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 240, Spatial MSE: [(0, 1.1802135304606054e-06), (1, 9.181685527437367e-06), (2, 1.4012021892995108e-05), (3, 1.2298242836550344e-05), (4, 1.5853482182137668e-05), (5, 3.088322773692198e-05), (6, 5.187165879760869e-05), (7, 9.94253350654617e-05), (8, 0.00012094574049115181), (9, 0.00015166608500294387), (10, 0.00011278927559033036), (11, 0.00014552618085872382), (12, 0.00017317375750280917), (13, 0.0005858134245499969), (14, 0.001338604255579412), (15, 0.0010836966102942824), (16, 0.02463349513709545), (17, 0.00036618331796489656), (18, 0.0005217607249505818), (19, 0.0014920505927875638), (20, 0.0011958011891692877), (21, 0.0006422111764550209), (22, 0.0007572825415991247), (23, 0.001182486186735332), (24, 0.0019373043905943632), (25, 0.004346597474068403), (26, 0.008102373220026493), (27, 0.001962800044566393)], Temporal MSE: [(0, 1.126870529333246e-06), (1, 4.997590394850704e-07), (2, 2.120870021826704e-06), (3, 6.2997069107950665e-06), (4, 1.1568531590455677e-05), (5, 1.7179221686092205e-05), (6, 4.240102498442866e-05), (7, 3.718920925166458e-05), (8, 2.1269368517096154e-05), (9, 3.3367075957357883e-05), (10, 4.90492457174696e-05), (11, 5.2875675464747474e-05), (12, 5.9532256273087114e-05), (13, 0.00011651792738121003), (14, 0.0002066894230665639), (15, 0.00018270633881911635), (16, 0.0001738806167850271), (17, 0.0001543370308354497), (18, 0.0001552069588797167), (19, 0.0002253777056466788), (20, 0.0003516124270390719), (21, 0.0004026668320875615), (22, 0.0003698026994243264), (23, 0.00046699843369424343), (24, 0.0011256755096837878), (25, 0.002072768984362483), (26, 0.0008123349398374557), (27, 7.227024616440758e-05)]
+
+ 73%|███████▎  | 110/150 [16:23<05:28,  8.21s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 234, Spatial MSE: [(0, 8.012643775145989e-07), (1, 7.559875484730583e-06), (2, 1.2890505786344875e-05), (3, 8.885773240763228e-06), (4, 2.035869101746357e-06), (5, 2.683921820789692e-06), (6, 2.1954517706035404e-06), (7, 4.359454123914475e-06), (8, 3.0741607588424813e-06), (9, 2.483264779584715e-06), (10, 2.7362182208889863e-06), (11, 3.819503035629168e-06), (12, 7.99403278506361e-06), (13, 0.00015978125156834722), (14, 0.00024445619783364236), (15, 0.0001256722171092406), (16, 0.001748264068737626), (17, 1.829839857236948e-05), (18, 4.5844499254599214e-05), (19, 7.28027371224016e-05), (20, 5.0605052820174024e-05), (21, 7.044841913739219e-05), (22, 0.00012291590974200517), (23, 7.000845653237775e-05), (24, 0.00013210093311499804), (25, 0.00023348582908511162), (26, 0.0002782437950372696), (27, 0.002037309342995286)], Temporal MSE: [(0, 7.807519750713254e-07), (1, 4.432100979556708e-07), (2, 6.748580858584319e-07), (3, 1.168005951512896e-06), (4, 1.1992997315246612e-06), (5, 1.4848142200207803e-06), (6, 2.007713192142546e-06), (7, 1.9485676148178754e-06), (8, 1.1766337593144272e-06), (9, 1.305669684370514e-06), (10, 1.683437972133106e-06), (11, 3.4662039070099127e-06), (12, 4.5518499973695725e-06), (13, 8.179110409400892e-06), (14, 9.25932818063302e-06), (15, 6.058247436158126e-06), (16, 5.762999535363633e-06), (17, 4.468213774089236e-06), (18, 6.621636657655472e-06), (19, 5.706925549020525e-06), (20, 7.647580787306651e-06), (21, 5.0715541874524206e-05), (22, 5.0884413212770596e-05), (23, 5.839364530402236e-05), (24, 5.266005973680876e-05), (25, 9.530538227409124e-05), (26, 3.7444671761477366e-05), (27, 1.161790805781493e-05)]
+
+ 74%|███████▍  | 111/150 [16:31<05:17,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 228, Spatial MSE: [(0, 8.831007676235458e-07), (1, 7.5377306529844645e-06), (2, 9.594825314707123e-06), (3, 6.613225650653476e-06), (4, 4.80153130411054e-06), (5, 9.36359901970718e-06), (6, 1.9655071810120717e-05), (7, 1.4063264643482398e-05), (8, 2.163683711842168e-05), (9, 3.902769822161645e-05), (10, 5.587674240814522e-05), (11, 7.053122681099921e-05), (12, 6.367688183672726e-05), (13, 0.00022166702547110617), (14, 0.0003442172601353377), (15, 0.00019394229457248002), (16, 0.0017705617938190699), (17, 5.2792951464653015e-05), (18, 0.00019479934417176992), (19, 0.0003377194225322455), (20, 0.0005661246250383556), (21, 0.0002547170442994684), (22, 0.00020833328017033637), (23, 0.00022750238713342696), (24, 0.0006506096106022596), (25, 0.002024357905611396), (26, 0.0043584927916526794), (27, 0.0019473149441182613)], Temporal MSE: [(0, 8.044855235311843e-07), (1, 4.570019029870309e-07), (2, 7.858935759941232e-07), (3, 2.284898073412478e-06), (4, 3.5979976473754505e-06), (5, 3.9905849007482175e-06), (6, 7.373842436209088e-06), (7, 9.639873496780638e-06), (8, 5.953925210633315e-06), (9, 7.3346050157852005e-06), (10, 9.752910045790486e-06), (11, 1.3933331501903012e-05), (12, 1.4224521692085546e-05), (13, 2.2533960873261094e-05), (14, 3.396240572328679e-05), (15, 2.1443209334393032e-05), (16, 1.7684003978502005e-05), (17, 1.3400751413428225e-05), (18, 2.0414352547959425e-05), (19, 3.244832259952091e-05), (20, 5.668998710461892e-05), (21, 9.041123848874122e-05), (22, 8.733248250791803e-05), (23, 0.00010136981290997937), (24, 0.00023021006199996918), (25, 0.0006191353313624859), (26, 0.00026466333656571805), (27, 3.9121416193665937e-05)]
+
+ 75%|███████▍  | 112/150 [16:39<05:11,  8.19s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 222, Spatial MSE: [(0, 7.817064897608361e-07), (1, 7.260742222570116e-06), (2, 1.2941495697305072e-05), (3, 8.412944225710817e-06), (4, 2.1041082618467044e-06), (5, 3.1386755381390685e-06), (6, 2.318839051440591e-06), (7, 5.263012553768931e-06), (8, 3.4296383546461584e-06), (9, 2.669236891961191e-06), (10, 3.987679974670755e-06), (11, 5.859071279701311e-06), (12, 1.4054588973522186e-05), (13, 0.0002619219885673374), (14, 0.0005342544754967093), (15, 0.00015282111417036504), (16, 0.010279017500579357), (17, 9.158358443528414e-05), (18, 4.060811625095084e-05), (19, 0.000875601195730269), (20, 0.0007092522573657334), (21, 7.749629730824381e-05), (22, 0.00010442936763865873), (23, 8.2705395470839e-05), (24, 0.0001495742326369509), (25, 0.00023739878088235855), (26, 0.00027381902327761054), (27, 0.004955217707902193)], Temporal MSE: [(0, 9.020687343763711e-07), (1, 5.030296961194836e-07), (2, 7.309819807232998e-07), (3, 1.2014792218906223e-06), (4, 1.2384632555040298e-06), (5, 1.5385294318548404e-06), (6, 2.057638766927994e-06), (7, 2.0453571778489277e-06), (8, 1.1934638450838975e-06), (9, 1.4091727962295408e-06), (10, 2.2448512027040124e-06), (11, 6.4884252424235456e-06), (12, 7.767659553792328e-06), (13, 7.992654900590423e-06), (14, 1.1708038073265925e-05), (15, 4.7452795115532354e-05), (16, 2.8656560971285217e-05), (17, 3.270264278398827e-05), (18, 1.4843747521808837e-05), (19, 1.4040065252629574e-05), (20, 1.2361741028144024e-05), (21, 7.598930096719414e-05), (22, 5.4752686992287636e-05), (23, 5.917685848544352e-05), (24, 5.441118992166594e-05), (25, 8.247842924902216e-05), (26, 3.936140637961216e-05), (27, 2.6338901079725474e-05)]
+
+ 75%|███████▌  | 113/150 [16:47<04:53,  7.94s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 216, Spatial MSE: [(0, 8.904239052753837e-07), (1, 9.630946806282736e-06), (2, 1.622671879886184e-05), (3, 1.2841559509979561e-05), (4, 1.5914562027319334e-05), (5, 3.1280233088182285e-05), (6, 5.3966934501659125e-05), (7, 0.00010541288065724075), (8, 0.0001305626065004617), (9, 0.00016122485976666212), (10, 0.00012656785838771611), (11, 0.00016239697288256139), (12, 0.00018868122424464673), (13, 0.0007177460938692093), (14, 0.0017557370010763407), (15, 0.0011220582528039813), (16, 0.02940332517027855), (17, 0.00046307628508657217), (18, 0.0004525847907643765), (19, 0.0021875351667404175), (20, 0.001960358116775751), (21, 0.0006525220233015716), (22, 0.0007295351242646575), (23, 0.0012659882195293903), (24, 0.0020725911017507315), (25, 0.0049053276889026165), (26, 0.008099410682916641), (27, 0.004959483630955219)], Temporal MSE: [(0, 9.039599149218702e-07), (1, 4.785323994838109e-07), (2, 2.162145165129914e-06), (3, 5.3477310757443774e-06), (4, 1.2654290003411006e-05), (5, 1.7550184566061944e-05), (6, 4.279099448467605e-05), (7, 4.0096947486745194e-05), (8, 2.3500382667407393e-05), (9, 3.624657983891666e-05), (10, 5.31495752511546e-05), (11, 6.141944322735071e-05), (12, 6.927821232238784e-05), (13, 0.00012679713836405426), (14, 0.00022516555327456445), (15, 0.00023634982062503695), (16, 0.0002099601406371221), (17, 0.00019169272854924202), (18, 0.0001794601557776332), (19, 0.00024314782058354467), (20, 0.000381001562345773), (21, 0.00039750809082761407), (22, 0.0003940252645406872), (23, 0.0005141316796652973), (24, 0.0011565330205485225), (25, 0.0019997041672468185), (26, 0.0009013224625959992), (27, 0.00010187285079155117)]
+
+ 76%|███████▌  | 114/150 [16:55<04:52,  8.14s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 210, Spatial MSE: [(0, 1.0906253464781912e-06), (1, 9.059535841515753e-06), (2, 1.3663894605997484e-05), (3, 8.738349606574047e-06), (4, 2.2439862732426263e-06), (5, 3.6246578929421958e-06), (6, 2.665225338205346e-06), (7, 5.893354227737291e-06), (8, 3.967286829720251e-06), (9, 2.973935579575482e-06), (10, 3.28477585753717e-06), (11, 4.960490969097009e-06), (12, 1.1141861250507645e-05), (13, 0.00030375257483683527), (14, 0.00036103956517763436), (15, 0.00018832046771422029), (16, 0.010317593812942505), (17, 5.703363422071561e-05), (18, 4.084729516762309e-05), (19, 0.00073158711893484), (20, 0.0006161557394079864), (21, 0.0001821407931856811), (22, 0.0001585226273164153), (23, 0.00010295184620190412), (24, 0.00019556109327822924), (25, 0.00034448382211849093), (26, 0.0003390816564206034), (27, 0.005646991543471813)], Temporal MSE: [(0, 9.294565188611159e-07), (1, 5.088627972327231e-07), (2, 7.706951805630524e-07), (3, 1.2833695564040681e-06), (4, 1.337180833616003e-06), (5, 1.6531292885701987e-06), (6, 2.0551137822621968e-06), (7, 2.478476517353556e-06), (8, 1.3873792568119825e-06), (9, 1.5843693290662486e-06), (10, 2.23716801883711e-06), (11, 6.895651495142374e-06), (12, 7.180522061389638e-06), (13, 8.304902621603105e-06), (14, 1.2097755643480923e-05), (15, 1.6024443539208733e-05), (16, 1.7971307897823863e-05), (17, 3.462891254457645e-05), (18, 1.8280883523402736e-05), (19, 1.632501880521886e-05), (20, 1.4563489457941614e-05), (21, 8.534455992048606e-05), (22, 7.3228518886026e-05), (23, 8.459224045509472e-05), (24, 7.452881254721433e-05), (25, 0.0001524347608210519), (26, 5.0322571041760966e-05), (27, 3.256793206674047e-05)]
+
+ 77%|███████▋  | 115/150 [17:05<04:58,  8.53s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 204, Spatial MSE: [(0, 1.0478598824192886e-06), (1, 8.899361091607716e-06), (2, 1.6411115211667493e-05), (3, 7.248064321174752e-06), (4, 5.016249360778602e-06), (5, 9.299538760387804e-06), (6, 1.8840950360754505e-05), (7, 1.400041401211638e-05), (8, 2.4749318981776014e-05), (9, 3.9686776290182024e-05), (10, 5.559697456192225e-05), (11, 7.355085836024955e-05), (12, 7.057494804030284e-05), (13, 0.0004415445728227496), (14, 0.0008521986892446876), (15, 0.0005732495919801295), (16, 0.014859465882182121), (17, 0.0001132551406044513), (18, 9.654322639107704e-05), (19, 0.0009366106241941452), (20, 0.0009522192412987351), (21, 0.00028051622211933136), (22, 0.00026096365763805807), (23, 0.000259260181337595), (24, 0.000708270410541445), (25, 0.0019636780489236116), (26, 0.004308382049202919), (27, 0.004348783288151026)], Temporal MSE: [(0, 8.610183499513369e-07), (1, 5.258784199213551e-07), (2, 8.336090218108438e-07), (3, 2.5580061446817126e-06), (4, 3.7410425193229457e-06), (5, 4.125618943362497e-06), (6, 7.497887963836547e-06), (7, 1.0257044777972624e-05), (8, 6.6906623032991774e-06), (9, 7.594908311148174e-06), (10, 1.110173798224423e-05), (11, 1.967967727978248e-05), (12, 1.942903509188909e-05), (13, 2.3423684979206882e-05), (14, 3.947796722059138e-05), (15, 3.837118492810987e-05), (16, 3.236379416193813e-05), (17, 4.137916039326228e-05), (18, 3.246749474783428e-05), (19, 3.9865251892479137e-05), (20, 7.115067273844033e-05), (21, 0.0001115914637921378), (22, 0.00010050104901893064), (23, 0.00012372735363896936), (24, 0.00026416091714054346), (25, 0.0006439125863835216), (26, 0.0003046161727979779), (27, 5.676357977790758e-05)]
+
+ 77%|███████▋  | 116/150 [17:14<04:54,  8.67s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 198, Spatial MSE: [(0, 9.112464454119618e-07), (1, 1.0706188732001465e-05), (2, 1.47669852594845e-05), (3, 7.98083874542499e-06), (4, 2.370900801906828e-06), (5, 3.275456947449129e-06), (6, 3.228502464480698e-06), (7, 5.486888312589144e-06), (8, 4.594120582623873e-06), (9, 4.076105597050628e-06), (10, 1.3322164704732131e-05), (11, 5.318606781656854e-05), (12, 7.542184175690636e-05), (13, 0.0013604698469862342), (14, 0.0011353015433996916), (15, 0.0004348282527644187), (16, 0.014093473553657532), (17, 6.458728603320196e-05), (18, 5.191779928281903e-05), (19, 0.0011654399568215013), (20, 0.0008814362226985395), (21, 0.0001590493047842756), (22, 0.00024331387248821557), (23, 0.0001449919946026057), (24, 0.0002489958715159446), (25, 0.00043211146839894354), (26, 0.00040590594289824367), (27, 0.01091192476451397)], Temporal MSE: [(0, 9.126223972089065e-07), (1, 5.254001393950603e-07), (2, 7.502257517444377e-07), (3, 1.222712739945564e-06), (4, 1.3461042271956103e-06), (5, 1.605637635293533e-06), (6, 2.4182527340599336e-06), (7, 2.3742397843307117e-06), (8, 1.4394006484508282e-06), (9, 2.21047957893461e-06), (10, 7.835076758055948e-06), (11, 0.00010841383482329547), (12, 2.848525036824867e-05), (13, 2.0515992218861356e-05), (14, 1.9079869161942042e-05), (15, 4.144609556533396e-05), (16, 2.116294126608409e-05), (17, 3.6262135836295784e-05), (18, 3.671107697300613e-05), (19, 3.1945586670190096e-05), (20, 4.418630851432681e-05), (21, 0.00013194454368203878), (22, 9.238519123755395e-05), (23, 0.0001144544075941667), (24, 9.889246575767174e-05), (25, 0.0002158236165996641), (26, 6.9082023401279e-05), (27, 4.934176104143262e-05)]
+
+ 78%|███████▊  | 117/150 [17:24<04:58,  9.05s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 192, Spatial MSE: [(0, 1.2471728041418828e-06), (1, 1.0068423762277234e-05), (2, 2.0867888451903127e-05), (3, 1.131642875407124e-05), (4, 1.638057619857136e-05), (5, 3.1898340239422396e-05), (6, 5.582592712016776e-05), (7, 0.00010416381701361388), (8, 0.00013175909407436848), (9, 0.0001726766349747777), (10, 0.0001417687744833529), (11, 0.00021896026737522334), (12, 0.0002463257987983525), (13, 0.0017139810370281339), (14, 0.0016909510595723987), (15, 0.0008700587204657495), (16, 0.014631626196205616), (17, 0.0004310224612709135), (18, 0.0004511773004196584), (19, 0.0016487293178215623), (20, 0.001389791606925428), (21, 0.0006953373667784035), (22, 0.0008303578360937536), (23, 0.0013536422047764063), (24, 0.002201198600232601), (25, 0.00553717277944088), (26, 0.009748412296175957), (27, 0.01007989514619112)], Temporal MSE: [(0, 1.4006017181600328e-06), (1, 6.597085189241625e-07), (2, 2.1916162040724885e-06), (3, 5.284079634293448e-06), (4, 1.3350714652915485e-05), (5, 1.7833344827522524e-05), (6, 4.67009303974919e-05), (7, 4.378112862468697e-05), (8, 2.5205905330949463e-05), (9, 3.936554639949463e-05), (10, 6.151718844193965e-05), (11, 0.00016713765216991305), (12, 8.94069453352131e-05), (13, 0.00014040821406524628), (14, 0.00023084614076651633), (15, 0.00020601152209565043), (16, 0.00019587976566981524), (17, 0.00017869225121103227), (18, 0.00018294768233317882), (19, 0.00025101122446358204), (20, 0.0003792661882471293), (21, 0.00043905043276026845), (22, 0.0004160802054684609), (23, 0.0005509066977538168), (24, 0.00122752133756876), (25, 0.002276636892929673), (26, 0.0010224076686426997), (27, 0.00013058965851087123)]
+
+ 79%|███████▊  | 118/150 [17:34<05:03,  9.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 186, Spatial MSE: [(0, 9.549704600431141e-07), (1, 1.0241489690088201e-05), (2, 1.526698179077357e-05), (3, 8.956573765317444e-06), (4, 2.3858631266193697e-06), (5, 3.5053326428169385e-06), (6, 2.94639994535828e-06), (7, 5.772337772214087e-06), (8, 4.112998794880696e-06), (9, 3.274573600720032e-06), (10, 3.981293502874905e-06), (11, 4.84728980154614e-06), (12, 8.89057082531508e-06), (13, 0.0002460766118019819), (14, 0.00042548507917672396), (15, 0.0002985452301800251), (16, 0.013133499771356583), (17, 3.928266596631147e-05), (18, 0.00010259084228891879), (19, 0.0011735595762729645), (20, 0.0009748744778335094), (21, 0.00013797159772366285), (22, 0.00017290150572080165), (23, 0.00012166586384410039), (24, 0.00022623685072176158), (25, 0.00042445247527211905), (26, 0.0003227191336918622), (27, 0.006462533492594957)], Temporal MSE: [(0, 8.75545254075405e-07), (1, 5.204886974752299e-07), (2, 8.059120659709151e-07), (3, 1.3125713849149179e-06), (4, 1.5620191788912052e-06), (5, 1.8444744682710734e-06), (6, 2.7893102014786564e-06), (7, 2.572173571024905e-06), (8, 1.6555524098293972e-06), (9, 1.8592679680295987e-06), (10, 2.4654113985889126e-06), (11, 4.3572435970418155e-06), (12, 5.496555331774289e-06), (13, 9.020435754791833e-06), (14, 1.5245260328811128e-05), (15, 1.5845591406105086e-05), (16, 1.2396341844578274e-05), (17, 1.88673147931695e-05), (18, 3.37535711878445e-05), (19, 2.78349980362691e-05), (20, 2.6552481358521618e-05), (21, 0.00013575327466242015), (22, 9.649395360611379e-05), (23, 0.00010374924022471532), (24, 8.64985995576717e-05), (25, 0.00019337779667694122), (26, 5.0898812332889065e-05), (27, 4.078737401869148e-05)]
+
+ 79%|███████▉  | 119/150 [17:45<05:08,  9.95s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 180, Spatial MSE: [(0, 1.2300454272917705e-06), (1, 9.05157776287524e-06), (2, 1.8552638721303083e-05), (3, 7.436021860485198e-06), (4, 5.7485676734359e-06), (5, 1.0760596524050925e-05), (6, 1.9552042431314476e-05), (7, 1.599936149432324e-05), (8, 2.453024899295997e-05), (9, 4.5627290091942996e-05), (10, 5.716222949558869e-05), (11, 8.212280226871371e-05), (12, 6.646057590842247e-05), (13, 0.0003014357353094965), (14, 0.0005019199452362955), (15, 0.00037270254688337445), (16, 0.013201352208852768), (17, 7.794357225066051e-05), (18, 0.00016285367019008845), (19, 0.0012195775052532554), (20, 0.0010294493986293674), (21, 0.00019313018128741533), (22, 0.00027255076565779746), (23, 0.00031653515179641545), (24, 0.0008634747355245054), (25, 0.002403885591775179), (26, 0.004777624271810055), (27, 0.007212725933641195)], Temporal MSE: [(0, 1.1947156508540502e-06), (1, 7.454580099874875e-07), (2, 9.98571749732946e-07), (3, 2.717882807701244e-06), (4, 4.4138669181847945e-06), (5, 4.599150997819379e-06), (6, 9.348975254397374e-06), (7, 1.206529759656405e-05), (8, 8.145043466356583e-06), (9, 9.208278243022505e-06), (10, 1.2603092727658805e-05), (11, 1.7207799828611314e-05), (12, 1.7378075426677242e-05), (13, 2.4913764718803577e-05), (14, 4.2933290387736633e-05), (15, 3.486252899165265e-05), (16, 2.866192335204687e-05), (17, 3.137369276373647e-05), (18, 4.872522549703717e-05), (19, 5.2544823120115325e-05), (20, 6.0931000916752964e-05), (21, 0.0001672616635914892), (22, 0.00013937124458607286), (23, 0.00016510770365130156), (24, 0.00032788299722597003), (25, 0.0007644828292541206), (26, 0.0003658413770608604), (27, 8.566554606659338e-05)]
+
+ 80%|████████  | 120/150 [17:57<05:14, 10.50s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 174, Spatial MSE: [(0, 8.263682502729353e-07), (1, 1.066741424438078e-05), (2, 1.5827255992917344e-05), (3, 8.156593139574397e-06), (4, 2.351331886529806e-06), (5, 3.769483328142087e-06), (6, 3.071099172302638e-06), (7, 7.149042176024523e-06), (8, 5.175993010197999e-06), (9, 3.3210137644346105e-06), (10, 5.999899713060586e-06), (11, 9.21449282031972e-06), (12, 1.781473656592425e-05), (13, 0.0003880567674059421), (14, 0.0008219571900554001), (15, 0.0002700586919672787), (16, 0.020404519513249397), (17, 0.00012978061567991972), (18, 5.6843229685910046e-05), (19, 0.0017043199623003602), (20, 0.0013996957568451762), (21, 0.0001464894594391808), (22, 0.0001622081035748124), (23, 0.00012506173516158015), (24, 0.00022869445092510432), (25, 0.00039370032027363777), (26, 0.0003509167581796646), (27, 0.00872507132589817)], Temporal MSE: [(0, 9.781850849321927e-07), (1, 5.761525585512572e-07), (2, 8.283323040814139e-07), (3, 1.299477844440844e-06), (4, 1.5554730907751946e-06), (5, 1.823347020035726e-06), (6, 2.8427612051018514e-06), (7, 2.816495907609351e-06), (8, 1.6662748976159492e-06), (9, 2.0353281797724776e-06), (10, 3.139934733553673e-06), (11, 8.998161320050713e-06), (12, 1.0629019016050734e-05), (13, 1.2263322787475772e-05), (14, 1.91918334166985e-05), (15, 6.215004395926371e-05), (16, 4.483137308852747e-05), (17, 6.589181430172175e-05), (18, 2.9007824196014553e-05), (19, 2.6267314751748927e-05), (20, 2.1550824385485612e-05), (21, 0.00013344323087949306), (22, 9.574751311447471e-05), (23, 0.0001040577408275567), (24, 8.91541494638659e-05), (25, 0.00015650306886527687), (26, 5.4225532949203625e-05), (27, 5.149204662302509e-05)]
+
+ 81%|████████  | 121/150 [18:08<05:12, 10.78s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 168, Spatial MSE: [(0, 1.1857900972245261e-06), (1, 1.0993596333719324e-05), (2, 2.7658146791509353e-05), (3, 1.2731042261293624e-05), (4, 1.5952051398926415e-05), (5, 3.191909854649566e-05), (6, 5.938623507972807e-05), (7, 0.00010843506606761366), (8, 0.00013780524022877216), (9, 0.00016678081010468304), (10, 0.0001328554644715041), (11, 0.0001778158766683191), (12, 0.00018631707644090056), (13, 0.0006058048456907272), (14, 0.001225424581207335), (15, 0.0007436490850523114), (16, 0.02089511603116989), (17, 0.0005051909829489887), (18, 0.0004604957939591259), (19, 0.002152896486222744), (20, 0.0019382643513381481), (21, 0.0007343327160924673), (22, 0.0008116642129607499), (23, 0.0014724205248057842), (24, 0.00240030768327415), (25, 0.006307049188762903), (26, 0.011311638168990612), (27, 0.008647611364722252)], Temporal MSE: [(0, 1.5187500821411959e-06), (1, 7.718971346548642e-07), (2, 2.44218449552136e-06), (3, 6.074259090382839e-06), (4, 1.5485076801269315e-05), (5, 1.833545320550911e-05), (6, 4.9831418436951935e-05), (7, 4.616820297087543e-05), (8, 2.7560094167711213e-05), (9, 4.075584001839161e-05), (10, 5.8409710618434474e-05), (11, 6.983405182836577e-05), (12, 7.334975816775113e-05), (13, 0.00013522626250050962), (14, 0.0002461372932884842), (15, 0.000252067344263196), (16, 0.00023115314252208918), (17, 0.0002247517986688763), (18, 0.00018501385056879371), (19, 0.0002637873694766313), (20, 0.0004004504007752985), (21, 0.00046788089093752205), (22, 0.0004552783502731472), (23, 0.0005822349921800196), (24, 0.0012660461943596601), (25, 0.0022595254704356194), (26, 0.0011926890583708882), (27, 0.0001619597605895251)]
+
+ 81%|████████▏ | 122/150 [18:20<05:12, 11.15s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 162, Spatial MSE: [(0, 1.0927078619715758e-06), (1, 1.2233206689415965e-05), (2, 1.7603833839530125e-05), (3, 9.214600140694529e-06), (4, 2.6771788270707475e-06), (5, 5.280453024170129e-06), (6, 3.6387687032402027e-06), (7, 8.68917595653329e-06), (8, 5.980833066132618e-06), (9, 3.5563234632718377e-06), (10, 4.539196197583806e-06), (11, 5.316153419698821e-06), (12, 8.953631549957208e-06), (13, 0.00016333967505488545), (14, 0.00015115497808437794), (15, 6.332138582365587e-05), (16, 0.0001667978649493307), (17, 1.5559660823782906e-05), (18, 3.8062880776124075e-05), (19, 3.14299322781153e-05), (20, 1.764713306329213e-05), (21, 0.00018029462080448866), (22, 9.769330063136294e-05), (23, 7.346292113652453e-05), (24, 0.00012962166510988027), (25, 0.00022939823975320905), (26, 0.00031485530780628324), (27, 0.002238212153315544)], Temporal MSE: [(0, 1.0799234360092669e-06), (1, 6.592420049855718e-07), (2, 9.395753295393661e-07), (3, 1.4524624702971778e-06), (4, 1.7516282468932332e-06), (5, 2.1020080112066353e-06), (6, 3.4155164030380547e-06), (7, 3.4212880564155057e-06), (8, 2.0286806829972193e-06), (9, 2.3002532998361858e-06), (10, 2.8948861654498614e-06), (11, 5.060533112555277e-06), (12, 5.292343757901108e-06), (13, 6.984426363487728e-06), (14, 9.132323611993343e-06), (15, 8.430889465671498e-06), (16, 8.223465556511655e-06), (17, 6.517605925182579e-06), (18, 5.959229383734055e-06), (19, 6.514108463306911e-06), (20, 8.609717951912899e-06), (21, 3.7103989598108456e-05), (22, 3.672789171105251e-05), (23, 3.99843993363902e-05), (24, 4.331556920078583e-05), (25, 9.507468348601833e-05), (26, 5.247575245448388e-05), (27, 1.672015423537232e-05)]
+
+ 82%|████████▏ | 123/150 [18:31<04:59, 11.10s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 156, Spatial MSE: [(0, 1.2431273717083968e-06), (1, 1.0398399354016874e-05), (2, 2.485749064362608e-05), (3, 7.349500265263487e-06), (4, 6.9210082074278034e-06), (5, 1.1922430530830752e-05), (6, 1.914365748234559e-05), (7, 1.8085627743857913e-05), (8, 3.062854375457391e-05), (9, 5.532828072318807e-05), (10, 6.861601286800578e-05), (11, 9.279338701162487e-05), (12, 7.544501568190753e-05), (13, 0.0002304887748323381), (14, 0.0002667955122888088), (15, 0.00015716072812210768), (16, 0.00024204964574892074), (17, 6.732763722538948e-05), (18, 6.407750333892182e-05), (19, 8.400835213251412e-05), (20, 8.410584268858656e-05), (21, 0.00024216785095632076), (22, 0.00018684734823182225), (23, 0.00027570867678150535), (24, 0.0007530669099651277), (25, 0.0021950150839984417), (26, 0.00448097288608551), (27, 0.0019318675622344017)], Temporal MSE: [(0, 1.3441908777167555e-06), (1, 8.36592164432659e-07), (2, 1.0623612070048694e-06), (3, 2.7489907097333344e-06), (4, 4.758938302984461e-06), (5, 4.871973033004906e-06), (6, 9.174502338282764e-06), (7, 1.2964042070962023e-05), (8, 8.403199899476022e-06), (9, 9.995116670324933e-06), (10, 1.2390570191200823e-05), (11, 1.7157553884317167e-05), (12, 1.6340851288987324e-05), (13, 2.2809970687376335e-05), (14, 3.5785204090643674e-05), (15, 2.542754191381391e-05), (16, 2.3343172870227136e-05), (17, 1.6917534594540484e-05), (18, 2.135298927896656e-05), (19, 2.986549043271225e-05), (20, 4.230250488035381e-05), (21, 7.023208308964968e-05), (22, 7.653923967154697e-05), (23, 0.00010016612213803455), (24, 0.0002849541779141873), (25, 0.000608532689511776), (26, 0.00039676050073467195), (27, 7.320696749957278e-05)]
+
+ 83%|████████▎ | 124/150 [18:42<04:48, 11.09s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 150, Spatial MSE: [(0, 1.0636335900926497e-06), (1, 1.2565336874104105e-05), (2, 1.8715350961429067e-05), (3, 9.14120209927205e-06), (4, 2.8714057407341897e-06), (5, 4.987496140529402e-06), (6, 4.246808202879038e-06), (7, 8.559500201954506e-06), (8, 6.390193902916508e-06), (9, 3.992298388766358e-06), (10, 5.037822120357305e-06), (11, 5.3968924476066604e-06), (12, 1.224956304213265e-05), (13, 0.0005616165581159294), (14, 0.0013652696507051587), (15, 0.0006244305986911058), (16, 0.05112767964601517), (17, 0.0002217433793703094), (18, 0.000134319081553258), (19, 0.004251527134329081), (20, 0.0035145801957696676), (21, 0.0002580724249128252), (22, 0.00029451213777065277), (23, 0.0002337466721655801), (24, 0.00046842393931001425), (25, 0.0009268578724004328), (26, 0.0005595121765509248), (27, 0.01866295374929905)], Temporal MSE: [(0, 1.24425889680424e-06), (1, 7.745512107248942e-07), (2, 1.062325850398338e-06), (3, 1.5892878764134366e-06), (4, 2.0004833913844777e-06), (5, 2.2982746941124788e-06), (6, 3.778727887038258e-06), (7, 3.768793476410792e-06), (8, 2.37238964473363e-06), (9, 2.6613090540195117e-06), (10, 3.1874835713097127e-06), (11, 6.1409282352542505e-06), (12, 9.825853339862078e-06), (13, 2.1995818315190263e-05), (14, 3.6538156564347446e-05), (15, 6.577519525308162e-05), (16, 8.408551366301253e-05), (17, 0.00016251120541710407), (18, 7.434361759806052e-05), (19, 6.522412149934098e-05), (20, 4.9821363063529134e-05), (21, 0.0003177217731717974), (22, 0.00021804423886351287), (23, 0.0002326869434909895), (24, 0.00017882732208818197), (25, 0.0004455379385035485), (26, 8.976544631877914e-05), (27, 0.00011105997691629454)]
+
+ 83%|████████▎ | 125/150 [18:53<04:34, 10.98s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 144, Spatial MSE: [(0, 1.2081352451787097e-06), (1, 1.2410951057972852e-05), (2, 3.47888380929362e-05), (3, 1.2957507351529785e-05), (4, 1.666549542278517e-05), (5, 3.440073487581685e-05), (6, 6.367201422108337e-05), (7, 0.00012265190889593214), (8, 0.00016447958478238434), (9, 0.00018319320224691182), (10, 0.00013984585530124605), (11, 0.00019386294297873974), (12, 0.00019334744138177484), (13, 0.0007962554227560759), (14, 0.0017363752704113722), (15, 0.0011385242687538266), (16, 0.05175768584012985), (17, 0.0006075989222154021), (18, 0.0005495924269780517), (19, 0.00470622256398201), (20, 0.0040886858478188515), (21, 0.0008759634802117944), (22, 0.0009987978264689445), (23, 0.0016888438258320093), (24, 0.002892659744247794), (25, 0.007589518558233976), (26, 0.012101907283067703), (27, 0.018379533663392067)], Temporal MSE: [(0, 1.692093633209879e-06), (1, 8.427387001574971e-07), (2, 2.452663920848863e-06), (3, 7.068641934893094e-06), (4, 1.7949216271517798e-05), (5, 1.9951921785832383e-05), (6, 6.356155790854245e-05), (7, 5.475762372952886e-05), (8, 3.2580646802671254e-05), (9, 4.696052928920835e-05), (10, 6.439186836360022e-05), (11, 7.630442996742204e-05), (12, 8.018648804863915e-05), (13, 0.00015369530592579395), (14, 0.0002734961162786931), (15, 0.000261354522081092), (16, 0.00027381384279578924), (17, 0.00032809373806230724), (18, 0.0002416927454760298), (19, 0.0003084921045228839), (20, 0.00043891993118450046), (21, 0.000651011650916189), (22, 0.0005967976176179945), (23, 0.0007562047685496509), (24, 0.0014470161404460669), (25, 0.002611299278214574), (26, 0.0013659901451319456), (27, 0.00023794618027750403)]
+
+ 84%|████████▍ | 126/150 [19:02<04:08, 10.37s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 138, Spatial MSE: [(0, 1.132287707150681e-06), (1, 1.4351854588312563e-05), (2, 2.2560596335097216e-05), (3, 1.0749701687018387e-05), (4, 3.322641305203433e-06), (5, 5.198377493798034e-06), (6, 4.616020760295214e-06), (7, 8.074588549789041e-06), (8, 6.635353656747611e-06), (9, 5.1242459448985755e-06), (10, 6.072319592931308e-06), (11, 1.0660428415576462e-05), (12, 2.8091357307857834e-05), (13, 0.000781979993917048), (14, 0.0008814035099931061), (15, 0.0002210792590631172), (16, 0.020467234775424004), (17, 0.00016188574954867363), (18, 5.3545594710158184e-05), (19, 0.0016810274682939053), (20, 0.0013857899466529489), (21, 9.12761315703392e-05), (22, 0.00017705779464449733), (23, 0.00015800996334291995), (24, 0.00029408562113530934), (25, 0.0005217068828642368), (26, 0.0004579727246891707), (27, 0.008563303388655186)], Temporal MSE: [(0, 1.3830847365170484e-06), (1, 8.240580200435943e-07), (2, 1.2136056284361985e-06), (3, 1.7896716144605307e-06), (4, 2.402344080110197e-06), (5, 2.7336368475516792e-06), (6, 4.308012194087496e-06), (7, 4.2624419620551635e-06), (8, 2.9352681849559303e-06), (9, 3.206222800145042e-06), (10, 3.972173544752877e-06), (11, 2.070957634714432e-05), (12, 2.0822599253733642e-05), (13, 1.7204818504978903e-05), (14, 1.692938167252578e-05), (15, 8.300290210172534e-05), (16, 6.283335824264213e-05), (17, 7.10251260898076e-05), (18, 3.0622195481555536e-05), (19, 2.878526720451191e-05), (20, 2.4556364223826677e-05), (21, 0.0001225017331307754), (22, 8.713627903489396e-05), (23, 0.00011473314953036606), (24, 0.00010792339162435383), (25, 0.0001980118831852451), (26, 8.362748485524207e-05), (27, 5.852599497302435e-05)]
+
+ 85%|████████▍ | 127/150 [19:11<03:48,  9.93s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 132, Spatial MSE: [(0, 1.3223415180618758e-06), (1, 1.2492579116951674e-05), (2, 3.2930121960816905e-05), (3, 8.295664883917198e-06), (4, 8.149132554535754e-06), (5, 1.4534690308209974e-05), (6, 2.402360041742213e-05), (7, 2.1990654204273596e-05), (8, 3.20056606142316e-05), (9, 5.72069657209795e-05), (10, 7.455542800016701e-05), (11, 0.00011766351235564798), (12, 0.00010157092765439302), (13, 0.0008632061653770506), (14, 0.001022388692945242), (15, 0.00032947221188806), (16, 0.020663227885961533), (17, 0.0002159516152460128), (18, 0.00012102220352971926), (19, 0.001743948902003467), (20, 0.0014712351839989424), (21, 0.00018726789858192205), (22, 0.0003092557890340686), (23, 0.00043851652299053967), (24, 0.0011238233419135213), (25, 0.0032763162162154913), (26, 0.006098304875195026), (27, 0.009419308975338936)], Temporal MSE: [(0, 1.6477323470098781e-06), (1, 1.0973398048008676e-06), (2, 1.2918466154587804e-06), (3, 3.119919938399107e-06), (4, 5.6766716625133995e-06), (5, 5.675192369380966e-06), (6, 1.3404352102952544e-05), (7, 1.5623256331309676e-05), (8, 1.1092378372268286e-05), (9, 1.2602243259607349e-05), (10, 1.657849679759238e-05), (11, 3.635971006588079e-05), (12, 3.4389784559607506e-05), (13, 3.6882542190141976e-05), (14, 5.3435374866239727e-05), (15, 0.00010660327097866684), (16, 8.378107304451987e-05), (17, 8.537616668036208e-05), (18, 5.12134465679992e-05), (19, 5.916743975831196e-05), (20, 7.230664778035134e-05), (21, 0.00017224985640496016), (22, 0.00014314457075670362), (23, 0.00019912458083126694), (24, 0.00045651631080545485), (25, 0.000949417648371309), (26, 0.0005733125144615769), (27, 0.00014477968215942383)]
+
+ 85%|████████▌ | 128/150 [19:19<03:29,  9.51s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 126, Spatial MSE: [(0, 1.0689661849028198e-06), (1, 1.502750728832325e-05), (2, 2.4005144950933754e-05), (3, 1.0068238225358073e-05), (4, 3.4073023016389925e-06), (5, 5.388149475038517e-06), (6, 4.71195789941703e-06), (7, 8.404747859458439e-06), (8, 7.124753210518975e-06), (9, 4.7946846279955935e-06), (10, 7.179334261309123e-06), (11, 9.04711487237364e-06), (12, 2.0408248019521125e-05), (13, 0.000738301663659513), (14, 0.0013168588047847152), (15, 0.000433550652815029), (16, 0.030698321759700775), (17, 0.00016386639617849141), (18, 8.544899174012244e-05), (19, 0.0025566888507455587), (20, 0.0020977556705474854), (21, 0.00033860706025734544), (22, 0.00030271170544438064), (23, 0.00021311576711013913), (24, 0.00041825894732028246), (25, 0.0007689055637456477), (26, 0.000500026042573154), (27, 0.01166088879108429)], Temporal MSE: [(0, 1.5023272226244444e-06), (1, 8.817657999315998e-07), (2, 1.2355486660453607e-06), (3, 1.8112864381691907e-06), (4, 2.435199121464393e-06), (5, 2.8301637939875945e-06), (6, 3.957130047638202e-06), (7, 4.445385457074735e-06), (8, 3.1449303605768364e-06), (9, 3.2594421099929605e-06), (10, 4.074493517691735e-06), (11, 1.2743158549710643e-05), (12, 1.5205426279862877e-05), (13, 2.395784031250514e-05), (14, 2.99040930258343e-05), (15, 8.42874578665942e-05), (16, 7.261999417096376e-05), (17, 0.0001092262173187919), (18, 4.822552000405267e-05), (19, 4.289211210561916e-05), (20, 3.637537156464532e-05), (21, 0.00025313091464340687), (22, 0.00018147584341932088), (23, 0.0001854792790254578), (24, 0.00015098814037628472), (25, 0.0003098130982834846), (26, 9.958240116247907e-05), (27, 7.939839997561648e-05)]
+
+ 86%|████████▌ | 129/150 [19:28<03:11,  9.13s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 120, Spatial MSE: [(0, 1.2662374047067715e-06), (1, 1.4602484043280128e-05), (2, 4.281042856746353e-05), (3, 1.321151194133563e-05), (4, 1.8604991055326536e-05), (5, 3.9623566408408806e-05), (6, 8.446350693702698e-05), (7, 0.00013696763198822737), (8, 0.0001714794198051095), (9, 0.00022176621132530272), (10, 0.00015971550601534545), (11, 0.0002123018348356709), (12, 0.0002118959673680365), (13, 0.001060991664417088), (14, 0.002036057645455003), (15, 0.0012864107266068459), (16, 0.03233722969889641), (17, 0.0006606560782529414), (18, 0.0005991682410240173), (19, 0.0032852194271981716), (20, 0.0029769211541861296), (21, 0.001045174547471106), (22, 0.0011179771972820163), (23, 0.0018516675336286426), (24, 0.003228256478905678), (25, 0.009684422984719276), (26, 0.014868359081447124), (27, 0.012523542158305645)], Temporal MSE: [(0, 1.9040859342567273e-06), (1, 1.109823074330052e-06), (2, 2.7185167255083797e-06), (3, 1.1733886822185013e-05), (4, 2.1524116164073348e-05), (5, 2.295814556418918e-05), (6, 8.612028614152223e-05), (7, 6.834083615103737e-05), (8, 3.971378464484587e-05), (9, 5.6257907999679446e-05), (10, 7.467793329851702e-05), (11, 9.533455886412412e-05), (12, 9.751348261488602e-05), (13, 0.00016852648695930839), (14, 0.0002963927108794451), (15, 0.00029249207000248134), (16, 0.0002812882012221962), (17, 0.0002941686543636024), (18, 0.00022728693147655576), (19, 0.0003218264027964324), (20, 0.0004839606408495456), (21, 0.0006093965494073927), (22, 0.0005926124285906553), (23, 0.0007690124330110848), (24, 0.0016120835207402706), (25, 0.0028017591685056686), (26, 0.0017807104159146547), (27, 0.00027670356212183833)]
+
+ 87%|████████▋ | 130/150 [19:36<02:59,  8.96s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 114, Spatial MSE: [(0, 1.1736900660253013e-06), (1, 1.825614344852511e-05), (2, 2.63023903244175e-05), (3, 1.1120121598651167e-05), (4, 3.826984993793303e-06), (5, 6.8346735133673064e-06), (6, 5.947842510067858e-06), (7, 1.1581054423004389e-05), (8, 9.235664947482292e-06), (9, 6.022055913490476e-06), (10, 7.307690339075634e-06), (11, 1.0321660738554783e-05), (12, 2.321216743439436e-05), (13, 0.0006834569503553212), (14, 0.0008186805061995983), (15, 0.00033829532912932336), (16, 0.02459961175918579), (17, 0.00010790936357807368), (18, 8.557667024433613e-05), (19, 0.0019223267445340753), (20, 0.0015054454561322927), (21, 0.00016311017679981887), (22, 0.00022598884243052453), (23, 0.0002040070976363495), (24, 0.0003597477625589818), (25, 0.0007198690436780453), (26, 0.0006116863223724067), (27, 0.011897411197423935)], Temporal MSE: [(0, 1.671346240073035e-06), (1, 1.0383506605649018e-06), (2, 1.4006292303747614e-06), (3, 1.9919855276384624e-06), (4, 2.938221768999938e-06), (5, 3.2052669212134788e-06), (6, 6.470179414463928e-06), (7, 5.323699497239431e-06), (8, 3.823668521363288e-06), (9, 4.1792673073359765e-06), (10, 4.910909410682507e-06), (11, 1.750519368215464e-05), (12, 1.7353457224089652e-05), (13, 2.072540155495517e-05), (14, 2.3965167201822624e-05), (15, 4.763803372043185e-05), (16, 5.214846896706149e-05), (17, 8.348625124199316e-05), (18, 4.807701407116838e-05), (19, 4.3568859837250784e-05), (20, 4.005151640740223e-05), (21, 0.00014587774057872593), (22, 0.0001155464633484371), (23, 0.00015130122483242303), (24, 0.0001427343813702464), (25, 0.00033975139376707375), (26, 0.00011929573520319536), (27, 8.889508171705529e-05)]
+
+ 87%|████████▋ | 131/150 [19:44<02:45,  8.71s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 108, Spatial MSE: [(0, 1.4723551657880307e-06), (1, 1.5818883184692822e-05), (2, 5.194290497456677e-05), (3, 9.473117643210571e-06), (4, 9.58343571255682e-06), (5, 1.9431054170127027e-05), (6, 2.711731576710008e-05), (7, 2.9768585591227747e-05), (8, 4.7929912398103625e-05), (9, 6.281406967900693e-05), (10, 9.879108256427571e-05), (11, 0.000131515073007904), (12, 0.0001271657965844497), (13, 0.0010480120545253158), (14, 0.001499504898674786), (15, 0.0007359524024650455), (16, 0.02930220402777195), (17, 0.0002077559765893966), (18, 0.00018177203310187906), (19, 0.0023240214213728905), (20, 0.0019391413079574704), (21, 0.00036043176078237593), (22, 0.000392782996641472), (23, 0.0005867393338121474), (24, 0.0015147164231166244), (25, 0.004023888148367405), (26, 0.007148503325879574), (27, 0.012028439901769161)], Temporal MSE: [(0, 2.0125071387155913e-06), (1, 1.4058225588087225e-06), (2, 1.6571910919083166e-06), (3, 3.6481692404777277e-06), (4, 6.901336291775806e-06), (5, 6.847578788438113e-06), (6, 1.5152076230151579e-05), (7, 2.018261147895828e-05), (8, 1.4704758541483898e-05), (9, 1.554068148834631e-05), (10, 2.1572242985712364e-05), (11, 4.4559510570252314e-05), (12, 4.3334162910468876e-05), (13, 4.847693344345316e-05), (14, 7.487507537007332e-05), (15, 9.201477951137349e-05), (16, 8.03784787422046e-05), (17, 0.00010860789188882336), (18, 7.370626553893089e-05), (19, 8.038318628678098e-05), (20, 0.00010440865298733115), (21, 0.00022618088405579329), (22, 0.00019938494369853288), (23, 0.0002742178039625287), (24, 0.0006317779771052301), (25, 0.001371100777760148), (26, 0.0008230630191974342), (27, 0.00023481006792280823)]
+
+ 88%|████████▊ | 132/150 [19:52<02:31,  8.39s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 102, Spatial MSE: [(0, 1.270054781343788e-06), (1, 1.955323023139499e-05), (2, 3.059744631173089e-05), (3, 1.1186852134414949e-05), (4, 4.250529400451342e-06), (5, 7.3188843998650555e-06), (6, 6.572576239705086e-06), (7, 1.1672257642203476e-05), (8, 9.772105840966105e-06), (9, 8.263587915280368e-06), (10, 1.6926307580433786e-05), (11, 6.34402604191564e-05), (12, 9.205167589243501e-05), (13, 0.001934072352014482), (14, 0.003230026690289378), (15, 0.0012599054025486112), (16, 0.09437120705842972), (17, 0.00042450809269212186), (18, 0.0002299011539435014), (19, 0.007692351005971432), (20, 0.0061923847533762455), (21, 0.000399829528760165), (22, 0.000515644671395421), (23, 0.0004163255507592112), (24, 0.0008103065774776042), (25, 0.001530370325781405), (26, 0.0009552970877848566), (27, 0.03470950573682785)], Temporal MSE: [(0, 1.8038452935797977e-06), (1, 1.1388775646992144e-06), (2, 1.5659032897019642e-06), (3, 2.110245986841619e-06), (4, 3.11343455905444e-06), (5, 3.5395805753069e-06), (6, 6.088091595302103e-06), (7, 6.173596375447232e-06), (8, 4.551260644802824e-06), (9, 4.996551069780253e-06), (10, 9.452675840293523e-06), (11, 0.0001134031917899847), (12, 4.487797923502512e-05), (13, 5.043261990067549e-05), (14, 6.487295468105003e-05), (15, 0.0001916912297019735), (16, 0.00016126706032082438), (17, 0.00028963733348064125), (18, 0.00014476603246293962), (19, 0.00012008095654891804), (20, 9.191019489662722e-05), (21, 0.0004881370987277478), (22, 0.0003479426959529519), (23, 0.0003995976294390857), (24, 0.000323111773468554), (25, 0.0007394755957648158), (26, 0.0001740630978019908), (27, 0.0002182325697503984)]
+
+ 89%|████████▊ | 133/150 [20:00<02:19,  8.21s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 96, Spatial MSE: [(0, 1.4548676290360163e-06), (1, 1.8190890841651708e-05), (2, 6.0221875173738226e-05), (3, 1.4420693332795054e-05), (4, 2.136158218490891e-05), (5, 4.6636905608465895e-05), (6, 0.00010090194700751454), (7, 0.00017759413458406925), (8, 0.0002362160012125969), (9, 0.00025591161102056503), (10, 0.00020736436999868602), (11, 0.00032531723263673484), (12, 0.0003175518359057605), (13, 0.0024445157032459974), (14, 0.0042156074196100235), (15, 0.0021386994048953056), (16, 0.10112062096595764), (17, 0.0009908458450809121), (18, 0.000755203131120652), (19, 0.009055395610630512), (20, 0.007657114416360855), (21, 0.0012149896938353777), (22, 0.0014266212237998843), (23, 0.0022967096883803606), (24, 0.004219838418066502), (25, 0.0118605587631464), (26, 0.017960457131266594), (27, 0.03459681570529938)], Temporal MSE: [(0, 2.3601201064593624e-06), (1, 1.4250439335228293e-06), (2, 3.225220780223026e-06), (3, 1.5649726265110075e-05), (4, 2.541557842050679e-05), (5, 2.728790968831163e-05), (6, 0.00012476209667511284), (7, 8.596399857196957e-05), (8, 5.368211350287311e-05), (9, 6.910954834893346e-05), (10, 9.435118408873677e-05), (11, 0.000211403748835437), (12, 0.00014011649182066321), (13, 0.00022336836264003068), (14, 0.00036755952169187367), (15, 0.0004353822150733322), (16, 0.0003931451647076756), (17, 0.0005087848985567689), (18, 0.0003586665552575141), (19, 0.0004380800237413496), (20, 0.0006023269379511476), (21, 0.000915715703740716), (22, 0.0008369029965251684), (23, 0.0010912265861406922), (24, 0.0020455268677324057), (25, 0.0037403893657028675), (26, 0.0023704140912741423), (27, 0.0005349454586394131)]
+
+ 89%|████████▉ | 134/150 [20:08<02:12,  8.27s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 90, Spatial MSE: [(0, 1.3643717693412327e-06), (1, 2.1849744371138513e-05), (2, 4.3864660256076604e-05), (3, 7.441409707098501e-06), (4, 6.1973751144250855e-06), (5, 1.1334073860780336e-05), (6, 2.4272003429359756e-05), (7, 3.737994120456278e-05), (8, 4.442113277036697e-05), (9, 5.3654308430850506e-05), (10, 4.970413647242822e-05), (11, 7.261933933477849e-05), (12, 4.3269905290799215e-05), (13, 0.0001368392986478284), (14, 0.0004729933280032128), (15, 0.0003091440303251147), (16, 0.006201003212481737), (17, 0.00012575770961120725), (18, 9.268887515645474e-05), (19, 0.0005535919917747378), (20, 0.0005819503567181528), (21, 0.00017292180564254522), (22, 0.0001778397854650393), (23, 0.00040910966345109046), (24, 0.0007744118920527399), (25, 0.002272337907925248), (26, 0.0040572225116193295), (27, 0.003036675974726677)], Temporal MSE: [(0, 2.293630814165226e-06), (1, 1.4524404150506598e-06), (2, 1.9736353351618163e-06), (3, 3.818290679191705e-06), (4, 4.5762267291138414e-06), (5, 3.570334001778974e-06), (6, 2.8026150175719522e-05), (7, 1.7209964426001534e-05), (8, 1.1285465916444082e-05), (9, 1.4067679330764804e-05), (10, 1.5664700185880065e-05), (11, 1.661391979723703e-05), (12, 1.376078125758795e-05), (13, 2.5706563974381424e-05), (14, 4.403825005283579e-05), (15, 5.1529190386645496e-05), (16, 4.515455293585546e-05), (17, 3.420648135943338e-05), (18, 3.506713983369991e-05), (19, 5.389707439462654e-05), (20, 8.469385647913441e-05), (21, 7.629438914591447e-05), (22, 7.810092938598245e-05), (23, 0.0001238503900822252), (24, 0.00028510449919849634), (25, 0.00048138853162527084), (26, 0.00021250118152238429), (27, 0.00010997203207807615)]
+
+ 90%|█████████ | 135/150 [20:17<02:05,  8.36s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 84, Spatial MSE: [(0, 1.3262958873383468e-06), (1, 2.1141227989573963e-05), (2, 4.456973692867905e-05), (3, 7.501417712774128e-06), (4, 5.827700533700408e-06), (5, 1.0292455044691451e-05), (6, 2.4350925741600804e-05), (7, 3.362183997523971e-05), (8, 3.9905207813717425e-05), (9, 4.9633439630270004e-05), (10, 3.8573023630306125e-05), (11, 5.993263403070159e-05), (12, 3.829902925645001e-05), (13, 5.553970913751982e-05), (14, 0.00010450639820192009), (15, 0.00010865974036278203), (16, 9.869825589703396e-05), (17, 9.283644612878561e-05), (18, 7.137624197639525e-05), (19, 9.287225111620501e-05), (20, 0.00012666235852520913), (21, 0.0001336892310064286), (22, 0.0001360801252303645), (23, 0.0003249262226745486), (24, 0.0006900145672261715), (25, 0.0019224300049245358), (26, 0.0035322110634297132), (27, 0.0029953811317682266)], Temporal MSE: [(0, 2.4235225737356814e-06), (1, 1.5416057976835873e-06), (2, 1.924877096826094e-06), (3, 3.878632924170233e-06), (4, 4.242384875396965e-06), (5, 3.396433385205455e-06), (6, 2.775703433144372e-05), (7, 1.679140950727742e-05), (8, 1.1552295291039627e-05), (9, 1.2936931852891576e-05), (10, 1.3800590750179254e-05), (11, 1.4451264178205747e-05), (12, 1.124202026403509e-05), (13, 2.1446212485898286e-05), (14, 3.572679270291701e-05), (15, 3.9650134567636997e-05), (16, 3.5123615816701204e-05), (17, 2.6577627068036236e-05), (18, 2.677036900422536e-05), (19, 4.0511844417778775e-05), (20, 5.9528203564696014e-05), (21, 5.647880971082486e-05), (22, 5.8524004998616874e-05), (23, 9.981488983612508e-05), (24, 0.0002345247776247561), (25, 0.0004027247487101704), (26, 0.00018404908769298345), (27, 0.00011586114123929292)]
+
+ 91%|█████████ | 136/150 [20:26<01:58,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 78, Spatial MSE: [(0, 1.3667041685039294e-06), (1, 2.3946859073475935e-05), (2, 5.101767601445317e-05), (3, 7.871759407862555e-06), (4, 6.662757641606731e-06), (5, 1.1247630936850328e-05), (6, 2.589554242149461e-05), (7, 4.1819112084340304e-05), (8, 4.954624455422163e-05), (9, 5.171398152015172e-05), (10, 4.6580240450566635e-05), (11, 6.300182576524094e-05), (12, 4.0267517761094496e-05), (13, 6.011114965076558e-05), (14, 0.00012950782547704875), (15, 0.00013635314826387912), (16, 0.00011400921357562765), (17, 0.000102453566796612), (18, 8.191054803319275e-05), (19, 0.00010804947669385001), (20, 0.00014864158583804965), (21, 0.00015592183626722544), (22, 0.0001624935830477625), (23, 0.00039152769022621214), (24, 0.0007752637611702085), (25, 0.0020562736317515373), (26, 0.00392081867903471), (27, 0.0030958587303757668)], Temporal MSE: [(0, 2.4756245693424717e-06), (1, 1.6035666021707584e-06), (2, 2.1200294213485904e-06), (3, 4.285242084733909e-06), (4, 4.896707650914323e-06), (5, 3.716034825629322e-06), (6, 3.4045326174236834e-05), (7, 1.917060581035912e-05), (8, 1.2911063095089048e-05), (9, 1.4448678484768607e-05), (10, 1.4016262866789475e-05), (11, 1.5829347830731422e-05), (12, 1.1550476301636081e-05), (13, 2.261285590066109e-05), (14, 4.090323636773974e-05), (15, 5.064939250587486e-05), (16, 4.019484913442284e-05), (17, 3.081139948335476e-05), (18, 3.0581119062844664e-05), (19, 4.824619827559218e-05), (20, 7.372008985839784e-05), (21, 7.448407995980233e-05), (22, 6.860036955913529e-05), (23, 0.00011492575140437111), (24, 0.0002703479549381882), (25, 0.0004369846428744495), (26, 0.0002208980731666088), (27, 0.00016223016427829862)]
+
+ 91%|█████████▏| 137/150 [20:34<01:50,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 72, Spatial MSE: [(0, 1.448347120458493e-06), (1, 2.5087887479458004e-05), (2, 5.0512069719843566e-05), (3, 8.346501999767497e-06), (4, 6.957107416383224e-06), (5, 1.2195656381663866e-05), (6, 2.8710881451843306e-05), (7, 4.1594310459913686e-05), (8, 4.958526187692769e-05), (9, 5.648381193168461e-05), (10, 4.498363705351949e-05), (11, 5.9354726545279846e-05), (12, 3.6881468986393884e-05), (13, 6.469803338404745e-05), (14, 0.00012155718286521733), (15, 0.00012276087363716215), (16, 0.00010972815653076395), (17, 0.00011889976303791627), (18, 8.453054033452645e-05), (19, 0.00011639812146313488), (20, 0.00015801672998350114), (21, 0.00015005801105871797), (22, 0.00017380609642714262), (23, 0.0004156196373514831), (24, 0.0008340973872691393), (25, 0.002308727940544486), (26, 0.004301135893911123), (27, 0.0034830777440220118)], Temporal MSE: [(0, 2.6141728994844016e-06), (1, 1.7121024029620457e-06), (2, 2.1932460185780656e-06), (3, 4.512879513640655e-06), (4, 5.162827619642485e-06), (5, 4.02607474825345e-06), (6, 4.0722021367400885e-05), (7, 1.9954017261625268e-05), (8, 1.4305662261904217e-05), (9, 1.573902045493014e-05), (10, 1.5518029613303952e-05), (11, 1.698991036391817e-05), (12, 1.2218349183967803e-05), (13, 2.4268725610454567e-05), (14, 4.172195622231811e-05), (15, 4.9678470531944185e-05), (16, 4.319518120610155e-05), (17, 3.280495366198011e-05), (18, 3.1330746423918754e-05), (19, 5.254656571196392e-05), (20, 8.08041004347615e-05), (21, 7.1148962888401e-05), (22, 7.566736894659698e-05), (23, 0.0001254469098057598), (24, 0.00028940499760210514), (25, 0.00047933339374139905), (26, 0.0002396627824055031), (27, 0.00021535035921260715)]
+
+ 92%|█████████▏| 138/150 [20:43<01:42,  8.57s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 66, Spatial MSE: [(0, 1.69633142377279e-06), (1, 2.864732414309401e-05), (2, 5.8084526244783774e-05), (3, 9.416900866199285e-06), (4, 8.064945177466143e-06), (5, 1.4199385077517945e-05), (6, 3.467454735073261e-05), (7, 4.881665881839581e-05), (8, 6.121543265180662e-05), (9, 6.979295721976086e-05), (10, 5.848624277859926e-05), (11, 7.889133848948404e-05), (12, 5.657266592606902e-05), (13, 0.0004219831607770175), (14, 0.0015795585932210088), (15, 0.001173919066786766), (16, 0.030174031853675842), (17, 0.00018216499302070588), (18, 0.00016461373888887465), (19, 0.0027579443994909525), (20, 0.0025925026275217533), (21, 0.00027680673520080745), (22, 0.0002655072312336415), (23, 0.0005404642433859408), (24, 0.0010956642217934132), (25, 0.0028596825432032347), (26, 0.005105074029415846), (27, 0.003777055535465479)], Temporal MSE: [(0, 3.2706946058169706e-06), (1, 2.114886001436389e-06), (2, 2.7918638352275593e-06), (3, 5.506356501427945e-06), (4, 6.0218862927285954e-06), (5, 4.905067726213019e-06), (6, 4.615834404830821e-05), (7, 2.288661380589474e-05), (8, 1.8514505427447148e-05), (9, 1.883737786556594e-05), (10, 1.8888807971961796e-05), (11, 2.3142982172430493e-05), (12, 2.3879156287875958e-05), (13, 4.0831549995345995e-05), (14, 6.521911564050242e-05), (15, 7.965994882397354e-05), (16, 6.106326327426359e-05), (17, 5.182974928175099e-05), (18, 6.118606688687578e-05), (19, 8.513392822351307e-05), (20, 0.00012869038619101048), (21, 0.00013859389582648873), (22, 0.00011268487287452444), (23, 0.0001684694434516132), (24, 0.00036258093314245343), (25, 0.0006340381223708391), (26, 0.0003247424028813839), (27, 0.0003282176039647311)]
+
+ 93%|█████████▎| 139/150 [20:51<01:33,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 60, Spatial MSE: [(0, 1.833745272961096e-06), (1, 3.106099393335171e-05), (2, 6.335341458907351e-05), (3, 1.0490869499335531e-05), (4, 8.69550058268942e-06), (5, 1.5172051462286618e-05), (6, 3.8305792259052396e-05), (7, 5.6952798331622034e-05), (8, 6.720849341945723e-05), (9, 7.327438652282581e-05), (10, 6.58260250929743e-05), (11, 8.530830382369459e-05), (12, 6.516936264233664e-05), (13, 0.001142154331319034), (14, 0.004426267929375172), (15, 0.0029962831176817417), (16, 0.08020159602165222), (17, 0.0002585150650702417), (18, 0.0002854425983969122), (19, 0.0069669876247644424), (20, 0.006408313289284706), (21, 0.0003997173625975847), (22, 0.0003656295593827963), (23, 0.0006449546199291945), (24, 0.001285966020077467), (25, 0.0031805660109966993), (26, 0.0057646119967103004), (27, 0.004041876178234816)], Temporal MSE: [(0, 3.443059995333897e-06), (1, 2.2912813619768713e-06), (2, 3.1319175377575448e-06), (3, 5.464564310386777e-06), (4, 6.303866939560976e-06), (5, 5.400223926699255e-06), (6, 5.538331970456056e-05), (7, 2.5950774215743877e-05), (8, 2.1618703613057733e-05), (9, 2.1111942260176875e-05), (10, 2.1952153474558145e-05), (11, 2.912923264375422e-05), (12, 3.8636793760815635e-05), (13, 6.653657328570262e-05), (14, 9.592639253241941e-05), (15, 0.00012917682761326432), (16, 8.46125913085416e-05), (17, 7.541367813246325e-05), (18, 9.8523756605573e-05), (19, 0.00013580135419033468), (20, 0.0001718476414680481), (21, 0.00020375062013044953), (22, 0.00016562765813432634), (23, 0.00022300668933894485), (24, 0.0004276656254660338), (25, 0.0007678073598071933), (26, 0.0003775622753892094), (27, 0.0004730535438284278)]
+
+ 93%|█████████▎| 140/150 [20:59<01:24,  8.46s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 54, Spatial MSE: [(0, 1.962023588930606e-06), (1, 3.413521699258126e-05), (2, 7.076590554788709e-05), (3, 1.1016783901141025e-05), (4, 9.648131708672736e-06), (5, 1.6855885405675508e-05), (6, 4.364674896351062e-05), (7, 6.43864696030505e-05), (8, 7.5280993769411e-05), (9, 8.45157919684425e-05), (10, 7.085916149662808e-05), (11, 9.576274896971881e-05), (12, 6.677364581264555e-05), (13, 0.000964012579061091), (14, 0.0035596198868006468), (15, 0.002320500323548913), (16, 0.06363610923290253), (17, 0.00025201233802363276), (18, 0.00025549044949002564), (19, 0.005342873279005289), (20, 0.005012536887079477), (21, 0.0003758905513677746), (22, 0.0003646450932137668), (23, 0.0006776496302336454), (24, 0.0013578134821727872), (25, 0.003275367198511958), (26, 0.0056834109127521515), (27, 0.004339255392551422)], Temporal MSE: [(0, 3.679256678879028e-06), (1, 2.4888900043151807e-06), (2, 3.4475217489671195e-06), (3, 5.991564648866188e-06), (4, 6.988110271777259e-06), (5, 5.973311544948956e-06), (6, 7.021144847385585e-05), (7, 2.9682099921046756e-05), (8, 2.4639635739731602e-05), (9, 2.432766996207647e-05), (10, 2.362604755035136e-05), (11, 3.130235563730821e-05), (12, 3.901276795659214e-05), (13, 6.152385321911424e-05), (14, 8.421524398727342e-05), (15, 0.00012191841233288869), (16, 8.653556869830936e-05), (17, 7.430984987877309e-05), (18, 9.05673805391416e-05), (19, 0.00012245523976162076), (20, 0.00017114347429014742), (21, 0.00019641686230897903), (22, 0.00016831625544000417), (23, 0.00024437016691081226), (24, 0.000457577349152416), (25, 0.0007477648323401809), (26, 0.00042787202983163297), (27, 0.0007542740786448121)]
+
+ 94%|█████████▍| 141/150 [21:08<01:16,  8.47s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 48, Spatial MSE: [(0, 2.201662255174597e-06), (1, 3.752338670892641e-05), (2, 7.792237011017278e-05), (3, 1.2087846698705107e-05), (4, 1.065860669768881e-05), (5, 1.803552186174784e-05), (6, 4.5604563638335094e-05), (7, 6.674988253507763e-05), (8, 8.545772288925946e-05), (9, 9.057827992364764e-05), (10, 8.390886068809777e-05), (11, 9.261012746719643e-05), (12, 6.100005339249037e-05), (13, 0.0007216596277430654), (14, 0.0029237843118608), (15, 0.0019417841685935855), (16, 0.05052580684423447), (17, 0.000254750601015985), (18, 0.00023870360746514052), (19, 0.004278681706637144), (20, 0.0040479605086147785), (21, 0.0003795528900809586), (22, 0.0003439654246903956), (23, 0.0007156392675824463), (24, 0.0014194888062775135), (25, 0.0031229997985064983), (26, 0.005677204579114914), (27, 0.004440994933247566)], Temporal MSE: [(0, 4.052930307807401e-06), (1, 2.740091076702811e-06), (2, 3.885501882905373e-06), (3, 6.213075266714441e-06), (4, 7.564646239188733e-06), (5, 6.702996415697271e-06), (6, 8.037710358621553e-05), (7, 3.094709609285928e-05), (8, 2.8061747798346914e-05), (9, 2.4807999579934403e-05), (10, 2.4877657779143192e-05), (11, 2.8371394364512525e-05), (12, 3.0432538551394828e-05), (13, 5.785523899248801e-05), (14, 8.49551142891869e-05), (15, 0.00011663259647320956), (16, 8.222632459364831e-05), (17, 6.845004099886864e-05), (18, 8.235086716013029e-05), (19, 0.0001239521661773324), (20, 0.00016658368986099958), (21, 0.00018371138139627874), (22, 0.0001655990636209026), (23, 0.00024175648286473006), (24, 0.00048246406367979944), (25, 0.0007764873444102705), (26, 0.000516155909281224), (27, 0.001063246396370232)]
+
+ 95%|█████████▍| 142/150 [21:17<01:08,  8.54s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 42, Spatial MSE: [(0, 2.612367325127707e-06), (1, 4.520605580182746e-05), (2, 9.334936476079747e-05), (3, 1.4442044630413875e-05), (4, 1.2675383004534524e-05), (5, 2.2066920791985467e-05), (6, 6.178904004627839e-05), (7, 8.63665045471862e-05), (8, 0.00010723788727773353), (9, 0.0001136535283876583), (10, 0.0001067088232957758), (11, 0.00012514741683844477), (12, 7.122518582036719e-05), (13, 0.00020393237355165184), (14, 0.0004918437916785479), (15, 0.0003377636894583702), (16, 0.0036876671947538853), (17, 0.00024540614685975015), (18, 0.0001859482581494376), (19, 0.00040973129216581583), (20, 0.0005821527447551489), (21, 0.0005519536207430065), (22, 0.00035751648829318583), (23, 0.000857493665535003), (24, 0.0018356333021074533), (25, 0.003906893543899059), (26, 0.007481805514544249), (27, 0.0055333650670945644)], Temporal MSE: [(0, 4.9211844270757865e-06), (1, 3.3466278637206415e-06), (2, 4.859936325374292e-06), (3, 7.4779436545213684e-06), (4, 9.096586836676579e-06), (5, 8.270354555861559e-06), (6, 9.842571307672188e-05), (7, 4.115020055905916e-05), (8, 3.863393430947326e-05), (9, 3.2609004847472534e-05), (10, 3.324188946862705e-05), (11, 3.549003668013029e-05), (12, 2.778769521682989e-05), (13, 5.0545921112643555e-05), (14, 8.920133404899389e-05), (15, 0.00011393247405067086), (16, 8.775902824709192e-05), (17, 6.556037260452285e-05), (18, 6.44755782559514e-05), (19, 0.00010491162538528442), (20, 0.00016723353473935276), (21, 0.00016414534184150398), (22, 0.00014781922800466418), (23, 0.0002528832119423896), (24, 0.0005718131433241069), (25, 0.0010032722493633628), (26, 0.0007128503639250994), (27, 0.002124876482412219)]
+
+ 95%|█████████▌| 143/150 [21:25<00:59,  8.45s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 36, Spatial MSE: [(0, 3.1067531836015405e-06), (1, 5.232413604971953e-05), (2, 0.00010696550452848896), (3, 1.6555317415622994e-05), (4, 1.5382211131509393e-05), (5, 2.621318526507821e-05), (6, 7.201834523584694e-05), (7, 0.0001010457708616741), (8, 0.00013456585293170065), (9, 0.0001391414989484474), (10, 0.00013433551066555083), (11, 0.00015468269702978432), (12, 9.172279533231631e-05), (13, 0.0007562771788798273), (14, 0.002771517727524042), (15, 0.0017808577977120876), (16, 0.04398013651371002), (17, 0.0003750529431272298), (18, 0.0002858502557501197), (19, 0.0038282559253275394), (20, 0.0034131582360714674), (21, 0.0007051078719086945), (22, 0.0005126784672029316), (23, 0.0011007143184542656), (24, 0.0023016887716948986), (25, 0.004935369826853275), (26, 0.008946267887949944), (27, 0.007213226519525051)], Temporal MSE: [(0, 5.663940100930631e-06), (1, 4.014939804619644e-06), (2, 5.951188995823031e-06), (3, 8.710809197509661e-06), (4, 1.0594978448352776e-05), (5, 1.0231903615931515e-05), (6, 0.00013594065967481583), (7, 4.6501183533109725e-05), (8, 4.736898699775338e-05), (9, 3.953778286813758e-05), (10, 4.039364284835756e-05), (11, 4.39395043940749e-05), (12, 4.1490337025607005e-05), (13, 7.741538865957409e-05), (14, 0.00012405282177496701), (15, 0.0001537289354018867), (16, 0.00011500276741571724), (17, 8.954184886533767e-05), (18, 0.0001036685353028588), (19, 0.00015537708532065153), (20, 0.00022358197020366788), (21, 0.00024238656624220312), (22, 0.00022492882271762937), (23, 0.00034976645838469267), (24, 0.0007327145431190729), (25, 0.0012948851799592376), (26, 0.0009764957940205932), (27, 0.0038924901746213436)]
+
+ 96%|█████████▌| 144/150 [21:33<00:50,  8.44s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 30, Spatial MSE: [(0, 3.622014673965168e-06), (1, 5.994475577608682e-05), (2, 0.00011741416528820992), (3, 1.936017542902846e-05), (4, 1.7787298929761164e-05), (5, 2.9318071028683335e-05), (6, 8.427628927165642e-05), (7, 0.00011930352047784254), (8, 0.00015381128469016403), (9, 0.00015544731286354363), (10, 0.00015711785817984492), (11, 0.00016887177480384707), (12, 0.00012037836131639779), (13, 0.0022077136673033237), (14, 0.007855406031012535), (15, 0.004874375183135271), (16, 0.1355351209640503), (17, 0.0004874996666330844), (18, 0.0004953722818754613), (19, 0.011131027713418007), (20, 0.010250045917928219), (21, 0.0008195164264179766), (22, 0.000762684503570199), (23, 0.0013763719471171498), (24, 0.002897800412029028), (25, 0.005647016689181328), (26, 0.010496537201106548), (27, 0.009820652194321156)], Temporal MSE: [(0, 6.859066161268856e-06), (1, 4.795494078280171e-06), (2, 7.16996555638616e-06), (3, 9.631802640797105e-06), (4, 1.2561468793137465e-05), (5, 1.2277053428988438e-05), (6, 0.00015765268472023308), (7, 5.554125891649164e-05), (8, 6.0408310673665255e-05), (9, 4.637604797608219e-05), (10, 4.731002263724804e-05), (11, 6.073436816222966e-05), (12, 7.775676203891635e-05), (13, 0.0001223730214405805), (14, 0.000167184611200355), (15, 0.00024055616813711822), (16, 0.00015296708443202078), (17, 0.00012815717491321266), (18, 0.00016809254884719849), (19, 0.0002429245796520263), (20, 0.0003364460717421025), (21, 0.0004221878189127892), (22, 0.0003193357842974365), (23, 0.00047250837087631226), (24, 0.0009240230428986251), (25, 0.0015925082843750715), (26, 0.0013966213446110487), (27, 0.007562102749943733)]
+
+ 97%|█████████▋| 145/150 [21:42<00:42,  8.49s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 24, Spatial MSE: [(0, 4.555148734652903e-06), (1, 7.115835614968091e-05), (2, 0.00013784156180918217), (3, 2.3655085897189565e-05), (4, 2.2713758880854584e-05), (5, 3.5379096516408026e-05), (6, 0.00011003266990883276), (7, 0.00012617681932169944), (8, 0.0001725539768813178), (9, 0.0002043685526587069), (10, 0.00019546296971384436), (11, 0.00022039742907509208), (12, 0.00012620724737644196), (13, 0.0015390586340799928), (14, 0.005770798772573471), (15, 0.0036283174995332956), (16, 0.09387069940567017), (17, 0.0005412481259554625), (18, 0.00047291466034948826), (19, 0.007920730859041214), (20, 0.00739244231954217), (21, 0.0008116321987472475), (22, 0.0007655310910195112), (23, 0.001608168357051909), (24, 0.0036128105130046606), (25, 0.006654410623013973), (26, 0.012571962550282478), (27, 0.015815965831279755)], Temporal MSE: [(0, 8.264468306151684e-06), (1, 5.994319053570507e-06), (2, 9.133387720794417e-06), (3, 1.077427532436559e-05), (4, 1.5991889085853472e-05), (5, 1.541298297524918e-05), (6, 0.00019957020413130522), (7, 6.812214269302785e-05), (8, 8.144172898028046e-05), (9, 5.892687477171421e-05), (10, 6.23982195975259e-05), (11, 6.965760985622182e-05), (12, 7.323322643060237e-05), (13, 0.0001251088106073439), (14, 0.00018451068899594247), (15, 0.00024331716122105718), (16, 0.00017051748000085354), (17, 0.00013587597641162574), (18, 0.000163805962074548), (19, 0.00025454023852944374), (20, 0.00035380563349463046), (21, 0.0003818330878857523), (22, 0.00032857866608537734), (23, 0.000531925237737596), (24, 0.0011392813175916672), (25, 0.001976356841623783), (26, 0.002151647349819541), (27, 0.016429485753178596)]
+
+ 97%|█████████▋| 146/150 [21:51<00:34,  8.61s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 18, Spatial MSE: [(0, 5.778076683782274e-06), (1, 9.427884651813656e-05), (2, 0.00017397590272594243), (3, 2.974965718749445e-05), (4, 2.981322722916957e-05), (5, 4.458214243641123e-05), (6, 0.0001397212326992303), (7, 0.0001857196184573695), (8, 0.0002584235917311162), (9, 0.00026048594736494124), (10, 0.00027407280867919326), (11, 0.00026259280275553465), (12, 0.00015353817434515804), (13, 0.0015179745387285948), (14, 0.006213858723640442), (15, 0.0043878499418497086), (16, 0.10778936743736267), (17, 0.0007775103440508246), (18, 0.0006271795136854053), (19, 0.009006830863654613), (20, 0.008359953761100769), (21, 0.0010733900126069784), (22, 0.0009877750417217612), (23, 0.0020989838521927595), (24, 0.004829396493732929), (25, 0.008552142418920994), (26, 0.016905738040804863), (27, 0.03308499976992607)], Temporal MSE: [(0, 1.0347620445827488e-05), (1, 7.5315783760743216e-06), (2, 1.2226797480252571e-05), (3, 1.2974903256690595e-05), (4, 2.0153554942226037e-05), (5, 1.9984991013188846e-05), (6, 0.00027383933775126934), (7, 9.270196460420266e-05), (8, 0.00011186845949850976), (9, 7.75482039898634e-05), (10, 7.934360473882407e-05), (11, 8.549125777790323e-05), (12, 8.379184873774648e-05), (13, 0.0001470074785174802), (14, 0.0002225808857474476), (15, 0.00030066072940826416), (16, 0.00021618467872031033), (17, 0.00016928103286772966), (18, 0.0001949574943864718), (19, 0.0003099188907071948), (20, 0.0004258511180523783), (21, 0.0004526202392298728), (22, 0.00041940936353057623), (23, 0.0006936651770956814), (24, 0.0015430101193487644), (25, 0.0025988442357629538), (26, 0.003572796005755663), (27, 0.03951173648238182)]
+
+ 98%|█████████▊| 147/150 [21:59<00:25,  8.58s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 12, Spatial MSE: [(0, 7.885626473580487e-06), (1, 0.0001242631406057626), (2, 0.0002426091959932819), (3, 4.0908442315412685e-05), (4, 4.21556833316572e-05), (5, 6.012438825564459e-05), (6, 0.00020003376994282007), (7, 0.0002291886485181749), (8, 0.00032530553289689124), (9, 0.0003469923394732177), (10, 0.00039175053825601935), (11, 0.00035189540358260274), (12, 0.00021153788839001209), (13, 0.0023570440243929625), (14, 0.00982877891510725), (15, 0.006971868686378002), (16, 0.1766744703054428), (17, 0.0009417335386388004), (18, 0.0008142011938616633), (19, 0.014189026318490505), (20, 0.013209880329668522), (21, 0.0014512070920318365), (22, 0.0014028361765667796), (23, 0.0029002660885453224), (24, 0.007063231430947781), (25, 0.012451385147869587), (26, 0.024801725521683693), (27, 0.09670707583427429)], Temporal MSE: [(0, 1.3820490494254045e-05), (1, 1.0309869139746297e-05), (2, 1.7179821952595375e-05), (3, 1.61766311066458e-05), (4, 2.7405061700847e-05), (5, 2.7904938178835437e-05), (6, 0.0003577367460820824), (7, 0.00012881774455308914), (8, 0.00016472797142341733), (9, 0.00010178895900025964), (10, 0.00011446370626799762), (11, 0.00012363799032755196), (12, 0.00012096775026293471), (13, 0.0002053706703009084), (14, 0.0002997653791680932), (15, 0.0004280705179553479), (16, 0.00028484902577474713), (17, 0.00022935462766326964), (18, 0.00027114013209939003), (19, 0.0004444641526788473), (20, 0.0005906810401938856), (21, 0.0006646868423558772), (22, 0.0005932754720561206), (23, 0.0009906226769089699), (24, 0.0022603110410273075), (25, 0.003678377252072096), (26, 0.007141702342778444), (27, 0.11025430262088776)]
+
+ 99%|█████████▊| 148/150 [22:08<00:16,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 6, Spatial MSE: [(0, 1.3798394320474472e-05), (1, 0.00021041596482973546), (2, 0.0003656734770629555), (3, 6.974415009608492e-05), (4, 7.75603621150367e-05), (5, 0.00010023439244832844), (6, 0.00034176401095464826), (7, 0.0003748952003661543), (8, 0.0005631754756905138), (9, 0.0006008424679748714), (10, 0.0006917820428498089), (11, 0.0006185898091644049), (12, 0.0003757528029382229), (13, 0.0027680210769176483), (14, 0.010767525993287563), (15, 0.007862051948904991), (16, 0.18880034983158112), (17, 0.0014219781151041389), (18, 0.0012177039170637727), (19, 0.015612469986081123), (20, 0.014588971622288227), (21, 0.002373338444158435), (22, 0.0024072665255516768), (23, 0.004931539762765169), (24, 0.01399307046085596), (25, 0.02674505114555359), (26, 0.05574282631278038), (27, 0.5315816402435303)], Temporal MSE: [(0, 2.3225697077577934e-05), (1, 1.7865151676232927e-05), (2, 3.130084587610327e-05), (3, 2.6303170670871623e-05), (4, 4.753820030600764e-05), (5, 4.9374939408153296e-05), (6, 0.0006029396317899227), (7, 0.00022963766241446137), (8, 0.00030741197406314313), (9, 0.0001793957344489172), (10, 0.00020931509789079428), (11, 0.0002210672973887995), (12, 0.0002004744455916807), (13, 0.00030215492006391287), (14, 0.0004685312742367387), (15, 0.0006638921913690865), (16, 0.0004261823196429759), (17, 0.0003406500618439168), (18, 0.0003830741625279188), (19, 0.0006313379271887243), (20, 0.0008539421833120286), (21, 0.0009321801480837166), (22, 0.0009436476975679398), (23, 0.0016621197573840618), (24, 0.004042302258312702), (25, 0.006876618135720491), (26, 0.020303059369325638), (27, 0.4503142833709717)]
+
+ 99%|█████████▉| 149/150 [22:16<00:08,  8.48s/it]Time step tensor([894, 891, 891, 888, 888, 885, 885, 882, 882, 879, 879, 876, 876, 870,
+        864, 858, 852, 846, 840, 834, 828, 822, 816, 810, 804, 798, 792, 786,
+        780, 774, 768, 762, 756, 750, 744, 738, 732, 726, 720, 714, 708, 702,
+        696, 690, 684, 678, 672, 666, 660, 654, 648, 642, 636, 630, 624, 618,
+        612, 606, 600, 594, 588, 582, 576, 570, 564, 558, 552, 546, 540, 534,
+        528, 522, 516, 510, 504, 498, 492, 486, 480, 474, 468, 462, 456, 450,
+        444, 438, 432, 426, 420, 414, 408, 402, 396, 390, 384, 378, 372, 366,
+        360, 354, 348, 342, 336, 330, 324, 318, 312, 306, 300, 294, 288, 282,
+        276, 270, 264, 258, 252, 246, 240, 234, 228, 222, 216, 210, 204, 198,
+        192, 186, 180, 174, 168, 162, 156, 150, 144, 138, 132, 126, 120, 114,
+        108, 102,  96,  90,  84,  78,  72,  66,  60,  54,  48,  42,  36,  30,
+         24,  18,  12,   6,   0], device='cuda:0')
+Time step 0, Spatial MSE: [(0, 8.682196494191885e-05), (1, 0.0007777783321216702), (2, 0.002649035770446062), (3, 0.0006305871065706015), (4, 0.00045698584290221334), (5, 0.0006599664920940995), (6, 0.001274924143217504), (7, 0.001023141318000853), (8, 0.0017148785991594195), (9, 0.001984645379707217), (10, 0.001975002232939005), (11, 0.002021169988438487), (12, 0.0018089690711349249), (13, 0.0067273955792188644), (14, 0.022210383787751198), (15, 0.017595674842596054), (16, 0.39976388216018677), (17, 0.004591927397996187), (18, 0.00523647153750062), (19, 0.036998916417360306), (20, 0.035532236099243164), (21, 0.017644263803958893), (22, 0.0220019593834877), (23, 0.043907906860113144), (24, 0.1113404929637909), (25, 0.6924773454666138), (26, 0.9954642653465271), (27, 4.255101203918457)], Temporal MSE: [(0, 0.00014502315025310963), (1, 8.854523184709251e-05), (2, 0.00013451733684632927), (3, 0.00011843206448247656), (4, 0.00031305287848226726), (5, 0.00023918446095194668), (6, 0.0014470111345872283), (7, 0.0008959615370258689), (8, 0.0011110380291938782), (9, 0.0005824737017974257), (10, 0.0009813603246584535), (11, 0.0010649949545040727), (12, 0.0008089951588772237), (13, 0.0013155165361240506), (14, 0.0017760078189894557), (15, 0.0019465675577521324), (16, 0.001509269350208342), (17, 0.0012741825776174664), (18, 0.001559382420964539), (19, 0.0026681094896048307), (20, 0.003321603173390031), (21, 0.004365516826510429), (22, 0.006303249858319759), (23, 0.011030970141291618), (24, 0.030485467985272408), (25, 0.06276512145996094), (26, 0.13214068114757538), (27, 1.68144953250885)]
+
+100%|██████████| 150/150 [22:25<00:00,  8.50s/it]
+100%|██████████| 150/150 [22:25<00:00,  8.97s/it]
+Saved spatial MLP MSE outputs to ./samples/opensora_plan/skip_mse/4_spatial_mlp_mse.pt
+Saved temporal MLP MSE outputs to ./samples/opensora_plan/skip_mse/4_temporal_mlp_mse.pt
+IMAGEIO FFMPEG_WRITER WARNING: input image is not divisible by macro_block_size=16, resizing from (1027, 1540) to (1040, 1552) to ensure video compatibility with most codecs and players. To prevent resizing, make your input image divisible by the macro_block_size or set the macro_block_size to 1 (risking incompatibility).
+[swscaler @ 0x5f293c0] Warning: data is not aligned! This can lead to a speed loss
+save path ./samples/opensora_plan/skip_mse
diff --git a/setup.py b/setup.py
deleted file mode 100644
index d3b2086b..00000000
--- a/setup.py
+++ /dev/null
@@ -1,87 +0,0 @@
-from typing import List
-
-from setuptools import find_packages, setup
-from setuptools.command.develop import develop
-from setuptools.command.egg_info import egg_info
-from setuptools.command.install import install
-
-
-def fetch_requirements(path) -> List[str]:
-    """
-    This function reads the requirements file.
-
-    Args:
-        path (str): the path to the requirements file.
-
-    Returns:
-        The lines in the requirements file.
-    """
-    with open(path, "r") as fd:
-        requirements = [r.strip() for r in fd.readlines()]
-        requirements.remove("colossalai")
-        return requirements
-
-
-def fetch_readme() -> str:
-    """
-    This function reads the README.md file in the current directory.
-
-    Returns:
-        The lines in the README file.
-    """
-    with open("README.md", encoding="utf-8") as f:
-        return f.read()
-
-
-def custom_install():
-    return ["pip", "install", "colossalai", "--no-deps"]
-
-
-class CustomInstallCommand(install):
-    def run(self):
-        install.run(self)
-        self.spawn(custom_install())
-
-
-class CustomDevelopCommand(develop):
-    def run(self):
-        develop.run(self)
-        self.spawn(custom_install())
-
-
-class CustomEggInfoCommand(egg_info):
-    def run(self):
-        egg_info.run(self)
-        self.spawn(custom_install())
-
-
-setup(
-    name="videosys",
-    version="2.0.0",
-    packages=find_packages(
-        exclude=(
-            "videos",
-            "tests",
-            "figure",
-            "*.egg-info",
-        )
-    ),
-    description="VideoSys",
-    long_description=fetch_readme(),
-    long_description_content_type="text/markdown",
-    license="Apache Software License 2.0",
-    install_requires=fetch_requirements("requirements.txt"),
-    python_requires=">=3.7",
-    cmdclass={
-        "install": CustomInstallCommand,
-        "develop": CustomDevelopCommand,
-        "egg_info": CustomEggInfoCommand,
-    },
-    classifiers=[
-        "Programming Language :: Python :: 3",
-        "License :: OSI Approved :: Apache Software License",
-        "Environment :: GPU :: NVIDIA CUDA",
-        "Topic :: Scientific/Engineering :: Artificial Intelligence",
-        "Topic :: System :: Distributed Computing",
-    ],
-)
diff --git a/taming/modules/autoencoder/lpips/vgg.pth b/taming/modules/autoencoder/lpips/vgg.pth
new file mode 100644
index 00000000..47e943cf
Binary files /dev/null and b/taming/modules/autoencoder/lpips/vgg.pth differ
diff --git a/videosys/models/transformers/flux_transformer.py b/videosys/models/transformers/flux_transformer.py
new file mode 100644
index 00000000..d7c6e91e
--- /dev/null
+++ b/videosys/models/transformers/flux_transformer.py
@@ -0,0 +1,645 @@
+# Copyright 2024 Black Forest Labs, The HuggingFace Team and The InstantX Team. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+
+from typing import Any, Dict, Optional, Tuple, Union
+
+import numpy as np
+import torch
+import torch.nn as nn
+import torch.nn.functional as F
+from diffusers.configuration_utils import ConfigMixin, register_to_config
+from diffusers.loaders import FromOriginalModelMixin, PeftAdapterMixin
+from diffusers.models.attention import FeedForward
+from diffusers.models.attention_processor import (
+    Attention,
+    AttentionProcessor,
+    FluxAttnProcessor2_0,
+    FusedFluxAttnProcessor2_0,
+)
+from diffusers.models.embeddings import (
+    CombinedTimestepGuidanceTextProjEmbeddings,
+    CombinedTimestepTextProjEmbeddings,
+    FluxPosEmbed,
+)
+from diffusers.models.modeling_outputs import Transformer2DModelOutput
+from diffusers.models.modeling_utils import ModelMixin
+from diffusers.models.normalization import AdaLayerNormContinuous, AdaLayerNormZero, AdaLayerNormZeroSingle
+from diffusers.utils import USE_PEFT_BACKEND, is_torch_version, logging, scale_lora_layers, unscale_lora_layers
+from diffusers.utils.torch_utils import maybe_allow_in_graph
+from videosys.core.parallel_mgr import ParallelManager
+from videosys.core.pab_mgr import (
+    enable_pab,
+    get_mlp_output,
+    if_broadcast_cross,
+    if_broadcast_mlp,
+    if_broadcast_spatial,
+    save_mlp_output,
+)
+
+logger = logging.get_logger(__name__)  # pylint: disable=invalid-name
+
+
+@maybe_allow_in_graph
+class FluxSingleTransformerBlock(nn.Module):
+    r"""
+    A Transformer block following the MMDiT architecture, introduced in Stable Diffusion 3.
+
+    Reference: https://arxiv.org/abs/2403.03206
+
+    Parameters:
+        dim (`int`): The number of channels in the input and output.
+        num_attention_heads (`int`): The number of heads to use for multi-head attention.
+        attention_head_dim (`int`): The number of channels in each head.
+        context_pre_only (`bool`): Boolean to determine if we should add some blocks associated with the
+            processing of `context` conditions.
+    """
+
+    def __init__(self, dim, num_attention_heads, attention_head_dim, mlp_ratio=4.0):
+        super().__init__()
+        self.mlp_hidden_dim = int(dim * mlp_ratio)
+
+        self.norm = AdaLayerNormZeroSingle(dim)
+        self.proj_mlp = nn.Linear(dim, self.mlp_hidden_dim)
+        self.act_mlp = nn.GELU(approximate="tanh")
+        self.proj_out = nn.Linear(dim + self.mlp_hidden_dim, dim)
+
+        processor = FluxAttnProcessor2_0()
+        self.attn = Attention(  # TODO: Add support for `cross_attention_dim` in the future.
+            query_dim=dim,
+            cross_attention_dim=None,
+            dim_head=attention_head_dim,
+            heads=num_attention_heads,
+            out_dim=dim,
+            bias=True,
+            processor=processor,
+            qk_norm="rms_norm",
+            eps=1e-6,
+            pre_only=True,
+        )
+
+        # pab
+        self.attn_count = 0
+        self.last_attn = None
+
+        # parallel
+        self.parallel_manager: ParallelManager = None
+
+    def forward(
+        self,
+        hidden_states: torch.FloatTensor,
+        temb: torch.FloatTensor,
+        image_rotary_emb=None,
+        joint_attention_kwargs=None,
+        timestep = None
+    ):
+        residual = hidden_states
+        norm_hidden_states, gate = self.norm(hidden_states, emb=temb)
+        mlp_hidden_states = self.act_mlp(self.proj_mlp(norm_hidden_states))
+        joint_attention_kwargs = joint_attention_kwargs or {}
+        
+        if enable_pab():
+            broadcast_attn, self.attn_count = if_broadcast_spatial(int(timestep[0]), self.attn_count)
+        
+        if enable_pab() and broadcast_attn:
+            attn_output = self.last_attn
+            # print(f'SingleBlock | skip t={int(timestep[0])}')
+        else:
+            attn_output = self.attn(
+                hidden_states=norm_hidden_states,
+                image_rotary_emb=image_rotary_emb,
+                **joint_attention_kwargs,
+            )
+            if enable_pab():
+                self.last_attn = attn_output
+
+        hidden_states = torch.cat([attn_output, mlp_hidden_states], dim=2)
+        gate = gate.unsqueeze(1)
+        hidden_states = gate * self.proj_out(hidden_states)
+        hidden_states = residual + hidden_states
+        if hidden_states.dtype == torch.float16:
+            hidden_states = hidden_states.clip(-65504, 65504)
+
+        return hidden_states
+
+
+@maybe_allow_in_graph
+class FluxTransformerBlock(nn.Module):
+    r"""
+    A Transformer block following the MMDiT architecture, introduced in Stable Diffusion 3.
+
+    Reference: https://arxiv.org/abs/2403.03206
+
+    Parameters:
+        dim (`int`): The number of channels in the input and output.
+        num_attention_heads (`int`): The number of heads to use for multi-head attention.
+        attention_head_dim (`int`): The number of channels in each head.
+        context_pre_only (`bool`): Boolean to determine if we should add some blocks associated with the
+            processing of `context` conditions.
+    """
+
+    def __init__(self, dim, num_attention_heads, attention_head_dim, qk_norm="rms_norm", eps=1e-6):
+        super().__init__()
+
+        self.norm1 = AdaLayerNormZero(dim)
+
+        self.norm1_context = AdaLayerNormZero(dim)
+
+        if hasattr(F, "scaled_dot_product_attention"):
+            processor = FluxAttnProcessor2_0()
+        else:
+            raise ValueError(
+                "The current PyTorch version does not support the `scaled_dot_product_attention` function."
+            )
+        self.attn = Attention(
+            query_dim=dim,
+            cross_attention_dim=None,
+            added_kv_proj_dim=dim,
+            dim_head=attention_head_dim,
+            heads=num_attention_heads,
+            out_dim=dim,
+            context_pre_only=False,
+            bias=True,
+            processor=processor,
+            qk_norm=qk_norm,
+            eps=eps,
+        )
+
+        self.norm2 = nn.LayerNorm(dim, elementwise_affine=False, eps=1e-6)
+        self.ff = FeedForward(dim=dim, dim_out=dim, activation_fn="gelu-approximate")
+
+        self.norm2_context = nn.LayerNorm(dim, elementwise_affine=False, eps=1e-6)
+        self.ff_context = FeedForward(dim=dim, dim_out=dim, activation_fn="gelu-approximate")
+
+        # let chunk size default to None
+        self._chunk_size = None
+        self._chunk_dim = 0
+
+        # pab
+        self.cross_count = 0
+        self.last_cross = None
+        self.context_attn_output = None
+        
+        # parallel
+        self.parallel_manager: ParallelManager = None
+
+    def forward(
+        self,
+        hidden_states: torch.FloatTensor,
+        encoder_hidden_states: torch.FloatTensor,
+        temb: torch.FloatTensor,
+        image_rotary_emb=None,
+        joint_attention_kwargs=None,
+        timestep = None
+    ):
+        norm_hidden_states, gate_msa, shift_mlp, scale_mlp, gate_mlp = self.norm1(hidden_states, emb=temb)
+
+        norm_encoder_hidden_states, c_gate_msa, c_shift_mlp, c_scale_mlp, c_gate_mlp = self.norm1_context(
+            encoder_hidden_states, emb=temb
+        )
+        joint_attention_kwargs = joint_attention_kwargs or {}
+        # Attention. # TODO: Add support for `cross_attention_dim` in the future.
+        
+        # cross attention
+        if enable_pab():
+            broadcast_cross, self.cross_count = if_broadcast_cross(int(timestep[0]), self.cross_count)
+
+        if enable_pab() and broadcast_cross:
+            attn_output = self.last_cross
+            context_attn_output = self.context_attn_output
+            # print(f'FluxTransformerBlock | skip t={int(timestep[0])}')
+        else:
+            attn_output, context_attn_output = self.attn(
+                hidden_states=norm_hidden_states,
+                encoder_hidden_states=norm_encoder_hidden_states,
+                image_rotary_emb=image_rotary_emb,
+                **joint_attention_kwargs,
+            )
+            if enable_pab():
+                self.last_cross = attn_output
+                self.context_attn_output = context_attn_output
+                
+        # Process attention outputs for the `hidden_states`.
+        attn_output = gate_msa.unsqueeze(1) * attn_output
+        hidden_states = hidden_states + attn_output
+        
+        norm_hidden_states = self.norm2(hidden_states)
+        norm_hidden_states = norm_hidden_states * (1 + scale_mlp[:, None]) + shift_mlp[:, None]
+
+        ff_output = self.ff(norm_hidden_states)
+        ff_output = gate_mlp.unsqueeze(1) * ff_output
+
+        hidden_states = hidden_states + ff_output
+
+        # Process attention outputs for the `encoder_hidden_states`.
+
+        context_attn_output = c_gate_msa.unsqueeze(1) * context_attn_output
+        encoder_hidden_states = encoder_hidden_states + context_attn_output
+
+        norm_encoder_hidden_states = self.norm2_context(encoder_hidden_states)
+        norm_encoder_hidden_states = norm_encoder_hidden_states * (1 + c_scale_mlp[:, None]) + c_shift_mlp[:, None]
+
+        context_ff_output = self.ff_context(norm_encoder_hidden_states)
+        encoder_hidden_states = encoder_hidden_states + c_gate_mlp.unsqueeze(1) * context_ff_output
+        if encoder_hidden_states.dtype == torch.float16:
+            encoder_hidden_states = encoder_hidden_states.clip(-65504, 65504)
+
+        return encoder_hidden_states, hidden_states
+
+
+class FluxTransformer2DModel(ModelMixin, ConfigMixin, PeftAdapterMixin, FromOriginalModelMixin):
+    """
+    The Transformer model introduced in Flux.
+
+    Reference: https://blackforestlabs.ai/announcing-black-forest-labs/
+
+    Parameters:
+        patch_size (`int`): Patch size to turn the input data into small patches.
+        in_channels (`int`, *optional*, defaults to 16): The number of channels in the input.
+        num_layers (`int`, *optional*, defaults to 18): The number of layers of MMDiT blocks to use.
+        num_single_layers (`int`, *optional*, defaults to 18): The number of layers of single DiT blocks to use.
+        attention_head_dim (`int`, *optional*, defaults to 64): The number of channels in each head.
+        num_attention_heads (`int`, *optional*, defaults to 18): The number of heads to use for multi-head attention.
+        joint_attention_dim (`int`, *optional*): The number of `encoder_hidden_states` dimensions to use.
+        pooled_projection_dim (`int`): Number of dimensions to use when projecting the `pooled_projections`.
+        guidance_embeds (`bool`, defaults to False): Whether to use guidance embeddings.
+    """
+
+    _supports_gradient_checkpointing = True
+    _no_split_modules = ["FluxTransformerBlock", "FluxSingleTransformerBlock"]
+
+    @register_to_config
+    def __init__(
+        self,
+        patch_size: int = 1,
+        in_channels: int = 64,
+        num_layers: int = 19,
+        num_single_layers: int = 38,
+        attention_head_dim: int = 128,
+        num_attention_heads: int = 24,
+        joint_attention_dim: int = 4096,
+        pooled_projection_dim: int = 768,
+        guidance_embeds: bool = False,
+        axes_dims_rope: Tuple[int] = (16, 56, 56),
+    ):
+        super().__init__()
+        self.out_channels = in_channels
+        self.inner_dim = self.config.num_attention_heads * self.config.attention_head_dim
+
+        self.pos_embed = FluxPosEmbed(theta=10000, axes_dim=axes_dims_rope)
+
+        text_time_guidance_cls = (
+            CombinedTimestepGuidanceTextProjEmbeddings if guidance_embeds else CombinedTimestepTextProjEmbeddings
+        )
+        self.time_text_embed = text_time_guidance_cls(
+            embedding_dim=self.inner_dim, pooled_projection_dim=self.config.pooled_projection_dim
+        )
+
+        self.context_embedder = nn.Linear(self.config.joint_attention_dim, self.inner_dim)
+        self.x_embedder = torch.nn.Linear(self.config.in_channels, self.inner_dim)
+
+        self.transformer_blocks = nn.ModuleList(
+            [
+                FluxTransformerBlock(
+                    dim=self.inner_dim,
+                    num_attention_heads=self.config.num_attention_heads,
+                    attention_head_dim=self.config.attention_head_dim,
+                )
+                for i in range(self.config.num_layers)
+            ]
+        )
+
+        self.single_transformer_blocks = nn.ModuleList(
+            [
+                FluxSingleTransformerBlock(
+                    dim=self.inner_dim,
+                    num_attention_heads=self.config.num_attention_heads,
+                    attention_head_dim=self.config.attention_head_dim,
+                )
+                for i in range(self.config.num_single_layers)
+            ]
+        )
+
+        self.norm_out = AdaLayerNormContinuous(self.inner_dim, self.inner_dim, elementwise_affine=False, eps=1e-6)
+        self.proj_out = nn.Linear(self.inner_dim, patch_size * patch_size * self.out_channels, bias=True)
+
+        self.gradient_checkpointing = False
+
+        # parallel
+        self.parallel_manager: ParallelManager = None
+
+
+    def enable_parallel(self, dp_size, sp_size, enable_cp):
+        # update cfg parallel
+        if enable_cp and sp_size % 2 == 0:
+            sp_size = sp_size // 2
+            cp_size = 2
+        else:
+            cp_size = 1
+
+        self.parallel_manager = ParallelManager(dp_size, cp_size, sp_size)
+
+        for _, module in self.named_modules():
+            if hasattr(module, "parallel_manager"):
+                module.parallel_manager = self.parallel_manager
+
+    @property
+    # Copied from diffusers.models.unets.unet_2d_condition.UNet2DConditionModel.attn_processors
+    def attn_processors(self) -> Dict[str, AttentionProcessor]:
+        r"""
+        Returns:
+            `dict` of attention processors: A dictionary containing all attention processors used in the model with
+            indexed by its weight name.
+        """
+        # set recursively
+        processors = {}
+
+        def fn_recursive_add_processors(name: str, module: torch.nn.Module, processors: Dict[str, AttentionProcessor]):
+            if hasattr(module, "get_processor"):
+                processors[f"{name}.processor"] = module.get_processor()
+
+            for sub_name, child in module.named_children():
+                fn_recursive_add_processors(f"{name}.{sub_name}", child, processors)
+
+            return processors
+
+        for name, module in self.named_children():
+            fn_recursive_add_processors(name, module, processors)
+
+        return processors
+
+    # Copied from diffusers.models.unets.unet_2d_condition.UNet2DConditionModel.set_attn_processor
+    def set_attn_processor(self, processor: Union[AttentionProcessor, Dict[str, AttentionProcessor]]):
+        r"""
+        Sets the attention processor to use to compute attention.
+
+        Parameters:
+            processor (`dict` of `AttentionProcessor` or only `AttentionProcessor`):
+                The instantiated processor class or a dictionary of processor classes that will be set as the processor
+                for **all** `Attention` layers.
+
+                If `processor` is a dict, the key needs to define the path to the corresponding cross attention
+                processor. This is strongly recommended when setting trainable attention processors.
+
+        """
+        count = len(self.attn_processors.keys())
+
+        if isinstance(processor, dict) and len(processor) != count:
+            raise ValueError(
+                f"A dict of processors was passed, but the number of processors {len(processor)} does not match the"
+                f" number of attention layers: {count}. Please make sure to pass {count} processor classes."
+            )
+
+        def fn_recursive_attn_processor(name: str, module: torch.nn.Module, processor):
+            if hasattr(module, "set_processor"):
+                if not isinstance(processor, dict):
+                    module.set_processor(processor)
+                else:
+                    module.set_processor(processor.pop(f"{name}.processor"))
+
+            for sub_name, child in module.named_children():
+                fn_recursive_attn_processor(f"{name}.{sub_name}", child, processor)
+
+        for name, module in self.named_children():
+            fn_recursive_attn_processor(name, module, processor)
+
+    # Copied from diffusers.models.unets.unet_2d_condition.UNet2DConditionModel.fuse_qkv_projections with FusedAttnProcessor2_0->FusedFluxAttnProcessor2_0
+    def fuse_qkv_projections(self):
+        """
+        Enables fused QKV projections. For self-attention modules, all projection matrices (i.e., query, key, value)
+        are fused. For cross-attention modules, key and value projection matrices are fused.
+
+        <Tip warning={true}>
+
+        This API is 🧪 experimental.
+
+        </Tip>
+        """
+        self.original_attn_processors = None
+
+        for _, attn_processor in self.attn_processors.items():
+            if "Added" in str(attn_processor.__class__.__name__):
+                raise ValueError("`fuse_qkv_projections()` is not supported for models having added KV projections.")
+
+        self.original_attn_processors = self.attn_processors
+
+        for module in self.modules():
+            if isinstance(module, Attention):
+                module.fuse_projections(fuse=True)
+
+        self.set_attn_processor(FusedFluxAttnProcessor2_0())
+
+    # Copied from diffusers.models.unets.unet_2d_condition.UNet2DConditionModel.unfuse_qkv_projections
+    def unfuse_qkv_projections(self):
+        """Disables the fused QKV projection if enabled.
+
+        <Tip warning={true}>
+
+        This API is 🧪 experimental.
+
+        </Tip>
+
+        """
+        if self.original_attn_processors is not None:
+            self.set_attn_processor(self.original_attn_processors)
+
+    def _set_gradient_checkpointing(self, module, value=False):
+        if hasattr(module, "gradient_checkpointing"):
+            module.gradient_checkpointing = value
+
+    def forward(
+        self,
+        hidden_states: torch.Tensor,
+        encoder_hidden_states: torch.Tensor = None,
+        pooled_projections: torch.Tensor = None,
+        timestep: torch.LongTensor = None,
+        img_ids: torch.Tensor = None,
+        txt_ids: torch.Tensor = None,
+        guidance: torch.Tensor = None,
+        joint_attention_kwargs: Optional[Dict[str, Any]] = None,
+        controlnet_block_samples=None,
+        controlnet_single_block_samples=None,
+        return_dict: bool = True,
+        controlnet_blocks_repeat: bool = False,
+    ) -> Union[torch.FloatTensor, Transformer2DModelOutput]:
+        """
+        The [`FluxTransformer2DModel`] forward method.
+
+        Args:
+            hidden_states (`torch.FloatTensor` of shape `(batch size, channel, height, width)`):
+                Input `hidden_states`.
+            encoder_hidden_states (`torch.FloatTensor` of shape `(batch size, sequence_len, embed_dims)`):
+                Conditional embeddings (embeddings computed from the input conditions such as prompts) to use.
+            pooled_projections (`torch.FloatTensor` of shape `(batch_size, projection_dim)`): Embeddings projected
+                from the embeddings of input conditions.
+            timestep ( `torch.LongTensor`):
+                Used to indicate denoising step.
+            block_controlnet_hidden_states: (`list` of `torch.Tensor`):
+                A list of tensors that if specified are added to the residuals of transformer blocks.
+            joint_attention_kwargs (`dict`, *optional*):
+                A kwargs dictionary that if specified is passed along to the `AttentionProcessor` as defined under
+                `self.processor` in
+                [diffusers.models.attention_processor](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py).
+            return_dict (`bool`, *optional*, defaults to `True`):
+                Whether or not to return a [`~models.transformer_2d.Transformer2DModelOutput`] instead of a plain
+                tuple.
+
+        Returns:
+            If `return_dict` is True, an [`~models.transformer_2d.Transformer2DModelOutput`] is returned, otherwise a
+            `tuple` where the first element is the sample tensor.
+        """
+        if joint_attention_kwargs is not None:
+            joint_attention_kwargs = joint_attention_kwargs.copy()
+            lora_scale = joint_attention_kwargs.pop("scale", 1.0)
+        else:
+            lora_scale = 1.0
+
+        if USE_PEFT_BACKEND:
+            # weight the lora layers by setting `lora_scale` for each PEFT layer
+            scale_lora_layers(self, lora_scale)
+        else:
+            if joint_attention_kwargs is not None and joint_attention_kwargs.get("scale", None) is not None:
+                logger.warning(
+                    "Passing `scale` via `joint_attention_kwargs` when not using the PEFT backend is ineffective."
+                )
+        hidden_states = self.x_embedder(hidden_states)
+
+        timestep = timestep.to(hidden_states.dtype) * 1000
+        if guidance is not None:
+            guidance = guidance.to(hidden_states.dtype) * 1000
+        else:
+            guidance = None
+        temb = (
+            self.time_text_embed(timestep, pooled_projections)
+            if guidance is None
+            else self.time_text_embed(timestep, guidance, pooled_projections)
+        )
+        encoder_hidden_states = self.context_embedder(encoder_hidden_states)
+
+        if txt_ids.ndim == 3:
+            logger.warning(
+                "Passing `txt_ids` 3d torch.Tensor is deprecated."
+                "Please remove the batch dimension and pass it as a 2d torch Tensor"
+            )
+            txt_ids = txt_ids[0]
+        if img_ids.ndim == 3:
+            logger.warning(
+                "Passing `img_ids` 3d torch.Tensor is deprecated."
+                "Please remove the batch dimension and pass it as a 2d torch Tensor"
+            )
+            img_ids = img_ids[0]
+
+        ids = torch.cat((txt_ids, img_ids), dim=0)
+        image_rotary_emb = self.pos_embed(ids)
+
+        # TODO parallel
+        for index_block, block in enumerate(self.transformer_blocks):
+            if self.training and self.gradient_checkpointing:
+
+                def create_custom_forward(module, return_dict=None):
+                    def custom_forward(*inputs):
+                        if return_dict is not None:
+                            return module(*inputs, return_dict=return_dict)
+                        else:
+                            return module(*inputs)
+
+                    return custom_forward
+
+                ckpt_kwargs: Dict[str, Any] = {"use_reentrant": False} if is_torch_version(">=", "1.11.0") else {}
+                encoder_hidden_states, hidden_states = torch.utils.checkpoint.checkpoint(
+                    create_custom_forward(block),
+                    hidden_states,
+                    encoder_hidden_states,
+                    temb,
+                    image_rotary_emb,
+                    **ckpt_kwargs,
+                )
+
+            else:
+                encoder_hidden_states, hidden_states = block(
+                    hidden_states=hidden_states,
+                    encoder_hidden_states=encoder_hidden_states,
+                    temb=temb,
+                    image_rotary_emb=image_rotary_emb,
+                    joint_attention_kwargs=joint_attention_kwargs,
+                    timestep = timestep
+                )
+
+            # controlnet residual
+            if controlnet_block_samples is not None:
+                interval_control = len(self.transformer_blocks) / len(controlnet_block_samples)
+                interval_control = int(np.ceil(interval_control))
+                # For Xlabs ControlNet.
+                if controlnet_blocks_repeat:
+                    hidden_states = (
+                        hidden_states + controlnet_block_samples[index_block % len(controlnet_block_samples)]
+                    )
+                else:
+                    hidden_states = hidden_states + controlnet_block_samples[index_block // interval_control]
+
+        hidden_states = torch.cat([encoder_hidden_states, hidden_states], dim=1)
+
+        # TODO parallel
+        for index_block, block in enumerate(self.single_transformer_blocks):
+            if self.training and self.gradient_checkpointing:
+
+                def create_custom_forward(module, return_dict=None):
+                    def custom_forward(*inputs):
+                        if return_dict is not None:
+                            return module(*inputs, return_dict=return_dict)
+                        else:
+                            return module(*inputs)
+
+                    return custom_forward
+
+                ckpt_kwargs: Dict[str, Any] = {"use_reentrant": False} if is_torch_version(">=", "1.11.0") else {}
+                hidden_states = torch.utils.checkpoint.checkpoint(
+                    create_custom_forward(block),
+                    hidden_states,
+                    temb,
+                    image_rotary_emb,
+                    **ckpt_kwargs,
+                )
+
+            else:
+                hidden_states = block(
+                    hidden_states=hidden_states,
+                    temb=temb,
+                    image_rotary_emb=image_rotary_emb,
+                    joint_attention_kwargs=joint_attention_kwargs,
+                    timestep = timestep
+                )
+
+            # controlnet residual
+            if controlnet_single_block_samples is not None:
+                interval_control = len(self.single_transformer_blocks) / len(controlnet_single_block_samples)
+                interval_control = int(np.ceil(interval_control))
+                hidden_states[:, encoder_hidden_states.shape[1] :, ...] = (
+                    hidden_states[:, encoder_hidden_states.shape[1] :, ...]
+                    + controlnet_single_block_samples[index_block // interval_control]
+                )
+
+        hidden_states = hidden_states[:, encoder_hidden_states.shape[1] :, ...]
+
+        hidden_states = self.norm_out(hidden_states, temb)
+        output = self.proj_out(hidden_states)
+
+        if USE_PEFT_BACKEND:
+            # remove `lora_scale` from each PEFT layer
+            unscale_lora_layers(self, lora_scale)
+
+        if not return_dict:
+            return (output,)
+
+        return Transformer2DModelOutput(sample=output)
diff --git a/videosys/pipelines/flux/__init__.py b/videosys/pipelines/flux/__init__.py
new file mode 100644
index 00000000..fd0a154e
--- /dev/null
+++ b/videosys/pipelines/flux/__init__.py
@@ -0,0 +1,3 @@
+from .pipeline_flux_pab import FluxConfig, FluxPABConfig, FluxPipeline
+
+__all__ = ["FluxConfig", "FluxPipeline", "FluxPABConfig"]
diff --git a/videosys/pipelines/flux/pipeline_flux_pab.py b/videosys/pipelines/flux/pipeline_flux_pab.py
new file mode 100644
index 00000000..529781c6
--- /dev/null
+++ b/videosys/pipelines/flux/pipeline_flux_pab.py
@@ -0,0 +1,899 @@
+# Adapted from Flux
+
+# This source code is licensed under the license found in the
+# LICENSE file in the root directory of this source tree.
+# --------------------------------------------------------
+# References:
+# Flux:
+# --------------------------------------------------------
+import os
+import inspect
+import re
+from typing import Any, Callable, Dict, List, Optional, Union
+from PIL import Image
+import numpy as np
+import torch
+from diffusers.image_processor import VaeImageProcessor
+from diffusers.loaders import FluxLoraLoaderMixin, TextualInversionLoaderMixin
+from diffusers.models import AutoencoderKL
+from diffusers.schedulers import FlowMatchEulerDiscreteScheduler
+from diffusers.utils import USE_PEFT_BACKEND, replace_example_docstring, scale_lora_layers, unscale_lora_layers
+from diffusers.utils.torch_utils import randn_tensor
+from diffusers.pipelines.flux.pipeline_output import FluxPipelineOutput
+from transformers import CLIPTextModel, CLIPTokenizer, T5EncoderModel, T5TokenizerFast
+from videosys.core.pab_mgr import PABConfig, set_pab_manager, update_steps
+
+from videosys.core.pipeline import VideoSysPipeline
+from videosys.models.transformers.flux_transformer import FluxTransformer2DModel
+from videosys.utils.logging import logger
+from videosys.utils.utils import save_video, set_seed
+
+from diffusers.utils import (
+    USE_PEFT_BACKEND,
+    is_torch_xla_available,
+    logging,
+    replace_example_docstring,
+    scale_lora_layers,
+    unscale_lora_layers,
+)
+
+EXAMPLE_DOC_STRING = """
+    Examples:
+        ```py
+        >>> import torch
+        >>> from diffusers import FluxPipeline
+
+        >>> pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16)
+        >>> pipe.to("cuda")
+        >>> prompt = "A cat holding a sign that says hello world"
+        >>> # Depending on the variant being used, the pipeline call will slightly vary.
+        >>> # Refer to the pipeline documentation for more details.
+        >>> image = pipe(prompt, num_inference_steps=4, guidance_scale=0.0).images[0]
+        >>> image.save("flux.png")
+        ```
+"""
+
+if is_torch_xla_available():
+    import torch_xla.core.xla_model as xm
+
+    XLA_AVAILABLE = True
+else:
+    XLA_AVAILABLE = False
+
+
+def calculate_shift(
+    image_seq_len,
+    base_seq_len: int = 256,
+    max_seq_len: int = 4096,
+    base_shift: float = 0.5,
+    max_shift: float = 1.16,
+):
+    m = (max_shift - base_shift) / (max_seq_len - base_seq_len)
+    b = base_shift - m * base_seq_len
+    mu = image_seq_len * m + b
+    return mu
+
+
+# Copied from diffusers.pipelines.stable_diffusion.pipeline_stable_diffusion.retrieve_timesteps
+def retrieve_timesteps(
+    scheduler,
+    num_inference_steps: Optional[int] = None,
+    device: Optional[Union[str, torch.device]] = None,
+    timesteps: Optional[List[int]] = None,
+    sigmas: Optional[List[float]] = None,
+    **kwargs,
+):
+    r"""
+    Calls the scheduler's `set_timesteps` method and retrieves timesteps from the scheduler after the call. Handles
+    custom timesteps. Any kwargs will be supplied to `scheduler.set_timesteps`.
+
+    Args:
+        scheduler (`SchedulerMixin`):
+            The scheduler to get timesteps from.
+        num_inference_steps (`int`):
+            The number of diffusion steps used when generating samples with a pre-trained model. If used, `timesteps`
+            must be `None`.
+        device (`str` or `torch.device`, *optional*):
+            The device to which the timesteps should be moved to. If `None`, the timesteps are not moved.
+        timesteps (`List[int]`, *optional*):
+            Custom timesteps used to override the timestep spacing strategy of the scheduler. If `timesteps` is passed,
+            `num_inference_steps` and `sigmas` must be `None`.
+        sigmas (`List[float]`, *optional*):
+            Custom sigmas used to override the timestep spacing strategy of the scheduler. If `sigmas` is passed,
+            `num_inference_steps` and `timesteps` must be `None`.
+
+    Returns:
+        `Tuple[torch.Tensor, int]`: A tuple where the first element is the timestep schedule from the scheduler and the
+        second element is the number of inference steps.
+    """
+    if timesteps is not None and sigmas is not None:
+        raise ValueError("Only one of `timesteps` or `sigmas` can be passed. Please choose one to set custom values")
+    if timesteps is not None:
+        accepts_timesteps = "timesteps" in set(inspect.signature(scheduler.set_timesteps).parameters.keys())
+        if not accepts_timesteps:
+            raise ValueError(
+                f"The current scheduler class {scheduler.__class__}'s `set_timesteps` does not support custom"
+                f" timestep schedules. Please check whether you are using the correct scheduler."
+            )
+        scheduler.set_timesteps(timesteps=timesteps, device=device, **kwargs)
+        timesteps = scheduler.timesteps
+        num_inference_steps = len(timesteps)
+    elif sigmas is not None:
+        accept_sigmas = "sigmas" in set(inspect.signature(scheduler.set_timesteps).parameters.keys())
+        if not accept_sigmas:
+            raise ValueError(
+                f"The current scheduler class {scheduler.__class__}'s `set_timesteps` does not support custom"
+                f" sigmas schedules. Please check whether you are using the correct scheduler."
+            )
+        scheduler.set_timesteps(sigmas=sigmas, device=device, **kwargs)
+        timesteps = scheduler.timesteps
+        num_inference_steps = len(timesteps)
+    else:
+        scheduler.set_timesteps(num_inference_steps, device=device, **kwargs)
+        timesteps = scheduler.timesteps
+    return timesteps, num_inference_steps
+
+
+class FluxPABConfig(PABConfig):
+    def __init__(
+        self,
+        spatial_broadcast: bool = True,
+        spatial_threshold: list = [100, 800],
+        spatial_range: int = 20,
+        
+        temporal_broadcast: bool = False,
+        cross_broadcast: bool = True,
+        
+        cross_threshold: list = [100, 800],
+        cross_range: int = 20,
+        mlp_broadcast: bool = True,
+    ):
+        super().__init__(
+            spatial_broadcast=spatial_broadcast,
+            spatial_threshold=spatial_threshold,
+            spatial_range=spatial_range,
+            temporal_broadcast=temporal_broadcast,
+            cross_broadcast=cross_broadcast,
+            cross_threshold=cross_threshold,
+            cross_range=cross_range,
+            mlp_broadcast=mlp_broadcast,
+        )
+
+
+class FluxConfig:
+    def __init__(
+        self,
+        model_path: str = "black-forest-labs/FLUX.1-dev",
+        # ======= distributed =======
+        num_gpus: int = 1,
+        # ======= vae ========
+        enable_vae_temporal_decoder: bool = True,
+        # ======= scheduler ========
+        beta_start: float = 0.0001,
+        beta_end: float = 0.02,
+        beta_schedule: str = "linear",
+        variance_type: str = "learned_range",
+        # ======= memory =======
+        cpu_offload: bool = False,
+        # ======= pab ========
+        enable_pab: bool = False,
+        pab_config: PABConfig = FluxPABConfig(),
+    ):
+        self.model_path = model_path
+        self.steps = 50
+        
+        # self.pipeline_cls = FluxPipeline
+        # ======= distributed =======
+        self.num_gpus = num_gpus
+        # ======= vae ========
+        self.enable_vae_temporal_decoder = enable_vae_temporal_decoder
+        # ======= memory ========
+        self.cpu_offload = cpu_offload
+        # ======= scheduler ========
+        self.beta_start = beta_start
+        self.beta_end = beta_end
+        self.beta_schedule = beta_schedule
+        self.variance_type = variance_type
+        # ======= pab ========
+        self.enable_pab = enable_pab
+        self.pab_config = pab_config
+
+
+class FluxPipeline(VideoSysPipeline):
+    r"""
+    Pipeline for text-to-image generation using PixArt-Alpha.
+
+    This model inherits from [`DiffusionPipeline`]. Check the superclass documentation for the generic methods the
+    library implements for all the pipelines (such as downloading or saving, running on a particular device, etc.)
+
+    Args:
+        vae ([`AutoencoderKL`]):
+            Variational Auto-Encoder (VAE) Model to encode and decode images to and from latent representations.
+        text_encoder ([`T5EncoderModel`]):
+            Frozen text-encoder. PixArt-Alpha uses
+            [T5](https://huggingface.co/docs/transformers/model_doc/t5#transformers.T5EncoderModel), specifically the
+            [t5-v1_1-xxl](https://huggingface.co/PixArt-alpha/PixArt-alpha/tree/main/t5-v1_1-xxl) variant.
+        tokenizer (`T5Tokenizer`):
+            Tokenizer of class
+            [T5Tokenizer](https://huggingface.co/docs/transformers/model_doc/t5#transformers.T5Tokenizer).
+        transformer ([`Transformer2DModel`]):
+            A text conditioned `Transformer2DModel` to denoise the encoded image latents.
+        scheduler ([`SchedulerMixin`]):
+            A scheduler to be used in combination with `transformer` to denoise the encoded image latents.
+    """
+    bad_punct_regex = re.compile(
+        r"[" + "#®•©™&@·º½¾¿¡§~" + "\)" + "\(" + "\]" + "\[" + "\}" + "\{" + "\|" + "\\" + "\/" + "\*" + r"]{1,}"
+    )  # noqa
+
+    # _optional_components = ["tokenizer", "text_encoder", "vae", "transformer", "scheduler"]
+    # model_cpu_offload_seq = "text_encoder->transformer->vae"
+    _callback_tensor_inputs = ["latents"]
+
+    def __init__(
+        self,
+        # model components
+        scheduler: Optional[FlowMatchEulerDiscreteScheduler] = None,
+        vae: Optional[AutoencoderKL] = None,
+        text_encoder: Optional[CLIPTextModel] = None,
+        tokenizer: Optional[CLIPTokenizer] = None,
+        text_encoder_2: Optional[T5EncoderModel] = None,
+        tokenizer_2: Optional[T5TokenizerFast] = None,
+        transformer: Optional[FluxTransformer2DModel] = None,
+        config: FluxConfig = FluxConfig(),
+        device: torch.device = torch.device("cuda:0"),
+        dtype: torch.dtype = torch.float16,
+    ):
+        super().__init__()
+        self._config = config
+        self._device = device
+        # print("Loading FluxPipeline modules")
+
+        # init
+        if tokenizer is None:
+            tokenizer = CLIPTokenizer.from_pretrained(config.model_path, subfolder="tokenizer", torch_dtype=dtype)
+
+        if tokenizer_2 is None:
+            tokenizer_2 = T5TokenizerFast.from_pretrained(config.model_path, subfolder="tokenizer_2", torch_dtype=dtype)
+
+        if text_encoder is None:
+            text_encoder = CLIPTextModel.from_pretrained(config.model_path, subfolder="text_encoder", torch_dtype=dtype)
+
+        if text_encoder_2 is None:
+            text_encoder_2 = T5EncoderModel.from_pretrained(
+                config.model_path, subfolder="text_encoder_2", torch_dtype=dtype
+            )
+
+        if vae is None:
+            vae = AutoencoderKL.from_pretrained(config.model_path, subfolder="vae", torch_dtype=dtype)
+
+        if transformer is None:
+            transformer = FluxTransformer2DModel.from_pretrained(
+                config.model_path, subfolder="transformer", torch_dtype=dtype
+            )
+
+        if scheduler is None:
+            scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(config.model_path, subfolder="scheduler")
+
+        # cpu offload
+        if config.cpu_offload:
+            self.enable_model_cpu_offload()
+        else:
+            self.set_eval_and_device(
+                self._device, vae, transformer, text_encoder, text_encoder_2
+            )
+            
+        # pab
+        if config.enable_pab:
+            set_pab_manager(config.pab_config)
+
+        # print("Registering FluxPipeline modules")
+        self.register_modules(
+            vae=vae,
+            text_encoder=text_encoder,
+            text_encoder_2=text_encoder_2,
+            tokenizer=tokenizer,
+            tokenizer_2=tokenizer_2,
+            transformer=transformer,
+            scheduler=scheduler,
+        )
+
+        # model setting
+        self.vae_scale_factor = (
+            2 ** (len(self.vae.config.block_out_channels)) if hasattr(self, "vae") and self.vae is not None else 16
+        )
+        self.image_processor = VaeImageProcessor(vae_scale_factor=self.vae_scale_factor)
+        self.tokenizer_max_length = (
+            self.tokenizer.model_max_length if hasattr(self, "tokenizer") and self.tokenizer is not None else 77
+        )
+        self.default_sample_size = 64
+
+        # TODO parallel 
+        # self._set_parallel()
+
+    def _get_t5_prompt_embeds(
+        self,
+        prompt: Union[str, List[str]] = None,
+        num_images_per_prompt: int = 1,
+        max_sequence_length: int = 512,
+        device: Optional[torch.device] = None,
+        dtype: Optional[torch.dtype] = None,
+    ):
+        device = device or self._device
+        dtype = dtype or self.text_encoder.dtype
+
+        prompt = [prompt] if isinstance(prompt, str) else prompt
+        batch_size = len(prompt)
+
+        if isinstance(self, TextualInversionLoaderMixin):
+            prompt = self.maybe_convert_prompt(prompt, self.tokenizer_2)
+
+        text_inputs = self.tokenizer_2(
+            prompt,
+            padding="max_length",
+            max_length=max_sequence_length,
+            truncation=True,
+            return_length=False,
+            return_overflowing_tokens=False,
+            return_tensors="pt",
+        )
+        text_input_ids = text_inputs.input_ids
+        untruncated_ids = self.tokenizer_2(prompt, padding="longest", return_tensors="pt").input_ids
+
+        if untruncated_ids.shape[-1] >= text_input_ids.shape[-1] and not torch.equal(text_input_ids, untruncated_ids):
+            removed_text = self.tokenizer_2.batch_decode(untruncated_ids[:, self.tokenizer_max_length - 1 : -1])
+            logger.warning(
+                "The following part of your input was truncated because `max_sequence_length` is set to "
+                f" {max_sequence_length} tokens: {removed_text}"
+            )
+
+        prompt_embeds = self.text_encoder_2(text_input_ids.to(device), output_hidden_states=False)[0]
+
+        dtype = self.text_encoder_2.dtype
+        prompt_embeds = prompt_embeds.to(dtype=dtype, device=device)
+
+        _, seq_len, _ = prompt_embeds.shape
+
+        # duplicate text embeddings and attention mask for each generation per prompt, using mps friendly method
+        prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt, 1)
+        prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, seq_len, -1)
+
+        return prompt_embeds
+
+    def _get_clip_prompt_embeds(
+        self,
+        prompt: Union[str, List[str]],
+        num_images_per_prompt: int = 1,
+        device: Optional[torch.device] = None,
+    ):
+        device = device or self._device
+
+        prompt = [prompt] if isinstance(prompt, str) else prompt
+        batch_size = len(prompt)
+
+        if isinstance(self, TextualInversionLoaderMixin):
+            prompt = self.maybe_convert_prompt(prompt, self.tokenizer)
+
+        text_inputs = self.tokenizer(
+            prompt,
+            padding="max_length",
+            max_length=self.tokenizer_max_length,
+            truncation=True,
+            return_overflowing_tokens=False,
+            return_length=False,
+            return_tensors="pt",
+        )
+
+        text_input_ids = text_inputs.input_ids
+        untruncated_ids = self.tokenizer(prompt, padding="longest", return_tensors="pt").input_ids
+        if untruncated_ids.shape[-1] >= text_input_ids.shape[-1] and not torch.equal(text_input_ids, untruncated_ids):
+            removed_text = self.tokenizer.batch_decode(untruncated_ids[:, self.tokenizer_max_length - 1 : -1])
+            logger.warning(
+                "The following part of your input was truncated because CLIP can only handle sequences up to"
+                f" {self.tokenizer_max_length} tokens: {removed_text}"
+            )
+        
+        prompt_embeds = self.text_encoder(text_input_ids.to(device), output_hidden_states=False)
+
+        # Use pooled output of CLIPTextModel
+        prompt_embeds = prompt_embeds.pooler_output
+        prompt_embeds = prompt_embeds.to(dtype=self.text_encoder.dtype, device=device)
+
+        # duplicate text embeddings for each generation per prompt, using mps friendly method
+        prompt_embeds = prompt_embeds.repeat(1, num_images_per_prompt)
+        prompt_embeds = prompt_embeds.view(batch_size * num_images_per_prompt, -1)
+
+        return prompt_embeds
+
+    def encode_prompt(
+        self,
+        prompt: Union[str, List[str]],
+        prompt_2: Union[str, List[str]],
+        device: Optional[torch.device] = None,
+        num_images_per_prompt: int = 1,
+        prompt_embeds: Optional[torch.FloatTensor] = None,
+        pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
+        max_sequence_length: int = 512,
+        lora_scale: Optional[float] = None,
+    ):
+        r"""
+
+        Args:
+            prompt (`str` or `List[str]`, *optional*):
+                prompt to be encoded
+            prompt_2 (`str` or `List[str]`, *optional*):
+                The prompt or prompts to be sent to the `tokenizer_2` and `text_encoder_2`. If not defined, `prompt` is
+                used in all text-encoders
+            device: (`torch.device`):
+                torch device
+            num_images_per_prompt (`int`):
+                number of images that should be generated per prompt
+            prompt_embeds (`torch.FloatTensor`, *optional*):
+                Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
+                provided, text embeddings will be generated from `prompt` input argument.
+            pooled_prompt_embeds (`torch.FloatTensor`, *optional*):
+                Pre-generated pooled text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting.
+                If not provided, pooled text embeddings will be generated from `prompt` input argument.
+            lora_scale (`float`, *optional*):
+                A lora scale that will be applied to all LoRA layers of the text encoder if LoRA layers are loaded.
+        """
+        device = device or self._device
+
+        # set lora scale so that monkey patched LoRA
+        # function of text encoder can correctly access it
+        if lora_scale is not None and isinstance(self, FluxLoraLoaderMixin):
+            self._lora_scale = lora_scale
+
+            # dynamically adjust the LoRA scale
+            if self.text_encoder is not None and USE_PEFT_BACKEND:
+                scale_lora_layers(self.text_encoder, lora_scale)
+            if self.text_encoder_2 is not None and USE_PEFT_BACKEND:
+                scale_lora_layers(self.text_encoder_2, lora_scale)
+
+        prompt = [prompt] if isinstance(prompt, str) else prompt
+
+        if prompt_embeds is None:
+            prompt_2 = prompt_2 or prompt
+            prompt_2 = [prompt_2] if isinstance(prompt_2, str) else prompt_2
+
+            # We only use the pooled prompt output from the CLIPTextModel
+            pooled_prompt_embeds = self._get_clip_prompt_embeds(
+                prompt=prompt,
+                device=device,
+                num_images_per_prompt=num_images_per_prompt,
+            )
+            prompt_embeds = self._get_t5_prompt_embeds(
+                prompt=prompt_2,
+                num_images_per_prompt=num_images_per_prompt,
+                max_sequence_length=max_sequence_length,
+                device=device,
+            )
+
+        if self.text_encoder is not None:
+            if isinstance(self, FluxLoraLoaderMixin) and USE_PEFT_BACKEND:
+                # Retrieve the original scale by scaling back the LoRA layers
+                unscale_lora_layers(self.text_encoder, lora_scale)
+
+        if self.text_encoder_2 is not None:
+            if isinstance(self, FluxLoraLoaderMixin) and USE_PEFT_BACKEND:
+                # Retrieve the original scale by scaling back the LoRA layers
+                unscale_lora_layers(self.text_encoder_2, lora_scale)
+
+        dtype = self.text_encoder.dtype if self.text_encoder is not None else self.transformer.dtype
+        text_ids = torch.zeros(prompt_embeds.shape[1], 3).to(device=device, dtype=dtype)
+
+        return prompt_embeds, pooled_prompt_embeds, text_ids
+
+    def check_inputs(
+        self,
+        prompt,
+        prompt_2,
+        height,
+        width,
+        prompt_embeds=None,
+        pooled_prompt_embeds=None,
+        callback_on_step_end_tensor_inputs=None,
+        max_sequence_length=None,
+    ):
+        if height % 8 != 0 or width % 8 != 0:
+            raise ValueError(f"`height` and `width` have to be divisible by 8 but are {height} and {width}.")
+        # print(f"Checking FluxPipeline inputs callback_on_step_end_tensor_inputs: {callback_on_step_end_tensor_inputs}")
+        if callback_on_step_end_tensor_inputs is not None and not all(
+            k in self._callback_tensor_inputs for k in callback_on_step_end_tensor_inputs
+        ):
+            raise ValueError(
+                f"`callback_on_step_end_tensor_inputs` has to be in {self._callback_tensor_inputs}, but found {[k for k in callback_on_step_end_tensor_inputs if k not in self._callback_tensor_inputs]}"
+            )
+
+        if prompt is not None and prompt_embeds is not None:
+            raise ValueError(
+                f"Cannot forward both `prompt`: {prompt} and `prompt_embeds`: {prompt_embeds}. Please make sure to"
+                " only forward one of the two."
+            )
+        elif prompt_2 is not None and prompt_embeds is not None:
+            raise ValueError(
+                f"Cannot forward both `prompt_2`: {prompt_2} and `prompt_embeds`: {prompt_embeds}. Please make sure to"
+                " only forward one of the two."
+            )
+        elif prompt is None and prompt_embeds is None:
+            raise ValueError(
+                "Provide either `prompt` or `prompt_embeds`. Cannot leave both `prompt` and `prompt_embeds` undefined."
+            )
+        elif prompt is not None and (not isinstance(prompt, str) and not isinstance(prompt, list)):
+            raise ValueError(f"`prompt` has to be of type `str` or `list` but is {type(prompt)}")
+        elif prompt_2 is not None and (not isinstance(prompt_2, str) and not isinstance(prompt_2, list)):
+            raise ValueError(f"`prompt_2` has to be of type `str` or `list` but is {type(prompt_2)}")
+
+        if prompt_embeds is not None and pooled_prompt_embeds is None:
+            raise ValueError(
+                "If `prompt_embeds` are provided, `pooled_prompt_embeds` also have to be passed. Make sure to generate `pooled_prompt_embeds` from the same text encoder that was used to generate `prompt_embeds`."
+            )
+
+        if max_sequence_length is not None and max_sequence_length > 512:
+            raise ValueError(f"`max_sequence_length` cannot be greater than 512 but is {max_sequence_length}")
+
+    @staticmethod
+    def _prepare_latent_image_ids(batch_size, height, width, device, dtype):
+        latent_image_ids = torch.zeros(height // 2, width // 2, 3)
+        latent_image_ids[..., 1] = latent_image_ids[..., 1] + torch.arange(height // 2)[:, None]
+        latent_image_ids[..., 2] = latent_image_ids[..., 2] + torch.arange(width // 2)[None, :]
+
+        latent_image_id_height, latent_image_id_width, latent_image_id_channels = latent_image_ids.shape
+
+        latent_image_ids = latent_image_ids.reshape(
+            latent_image_id_height * latent_image_id_width, latent_image_id_channels
+        )
+
+        return latent_image_ids.to(device=device, dtype=dtype)
+
+    @staticmethod
+    def _pack_latents(latents, batch_size, num_channels_latents, height, width):
+        latents = latents.view(batch_size, num_channels_latents, height // 2, 2, width // 2, 2)
+        latents = latents.permute(0, 2, 4, 1, 3, 5)
+        latents = latents.reshape(batch_size, (height // 2) * (width // 2), num_channels_latents * 4)
+
+        return latents
+
+    @staticmethod
+    def _unpack_latents(latents, height, width, vae_scale_factor):
+        batch_size, num_patches, channels = latents.shape
+
+        height = height // vae_scale_factor
+        width = width // vae_scale_factor
+
+        latents = latents.view(batch_size, height, width, channels // 4, 2, 2)
+        latents = latents.permute(0, 3, 1, 4, 2, 5)
+
+        latents = latents.reshape(batch_size, channels // (2 * 2), height * 2, width * 2)
+
+        return latents
+
+    def enable_vae_slicing(self):
+        r"""
+        Enable sliced VAE decoding. When this option is enabled, the VAE will split the input tensor in slices to
+        compute decoding in several steps. This is useful to save some memory and allow larger batch sizes.
+        """
+        self.vae.enable_slicing()
+
+    def disable_vae_slicing(self):
+        r"""
+        Disable sliced VAE decoding. If `enable_vae_slicing` was previously enabled, this method will go back to
+        computing decoding in one step.
+        """
+        self.vae.disable_slicing()
+
+    def enable_vae_tiling(self):
+        r"""
+        Enable tiled VAE decoding. When this option is enabled, the VAE will split the input tensor into tiles to
+        compute decoding and encoding in several steps. This is useful for saving a large amount of memory and to allow
+        processing larger images.
+        """
+        self.vae.enable_tiling()
+
+    def disable_vae_tiling(self):
+        r"""
+        Disable tiled VAE decoding. If `enable_vae_tiling` was previously enabled, this method will go back to
+        computing decoding in one step.
+        """
+        self.vae.disable_tiling()
+
+    def prepare_latents(
+        self,
+        batch_size,
+        num_channels_latents,
+        height,
+        width,
+        dtype,
+        device,
+        generator,
+        latents=None,
+    ):
+        height = 2 * (int(height) // self.vae_scale_factor)
+        width = 2 * (int(width) // self.vae_scale_factor)
+
+        shape = (batch_size, num_channels_latents, height, width)
+
+        if latents is not None:
+            latent_image_ids = self._prepare_latent_image_ids(batch_size, height, width, device, dtype)
+            return latents.to(device=device, dtype=dtype), latent_image_ids
+
+        if isinstance(generator, list) and len(generator) != batch_size:
+            raise ValueError(
+                f"You have passed a list of generators of length {len(generator)}, but requested an effective batch"
+                f" size of {batch_size}. Make sure the batch size matches the length of the generators."
+            )
+
+        latents = randn_tensor(shape, generator=generator, device=device, dtype=dtype)
+        latents = self._pack_latents(latents, batch_size, num_channels_latents, height, width)
+
+        latent_image_ids = self._prepare_latent_image_ids(batch_size, height, width, device, dtype)
+
+        return latents, latent_image_ids
+
+    @property
+    def guidance_scale(self):
+        return self._guidance_scale
+
+    @property
+    def joint_attention_kwargs(self):
+        return self._joint_attention_kwargs
+
+    @property
+    def num_timesteps(self):
+        return self._num_timesteps
+
+    @property
+    def interrupt(self):
+        return self._interrupt
+
+    @torch.no_grad()
+    @replace_example_docstring(EXAMPLE_DOC_STRING)
+    def __call__(
+        self,
+        prompt: Union[str, List[str]] = None,
+        prompt_2: Optional[Union[str, List[str]]] = None,
+        height: Optional[int] = None,
+        width: Optional[int] = None,
+        num_inference_steps: int = 28,
+        timesteps: List[int] = None,
+        guidance_scale: float = 3.5,
+        num_images_per_prompt: Optional[int] = 1,
+        generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
+        latents: Optional[torch.FloatTensor] = None,
+        prompt_embeds: Optional[torch.FloatTensor] = None,
+        pooled_prompt_embeds: Optional[torch.FloatTensor] = None,
+        output_type: Optional[str] = "pil",
+        return_dict: bool = True,
+        joint_attention_kwargs: Optional[Dict[str, Any]] = None,
+        callback_on_step_end: Optional[Callable[[int, int, Dict], None]] = None,
+        callback_on_step_end_tensor_inputs: List[str] = ["latents"],
+        max_sequence_length: int = 512,
+    ):
+        r"""
+        Function invoked when calling the pipeline for generation.
+
+        Args:
+            prompt (`str` or `List[str]`, *optional*):
+                The prompt or prompts to guide the image generation. If not defined, one has to pass `prompt_embeds`.
+                instead.
+            prompt_2 (`str` or `List[str]`, *optional*):
+                The prompt or prompts to be sent to `tokenizer_2` and `text_encoder_2`. If not defined, `prompt` is
+                will be used instead
+            height (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
+                The height in pixels of the generated image. This is set to 1024 by default for the best results.
+            width (`int`, *optional*, defaults to self.unet.config.sample_size * self.vae_scale_factor):
+                The width in pixels of the generated image. This is set to 1024 by default for the best results.
+            num_inference_steps (`int`, *optional*, defaults to 50):
+                The number of denoising steps. More denoising steps usually lead to a higher quality image at the
+                expense of slower inference.
+            timesteps (`List[int]`, *optional*):
+                Custom timesteps to use for the denoising process with schedulers which support a `timesteps` argument
+                in their `set_timesteps` method. If not defined, the default behavior when `num_inference_steps` is
+                passed will be used. Must be in descending order.
+            guidance_scale (`float`, *optional*, defaults to 7.0):
+                Guidance scale as defined in [Classifier-Free Diffusion Guidance](https://arxiv.org/abs/2207.12598).
+                `guidance_scale` is defined as `w` of equation 2. of [Imagen
+                Paper](https://arxiv.org/pdf/2205.11487.pdf). Guidance scale is enabled by setting `guidance_scale >
+                1`. Higher guidance scale encourages to generate images that are closely linked to the text `prompt`,
+                usually at the expense of lower image quality.
+            num_images_per_prompt (`int`, *optional*, defaults to 1):
+                The number of images to generate per prompt.
+            generator (`torch.Generator` or `List[torch.Generator]`, *optional*):
+                One or a list of [torch generator(s)](https://pytorch.org/docs/stable/generated/torch.Generator.html)
+                to make generation deterministic.
+            latents (`torch.FloatTensor`, *optional*):
+                Pre-generated noisy latents, sampled from a Gaussian distribution, to be used as inputs for image
+                generation. Can be used to tweak the same generation with different prompts. If not provided, a latents
+                tensor will ge generated by sampling using the supplied random `generator`.
+            prompt_embeds (`torch.FloatTensor`, *optional*):
+                Pre-generated text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting. If not
+                provided, text embeddings will be generated from `prompt` input argument.
+            pooled_prompt_embeds (`torch.FloatTensor`, *optional*):
+                Pre-generated pooled text embeddings. Can be used to easily tweak text inputs, *e.g.* prompt weighting.
+                If not provided, pooled text embeddings will be generated from `prompt` input argument.
+            output_type (`str`, *optional*, defaults to `"pil"`):
+                The output format of the generate image. Choose between
+                [PIL](https://pillow.readthedocs.io/en/stable/): `PIL.Image.Image` or `np.array`.
+            return_dict (`bool`, *optional*, defaults to `True`):
+                Whether or not to return a [`~pipelines.flux.FluxPipelineOutput`] instead of a plain tuple.
+            joint_attention_kwargs (`dict`, *optional*):
+                A kwargs dictionary that if specified is passed along to the `AttentionProcessor` as defined under
+                `self.processor` in
+                [diffusers.models.attention_processor](https://github.com/huggingface/diffusers/blob/main/src/diffusers/models/attention_processor.py).
+            callback_on_step_end (`Callable`, *optional*):
+                A function that calls at the end of each denoising steps during the inference. The function is called
+                with the following arguments: `callback_on_step_end(self: DiffusionPipeline, step: int, timestep: int,
+                callback_kwargs: Dict)`. `callback_kwargs` will include a list of all tensors as specified by
+                `callback_on_step_end_tensor_inputs`.
+            callback_on_step_end_tensor_inputs (`List`, *optional*):
+                The list of tensor inputs for the `callback_on_step_end` function. The tensors specified in the list
+                will be passed as `callback_kwargs` argument. You will only be able to include variables listed in the
+                `._callback_tensor_inputs` attribute of your pipeline class.
+            max_sequence_length (`int` defaults to 512): Maximum sequence length to use with the `prompt`.
+
+        Examples:
+
+        Returns:
+            [`~pipelines.flux.FluxPipelineOutput`] or `tuple`: [`~pipelines.flux.FluxPipelineOutput`] if `return_dict`
+            is True, otherwise a `tuple`. When returning a tuple, the first element is a list with the generated
+            images.
+        """
+        # print("Running FluxPipeline Generation")
+        height = height or self.default_sample_size * self.vae_scale_factor
+        width = width or self.default_sample_size * self.vae_scale_factor
+
+        update_steps(num_inference_steps)
+
+        # 1. Check inputs. Raise error if not correct
+        self.check_inputs(
+            prompt,
+            prompt_2,
+            height,
+            width,
+            prompt_embeds=prompt_embeds,
+            pooled_prompt_embeds=pooled_prompt_embeds,
+            callback_on_step_end_tensor_inputs=callback_on_step_end_tensor_inputs,
+            max_sequence_length=max_sequence_length,
+        )
+
+        self._guidance_scale = guidance_scale
+        self._joint_attention_kwargs = joint_attention_kwargs
+        self._interrupt = False
+
+        # 2. Define call parameters
+        if prompt is not None and isinstance(prompt, str):
+            batch_size = 1
+        elif prompt is not None and isinstance(prompt, list):
+            batch_size = len(prompt)
+        else:
+            batch_size = prompt_embeds.shape[0]
+
+        device = self._device
+
+        lora_scale = self.joint_attention_kwargs.get("scale", None) if self.joint_attention_kwargs is not None else None
+        (
+            prompt_embeds,
+            pooled_prompt_embeds,
+            text_ids,
+        ) = self.encode_prompt(
+            prompt=prompt,
+            prompt_2=prompt_2,
+            prompt_embeds=prompt_embeds,
+            pooled_prompt_embeds=pooled_prompt_embeds,
+            device=device,
+            num_images_per_prompt=num_images_per_prompt,
+            max_sequence_length=max_sequence_length,
+            lora_scale=lora_scale,
+        )
+
+        # 4. Prepare latent variables
+        num_channels_latents = self.transformer.config.in_channels // 4
+        latents, latent_image_ids = self.prepare_latents(
+            batch_size * num_images_per_prompt,
+            num_channels_latents,
+            height,
+            width,
+            prompt_embeds.dtype,
+            device,
+            generator,
+            latents,
+        )
+
+        # 5. Prepare timesteps
+        sigmas = np.linspace(1.0, 1 / num_inference_steps, num_inference_steps)
+        image_seq_len = latents.shape[1]
+        mu = calculate_shift(
+            image_seq_len,
+            self.scheduler.config.base_image_seq_len,
+            self.scheduler.config.max_image_seq_len,
+            self.scheduler.config.base_shift,
+            self.scheduler.config.max_shift,
+        )
+        timesteps, num_inference_steps = retrieve_timesteps(
+            self.scheduler,
+            num_inference_steps,
+            device,
+            timesteps,
+            sigmas,
+            mu=mu,
+        )
+        num_warmup_steps = max(len(timesteps) - num_inference_steps * self.scheduler.order, 0)
+        self._num_timesteps = len(timesteps)
+
+        # handle guidance
+        if self.transformer.config.guidance_embeds:
+            guidance = torch.full([1], guidance_scale, device=device, dtype=torch.float32)
+            guidance = guidance.expand(latents.shape[0])
+        else:
+            guidance = None
+        
+        # 6. Denoising loop
+        # print(f'timesteps {timesteps}')
+        with self.progress_bar(total=num_inference_steps) as progress_bar:
+            for i, t in enumerate(timesteps):
+                if self.interrupt:
+                    continue
+
+                # broadcast to batch dimension in a way that's compatible with ONNX/Core ML
+                timestep = t.expand(latents.shape[0]).to(latents.dtype)
+
+                noise_pred = self.transformer(
+                    hidden_states=latents,
+                    timestep=timestep / 1000,
+                    guidance=guidance,
+                    pooled_projections=pooled_prompt_embeds,
+                    encoder_hidden_states=prompt_embeds,
+                    txt_ids=text_ids,
+                    img_ids=latent_image_ids,
+                    joint_attention_kwargs=self.joint_attention_kwargs,
+                    return_dict=False,
+                )[0]
+
+                # compute the previous noisy sample x_t -> x_t-1
+                latents_dtype = latents.dtype
+                latents = self.scheduler.step(noise_pred, t, latents, return_dict=False)[0]
+
+                if latents.dtype != latents_dtype:
+                    if torch.backends.mps.is_available():
+                        # some platforms (eg. apple mps) misbehave due to a pytorch bug: https://github.com/pytorch/pytorch/pull/99272
+                        latents = latents.to(latents_dtype)
+
+                if callback_on_step_end is not None:
+                    callback_kwargs = {}
+                    for k in callback_on_step_end_tensor_inputs:
+                        callback_kwargs[k] = locals()[k]
+                    callback_outputs = callback_on_step_end(self, i, t, callback_kwargs)
+
+                    latents = callback_outputs.pop("latents", latents)
+                    prompt_embeds = callback_outputs.pop("prompt_embeds", prompt_embeds)
+
+                # call the callback, if provided
+                if i == len(timesteps) - 1 or ((i + 1) > num_warmup_steps and (i + 1) % self.scheduler.order == 0):
+                    progress_bar.update()
+
+                if XLA_AVAILABLE:
+                    xm.mark_step()
+
+        if output_type == "latent":
+            image = latents
+
+        else:
+            latents = self._unpack_latents(latents, height, width, self.vae_scale_factor)
+            latents = (latents / self.vae.config.scaling_factor) + self.vae.config.shift_factor
+            image = self.vae.decode(latents, return_dict=False)[0]
+            image = self.image_processor.postprocess(image, output_type=output_type)
+
+        # Offload all models
+        self.maybe_free_model_hooks()
+
+        if not return_dict:
+            return (image,)
+
+        return FluxPipelineOutput(images=image)
+
+
+    def save_image(self, image, output_path):
+        if isinstance(image, Image.Image):
+            os.makedirs(os.path.dirname(output_path), exist_ok=True)
+            image.save(output_path)
+        else:
+            raise ValueError("Image must be a PIL Image object.")